140 lines
5.1 KiB
Rust
140 lines
5.1 KiB
Rust
pub mod winget;
|
|
|
|
use std::fs;
|
|
use std::process::Command;
|
|
use std::os::windows::process::CommandExt;
|
|
use tokio::sync::mpsc;
|
|
use tauri::{AppHandle, Manager, State, Emitter};
|
|
use serde::Serialize;
|
|
use winget::{Software, list_all_software, list_updates, ensure_winget_dependencies};
|
|
|
|
struct AppState {
|
|
install_tx: mpsc::Sender<String>,
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn get_essentials(app: AppHandle) -> Vec<Software> {
|
|
let app_data_dir = app.path().app_data_dir().unwrap_or_default();
|
|
if !app_data_dir.exists() {
|
|
let _ = fs::create_dir_all(&app_data_dir);
|
|
}
|
|
|
|
let file_path = app_data_dir.join("setup-essentials.json");
|
|
if !file_path.exists() {
|
|
let default_essentials = vec![
|
|
Software {
|
|
id: "Microsoft.PowerToys".to_string(),
|
|
name: "PowerToys".to_string(),
|
|
description: Some("Microsoft PowerToys 是一组实用程序，供高级用户调整和简化其 Windows 10 和 11 体验。".to_string()),
|
|
version: None,
|
|
available_version: None,
|
|
icon_url: Some("https://raw.githubusercontent.com/microsoft/PowerToys/main/doc/images/icons/PowerToys icon/PNG/PowerToysAppList.targetsize-48.png".to_string()),
|
|
status: "idle".to_string(),
|
|
progress: 0.0,
|
|
},
|
|
Software {
|
|
id: "Google.Chrome.EXE".to_string(),
|
|
name: "Google Chrome".to_string(),
|
|
description: Some("Google Chrome 是一款快速、安全且免费的浏览器。".to_string()),
|
|
version: None,
|
|
available_version: None,
|
|
icon_url: Some("https://www.google.com/chrome/static/images/chrome-logo.svg".to_string()),
|
|
status: "idle".to_string(),
|
|
progress: 0.0,
|
|
}
|
|
];
|
|
let _ = fs::write(&file_path, serde_json::to_string_pretty(&default_essentials).unwrap());
|
|
return default_essentials;
|
|
}
|
|
|
|
let content = fs::read_to_string(file_path).unwrap_or_else(|_| "[]".to_string());
|
|
serde_json::from_str(&content).unwrap_or_default()
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_all_software() -> Vec<Software> {
|
|
tokio::task::spawn_blocking(move || list_all_software()).await.unwrap_or_default()
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn get_updates() -> Vec<Software> {
|
|
tokio::task::spawn_blocking(move || list_updates()).await.unwrap_or_default()
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn install_software(id: String, state: State<'_, AppState>) -> Result<(), String> {
|
|
state.install_tx.send(id).await.map_err(|e| e.to_string())
|
|
}
|
|
|
|
#[derive(Clone, Serialize)]
|
|
struct InstallProgress {
|
|
id: String,
|
|
status: String,
|
|
progress: f32,
|
|
}
|
|
|
|
pub fn run() {
|
|
tauri::Builder::default()
|
|
.plugin(tauri_plugin_opener::init())
|
|
.setup(move |app| {
|
|
let handle = app.handle().clone();
|
|
|
|
// 确保依赖项已安装 (这是一个耗时的过程，建议在异步任务中运行)
|
|
tauri::async_runtime::spawn(async move {
|
|
let _ = tokio::task::spawn_blocking(|| {
|
|
let _ = ensure_winget_dependencies();
|
|
}).await;
|
|
});
|
|
|
|
// 在 setup 闭包中初始化，此时运行时已就绪
|
|
let (tx, mut rx) = mpsc::channel::<String>(100);
|
|
|
|
// 托管状态
|
|
app.manage(AppState { install_tx: tx });
|
|
|
|
// 后台安装队列
|
|
tauri::async_runtime::spawn(async move {
|
|
while let Some(id) = rx.recv().await {
|
|
let _ = handle.emit("install-status", InstallProgress {
|
|
id: id.clone(),
|
|
status: "installing".to_string(),
|
|
progress: 0.5,
|
|
});
|
|
|
|
let id_for_cmd = id.clone();
|
|
let status_result = tokio::task::spawn_blocking(move || {
|
|
let output = Command::new("winget")
|
|
.args([
|
|
"install", "--id", &id_for_cmd, "-e", "--silent",
|
|
"--accept-package-agreements", "--accept-source-agreements",
|
|
"--disable-interactivity"
|
|
])
|
|
.creation_flags(0x08000000)
|
|
.output();
|
|
|
|
match output {
|
|
Ok(out) if out.status.success() => "success",
|
|
_ => "error",
|
|
}
|
|
}).await.unwrap_or("error");
|
|
|
|
let _ = handle.emit("install-status", InstallProgress {
|
|
id: id.clone(),
|
|
status: status_result.to_string(),
|
|
progress: 1.0,
|
|
});
|
|
}
|
|
});
|
|
|
|
Ok(())
|
|
})
|
|
.invoke_handler(tauri::generate_handler![
|
|
get_essentials,
|
|
get_all_software,
|
|
get_updates,
|
|
install_software
|
|
])
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
}
|