mirror of
https://github.com/TommyFang2077/dsh-desktop.git
synced 2026-08-17 09:06:36 +08:00
Add README, plugin citations, and multi-platform release CI.
Ship screenshots and third-party notices for the desktop shell, and package Windows, macOS, deb, rpm, and Flatpak from GitHub Releases. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Tommy
383
src-tauri/src/lib.rs
Normal file
383
src-tauri/src/lib.rs
Normal file
@@ -0,0 +1,383 @@
|
||||
use std::io::Cursor;
|
||||
use std::path::PathBuf;
|
||||
use std::sync::{Arc, Mutex};
|
||||
use std::thread;
|
||||
use std::time::{Duration, Instant};
|
||||
|
||||
use dsh_core::clipboard::{
|
||||
file_from_bytes, filename_for_mime, files_from_paths, parse_uri_list, ClipboardFile,
|
||||
};
|
||||
use dsh_core::launcher::{DshLauncher, DshProcess, URL_TIMEOUT_SECONDS};
|
||||
use dsh_core::modlens::ensure_modlens;
|
||||
use dsh_core::paths::BundledPaths;
|
||||
use dsh_core::preset::ensure_anchored_standard;
|
||||
use dsh_core::updater::{mark_update_checked, update_check_due, update_dsh};
|
||||
use dsh_core::{APP_NAME, ENV_NO_UPDATE};
|
||||
use serde::Serialize;
|
||||
use tauri::{AppHandle, Emitter, Manager, WebviewUrl, WebviewWindowBuilder};
|
||||
use url::Url;
|
||||
|
||||
const INJECT: &str = concat!(
|
||||
include_str!("../../ui/inject/ingest.js"),
|
||||
include_str!("../../ui/inject/hide-twins.js"),
|
||||
include_str!("../../ui/inject/chrome.js"),
|
||||
);
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct Args {
|
||||
pub dsh: Option<String>,
|
||||
pub cwd: Option<PathBuf>,
|
||||
pub no_update: bool,
|
||||
pub force_update: bool,
|
||||
pub dev: bool,
|
||||
pub verbose: bool,
|
||||
}
|
||||
|
||||
impl Args {
|
||||
pub fn parse() -> Self {
|
||||
let mut args = Args {
|
||||
dsh: std::env::var(dsh_core::ENV_BIN_OVERRIDE).ok().filter(|s| !s.is_empty()),
|
||||
cwd: std::env::var(dsh_core::ENV_CWD_OVERRIDE)
|
||||
.ok()
|
||||
.filter(|s| !s.is_empty())
|
||||
.map(PathBuf::from),
|
||||
no_update: std::env::var(ENV_NO_UPDATE)
|
||||
.map(|v| matches!(v.to_ascii_lowercase().as_str(), "1" | "true" | "yes"))
|
||||
.unwrap_or(false),
|
||||
force_update: false,
|
||||
dev: false,
|
||||
verbose: false,
|
||||
};
|
||||
let mut argv = std::env::args().skip(1);
|
||||
while let Some(arg) = argv.next() {
|
||||
match arg.as_str() {
|
||||
"--dsh" => args.dsh = argv.next(),
|
||||
"--cwd" => args.cwd = argv.next().map(PathBuf::from),
|
||||
"--no-update" => args.no_update = true,
|
||||
"--update" => args.force_update = true,
|
||||
"--dev" => args.dev = true,
|
||||
"--verbose" => args.verbose = true,
|
||||
"--help" | "-h" => {
|
||||
print_help();
|
||||
std::process::exit(0);
|
||||
}
|
||||
"--version" | "-V" => {
|
||||
println!("{APP_NAME} {}", dsh_core::VERSION);
|
||||
std::process::exit(0);
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
args
|
||||
}
|
||||
}
|
||||
|
||||
fn print_help() {
|
||||
println!(
|
||||
"{APP_NAME} — native window around the official dsh WebUI.\n\n\
|
||||
--dsh PATH dsh 可执行文件路径或命令\n\
|
||||
--cwd DIR 传给 dsh 的工作目录\n\
|
||||
--no-update 不检查 dsh 更新\n\
|
||||
--update 启动前强制检查/安装 dsh 更新\n\
|
||||
--dev 打开 WebView 开发者工具\n\
|
||||
--verbose 输出调试日志"
|
||||
);
|
||||
}
|
||||
|
||||
struct AppState {
|
||||
args: Args,
|
||||
paths: BundledPaths,
|
||||
process: Mutex<Option<Arc<DshProcess>>>,
|
||||
url: Mutex<Option<String>>,
|
||||
}
|
||||
|
||||
impl AppState {
|
||||
fn stop(&self) {
|
||||
if let Ok(mut slot) = self.process.lock() {
|
||||
if let Some(proc) = slot.take() {
|
||||
proc.stop();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Serialize)]
|
||||
struct StatusPayload {
|
||||
message: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, Serialize)]
|
||||
struct ErrorPayload {
|
||||
message: String,
|
||||
detail: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, Serialize)]
|
||||
struct ReadyPayload {
|
||||
url: String,
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
fn restart(app: AppHandle) {
|
||||
thread::spawn(move || boot(app));
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
fn open_in_browser(state: tauri::State<AppState>) -> Result<(), String> {
|
||||
let url = state
|
||||
.url
|
||||
.lock()
|
||||
.ok()
|
||||
.and_then(|g| g.clone())
|
||||
.ok_or_else(|| "服务尚未就绪".to_string())?;
|
||||
open::that(&url).map_err(|e| e.to_string())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
fn read_clipboard_images() -> Result<Vec<ClipboardFile>, String> {
|
||||
read_images().map_err(|e| e.to_string())
|
||||
}
|
||||
|
||||
fn read_images() -> Result<Vec<ClipboardFile>, String> {
|
||||
let mut clipboard = arboard::Clipboard::new().map_err(|e| e.to_string())?;
|
||||
if let Ok(img) = clipboard.get_image() {
|
||||
let width = img.width as u32;
|
||||
let height = img.height as u32;
|
||||
let bytes = img.bytes.into_owned();
|
||||
if let Some(buffer) = image::RgbaImage::from_raw(width, height, bytes) {
|
||||
let mut png = Vec::new();
|
||||
if buffer
|
||||
.write_to(&mut Cursor::new(&mut png), image::ImageFormat::Png)
|
||||
.is_ok()
|
||||
{
|
||||
if let Some(file) = file_from_bytes(
|
||||
filename_for_mime("image/png", 0),
|
||||
"image/png".into(),
|
||||
png,
|
||||
) {
|
||||
return Ok(vec![file]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if let Ok(text) = clipboard.get_text() {
|
||||
let loaded = files_from_paths(parse_uri_list(&text));
|
||||
if !loaded.is_empty() {
|
||||
return Ok(loaded);
|
||||
}
|
||||
}
|
||||
Ok(Vec::new())
|
||||
}
|
||||
|
||||
fn is_internal(url: &Url) -> bool {
|
||||
matches!(url.scheme(), "tauri" | "asset" | "about" | "data" | "blob")
|
||||
|| matches!(
|
||||
url.host_str(),
|
||||
Some("127.0.0.1" | "localhost" | "::1" | "tauri.localhost")
|
||||
)
|
||||
}
|
||||
|
||||
fn boot(app: AppHandle) {
|
||||
let Some(state) = app.try_state::<AppState>() else {
|
||||
return;
|
||||
};
|
||||
state.stop();
|
||||
let started = Instant::now();
|
||||
let _ = app.emit(
|
||||
"status",
|
||||
StatusPayload {
|
||||
message: "正在启动…".into(),
|
||||
},
|
||||
);
|
||||
|
||||
if state.args.force_update && !state.args.no_update {
|
||||
let _ = app.emit(
|
||||
"status",
|
||||
StatusPayload {
|
||||
message: "正在检查 dsh 更新…".into(),
|
||||
},
|
||||
);
|
||||
let result = update_dsh(true);
|
||||
mark_update_checked();
|
||||
log::info!("dsh update: {} ({})", result.status, result.message);
|
||||
}
|
||||
|
||||
let plugin = ensure_modlens(&state.paths);
|
||||
log::info!(
|
||||
"modlens: {} ({}) in {:?}",
|
||||
plugin.status,
|
||||
plugin.message,
|
||||
started.elapsed()
|
||||
);
|
||||
let preset = ensure_anchored_standard(&state.paths);
|
||||
log::info!(
|
||||
"anchored-standard: {} ({}) in {:?}",
|
||||
preset.status,
|
||||
preset.message,
|
||||
started.elapsed()
|
||||
);
|
||||
let _ = app.emit(
|
||||
"status",
|
||||
StatusPayload {
|
||||
message: "正在启动 dsh web 服务…".into(),
|
||||
},
|
||||
);
|
||||
|
||||
let launcher = DshLauncher::new(state.args.dsh.clone(), state.args.cwd.clone());
|
||||
let process = match launcher.start() {
|
||||
Ok(p) => Arc::new(p),
|
||||
Err(err) => {
|
||||
let _ = app.emit(
|
||||
"error",
|
||||
ErrorPayload {
|
||||
message: err.to_string(),
|
||||
detail: String::new(),
|
||||
},
|
||||
);
|
||||
return;
|
||||
}
|
||||
};
|
||||
if let Ok(mut slot) = state.process.lock() {
|
||||
*slot = Some(Arc::clone(&process));
|
||||
}
|
||||
log::info!("dsh web spawned in {:?}", started.elapsed());
|
||||
|
||||
if !state.args.no_update && !state.args.force_update && update_check_due() {
|
||||
thread::spawn(|| {
|
||||
log::info!("background dsh update check");
|
||||
let result = update_dsh(true);
|
||||
mark_update_checked();
|
||||
log::info!(
|
||||
"background dsh update: {} ({})",
|
||||
result.status,
|
||||
result.message
|
||||
);
|
||||
});
|
||||
}
|
||||
|
||||
let deadline = Instant::now() + Duration::from_secs(URL_TIMEOUT_SECONDS);
|
||||
loop {
|
||||
if let Some(url) = process.take_url() {
|
||||
if let Ok(mut slot) = state.url.lock() {
|
||||
*slot = Some(url.clone());
|
||||
}
|
||||
let _ = app.emit("ready", ReadyPayload { url });
|
||||
log::info!("dsh web ready in {:?}", started.elapsed());
|
||||
return;
|
||||
}
|
||||
if let Some(code) = process.poll() {
|
||||
let detail = process.snapshot_lines().join("\n");
|
||||
let _ = app.emit(
|
||||
"error",
|
||||
ErrorPayload {
|
||||
message: format!("服务已退出（exit {code}）。"),
|
||||
detail,
|
||||
},
|
||||
);
|
||||
return;
|
||||
}
|
||||
if Instant::now() >= deadline {
|
||||
let detail = process
|
||||
.snapshot_lines()
|
||||
.into_iter()
|
||||
.rev()
|
||||
.take(80)
|
||||
.collect::<Vec<_>>()
|
||||
.into_iter()
|
||||
.rev()
|
||||
.collect::<Vec<_>>()
|
||||
.join("\n");
|
||||
process.stop();
|
||||
let _ = app.emit(
|
||||
"error",
|
||||
ErrorPayload {
|
||||
message: format!("等待 dsh web 输出 URL 超时（{URL_TIMEOUT_SECONDS} 秒）。"),
|
||||
detail,
|
||||
},
|
||||
);
|
||||
return;
|
||||
}
|
||||
thread::sleep(Duration::from_millis(80));
|
||||
}
|
||||
}
|
||||
|
||||
pub fn run() {
|
||||
let args = Args::parse();
|
||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or(if args.verbose {
|
||||
"debug"
|
||||
} else {
|
||||
"info"
|
||||
}))
|
||||
.init();
|
||||
|
||||
let mut paths = BundledPaths::discover();
|
||||
let dev = args.dev;
|
||||
|
||||
tauri::Builder::default()
|
||||
.plugin(tauri_plugin_opener::init())
|
||||
.plugin(tauri_plugin_single_instance::init(|app, _argv, _cwd| {
|
||||
if let Some(window) = app.get_webview_window("main") {
|
||||
let _ = window.unminimize();
|
||||
let _ = window.set_focus();
|
||||
}
|
||||
}))
|
||||
.setup(move |app| {
|
||||
if let Ok(dir) = app.path().resource_dir() {
|
||||
paths = paths.with_resource_dir(dir);
|
||||
}
|
||||
app.manage(AppState {
|
||||
args: args.clone(),
|
||||
paths,
|
||||
process: Mutex::new(None),
|
||||
url: Mutex::new(None),
|
||||
});
|
||||
|
||||
let mut builder = WebviewWindowBuilder::new(app, "main", WebviewUrl::App("index.html".into()))
|
||||
.title(APP_NAME)
|
||||
.inner_size(1320.0, 860.0)
|
||||
.min_inner_size(800.0, 560.0)
|
||||
.decorations(false)
|
||||
.resizable(true)
|
||||
.initialization_script(INJECT)
|
||||
.on_navigation(|url| {
|
||||
if is_internal(&url) {
|
||||
true
|
||||
} else {
|
||||
let _ = open::that(url.as_str());
|
||||
false
|
||||
}
|
||||
});
|
||||
|
||||
if let Some(icon) = app.default_window_icon().cloned() {
|
||||
builder = builder.icon(icon)?;
|
||||
}
|
||||
|
||||
if dev {
|
||||
builder = builder.devtools(true);
|
||||
}
|
||||
|
||||
let window = builder.build()?;
|
||||
if dev {
|
||||
window.open_devtools();
|
||||
}
|
||||
|
||||
let app_handle = app.handle().clone();
|
||||
let _ = window;
|
||||
thread::spawn(move || boot(app_handle));
|
||||
Ok(())
|
||||
})
|
||||
.invoke_handler(tauri::generate_handler![
|
||||
restart,
|
||||
open_in_browser,
|
||||
read_clipboard_images
|
||||
])
|
||||
.on_window_event(|window, event| {
|
||||
if let tauri::WindowEvent::CloseRequested { .. } = event {
|
||||
if let Some(state) = window.try_state::<AppState>() {
|
||||
state.stop();
|
||||
}
|
||||
}
|
||||
})
|
||||
.run(tauri::generate_context!())
|
||||
.expect("error while running DeepSeek Harness Desktop");
|
||||
}
|
||||
Reference in New Issue
Block a user