Firelink/src-tauri/src/parity.rs

use serde::{Deserialize, Serialize};
#[cfg(any(target_os = "macos", target_os = "linux"))]
use std::process::Command;
use ts_rs::TS;

use crate::ipc::DownloadCategory;

#[tauri::command]
pub async fn get_system_proxy() -> Result<Option<String>, String> {
    match native_system_proxy() {
        Ok(Some(proxy)) => Ok(Some(proxy)),
        Ok(None) => Ok(proxy_from_environment()),
        Err(native_error) => match sysproxy::Sysproxy::get_system_proxy() {
            Ok(proxy) if proxy.enable => {
                if proxy.host.contains('=') {
                    Ok(parse_windows_proxy_server(&proxy.host).or_else(proxy_from_environment))
                } else {
                    Ok(normalize_sysproxy_address(&proxy.host, proxy.port)
                        .or_else(proxy_from_environment))
                }
            }
            Ok(_) => Ok(proxy_from_environment()),
            Err(error) => proxy_from_environment().map(Some).ok_or_else(|| {
                format!(
                    "failed to read system proxy settings: {native_error}; sysproxy fallback: {error}"
                )
            }),
        },
    }
}

#[cfg(target_os = "windows")]
fn native_system_proxy() -> Result<Option<String>, String> {
    fallback_windows_proxy().map_err(|_| "failed to read Windows proxy registry".to_string())
}

#[cfg(target_os = "macos")]
fn native_system_proxy() -> Result<Option<String>, String> {
    let proxy = sysproxy::Sysproxy::get_system_proxy().map_err(|error| error.to_string())?;
    if !proxy.enable {
        return Ok(None);
    }
    Ok(macos_proxy_for_host_port(&proxy.host, proxy.port)
        .unwrap_or_else(|| {
            normalize_sysproxy_address(&proxy.host, proxy.port)
                .unwrap_or_else(|| format!("http://{}:{}", proxy.host, proxy.port))
        })
        .into())
}

#[cfg(target_os = "linux")]
fn native_system_proxy() -> Result<Option<String>, String> {
    let mode =
        command_stdout(Command::new("gsettings").args(["get", "org.gnome.system.proxy", "mode"]))
            .map_err(|error| error.to_string())?;
    if strip_gsettings_string(&mode) != "manual" {
        return Ok(None);
    }

    Ok(linux_gsettings_proxy("https", "http")
        .or_else(|| linux_gsettings_proxy("http", "http"))
        .or_else(|| linux_gsettings_proxy("socks", "socks5")))
}

#[cfg(not(any(target_os = "windows", target_os = "macos", target_os = "linux")))]
fn native_system_proxy() -> Result<Option<String>, String> {
    Ok(None)
}

fn proxy_from_environment() -> Option<String> {
    [
        "HTTPS_PROXY",
        "https_proxy",
        "HTTP_PROXY",
        "http_proxy",
        "ALL_PROXY",
        "all_proxy",
    ]
    .into_iter()
    .find_map(|name| {
        std::env::var(name)
            .ok()
            .and_then(|value| normalize_proxy_address(&value, "http"))
    })
}

#[cfg(any(target_os = "macos", target_os = "linux"))]
fn command_stdout(command: &mut Command) -> std::io::Result<String> {
    let output = command.output()?;
    if !output.status.success() {
        return Err(std::io::Error::other(format!(
            "command exited with {}",
            output.status
        )));
    }
    Ok(String::from_utf8_lossy(&output.stdout).to_string())
}

fn normalize_proxy_address(raw: &str, default_scheme: &str) -> Option<String> {
    let trimmed = raw.trim().trim_matches('"').trim_end_matches('/');
    if trimmed.is_empty() {
        return None;
    }

    let candidate = if trimmed.contains("://") {
        trimmed.to_string()
    } else {
        format!("{default_scheme}://{trimmed}")
    };
    let parsed = url::Url::parse(&candidate).ok()?;
    match parsed.scheme() {
        "http" | "https" | "socks4" | "socks4a" | "socks5" | "socks5h" => {}
        _ => return None,
    }
    parsed.host_str()?;
    Some(candidate)
}

fn normalize_sysproxy_address(host: &str, port: u16) -> Option<String> {
    let host = host.trim();
    if host.is_empty() {
        return None;
    }

    if host.contains("://") {
        let mut parsed = url::Url::parse(host).ok()?;
        if parsed.port().is_none() && port != 0 {
            parsed.set_port(Some(port)).ok()?;
        }
        return normalize_proxy_address(parsed.as_str(), "http");
    }

    if port == 0 {
        normalize_proxy_address(host, "http")
    } else {
        normalize_proxy_address(&format!("{host}:{port}"), "http")
    }
}

