resopt/

web.rs

//! Local project companion: filesystem analysis and review, never an upload API.
use crate::{AnalysisOptions, analyze_with_progress, server};
use anyhow::{Context, Result, ensure};
use maud::{DOCTYPE, PreEscaped, html};
use serde::Serialize;
use std::{
    fs,
    io::Write,
    path::{Path, PathBuf},
    process::{Command, Stdio},
    sync::{Arc, Mutex},
    time::Duration,
};
use tiny_http::{Method, Server};

#[derive(Default)]
pub struct WebOptions {
    pub out: Option<PathBuf>,
    pub port: u16,
    pub no_open: bool,
    pub analysis: AnalysisOptions,
}
#[derive(Default, Serialize)]
struct Progress {
    completed: usize,
    total: usize,
    done: bool,
    error: Option<String>,
}

/// Start a local-only project analysis and transition to the existing review UI.
/// Generated reports and restore backups are retained after the process exits.
pub fn web(root: impl AsRef<Path>, options: WebOptions) -> Result<()> {
    let root = fs::canonicalize(root).context("project directory not found")?;
    ensure!(root.is_dir(), "project must be a directory");
    options.analysis.validate()?;
    let out = match options.out {
        Some(path) => {
            let parent = fs::canonicalize(
                path.parent()
                    .filter(|p| !p.as_os_str().is_empty())
                    .unwrap_or(Path::new(".")),
            )?;
            parent.join(path.file_name().context("report directory has no name")?)
        }
        None => tempfile::Builder::new()
            .prefix("resopt-web-")
            .tempdir()?
            .keep()
            .join("analysis"),
    };
    ensure!(
        !out.starts_with(&root),
        "analysis output must be outside the scanned project"
    );
    ensure!(
        !out.try_exists()?,
        "analysis output must be a new directory"
    );
    let server = Server::http(("127.0.0.1", options.port)).map_err(|e| anyhow::anyhow!("{e}"))?;
    let address = server.server_addr().to_string();
    let origin = format!("http://{address}");
    let page = progress_page(&root, &out);
    println!(
        "Local web: {origin}/\nProject: {}\nReport: {}\nStop with Ctrl-C. Files stay on this device; reports and restore backups are retained.",
        root.display(),
        out.display()
    );
    std::io::stdout().flush()?;
    let state = Arc::new(Mutex::new(Progress::default()));
    let worker_state = state.clone();
    let worker_out = out.clone();
    let worker = std::thread::spawn(move || {
        let result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
            analyze_with_progress(&root, &worker_out, options.analysis, |completed, total| {
                let mut state = worker_state.lock().unwrap();
                state.completed = state.completed.max(completed);
                state.total = total;
            })
        }));
        let mut state = worker_state.lock().unwrap();
        match result {
            Ok(Ok(_)) => state.done = true,
            Ok(Err(error)) => state.error = Some(format!("{error:#}")),
            Err(_) => state.error = Some("分析进程异常；请查看终端并重试。".into()),
        }
    });
    if !options.no_open {
        open_browser(&origin);
    }
    loop {
        if state.lock().unwrap().done {
            worker
                .join()
                .map_err(|_| anyhow::anyhow!("analysis thread failed"))?;
            return server::serve_on(&out, server);
        }
        let Some(request) = server.recv_timeout(Duration::from_millis(200))? else {
            continue;
        };
        if server::header(&request, "Host") != Some(address.as_str()) {
            server::respond(request, 403, "text/plain", b"Invalid host".to_vec());
            continue;
        }
        if request.method() != &Method::Get {
            server::respond(
                request,
                405,
                "text/plain",
                b"No uploads; analysis reads the selected local project".to_vec(),
            );
            continue;
        }
        match request.url() {
            "/" => server::respond(
                request,
                200,
                "text/html; charset=utf-8",
                page.as_bytes().to_vec(),
            ),
            "/api/progress" => {
                let progress = state.lock().unwrap();
                server::respond(
                    request,
                    200,
                    "application/json",
                    serde_json::to_vec(&*progress)?,
                );
            }
            _ => server::respond(request, 404, "text/plain", b"Not found".to_vec()),
        }
    }
}
fn open_browser(origin: &str) {
    #[cfg(target_os = "macos")]
    let mut cmd = Command::new("open");
    #[cfg(target_os = "windows")]
    let mut cmd = {
        let mut cmd = Command::new("cmd");
        cmd.args(["/C", "start", ""]);
        cmd
    };
    #[cfg(not(any(target_os = "macos", target_os = "windows")))]
    let mut cmd = Command::new("xdg-open");
    match cmd
        .arg(origin)
        .stdin(Stdio::null())
        .stdout(Stdio::null())
        .stderr(Stdio::null())
        .spawn()
    {
        Ok(mut child) => {
            std::thread::spawn(move || {
                let _ = child.wait();
            });
        }
        Err(error) => eprintln!("Could not open browser ({error}); open {origin}/ manually."),
    }
}
fn progress_page(root: &Path, out: &Path) -> String {
    html! {
        (DOCTYPE) html lang="zh-CN" {
            head {
                meta charset="utf-8";meta name="viewport" content="width=device-width,initial-scale=1";
                title {"resopt · 本地项目分析"}
                style {(PreEscaped(include_str!("report.css")))}
                script {(PreEscaped("try{const t=localStorage.getItem('resopt-theme')||'system';document.documentElement.dataset.theme=t==='system'?(matchMedia('(prefers-color-scheme: dark)').matches?'dark':'light'):t}catch{}"))}
            }
            body {main {
                h1 {"正在分析本地项目"}
                p {(root.display())}
                p {"图片直接从本机磁盘读取，不上传到远程服务器。完成后自动进入审核页面。"}
                p { @if cfg!(target_os="macos") {"当前能力：PNG 无损 · JPEG / HEIC · 透明度与画质检测"} @else {"当前能力：PNG 无损；JPEG / HEIC 编码仅在 macOS 可用。"} }
                p id="progress" role="status" aria-live="polite" {"正在扫描目录与 Git 忽略规则…"}
                p {"报告与恢复备份保留在：" (out.display())}
            }
            script {(PreEscaped(r#"async function poll(){try{const r=await fetch('/api/progress',{cache:'no-store'});if(!r.ok)throw Error('connection');const p=await r.json();if(p.done){location.reload();return;}if(p.error){document.getElementById('progress').textContent='分析失败：'+p.error;return;}document.getElementById('progress').textContent=p.total?'已分析 '+p.completed+' / '+p.total+' 个资源':'正在扫描目录与 Git 忽略规则…';setTimeout(poll,700);}catch{document.getElementById('progress').textContent='本地服务连接中断，请确认终端进程仍在运行。';setTimeout(poll,2000);}}poll();"#))}
            }
        }
    }.into_string()
}