75 lines
2.2 KiB
Rust
75 lines
2.2 KiB
Rust
use std::sync::atomic::{AtomicBool, Ordering};
|
|
|
|
const PORTABLE_DIR_NAME: &str = ".Axolotl";
|
|
|
|
static PORTABLE_MODE: AtomicBool = AtomicBool::new(false);
|
|
|
|
/// 在启动时初始化便携模式
|
|
/// 检查 `.Axolotl` 文件夹是否存在且可写
|
|
/// 如果存在且可写，将 `THESEUS_CONFIG_DIR` 环境变量设置为该路径
|
|
/// 返回 `true` 如果便携模式已启用，否则返回 `false`
|
|
///
|
|
/// 必须在 main() 开头、任何其他线程（包括 tokio runtime）启动之前调用。
|
|
/// 此函数内部会调用 `std::env::set_var`，该函数在 Rust 中不是线程安全的。
|
|
pub unsafe fn init_portable_mode() -> bool {
|
|
let exe_path = match std::env::current_exe() {
|
|
Ok(p) => p,
|
|
Err(_) => return false,
|
|
};
|
|
|
|
let Some(app_dir) = exe_path.parent() else {
|
|
return false;
|
|
};
|
|
|
|
let portable_dir = app_dir.join(PORTABLE_DIR_NAME);
|
|
|
|
if !portable_dir.is_dir() {
|
|
return false;
|
|
}
|
|
if !try_write_test(&portable_dir) {
|
|
return false;
|
|
}
|
|
|
|
// SAFETY: 调用者保证此时没有其他线程访问环境变量
|
|
unsafe {
|
|
std::env::set_var("THESEUS_CONFIG_DIR", &portable_dir);
|
|
}
|
|
PORTABLE_MODE.store(true, Ordering::Relaxed);
|
|
|
|
tracing::info!(
|
|
"Portable mode enabled: THESEUS_CONFIG_DIR={}",
|
|
portable_dir.display()
|
|
);
|
|
|
|
true
|
|
}
|
|
|
|
/// 尝试在目标目录中创建并删除临时文件以验证可写性。
|
|
/// 先删除可能残留的 `.write_test`，避免已有的只读文件导致 `File::create` 失败。
|
|
fn try_write_test(dir: &std::path::Path) -> bool {
|
|
let test_file = dir.join(".write_test");
|
|
|
|
let _ = std::fs::remove_file(&test_file);
|
|
|
|
match std::fs::File::create(&test_file) {
|
|
Ok(_) => {
|
|
let _ = std::fs::remove_file(&test_file);
|
|
true
|
|
}
|
|
Err(e) => {
|
|
tracing::warn!(
|
|
"Portable directory {} exists but is not writable: {}",
|
|
dir.display(),
|
|
e
|
|
);
|
|
false
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Tauri 命令：检查应用程序是否运行在便携模式下
|
|
#[tauri::command]
|
|
pub fn is_portable_mode() -> bool {
|
|
PORTABLE_MODE.load(Ordering::Relaxed)
|
|
}
|