fn parse_windows_proxy_server(value: &str) -> Option<String> {
    let value = value.trim().trim_matches('"');
    if value.is_empty() {
        return None;
    }

    if !value.contains('=') {
        return normalize_proxy_address(value, "http");
    }

    let mut http = None;
    let mut https = None;
    let mut socks = None;
    for entry in value.split(';') {
        let Some((kind, address)) = entry.split_once('=') else {
            continue;
        };
        let kind = kind.trim().to_ascii_lowercase();
        let address = address.trim();
        match kind.as_str() {
            "http" => http = normalize_proxy_address(address, "http"),
            "https" => https = normalize_proxy_address(address, "http"),
            "socks" => socks = normalize_proxy_address(address, "socks5"),
            _ => {}
        }
    }

    https.or(http).or(socks)
}

#[cfg(target_os = "macos")]
fn macos_proxy_for_host_port(host: &str, port: u16) -> Option<String> {
    let services_output =
        command_stdout(Command::new("networksetup").arg("-listallnetworkservices")).ok()?;
    for service in parse_macos_network_services(&services_output) {
        for (target, scheme) in [
            ("securewebproxy", "http"),
            ("webproxy", "http"),
            ("socksfirewallproxy", "socks5"),
        ] {
            let output = command_stdout(
                Command::new("networksetup").args([format!("-get{target}"), service.clone()]),
            )
            .ok()?;
            if let Some(proxy) = parse_macos_networksetup_proxy(&output, scheme)
                .filter(|proxy| proxy_matches_host_port(proxy, host, port))
            {
                return Some(proxy);
            }
        }
    }
    None
}

#[cfg_attr(not(target_os = "macos"), allow(dead_code))]
fn parse_macos_network_services(output: &str) -> Vec<String> {
    output
        .lines()
        .map(str::trim)
        .filter(|line| !line.is_empty())
        .filter(|line| !line.starts_with("An asterisk"))
        .filter(|line| !line.starts_with('*'))
        .map(str::to_string)
        .collect()
}

#[cfg_attr(not(target_os = "macos"), allow(dead_code))]
fn parse_macos_networksetup_proxy(output: &str, scheme: &str) -> Option<String> {
    let enabled = macos_networksetup_value(output, "Enabled:")
        .is_some_and(|value| value.eq_ignore_ascii_case("yes"));
    if !enabled {
        return None;
    }
    let server = macos_networksetup_value(output, "Server:")?;
    let port = macos_networksetup_value(output, "Port:")?;
    normalize_proxy_address(&format!("{scheme}://{server}:{port}"), scheme)
}

#[cfg_attr(not(target_os = "macos"), allow(dead_code))]
fn proxy_matches_host_port(proxy: &str, host: &str, port: u16) -> bool {
    let Ok(parsed) = url::Url::parse(proxy) else {
        return false;
    };
    parsed.host_str() == Some(host) && parsed.port() == Some(port)
}

#[cfg_attr(not(target_os = "macos"), allow(dead_code))]
fn macos_networksetup_value<'a>(output: &'a str, key: &str) -> Option<&'a str> {
    output
        .lines()
        .map(str::trim)
        .find_map(|line| line.strip_prefix(key).map(str::trim))
        .filter(|value| !value.is_empty())
}

#[cfg(target_os = "linux")]
fn linux_gsettings_proxy(service: &str, scheme: &str) -> Option<String> {
    let schema = format!("org.gnome.system.proxy.{service}");
    let host = command_stdout(Command::new("gsettings").args(["get", &schema, "host"])).ok()?;
    let host = strip_gsettings_string(&host);
    if host.is_empty() {
        return None;
    }
    let port = command_stdout(Command::new("gsettings").args(["get", &schema, "port"])).ok()?;
    let port = port.trim();
    normalize_proxy_address(&format!("{scheme}://{host}:{port}"), scheme)
}

#[cfg_attr(not(target_os = "linux"), allow(dead_code))]
fn strip_gsettings_string(value: &str) -> String {
    value
        .trim()
        .trim_matches('\'')
        .trim_matches('"')
        .to_string()
}

#[cfg(target_os = "windows")]
fn fallback_windows_proxy() -> Result<Option<String>, ()> {
    use std::os::windows::process::CommandExt;
    use std::process::Command;
    const CREATE_NO_WINDOW: u32 = 0x08000000;

    let output = Command::new("reg")
        .args(&[
            "query",
            "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
            "/v",
            "ProxyEnable",
        ])
        .creation_flags(CREATE_NO_WINDOW)
        .output()
        .map_err(|_| ())?;

    if !output.status.success() {
        return Err(());
    }

    let stdout = String::from_utf8_lossy(&output.stdout);
    let enabled = registry_value(&stdout, "ProxyEnable")
        .as_deref()
        .is_some_and(windows_proxy_enabled);
    if !enabled {
        return Ok(None);
    }

    let output = Command::new("reg")
        .args(&[
            "query",
            "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
            "/v",
            "ProxyServer",
        ])
        .creation_flags(CREATE_NO_WINDOW)
        .output()
        .map_err(|_| ())?;

    if !output.status.success() {
        return Err(());
    }

    let stdout = String::from_utf8_lossy(&output.stdout);
    Ok(registry_value(&stdout, "ProxyServer").and_then(|value| parse_windows_proxy_server(&value)))
}

#[cfg_attr(not(target_os = "windows"), allow(dead_code))]
fn windows_proxy_enabled(value: &str) -> bool {
    let value = value.trim();
    if value == "1" {
        return true;
    }
    value
        .strip_prefix("0x")
        .or_else(|| value.strip_prefix("0X"))
        .and_then(|hex| u32::from_str_radix(hex, 16).ok())
        .is_some_and(|enabled| enabled == 1)
}

#[cfg_attr(not(target_os = "windows"), allow(dead_code))]
fn registry_value(output: &str, name: &str) -> Option<String> {
    for line in output.lines() {
        let trimmed = line.trim();
        let mut parts = trimmed.split_whitespace();
        let Some(key) = parts.next() else {
            continue;
        };
        if !key.eq_ignore_ascii_case(name) {
            continue;
        }
        if parts.next().is_none() {
            continue;
        }
        let data = parts.collect::<Vec<_>>().join(" ");
        if !data.is_empty() {
            return Some(data);
        }
    }
    None
}

#[cfg(test)]
mod proxy_tests {
    use super::{
        normalize_proxy_address, normalize_sysproxy_address, parse_macos_network_services,
        parse_macos_networksetup_proxy, parse_windows_proxy_server, proxy_matches_host_port,
        registry_value, strip_gsettings_string, windows_proxy_enabled,
    };

    #[test]
    fn normalizes_bare_proxy_addresses() {
        assert_eq!(
            normalize_proxy_address("127.0.0.1:8080", "http").as_deref(),
            Some("http://127.0.0.1:8080")
        );
        assert_eq!(
            normalize_proxy_address(" socks5://127.0.0.1:1080/ ", "http").as_deref(),
            Some("socks5://127.0.0.1:1080")
        );
        assert_eq!(normalize_proxy_address("file:///tmp/proxy", "http"), None);
    }

    #[test]
    fn parses_windows_protocol_proxy_server_values() {
        assert_eq!(
            parse_windows_proxy_server("http=127.0.0.1:8080;https=127.0.0.1:8081").as_deref(),
            Some("http://127.0.0.1:8081")
        );
        assert_eq!(
            parse_windows_proxy_server("socks=127.0.0.1:1080").as_deref(),
            Some("socks5://127.0.0.1:1080")
        );
        assert_eq!(
            parse_windows_proxy_server("proxy.local:9000").as_deref(),
            Some("http://proxy.local:9000")
        );
    }

    #[test]
    fn normalizes_sysproxy_host_without_duplicating_ports() {
        assert_eq!(
            normalize_sysproxy_address("http://proxy.local", 8080).as_deref(),
            Some("http://proxy.local:8080")
        );
        assert_eq!(
            normalize_sysproxy_address("http://proxy.local:9000", 8080).as_deref(),
            Some("http://proxy.local:9000")
        );
        assert_eq!(
            normalize_sysproxy_address("proxy.local", 8080).as_deref(),
            Some("http://proxy.local:8080")
        );
    }

