Creepso/stremio-shell-ng
1
0
Fork 0
mirror of https://github.com/Stremio/stremio-shell-ng.git synced 2026-01-11 22:40:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(webview): redirect to warning page for non-whitelisted host

Browse source
This commit is contained in:
Tim 2025-03-31 14:25:23 +02:00
parent c665c2fb75
commit 98edc086ea
3 changed files with 35 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/stremio_app/stremio_wevbiew/constants.rs Normal file
View file

@ -0,0 +1,15 @@
pub const WARNING_URL: &str = "https://www.stremio.com/warning#";
pub const WHITELISTED_HOSTS: &[&str] = &[
"stremio.com",
"strem.io",
"stremio.zendesk.com",
"google.com",
"youtube.com",
"twitch.tv",
"twitter.com",
"x.com",
"netflix.com",
"adex.network",
"amazon.com",
"forms.gle",
];

2
src/stremio_app/stremio_wevbiew/mod.rs
View file

@ -1,2 +1,4 @@
mod constants;
pub mod wevbiew;
pub use wevbiew::WebView;

19
src/stremio_app/stremio_wevbiew/wevbiew.rs
View file

@ -9,11 +9,14 @@ use std::mem;
use std::rc::Rc;
use std::sync::{Arc, Mutex};
use std::thread;
use url::Url;
use urlencoding::decode;
use webview2::Controller;
use winapi::shared::windef::HWND;
use winapi::um::winuser::{GetClientRect, VK_F7, WM_SETFOCUS};
use super::constants::{WARNING_URL, WHITELISTED_HOSTS};
#[derive(Default)]
pub struct WebView {
pub endpoint: Rc<OnceCell<String>>,
@ -104,7 +107,21 @@ impl PartialUi for WebView {
// Handle window.open and href
webview.add_new_window_requested(move |_webview, event| {
if let Ok(uri) = event.get_uri() {
open::that(uri)?;
if let Ok(url) = Url::parse(&uri) {
let is_whitelisted = url.host().is_some_and(|host| {
WHITELISTED_HOSTS.iter().any(|whitelisted_host| host.to_string().ends_with(whitelisted_host))
});
let final_url = if is_whitelisted {
url.to_string()
} else {
format!("{}{}", WARNING_URL, urlencoding::encode(url.as_ref()))
};
if let Err(e) = open::that(final_url) {
eprintln!("Failed to open URL: {}", e);
}
}
}
Ok(())