mirror of
https://github.com/Stremio/stremio-shell-ng.git
synced 2026-05-17 18:21:59 +00:00
feat(webview): redirect to warning page for non-whitelisted host
This commit is contained in:
parent
c665c2fb75
commit
98edc086ea
3 changed files with 35 additions and 1 deletions
15
src/stremio_app/stremio_wevbiew/constants.rs
Normal file
15
src/stremio_app/stremio_wevbiew/constants.rs
Normal 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",
|
||||||
|
];
|
||||||
|
|
@ -1,2 +1,4 @@
|
||||||
|
mod constants;
|
||||||
|
|
||||||
pub mod wevbiew;
|
pub mod wevbiew;
|
||||||
pub use wevbiew::WebView;
|
pub use wevbiew::WebView;
|
||||||
|
|
|
||||||
|
|
@ -9,11 +9,14 @@ use std::mem;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::sync::{Arc, Mutex};
|
use std::sync::{Arc, Mutex};
|
||||||
use std::thread;
|
use std::thread;
|
||||||
|
use url::Url;
|
||||||
use urlencoding::decode;
|
use urlencoding::decode;
|
||||||
use webview2::Controller;
|
use webview2::Controller;
|
||||||
use winapi::shared::windef::HWND;
|
use winapi::shared::windef::HWND;
|
||||||
use winapi::um::winuser::{GetClientRect, VK_F7, WM_SETFOCUS};
|
use winapi::um::winuser::{GetClientRect, VK_F7, WM_SETFOCUS};
|
||||||
|
|
||||||
|
use super::constants::{WARNING_URL, WHITELISTED_HOSTS};
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct WebView {
|
pub struct WebView {
|
||||||
pub endpoint: Rc<OnceCell<String>>,
|
pub endpoint: Rc<OnceCell<String>>,
|
||||||
|
|
@ -104,7 +107,21 @@ impl PartialUi for WebView {
|
||||||
// Handle window.open and href
|
// Handle window.open and href
|
||||||
webview.add_new_window_requested(move |_webview, event| {
|
webview.add_new_window_requested(move |_webview, event| {
|
||||||
if let Ok(uri) = event.get_uri() {
|
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(())
|
Ok(())
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue