freenet 0.2.48

//! Background installation logic for the Freenet setup wizard.
//!
//! Runs on a background thread and reports progress via a callback.
//! Each step is independent — failures in optional steps (like downloading
//! fdev) don't block the installation.

use anyhow::{Context, Result};
use std::path::{Path, PathBuf};

use super::detection::get_install_dir;
use crate::commands::service::DASHBOARD_URL;

/// Progress updates sent from the installer thread to the UI.
#[derive(Debug, Clone)]
pub enum InstallProgress {
    StoppingExisting,
    CopyingBinary,
    DownloadingFdev,
    FdevSkipped(String),
    AddingToPath,
    InstallingService,
    LaunchingService,
    OpeningDashboard,
    Complete,
    Error(String),
}

/// Run the full installation, reporting progress via the callback.
///
/// Steps:
/// 1. Stop any running Freenet service
/// 2. Copy freenet.exe to install directory
/// 3. Download fdev.exe (optional, non-blocking on failure)
/// 4. Add install directory to user PATH
/// 5. Install service (Task Scheduler)
/// 6. Start service (launches run-wrapper + tray icon)
/// 7. Open dashboard in browser
pub fn run_install(progress: impl Fn(InstallProgress) + Send) -> Result<()> {
    let install_dir =
        get_install_dir().context("Could not determine install directory (%LOCALAPPDATA%)")?;

    let current_exe =
        std::env::current_exe().context("Could not determine current executable path")?;

    let installed_exe = install_dir.join("freenet.exe");

    // Step 1: Stop existing service if running
    progress(InstallProgress::StoppingExisting);
    if installed_exe.exists() {
        // Try to stop gracefully — ignore errors (may not be running)
        drop(
            std::process::Command::new(&installed_exe)
                .args(["service", "stop"])
                .stdout(std::process::Stdio::null())
                .stderr(std::process::Stdio::null())
                .status(),
        );
        std::thread::sleep(std::time::Duration::from_secs(2));
    }

    // Step 2: Copy binary to install directory
    progress(InstallProgress::CopyingBinary);
    std::fs::create_dir_all(&install_dir)
        .with_context(|| format!("Failed to create directory: {}", install_dir.display()))?;
    std::fs::copy(&current_exe, &installed_exe)
        .with_context(|| format!("Failed to copy binary to {}", installed_exe.display()))?;

    // Step 3: Download fdev.exe (optional)
    progress(InstallProgress::DownloadingFdev);
    match download_fdev(&install_dir) {
        Ok(()) => {}
        Err(e) => {
            progress(InstallProgress::FdevSkipped(format!("{e:#}")));
        }
    }

    // Step 4: Add to user PATH
    progress(InstallProgress::AddingToPath);
    if let Err(e) = add_to_user_path(&install_dir) {
        // Non-fatal — warn but continue
        eprintln!("Warning: could not add to PATH: {e:#}");
    }

    // Step 5: Install service
    progress(InstallProgress::InstallingService);
    let output = std::process::Command::new(&installed_exe)
        .args(["service", "install"])
        .output()
        .context("Failed to run freenet service install")?;
    if !output.status.success() {
        let stderr = String::from_utf8_lossy(&output.stderr);
        let detail = if stderr.trim().is_empty() {
            format!("exit code {}", output.status.code().unwrap_or(-1))
        } else {
            stderr.trim().to_string()
        };
        return Err(anyhow::anyhow!("Service installation failed: {detail}"));
    }

    // Step 6: Start service
    //
    // Use Stdio::null() + .status() instead of .output() to prevent Windows
    // handle inheritance. With .output(), the installer creates pipes that get
    // inherited by the detached wrapper process (via CreateProcess's default
    // bInheritHandles=TRUE). The wrapper runs forever, so the pipe handles
    // never close, and .output() blocks indefinitely waiting for EOF.
    progress(InstallProgress::LaunchingService);
    let status = std::process::Command::new(&installed_exe)
        .args(["service", "start"])
        .stdin(std::process::Stdio::null())
        .stdout(std::process::Stdio::null())
        .stderr(std::process::Stdio::null())
        .status()
        .context("Failed to start freenet service")?;
    if !status.success() {
        return Err(anyhow::anyhow!(
            "Service failed to start (exit code {}). Try running 'freenet service start' manually.",
            status.code().unwrap_or(-1)
        ));
    }

    // Brief wait for the node to initialize before opening the dashboard
    std::thread::sleep(std::time::Duration::from_secs(3));

    // Step 7: Open dashboard
    progress(InstallProgress::OpeningDashboard);
    crate::commands::open_url_in_browser(DASHBOARD_URL);

    progress(InstallProgress::Complete);
    Ok(())
}

/// Download fdev.exe from the GitHub release matching the current version.
fn download_fdev(install_dir: &Path) -> Result<()> {
    let version = env!("CARGO_PKG_VERSION");
    let url = format!(
        "https://github.com/freenet/freenet-core/releases/download/v{version}/fdev-x86_64-pc-windows-msvc.zip"
    );

    let zip_path = install_dir.join("fdev-download.zip");

    // Use Win32 URLDownloadToFile for zero additional dependencies
    download_url_to_file(&url, &zip_path)?;

    // Extract fdev.exe from the zip
    let zip_file = std::fs::File::open(&zip_path).context("Failed to open downloaded fdev zip")?;
    let mut archive = zip::ZipArchive::new(zip_file).context("Failed to read fdev zip")?;

    for i in 0..archive.len() {
        let mut file = archive.by_index(i)?;
        let name = file.name().to_string();
        if name.ends_with("fdev.exe") {
            let dest = install_dir.join("fdev.exe");
            let mut out = std::fs::File::create(&dest).context("Failed to create fdev.exe")?;
            std::io::copy(&mut file, &mut out)?;
            break;
        }
    }

    // Clean up zip
    drop(std::fs::remove_file(&zip_path));

    Ok(())
}

/// Download a URL to a file using Win32 URLDownloadToFileW.
#[cfg(target_os = "windows")]
fn download_url_to_file(url: &str, dest: &Path) -> Result<()> {
    use std::ffi::OsStr;
    use std::os::windows::ffi::OsStrExt;

    // winapi 0.3.9 does not include URLDownloadToFileW bindings,
    // so we declare the FFI import directly from urlmon.dll.
    #[link(name = "urlmon")]
    unsafe extern "system" {
        // SAFETY: Standard Win32 API from urlmon.dll.
        // Signature: HRESULT URLDownloadToFileW(LPUNKNOWN, LPCWSTR, LPCWSTR, DWORD, LPBINDSTATUSCALLBACK)
        fn URLDownloadToFileW(
            caller: *mut std::ffi::c_void,
            url: *const u16,
            file_name: *const u16,
            reserved: u32,
            status_cb: *mut std::ffi::c_void,
        ) -> i32;
    }

    let url_wide: Vec<u16> = OsStr::new(url).encode_wide().chain(Some(0)).collect();
    let dest_wide: Vec<u16> = dest.as_os_str().encode_wide().chain(Some(0)).collect();

    // SAFETY: FFI call to urlmon.dll. Both strings are null-terminated wide strings.
    // All pointer parameters except the two strings are null (unused).
    let hr = unsafe {
        URLDownloadToFileW(
            std::ptr::null_mut(),
            url_wide.as_ptr(),
            dest_wide.as_ptr(),
            0,
            std::ptr::null_mut(),
        )
    };

    if hr != 0 {
        anyhow::bail!("URLDownloadToFileW failed with HRESULT 0x{hr:08x}");
    }

    Ok(())
}

#[cfg(not(target_os = "windows"))]
fn download_url_to_file(_url: &str, _dest: &Path) -> Result<()> {
    anyhow::bail!("URLDownloadToFile is only available on Windows")
}

/// Add a directory to the user's PATH environment variable via the registry.
/// Preserves the existing registry value type (REG_EXPAND_SZ) to avoid
/// breaking PATH entries that use `%VARIABLE%` expansion.
#[cfg(target_os = "windows")]
fn add_to_user_path(dir: &Path) -> Result<()> {
    #[rustfmt::skip] // CI and local rustfmt disagree on ordering
    use winreg::{RegKey, enums::*};

    let hkcu = RegKey::predef(HKEY_CURRENT_USER);
    let env = hkcu.open_subkey_with_flags("Environment", KEY_READ | KEY_WRITE)?;

    // Read as raw value to preserve REG_EXPAND_SZ type
    let raw_val = env
        .get_raw_value("Path")
        .unwrap_or_else(|_| winreg::RegValue {
            bytes: std::borrow::Cow::Owned(Vec::new()),
            vtype: REG_EXPAND_SZ,
        });

    // Decode the wide string from the registry
    let current_path = String::from_utf16_lossy(
        &raw_val
            .bytes
            .chunks_exact(2)
            .map(|c| u16::from_le_bytes([c[0], c[1]]))
            .collect::<Vec<u16>>(),
    )
    .trim_end_matches('\0')
    .to_string();

    let dir_str = dir.to_string_lossy();

    // Check if already in PATH — compare individual entries, not substrings
    let already_present = current_path
        .split(';')
        .any(|entry| entry.trim().eq_ignore_ascii_case(dir_str.as_ref()));

    if already_present {
        return Ok(());
    }

    // Append to PATH
    let new_path = if current_path.is_empty() {
        dir_str.to_string()
    } else {
        format!("{current_path};{dir_str}")
    };

    // Write back preserving the original value type (typically REG_EXPAND_SZ)
    let mut new_bytes: Vec<u8> = new_path
        .encode_utf16()
        .chain(Some(0)) // null terminator
        .flat_map(|c| c.to_le_bytes())
        .collect();
    // Ensure double-null termination for REG_EXPAND_SZ
    new_bytes.extend_from_slice(&[0, 0]);

    env.set_raw_value(
        "Path",
        &winreg::RegValue {
            bytes: std::borrow::Cow::Owned(new_bytes),
            vtype: raw_val.vtype, // preserve original type
        },
    )?;

    // Broadcast WM_SETTINGCHANGE so running Explorer picks up the change
    broadcast_environment_change();

    Ok(())
}

#[cfg(not(target_os = "windows"))]
fn add_to_user_path(_dir: &Path) -> Result<()> {
    Ok(()) // No-op on non-Windows
}

/// Broadcast WM_SETTINGCHANGE so other processes pick up the PATH change.
#[cfg(target_os = "windows")]
fn broadcast_environment_change() {
    use std::ffi::OsStr;
    use std::os::windows::ffi::OsStrExt;

    let environment: Vec<u16> = OsStr::new("Environment")
        .encode_wide()
        .chain(Some(0))
        .collect();

    unsafe {
        winapi::um::winuser::SendMessageTimeoutW(
            winapi::um::winuser::HWND_BROADCAST,
            winapi::um::winuser::WM_SETTINGCHANGE,
            0,
            environment.as_ptr() as isize,
            winapi::um::winuser::SMTO_ABORTIFHUNG,
            5000,
            std::ptr::null_mut(),
        );
    }
}