    #[test]
    fn parses_macos_proxy_outputs_with_scheme() {
        let services = r#"
An asterisk (*) denotes that a network service is disabled.
Wi-Fi
*USB 10/100/1000 LAN
Thunderbolt Bridge
"#;
        assert_eq!(
            parse_macos_network_services(services),
            vec!["Wi-Fi".to_string(), "Thunderbolt Bridge".to_string()]
        );

        let proxy = r#"
Enabled: Yes
Server: 127.0.0.1
Port: 1080
Authenticated Proxy Enabled: 0
"#;
        assert_eq!(
            parse_macos_networksetup_proxy(proxy, "socks5").as_deref(),
            Some("socks5://127.0.0.1:1080")
        );
        assert!(proxy_matches_host_port(
            "socks5://127.0.0.1:1080",
            "127.0.0.1",
            1080
        ));
        assert!(!proxy_matches_host_port(
            "socks5://127.0.0.1:1080",
            "127.0.0.1",
            1081
        ));

        let disabled = proxy.replace("Enabled: Yes", "Enabled: No");
        assert_eq!(parse_macos_networksetup_proxy(&disabled, "socks5"), None);
    }

    #[test]
    fn strips_gsettings_string_quotes() {
        assert_eq!(strip_gsettings_string("'manual'\n"), "manual");
        assert_eq!(strip_gsettings_string("\"127.0.0.1\"\n"), "127.0.0.1");
    }

    #[test]
    fn parses_reg_query_output_values() {
        let output = r#"
HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Internet Settings
    ProxyEnable    REG_DWORD    0x1
    ProxyServer    REG_SZ       http=127.0.0.1:8080;https=127.0.0.1:8081
"#;

        assert!(registry_value(output, "ProxyEnable")
            .as_deref()
            .is_some_and(windows_proxy_enabled));
        assert_eq!(
            registry_value(output, "ProxyServer").as_deref(),
            Some("http=127.0.0.1:8080;https=127.0.0.1:8081")
        );
        assert!(!windows_proxy_enabled("0X0"));
        assert!(windows_proxy_enabled("0X1"));
    }
}

#[tauri::command]
pub fn get_file_category(filename: String) -> DownloadCategory {
    let ext = std::path::Path::new(&filename)
        .extension()
        .and_then(|s| s.to_str())
        .map(|s| s.to_lowercase())
        .unwrap_or_default();

    let music_exts = [
        "mp3", "wav", "aac", "flac", "ogg", "m4a", "wma", "alac", "ape", "mid", "midi",
    ];
    let movie_exts = [
        "mp4", "mkv", "avi", "mov", "wmv", "flv", "webm", "m4v", "mpeg", "mpg", "3gp", "ts", "vob",
    ];
    let compressed_exts = [
        "zip", "rar", "7z", "tar", "gz", "xz", "bz2", "lz", "lzma", "zst", "iso", "cab", "tgz",
        "tbz", "z", "sit", "sitx",
    ];
    let picture_exts = [
        "jpg", "jpeg", "png", "gif", "webp", "bmp", "tiff", "svg", "ico", "heic", "raw", "psd",
        "ai",
    ];
    let document_exts = [
        "pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "txt", "rtf", "csv", "md", "epub",
        "mobi", "azw3",
    ];
    let app_exts = [
        "exe", "msi", "bat", "cmd", "app", "dmg", "pkg", "apk", "appx", "deb", "rpm", "appimage",
        "run", "sh", "bin", "jar",
    ];

    if music_exts.contains(&ext.as_str()) {
        DownloadCategory::Musics
    } else if movie_exts.contains(&ext.as_str()) {
        DownloadCategory::Movies
    } else if compressed_exts.contains(&ext.as_str()) {
        DownloadCategory::Compressed
    } else if picture_exts.contains(&ext.as_str()) {
        DownloadCategory::Pictures
    } else if document_exts.contains(&ext.as_str()) {
        DownloadCategory::Documents
    } else if app_exts.contains(&ext.as_str()) {
        DownloadCategory::Applications
    } else {
        DownloadCategory::Other
    }
}

#[derive(Serialize, Deserialize, Clone, TS)]
#[ts(export, export_to = "../../src/bindings/")]
pub struct AvailableReleaseUpdate {
    pub version: String,
    pub tag_name: String,
    pub title: String,
    pub release_notes: String,
    pub release_url: String,
    pub published_at: Option<String>,
}

#[derive(Serialize, TS)]
#[serde(tag = "type")]
#[ts(export, export_to = "../../src/bindings/")]
pub enum ReleaseCheckOutcome {
    UpdateAvailable {
        update: AvailableReleaseUpdate,
    },
    UpToDate {
        latest_version: String,
        local_version: String,
    },
}

#[derive(Deserialize)]
struct GitHubRelease {
    tag_name: String,
    name: Option<String>,
    body: Option<String>,
    html_url: String,
    draft: bool,
    prerelease: bool,
    published_at: Option<String>,
}

