🐛 修复重启异常

This commit is contained in:
BTMuli
2025-12-02 11:37:37 +08:00
parent 1d204c8284
commit b267599039
8 changed files with 102 additions and 66 deletions

12
src-tauri/Cargo.lock generated
View File

@@ -3422,9 +3422,9 @@ dependencies = [
[[package]]
name = "prost"
version = "0.14.2"
version = "0.14.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "101fec8d036f8d9d4a1e8ebf90d566d1d798f3b1aa379d2576a54a0d9acea5bd"
checksum = "7231bd9b3d3d33c86b58adbac74b5ec0ad9f496b19d22801d773636feaa95f3d"
dependencies = [
"bytes",
"prost-derive",
@@ -3432,9 +3432,9 @@ dependencies = [
[[package]]
name = "prost-derive"
version = "0.14.2"
version = "0.14.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d2d93e596a829ebe00afa41c3a056e6308d6b8a4c7d869edf184e2c91b1ba564"
checksum = "9120690fafc389a67ba3803df527d0ec9cbbc9cc45e4cc20b332996dfb672425"
dependencies = [
"anyhow",
"itertools",
@@ -3445,9 +3445,9 @@ dependencies = [
[[package]]
name = "prost-types"
version = "0.14.2"
version = "0.14.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f5d7b7346e150de32340ae3390b8b3ffa37ad93ec31fb5dad86afe817619e4e7"
checksum = "b9b4db3d6da204ed77bb26ba83b6122a73aeb2e87e25fbf7ad2e84c4ccbf8f72"
dependencies = [
"prost",
]

View File

@@ -26,8 +26,8 @@ prost = "0.14.1"
prost-types = "0.14.1"
serde = { version = "1.0.228", features = ["derive"] }
serde_json = "1.0.145"
tauri = { version = "2.9.3", features = [] }
tauri-utils = "2.8.0"
tauri = { version = "2.9.4", features = [] }
tauri-utils = "2.8.1"
url = "2.5.7"
walkdir = "2.5.0"

View File

@@ -136,29 +136,44 @@ pub fn run_with_admin() -> Result<(), String> {
use std::ffi::OsStr;
use std::iter::once;
use std::os::windows::ffi::OsStrExt;
use std::ptr::null_mut;
use windows_sys::Win32::Foundation::HWND;
use windows_sys::Win32::UI::Shell::ShellExecuteW;
use windows_sys::Win32::UI::WindowsAndMessaging::SW_SHOWNORMAL;
fn to_wide(s: &OsStr) -> Vec<u16> {
s.encode_wide().chain(once(0)).collect()
}
let exe_path = std::env::current_exe().map_err(|e| e.to_string())?;
let exe_str: Vec<u16> = exe_path.as_os_str().encode_wide().chain(once(0)).collect();
let verb: Vec<u16> = OsStr::new("runas").encode_wide().chain(once(0)).collect();
let workdir: Vec<u16> =
exe_path.parent().unwrap().as_os_str().encode_wide().chain(once(0)).collect();
if !exe_path.exists() {
return Err(format!("executable not found: {}", exe_path.display()));
}
let elevated_arg = "--elevated-action=post_install";
// /C start "" "<full_path>" --elevated-action=post_install
let params = format!("/C start \"\" \"{}\" {}", exe_path.display(), elevated_arg);
let cmd_w = to_wide(OsStr::new("cmd.exe"));
let verb_w = to_wide(OsStr::new("runas"));
let params_w = to_wide(OsStr::new(&params));
let workdir_w =
exe_path.parent().map(|p| to_wide(p.as_os_str())).unwrap_or_else(|| to_wide(OsStr::new("")));
unsafe {
let result = ShellExecuteW(
0 as HWND,
verb.as_ptr(),
exe_str.as_ptr(),
std::ptr::null(),
workdir.as_ptr(),
verb_w.as_ptr(),
cmd_w.as_ptr(),
params_w.as_ptr(),
if workdir_w.len() > 1 { workdir_w.as_ptr() } else { null_mut() },
SW_SHOWNORMAL,
);
if result as usize > 32 {
std::process::exit(0);
if (result as usize) > 32 {
Ok(())
} else {
Err("Failed to restart as administrator.".into())
Err("Failed to restart as administrator via cmd.".into())
}
}
}

View File

@@ -1,6 +1,5 @@
//! @file src/plugins.rs
//! @desc 插件模块,用于注册插件
//! @since Beta v0.6.2
//! 插件模块,用于注册插件
//! @since Beta v0.7.8
use crate::utils::get_current_date;
use log::LevelFilter;
@@ -11,7 +10,29 @@ use tauri_plugin_single_instance::init;
// 单例插件
pub fn build_si_plugin<R: Runtime>() -> TauriPlugin<R> {
init(move |app, argv, _cwd| app.emit("active_deep_link", argv).unwrap())
init(move |app, argv, _cwd| {
// 把 argv 转成 Vec<String>
let args: Vec<String> = argv.iter().map(|s| s.to_string()).collect();
// 如果包含提升约定参数,发出专门事件并短路退出
if args.iter().any(|a| a.starts_with("--elevated-action"))
|| args.iter().any(|a| a == "--elevated")
{
if let Err(e) = app.emit("elevated_launch", args.clone()) {
// 记录错误但不要 panic
eprintln!("emit elevated_launch failed: {}", e);
}
// 提升实例通常只负责传参或执行一次性任务,退出避免与主实例冲突
std::process::exit(0);
}
// 非提升启动:按原逻辑广播 deep link
if let Err(e) = app.emit("active_deep_link", argv) {
eprintln!("emit active_deep_link failed: {}", e);
}
// 回调必须返回 unit直接结束即可
})
}
// 日志插件