7c71d07ad2
* refactor: convert synchronous file operations to asynchronous for improved performance * fix: update copy_icon_file to use asynchronous directory creation * refactor: remove unnecessary variable assignments in shortcut management functions
240 lines
6.6 KiB
Rust
240 lines
6.6 KiB
Rust
use super::CmdResult;
|
|
use crate::{
|
|
feat, logging,
|
|
utils::{
|
|
dirs::{self, PathBufExec},
|
|
logging::Type,
|
|
},
|
|
wrap_err,
|
|
};
|
|
use std::path::Path;
|
|
use tauri::{AppHandle, Manager};
|
|
use tokio::fs;
|
|
|
|
/// 打开应用程序所在目录
|
|
#[tauri::command]
|
|
pub async fn open_app_dir() -> CmdResult<()> {
|
|
let app_dir = wrap_err!(dirs::app_home_dir())?;
|
|
wrap_err!(open::that(app_dir))
|
|
}
|
|
|
|
/// 打开核心所在目录
|
|
#[tauri::command]
|
|
pub async fn open_core_dir() -> CmdResult<()> {
|
|
let core_dir = wrap_err!(tauri::utils::platform::current_exe())?;
|
|
let core_dir = core_dir.parent().ok_or("failed to get core dir")?;
|
|
wrap_err!(open::that(core_dir))
|
|
}
|
|
|
|
/// 打开日志目录
|
|
#[tauri::command]
|
|
pub async fn open_logs_dir() -> CmdResult<()> {
|
|
let log_dir = wrap_err!(dirs::app_logs_dir())?;
|
|
wrap_err!(open::that(log_dir))
|
|
}
|
|
|
|
/// 打开网页链接
|
|
#[tauri::command]
|
|
pub fn open_web_url(url: String) -> CmdResult<()> {
|
|
wrap_err!(open::that(url))
|
|
}
|
|
|
|
/// 打开/关闭开发者工具
|
|
#[tauri::command]
|
|
pub fn open_devtools(app_handle: AppHandle) {
|
|
if let Some(window) = app_handle.get_webview_window("main") {
|
|
if !window.is_devtools_open() {
|
|
window.open_devtools();
|
|
} else {
|
|
window.close_devtools();
|
|
}
|
|
}
|
|
}
|
|
|
|
/// 退出应用
|
|
#[tauri::command]
|
|
pub async fn exit_app() {
|
|
feat::quit().await;
|
|
}
|
|
|
|
/// 重启应用
|
|
#[tauri::command]
|
|
pub async fn restart_app() -> CmdResult<()> {
|
|
feat::restart_app().await;
|
|
Ok(())
|
|
}
|
|
|
|
/// 获取便携版标识
|
|
#[tauri::command]
|
|
pub fn get_portable_flag() -> CmdResult<bool> {
|
|
Ok(*dirs::PORTABLE_FLAG.get().unwrap_or(&false))
|
|
}
|
|
|
|
/// 获取应用目录
|
|
#[tauri::command]
|
|
pub fn get_app_dir() -> CmdResult<String> {
|
|
let app_home_dir = wrap_err!(dirs::app_home_dir())?.to_string_lossy().into();
|
|
Ok(app_home_dir)
|
|
}
|
|
|
|
/// 获取当前自启动状态
|
|
#[tauri::command]
|
|
pub fn get_auto_launch_status() -> CmdResult<bool> {
|
|
use crate::core::sysopt::Sysopt;
|
|
wrap_err!(Sysopt::global().get_launch_status())
|
|
}
|
|
|
|
/// 下载图标缓存
|
|
#[tauri::command]
|
|
pub async fn download_icon_cache(url: String, name: String) -> CmdResult<String> {
|
|
let icon_cache_dir = wrap_err!(dirs::app_home_dir())?.join("icons").join("cache");
|
|
let icon_path = icon_cache_dir.join(&name);
|
|
|
|
if icon_path.exists() {
|
|
return Ok(icon_path.to_string_lossy().into());
|
|
}
|
|
|
|
if !icon_cache_dir.exists() {
|
|
let _ = std::fs::create_dir_all(&icon_cache_dir);
|
|
}
|
|
|
|
let temp_path = icon_cache_dir.join(format!("{}.downloading", &name));
|
|
|
|
let response = wrap_err!(reqwest::get(&url).await)?;
|
|
|
|
let content_type = response
|
|
.headers()
|
|
.get(reqwest::header::CONTENT_TYPE)
|
|
.and_then(|v| v.to_str().ok())
|
|
.unwrap_or("");
|
|
|
|
let is_image = content_type.starts_with("image/");
|
|
|
|
let content = wrap_err!(response.bytes().await)?;
|
|
|
|
let is_html = content.len() > 15
|
|
&& (content.starts_with(b"<!DOCTYPE html")
|
|
|| content.starts_with(b"<html")
|
|
|| content.starts_with(b"<?xml"));
|
|
|
|
if is_image && !is_html {
|
|
{
|
|
let mut file = match std::fs::File::create(&temp_path) {
|
|
Ok(file) => file,
|
|
Err(_) => {
|
|
if icon_path.exists() {
|
|
return Ok(icon_path.to_string_lossy().into());
|
|
}
|
|
return Err("Failed to create temporary file".into());
|
|
}
|
|
};
|
|
|
|
wrap_err!(std::io::copy(&mut content.as_ref(), &mut file))?;
|
|
}
|
|
|
|
if !icon_path.exists() {
|
|
match std::fs::rename(&temp_path, &icon_path) {
|
|
Ok(_) => {}
|
|
Err(_) => {
|
|
let _ = temp_path.remove_if_exists().await;
|
|
if icon_path.exists() {
|
|
return Ok(icon_path.to_string_lossy().into());
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
let _ = temp_path.remove_if_exists().await;
|
|
}
|
|
|
|
Ok(icon_path.to_string_lossy().into())
|
|
} else {
|
|
let _ = temp_path.remove_if_exists().await;
|
|
Err(format!("下载的内容不是有效图片: {url}"))
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize, serde::Deserialize)]
|
|
pub struct IconInfo {
|
|
name: String,
|
|
previous_t: String,
|
|
current_t: String,
|
|
}
|
|
|
|
/// 复制图标文件
|
|
#[tauri::command]
|
|
pub async fn copy_icon_file(path: String, icon_info: IconInfo) -> CmdResult<String> {
|
|
let file_path = Path::new(&path);
|
|
|
|
let icon_dir = wrap_err!(dirs::app_home_dir())?.join("icons");
|
|
if !icon_dir.exists() {
|
|
let _ = fs::create_dir_all(&icon_dir).await;
|
|
}
|
|
let ext: String = match file_path.extension() {
|
|
Some(e) => e.to_string_lossy().into(),
|
|
None => "ico".into(),
|
|
};
|
|
|
|
let dest_path = icon_dir.join(format!(
|
|
"{0}-{1}.{ext}",
|
|
icon_info.name, icon_info.current_t
|
|
));
|
|
if file_path.exists() {
|
|
if icon_info.previous_t.trim() != "" {
|
|
icon_dir
|
|
.join(format!("{0}-{1}.png", icon_info.name, icon_info.previous_t))
|
|
.remove_if_exists()
|
|
.await
|
|
.unwrap_or_default();
|
|
icon_dir
|
|
.join(format!("{0}-{1}.ico", icon_info.name, icon_info.previous_t))
|
|
.remove_if_exists()
|
|
.await
|
|
.unwrap_or_default();
|
|
}
|
|
logging!(
|
|
info,
|
|
Type::Cmd,
|
|
"Copying icon file path: {:?} -> file dist: {:?}",
|
|
path,
|
|
dest_path
|
|
);
|
|
match fs::copy(file_path, &dest_path).await {
|
|
Ok(_) => Ok(dest_path.to_string_lossy().into()),
|
|
Err(err) => Err(err.to_string()),
|
|
}
|
|
} else {
|
|
Err("file not found".into())
|
|
}
|
|
}
|
|
|
|
/// 通知UI已准备就绪
|
|
#[tauri::command]
|
|
pub fn notify_ui_ready() -> CmdResult<()> {
|
|
log::info!(target: "app", "前端UI已准备就绪");
|
|
crate::utils::resolve::ui::mark_ui_ready();
|
|
Ok(())
|
|
}
|
|
|
|
/// UI加载阶段
|
|
#[tauri::command]
|
|
pub fn update_ui_stage(stage: String) -> CmdResult<()> {
|
|
log::info!(target: "app", "UI加载阶段更新: {stage}");
|
|
|
|
use crate::utils::resolve::ui::UiReadyStage;
|
|
|
|
let stage_enum = match stage.as_str() {
|
|
"NotStarted" => UiReadyStage::NotStarted,
|
|
"Loading" => UiReadyStage::Loading,
|
|
"DomReady" => UiReadyStage::DomReady,
|
|
"ResourcesLoaded" => UiReadyStage::ResourcesLoaded,
|
|
"Ready" => UiReadyStage::Ready,
|
|
_ => {
|
|
log::warn!(target: "app", "未知的UI加载阶段: {stage}");
|
|
return Err(format!("未知的UI加载阶段: {stage}"));
|
|
}
|
|
};
|
|
|
|
crate::utils::resolve::ui::update_ui_ready_stage(stage_enum);
|
|
Ok(())
|
|
}
|