#[tauri::command]
pub async fn check_for_updates(
    app_handle: tauri::AppHandle,
) -> Result<ReleaseCheckOutcome, String> {
    let current_version = app_handle.package_info().version.to_string();

    crate::ensure_reqwest_crypto_provider();
    let client = reqwest::Client::new();
    let res = client
        .get("https://api.github.com/repos/nimbold/Firelink/releases?per_page=30")
        .header("User-Agent", "Firelink")
        .header("Accept", "application/vnd.github+json")
        .send()
        .await
        .map_err(|e| e.to_string())?;

    if !res.status().is_success() {
        return Err(format!("GitHub returned HTTP {}", res.status().as_u16()));
    }

    let releases: Vec<GitHubRelease> = res.json().await.map_err(|e| e.to_string())?;

    let latest_stable = releases
        .into_iter()
        .filter(|r| !r.draft && !r.prerelease)
        .max_by(|a, b| cmp_versions(&a.tag_name, &b.tag_name));

    let release = match latest_stable {
        Some(r) => r,
        None => return Err("No stable release was found.".to_string()),
    };

    let latest_version = release.tag_name.trim_start_matches(['v', 'V']).to_string();

    if cmp_versions(&latest_version, &current_version) == std::cmp::Ordering::Greater {
        Ok(ReleaseCheckOutcome::UpdateAvailable {
            update: AvailableReleaseUpdate {
                version: latest_version.clone(),
                tag_name: release.tag_name.clone(),
                title: release.name.unwrap_or(release.tag_name),
                release_notes: release.body.unwrap_or_else(|| {
                    "No release notes were provided for this version.".to_string()
                }),
                release_url: release.html_url,
                published_at: release.published_at,
            },
        })
    } else {
        Ok(ReleaseCheckOutcome::UpToDate {
            latest_version,
            local_version: current_version,
        })
    }
}

fn cmp_versions(a: &str, b: &str) -> std::cmp::Ordering {
    use semver::Version;

    let a_clean = a.trim_start_matches(['v', 'V']);
    let b_clean = b.trim_start_matches(['v', 'V']);

    let a_ver = Version::parse(a_clean).unwrap_or_else(|_| Version::new(0, 0, 0));
    let b_ver = Version::parse(b_clean).unwrap_or_else(|_| Version::new(0, 0, 0));

    a_ver.cmp(&b_ver)
}

#[tauri::command]
pub async fn create_category_directories(
    app_handle: tauri::AppHandle,
    base_folder: String,
    subfolders: std::collections::HashMap<String, String>,
) -> Result<(), String> {
    let base = crate::resolve_path(&base_folder, &app_handle);
    let mut errors = Vec::new();

    for subfolder in subfolders.values() {
        let normalized = subfolder.replace('\\', "/");
        let relative = std::path::Path::new(&normalized);
        if relative.is_absolute()
            || normalized
                .split('/')
                .any(|part| part == ".." || part.ends_with(':'))
        {
            return Err(format!(
                "Category subfolder must be a relative path: {subfolder}"
            ));
        }

        let expanded = base.join(relative);
        if !expanded.exists() {
            if let Err(e) = tokio::fs::create_dir_all(&expanded).await {
                errors.push(format!("{}: {}", expanded.display(), e));
            }
        }
    }

    if errors.is_empty() {
        Ok(())
    } else {
        Err(format!(
            "Failed to create one or more category directories ({})",
            errors.join("; ")
        ))
    }
}

pub static SUPPORTED_DOMAINS: &[&str] = &[
    "youtube.com",
    "youtu.be",
    "twitter.com",
    "x.com",
    "vimeo.com",
    "twitch.tv",
    "instagram.com",
    "tiktok.com",
    "facebook.com",
    "fb.watch",
    "reddit.com",
    "v.redd.it",
    "soundcloud.com",
    "pornhub.com",
    "redtube.com",
    "xhamster.com",
    "xnxx.com",
    "xvideos.com",
];

#[tauri::command]
pub fn get_supported_media_domains() -> Vec<String> {
    SUPPORTED_DOMAINS.iter().map(|s| s.to_string()).collect()
}

#[tauri::command]
pub fn is_supported_media(url: String) -> bool {
    if let Ok(parsed_url) = reqwest::Url::parse(&url) {
        if let Some(host) = parsed_url.host_str() {
            let host_lower = host.to_lowercase();
            for domain in SUPPORTED_DOMAINS.iter() {
                if host_lower == *domain || host_lower.ends_with(&format!(".{}", domain)) {
                    return true;
                }
            }
        }
    }
    false
}