From 72f9ced533ba904f1a31346378cba47c1b43c2e8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=95=E6=9F=8F=E9=9D=92?= Date: Sun, 7 Jun 2026 13:05:53 +0800 Subject: [PATCH] =?UTF-8?q?refactor(commands):=20commands.rs=20=E6=8B=86?= =?UTF-8?q?=E4=B8=BA=20commands/=20=E5=9F=9F=E6=A8=A1=E5=9D=97=EF=BC=88glo?= =?UTF-8?q?b=20=E9=87=8D=E5=AF=BC=E5=87=BA=EF=BC=8Clib.rs=20=E9=9B=B6?= =?UTF-8?q?=E6=94=B9=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 把 4857 行的单文件 commands.rs 按单一职责拆成 commands/ 下 16 个域模块 （settings / credentials / providers / history / dictionary / dictation / style_packs / permissions_cmds / qa / hotkeys / local_asr / foundry_asr / sherpa_asr / marketplace / github_oauth / misc）。纯机械式、行为保持的剪切 + 可见性提升，未改动任何 #[tauri::command] 函数体。 - commands.rs → commands/mod.rs：内含 `mod ;` + `pub use ::*;`， glob 重导出每个域模块。`commands::` 路径全部不变 → lib.rs 的 generate_handler! 清单（346-492）与所有类型/helper 引用零改动。 #[tauri::command] 生成的 __cmd__ 伴生项随 glob 一并重导出。 - 跨域共享项提到 mod.rs 并 pub(crate)：is_valid_session_id / is_valid_local_pack_id、open_path_in_file_manager（3 个 cfg 变体）、 共享状态类型（MicrophoneMonitorState / TrayMicrophoneMenuState 等）、 常用 crate 级 import 以 `pub(crate) use` 重导出供子模块 `use super::*` 取用。 - MARKETPLACE_BASE_URL 提为 pub(crate)（misc::check_network 引用）； hotkeys.rs 的 reject_* 整套 + sync_dictation_hotkey_legacy_fields 保留在 hotkeys.rs 但提 pub(crate)（被 settings::persist_settings、qa::set_qa_hotkey 引用）； style_packs.rs 的 activate_builtin_style_mode 等保留并 pub(crate)（lib.rs 引用）。 - 测试：保留单个 tests mod 在 mod.rs，把 `use super::{...}` 改为 per-module 路径，被测 helper 提为 pub(crate)。不改任何测试逻辑。 - sherpa_onnx_asr_* 整组 #[cfg(windows)]，非 Windows 平台 glob 重导出无人引用， 对该 `pub use sherpa_asr::*` 加 #[allow(unused_imports)]。 验证：cargo check 0 error；cargo test --lib 417 passed（基线不回归）； npm run build 通过；capsule Spaces / hotkey-injection 契约测试通过。 命令面 127 个 #[tauri::command] 与原文件逐一对齐，无新增/缺失/重复。 Co-Authored-By: Claude Opus 4.8 (1M context) --- openless-all/app/src-tauri/src/commands.rs | 4857 ----------------- .../app/src-tauri/src/commands/credentials.rs | 227 + .../app/src-tauri/src/commands/dictation.rs | 49 + .../app/src-tauri/src/commands/dictionary.rs | 79 + .../app/src-tauri/src/commands/foundry_asr.rs | 190 + .../src-tauri/src/commands/github_oauth.rs | 154 + .../app/src-tauri/src/commands/history.rs | 50 + .../app/src-tauri/src/commands/hotkeys.rs | 408 ++ .../app/src-tauri/src/commands/local_asr.rs | 309 ++ .../app/src-tauri/src/commands/marketplace.rs | 404 ++ .../app/src-tauri/src/commands/misc.rs | 123 + .../app/src-tauri/src/commands/mod.rs | 1364 +++++ .../src/commands/permissions_cmds.rs | 100 + .../app/src-tauri/src/commands/providers.rs | 496 ++ openless-all/app/src-tauri/src/commands/qa.rs | 67 + .../app/src-tauri/src/commands/settings.rs | 412 ++ .../app/src-tauri/src/commands/sherpa_asr.rs | 192 + .../app/src-tauri/src/commands/style_packs.rs | 287 + 18 files changed, 4911 insertions(+), 4857 deletions(-) delete mode 100644 openless-all/app/src-tauri/src/commands.rs create mode 100644 openless-all/app/src-tauri/src/commands/credentials.rs create mode 100644 openless-all/app/src-tauri/src/commands/dictation.rs create mode 100644 openless-all/app/src-tauri/src/commands/dictionary.rs create mode 100644 openless-all/app/src-tauri/src/commands/foundry_asr.rs create mode 100644 openless-all/app/src-tauri/src/commands/github_oauth.rs create mode 100644 openless-all/app/src-tauri/src/commands/history.rs create mode 100644 openless-all/app/src-tauri/src/commands/hotkeys.rs create mode 100644 openless-all/app/src-tauri/src/commands/local_asr.rs create mode 100644 openless-all/app/src-tauri/src/commands/marketplace.rs create mode 100644 openless-all/app/src-tauri/src/commands/misc.rs create mode 100644 openless-all/app/src-tauri/src/commands/mod.rs create mode 100644 openless-all/app/src-tauri/src/commands/permissions_cmds.rs create mode 100644 openless-all/app/src-tauri/src/commands/providers.rs create mode 100644 openless-all/app/src-tauri/src/commands/qa.rs create mode 100644 openless-all/app/src-tauri/src/commands/settings.rs create mode 100644 openless-all/app/src-tauri/src/commands/sherpa_asr.rs create mode 100644 openless-all/app/src-tauri/src/commands/style_packs.rs diff --git a/openless-all/app/src-tauri/src/commands.rs b/openless-all/app/src-tauri/src/commands.rs deleted file mode 100644 index 7a2e12b9..00000000 --- a/openless-all/app/src-tauri/src/commands.rs +++ /dev/null @@ -1,4857 +0,0 @@ -#![cfg_attr(target_os = "linux", allow(dead_code, unused_variables))] -//! Tauri command surface — every IPC entry the React UI invokes lives here. - -use std::sync::Arc; - -use parking_lot::Mutex; -use serde::Serialize; -use serde_json::Value; -use tauri::{AppHandle, Emitter, Manager, State, Window}; - -use crate::asr::local::foundry::{ - model_alias_is_known, FoundryCatalogModel, FoundryPrepareProgressPayload, FoundryRuntimeStatus, - DEFAULT_MODEL_ALIAS, PROVIDER_ID as FOUNDRY_LOCAL_PROVIDER_ID, -}; -use crate::asr::local::sherpa::{ - model_alias_is_known as sherpa_model_alias_is_known, SherpaCatalogModel, - SherpaPrepareProgressPayload, SherpaRuntimeStatus, - DEFAULT_MODEL_ALIAS as SHERPA_DEFAULT_MODEL_ALIAS, -}; -use crate::asr::local::sherpa_download::{ - fetch_remote_info as fetch_sherpa_remote_info, SherpaDownloadManager, SherpaRemoteInfo, -}; -use crate::asr::local::{FoundryLocalRuntime, SherpaOnnxRuntime}; -use crate::coordinator::Coordinator; -use crate::net; -use crate::permissions::{self, PermissionStatus}; -use crate::persistence::{ - sync_style_pack_preferences, CredentialAccount, CredentialsSnapshot, CredentialsVault, - PreferencesStore, -}; -use crate::polish::{ - http_client_builder, CodexOAuthConfig, CodexOAuthCredentials, CodexOAuthLLMProvider, LLMError, - OpenAICompatibleConfig, OpenAICompatibleLLMProvider, CODEX_DEFAULT_MODEL, - CODEX_OAUTH_PROVIDER_ID, -}; -use crate::recorder::{AudioConsumer, Recorder}; -use crate::types::{ - builtin_style_pack_id, default_active_style_pack_id, ChineseScriptPreference, ComboBinding, - CorrectionRule, CredentialsStatus, DictationSession, DictionaryEntry, HotkeyCapability, - HotkeyStatus, OutputLanguagePreference, PolishMode, ShortcutBinding, StylePack, StylePackKind, - StylePackRuntimeDiagnostics, StyleSystemPrompts, UpdateChannel, UserPreferences, - VocabPresetStore, WindowsImeStatus, -}; - -type CoordinatorState<'a> = State<'a, Arc>; -pub type MicrophoneMonitorState = Mutex>; -pub type TrayMicrophoneMenuState = Mutex>; - -pub struct TrayMicrophoneMenuItem { - pub id: String, - pub device_name: String, - pub item: tauri::menu::CheckMenuItem, -} - -pub fn sync_tray_microphone_selection(items: &[TrayMicrophoneMenuItem], device_name: &str) { - for item in items { - let _ = item.item.set_checked(item.device_name == device_name); - } -} - -struct LevelProbeConsumer; - -impl AudioConsumer for LevelProbeConsumer { - fn consume_pcm_chunk(&self, _pcm: &[u8]) {} -} - -// ─────────────────────────── settings + credentials ─────────────────────────── - -#[tauri::command] -pub fn get_settings(coord: CoordinatorState<'_>) -> UserPreferences { - coord.prefs().get() -} - -#[tauri::command] -pub fn get_default_style_system_prompts() -> StyleSystemPrompts { - StyleSystemPrompts::default() -} - -trait SettingsWriter { - fn read_settings(&self) -> UserPreferences; - fn write_settings(&self, prefs: UserPreferences) -> Result<(), String>; - fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String>; - fn refresh_dictation_hotkey(&self); - fn refresh_qa_hotkey(&self); - fn refresh_combo_hotkey(&self); - fn refresh_translation_hotkey(&self); - fn refresh_switch_style_hotkey(&self); - fn refresh_open_app_hotkey(&self); - fn refresh_coding_agent_hotkey(&self); -} - -impl SettingsWriter for Coordinator { - fn read_settings(&self) -> UserPreferences { - self.prefs().get() - } - - fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { - self.prefs().set(prefs).map_err(|e| e.to_string()) - } - - fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { - self.sync_active_asr_provider_to_vault(provider) - } - - fn refresh_dictation_hotkey(&self) { - self.update_hotkey_binding(); - } - - fn refresh_qa_hotkey(&self) { - self.update_qa_hotkey_binding(); - } - - fn refresh_combo_hotkey(&self) { - self.update_combo_hotkey_binding(); - } - - fn refresh_translation_hotkey(&self) { - self.update_translation_hotkey_binding(); - } - - fn refresh_switch_style_hotkey(&self) { - self.update_switch_style_hotkey_binding(); - } - - fn refresh_open_app_hotkey(&self) { - self.update_open_app_hotkey_binding(); - } - - fn refresh_coding_agent_hotkey(&self) { - self.update_coding_agent_hotkey_binding(); - } -} - -impl SettingsWriter for Arc { - fn read_settings(&self) -> UserPreferences { - (**self).read_settings() - } - - fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { - (**self).write_settings(prefs) - } - - fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { - (**self).sync_active_asr_provider(provider) - } - - fn refresh_dictation_hotkey(&self) { - (**self).refresh_dictation_hotkey(); - } - - fn refresh_qa_hotkey(&self) { - (**self).refresh_qa_hotkey(); - } - - fn refresh_combo_hotkey(&self) { - (**self).refresh_combo_hotkey(); - } - - fn refresh_translation_hotkey(&self) { - (**self).refresh_translation_hotkey(); - } - - fn refresh_switch_style_hotkey(&self) { - (**self).refresh_switch_style_hotkey(); - } - - fn refresh_open_app_hotkey(&self) { - (**self).refresh_open_app_hotkey(); - } - - fn refresh_coding_agent_hotkey(&self) { - (**self).refresh_coding_agent_hotkey(); - } -} - -fn persist_settings( - coord: &T, - mut prefs: UserPreferences, -) -> Result<(), String> { - let mut previous = coord.read_settings(); - sync_dictation_hotkey_legacy_fields(&mut previous); - sync_dictation_hotkey_legacy_fields(&mut prefs); - reject_hotkey_collisions(&prefs)?; - let dictation_shortcut_changed = previous.dictation_hotkey != prefs.dictation_hotkey; - let dictation_mode_changed = previous.hotkey.mode != prefs.hotkey.mode; - let qa_changed = previous.qa_hotkey != prefs.qa_hotkey; - let translation_changed = previous.translation_hotkey != prefs.translation_hotkey; - let switch_style_changed = previous.switch_style_hotkey != prefs.switch_style_hotkey; - let open_app_changed = previous.open_app_hotkey != prefs.open_app_hotkey; - let coding_agent_changed = previous.coding_agent_enabled != prefs.coding_agent_enabled - || previous.coding_agent_voice_hotkey != prefs.coding_agent_voice_hotkey; - let active_asr_provider_changed = previous.active_asr_provider != prefs.active_asr_provider; - let active_asr_provider = prefs.active_asr_provider.clone(); - if active_asr_provider_changed { - coord.sync_active_asr_provider(&active_asr_provider)?; - } - if let Err(error) = coord.write_settings(prefs.clone()) { - if active_asr_provider_changed { - if let Err(rollback_error) = - coord.sync_active_asr_provider(&previous.active_asr_provider) - { - coord.write_settings(prefs).map_err(|roll_forward_error| { - format!( - "{error}; additionally failed to restore active ASR provider: {rollback_error}; additionally failed to preserve active ASR provider consistency: {roll_forward_error}" - ) - })?; - } else { - return Err(error); - } - } else { - return Err(error); - } - } - if dictation_shortcut_changed || dictation_mode_changed { - coord.refresh_dictation_hotkey(); - } - if dictation_shortcut_changed { - coord.refresh_combo_hotkey(); - } - if qa_changed { - coord.refresh_qa_hotkey(); - } - if translation_changed { - coord.refresh_translation_hotkey(); - } - if switch_style_changed { - coord.refresh_switch_style_hotkey(); - } - if open_app_changed { - coord.refresh_open_app_hotkey(); - } - if coding_agent_changed { - coord.refresh_coding_agent_hotkey(); - } - Ok(()) -} - -#[tauri::command] -pub fn set_settings( - coord: CoordinatorState<'_>, - app: AppHandle, - tray_microphones: State<'_, TrayMicrophoneMenuState>, - mut prefs: UserPreferences, -) -> Result<(), String> { - let packs = coord.style_packs().list().map_err(|e| e.to_string())?; - sync_style_pack_preferences(&mut prefs, &packs); - // 广播给所有 webview。issue #205：QaPanel 跑在独立 webview， - // 没有 HotkeySettingsContext，必须靠事件感知录音键变化，否则面板可见时 - // 用户改键会让浮窗里的 "{recordHotkey}" 文案一直停留在旧值。 - persist_settings(&*coord, prefs.clone())?; - // refresh_tray_microphone_menu 内部会调用 NSStatusItem.set_menu，必须在主线程上跑。 - // set_settings 本身是同步 Tauri command，在 IPC handler 线程上执行；从这里直接调 - // 会触发 macOS 主线程断言或在 dispatch 队列上死锁，导致整个 UI 无响应（用户改 - // 偏好后所有按键都没反应即此根因）。dispatch 到主线程后立即返回，IPC 线程不阻塞。 - let app_for_main = app.clone(); - let prefs_for_main = prefs.clone(); - let _ = app.run_on_main_thread(move || { - if let Err(err) = crate::refresh_tray_microphone_menu(&app_for_main) { - log::warn!("[tray] refresh microphone menu after settings save failed: {err}"); - let tray_state = app_for_main.state::(); - sync_tray_microphone_selection( - &tray_state.lock(), - &prefs_for_main.microphone_device_name, - ); - } - }); - // 抑制 unused 警告：tray_microphones 现在改在闭包里通过 app.state 取， - // 但函数签名保留 State 入参，以便 Tauri 在调用前注入。 - let _ = tray_microphones; - let _ = app.emit("prefs:changed", &prefs); - Ok(()) -} - -fn refresh_tray_menu_async(app: &AppHandle) { - let app_for_main = app.clone(); - let _ = app.run_on_main_thread(move || { - if let Err(err) = crate::refresh_tray_microphone_menu(&app_for_main) { - log::warn!("[tray] refresh after style change failed: {err}"); - } - }); -} - -fn emit_prefs_changed(app: &AppHandle, prefs: &UserPreferences) { - let _ = app.emit("prefs:changed", prefs); - let _ = app.emit_to("main", "prefs:changed", prefs); -} - -pub(crate) fn sync_style_pack_prefs_and_persist( - coord: &Coordinator, - app: &AppHandle, - mut prefs: UserPreferences, -) -> Result { - let packs = coord.style_packs().list().map_err(|e| e.to_string())?; - sync_style_pack_preferences(&mut prefs, &packs); - coord - .prefs() - .set(prefs.clone()) - .map_err(|e| e.to_string())?; - emit_prefs_changed(app, &prefs); - refresh_tray_menu_async(app); - Ok(prefs) -} - -pub(crate) fn activate_style_pack_by_id( - coord: &Coordinator, - app: &AppHandle, - id: &str, -) -> Result { - let mut prefs = coord.prefs().get(); - let pack = coord.style_packs().get(id).map_err(|e| e.to_string())?; - log::info!( - "[style-pack] activate helper requested id={} kind={:?} base_mode={:?} enabled={}", - pack.id, - pack.kind, - pack.base_mode, - pack.enabled - ); - if !pack.enabled { - coord - .style_packs() - .set_enabled(id, true) - .map_err(|e| e.to_string())?; - } - prefs.active_style_pack_id = id.to_string(); - sync_style_pack_prefs_and_persist(coord, app, prefs)?; - log::info!("[style-pack] activate helper applied id={id}"); - coord - .style_packs() - .get(id) - .map(|mut pack| { - pack.active = true; - pack - }) - .map_err(|e| e.to_string()) -} - -pub(crate) fn activate_builtin_style_mode( - coord: &Coordinator, - app: &AppHandle, - mode: PolishMode, -) -> Result<(), String> { - let pack_id = builtin_style_pack_id(mode).to_string(); - log::info!( - "[style-pack] activate builtin mode helper mode={:?} pack_id={}", - mode, - pack_id - ); - let _ = activate_style_pack_by_id(coord, app, &pack_id)?; - Ok(()) -} - -// ─────────────────────────── release channel (Beta opt-in) ─────────────────────────── -// -// 渠道偏好的写入路径跟 set_settings 复用 persist_settings：保持热键兜底归一化 -// 跟其他 prefs 写入一致，且写完后 emit "prefs:changed"，让前端跨 webview 同步。 -// -// 更新：plugin-updater 2.10.1 的 Builder 现在暴露 .endpoints() runtime API（CLAUDE.md -// 当年记的"不支持"已不成立）。本节配合 `app_check_update_with_channel` 命令实现 -// Beta auto-update：Stable 渠道 → 走 tauri.conf 的默认 endpoints；Beta 渠道 → -// fetch_latest_beta_release 拿最新 prerelease tag → 拼成 -beta manifest URL → -// builder.endpoints(vec![url]).build().check()。Stable 用户绝对不会撞到 Beta 包 -// （Beta tag 的 manifest 文件名带 `-beta` 后缀，跟 Stable manifest 在 GitHub -// Release assets 里物理分离）。 - -#[tauri::command] -pub fn get_update_channel(coord: CoordinatorState<'_>) -> UpdateChannel { - coord.prefs().get().update_channel -} - -#[tauri::command] -pub fn set_update_channel( - coord: CoordinatorState<'_>, - app: AppHandle, - channel: UpdateChannel, -) -> Result<(), String> { - let mut prefs = coord.prefs().get(); - if prefs.update_channel == channel { - return Ok(()); - } - prefs.update_channel = channel; - persist_settings(&*coord, prefs.clone())?; - let _ = app.emit("prefs:changed", &prefs); - Ok(()) -} - -#[derive(Debug, Clone, Serialize)] -#[serde(rename_all = "camelCase")] -pub struct LatestBetaRelease { - pub tag_name: String, - pub html_url: String, - pub published_at: String, -} - -/// 拉 GitHub Releases atom feed 找最新 Beta release（tag 以 `-beta-tauri` 结尾）。 -/// -/// 历史：之前用 `api.github.com/repos/.../releases` REST 端点，**未认证 60 req/h/IP**， -/// 多人多次切 Beta toggle 很容易撞 403 rate limit（用户报"获取 Beta 版本信息失败" -/// 即是这个）。换成 `releases.atom` 后是公开页面 + CDN cache，没有同等 rate 限制。 -/// Atom feed 不显式标 prerelease，但项目约定 tag 后缀 `-beta-tauri` 必为 Beta， -/// 所以只用 tag 后缀过滤就够了。 -/// -/// 返回 `Ok(None)` = 当前没发过 Beta 版；`Err(String)` = 网络/解析故障。 -#[tauri::command] -pub async fn fetch_latest_beta_release() -> Result, String> { - let resp = net::send_with_retry(|| { - net::http() - .get("https://github.com/appergb/openless/releases.atom") - .timeout(std::time::Duration::from_secs(15)) - }) - .await - .map_err(|e| format!("fetch releases.atom: {e}"))?; - if !resp.status().is_success() { - return Err(format!("releases.atom status {}", resp.status())); - } - let body = resp - .text() - .await - .map_err(|e| format!("read atom body: {e}"))?; - Ok(parse_latest_beta_from_atom(&body)) -} - -/// 简单字符串解析 atom feed，避免引 XML 库。每个 `...` 内含一行 -/// ``， -/// 用 `/releases/tag/` 这个唯一锚点抓 tag。 -fn parse_latest_beta_from_atom(body: &str) -> Option { - for entry in body.split("").skip(1) { - let entry_body = entry - .split_once("") - .map(|(b, _)| b) - .unwrap_or(entry); - let needle = "/releases/tag/"; - let tag_start = match entry_body.find(needle) { - Some(i) => i + needle.len(), - None => continue, - }; - let tag_after = &entry_body[tag_start..]; - let tag_end = tag_after - .find(|c: char| c == '"' || c == '<' || c == ' ' || c == '/') - .unwrap_or(tag_after.len()); - let tag_name = tag_after[..tag_end].to_string(); - if !tag_name.ends_with("-beta-tauri") { - continue; - } - let html_url = format!("https://github.com/appergb/openless/releases/tag/{tag_name}"); - let published_at = - extract_between(entry_body, "", "").unwrap_or_default(); - return Some(LatestBetaRelease { - tag_name, - html_url, - published_at, - }); - } - None -} - -fn extract_between(haystack: &str, open: &str, close: &str) -> Option { - let start = haystack.find(open)? + open.len(); - let end = haystack[start..].find(close)?; - Some(haystack[start..start + end].to_string()) -} - -// ─────────────────────── Channel-aware updater check ──────────────────────── -// -// 替换前端原来直接 import('@tauri-apps/plugin-updater').check() 的路径： -// - Stable 渠道：builder 不动 endpoints，沿用 tauri.conf 配的 stable manifest URL。 -// - Beta 渠道：先 fetch_latest_beta_release 拿最新 prerelease tag，拼成 -beta manifest -// URL（同时给一对 mirror + direct），再 builder.endpoints(vec![url])?.build()?.check()。 -// -// 返回的 Metadata 形状与 plugin-updater 的 JS UpdateMetadata 完全一致（rid + -// currentVersion 等驼峰字段），前端可以直接 `new Update(metadata)` 复用 plugin -// 的 download / install / close 实现，无需我们自己写下载和签名校验。 -// -// 物理隔离：Beta tag 推出来的 manifest 文件名带 `-beta` 后缀（参见 release-tauri.yml -// 第 382 行注释），跟 Stable 的 `latest-{tgt}-{arch}.json` 在 GitHub Release assets -// 里是分开的两份文件 —— 即使代码逻辑写错把 Beta URL 传给 Stable 用户，HTTP 也是 -// 直接 404，绝不会拿到错档。 - -#[derive(Debug, Clone, Serialize)] -#[serde(rename_all = "camelCase")] -pub struct AppUpdateMetadata { - pub rid: tauri::ResourceId, - pub current_version: String, - pub version: String, - pub date: Option, - pub body: Option, - /// 原始 manifest JSON——`new Update(metadata)` 在 JS 那边会校验它存在； - /// 我们透传 plugin 自己 check 时拿到的字段。 - pub raw_json: serde_json::Value, -} - -/// 决定 manifest 来源后走 plugin-updater 的标准 check 流程。 -/// 渠道：显式传入 `channel` 时用它（关于页固定查 Stable、高级页 Beta 区查 Beta）； -/// 不传则回落到 `prefs.update_channel`（后台 AutoUpdateGate 自动检查走这条）。 -/// 返回 None = 当前是最新；Some(metadata) = 有新版可装。 -#[tauri::command] -pub async fn app_check_update_with_channel( - coord: CoordinatorState<'_>, - webview: tauri::Webview, - timeout_ms: Option, - channel: Option, -) -> Result, String> { - use tauri_plugin_updater::UpdaterExt; - - let channel = channel.unwrap_or_else(|| coord.prefs().get().update_channel); - let mut builder = webview.updater_builder(); - if let Some(ms) = timeout_ms { - builder = builder.timeout(std::time::Duration::from_millis(ms)); - } - if matches!(channel, UpdateChannel::Beta) { - let urls = resolve_beta_manifest_endpoints().await?; - builder = builder - .endpoints(urls) - .map_err(|e| format!("set beta endpoints: {e}"))?; - } - let updater = builder.build().map_err(|e| format!("build updater: {e}"))?; - let update = updater - .check() - .await - .map_err(|e| format!("check update failed: {e}"))?; - - let Some(update) = update else { - return Ok(None); - }; - // date 字段透传需要引 time crate；前端 AutoUpdate.tsx 实际并不用 date，所以这里 - // 直接置 None，避免拉一个新 dep 进 src-tauri/Cargo.toml。 - let metadata = AppUpdateMetadata { - current_version: update.current_version.clone(), - version: update.version.clone(), - date: None, - body: update.body.clone(), - raw_json: update.raw_json.clone(), - rid: webview.resources_table().add(update), - }; - Ok(Some(metadata)) -} - -/// 把 fetch_latest_beta_release 找到的最新 prerelease tag 拼成 -beta manifest URL 对。 -/// 顺序：先镜像（fastgit.cc 代理 GitHub），后直连 —— 跟 tauri.conf 现有 Stable -/// endpoints 一致，让国内访问优先打到 CDN。 -async fn resolve_beta_manifest_endpoints() -> Result, String> { - let Some(latest) = fetch_latest_beta_release().await? else { - return Err("尚未发布过 Beta 版本".to_string()); - }; - let tag = latest.tag_name; - // {{target}} / {{arch}} 占位符由 plugin 在 check 时替换。Rust raw string 用 r#""# - // 不需要转义双花括号，比 format! 干净。 - let mirror = format!( - "https://fastgit.cc/https://github.com/appergb/openless/releases/download/{tag}/latest-{{{{target}}}}-{{{{arch}}}}-beta-mirror.json" - ); - let direct = format!( - "https://github.com/appergb/openless/releases/download/{tag}/latest-{{{{target}}}}-{{{{arch}}}}-beta.json" - ); - let mirror_url = url::Url::parse(&mirror).map_err(|e| format!("parse beta mirror url: {e}"))?; - let direct_url = url::Url::parse(&direct).map_err(|e| format!("parse beta direct url: {e}"))?; - Ok(vec![mirror_url, direct_url]) -} - -#[derive(Debug, Clone, Serialize)] -#[serde(rename_all = "camelCase")] -pub struct NetworkCheckResult { - pub online: bool, - pub latency_ms: Option, -} - -#[tauri::command] -pub async fn check_network() -> NetworkCheckResult { - // 探一个真实存在的接口。旧逻辑探 `/health` —— 实测返回 404，链路正常也永远判 - // 离线；且用 HEAD（后端只挂 GET）。改成 GET `/packs`，拿到任意 HTTP 响应即算通。 - // - // 单发、不走 send_with_retry：这是每 30s 跑一次的状态探针，要的是「快」。10 次 - // 退避重试会让被过滤 / 黑洞的网络下探测拖到近一分钟、状态灯像卡死。偶发的瞬时 - // 误判由下一个 30s 周期自动纠正。仍用 net::http() 共享连接池。 - let url = format!("{MARKETPLACE_BASE_URL}/packs?limit=1"); - let start = std::time::Instant::now(); - match net::http() - .get(&url) - .timeout(std::time::Duration::from_secs(8)) - .send() - .await - { - Ok(_) => NetworkCheckResult { - online: true, - latency_ms: Some(start.elapsed().as_millis() as u64), - }, - Err(_) => NetworkCheckResult { - online: false, - latency_ms: None, - }, - } -} - -#[tauri::command] -pub fn get_hotkey_status(coord: CoordinatorState<'_>) -> HotkeyStatus { - coord.hotkey_status() -} - -#[tauri::command] -pub fn get_hotkey_capability(coord: CoordinatorState<'_>) -> HotkeyCapability { - coord.hotkey_capability() -} - -#[tauri::command] -pub fn set_shortcut_recording_active(coord: CoordinatorState<'_>, active: bool) { - coord.set_shortcut_recording_active(active); -} - -#[tauri::command] -pub fn get_windows_ime_status() -> WindowsImeStatus { - crate::windows_ime_profile::get_windows_ime_status() -} - -#[tauri::command] -pub fn list_microphone_devices() -> Result, String> { - crate::recorder::list_input_devices().map_err(|e| e.to_string()) -} - -#[tauri::command] -pub async fn start_microphone_level_monitor( - app: AppHandle, - device_name: String, -) -> Result<(), String> { - tauri::async_runtime::spawn_blocking(move || { - let state = app.state::(); - if let Some(existing) = state.lock().take() { - existing.stop(); - } - - let selected = device_name.trim().to_string(); - let microphone_device_name = if selected.is_empty() { - None - } else { - Some(selected) - }; - let consumer: Arc = Arc::new(LevelProbeConsumer); - let level_app = app.clone(); - let level_handler: Arc = Arc::new(move |level| { - let _ = level_app.emit("microphone:level", serde_json::json!({ "level": level })); - }); - let (recorder, _runtime_errors, _archive_active) = - Recorder::start(microphone_device_name, consumer, level_handler, None) - .map_err(|e| e.to_string())?; - *state.lock() = Some(recorder); - Ok(()) - }) - .await - .map_err(|e| format!("start microphone monitor task failed: {e}"))? -} - -#[tauri::command] -pub async fn stop_microphone_level_monitor(app: AppHandle) { - let _ = tauri::async_runtime::spawn_blocking(move || { - let state = app.state::(); - let recorder = state.lock().take(); - if let Some(recorder) = recorder { - recorder.stop(); - } - }) - .await; -} - -#[tauri::command] -pub fn get_credentials() -> CredentialsStatus { - let snap = CredentialsVault::snapshot(); - let active_asr_provider = CredentialsVault::get_active_asr(); - let active_llm_provider = CredentialsVault::get_active_llm(); - let volcengine_configured = volcengine_configured(&snap); - let asr_configured = asr_configured_for_provider(&active_asr_provider, &snap); - let llm_configured = llm_configured_for_provider(&active_llm_provider, &snap); - CredentialsStatus { - active_asr_provider, - active_llm_provider, - asr_configured, - llm_configured, - volcengine_configured, - ark_configured: llm_configured, - } -} - -fn volcengine_configured(snap: &CredentialsSnapshot) -> bool { - configured(&snap.volcengine_app_key) - && configured(&snap.volcengine_access_key) - && configured(&snap.volcengine_resource_id) -} - -fn asr_configured_for_provider(provider: &str, snap: &CredentialsSnapshot) -> bool { - if provider == "volcengine" { - return volcengine_configured(snap); - } - if provider == crate::asr::local::PROVIDER_ID - || active_foundry_asr_is_supported(provider) - || active_sherpa_asr_is_supported(provider) - { - // 本地 ASR 不依赖云端凭据。 - return true; - } - if provider == crate::asr::bailian::PROVIDER_ID { - return configured(&snap.asr_api_key); - } - if provider == crate::asr::mimo::PROVIDER_ID { - return configured(&snap.asr_api_key) - && configured(&snap.asr_endpoint) - && configured(&snap.asr_model); - } - configured(&snap.asr_endpoint) && configured(&snap.asr_model) -} - -fn llm_configured_for_provider(provider: &str, snap: &CredentialsSnapshot) -> bool { - if provider == CODEX_OAUTH_PROVIDER_ID { - return CodexOAuthCredentials::load_default().is_ok(); - } - let endpoint = snap.ark_endpoint.as_deref().unwrap_or_default(); - let endpoint_and_model = configured(&snap.ark_endpoint) && configured(&snap.ark_model_id); - if endpoint_and_model - && llm_provider_default_endpoint(provider) - .map(|default| same_llm_endpoint(endpoint, default)) - .unwrap_or(false) - { - return configured(&snap.ark_api_key); - } - endpoint_and_model -} - -fn llm_provider_default_endpoint(provider: &str) -> Option<&'static str> { - match provider { - "ark" => Some("https://ark.cn-beijing.volces.com/api/v3"), - "deepseek" => Some("https://api.deepseek.com/v1"), - "siliconflow" => Some("https://api.siliconflow.cn/v1"), - "openai" => Some("https://api.openai.com/v1"), - // 谷歌 Gemini 原生 API（v1beta）。后端 llm_gemini.rs 会拼成 - // `{baseUrl}/models/{model}:generateContent`，认证用 x-goog-api-key 头。 - "gemini" => Some("https://generativelanguage.googleapis.com/v1beta"), - "mimo" => Some("https://api.xiaomimimo.com/v1"), - "cometapi" => Some("https://api.cometapi.com/v1"), - "openrouterFree" => Some("https://openrouter.ai/api/v1"), - "alibabaCoding" => Some("https://coding-intl.dashscope.aliyuncs.com/v1"), - "codingPlanX" => Some("https://api.codingplanx.ai/v1"), - _ => None, - } -} - -fn same_llm_endpoint(a: &str, b: &str) -> bool { - fn normalize(value: &str) -> &str { - value - .trim() - .trim_end_matches('/') - .trim_end_matches("/chat/completions") - .trim_end_matches('/') - } - normalize(a).eq_ignore_ascii_case(normalize(b)) -} - -fn configured(field: &Option) -> bool { - field - .as_ref() - .map(|s| !s.trim().is_empty()) - .unwrap_or(false) -} - -#[derive(Debug, Clone, Copy, PartialEq, Eq)] -struct LocalAsrReleasePlan { - qwen: bool, - foundry: bool, - sherpa: bool, -} - -fn local_asr_release_plan_for_provider(provider: &str) -> LocalAsrReleasePlan { - LocalAsrReleasePlan { - qwen: provider != crate::asr::local::PROVIDER_ID, - foundry: provider != FOUNDRY_LOCAL_PROVIDER_ID, - sherpa: provider != crate::asr::local::sherpa::PROVIDER_ID, - } -} - -async fn release_foundry_runtime_if_inactive( - runtime: &Arc, - release_foundry: bool, -) { - if release_foundry { - runtime.request_cancel_prepare(); - if let Err(error) = runtime.release_now().await { - log::warn!("[foundry-asr] release inactive runtime failed: {error:#}"); - } - } -} - -async fn release_sherpa_runtime_if_inactive( - runtime: &Arc, - release_sherpa: bool, -) { - if release_sherpa { - runtime.request_cancel_prepare(); - if let Err(error) = runtime.release_now().await { - log::warn!("[sherpa-asr] release inactive runtime failed: {error:#}"); - } - } -} - -#[tauri::command] -pub fn set_credential(window: Window, account: String, value: String) -> Result<(), String> { - ensure_main_window(&window)?; - let acc = parse_account(&account)?; - if value.is_empty() { - CredentialsVault::remove(acc).map_err(|e| e.to_string())?; - } else { - CredentialsVault::set(acc, &value).map_err(|e| e.to_string())?; - } - // 通知前端凭据已变更（如 Overview 页需要刷新 asrConfigured 状态）。 - // issue #532 / #573：在 Settings 填写凭据但不切换提供商时，Overview 不会重拉状态， - // 仍显示「未配置」。该修复曾随 #538 合入 main，但被 beta→main 合并覆盖，beta 上缺失。 - let _ = window.emit("credentials:changed", ()); - Ok(()) -} - -#[tauri::command] -pub async fn set_active_asr_provider( - coord: CoordinatorState<'_>, - runtime: State<'_, Arc>, - sherpa_runtime: State<'_, Arc>, - provider: String, -) -> Result<(), String> { - if provider == FOUNDRY_LOCAL_PROVIDER_ID && !active_foundry_asr_is_supported(&provider) { - return Err("Foundry Local Whisper is only available on Windows".to_string()); - } - if provider == crate::asr::local::sherpa::PROVIDER_ID - && !active_sherpa_asr_is_supported(&provider) - { - return Err("sherpa-onnx local ASR is only available on Windows".to_string()); - } - if CredentialsVault::get_active_asr() == provider { - return Ok(()); - } - CredentialsVault::set_active_asr_provider(&provider).map_err(|e| e.to_string())?; - let release_plan = local_asr_release_plan_for_provider(&provider); - if provider == crate::asr::local::PROVIDER_ID { - // 切到本地 ASR → 后台预加载模型，下次按 hotkey 时不必等数秒。 - coord.preload_local_asr_in_background(); - } - if release_plan.qwen { - // 切回云端 → 用户已不需要本地引擎，立刻释放 1.2GB+ RAM；不释放的话只会等到 - // schedule_local_asr_release 的下一次 dictation 才触发，而切回云端后根本不会 - // 再走 local 路径，引擎会驻留到进程退出。 - coord.release_local_asr_engine(); - } - release_foundry_runtime_if_inactive(runtime.inner(), release_plan.foundry).await; - release_sherpa_runtime_if_inactive(sherpa_runtime.inner(), release_plan.sherpa).await; - Ok(()) -} - -#[tauri::command] -pub fn set_active_llm_provider(provider: String) -> Result<(), String> { - CredentialsVault::set_active_llm_provider(&provider).map_err(|e| e.to_string()) -} - -/// 读出某个账号的实际值（用于设置页预填表单）。 -/// 凭据来自系统凭据库；只允许主设置窗口读取 raw secret，避免胶囊 / QA 等辅助窗口默认暴露。 -#[tauri::command] -pub fn read_credential(window: Window, account: String) -> Result, String> { - ensure_main_window(&window)?; - let acc = parse_account(&account)?; - CredentialsVault::get(acc).map_err(|e| e.to_string()) -} - -fn ensure_main_window(window: &Window) -> Result<(), String> { - if window.label() == "main" { - Ok(()) - } else { - Err("credential access is only allowed from the main window".to_string()) - } -} - -#[derive(Serialize)] -#[serde(rename_all = "camelCase")] -pub struct ProviderCheckResult { - ok: bool, -} - -#[derive(Serialize)] -pub struct ProviderModelsResult { - models: Vec, -} - -#[tauri::command] -pub async fn validate_provider_credentials(kind: String) -> Result { - match kind.as_str() { - "llm" => validate_llm_provider() - .await - .map(|()| ProviderCheckResult { ok: true }), - "asr" => validate_asr_provider() - .await - .map(|()| ProviderCheckResult { ok: true }), - _ => Err(format!("unknown provider kind: {kind}")), - } -} - -#[tauri::command] -pub async fn list_provider_models(kind: String) -> Result { - if kind == "asr" && CredentialsVault::get_active_asr() == crate::asr::bailian::PROVIDER_ID { - return Ok(ProviderModelsResult { - models: vec![crate::asr::bailian::DEFAULT_MODEL.to_string()], - }); - } - if kind == "asr" && CredentialsVault::get_active_asr() == crate::asr::mimo::PROVIDER_ID { - return Ok(ProviderModelsResult { - models: vec![crate::asr::mimo::DEFAULT_MODEL.to_string()], - }); - } - if kind == "llm" && CredentialsVault::get_active_llm() == CODEX_OAUTH_PROVIDER_ID { - return Ok(ProviderModelsResult { - models: vec![ - CODEX_DEFAULT_MODEL.to_string(), - "gpt-5.3-codex".to_string(), - "gpt-5.4".to_string(), - "gpt-5.5".to_string(), - ], - }); - } - let config = read_openai_provider_config(&kind)?; - fetch_provider_models(&config) - .await - .map(|models| ProviderModelsResult { models }) -} - -struct ProviderConfig { - base_url: String, - api_key: String, -} - -fn read_openai_provider_config(kind: &str) -> Result { - let (api_key_account, endpoint_account, api_key_required) = match kind { - "llm" => ( - CredentialAccount::ArkApiKey, - CredentialAccount::ArkEndpoint, - false, - ), - "asr" => ( - CredentialAccount::AsrApiKey, - CredentialAccount::AsrEndpoint, - true, - ), - _ => return Err(format!("unknown provider kind: {kind}")), - }; - let api_key = CredentialsVault::get(api_key_account) - .map_err(|e| e.to_string())? - .unwrap_or_default(); - let base_url = CredentialsVault::get(endpoint_account) - .map_err(|e| e.to_string())? - .unwrap_or_default(); - if api_key_required && api_key.trim().is_empty() { - return Err("API Key 为空".to_string()); - } - if base_url.trim().is_empty() { - return Err("Endpoint 为空".to_string()); - } - Ok(ProviderConfig { base_url, api_key }) -} - -async fn validate_llm_provider() -> Result<(), String> { - let llm_thinking_enabled = PreferencesStore::new() - .map_err(|e| e.to_string())? - .get() - .llm_thinking_enabled; - if CredentialsVault::get_active_llm() == CODEX_OAUTH_PROVIDER_ID { - let model = CredentialsVault::get(CredentialAccount::ArkModelId) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()) - .unwrap_or_else(|| CODEX_DEFAULT_MODEL.to_string()); - let provider = CodexOAuthLLMProvider::new( - CodexOAuthConfig::new(model).with_thinking_enabled(llm_thinking_enabled), - ); - return provider - .polish( - "验证连接", - PolishMode::Raw, - &[], - "", - &[], - ChineseScriptPreference::Auto, - OutputLanguagePreference::Auto, - None, - &[], - ) - .await - .map(|_| ()) - .map_err(|e| match e { - LLMError::InvalidResponse { status, .. } => { - format!("providerHttpStatus:{status}") - } - other => other.to_string(), - }); - } - - let config = read_openai_provider_config("llm")?; - let active_llm = CredentialsVault::get_active_llm(); - let model = CredentialsVault::get(CredentialAccount::ArkModelId) - .map_err(|e| e.to_string())? - .filter(|s| !s.is_empty()) - .ok_or_else(|| "llmModelMissing".to_string())?; - let provider = OpenAICompatibleLLMProvider::new( - OpenAICompatibleConfig::new( - active_llm.clone(), - active_llm, - config.base_url, - config.api_key, - model, - ) - .with_thinking_enabled(llm_thinking_enabled), - ); - provider - .polish( - "验证连接", - PolishMode::Raw, - &[], - "", - &[], - ChineseScriptPreference::Auto, - OutputLanguagePreference::Auto, - None, - &[], - ) - .await - .map(|_| ()) - .map_err(|e| match e { - LLMError::InvalidResponse { status, .. } => { - format!("providerHttpStatus:{status}") - } - other => other.to_string(), - }) -} - -async fn validate_asr_provider() -> Result<(), String> { - let active_asr = CredentialsVault::get_active_asr(); - if active_asr_is_keyless_for_validation(&active_asr) { - return Ok(()); - } - - if active_asr == crate::asr::bailian::PROVIDER_ID { - return validate_bailian_asr_provider().await; - } - if active_asr == crate::asr::mimo::PROVIDER_ID { - return validate_mimo_asr_provider().await; - } - - let config = read_openai_provider_config("asr")?; - let model = CredentialsVault::get(CredentialAccount::AsrModel) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()) - .ok_or_else(|| "asrModelMissing".to_string())?; - validate_asr_transcription(&config, model.trim()).await -} - -async fn validate_mimo_asr_provider() -> Result<(), String> { - let config = read_openai_provider_config("asr")?; - let model = CredentialsVault::get(CredentialAccount::AsrModel) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()) - .unwrap_or_else(|| crate::asr::mimo::DEFAULT_MODEL.to_string()); - let asr = crate::asr::MimoBatchASR::new(config.api_key, config.base_url, model); - crate::recorder::AudioConsumer::consume_pcm_chunk( - &asr, - &encode_wav_16k_mono_silence(250)[44..], - ); - asr.transcribe() - .await - .map(|_| ()) - .map_err(|e| e.to_string()) -} - -async fn validate_bailian_asr_provider() -> Result<(), String> { - let api_key = CredentialsVault::get(CredentialAccount::AsrApiKey) - .map_err(|e| e.to_string())? - .unwrap_or_default(); - if api_key.trim().is_empty() { - return Err("API Key 为空".to_string()); - } - let endpoint = CredentialsVault::get(CredentialAccount::AsrEndpoint) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()) - .unwrap_or_else(|| crate::asr::bailian::DEFAULT_ENDPOINT.to_string()); - let model = CredentialsVault::get(CredentialAccount::AsrModel) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()) - .unwrap_or_else(|| crate::asr::bailian::DEFAULT_MODEL.to_string()); - let vocabulary_id = CredentialsVault::get(CredentialAccount::AsrVocabularyId) - .map_err(|e| e.to_string())? - .filter(|s| !s.trim().is_empty()); - let asr = std::sync::Arc::new(crate::asr::BailianRealtimeASR::new( - crate::asr::BailianCredentials { - api_key, - endpoint, - model, - vocabulary_id, - }, - )); - asr.open_session().await.map_err(|e| e.to_string())?; - crate::asr::AudioConsumer::consume_pcm_chunk( - &*asr, - &vec![0u8; crate::asr::bailian::TARGET_AUDIO_CHUNK_BYTES], - ); - asr.send_last_frame().await.map_err(|e| e.to_string())?; - asr.await_final_result() - .await - .map(|_| ()) - .map_err(|e| e.to_string()) -} - -fn active_asr_is_keyless_for_validation(provider: &str) -> bool { - provider == crate::asr::local::PROVIDER_ID - || active_foundry_asr_is_supported(provider) - || active_sherpa_asr_is_supported(provider) -} - -fn active_foundry_asr_is_supported(provider: &str) -> bool { - #[cfg(target_os = "windows")] - { - provider == FOUNDRY_LOCAL_PROVIDER_ID - } - #[cfg(not(target_os = "windows"))] - { - let _ = provider; - false - } -} - -fn active_sherpa_asr_is_supported(provider: &str) -> bool { - #[cfg(target_os = "windows")] - { - provider == crate::asr::local::sherpa::PROVIDER_ID - } - #[cfg(not(target_os = "windows"))] - { - let _ = provider; - false - } -} - -async fn validate_asr_transcription(config: &ProviderConfig, model: &str) -> Result<(), String> { - const MAX_ASR_VALIDATE_BODY_BYTES: usize = 1024 * 1024; - const MAX_ATTEMPTS: u32 = 6; - let url = asr_transcriptions_url(&config.base_url)?; - let wav = encode_wav_16k_mono_silence(250); - let client = http_client_builder(&url, 20) - .build() - .map_err(|_| "providerClientInitFailed".to_string())?; - // 连接 / 请求未送出类失败做指数退避重试 —— 这类失败请求尚未送达服务端，重试 - // 安全。超时不重试（服务端可能已在处理）。multipart 是流式 body，每次重建。 - let mut attempt: u32 = 0; - let response = loop { - attempt += 1; - let wav_part = reqwest::multipart::Part::bytes(wav.clone()) - .file_name("openless-asr-check.wav") - .mime_str("audio/wav") - .map_err(|e| format!("请求体构建失败: {e}"))?; - let form = reqwest::multipart::Form::new() - .part("file", wav_part) - .text("model", model.to_string()); - match client - .post(&url) - .header("Authorization", format!("Bearer {}", config.api_key)) - .multipart(form) - .send() - .await - { - Ok(resp) => break resp, - Err(e) if e.is_timeout() => return Err("providerRequestTimeout".to_string()), - Err(e) if (e.is_connect() || e.is_request()) && attempt < MAX_ATTEMPTS => { - let backoff = (200u64 * 2u64.pow((attempt - 1).min(3))).min(900); - tokio::time::sleep(std::time::Duration::from_millis(backoff)).await; - continue; - } - Err(_) => return Err("providerNetworkError".to_string()), - } - }; - let status = response.status(); - if !status.is_success() { - return Err(format!("providerHttpStatus:{}", status.as_u16())); - } - if let Some(len) = response.content_length() { - if len as usize > MAX_ASR_VALIDATE_BODY_BYTES { - return Err("providerResponseTooLarge".to_string()); - } - } - use futures_util::StreamExt; - let mut body = Vec::::new(); - let mut stream = response.bytes_stream(); - while let Some(chunk) = stream.next().await { - let chunk = chunk.map_err(|_| "providerReadResponseFailed".to_string())?; - if body.len().saturating_add(chunk.len()) > MAX_ASR_VALIDATE_BODY_BYTES { - return Err("providerResponseTooLarge".to_string()); - } - body.extend_from_slice(&chunk); - } - let json: Value = serde_json::from_slice(&body).map_err(|_| "asrInvalidJson".to_string())?; - if !json.is_object() || json.get("text").is_none() { - return Err("asrMissingTextField".to_string()); - } - Ok(()) -} - -fn asr_transcriptions_url(base_url: &str) -> Result { - let parsed = reqwest::Url::parse(base_url.trim()).map_err(|_| "endpointInvalid".to_string())?; - - // Work on the URL path only so we don't corrupt query parameters. - let mut url = parsed.clone(); - let path = parsed.path().trim_end_matches('/'); - let next_path = if path.ends_with("/audio/transcriptions") { - path.to_string() - } else if path.ends_with("/audio") { - format!("{path}/transcriptions") - } else if let Some(prefix) = path.strip_suffix("/chat/completions") { - format!("{prefix}/audio/transcriptions") - } else { - format!("{path}/audio/transcriptions") - }; - url.set_path(&next_path); - Ok(url.to_string()) -} - -fn encode_wav_16k_mono_silence(duration_ms: u32) -> Vec { - let sample_rate: u32 = 16_000; - let num_channels: u16 = 1; - let bits_per_sample: u16 = 16; - let bytes_per_sample = (bits_per_sample / 8) as usize; - let samples = (sample_rate as usize * duration_ms as usize) / 1000; - let pcm_len = samples * bytes_per_sample; - let data_size = pcm_len as u32; - let byte_rate = sample_rate * num_channels as u32 * bits_per_sample as u32 / 8; - let block_align = num_channels * bits_per_sample / 8; - let chunk_size = 36 + data_size; - - let mut wav = Vec::with_capacity(44 + pcm_len); - wav.extend_from_slice(b"RIFF"); - wav.extend_from_slice(&chunk_size.to_le_bytes()); - wav.extend_from_slice(b"WAVE"); - wav.extend_from_slice(b"fmt "); - wav.extend_from_slice(&16u32.to_le_bytes()); - wav.extend_from_slice(&1u16.to_le_bytes()); - wav.extend_from_slice(&num_channels.to_le_bytes()); - wav.extend_from_slice(&sample_rate.to_le_bytes()); - wav.extend_from_slice(&byte_rate.to_le_bytes()); - wav.extend_from_slice(&block_align.to_le_bytes()); - wav.extend_from_slice(&bits_per_sample.to_le_bytes()); - wav.extend_from_slice(b"data"); - wav.extend_from_slice(&data_size.to_le_bytes()); - wav.resize(44 + pcm_len, 0); - wav -} - -async fn fetch_provider_models(config: &ProviderConfig) -> Result, String> { - let url = models_url(&config.base_url); - let is_gemini = is_gemini_base_url(&config.base_url); - log::info!("[provider-check] GET {url} (gemini={is_gemini})"); - let client = http_client_builder(&config.base_url, 15) - .build() - .map_err(|e| format!("HTTP client 初始化失败: {e}"))?; - let mut request = client.get(&url); - if !config.api_key.trim().is_empty() { - // 谷歌原生 generativelanguage.googleapis.com 不识别 Bearer Authorization, - // 必须用 x-goog-api-key 头。其它 OpenAI 兼容 provider 仍走 Bearer。 - if is_gemini { - request = request.header("x-goog-api-key", config.api_key.as_str()); - } else { - request = request.header("Authorization", format!("Bearer {}", config.api_key)); - } - } - let response = request.send().await.map_err(|e| { - if e.is_timeout() { - "请求超时".to_string() - } else { - format!("网络错误: {e}") - } - })?; - let status = response.status(); - let body = response - .text() - .await - .map_err(|e| format!("读取响应失败: {e}"))?; - if !status.is_success() { - return Err(format!("providerHttpStatus:{}", status.as_u16())); - } - if is_gemini { - parse_gemini_model_ids(&body) - } else { - parse_model_ids(&body) - } -} - -fn is_gemini_base_url(base_url: &str) -> bool { - base_url.contains("generativelanguage.googleapis.com") -} - -fn models_url(base_url: &str) -> String { - let trimmed = base_url.trim().trim_end_matches('/'); - if trimmed.ends_with("/models") { - return trimmed.to_string(); - } - if let Some(prefix) = trimmed.strip_suffix("/chat/completions") { - return format!("{prefix}/models"); - } - format!("{trimmed}/models") -} - -fn parse_model_ids(body: &str) -> Result, String> { - let json: Value = - serde_json::from_str(body).map_err(|e| format!("模型列表不是有效 JSON: {e}"))?; - let data = json - .get("data") - .and_then(|v| v.as_array()) - .ok_or_else(|| "模型列表缺少 data 数组".to_string())?; - let mut models = data - .iter() - .filter_map(|item| item.get("id").and_then(|id| id.as_str())) - .map(str::trim) - .filter(|id| !id.is_empty()) - .map(ToOwned::to_owned) - .collect::>(); - models.sort(); - models.dedup(); - Ok(models) -} - -/// 谷歌 v1beta/models 响应形状：`{models: [{name: "models/gemini-2.5-flash", -/// supportedGenerationMethods: ["generateContent", ...], ...}, ...]}`。 -/// 与 OpenAI `{data: [{id: "..."}]}` 不兼容，所以单独解析；name 字段去掉 -/// "models/" 前缀后即是 ProviderTools「拉取模型」按钮可直接写入 ark.model_id -/// 的字符串。 -/// -/// 过滤：只保留声明支持 `generateContent` 的模型——Google 的 model list 同时 -/// 暴露 embedding (`gemini-embedding-2`)、TTS、image 等不支持 -/// generateContent 的家族；用户选中那种 ID 后 polish 必失败（PR #398 pr_agent -/// 漏洞反馈）。`supportedGenerationMethods` 字段缺失时保守保留——某些 preview -/// 模型可能未暴露这个字段，宁误显示也不要把新模型挡在外面。 -fn parse_gemini_model_ids(body: &str) -> Result, String> { - let json: Value = - serde_json::from_str(body).map_err(|e| format!("模型列表不是有效 JSON: {e}"))?; - let models = json - .get("models") - .and_then(|v| v.as_array()) - .ok_or_else(|| "Gemini 模型列表缺少 models 数组".to_string())?; - let mut ids = models - .iter() - .filter(|item| { - match item - .get("supportedGenerationMethods") - .and_then(|v| v.as_array()) - { - Some(methods) => methods - .iter() - .any(|m| m.as_str() == Some("generateContent")), - None => true, // 字段缺失：保守包含 - } - }) - .filter_map(|item| item.get("name").and_then(|n| n.as_str())) - .map(|name| { - name.strip_prefix("models/") - .unwrap_or(name) - .trim() - .to_string() - }) - .filter(|id| !id.is_empty()) - .collect::>(); - ids.sort(); - ids.dedup(); - Ok(ids) -} - -fn parse_account(s: &str) -> Result { - match s { - "volcengine.app_key" => Ok(CredentialAccount::VolcengineAppKey), - "volcengine.access_key" => Ok(CredentialAccount::VolcengineAccessKey), - "volcengine.resource_id" => Ok(CredentialAccount::VolcengineResourceId), - "ark.api_key" => Ok(CredentialAccount::ArkApiKey), - "ark.model_id" => Ok(CredentialAccount::ArkModelId), - "ark.endpoint" => Ok(CredentialAccount::ArkEndpoint), - "asr.api_key" => Ok(CredentialAccount::AsrApiKey), - "asr.endpoint" => Ok(CredentialAccount::AsrEndpoint), - "asr.model" => Ok(CredentialAccount::AsrModel), - "asr.vocabulary_id" => Ok(CredentialAccount::AsrVocabularyId), - _ => Err(format!("unknown account: {s}")), - } -} - -// ─────────────────────────── history ─────────────────────────── - -#[tauri::command] -pub fn list_history(coord: CoordinatorState<'_>) -> Result, String> { - coord.history().list().map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn delete_history_entry(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { - coord.history().delete(&id).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn clear_history(coord: CoordinatorState<'_>) -> Result<(), String> { - coord.history().clear().map_err(|e| e.to_string()) -} - -/// 读取某次会话的原始麦克风 wav 字节流。仅当用户开过 -/// `prefs.record_audio_for_debug` 并且这条 session 是开关打开后录的，才会有文件。 -/// 文件名规约：`/recordings/.wav`，与 DictationSession.id 同名。 -/// -/// 路径校验：session_id **必须**严格匹配 UUID-v4 字面（36 字符 = 8-4-4-4-12 + 4 个 `-`， -/// 内容仅 ASCII 十六进制 + `-`）。白名单胜过黑名单——绝对路径前缀、Windows ADS、 -/// 百分号编码、NUL 字节都不在合法字符集里，挡掉所有 Path::join 越界的可能。 -/// session_id 在仓库内由 `Uuid::new_v4()` 生成 (`dictation.rs:1531`)，前端只会回传 -/// 自己列出的合法 id，但 IPC = boundary，按 boundary 规则严格校验。 -/// -/// async fs：单条 5 分钟 wav 约 9.6MB，同步 `std::fs::read` 会阻塞 Tauri IPC 主循环。 -/// 改 `tokio::fs::read` 后让出线程给其它 IPC。 -#[tauri::command] -pub async fn read_audio_recording(session_id: String) -> Result, String> { - if !is_valid_session_id(&session_id) { - return Err("invalid session id".into()); - } - let path = - crate::persistence::recording_path_for_session(&session_id).map_err(|e| e.to_string())?; - if !path.exists() { - return Err("recording not found".into()); - } - // TOCTOU 兜底：exists() 通过到 read 之间文件可能被 prune（条数 cap / retention - // 清理 / 用户手动删）。把 NotFound 标准化成跟 exists() 失败同样的错误字符串， - // 前端单条 'recording not found' catch 就能稳定隐藏按钮，不依赖本地化 OS 错误。 - tokio::fs::read(&path).await.map_err(|e| { - if e.kind() == std::io::ErrorKind::NotFound { - "recording not found".into() - } else { - format!("read wav failed: {e}") - } - }) -} - -/// UUID-v4 字面校验：36 字符 + 5 段 `-` 分隔（8-4-4-4-12）+ 仅 ASCII 十六进制。 -/// 用于 install/detail/like —— pack_id 来自远端服务器，必须是它发的 UUID。 -fn is_valid_session_id(s: &str) -> bool { - if s.len() != 36 { - return false; - } - let bytes = s.as_bytes(); - for (i, b) in bytes.iter().enumerate() { - let is_dash_position = matches!(i, 8 | 13 | 18 | 23); - if is_dash_position { - if *b != b'-' { - return false; - } - } else if !b.is_ascii_hexdigit() { - return false; - } - } - true -} - -/// 本地 style pack id 白名单：`[A-Za-z0-9._-]`、长度 1..=128。 -/// 上传走本地 id（`builtin.light` / 用户自取 slug / UUID 都可），不是远端 UUID。 -/// 仍阻断 `..` / `/` / `\` / 控制字符，避免 path traversal 进临时 zip 文件名。 -fn is_valid_local_pack_id(s: &str) -> bool { - if s.is_empty() || s.len() > 128 { - return false; - } - s.bytes() - .all(|b| b.is_ascii_alphanumeric() || b == b'.' || b == b'-' || b == b'_') -} - -// ─────────────────────────── vocab ─────────────────────────── - -#[tauri::command] -pub fn list_vocab(coord: CoordinatorState<'_>) -> Result, String> { - coord.vocab().list().map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn add_vocab( - coord: CoordinatorState<'_>, - phrase: String, - note: Option, -) -> Result { - coord.vocab().add(phrase, note).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn remove_vocab(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { - coord.vocab().remove(&id).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn set_vocab_enabled( - coord: CoordinatorState<'_>, - id: String, - enabled: bool, -) -> Result<(), String> { - coord - .vocab() - .set_enabled(&id, enabled) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn list_correction_rules(coord: CoordinatorState<'_>) -> Result, String> { - coord.correction_rules().list().map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn add_correction_rule( - coord: CoordinatorState<'_>, - pattern: String, - replacement: String, -) -> Result { - coord - .correction_rules() - .add(pattern, replacement) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn remove_correction_rule(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { - coord - .correction_rules() - .remove(&id) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn set_correction_rule_enabled( - coord: CoordinatorState<'_>, - id: String, - enabled: bool, -) -> Result<(), String> { - coord - .correction_rules() - .set_enabled(&id, enabled) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn list_vocab_presets() -> Result { - crate::persistence::list_vocab_presets().map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn save_vocab_presets(store: VocabPresetStore) -> Result<(), String> { - crate::persistence::save_vocab_presets(&store).map_err(|e| e.to_string()) -} - -// ─────────────────────────── dictation lifecycle ─────────────────────────── - -#[tauri::command] -pub async fn start_dictation(coord: CoordinatorState<'_>) -> Result<(), String> { - coord.start_dictation().await -} - -#[tauri::command] -pub async fn stop_dictation(coord: CoordinatorState<'_>) -> Result<(), String> { - coord.stop_dictation().await -} - -#[tauri::command] -pub fn cancel_dictation(coord: CoordinatorState<'_>) { - coord.cancel_dictation(); -} - -#[tauri::command] -pub async fn handle_window_hotkey_event( - coord: CoordinatorState<'_>, - event_type: String, - key: String, - code: String, - repeat: bool, -) -> Result<(), String> { - coord - .handle_window_hotkey_event(event_type, key, code, repeat) - .await -} - -#[cfg(debug_assertions)] -#[tauri::command] -pub async fn inject_hotkey_click_for_dev(coord: CoordinatorState<'_>) -> Result<(), String> { - coord.inject_hotkey_click_for_dev().await -} - -#[tauri::command] -pub async fn repolish( - coord: CoordinatorState<'_>, - raw_text: String, - mode: PolishMode, -) -> Result { - log::info!( - "[style-pack] command repolish requested legacy_mode={:?} raw_chars={}", - mode, - raw_text.chars().count() - ); - coord.repolish(raw_text, mode).await -} - -// ─────────────────────────── style packs ─────────────────────────── - -#[tauri::command] -pub fn list_style_packs(coord: CoordinatorState<'_>) -> Result, String> { - let prefs = coord.prefs().get(); - coord - .style_packs() - .list_with_active(&prefs.active_style_pack_id) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn create_style_pack_from_template( - coord: CoordinatorState<'_>, - app: AppHandle, - template: StylePack, -) -> Result { - log::info!( - "[style-pack] command create_from_template name={} base_mode={:?}", - template.name, - template.base_mode - ); - let created = coord - .style_packs() - .create_from_template(template) - .map_err(|e| e.to_string())?; - let prefs = coord.prefs().get(); - let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - Ok(created) -} - -#[tauri::command] -pub fn save_style_pack( - coord: CoordinatorState<'_>, - app: AppHandle, - style_pack: StylePack, -) -> Result { - log::info!( - "[style-pack] command save id={} kind={:?} base_mode={:?}", - style_pack.id, - style_pack.kind, - style_pack.base_mode - ); - let saved = coord - .style_packs() - .upsert(style_pack) - .map_err(|e| e.to_string())?; - if saved.kind == StylePackKind::Builtin { - let prefs = coord.prefs().get(); - let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - } - Ok(saved) -} - -#[tauri::command] -pub fn preview_style_pack_runtime( - coord: CoordinatorState<'_>, - style_pack: StylePack, -) -> Result { - log::info!( - "[style-pack] command preview_runtime id={} base_mode={:?} prompt_chars={}", - style_pack.id, - style_pack.base_mode, - style_pack.prompt.chars().count() - ); - Ok(coord.preview_style_pack_runtime(&style_pack)) -} - -#[tauri::command] -pub fn set_active_style_pack( - coord: CoordinatorState<'_>, - app: AppHandle, - id: String, -) -> Result { - activate_style_pack_by_id(&coord, &app, &id) -} - -#[tauri::command] -pub fn set_style_pack_enabled( - coord: CoordinatorState<'_>, - app: AppHandle, - id: String, - enabled: bool, -) -> Result, String> { - log::info!( - "[style-pack] command set_enabled requested id={} enabled={}", - id, - enabled - ); - coord - .style_packs() - .set_enabled(&id, enabled) - .map_err(|e| e.to_string())?; - let mut prefs = coord.prefs().get(); - if !enabled && prefs.active_style_pack_id == id { - prefs.active_style_pack_id = default_active_style_pack_id(); - } - let prefs = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - coord - .style_packs() - .list_with_active(&prefs.active_style_pack_id) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn reset_builtin_style_pack( - coord: CoordinatorState<'_>, - app: AppHandle, - id: String, -) -> Result { - log::info!("[style-pack] command reset_builtin requested id={id}"); - let saved = coord - .style_packs() - .reset_builtin(&id) - .map_err(|e| e.to_string())?; - let prefs = coord.prefs().get(); - let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - Ok(saved) -} - -#[tauri::command] -pub fn delete_style_pack( - coord: CoordinatorState<'_>, - app: AppHandle, - id: String, -) -> Result<(), String> { - let mut prefs = coord.prefs().get(); - log::info!("[style-pack] command delete requested id={id}"); - coord - .style_packs() - .remove_imported(&id) - .map_err(|e| e.to_string())?; - if prefs.active_style_pack_id == id { - prefs.active_style_pack_id = default_active_style_pack_id(); - let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - } else { - refresh_tray_menu_async(&app); - } - Ok(()) -} - -#[tauri::command] -pub fn import_style_pack_from_zip( - coord: CoordinatorState<'_>, - zip_path: String, -) -> Result { - log::info!("[style-pack] command import requested zip_path={zip_path}"); - coord - .style_packs() - .import_from_zip(std::path::Path::new(&zip_path)) - .map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn export_style_pack_to_zip( - coord: CoordinatorState<'_>, - id: String, - target_path: String, -) -> Result { - log::info!( - "[style-pack] command export requested id={} target_path={}", - id, - target_path - ); - coord - .style_packs() - .export_to_zip(&id, std::path::Path::new(&target_path)) - .map_err(|e| e.to_string())?; - Ok(target_path) -} - -// ─────────────────────────── style toggles (compat) ─────────────────────────── - -#[tauri::command] -pub fn set_default_polish_mode( - coord: CoordinatorState<'_>, - app: AppHandle, - mode: PolishMode, -) -> Result<(), String> { - activate_builtin_style_mode(&coord, &app, mode) -} - -#[tauri::command] -pub fn set_style_enabled( - coord: CoordinatorState<'_>, - app: AppHandle, - mode: PolishMode, - enabled: bool, -) -> Result<(), String> { - let pack_id = builtin_style_pack_id(mode).to_string(); - log::info!( - "[style-pack] compat set_style_enabled mode={:?} pack_id={} enabled={}", - mode, - pack_id, - enabled - ); - coord - .style_packs() - .set_enabled(&pack_id, enabled) - .map_err(|e| e.to_string())?; - let mut prefs = coord.prefs().get(); - if !enabled && prefs.active_style_pack_id == pack_id { - prefs.active_style_pack_id = default_active_style_pack_id(); - } - let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; - Ok(()) -} - -// ─────────────────────────── 系统权限 ─────────────────────────── - -#[tauri::command] -pub fn check_accessibility_permission() -> PermissionStatus { - permissions::check_accessibility() -} - -#[tauri::command] -pub fn request_accessibility_permission() -> PermissionStatus { - permissions::request_accessibility() -} - -#[tauri::command] -pub fn check_microphone_permission() -> PermissionStatus { - permissions::check_microphone() -} - -#[tauri::command] -pub fn request_microphone_permission(app: AppHandle) -> PermissionStatus { - crate::request_microphone_from_foreground(&app) -} - -/// 跳到 macOS 系统设置的指定隐私面板。pane: "accessibility" | "microphone". -#[tauri::command] -pub fn open_system_settings(pane: String) -> Result<(), String> { - #[cfg(target_os = "macos")] - { - let url = match pane.as_str() { - "accessibility" => { - "x-apple.systempreferences:com.apple.preference.security?Privacy_Accessibility" - } - "microphone" => { - "x-apple.systempreferences:com.apple.preference.security?Privacy_Microphone" - } - _ => "x-apple.systempreferences:com.apple.preference.security?Privacy", - }; - std::process::Command::new("open") - .arg(url) - .spawn() - .map(|_| ()) - .map_err(|e| e.to_string()) - } - #[cfg(target_os = "windows")] - { - use windows::core::PCWSTR; - use windows::Win32::UI::Shell::ShellExecuteW; - use windows::Win32::UI::WindowsAndMessaging::SW_SHOWNORMAL; - - fn wide_null(value: &str) -> Vec { - value.encode_utf16().chain(std::iter::once(0)).collect() - } - - let uri = match pane.as_str() { - "microphone" => "ms-settings:privacy-microphone", - "sound" => "ms-settings:sound", - "accessibility" => "ms-settings:easeofaccess", - _ => "ms-settings:", - }; - - let operation = wide_null("open"); - let target = wide_null(uri); - let result = unsafe { - ShellExecuteW( - None, - PCWSTR(operation.as_ptr()), - PCWSTR(target.as_ptr()), - PCWSTR::null(), - PCWSTR::null(), - SW_SHOWNORMAL, - ) - }; - - if result.0 as isize <= 32 { - Err(format!("ShellExecuteW failed: {}", result.0 as isize)) - } else { - Ok(()) - } - } - #[cfg(all(not(target_os = "macos"), not(target_os = "windows")))] - { - let _ = pane; - Err("open_system_settings is only supported on macOS and Windows".to_string()) - } -} - -/// 触发 macOS 系统弹"是否允许 OpenLess 访问麦克风"对话框。 -/// 与 Swift `MicrophonePermission.request()` 同语义：只信系统权限回调， -/// 不用 cpal stream 成功与否伪造授权状态。 -#[tauri::command] -pub fn trigger_microphone_prompt(app: AppHandle) -> Result<(), String> { - let status = crate::request_microphone_from_foreground(&app); - if matches!( - status, - PermissionStatus::Granted | PermissionStatus::NotApplicable - ) { - Ok(()) - } else { - Err(format!("microphone permission is {status:?}")) - } -} - -// ─────────────────────────── QA (划词语音问答, issue #118) ─────────────────────────── - -/// 给前端 Settings 页渲染当前 QA 快捷键 label（如 `"Cmd+Shift+;"`）。 -/// 未启用时返回空串。 -#[tauri::command] -pub fn get_qa_hotkey_label(coord: CoordinatorState<'_>) -> String { - coord.qa_hotkey_label() -} - -/// 设置 QA 快捷键并热更新 monitor。 -/// 传入 `None` 形式的字段不在这里支持——前端用 `binding == null` 时调下面的 -/// "disable" 写法（写 prefs.qa_hotkey = None）即可。 -#[tauri::command] -pub fn set_qa_hotkey( - coord: CoordinatorState<'_>, - binding: Option, -) -> Result<(), String> { - if let Some(binding) = binding.as_ref() { - crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; - if binding.modifiers.is_empty() && binding.primary.eq_ignore_ascii_case("shift") { - return Err("Shift 单键目前只能用于翻译快捷键".into()); - } - } - let mut prefs = coord.prefs().get(); - if let Some(binding) = binding.as_ref() { - reject_dictation_qa_hotkey_overlap(&prefs.dictation_hotkey, binding)?; - reject_qa_translation_hotkey_overlap(binding, &prefs.translation_hotkey)?; - if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { - reject_qa_switch_style_hotkey_overlap(binding, switch_style)?; - } - if let Some(open_app) = prefs.open_app_hotkey.as_ref() { - reject_qa_open_app_hotkey_overlap(binding, open_app)?; - } - } - prefs.qa_hotkey = binding; - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - coord.update_qa_hotkey_binding(); - Ok(()) -} - -/// 用户点 ✕ / 按 Esc 关 QA 浮窗。 -#[tauri::command] -pub fn qa_window_dismiss(coord: CoordinatorState<'_>) { - coord.qa_window_dismiss(); -} - -/// 用户点 📌 / 取消 📌。pinned=true 时浮窗不会自动隐藏。 -#[tauri::command] -pub fn qa_window_pin(coord: CoordinatorState<'_>, pinned: bool) { - coord.qa_window_pin(pinned); -} - -/// 用户点 ✕ / 按 Esc 关 Less Computer 浮窗。 -#[tauri::command] -pub fn less_computer_window_dismiss(coord: CoordinatorState<'_>) { - coord.less_computer_window_dismiss(); -} - -/// 前端按内容测高后回传高度，后端 clamp + bottom-anchored 重新摆放浮窗。 -#[tauri::command] -pub fn less_computer_window_resize(coord: CoordinatorState<'_>, height: f64) { - coord.less_computer_window_resize(height); -} - -/// 内联审批卡的 Approve / Deny 回执。token 关联到等待中的拦截动作。 -#[tauri::command] -pub fn less_computer_approve(coord: CoordinatorState<'_>, token: String, approved: bool) { - coord.less_computer_approve(&token, approved); -} - -// ─────────────────────────── 自定义组合键 ─────────────────────────── - -/// 测试一个组合键是否可以注册（验证格式，不实际注册）。 -#[tauri::command] -pub fn validate_shortcut_binding(binding: ShortcutBinding) -> Result<(), String> { - crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn set_dictation_hotkey( - coord: CoordinatorState<'_>, - binding: ShortcutBinding, -) -> Result<(), String> { - crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string())?; - reject_bare_shift_dictation_shortcut(&binding)?; - let mut prefs = coord.prefs().get(); - if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { - reject_dictation_qa_hotkey_overlap(&binding, qa_hotkey)?; - } - reject_dictation_translation_hotkey_overlap(&binding, &prefs.translation_hotkey)?; - if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { - reject_dictation_switch_style_hotkey_overlap(&binding, switch_style)?; - } - if let Some(open_app) = prefs.open_app_hotkey.as_ref() { - reject_dictation_open_app_hotkey_overlap(&binding, open_app)?; - } - prefs.dictation_hotkey = binding; - sync_dictation_hotkey_legacy_fields(&mut prefs); - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - coord.update_hotkey_binding(); - coord.update_combo_hotkey_binding(); - Ok(()) -} - -#[tauri::command] -pub fn set_translation_hotkey( - coord: CoordinatorState<'_>, - binding: ShortcutBinding, -) -> Result<(), String> { - crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string())?; - let previous = coord.prefs().get(); - reject_dictation_translation_hotkey_overlap(&previous.dictation_hotkey, &binding)?; - if let Some(qa_hotkey) = previous.qa_hotkey.as_ref() { - reject_qa_translation_hotkey_overlap(qa_hotkey, &binding)?; - } - if let Some(switch_style) = previous.switch_style_hotkey.as_ref() { - reject_translation_switch_style_hotkey_overlap(&binding, switch_style)?; - } - if let Some(open_app) = previous.open_app_hotkey.as_ref() { - reject_translation_open_app_hotkey_overlap(&binding, open_app)?; - } - let mut prefs = previous.clone(); - prefs.translation_hotkey = binding; - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - if let Err(e) = coord.try_update_translation_hotkey_binding() { - if let Err(rollback_err) = coord.prefs().set(previous) { - log::warn!("[commands] 回滚翻译快捷键失败: {rollback_err}"); - } - coord.update_translation_hotkey_binding(); - return Err(e); - } - Ok(()) -} - -/// 设置「切换风格」全局快捷键。`binding == None`（前端传 null）= 停用：清空绑定并 -/// 反注册全局键。镜像 `set_qa_hotkey` 的 `Option=None` 停用模式（issue #576）。 -#[tauri::command] -pub fn set_switch_style_hotkey( - coord: CoordinatorState<'_>, - binding: Option, -) -> Result<(), String> { - if let Some(binding) = binding.as_ref() { - crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; - reject_modifier_only_action_shortcut(binding)?; - } - let mut prefs = coord.prefs().get(); - if let Some(binding) = binding.as_ref() { - reject_dictation_switch_style_hotkey_overlap(&prefs.dictation_hotkey, binding)?; - reject_translation_switch_style_hotkey_overlap(&prefs.translation_hotkey, binding)?; - if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { - reject_qa_switch_style_hotkey_overlap(qa_hotkey, binding)?; - } - if let Some(open_app) = prefs.open_app_hotkey.as_ref() { - reject_switch_style_open_app_hotkey_overlap(binding, open_app)?; - } - } - prefs.switch_style_hotkey = binding; - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - coord.update_switch_style_hotkey_binding(); - Ok(()) -} - -/// 设置「唤起 App」全局快捷键。`binding == None`（前端传 null）= 停用（同上）。 -#[tauri::command] -pub fn set_open_app_hotkey( - coord: CoordinatorState<'_>, - binding: Option, -) -> Result<(), String> { - if let Some(binding) = binding.as_ref() { - crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; - reject_modifier_only_action_shortcut(binding)?; - } - let mut prefs = coord.prefs().get(); - if let Some(binding) = binding.as_ref() { - reject_dictation_open_app_hotkey_overlap(&prefs.dictation_hotkey, binding)?; - reject_translation_open_app_hotkey_overlap(&prefs.translation_hotkey, binding)?; - if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { - reject_qa_open_app_hotkey_overlap(qa_hotkey, binding)?; - } - if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { - reject_switch_style_open_app_hotkey_overlap(switch_style, binding)?; - } - } - prefs.open_app_hotkey = binding; - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - coord.update_open_app_hotkey_binding(); - Ok(()) -} - -fn reject_modifier_only_action_shortcut(binding: &ShortcutBinding) -> Result<(), String> { - if binding.modifiers.is_empty() - && (binding.primary.eq_ignore_ascii_case("shift") - || crate::shortcut_binding::legacy_modifier_trigger(binding).is_some()) - { - return Err("该快捷键需要使用组合键或非修饰主键".into()); - } - Ok(()) -} - -#[tauri::command] -pub fn validate_combo_hotkey(binding: ComboBinding) -> Result<(), String> { - let shortcut = ShortcutBinding { - primary: binding.primary, - modifiers: binding.modifiers, - }; - reject_bare_shift_dictation_shortcut(&shortcut)?; - crate::combo_hotkey::validate_binding(&shortcut).map_err(|e| e.to_string()) -} - -/// 设置自定义录音组合键并热更新 monitor。 -#[tauri::command] -pub fn set_combo_hotkey(coord: CoordinatorState<'_>, binding: ComboBinding) -> Result<(), String> { - let mut prefs = coord.prefs().get(); - let shortcut = ShortcutBinding { - primary: binding.primary.clone(), - modifiers: binding.modifiers.clone(), - }; - reject_bare_shift_dictation_shortcut(&shortcut)?; - crate::combo_hotkey::validate_binding(&shortcut).map_err(|e| e.to_string())?; - if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { - reject_dictation_qa_hotkey_overlap(&shortcut, qa_hotkey)?; - } - reject_dictation_translation_hotkey_overlap(&shortcut, &prefs.translation_hotkey)?; - if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { - reject_dictation_switch_style_hotkey_overlap(&shortcut, switch_style)?; - } - if let Some(open_app) = prefs.open_app_hotkey.as_ref() { - reject_dictation_open_app_hotkey_overlap(&shortcut, open_app)?; - } - prefs.custom_combo_hotkey = Some(binding); - prefs.dictation_hotkey = shortcut; - sync_dictation_hotkey_legacy_fields(&mut prefs); - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - coord.update_hotkey_binding(); - coord.update_combo_hotkey_binding(); - Ok(()) -} - -fn reject_bare_shift_dictation_shortcut(binding: &ShortcutBinding) -> Result<(), String> { - if binding.modifiers.is_empty() && binding.primary.eq_ignore_ascii_case("shift") { - return Err("Shift 单键目前只能用于翻译快捷键".into()); - } - Ok(()) -} - -fn sync_dictation_hotkey_legacy_fields(prefs: &mut UserPreferences) { - if let Some(trigger) = crate::shortcut_binding::legacy_modifier_trigger(&prefs.dictation_hotkey) - { - prefs.hotkey.trigger = trigger; - prefs.custom_combo_hotkey = None; - return; - } - prefs.hotkey.trigger = crate::types::HotkeyTrigger::Custom; - prefs.custom_combo_hotkey = if prefs.dictation_hotkey.primary.trim().is_empty() { - None - } else { - Some(ComboBinding { - primary: prefs.dictation_hotkey.primary.clone(), - modifiers: prefs.dictation_hotkey.modifiers.clone(), - }) - }; -} - -fn reject_dictation_qa_hotkey_overlap( - dictation: &ShortcutBinding, - qa: &ShortcutBinding, -) -> Result<(), String> { - if shortcut_bindings_overlap(dictation, qa) { - return Err("QA 快捷键不能和听写快捷键相同".into()); - } - Ok(()) -} - -fn reject_hotkey_overlap( - left: &ShortcutBinding, - right: &ShortcutBinding, - message: &'static str, -) -> Result<(), String> { - if shortcut_bindings_overlap(left, right) { - return Err(message.into()); - } - Ok(()) -} - -fn reject_hotkey_collisions(prefs: &UserPreferences) -> Result<(), String> { - // 停用（None）的 action 快捷键不参与任何冲突检测。 - let switch_style = prefs.switch_style_hotkey.as_ref(); - let open_app = prefs.open_app_hotkey.as_ref(); - let less_computer = prefs.coding_agent_voice_hotkey.as_ref(); - if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { - reject_dictation_qa_hotkey_overlap(&prefs.dictation_hotkey, qa_hotkey)?; - reject_qa_translation_hotkey_overlap(qa_hotkey, &prefs.translation_hotkey)?; - if let Some(less_computer) = less_computer { - reject_qa_less_computer_hotkey_overlap(qa_hotkey, less_computer)?; - } - if let Some(switch_style) = switch_style { - reject_qa_switch_style_hotkey_overlap(qa_hotkey, switch_style)?; - } - if let Some(open_app) = open_app { - reject_qa_open_app_hotkey_overlap(qa_hotkey, open_app)?; - } - } - reject_dictation_translation_hotkey_overlap( - &prefs.dictation_hotkey, - &prefs.translation_hotkey, - )?; - if let Some(less_computer) = less_computer { - reject_dictation_less_computer_hotkey_overlap(&prefs.dictation_hotkey, less_computer)?; - reject_translation_less_computer_hotkey_overlap(&prefs.translation_hotkey, less_computer)?; - } - if let Some(switch_style) = switch_style { - reject_dictation_switch_style_hotkey_overlap(&prefs.dictation_hotkey, switch_style)?; - reject_translation_switch_style_hotkey_overlap(&prefs.translation_hotkey, switch_style)?; - if let Some(less_computer) = less_computer { - reject_less_computer_switch_style_hotkey_overlap(less_computer, switch_style)?; - } - } - if let Some(open_app) = open_app { - reject_dictation_open_app_hotkey_overlap(&prefs.dictation_hotkey, open_app)?; - reject_translation_open_app_hotkey_overlap(&prefs.translation_hotkey, open_app)?; - if let Some(less_computer) = less_computer { - reject_less_computer_open_app_hotkey_overlap(less_computer, open_app)?; - } - } - if let (Some(switch_style), Some(open_app)) = (switch_style, open_app) { - reject_switch_style_open_app_hotkey_overlap(switch_style, open_app)?; - } - Ok(()) -} - -fn reject_dictation_translation_hotkey_overlap( - dictation: &ShortcutBinding, - translation: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(dictation, translation, "翻译快捷键不能和听写快捷键相同") -} - -fn reject_dictation_switch_style_hotkey_overlap( - dictation: &ShortcutBinding, - switch_style: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - dictation, - switch_style, - "切换风格快捷键不能和听写快捷键相同", - ) -} - -fn reject_dictation_open_app_hotkey_overlap( - dictation: &ShortcutBinding, - open_app: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(dictation, open_app, "打开应用快捷键不能和听写快捷键相同") -} - -fn reject_dictation_less_computer_hotkey_overlap( - dictation: &ShortcutBinding, - less_computer: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - dictation, - less_computer, - "Less Computer 快捷键不能和听写快捷键相同", - ) -} - -fn reject_qa_translation_hotkey_overlap( - qa: &ShortcutBinding, - translation: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(qa, translation, "翻译快捷键不能和 QA 快捷键相同") -} - -fn reject_qa_switch_style_hotkey_overlap( - qa: &ShortcutBinding, - switch_style: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(qa, switch_style, "切换风格快捷键不能和 QA 快捷键相同") -} - -fn reject_qa_open_app_hotkey_overlap( - qa: &ShortcutBinding, - open_app: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(qa, open_app, "打开应用快捷键不能和 QA 快捷键相同") -} - -fn reject_qa_less_computer_hotkey_overlap( - qa: &ShortcutBinding, - less_computer: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - qa, - less_computer, - "Less Computer 快捷键不能和 QA 快捷键相同", - ) -} - -fn reject_translation_switch_style_hotkey_overlap( - translation: &ShortcutBinding, - switch_style: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - translation, - switch_style, - "切换风格快捷键不能和翻译快捷键相同", - ) -} - -fn reject_translation_open_app_hotkey_overlap( - translation: &ShortcutBinding, - open_app: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap(translation, open_app, "打开应用快捷键不能和翻译快捷键相同") -} - -fn reject_translation_less_computer_hotkey_overlap( - translation: &ShortcutBinding, - less_computer: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - translation, - less_computer, - "Less Computer 快捷键不能和翻译快捷键相同", - ) -} - -fn reject_switch_style_open_app_hotkey_overlap( - switch_style: &ShortcutBinding, - open_app: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - switch_style, - open_app, - "打开应用快捷键不能和切换风格快捷键相同", - ) -} - -fn reject_less_computer_switch_style_hotkey_overlap( - less_computer: &ShortcutBinding, - switch_style: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - less_computer, - switch_style, - "Less Computer 快捷键不能和切换风格快捷键相同", - ) -} - -fn reject_less_computer_open_app_hotkey_overlap( - less_computer: &ShortcutBinding, - open_app: &ShortcutBinding, -) -> Result<(), String> { - reject_hotkey_overlap( - less_computer, - open_app, - "Less Computer 快捷键不能和打开应用快捷键相同", - ) -} - -fn shortcut_bindings_overlap(left: &ShortcutBinding, right: &ShortcutBinding) -> bool { - let left_legacy = crate::shortcut_binding::legacy_modifier_trigger(left); - let right_legacy = crate::shortcut_binding::legacy_modifier_trigger(right); - match (left_legacy, right_legacy) { - (Some(left), Some(right)) => left == right, - (Some(_), None) | (None, Some(_)) => false, - (None, None) => { - let Ok(left) = crate::shortcut_binding::parse_global_hotkey(left) else { - return false; - }; - let Ok(right) = crate::shortcut_binding::parse_global_hotkey(right) else { - return false; - }; - left == right - } - } -} - -// ─────────────────────────── local ASR (Qwen3-ASR) ─────────────────────────── - -use crate::asr::local::{ - download::{fetch_remote_info, RemoteInfo}, - DownloadManager, Mirror, ModelId, ModelStatus, PROVIDER_ID as LOCAL_PROVIDER_ID, -}; - -#[derive(Serialize)] -#[serde(rename_all = "camelCase")] -pub struct LocalAsrSettings { - pub provider_id: String, - pub active_model: String, - pub mirror: String, - pub models_base_dir: Option, - pub models_root_dir: String, - /// macOS 才编入引擎；Windows 端 UI 需要据此把"开始下载"按钮灰掉。 - pub engine_available: bool, -} - -#[tauri::command] -pub fn local_asr_get_settings(coord: CoordinatorState<'_>) -> LocalAsrSettings { - let prefs = coord.prefs().get(); - let models_base_dir = non_empty_string(prefs.local_asr_models_base_dir.clone()); - let models_root_dir = crate::persistence::models_root_for_base_dir(models_base_dir.as_deref()) - .map(|path| path.display().to_string()) - .unwrap_or_default(); - LocalAsrSettings { - provider_id: LOCAL_PROVIDER_ID.into(), - active_model: prefs.local_asr_active_model, - mirror: prefs.local_asr_mirror, - models_base_dir, - models_root_dir, - engine_available: cfg!(target_os = "macos"), - } -} - -fn non_empty_string(value: String) -> Option { - let trimmed = value.trim(); - if trimmed.is_empty() { - None - } else { - Some(trimmed.to_string()) - } -} - -#[derive(Debug, Clone, Serialize)] -#[serde(rename_all = "camelCase")] -pub struct LocalAsrStorageSettings { - pub models_base_dir: Option, - pub models_root_dir: String, - pub is_default: bool, -} - -#[tauri::command] -pub fn local_asr_storage_settings( - coord: CoordinatorState<'_>, -) -> Result { - let prefs = coord.prefs().get(); - let models_base_dir = non_empty_string(prefs.local_asr_models_base_dir); - let models_root_dir = crate::persistence::models_root_for_base_dir(models_base_dir.as_deref()) - .map_err(|e| format!("{e:#}"))? - .display() - .to_string(); - Ok(LocalAsrStorageSettings { - is_default: models_base_dir.is_none(), - models_base_dir, - models_root_dir, - }) -} - -#[tauri::command] -pub async fn local_asr_set_models_base_dir( - coord: CoordinatorState<'_>, - qwen_manager: State<'_, Arc>, - foundry_runtime: State<'_, Arc>, - sherpa_manager: State<'_, Arc>, - sherpa_runtime: State<'_, Arc>, - models_base_dir: Option, -) -> Result { - let next_base_dir = models_base_dir - .as_deref() - .map(str::trim) - .filter(|value| !value.is_empty()) - .map(str::to_string); - let new_root = crate::persistence::validate_models_base_dir(next_base_dir.as_deref()) - .map_err(|e| format!("{e:#}"))?; - - let prefs = coord.prefs().get(); - let current_base_dir = non_empty_string(prefs.local_asr_models_base_dir.clone()); - let old_root = crate::persistence::models_root_for_base_dir(current_base_dir.as_deref()) - .map_err(|e| format!("{e:#}"))?; - let same_root = same_path_for_command(&old_root, &new_root); - if same_root && current_base_dir == next_base_dir { - return local_asr_storage_settings(coord); - } - if !same_root && foundry_runtime.storage_configuration_locked() { - return Err( - "Foundry Local has already been initialized in this app session; restart OpenLess before changing the model storage location" - .to_string(), - ); - } - - quiesce_local_asr_storage_users( - coord.inner(), - qwen_manager.inner(), - foundry_runtime.inner(), - sherpa_manager.inner(), - sherpa_runtime.inner(), - ) - .await?; - crate::persistence::migrate_models_root(&old_root, &new_root).map_err(|e| format!("{e:#}"))?; - - let mut prefs = prefs; - prefs.local_asr_models_base_dir = next_base_dir.clone().unwrap_or_default(); - coord.prefs().set(prefs).map_err(|e| e.to_string())?; - local_asr_storage_settings(coord) -} - -fn same_path_for_command(left: &std::path::Path, right: &std::path::Path) -> bool { - match (left.canonicalize(), right.canonicalize()) { - (Ok(left), Ok(right)) => left == right, - _ => false, - } -} - -async fn quiesce_local_asr_storage_users( - coord: &Arc, - qwen_manager: &Arc, - foundry_runtime: &Arc, - sherpa_manager: &Arc, - sherpa_runtime: &Arc, -) -> Result<(), String> { - for model_id in ModelId::all() { - qwen_manager.cancel(*model_id); - } - for model in crate::asr::local::sherpa::MODELS { - sherpa_manager.cancel(model.alias); - } - foundry_runtime.request_cancel_prepare(); - sherpa_runtime.request_cancel_prepare(); - coord.release_local_asr_engine(); - foundry_runtime - .release_now() - .await - .map_err(|e| format!("{e:#}"))?; - sherpa_runtime - .release_now() - .await - .map_err(|e| format!("{e:#}"))?; - - for _ in 0..50 { - let qwen_active = ModelId::all() - .iter() - .any(|model_id| qwen_manager.is_active(*model_id)); - let sherpa_active = crate::asr::local::sherpa::MODELS - .iter() - .any(|model| sherpa_manager.is_active(model.alias)); - if !qwen_active && !sherpa_active { - return Ok(()); - } - tokio::time::sleep(std::time::Duration::from_millis(100)).await; - } - Err("local ASR downloads are still stopping; retry after cancellation finishes".to_string()) -} - -#[tauri::command] -pub fn local_asr_set_active_model( - coord: CoordinatorState<'_>, - model_id: String, -) -> Result<(), String> { - if ModelId::from_str(&model_id).is_none() { - return Err(format!("unknown model id: {model_id}")); - } - let mut prefs = coord.prefs().get(); - prefs.local_asr_active_model = model_id; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn local_asr_set_mirror(coord: CoordinatorState<'_>, mirror: String) -> Result<(), String> { - let _normalized = Mirror::from_str(&mirror); - let mut prefs = coord.prefs().get(); - prefs.local_asr_mirror = mirror; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn local_asr_list_models() -> Vec { - crate::asr::local::models::list_status() -} - -/// 实时去 HuggingFace API 拉某个模型的真实文件清单 + 总尺寸； -/// 前端在显示模型卡时调一次，避免硬编码尺寸过期。 -#[tauri::command] -pub async fn local_asr_fetch_remote_info( - model_id: String, - mirror: Option, -) -> Result { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - let m = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); - fetch_remote_info(id, m).await.map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn local_asr_download_model( - app: AppHandle, - manager: State<'_, Arc>, - model_id: String, - mirror: Option, -) -> Result<(), String> { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - let m = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); - manager.start(app, id, m); - Ok(()) -} - -#[tauri::command] -pub fn local_asr_cancel_download( - manager: State<'_, Arc>, - model_id: String, -) -> Result<(), String> { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - manager.cancel(id); - Ok(()) -} - -#[tauri::command] -pub fn local_asr_delete_model(coord: CoordinatorState<'_>, model_id: String) -> Result<(), String> { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - // 如果内存里加载的就是要删的这个模型，先释放：否则 mmap 残留指向已 unlink 的文件， - // 且 RAM 直到下次切模型 / 用户手动按"释放"才回收。 - if coord.local_asr_loaded_model().as_deref() == Some(id.as_str()) { - coord.release_local_asr_engine(); - } - crate::asr::local::models::delete_model(id).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn local_asr_model_dir(model_id: String) -> Result { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - crate::asr::local::models::model_dir(id) - .map(|path| path.display().to_string()) - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn local_asr_reveal_model_dir(model_id: String) -> Result<(), String> { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - let dir = crate::asr::local::models::model_dir(id).map_err(|e| format!("{e:#}"))?; - std::fs::create_dir_all(&dir).map_err(|e| format!("create {} failed: {e}", dir.display()))?; - open_path_in_file_manager(&dir) -} - -#[tauri::command] -pub fn local_asr_reveal_models_root(coord: CoordinatorState<'_>) -> Result<(), String> { - let prefs = coord.prefs().get(); - let base_dir = non_empty_string(prefs.local_asr_models_base_dir); - let dir = crate::persistence::models_root_for_base_dir(base_dir.as_deref()) - .map_err(|e| format!("{e:#}"))?; - open_path_in_file_manager(&dir) -} - -#[tauri::command] -pub async fn local_asr_test_model( - model_id: String, -) -> Result { - let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; - crate::asr::local::test_run::run_test(id) - .await - .map_err(|e| format!("{e:#}")) -} - -#[derive(Serialize)] -#[serde(rename_all = "camelCase")] -pub struct LocalAsrEngineStatus { - pub loaded: bool, - pub model_id: Option, - pub keep_loaded_secs: u32, -} - -#[tauri::command] -pub fn local_asr_engine_status(coord: CoordinatorState<'_>) -> LocalAsrEngineStatus { - let prefs = coord.prefs().get(); - LocalAsrEngineStatus { - loaded: coord.local_asr_loaded_model().is_some(), - model_id: coord.local_asr_loaded_model(), - keep_loaded_secs: prefs.local_asr_keep_loaded_secs, - } -} - -#[tauri::command] -pub fn local_asr_release_engine(coord: CoordinatorState<'_>) { - coord.release_local_asr_engine(); -} - -#[tauri::command] -pub fn local_asr_preload(coord: tauri::State<'_, std::sync::Arc>) { - coord.preload_local_asr_in_background(); -} - -#[tauri::command] -pub fn local_asr_set_keep_loaded_secs( - coord: CoordinatorState<'_>, - seconds: u32, -) -> Result<(), String> { - let mut prefs = coord.prefs().get(); - prefs.local_asr_keep_loaded_secs = seconds; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -// ───────────────────── Windows local ASR (Foundry Local Whisper) ───────────────────── - -fn active_foundry_model_from_prefs(prefs: &UserPreferences) -> String { - if model_alias_is_known(&prefs.foundry_local_asr_model) { - prefs.foundry_local_asr_model.clone() - } else { - DEFAULT_MODEL_ALIAS.to_string() - } -} - -fn validate_foundry_model_alias(model_alias: &str) -> Result<(), String> { - if model_alias_is_known(model_alias) { - Ok(()) - } else { - Err(format!( - "unknown Foundry Whisper model alias: {model_alias}" - )) - } -} - -fn normalize_foundry_language_hint(language_hint: &str) -> Result { - let normalized = language_hint.trim().to_string(); - if normalized.is_empty() - || (normalized.len() == 2 && normalized.bytes().all(|b| b.is_ascii_lowercase())) - { - Ok(normalized) - } else { - Err("language hint must be empty or ISO 639-1 lowercase code".to_string()) - } -} - -fn normalize_foundry_runtime_source(source: &str) -> String { - crate::asr::local::foundry_native::normalize_runtime_source_str(source) -} - -#[tauri::command] -pub async fn foundry_local_asr_status( - coord: CoordinatorState<'_>, - runtime: State<'_, Arc>, -) -> Result { - let prefs = coord.prefs().get(); - let active_model = active_foundry_model_from_prefs(&prefs); - Ok(runtime - .status_snapshot(&active_model, &prefs.foundry_local_runtime_source) - .await) -} - -#[tauri::command] -pub async fn foundry_local_asr_catalog( - runtime: State<'_, Arc>, -) -> Result, String> { - runtime - .catalog_snapshot() - .await - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn foundry_local_asr_set_model( - coord: CoordinatorState<'_>, - model_alias: String, -) -> Result<(), String> { - validate_foundry_model_alias(&model_alias)?; - let mut prefs = coord.prefs().get(); - if prefs.foundry_local_asr_model == model_alias { - return Ok(()); - } - prefs.foundry_local_asr_model = model_alias; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn foundry_local_asr_set_language_hint( - coord: CoordinatorState<'_>, - language_hint: String, -) -> Result<(), String> { - let normalized = normalize_foundry_language_hint(&language_hint)?; - let mut prefs = coord.prefs().get(); - if prefs.foundry_local_asr_language_hint == normalized { - return Ok(()); - } - prefs.foundry_local_asr_language_hint = normalized; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn foundry_local_asr_set_runtime_source( - coord: CoordinatorState<'_>, - source: String, -) -> Result<(), String> { - let mut prefs = coord.prefs().get(); - let normalized = normalize_foundry_runtime_source(&source); - if prefs.foundry_local_runtime_source == normalized { - return Ok(()); - } - prefs.foundry_local_runtime_source = normalized; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub async fn foundry_local_asr_prepare( - app: AppHandle, - coord: CoordinatorState<'_>, - runtime: State<'_, Arc>, - model_alias: String, -) -> Result { - validate_foundry_model_alias(&model_alias)?; - let prefs = coord.prefs().get(); - let runtime_source = prefs.foundry_local_runtime_source.clone(); - let progress_app = app.clone(); - let result = runtime - .ensure_loaded_with_progress(&model_alias, &runtime_source, move |payload| { - emit_foundry_prepare_progress(&progress_app, payload); - }) - .await; - match result { - Ok(model_id) => Ok(model_id), - Err(error) => { - let message = format!("{error:#}"); - emit_foundry_prepare_progress( - &app, - FoundryPrepareProgressPayload::failed( - model_alias, - "Foundry Local Whisper prepare failed", - message.clone(), - ), - ); - Err(message) - } - } -} - -#[tauri::command] -pub fn foundry_local_asr_cancel_prepare( - runtime: State<'_, Arc>, -) -> Result<(), String> { - runtime.request_cancel_prepare(); - Ok(()) -} - -#[tauri::command] -pub async fn foundry_local_asr_release( - runtime: State<'_, Arc>, -) -> Result<(), String> { - runtime.release_now().await.map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub async fn foundry_local_asr_model_dir( - runtime: State<'_, Arc>, - model_alias: String, -) -> Result { - validate_foundry_model_alias(&model_alias)?; - runtime - .model_dir_for_alias(&model_alias) - .await - .map(|path| path.display().to_string()) - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub async fn foundry_local_asr_delete_model( - runtime: State<'_, Arc>, - model_alias: String, -) -> Result<(), String> { - validate_foundry_model_alias(&model_alias)?; - runtime - .delete_model(&model_alias) - .await - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub async fn foundry_local_asr_reveal_model_dir( - runtime: State<'_, Arc>, - model_alias: String, -) -> Result<(), String> { - validate_foundry_model_alias(&model_alias)?; - let dir = runtime - .model_dir_for_alias(&model_alias) - .await - .map_err(|e| format!("{e:#}"))?; - open_path_in_file_manager(&dir) -} - -fn emit_foundry_prepare_progress(app: &AppHandle, payload: FoundryPrepareProgressPayload) { - if let Err(error) = app.emit("foundry-local-asr-prepare-progress", payload) { - log::warn!("[foundry-asr] emit prepare progress failed: {error}"); - } -} - -// ───────────── Windows local ASR (sherpa-onnx-local, offline batch + online) ───────────── -// -// 命令形态与 Foundry 同形，让前端命令封装可以复用同一种 hook 模式；当前支持 -// catalog / 下载 / prepare / release / 删除 / 状态查询，推理由 coordinator 的 -// 听写链路触发。offline 模型停止录音后 batch decode；online 模型录音时输出 partial。 - -fn active_sherpa_model_from_prefs(prefs: &UserPreferences) -> String { - if sherpa_model_alias_is_known(&prefs.sherpa_onnx_model) { - prefs.sherpa_onnx_model.clone() - } else { - SHERPA_DEFAULT_MODEL_ALIAS.to_string() - } -} - -fn validate_sherpa_model_alias(model_alias: &str) -> Result<(), String> { - if sherpa_model_alias_is_known(model_alias) { - Ok(()) - } else { - Err(format!("unknown sherpa-onnx model alias: {model_alias}")) - } -} - -fn normalize_sherpa_language_hint(language_hint: &str) -> Result { - let normalized = language_hint.trim().to_lowercase(); - if normalized.is_empty() - || normalized - .chars() - .all(|c| c.is_ascii_lowercase() || c == '-') - { - Ok(normalized) - } else { - Err("language hint must be empty or BCP-47 lowercase code".to_string()) - } -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_status( - coord: CoordinatorState<'_>, - runtime: State<'_, Arc>, -) -> Result { - let prefs = coord.prefs().get(); - let active_model = active_sherpa_model_from_prefs(&prefs); - Ok(runtime.status_snapshot(&active_model).await) -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_catalog( - runtime: State<'_, Arc>, -) -> Result, String> { - runtime - .catalog_snapshot() - .await - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_fetch_remote_info( - model_alias: String, - mirror: Option, -) -> Result { - validate_sherpa_model_alias(&model_alias)?; - let mirror = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); - fetch_sherpa_remote_info(&model_alias, mirror) - .await - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_download_model( - app: AppHandle, - manager: State<'_, Arc>, - model_alias: String, - mirror: Option, -) -> Result<(), String> { - validate_sherpa_model_alias(&model_alias)?; - let mirror = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); - manager.start(app, model_alias, mirror); - Ok(()) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_cancel_download( - manager: State<'_, Arc>, - model_alias: String, -) -> Result<(), String> { - validate_sherpa_model_alias(&model_alias)?; - manager.cancel(&model_alias); - Ok(()) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_set_model( - coord: CoordinatorState<'_>, - model_alias: String, -) -> Result<(), String> { - validate_sherpa_model_alias(&model_alias)?; - let mut prefs = coord.prefs().get(); - if prefs.sherpa_onnx_model == model_alias { - return Ok(()); - } - prefs.sherpa_onnx_model = model_alias; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_set_language_hint( - coord: CoordinatorState<'_>, - language_hint: String, -) -> Result<(), String> { - let normalized = normalize_sherpa_language_hint(&language_hint)?; - let mut prefs = coord.prefs().get(); - if prefs.sherpa_onnx_language_hint == normalized { - return Ok(()); - } - prefs.sherpa_onnx_language_hint = normalized; - coord.prefs().set(prefs).map_err(|e| e.to_string()) -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_prepare( - app: AppHandle, - runtime: State<'_, Arc>, - model_alias: String, -) -> Result { - validate_sherpa_model_alias(&model_alias)?; - let progress_app = app.clone(); - let result = runtime - .ensure_loaded_with_progress(&model_alias, move |payload| { - emit_sherpa_prepare_progress(&progress_app, payload); - }) - .await; - match result { - Ok(loaded) => Ok(loaded), - Err(error) => { - let message = format!("{error:#}"); - emit_sherpa_prepare_progress( - &app, - SherpaPrepareProgressPayload::failed( - model_alias, - "sherpa-onnx prepare failed", - message.clone(), - ), - ); - Err(message) - } - } -} - -#[tauri::command] -pub fn sherpa_onnx_asr_cancel_prepare( - runtime: State<'_, Arc>, -) -> Result<(), String> { - runtime.request_cancel_prepare(); - Ok(()) -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_release( - runtime: State<'_, Arc>, -) -> Result<(), String> { - runtime.release_now().await.map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_model_dir(model_alias: String) -> Result { - validate_sherpa_model_alias(&model_alias)?; - SherpaOnnxRuntime::model_dir_for_alias(&model_alias) - .map(|path| path.display().to_string()) - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub async fn sherpa_onnx_asr_delete_model( - runtime: State<'_, Arc>, - model_alias: String, -) -> Result<(), String> { - validate_sherpa_model_alias(&model_alias)?; - runtime - .delete_model(&model_alias) - .await - .map_err(|e| format!("{e:#}")) -} - -#[tauri::command] -pub fn sherpa_onnx_asr_reveal_model_dir(model_alias: String) -> Result<(), String> { - validate_sherpa_model_alias(&model_alias)?; - let dir = SherpaOnnxRuntime::model_dir_for_alias(&model_alias).map_err(|e| format!("{e:#}"))?; - std::fs::create_dir_all(&dir).map_err(|e| format!("create {} failed: {e}", dir.display()))?; - open_path_in_file_manager(&dir) -} - -#[cfg(target_os = "windows")] -fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { - use windows::core::PCWSTR; - use windows::Win32::UI::Shell::ShellExecuteW; - use windows::Win32::UI::WindowsAndMessaging::SW_SHOWNORMAL; - - fn wide_null(value: &str) -> Vec { - value.encode_utf16().chain(std::iter::once(0)).collect() - } - - let operation = wide_null("open"); - let target = wide_null(&path.display().to_string()); - let result = unsafe { - ShellExecuteW( - None, - PCWSTR(operation.as_ptr()), - PCWSTR(target.as_ptr()), - PCWSTR::null(), - PCWSTR::null(), - SW_SHOWNORMAL, - ) - }; - if result.0 as isize <= 32 { - Err(format!("ShellExecuteW failed: {}", result.0 as isize)) - } else { - Ok(()) - } -} - -#[cfg(target_os = "macos")] -fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { - std::process::Command::new("open") - .arg(path) - .spawn() - .map(|_| ()) - .map_err(|e| e.to_string()) -} - -#[cfg(all(unix, not(target_os = "macos")))] -fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { - std::process::Command::new("xdg-open") - .arg(path) - .spawn() - .map(|_| ()) - .map_err(|e| e.to_string()) -} - -fn emit_sherpa_prepare_progress(app: &AppHandle, payload: SherpaPrepareProgressPayload) { - if let Err(error) = app.emit("sherpa-onnx-asr-prepare-progress", payload) { - log::warn!("[sherpa-asr] emit prepare progress failed: {error}"); - } -} - -/// 把当前会话的 openless.log 复制到用户选择的位置（前端用 plugin-dialog 拿 target_path）。 -/// 路径来自 lib::log_dir_path() —— mac: ~/Library/Logs/OpenLess/openless.log， -/// windows: %LOCALAPPDATA%\OpenLess\Logs\openless.log。 -#[tauri::command] -pub fn export_error_log(target_path: String) -> Result<(), String> { - let src = crate::log_dir_path().join("openless.log"); - if !src.exists() { - return Err(format!("日志文件不存在：{}", src.display())); - } - std::fs::copy(&src, std::path::Path::new(&target_path)) - .map(|_| ()) - .map_err(|e| format!("复制日志失败：{}", e)) -} - -// ─────────────────────────── unused but exported (silences dead_code) ─────────────────────────── - -#[allow(dead_code)] -fn _ensure_snapshot_used(_: CredentialsSnapshot) {} - -// ─────────────────────────── marketplace (Phase A) ─────────────────────────── -// -// 客户端跟 marketplace backend 的 HTTP 客户端封装。Backend URL 走 prefs -// `marketplace_base_url`（默认 http://127.0.0.1:8090 开发；生产用户填 https://api.）。 -// dev-mode auth：用户在 Settings 填 `marketplace_dev_login`（GitHub 风格 username）， -// 后续 OAuth 接入时换成 token 字段。 -// -// 5 个 IPC： -// - marketplace_list 列表 + 搜索 + 排序 -// - marketplace_detail 详情（含完整 prompt） -// - marketplace_install 下载 ZIP + 直接调 import_from_zip 装到本地 -// - marketplace_upload 把本地某个 style pack export ZIP → multipart 上传 -// - marketplace_like 点赞 - -#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] -#[serde(rename_all = "camelCase")] -pub struct MarketplaceListItem { - pub id: String, - pub slug: String, - pub name: String, - pub description: String, - #[serde(default)] - pub author_login: String, - pub version: String, - pub base_mode: String, - #[serde(default)] - pub tags: Vec, - pub like_count: i64, - pub download_count: i64, - pub published_at: String, - pub updated_at: String, - pub origin_pack_id: Option, - pub origin_author_login: Option, -} - -#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] -#[serde(rename_all = "camelCase")] -pub struct MarketplaceDetail { - #[serde(flatten)] - pub summary: MarketplaceListItem, - pub prompt: String, - pub state: String, -} - -#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] -#[serde(rename_all = "camelCase")] -pub struct MarketplaceMyPackItem { - #[serde(flatten)] - pub summary: MarketplaceListItem, - pub state: String, -} - -/// 风格市场 backend URL —— 硬编码到生产云端，不再读 prefs。 -/// -/// 历史上这里读 `prefs.marketplace_base_url`（dev 本地可填 127.0.0.1:8090）， -/// 现在风格市场已经稳定部署在 apic.openless.top，把 URL 锁死避免用户误改 / 写错。 -/// 参数 `_prefs` 保留是为不动调用点签名；将来需要白名单 / 多 endpoint 时再开口。 -const MARKETPLACE_BASE_URL: &str = "https://apic.openless.top"; - -fn marketplace_url_from_prefs(_prefs: &UserPreferences) -> String { - MARKETPLACE_BASE_URL.to_string() -} - -fn marketplace_dev_user(prefs: &UserPreferences) -> String { - prefs.marketplace_dev_login.trim().to_string() -} - -#[tauri::command] -pub async fn marketplace_list( - coord: CoordinatorState<'_>, - query: Option, - sort: Option, - limit: Option, -) -> Result, String> { - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let mut url = reqwest::Url::parse(&format!("{base}/packs")) - .map_err(|e| format!("invalid marketplace url: {e}"))?; - if let Some(q) = query.as_deref() { - if !q.trim().is_empty() { - url.query_pairs_mut().append_pair("q", q.trim()); - } - } - if let Some(s) = sort.as_deref() { - if !s.trim().is_empty() { - url.query_pairs_mut().append_pair("sort", s.trim()); - } - } - if let Some(n) = limit { - url.query_pairs_mut().append_pair("limit", &n.to_string()); - } - let resp = net::send_with_retry(|| { - net::http() - .get(url.clone()) - .timeout(std::time::Duration::from_secs(10)) - }) - .await - .map_err(|e| format!("marketplace request failed: {e}"))?; - if !resp.status().is_success() { - let status = resp.status(); - let body = resp.text().await.unwrap_or_default(); - return Err(format!("marketplace HTTP {status}: {body}")); - } - let items: Vec = resp - .json() - .await - .map_err(|e| format!("parse failed: {e}"))?; - Ok(items) -} - -#[tauri::command] -pub async fn marketplace_detail( - coord: CoordinatorState<'_>, - pack_id: String, -) -> Result { - if !is_valid_session_id(&pack_id) { - return Err("invalid pack id".into()); - } - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let url = format!("{base}/packs/{pack_id}"); - let resp = net::send_with_retry(|| { - net::http() - .get(&url) - .timeout(std::time::Duration::from_secs(10)) - }) - .await - .map_err(|e| format!("marketplace request failed: {e}"))?; - if !resp.status().is_success() { - let status = resp.status(); - return Err(format!("marketplace HTTP {status}")); - } - resp.json::() - .await - .map_err(|e| format!("parse failed: {e}")) -} - -#[tauri::command] -pub async fn marketplace_install( - coord: CoordinatorState<'_>, - pack_id: String, -) -> Result { - // 安全校验：pack_id 来自远端 backend，可能含路径遍历 segment。 - // 用跟 read_audio_recording 同样的 UUID-v4 白名单挡住 ../ / 绝对路径等。 - // backend 当前用 Uuid::new_v4 生成所有 id，合法 id 必然匹配。 - if !is_valid_session_id(&pack_id) { - return Err("invalid pack id".into()); - } - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - - // 先拉 detail 拿 authorLogin —— 装好后本地写 originAuthorLogin， - // 后续编辑+发布时 backend 据此判 supersede（原作者）vs derivative（他人 fork）。 - let detail_url = format!("{base}/packs/{pack_id}"); - let detail: serde_json::Value = net::send_with_retry(|| { - net::http() - .get(&detail_url) - .timeout(std::time::Duration::from_secs(15)) - }) - .await - .map_err(|e| format!("marketplace detail failed: {e}"))? - .error_for_status() - .map_err(|e| format!("marketplace detail HTTP error: {e}"))? - .json() - .await - .map_err(|e| format!("parse detail failed: {e}"))?; - let origin_author_login = detail - .get("authorLogin") - .and_then(|v| v.as_str()) - .map(|s| s.to_string()); - - let download_url = format!("{base}/packs/{pack_id}/download"); - let bytes = net::send_with_retry(|| { - net::http() - .get(&download_url) - .timeout(std::time::Duration::from_secs(30)) - }) - .await - .map_err(|e| format!("marketplace download failed: {e}"))? - .error_for_status() - .map_err(|e| format!("marketplace HTTP error: {e}"))? - .bytes() - .await - .map_err(|e| format!("read body failed: {e}"))?; - - // pack_id 已经过 UUID 白名单，拼临时文件路径安全。 - let tmp = std::env::temp_dir().join(format!("openless-marketplace-{pack_id}.zip")); - std::fs::write(&tmp, &bytes).map_err(|e| format!("write tmp zip: {e}"))?; - let imported_result = coord - .style_packs() - .import_from_zip(&tmp) - .map_err(|e| e.to_string()); - let _ = std::fs::remove_file(&tmp); - let imported = imported_result?; - - // 绑定 origin —— 后续编辑+发布走 derivative / supersede 分支。 - coord - .style_packs() - .set_origin(&imported.id, Some(pack_id), origin_author_login) - .map_err(|e| format!("set origin failed: {e}")) -} - -#[tauri::command] -pub async fn marketplace_upload( - coord: CoordinatorState<'_>, - pack_id: String, - origin_pack_id: Option, -) -> Result { - // 本地 pack id 形态：`builtin.light` / 用户 slug / Uuid。用 local 白名单挡 `..` / `/` / `\`。 - if !is_valid_local_pack_id(&pack_id) { - return Err("invalid pack id".into()); - } - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let dev_user = marketplace_dev_user(&prefs); - if dev_user.is_empty() { - return Err("未登录：先在 Settings 填发布者名字".into()); - } - - // 拉本地 pack 拿 origin_pack_id —— 装过的 pack 这里有值， - // backend 据此判同作者就 supersede 原行（新版本），他人就 derivative（独立新 row）。 - let local_pack = coord - .style_packs() - .get(&pack_id) - .map_err(|e| format!("local pack not found: {e}"))?; - let origin_pack_id = origin_pack_id - .filter(|id| is_valid_session_id(id)) - .or_else(|| local_pack.origin_pack_id.clone()); - - // 先 export 本地 pack → 临时 ZIP - let tmp = std::env::temp_dir().join(format!("openless-marketplace-upload-{pack_id}.zip")); - coord - .style_packs() - .export_to_zip(&pack_id, &tmp) - .map_err(|e| format!("export local pack failed: {e}"))?; - let bytes = std::fs::read(&tmp).map_err(|e| format!("read exported zip: {e}"))?; - let _ = std::fs::remove_file(&tmp); - - // multipart 上传：表单是流式 body，不走 send_with_retry 的闭包重试；改用共享 - // 客户端 —— 之前 list/detail 命令若已打开过连接，这里直接复用连接池里的连接。 - let part = reqwest::multipart::Part::bytes(bytes) - .file_name(format!("{pack_id}.zip")) - .mime_str("application/zip") - .map_err(|e| format!("multipart build failed: {e}"))?; - let mut form = reqwest::multipart::Form::new().part("file", part); - if let Some(ref oid) = origin_pack_id { - form = form.text("origin_pack_id", oid.clone()); - } - let resp = net::http() - .post(format!("{base}/packs")) - .header("X-Dev-User", dev_user) - .timeout(std::time::Duration::from_secs(30)) - .multipart(form) - .send() - .await - .map_err(|e| format!("upload request failed: {e}"))?; - let status = resp.status(); - let body = resp - .text() - .await - .unwrap_or_else(|e| format!("read body failed: {e}")) - .clone(); - if !status.is_success() { - return Err(format!("upload HTTP {status}: {body}")); - } - let parsed = serde_json::from_str::(&body) - .map_err(|e| format!("parse upload response failed: {e}"))?; - - // 本地从未绑定 origin（首次上传一个本地原创 pack）→ 把 backend 分配的 pack id 写回本地， - // 让用户在同设备上后续编辑能继续走「同作者 supersede」分支，更新自己原创的包。 - if origin_pack_id.is_none() { - if let Some(remote_id) = parsed.get("id").and_then(|v| v.as_str()) { - let prefs2 = coord.prefs().get(); - let dev_user2 = marketplace_dev_user(&prefs2); - let _ = coord.style_packs().set_origin( - &pack_id, - Some(remote_id.to_string()), - Some(dev_user2), - ); - } - } - - Ok(parsed) -} - -#[tauri::command] -pub async fn marketplace_like( - coord: CoordinatorState<'_>, - pack_id: String, -) -> Result { - if !is_valid_session_id(&pack_id) { - return Err("invalid pack id".into()); - } - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let dev_user = marketplace_dev_user(&prefs); - if dev_user.is_empty() { - return Err("未登录：先在 Settings 填发布者名字".into()); - } - let like_url = format!("{base}/packs/{pack_id}/like"); - let resp = net::send_with_retry(|| { - net::http() - .post(&like_url) - .header("X-Dev-User", dev_user.as_str()) - .timeout(std::time::Duration::from_secs(10)) - }) - .await - .map_err(|e| format!("like request failed: {e}"))?; - if !resp.status().is_success() { - return Err(format!("like HTTP {}", resp.status())); - } - resp.json::() - .await - .map_err(|e| format!("parse failed: {e}")) -} - -/// 撤回自己发布的 pack（后端软删 state='withdrawn'，前端列表不再可见）。 -/// pack_id 来自远端，必须是 UUID-v4。 -#[tauri::command] -pub async fn marketplace_delete( - coord: CoordinatorState<'_>, - pack_id: String, -) -> Result<(), String> { - if !is_valid_session_id(&pack_id) { - return Err("invalid pack id".into()); - } - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let dev_user = marketplace_dev_user(&prefs); - if dev_user.is_empty() { - return Err("未登录：先在 Settings 填发布者名字".into()); - } - let delete_url = format!("{base}/packs/{pack_id}"); - let resp = net::send_with_retry(|| { - net::http() - .delete(&delete_url) - .header("X-Dev-User", dev_user.as_str()) - .timeout(std::time::Duration::from_secs(15)) - }) - .await - .map_err(|e| format!("delete request failed: {e}"))?; - let status = resp.status(); - if !status.is_success() { - let body = resp.text().await.unwrap_or_default(); - return Err(format!("delete HTTP {status}: {body}")); - } - Ok(()) -} - -/// 拉当前用户赞过的所有 pack id，用于客户端市场页面渲染红心 + 「我赞过的」过滤。 -#[tauri::command] -pub async fn marketplace_my_likes(coord: CoordinatorState<'_>) -> Result, String> { - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let dev_user = marketplace_dev_user(&prefs); - if dev_user.is_empty() { - return Ok(Vec::new()); // 未登录就空集合，UI 渲染无红心 - } - let likes_url = format!("{base}/me/likes"); - let resp = net::send_with_retry(|| { - net::http() - .get(&likes_url) - .header("X-Dev-User", dev_user.as_str()) - .timeout(std::time::Duration::from_secs(10)) - }) - .await - .map_err(|e| format!("my-likes request failed: {e}"))?; - if !resp.status().is_success() { - return Err(format!("my-likes HTTP {}", resp.status())); - } - resp.json::>() - .await - .map_err(|e| format!("parse my-likes failed: {e}")) -} - -/// 拉当前用户发布过的 pack（含审核中/已通过/已拒绝/已撤回），用于「我的发布」页面。 -#[tauri::command] -pub async fn marketplace_my_packs( - coord: CoordinatorState<'_>, -) -> Result, String> { - let prefs = coord.prefs().get(); - let base = marketplace_url_from_prefs(&prefs); - let dev_user = marketplace_dev_user(&prefs); - if dev_user.is_empty() { - return Ok(Vec::new()); - } - let packs_url = format!("{base}/me/packs"); - let resp = net::send_with_retry(|| { - net::http() - .get(&packs_url) - .header("X-Dev-User", dev_user.as_str()) - .timeout(std::time::Duration::from_secs(10)) - }) - .await - .map_err(|e| format!("my-packs request failed: {e}"))?; - if !resp.status().is_success() { - return Err(format!("my-packs HTTP {}", resp.status())); - } - resp.json::>() - .await - .map_err(|e| format!("parse my-packs failed: {e}")) -} - -// ─────────────────────── GitHub OAuth Device Flow (Phase 1) ─────────────────────── -// -// 客户端直连 GitHub 拿 access_token + login，前端自动把 login 写进 -// prefs.marketplaceDevLogin。marketplace backend 完全不动（依然 X-Dev-User）。 -// Phase 2 才会让 backend 验证 GitHub identity（JWT 签发 + 防伪造）。 -// -// 配置 client_id 的两种方式（OAuth App client_id 非敏感，可硬编码）： -// 1. 在下方 GITHUB_OAUTH_CLIENT_ID 常量填值（生产推荐 — 直接 bake 进二进制） -// 2. 启动前设置环境变量 GITHUB_OAUTH_CLIENT_ID=（dev 方便） -// -// 注册 OAuth App： -// https://github.com/settings/applications/new -// - Application name: OpenLess (or your fork) -// - Homepage URL: https://openless.top (or任意) -// - Authorization callback URL: https://openless.top (Device Flow 不真用，但表单要求填) -// - 创建后在 General 页面勾选 "Enable Device Flow" -// - 抄 client_id 填到本常量 - -const GITHUB_OAUTH_CLIENT_ID: &str = "Ov23liyv3nEucG7oMHNE"; - -fn get_github_oauth_client_id() -> Result { - if let Ok(env_id) = std::env::var("GITHUB_OAUTH_CLIENT_ID") { - let trimmed = env_id.trim(); - if !trimmed.is_empty() { - return Ok(trimmed.to_string()); - } - } - if !GITHUB_OAUTH_CLIENT_ID.is_empty() { - return Ok(GITHUB_OAUTH_CLIENT_ID.to_string()); - } - Err( - "GitHub OAuth 未配置。请去 https://github.com/settings/applications/new 注册一个 OAuth App\ - （必须勾 Enable Device Flow），把 client_id 填到 \ - openless-all/app/src-tauri/src/commands.rs 的 GITHUB_OAUTH_CLIENT_ID 常量，\ - 或在启动前设置环境变量 GITHUB_OAUTH_CLIENT_ID=。" - .to_string(), - ) -} - -#[derive(Debug, serde::Serialize)] -#[serde(rename_all = "camelCase")] -pub struct GithubDeviceStartResponse { - pub device_code: String, - pub user_code: String, - pub verification_uri: String, - pub interval: u32, - pub expires_in: u32, -} - -#[tauri::command] -pub async fn github_device_flow_start() -> Result { - let client_id = get_github_oauth_client_id()?; - let resp = net::send_with_retry(|| { - net::http() - .post("https://github.com/login/device/code") - .header("Accept", "application/json") - .header("User-Agent", "OpenLess") - .timeout(std::time::Duration::from_secs(15)) - .form(&[("client_id", client_id.as_str()), ("scope", "read:user")]) - }) - .await - .map_err(|e| format!("调用 GitHub /login/device/code 失败：{e}"))?; - let status = resp.status(); - let body: serde_json::Value = resp - .json() - .await - .map_err(|e| format!("解析 device/code 响应失败：{e}"))?; - if !status.is_success() { - let err = body["error"].as_str().unwrap_or("unknown_error"); - let desc = body["error_description"].as_str().unwrap_or(""); - return Err(format!("GitHub device/code {status} {err}: {desc}")); - } - Ok(GithubDeviceStartResponse { - device_code: body["device_code"].as_str().unwrap_or("").to_string(), - user_code: body["user_code"].as_str().unwrap_or("").to_string(), - verification_uri: body["verification_uri"] - .as_str() - .unwrap_or("https://github.com/login/device") - .to_string(), - interval: body["interval"].as_u64().unwrap_or(5) as u32, - expires_in: body["expires_in"].as_u64().unwrap_or(900) as u32, - }) -} - -#[derive(Debug, serde::Serialize)] -#[serde(rename_all = "camelCase", tag = "kind")] -pub enum GithubDevicePollResult { - Authorized { login: String }, - Pending, - SlowDown, - Error { message: String }, -} - -#[tauri::command] -pub async fn github_device_flow_poll( - device_code: String, -) -> Result { - let client_id = get_github_oauth_client_id()?; - let token_resp = net::send_with_retry(|| { - net::http() - .post("https://github.com/login/oauth/access_token") - .header("Accept", "application/json") - .header("User-Agent", "OpenLess") - .timeout(std::time::Duration::from_secs(15)) - .form(&[ - ("client_id", client_id.as_str()), - ("device_code", device_code.as_str()), - ("grant_type", "urn:ietf:params:oauth:grant-type:device_code"), - ]) - }) - .await - .map_err(|e| format!("调用 GitHub /login/oauth/access_token 失败：{e}"))?; - let body: serde_json::Value = token_resp - .json() - .await - .map_err(|e| format!("解析 access_token 响应失败：{e}"))?; - - if let Some(token) = body["access_token"].as_str() { - let user_resp = net::send_with_retry(|| { - net::http() - .get("https://api.github.com/user") - .header("User-Agent", "OpenLess") - .header("Accept", "application/vnd.github+json") - .timeout(std::time::Duration::from_secs(15)) - .bearer_auth(token) - }) - .await - .map_err(|e| format!("调用 GitHub /user 失败：{e}"))?; - let user_body: serde_json::Value = user_resp - .json() - .await - .map_err(|e| format!("解析 /user 响应失败：{e}"))?; - let login = user_body["login"].as_str().unwrap_or("").to_string(); - if login.is_empty() { - return Ok(GithubDevicePollResult::Error { - message: "GitHub /user 返回空 login".to_string(), - }); - } - return Ok(GithubDevicePollResult::Authorized { login }); - } - - let err = body["error"].as_str().unwrap_or(""); - let msg = match err { - "authorization_pending" => return Ok(GithubDevicePollResult::Pending), - "slow_down" => return Ok(GithubDevicePollResult::SlowDown), - "expired_token" => "OAuth 设备码已过期，请重新发起登录".to_string(), - "access_denied" => "你在 GitHub 上拒绝了授权".to_string(), - other if !other.is_empty() => format!("OAuth 错误：{other}"), - _ => "未知 OAuth 错误（access_token 缺失）".to_string(), - }; - Ok(GithubDevicePollResult::Error { message: msg }) -} - -#[cfg(test)] -mod tests { - use super::{ - active_asr_is_keyless_for_validation, active_foundry_model_from_prefs, - active_sherpa_model_from_prefs, asr_configured_for_provider, asr_transcriptions_url, - fetch_provider_models, is_gemini_base_url, is_valid_local_pack_id, is_valid_session_id, - llm_configured_for_provider, local_asr_release_plan_for_provider, models_url, - normalize_foundry_language_hint, normalize_sherpa_language_hint, parse_gemini_model_ids, - parse_latest_beta_from_atom, parse_model_ids, persist_settings, - release_foundry_runtime_if_inactive, release_sherpa_runtime_if_inactive, - validate_foundry_model_alias, validate_sherpa_model_alias, ProviderConfig, SettingsWriter, - }; - use crate::persistence::CredentialsSnapshot; - use crate::types::{ - ComboBinding, HotkeyBinding, HotkeyMode, HotkeyTrigger, ShortcutBinding, UserPreferences, - }; - use std::io::{Read, Write}; - use std::net::TcpListener; - use std::sync::Mutex; - use std::thread; - - #[derive(Default)] - struct FakeSettingsWriter { - saved: Mutex>, - active_asr_provider_syncs: Mutex>, - write_settings_error: Mutex>, - write_settings_errors: Mutex>>, - active_asr_provider_sync_error: Mutex>, - active_asr_provider_sync_errors: Mutex>>, - dictation_refreshes: Mutex, - qa_refreshes: Mutex, - combo_refreshes: Mutex, - translation_refreshes: Mutex, - switch_style_refreshes: Mutex, - open_app_refreshes: Mutex, - coding_agent_refreshes: Mutex, - } - - fn snapshot() -> CredentialsSnapshot { - CredentialsSnapshot::default() - } - - #[test] - fn credentials_status_follows_active_asr_provider_requirements() { - let volcengine = CredentialsSnapshot { - volcengine_app_key: Some("app".into()), - volcengine_access_key: Some("access".into()), - volcengine_resource_id: Some("resource".into()), - ..snapshot() - }; - assert!(asr_configured_for_provider("volcengine", &volcengine)); - - let whisper_key_only = CredentialsSnapshot { - asr_api_key: Some("key".into()), - ..snapshot() - }; - assert!(!asr_configured_for_provider("whisper", &whisper_key_only)); - assert!(asr_configured_for_provider( - crate::asr::bailian::PROVIDER_ID, - &whisper_key_only - )); - - let whisper_keyless_ready = CredentialsSnapshot { - asr_endpoint: Some("https://api.openai.com/v1".into()), - asr_model: Some("whisper-1".into()), - ..snapshot() - }; - assert!(asr_configured_for_provider( - "whisper", - &whisper_keyless_ready - )); - assert!(!asr_configured_for_provider( - crate::asr::bailian::PROVIDER_ID, - &whisper_keyless_ready - )); - - assert!(asr_configured_for_provider( - crate::asr::local::PROVIDER_ID, - &snapshot() - )); - #[cfg(target_os = "windows")] - assert!(asr_configured_for_provider( - crate::asr::local::foundry::PROVIDER_ID, - &snapshot() - )); - #[cfg(target_os = "windows")] - assert!(asr_configured_for_provider( - crate::asr::local::sherpa::PROVIDER_ID, - &snapshot() - )); - #[cfg(not(target_os = "windows"))] - assert!(!asr_configured_for_provider( - crate::asr::local::foundry::PROVIDER_ID, - &snapshot() - )); - #[cfg(not(target_os = "windows"))] - assert!(!asr_configured_for_provider( - crate::asr::local::sherpa::PROVIDER_ID, - &snapshot() - )); - } - - #[test] - fn credentials_status_treats_foundry_local_asr_as_configured() { - #[cfg(target_os = "windows")] - { - assert!(asr_configured_for_provider( - crate::asr::local::foundry::PROVIDER_ID, - &CredentialsSnapshot::default() - )); - } - #[cfg(not(target_os = "windows"))] - { - assert!(!asr_configured_for_provider( - crate::asr::local::foundry::PROVIDER_ID, - &CredentialsSnapshot::default() - )); - } - } - - #[test] - fn local_asr_providers_skip_external_validation() { - assert!(active_asr_is_keyless_for_validation( - crate::asr::local::PROVIDER_ID - )); - #[cfg(target_os = "windows")] - assert!(active_asr_is_keyless_for_validation( - crate::asr::local::foundry::PROVIDER_ID - )); - #[cfg(target_os = "windows")] - assert!(active_asr_is_keyless_for_validation( - crate::asr::local::sherpa::PROVIDER_ID - )); - #[cfg(not(target_os = "windows"))] - assert!(!active_asr_is_keyless_for_validation( - crate::asr::local::foundry::PROVIDER_ID - )); - #[cfg(not(target_os = "windows"))] - assert!(!active_asr_is_keyless_for_validation( - crate::asr::local::sherpa::PROVIDER_ID - )); - assert!(!active_asr_is_keyless_for_validation("volcengine")); - assert!(!active_asr_is_keyless_for_validation("whisper")); - } - - #[test] - fn provider_switch_release_plan_covers_inactive_local_runtimes() { - let qwen = local_asr_release_plan_for_provider(crate::asr::local::PROVIDER_ID); - assert!(!qwen.qwen); - assert!(qwen.foundry); - assert!(qwen.sherpa); - - let foundry = local_asr_release_plan_for_provider(crate::asr::local::foundry::PROVIDER_ID); - assert!(foundry.qwen); - assert!(!foundry.foundry); - assert!(foundry.sherpa); - - let sherpa = local_asr_release_plan_for_provider(crate::asr::local::sherpa::PROVIDER_ID); - assert!(sherpa.qwen); - assert!(sherpa.foundry); - assert!(!sherpa.sherpa); - - let cloud = local_asr_release_plan_for_provider("volcengine"); - assert!(cloud.qwen); - assert!(cloud.foundry); - assert!(cloud.sherpa); - } - - #[cfg(target_os = "windows")] - #[tokio::test] - async fn provider_switch_release_requests_foundry_prepare_cancel_first() { - let runtime = std::sync::Arc::new(crate::asr::local::FoundryLocalRuntime::new()); - - release_foundry_runtime_if_inactive(&runtime, true).await; - - assert!(runtime.cancel_prepare_requested_for_tests()); - } - - #[tokio::test] - async fn provider_switch_release_requests_sherpa_prepare_cancel_first() { - let runtime = std::sync::Arc::new(crate::asr::local::SherpaOnnxRuntime::new()); - - release_sherpa_runtime_if_inactive(&runtime, true).await; - - assert!(runtime.cancel_prepare_requested_for_tests()); - let status = runtime.status_snapshot("sense-voice-small-zh").await; - assert!(!status.runtime_ready); - } - - #[test] - fn foundry_language_hint_accepts_empty_and_lowercase_iso_639_1() { - assert_eq!(normalize_foundry_language_hint("").unwrap(), ""); - assert_eq!(normalize_foundry_language_hint(" ").unwrap(), ""); - assert_eq!(normalize_foundry_language_hint("zh").unwrap(), "zh"); - assert_eq!(normalize_foundry_language_hint(" en ").unwrap(), "en"); - } - - #[test] - fn foundry_language_hint_rejects_non_lowercase_iso_639_1() { - assert!(normalize_foundry_language_hint("ZH").is_err()); - assert!(normalize_foundry_language_hint("zho").is_err()); - assert!(normalize_foundry_language_hint("z1").is_err()); - } - - #[test] - fn foundry_model_alias_validation_rejects_unknown_alias() { - assert!( - validate_foundry_model_alias(crate::asr::local::foundry::DEFAULT_MODEL_ALIAS).is_ok() - ); - assert!(validate_foundry_model_alias("whisper-medium").is_ok()); - assert!(validate_foundry_model_alias("whisper-large-v3-turbo").is_ok()); - assert!(validate_foundry_model_alias("whisper-large").is_err()); - } - - #[test] - fn foundry_active_model_pref_falls_back_to_default_for_unknown_alias() { - let prefs = UserPreferences { - foundry_local_asr_model: "whisper-large".to_string(), - ..Default::default() - }; - - assert_eq!( - active_foundry_model_from_prefs(&prefs), - crate::asr::local::foundry::DEFAULT_MODEL_ALIAS - ); - } - - #[test] - fn foundry_active_model_pref_preserves_large_model_aliases() { - for alias in ["whisper-medium", "whisper-large-v3-turbo"] { - let prefs = UserPreferences { - foundry_local_asr_model: alias.to_string(), - ..Default::default() - }; - - assert_eq!(active_foundry_model_from_prefs(&prefs), alias); - } - } - - #[test] - fn sherpa_language_hint_accepts_empty_and_supported_lowercase_tags() { - assert_eq!(normalize_sherpa_language_hint("").unwrap(), ""); - assert_eq!(normalize_sherpa_language_hint(" ").unwrap(), ""); - assert_eq!(normalize_sherpa_language_hint("zh").unwrap(), "zh"); - assert_eq!(normalize_sherpa_language_hint(" en ").unwrap(), "en"); - assert_eq!(normalize_sherpa_language_hint("zh-cn").unwrap(), "zh-cn"); - assert_eq!(normalize_sherpa_language_hint("yue").unwrap(), "yue"); - } - - #[test] - fn sherpa_language_hint_normalizes_uppercase_and_rejects_digits() { - assert_eq!(normalize_sherpa_language_hint("ZH").unwrap(), "zh"); - assert!(normalize_sherpa_language_hint("zh-1").is_err()); - assert!(normalize_sherpa_language_hint("zh_CN").is_err()); - } - - #[test] - fn sherpa_model_alias_validation_matches_catalog() { - assert!( - validate_sherpa_model_alias(crate::asr::local::sherpa::DEFAULT_MODEL_ALIAS).is_ok() - ); - assert!(validate_sherpa_model_alias("qwen3-asr-0.6b-int8").is_ok()); - assert!( - validate_sherpa_model_alias(crate::asr::local::sherpa::DEFAULT_ONLINE_MODEL_ALIAS) - .is_ok() - ); - assert!(validate_sherpa_model_alias("zipformer-zh-streaming").is_err()); - } - - #[test] - fn sherpa_active_model_pref_falls_back_to_default_for_unknown_alias() { - let prefs = UserPreferences { - sherpa_onnx_model: "zipformer-zh-streaming".to_string(), - ..Default::default() - }; - - assert_eq!( - active_sherpa_model_from_prefs(&prefs), - crate::asr::local::sherpa::DEFAULT_MODEL_ALIAS - ); - } - - #[test] - fn credentials_status_accepts_keyless_custom_llm_only() { - let keyless_ready = CredentialsSnapshot { - ark_endpoint: Some("http://localhost:11434/v1".into()), - ark_model_id: Some("qwen".into()), - ..snapshot() - }; - assert!(llm_configured_for_provider("custom", &keyless_ready)); - assert!(llm_configured_for_provider("self-hosted", &keyless_ready)); - assert!(llm_configured_for_provider( - "openrouterFree", - &keyless_ready - )); - - let hosted_keyless = CredentialsSnapshot { - ark_endpoint: Some("https://openrouter.ai/api/v1".into()), - ark_model_id: Some("qwen/qwen3-coder:free".into()), - ..snapshot() - }; - assert!(!llm_configured_for_provider( - "openrouterFree", - &hosted_keyless - )); - - let hosted_ready = CredentialsSnapshot { - ark_api_key: Some("key".into()), - ark_endpoint: Some("https://openrouter.ai/api/v1/chat/completions".into()), - ark_model_id: Some("qwen/qwen3-coder:free".into()), - ..snapshot() - }; - assert!(llm_configured_for_provider("openrouterFree", &hosted_ready)); - - let key_without_endpoint = CredentialsSnapshot { - ark_api_key: Some("key".into()), - ark_model_id: Some("qwen".into()), - ..snapshot() - }; - assert!(!llm_configured_for_provider( - "custom", - &key_without_endpoint - )); - - let endpoint_without_model = CredentialsSnapshot { - ark_endpoint: Some("http://localhost:11434/v1".into()), - ..snapshot() - }; - assert!(!llm_configured_for_provider( - "custom", - &endpoint_without_model - )); - } - - impl SettingsWriter for FakeSettingsWriter { - fn read_settings(&self) -> UserPreferences { - self.saved.lock().unwrap().clone().unwrap_or_default() - } - - fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { - if let Some(error) = { - let mut errors = self.write_settings_errors.lock().unwrap(); - if errors.is_empty() { - None - } else { - errors.remove(0) - } - } { - return Err(error); - } - if let Some(error) = self.write_settings_error.lock().unwrap().clone() { - return Err(error); - } - *self.saved.lock().unwrap() = Some(prefs); - Ok(()) - } - - fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { - self.active_asr_provider_syncs - .lock() - .unwrap() - .push(provider.to_string()); - if let Some(error) = { - let mut errors = self.active_asr_provider_sync_errors.lock().unwrap(); - if errors.is_empty() { - None - } else { - errors.remove(0) - } - } { - return Err(error); - } - if let Some(error) = self.active_asr_provider_sync_error.lock().unwrap().clone() { - return Err(error); - } - Ok(()) - } - - fn refresh_dictation_hotkey(&self) { - *self.dictation_refreshes.lock().unwrap() += 1; - } - - fn refresh_qa_hotkey(&self) { - *self.qa_refreshes.lock().unwrap() += 1; - } - - fn refresh_combo_hotkey(&self) { - *self.combo_refreshes.lock().unwrap() += 1; - } - - fn refresh_translation_hotkey(&self) { - *self.translation_refreshes.lock().unwrap() += 1; - } - - fn refresh_switch_style_hotkey(&self) { - *self.switch_style_refreshes.lock().unwrap() += 1; - } - - fn refresh_open_app_hotkey(&self) { - *self.open_app_refreshes.lock().unwrap() += 1; - } - - fn refresh_coding_agent_hotkey(&self) { - *self.coding_agent_refreshes.lock().unwrap() += 1; - } - } - - #[test] - fn models_url_accepts_base_or_chat_endpoint() { - assert_eq!( - models_url("https://api.openai.com/v1"), - "https://api.openai.com/v1/models" - ); - assert_eq!( - models_url("https://api.openai.com/v1/chat/completions"), - "https://api.openai.com/v1/models" - ); - } - - #[test] - fn asr_transcriptions_url_accepts_base_or_transcriptions_endpoint() { - assert_eq!( - asr_transcriptions_url("https://api.openai.com/v1").unwrap(), - "https://api.openai.com/v1/audio/transcriptions" - ); - assert_eq!( - asr_transcriptions_url("https://api.openai.com/v1/chat/completions").unwrap(), - "https://api.openai.com/v1/audio/transcriptions" - ); - assert_eq!( - asr_transcriptions_url("https://api.openai.com/v1/audio").unwrap(), - "https://api.openai.com/v1/audio/transcriptions" - ); - assert_eq!( - asr_transcriptions_url("https://api.openai.com/v1/audio/transcriptions").unwrap(), - "https://api.openai.com/v1/audio/transcriptions" - ); - assert_eq!( - asr_transcriptions_url("https://api.openai.com/v1?api-version=2024-12-01").unwrap(), - "https://api.openai.com/v1/audio/transcriptions?api-version=2024-12-01" - ); - assert_eq!( - asr_transcriptions_url("http://192.168.1.10:8000/v1").unwrap(), - "http://192.168.1.10:8000/v1/audio/transcriptions" - ); - } - - #[test] - fn parse_model_ids_sorts_and_deduplicates() { - let models = - parse_model_ids(r#"{ "data": [{ "id": "b" }, { "id": "a" }, { "id": "b" }] }"#) - .unwrap(); - assert_eq!(models, vec!["a".to_string(), "b".to_string()]); - } - - #[test] - fn parse_gemini_model_ids_strips_models_prefix_and_dedups() { - // Google v1beta/models 真实响应的子集——name 字段带 `models/` 前缀， - // ProviderTools 选中后写入 ark.model_id 时不能带这个前缀（generateContent - // URL 拼接已经会加 `models/`，不去前缀就会变成 `models/models/...`）。 - // 字段缺失时保守保留（视为支持 generateContent）。 - let body = r#"{"models":[ - {"name":"models/gemini-2.5-pro"}, - {"name":"models/gemini-2.5-flash"}, - {"name":"models/gemini-2.5-flash"}, - {"name":"models/gemini-3-flash-preview"} - ]}"#; - let ids = parse_gemini_model_ids(body).unwrap(); - assert_eq!( - ids, - vec![ - "gemini-2.5-flash".to_string(), - "gemini-2.5-pro".to_string(), - "gemini-3-flash-preview".to_string(), - ] - ); - } - - #[test] - fn parse_gemini_model_ids_filters_out_non_generate_content_families() { - // 真实 Google v1beta/models 响应里同时有 generateContent / embedContent / - // generateMessage 等多种家族。用户选中 embedding/TTS/image 模型写入 - // ark.model_id → polish 必败。这里是 PR #398 pr_agent advisory 的回归用例： - // 只把 supportedGenerationMethods 里含 generateContent 的过滤出来。 - let body = r#"{"models":[ - {"name":"models/gemini-2.5-flash","supportedGenerationMethods":["generateContent","streamGenerateContent","countTokens"]}, - {"name":"models/gemini-embedding-2","supportedGenerationMethods":["embedContent"]}, - {"name":"models/text-embedding-004","supportedGenerationMethods":["embedContent","countTextTokens"]}, - {"name":"models/gemini-2.5-pro-preview-tts","supportedGenerationMethods":["generateContent"]}, - {"name":"models/gemini-2.5-flash-image","supportedGenerationMethods":["predict"]} - ]}"#; - let ids = parse_gemini_model_ids(body).unwrap(); - // 只剩两条声明 generateContent 的；embedding 与 image (predict-only) 必须被过滤。 - assert_eq!( - ids, - vec![ - "gemini-2.5-flash".to_string(), - "gemini-2.5-pro-preview-tts".to_string(), - ] - ); - } - - #[test] - fn is_gemini_base_url_matches_official_domain() { - assert!(is_gemini_base_url( - "https://generativelanguage.googleapis.com/v1beta" - )); - assert!(is_gemini_base_url( - "https://generativelanguage.googleapis.com/v1beta/" - )); - assert!(!is_gemini_base_url("https://api.openai.com/v1")); - assert!(!is_gemini_base_url( - "https://ark.cn-beijing.volces.com/api/v3" - )); - } - - #[test] - fn persist_settings_refreshes_changed_hotkey_pipelines() { - let writer = FakeSettingsWriter::default(); - let previous = UserPreferences::default(); - *writer.saved.lock().unwrap() = Some(previous); - let prefs = UserPreferences { - dictation_hotkey: ShortcutBinding { - primary: "D".to_string(), - modifiers: vec!["ctrl".to_string()], - }, - qa_hotkey: Some(ShortcutBinding { - primary: "Q".to_string(), - modifiers: vec!["ctrl".to_string(), "alt".to_string()], - }), - translation_hotkey: ShortcutBinding { - primary: "T".to_string(), - modifiers: vec!["ctrl".to_string(), "alt".to_string()], - }, - switch_style_hotkey: Some(ShortcutBinding { - primary: "S".to_string(), - modifiers: vec!["ctrl".to_string(), "alt".to_string()], - }), - open_app_hotkey: Some(ShortcutBinding { - primary: "O".to_string(), - modifiers: vec!["ctrl".to_string(), "alt".to_string()], - }), - coding_agent_voice_hotkey: Some(ShortcutBinding { - primary: "RightControl".to_string(), - modifiers: vec![], - }), - hotkey: HotkeyBinding { - trigger: HotkeyTrigger::Custom, - mode: HotkeyMode::Hold, - ..Default::default() - }, - ..Default::default() - }; - - persist_settings(&writer, prefs.clone()).unwrap(); - - let saved = writer - .saved - .lock() - .unwrap() - .clone() - .expect("settings saved"); - assert_eq!(saved.hotkey.trigger, HotkeyTrigger::Custom); - assert_eq!(saved.hotkey.mode, prefs.hotkey.mode); - assert_eq!( - saved.dictation_hotkey.primary, - prefs.dictation_hotkey.primary - ); - assert_eq!(saved.qa_hotkey.unwrap().primary, "Q"); - assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.combo_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.qa_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.translation_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 1); - assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 1); - } - - #[test] - fn persist_settings_rejects_less_computer_dictation_overlap() { - let writer = FakeSettingsWriter::default(); - let binding = ShortcutBinding { - primary: "LeftControl".into(), - modifiers: vec![], - }; - let prefs = UserPreferences { - dictation_hotkey: binding.clone(), - coding_agent_voice_hotkey: Some(binding), - ..Default::default() - }; - - assert_eq!( - persist_settings(&writer, prefs), - Err("Less Computer 快捷键不能和听写快捷键相同".into()) - ); - assert!(writer.saved.lock().unwrap().is_none()); - } - - #[test] - fn persist_settings_syncs_active_asr_provider_without_hotkey_refresh() { - let writer = FakeSettingsWriter::default(); - let previous = UserPreferences::default(); - *writer.saved.lock().unwrap() = Some(previous.clone()); - let prefs = UserPreferences { - active_asr_provider: "whisper".to_string(), - microphone_device_name: "External Mic".to_string(), - hotkey: previous.hotkey, - dictation_hotkey: previous.dictation_hotkey, - qa_hotkey: previous.qa_hotkey, - translation_hotkey: previous.translation_hotkey, - switch_style_hotkey: previous.switch_style_hotkey, - open_app_hotkey: previous.open_app_hotkey, - ..Default::default() - }; - - persist_settings(&writer, prefs.clone()).unwrap(); - - let saved = writer - .saved - .lock() - .unwrap() - .clone() - .expect("settings saved"); - assert_eq!(saved.active_asr_provider, prefs.active_asr_provider); - assert_eq!(saved.microphone_device_name, prefs.microphone_device_name); - assert_eq!( - writer.active_asr_provider_syncs.lock().unwrap().clone(), - vec![prefs.active_asr_provider.clone()] - ); - assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); - } - - #[test] - fn persist_settings_does_not_save_when_active_asr_sync_fails() { - let writer = FakeSettingsWriter::default(); - let previous = UserPreferences::default(); - *writer.saved.lock().unwrap() = Some(previous.clone()); - *writer.active_asr_provider_sync_error.lock().unwrap() = Some("sync failed".to_string()); - let prefs = UserPreferences { - active_asr_provider: "whisper".to_string(), - microphone_device_name: "External Mic".to_string(), - hotkey: previous.hotkey, - dictation_hotkey: previous.dictation_hotkey, - qa_hotkey: previous.qa_hotkey, - translation_hotkey: previous.translation_hotkey, - switch_style_hotkey: previous.switch_style_hotkey, - open_app_hotkey: previous.open_app_hotkey, - ..Default::default() - }; - - let error = persist_settings(&writer, prefs).unwrap_err(); - - assert_eq!(error, "sync failed"); - let saved = writer - .saved - .lock() - .unwrap() - .clone() - .expect("previous settings remain saved"); - assert_eq!(saved.active_asr_provider, previous.active_asr_provider); - assert_eq!( - saved.microphone_device_name, - previous.microphone_device_name - ); - assert_eq!( - writer.active_asr_provider_syncs.lock().unwrap().clone(), - vec!["whisper".to_string()] - ); - assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); - } - - #[test] - fn persist_settings_restores_active_asr_provider_when_save_fails_after_sync() { - let writer = FakeSettingsWriter::default(); - let previous = UserPreferences::default(); - *writer.saved.lock().unwrap() = Some(previous.clone()); - *writer.write_settings_error.lock().unwrap() = Some("save failed".to_string()); - let prefs = UserPreferences { - active_asr_provider: "whisper".to_string(), - microphone_device_name: "External Mic".to_string(), - hotkey: previous.hotkey, - dictation_hotkey: previous.dictation_hotkey, - qa_hotkey: previous.qa_hotkey, - translation_hotkey: previous.translation_hotkey, - switch_style_hotkey: previous.switch_style_hotkey, - open_app_hotkey: previous.open_app_hotkey, - ..Default::default() - }; - - let error = persist_settings(&writer, prefs).unwrap_err(); - - assert_eq!(error, "save failed"); - let saved = writer - .saved - .lock() - .unwrap() - .clone() - .expect("previous settings remain saved"); - assert_eq!(saved.active_asr_provider, previous.active_asr_provider); - assert_eq!( - saved.microphone_device_name, - previous.microphone_device_name - ); - assert_eq!( - writer.active_asr_provider_syncs.lock().unwrap().clone(), - vec!["whisper".to_string(), previous.active_asr_provider.clone()] - ); - assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); - } - - #[test] - fn persist_settings_keeps_new_active_asr_provider_when_rollback_fails() { - let writer = FakeSettingsWriter::default(); - let previous = UserPreferences::default(); - *writer.saved.lock().unwrap() = Some(previous.clone()); - *writer.write_settings_errors.lock().unwrap() = vec![Some("save failed".to_string()), None]; - *writer.active_asr_provider_sync_errors.lock().unwrap() = - vec![None, Some("rollback failed".to_string())]; - let prefs = UserPreferences { - active_asr_provider: "whisper".to_string(), - microphone_device_name: "External Mic".to_string(), - hotkey: previous.hotkey, - dictation_hotkey: previous.dictation_hotkey, - qa_hotkey: previous.qa_hotkey, - translation_hotkey: previous.translation_hotkey, - switch_style_hotkey: previous.switch_style_hotkey, - open_app_hotkey: previous.open_app_hotkey, - ..Default::default() - }; - - persist_settings(&writer, prefs.clone()).expect("settings remain consistent"); - - let saved = writer - .saved - .lock() - .unwrap() - .clone() - .expect("new settings saved"); - assert_eq!(saved.active_asr_provider, prefs.active_asr_provider); - assert_eq!(saved.microphone_device_name, prefs.microphone_device_name); - assert_eq!( - writer.active_asr_provider_syncs.lock().unwrap().clone(), - vec!["whisper".to_string(), previous.active_asr_provider.clone()] - ); - assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); - assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); - } - - #[test] - fn sync_dictation_hotkey_sets_modifier_trigger_and_clears_combo() { - let mut prefs = UserPreferences { - hotkey: HotkeyBinding { - trigger: HotkeyTrigger::Custom, - mode: HotkeyMode::Toggle, - keys: None, - }, - custom_combo_hotkey: Some(ComboBinding { - primary: "D".into(), - modifiers: vec!["cmd".into(), "shift".into()], - }), - dictation_hotkey: ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }, - ..Default::default() - }; - - super::sync_dictation_hotkey_legacy_fields(&mut prefs); - - assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::RightControl); - assert!(prefs.custom_combo_hotkey.is_none()); - } - - #[test] - fn sync_dictation_hotkey_sets_custom_trigger_and_combo_binding() { - let mut prefs = UserPreferences { - hotkey: HotkeyBinding { - trigger: HotkeyTrigger::RightControl, - mode: HotkeyMode::Toggle, - keys: None, - }, - dictation_hotkey: ShortcutBinding { - primary: "D".into(), - modifiers: vec!["cmd".into(), "shift".into()], - }, - ..Default::default() - }; - - super::sync_dictation_hotkey_legacy_fields(&mut prefs); - - assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::Custom); - let combo = prefs.custom_combo_hotkey.expect("combo binding saved"); - assert_eq!(combo.primary, "D"); - assert_eq!( - combo.modifiers, - vec!["cmd".to_string(), "shift".to_string()] - ); - } - - #[test] - fn sync_dictation_hotkey_clears_empty_custom_binding() { - let mut prefs = UserPreferences { - hotkey: HotkeyBinding { - trigger: HotkeyTrigger::RightControl, - mode: HotkeyMode::Toggle, - keys: None, - }, - custom_combo_hotkey: Some(ComboBinding { - primary: "D".into(), - modifiers: vec!["cmd".into(), "shift".into()], - }), - dictation_hotkey: ShortcutBinding { - primary: " ".into(), - modifiers: vec!["cmd".into()], - }, - ..Default::default() - }; - - super::sync_dictation_hotkey_legacy_fields(&mut prefs); - - assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::Custom); - assert!(prefs.custom_combo_hotkey.is_none()); - } - - #[test] - fn validate_combo_hotkey_rejects_bare_shift() { - let result = super::validate_combo_hotkey(ComboBinding { - primary: "Shift".into(), - modifiers: vec![], - }); - - assert!(result.is_err()); - } - - #[test] - fn combo_hotkey_bare_shift_rejection_matches_dictation_setter() { - let binding = ShortcutBinding { - primary: "Shift".into(), - modifiers: vec![], - }; - - assert_eq!( - super::reject_bare_shift_dictation_shortcut(&binding), - Err("Shift 单键目前只能用于翻译快捷键".into()) - ); - } - - #[test] - fn dictation_qa_overlap_rejects_same_modifier_only_binding() { - let binding = ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }; - - assert_eq!( - super::reject_dictation_qa_hotkey_overlap(&binding, &binding), - Err("QA 快捷键不能和听写快捷键相同".into()) - ); - } - - #[test] - fn dictation_qa_overlap_rejects_same_combo_binding() { - let dictation = ShortcutBinding { - primary: ";".into(), - modifiers: vec!["ctrl".into(), "shift".into()], - }; - let qa = ShortcutBinding { - primary: ";".into(), - modifiers: vec!["control".into(), "shift".into()], - }; - - assert_eq!( - super::reject_dictation_qa_hotkey_overlap(&dictation, &qa), - Err("QA 快捷键不能和听写快捷键相同".into()) - ); - } - - #[test] - fn dictation_qa_overlap_allows_distinct_bindings() { - let dictation = ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }; - let qa = ShortcutBinding { - primary: ";".into(), - modifiers: vec!["ctrl".into(), "shift".into()], - }; - - assert!(super::reject_dictation_qa_hotkey_overlap(&dictation, &qa).is_ok()); - } - - #[test] - fn dictation_translation_overlap_rejects_same_modifier_only_binding() { - let binding = ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }; - - assert_eq!( - super::reject_dictation_translation_hotkey_overlap(&binding, &binding), - Err("翻译快捷键不能和听写快捷键相同".into()) - ); - } - - #[test] - fn dictation_translation_overlap_rejects_same_combo_binding() { - let dictation = ShortcutBinding { - primary: "T".into(), - modifiers: vec!["ctrl".into(), "shift".into()], - }; - let translation = ShortcutBinding { - primary: "T".into(), - modifiers: vec!["control".into(), "shift".into()], - }; - - assert_eq!( - super::reject_dictation_translation_hotkey_overlap(&dictation, &translation), - Err("翻译快捷键不能和听写快捷键相同".into()) - ); - } - - #[test] - fn dictation_translation_overlap_allows_distinct_bindings() { - let dictation = ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }; - let translation = ShortcutBinding { - primary: "Shift".into(), - modifiers: vec![], - }; - - assert!( - super::reject_dictation_translation_hotkey_overlap(&dictation, &translation).is_ok() - ); - } - - #[test] - fn persist_settings_rejects_dictation_translation_overlap() { - let writer = FakeSettingsWriter::default(); - let binding = ShortcutBinding { - primary: "RightControl".into(), - modifiers: vec![], - }; - let prefs = UserPreferences { - dictation_hotkey: binding.clone(), - translation_hotkey: binding, - ..Default::default() - }; - - assert_eq!( - persist_settings(&writer, prefs), - Err("翻译快捷键不能和听写快捷键相同".into()) - ); - assert!(writer.saved.lock().unwrap().is_none()); - } - - #[test] - fn persist_settings_rejects_translation_switch_style_overlap() { - let writer = FakeSettingsWriter::default(); - let binding = ShortcutBinding { - primary: "T".into(), - modifiers: vec!["cmd".into(), "shift".into()], - }; - let prefs = UserPreferences { - translation_hotkey: binding.clone(), - switch_style_hotkey: Some(binding), - ..Default::default() - }; - - assert_eq!( - persist_settings(&writer, prefs), - Err("切换风格快捷键不能和翻译快捷键相同".into()) - ); - assert!(writer.saved.lock().unwrap().is_none()); - } - - #[test] - fn persist_settings_rejects_switch_style_open_app_overlap() { - let writer = FakeSettingsWriter::default(); - let binding = ShortcutBinding { - primary: "K".into(), - modifiers: vec!["cmd".into(), "shift".into()], - }; - let prefs = UserPreferences { - switch_style_hotkey: Some(binding.clone()), - open_app_hotkey: Some(binding), - ..Default::default() - }; - - assert_eq!( - persist_settings(&writer, prefs), - Err("打开应用快捷键不能和切换风格快捷键相同".into()) - ); - assert!(writer.saved.lock().unwrap().is_none()); - } - - #[test] - fn parse_latest_beta_from_atom_picks_first_beta_tagged_entry() { - // Fixture trimmed from real `releases.atom`：包含一条 stable + 一条 Beta。 - // 解析必须跳过 stable（tag 不以 -beta-tauri 结尾），返回 Beta。 - let body = r#" - - - tag:github.com,2008:Repository/X/v1.2.23-tauri - 2026-05-07T09:05:00Z - - OpenLess v1.2.23-tauri - - - tag:github.com,2008:Repository/X/v1.2.24-2-beta-tauri - 2026-05-08T01:27:23Z - - OpenLess v1.2.24-2-beta-tauri - -"#; - let got = parse_latest_beta_from_atom(body).expect("must find a Beta entry"); - assert_eq!(got.tag_name, "v1.2.24-2-beta-tauri"); - assert_eq!( - got.html_url, - "https://github.com/appergb/openless/releases/tag/v1.2.24-2-beta-tauri" - ); - assert_eq!(got.published_at, "2026-05-08T01:27:23Z"); - } - - #[test] - fn parse_latest_beta_from_atom_returns_none_when_only_stable_releases() { - let body = r#" - - - 2026-05-07T09:05:00Z - -"#; - assert!(parse_latest_beta_from_atom(body).is_none()); - } - - #[tokio::test] - async fn fetch_provider_models_omits_authorization_when_api_key_is_empty() { - let listener = TcpListener::bind("127.0.0.1:0").unwrap(); - let addr = listener.local_addr().unwrap(); - - let server = thread::spawn(move || { - let (mut stream, _) = listener.accept().unwrap(); - let mut buf = [0u8; 8192]; - let mut request = Vec::new(); - loop { - let n = stream.read(&mut buf).unwrap(); - if n == 0 { - break; - } - request.extend_from_slice(&buf[..n]); - if request.windows(4).any(|w| w == b"\r\n\r\n") { - break; - } - } - let request_text = String::from_utf8_lossy(&request); - assert!(!request_text.contains("Authorization: Bearer")); - - let body = r#"{"data":[{"id":"m1"},{"id":"m2"}]}"#; - let response = format!( - "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\nContent-Length: {}\r\nConnection: close\r\n\r\n{}", - body.len(), - body - ); - stream.write_all(response.as_bytes()).unwrap(); - }); - - let models = fetch_provider_models(&ProviderConfig { - base_url: format!("http://{}", addr), - api_key: String::new(), - }) - .await - .unwrap(); - - assert_eq!(models, vec!["m1".to_string(), "m2".to_string()]); - server.join().unwrap(); - } - - #[test] - fn is_valid_session_id_accepts_canonical_uuid_v4() { - // canonical UUID-v4 字面：8-4-4-4-12，全小写、全大写、混合都接受。 - assert!(is_valid_session_id("550e8400-e29b-41d4-a716-446655440000")); - assert!(is_valid_session_id("550E8400-E29B-41D4-A716-446655440000")); - assert!(is_valid_session_id("Abc12345-6789-abcd-EF01-234567890abc")); - } - - #[test] - fn is_valid_session_id_rejects_path_traversal_and_garbage() { - assert!(!is_valid_session_id("")); - assert!(!is_valid_session_id("../../etc/passwd")); - assert!(!is_valid_session_id("..\\..\\windows\\system32")); - // 长度对但含 `/`：dash 位置错或非 hex 字符都不通过 - assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544/000")); - assert!(!is_valid_session_id("550e8400_e29b_41d4_a716_446655440000")); // 用 _ 代 - - // 非 hex 字符 - assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544000g")); - // 长度不对（35 / 37） - assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544000")); - assert!(!is_valid_session_id( - "550e8400-e29b-41d4-a716-4466554400000" - )); - // NUL 字节 - assert!(!is_valid_session_id( - "550e8400-e29b-41d4-a716-44665544\x00000" - )); - // 百分号编码与绝对路径 - assert!(!is_valid_session_id("%2e%2e/recordings/x")); - assert!(!is_valid_session_id("/Users/attacker/secret.wav")); - } - - #[test] - fn is_valid_local_pack_id_accepts_realistic_ids() { - assert!(is_valid_local_pack_id("builtin.light")); - assert!(is_valid_local_pack_id("builtin.structured")); - assert!(is_valid_local_pack_id("custom.meeting")); - assert!(is_valid_local_pack_id( - "550e8400-e29b-41d4-a716-446655440000" - )); - assert!(is_valid_local_pack_id("my_pack_v2")); - assert!(is_valid_local_pack_id("Pack-2026.05")); - } - - #[test] - fn is_valid_local_pack_id_rejects_path_traversal() { - assert!(!is_valid_local_pack_id("")); - assert!(!is_valid_local_pack_id("../etc/passwd")); - assert!(!is_valid_local_pack_id("..\\windows\\system32")); - assert!(!is_valid_local_pack_id("pack/../../etc")); - assert!(!is_valid_local_pack_id("/abs/path")); - assert!(!is_valid_local_pack_id("with space")); - assert!(!is_valid_local_pack_id("with\x00null")); - assert!(!is_valid_local_pack_id(&"a".repeat(129))); - } -} diff --git a/openless-all/app/src-tauri/src/commands/credentials.rs b/openless-all/app/src-tauri/src/commands/credentials.rs new file mode 100644 index 00000000..350d1e3e --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/credentials.rs @@ -0,0 +1,227 @@ +use super::*; + +#[tauri::command] +pub fn get_credentials() -> CredentialsStatus { + let snap = CredentialsVault::snapshot(); + let active_asr_provider = CredentialsVault::get_active_asr(); + let active_llm_provider = CredentialsVault::get_active_llm(); + let volcengine_configured = volcengine_configured(&snap); + let asr_configured = asr_configured_for_provider(&active_asr_provider, &snap); + let llm_configured = llm_configured_for_provider(&active_llm_provider, &snap); + CredentialsStatus { + active_asr_provider, + active_llm_provider, + asr_configured, + llm_configured, + volcengine_configured, + ark_configured: llm_configured, + } +} + +fn volcengine_configured(snap: &CredentialsSnapshot) -> bool { + configured(&snap.volcengine_app_key) + && configured(&snap.volcengine_access_key) + && configured(&snap.volcengine_resource_id) +} + +pub(crate) fn asr_configured_for_provider(provider: &str, snap: &CredentialsSnapshot) -> bool { + if provider == "volcengine" { + return volcengine_configured(snap); + } + if provider == crate::asr::local::PROVIDER_ID + || active_foundry_asr_is_supported(provider) + || active_sherpa_asr_is_supported(provider) + { + // 本地 ASR 不依赖云端凭据。 + return true; + } + if provider == crate::asr::bailian::PROVIDER_ID { + return configured(&snap.asr_api_key); + } + if provider == crate::asr::mimo::PROVIDER_ID { + return configured(&snap.asr_api_key) + && configured(&snap.asr_endpoint) + && configured(&snap.asr_model); + } + configured(&snap.asr_endpoint) && configured(&snap.asr_model) +} + +pub(crate) fn llm_configured_for_provider(provider: &str, snap: &CredentialsSnapshot) -> bool { + if provider == CODEX_OAUTH_PROVIDER_ID { + return CodexOAuthCredentials::load_default().is_ok(); + } + let endpoint = snap.ark_endpoint.as_deref().unwrap_or_default(); + let endpoint_and_model = configured(&snap.ark_endpoint) && configured(&snap.ark_model_id); + if endpoint_and_model + && llm_provider_default_endpoint(provider) + .map(|default| same_llm_endpoint(endpoint, default)) + .unwrap_or(false) + { + return configured(&snap.ark_api_key); + } + endpoint_and_model +} + +fn llm_provider_default_endpoint(provider: &str) -> Option<&'static str> { + match provider { + "ark" => Some("https://ark.cn-beijing.volces.com/api/v3"), + "deepseek" => Some("https://api.deepseek.com/v1"), + "siliconflow" => Some("https://api.siliconflow.cn/v1"), + "openai" => Some("https://api.openai.com/v1"), + // 谷歌 Gemini 原生 API（v1beta）。后端 llm_gemini.rs 会拼成 + // `{baseUrl}/models/{model}:generateContent`，认证用 x-goog-api-key 头。 + "gemini" => Some("https://generativelanguage.googleapis.com/v1beta"), + "mimo" => Some("https://api.xiaomimimo.com/v1"), + "cometapi" => Some("https://api.cometapi.com/v1"), + "openrouterFree" => Some("https://openrouter.ai/api/v1"), + "alibabaCoding" => Some("https://coding-intl.dashscope.aliyuncs.com/v1"), + "codingPlanX" => Some("https://api.codingplanx.ai/v1"), + _ => None, + } +} + +fn same_llm_endpoint(a: &str, b: &str) -> bool { + fn normalize(value: &str) -> &str { + value + .trim() + .trim_end_matches('/') + .trim_end_matches("/chat/completions") + .trim_end_matches('/') + } + normalize(a).eq_ignore_ascii_case(normalize(b)) +} + +fn configured(field: &Option) -> bool { + field + .as_ref() + .map(|s| !s.trim().is_empty()) + .unwrap_or(false) +} + +#[derive(Debug, Clone, Copy, PartialEq, Eq)] +pub(crate) struct LocalAsrReleasePlan { + pub(crate) qwen: bool, + pub(crate) foundry: bool, + pub(crate) sherpa: bool, +} + +pub(crate) fn local_asr_release_plan_for_provider(provider: &str) -> LocalAsrReleasePlan { + LocalAsrReleasePlan { + qwen: provider != crate::asr::local::PROVIDER_ID, + foundry: provider != FOUNDRY_LOCAL_PROVIDER_ID, + sherpa: provider != crate::asr::local::sherpa::PROVIDER_ID, + } +} + +pub(crate) async fn release_foundry_runtime_if_inactive( + runtime: &Arc, + release_foundry: bool, +) { + if release_foundry { + runtime.request_cancel_prepare(); + if let Err(error) = runtime.release_now().await { + log::warn!("[foundry-asr] release inactive runtime failed: {error:#}"); + } + } +} + +pub(crate) async fn release_sherpa_runtime_if_inactive( + runtime: &Arc, + release_sherpa: bool, +) { + if release_sherpa { + runtime.request_cancel_prepare(); + if let Err(error) = runtime.release_now().await { + log::warn!("[sherpa-asr] release inactive runtime failed: {error:#}"); + } + } +} + +#[tauri::command] +pub fn set_credential(window: Window, account: String, value: String) -> Result<(), String> { + ensure_main_window(&window)?; + let acc = parse_account(&account)?; + if value.is_empty() { + CredentialsVault::remove(acc).map_err(|e| e.to_string())?; + } else { + CredentialsVault::set(acc, &value).map_err(|e| e.to_string())?; + } + // 通知前端凭据已变更（如 Overview 页需要刷新 asrConfigured 状态）。 + // issue #532 / #573：在 Settings 填写凭据但不切换提供商时，Overview 不会重拉状态， + // 仍显示「未配置」。该修复曾随 #538 合入 main，但被 beta→main 合并覆盖，beta 上缺失。 + let _ = window.emit("credentials:changed", ()); + Ok(()) +} + +#[tauri::command] +pub async fn set_active_asr_provider( + coord: CoordinatorState<'_>, + runtime: State<'_, Arc>, + sherpa_runtime: State<'_, Arc>, + provider: String, +) -> Result<(), String> { + if provider == FOUNDRY_LOCAL_PROVIDER_ID && !active_foundry_asr_is_supported(&provider) { + return Err("Foundry Local Whisper is only available on Windows".to_string()); + } + if provider == crate::asr::local::sherpa::PROVIDER_ID + && !active_sherpa_asr_is_supported(&provider) + { + return Err("sherpa-onnx local ASR is only available on Windows".to_string()); + } + if CredentialsVault::get_active_asr() == provider { + return Ok(()); + } + CredentialsVault::set_active_asr_provider(&provider).map_err(|e| e.to_string())?; + let release_plan = local_asr_release_plan_for_provider(&provider); + if provider == crate::asr::local::PROVIDER_ID { + // 切到本地 ASR → 后台预加载模型，下次按 hotkey 时不必等数秒。 + coord.preload_local_asr_in_background(); + } + if release_plan.qwen { + // 切回云端 → 用户已不需要本地引擎，立刻释放 1.2GB+ RAM；不释放的话只会等到 + // schedule_local_asr_release 的下一次 dictation 才触发，而切回云端后根本不会 + // 再走 local 路径，引擎会驻留到进程退出。 + coord.release_local_asr_engine(); + } + release_foundry_runtime_if_inactive(runtime.inner(), release_plan.foundry).await; + release_sherpa_runtime_if_inactive(sherpa_runtime.inner(), release_plan.sherpa).await; + Ok(()) +} + +#[tauri::command] +pub fn set_active_llm_provider(provider: String) -> Result<(), String> { + CredentialsVault::set_active_llm_provider(&provider).map_err(|e| e.to_string()) +} + +/// 读出某个账号的实际值（用于设置页预填表单）。 +/// 凭据来自系统凭据库；只允许主设置窗口读取 raw secret，避免胶囊 / QA 等辅助窗口默认暴露。 +#[tauri::command] +pub fn read_credential(window: Window, account: String) -> Result, String> { + ensure_main_window(&window)?; + let acc = parse_account(&account)?; + CredentialsVault::get(acc).map_err(|e| e.to_string()) +} + +fn ensure_main_window(window: &Window) -> Result<(), String> { + if window.label() == "main" { + Ok(()) + } else { + Err("credential access is only allowed from the main window".to_string()) + } +} + +fn parse_account(s: &str) -> Result { + match s { + "volcengine.app_key" => Ok(CredentialAccount::VolcengineAppKey), + "volcengine.access_key" => Ok(CredentialAccount::VolcengineAccessKey), + "volcengine.resource_id" => Ok(CredentialAccount::VolcengineResourceId), + "ark.api_key" => Ok(CredentialAccount::ArkApiKey), + "ark.model_id" => Ok(CredentialAccount::ArkModelId), + "ark.endpoint" => Ok(CredentialAccount::ArkEndpoint), + "asr.api_key" => Ok(CredentialAccount::AsrApiKey), + "asr.endpoint" => Ok(CredentialAccount::AsrEndpoint), + "asr.model" => Ok(CredentialAccount::AsrModel), + "asr.vocabulary_id" => Ok(CredentialAccount::AsrVocabularyId), + _ => Err(format!("unknown account: {s}")), + } +} diff --git a/openless-all/app/src-tauri/src/commands/dictation.rs b/openless-all/app/src-tauri/src/commands/dictation.rs new file mode 100644 index 00000000..db36b799 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/dictation.rs @@ -0,0 +1,49 @@ +use super::*; + +#[tauri::command] +pub async fn start_dictation(coord: CoordinatorState<'_>) -> Result<(), String> { + coord.start_dictation().await +} + +#[tauri::command] +pub async fn stop_dictation(coord: CoordinatorState<'_>) -> Result<(), String> { + coord.stop_dictation().await +} + +#[tauri::command] +pub fn cancel_dictation(coord: CoordinatorState<'_>) { + coord.cancel_dictation(); +} + +#[tauri::command] +pub async fn handle_window_hotkey_event( + coord: CoordinatorState<'_>, + event_type: String, + key: String, + code: String, + repeat: bool, +) -> Result<(), String> { + coord + .handle_window_hotkey_event(event_type, key, code, repeat) + .await +} + +#[cfg(debug_assertions)] +#[tauri::command] +pub async fn inject_hotkey_click_for_dev(coord: CoordinatorState<'_>) -> Result<(), String> { + coord.inject_hotkey_click_for_dev().await +} + +#[tauri::command] +pub async fn repolish( + coord: CoordinatorState<'_>, + raw_text: String, + mode: PolishMode, +) -> Result { + log::info!( + "[style-pack] command repolish requested legacy_mode={:?} raw_chars={}", + mode, + raw_text.chars().count() + ); + coord.repolish(raw_text, mode).await +} diff --git a/openless-all/app/src-tauri/src/commands/dictionary.rs b/openless-all/app/src-tauri/src/commands/dictionary.rs new file mode 100644 index 00000000..e9e80202 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/dictionary.rs @@ -0,0 +1,79 @@ +use super::*; + +#[tauri::command] +pub fn list_vocab(coord: CoordinatorState<'_>) -> Result, String> { + coord.vocab().list().map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn add_vocab( + coord: CoordinatorState<'_>, + phrase: String, + note: Option, +) -> Result { + coord.vocab().add(phrase, note).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn remove_vocab(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { + coord.vocab().remove(&id).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn set_vocab_enabled( + coord: CoordinatorState<'_>, + id: String, + enabled: bool, +) -> Result<(), String> { + coord + .vocab() + .set_enabled(&id, enabled) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn list_correction_rules(coord: CoordinatorState<'_>) -> Result, String> { + coord.correction_rules().list().map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn add_correction_rule( + coord: CoordinatorState<'_>, + pattern: String, + replacement: String, +) -> Result { + coord + .correction_rules() + .add(pattern, replacement) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn remove_correction_rule(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { + coord + .correction_rules() + .remove(&id) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn set_correction_rule_enabled( + coord: CoordinatorState<'_>, + id: String, + enabled: bool, +) -> Result<(), String> { + coord + .correction_rules() + .set_enabled(&id, enabled) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn list_vocab_presets() -> Result { + crate::persistence::list_vocab_presets().map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn save_vocab_presets(store: VocabPresetStore) -> Result<(), String> { + crate::persistence::save_vocab_presets(&store).map_err(|e| e.to_string()) +} diff --git a/openless-all/app/src-tauri/src/commands/foundry_asr.rs b/openless-all/app/src-tauri/src/commands/foundry_asr.rs new file mode 100644 index 00000000..f222ae0c --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/foundry_asr.rs @@ -0,0 +1,190 @@ +use super::*; + +pub(crate) fn active_foundry_model_from_prefs(prefs: &UserPreferences) -> String { + if model_alias_is_known(&prefs.foundry_local_asr_model) { + prefs.foundry_local_asr_model.clone() + } else { + DEFAULT_MODEL_ALIAS.to_string() + } +} + +pub(crate) fn validate_foundry_model_alias(model_alias: &str) -> Result<(), String> { + if model_alias_is_known(model_alias) { + Ok(()) + } else { + Err(format!( + "unknown Foundry Whisper model alias: {model_alias}" + )) + } +} + +pub(crate) fn normalize_foundry_language_hint(language_hint: &str) -> Result { + let normalized = language_hint.trim().to_string(); + if normalized.is_empty() + || (normalized.len() == 2 && normalized.bytes().all(|b| b.is_ascii_lowercase())) + { + Ok(normalized) + } else { + Err("language hint must be empty or ISO 639-1 lowercase code".to_string()) + } +} + +fn normalize_foundry_runtime_source(source: &str) -> String { + crate::asr::local::foundry_native::normalize_runtime_source_str(source) +} + +#[tauri::command] +pub async fn foundry_local_asr_status( + coord: CoordinatorState<'_>, + runtime: State<'_, Arc>, +) -> Result { + let prefs = coord.prefs().get(); + let active_model = active_foundry_model_from_prefs(&prefs); + Ok(runtime + .status_snapshot(&active_model, &prefs.foundry_local_runtime_source) + .await) +} + +#[tauri::command] +pub async fn foundry_local_asr_catalog( + runtime: State<'_, Arc>, +) -> Result, String> { + runtime + .catalog_snapshot() + .await + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn foundry_local_asr_set_model( + coord: CoordinatorState<'_>, + model_alias: String, +) -> Result<(), String> { + validate_foundry_model_alias(&model_alias)?; + let mut prefs = coord.prefs().get(); + if prefs.foundry_local_asr_model == model_alias { + return Ok(()); + } + prefs.foundry_local_asr_model = model_alias; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn foundry_local_asr_set_language_hint( + coord: CoordinatorState<'_>, + language_hint: String, +) -> Result<(), String> { + let normalized = normalize_foundry_language_hint(&language_hint)?; + let mut prefs = coord.prefs().get(); + if prefs.foundry_local_asr_language_hint == normalized { + return Ok(()); + } + prefs.foundry_local_asr_language_hint = normalized; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn foundry_local_asr_set_runtime_source( + coord: CoordinatorState<'_>, + source: String, +) -> Result<(), String> { + let mut prefs = coord.prefs().get(); + let normalized = normalize_foundry_runtime_source(&source); + if prefs.foundry_local_runtime_source == normalized { + return Ok(()); + } + prefs.foundry_local_runtime_source = normalized; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub async fn foundry_local_asr_prepare( + app: AppHandle, + coord: CoordinatorState<'_>, + runtime: State<'_, Arc>, + model_alias: String, +) -> Result { + validate_foundry_model_alias(&model_alias)?; + let prefs = coord.prefs().get(); + let runtime_source = prefs.foundry_local_runtime_source.clone(); + let progress_app = app.clone(); + let result = runtime + .ensure_loaded_with_progress(&model_alias, &runtime_source, move |payload| { + emit_foundry_prepare_progress(&progress_app, payload); + }) + .await; + match result { + Ok(model_id) => Ok(model_id), + Err(error) => { + let message = format!("{error:#}"); + emit_foundry_prepare_progress( + &app, + FoundryPrepareProgressPayload::failed( + model_alias, + "Foundry Local Whisper prepare failed", + message.clone(), + ), + ); + Err(message) + } + } +} + +#[tauri::command] +pub fn foundry_local_asr_cancel_prepare( + runtime: State<'_, Arc>, +) -> Result<(), String> { + runtime.request_cancel_prepare(); + Ok(()) +} + +#[tauri::command] +pub async fn foundry_local_asr_release( + runtime: State<'_, Arc>, +) -> Result<(), String> { + runtime.release_now().await.map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub async fn foundry_local_asr_model_dir( + runtime: State<'_, Arc>, + model_alias: String, +) -> Result { + validate_foundry_model_alias(&model_alias)?; + runtime + .model_dir_for_alias(&model_alias) + .await + .map(|path| path.display().to_string()) + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub async fn foundry_local_asr_delete_model( + runtime: State<'_, Arc>, + model_alias: String, +) -> Result<(), String> { + validate_foundry_model_alias(&model_alias)?; + runtime + .delete_model(&model_alias) + .await + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub async fn foundry_local_asr_reveal_model_dir( + runtime: State<'_, Arc>, + model_alias: String, +) -> Result<(), String> { + validate_foundry_model_alias(&model_alias)?; + let dir = runtime + .model_dir_for_alias(&model_alias) + .await + .map_err(|e| format!("{e:#}"))?; + open_path_in_file_manager(&dir) +} + +fn emit_foundry_prepare_progress(app: &AppHandle, payload: FoundryPrepareProgressPayload) { + if let Err(error) = app.emit("foundry-local-asr-prepare-progress", payload) { + log::warn!("[foundry-asr] emit prepare progress failed: {error}"); + } +} diff --git a/openless-all/app/src-tauri/src/commands/github_oauth.rs b/openless-all/app/src-tauri/src/commands/github_oauth.rs new file mode 100644 index 00000000..7e5af6dc --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/github_oauth.rs @@ -0,0 +1,154 @@ +use super::*; + +// ─────────────────────── GitHub OAuth Device Flow (Phase 1) ─────────────────────── +// +// 客户端直连 GitHub 拿 access_token + login，前端自动把 login 写进 +// prefs.marketplaceDevLogin。marketplace backend 完全不动（依然 X-Dev-User）。 +// Phase 2 才会让 backend 验证 GitHub identity（JWT 签发 + 防伪造）。 +// +// 配置 client_id 的两种方式（OAuth App client_id 非敏感，可硬编码）： +// 1. 在下方 GITHUB_OAUTH_CLIENT_ID 常量填值（生产推荐 — 直接 bake 进二进制） +// 2. 启动前设置环境变量 GITHUB_OAUTH_CLIENT_ID=（dev 方便） +// +// 注册 OAuth App： +// https://github.com/settings/applications/new +// - Application name: OpenLess (or your fork) +// - Homepage URL: https://openless.top (or任意) +// - Authorization callback URL: https://openless.top (Device Flow 不真用，但表单要求填) +// - 创建后在 General 页面勾选 "Enable Device Flow" +// - 抄 client_id 填到本常量 + +const GITHUB_OAUTH_CLIENT_ID: &str = "Ov23liyv3nEucG7oMHNE"; + +fn get_github_oauth_client_id() -> Result { + if let Ok(env_id) = std::env::var("GITHUB_OAUTH_CLIENT_ID") { + let trimmed = env_id.trim(); + if !trimmed.is_empty() { + return Ok(trimmed.to_string()); + } + } + if !GITHUB_OAUTH_CLIENT_ID.is_empty() { + return Ok(GITHUB_OAUTH_CLIENT_ID.to_string()); + } + Err( + "GitHub OAuth 未配置。请去 https://github.com/settings/applications/new 注册一个 OAuth App\ + （必须勾 Enable Device Flow），把 client_id 填到 \ + openless-all/app/src-tauri/src/commands.rs 的 GITHUB_OAUTH_CLIENT_ID 常量，\ + 或在启动前设置环境变量 GITHUB_OAUTH_CLIENT_ID=。" + .to_string(), + ) +} + +#[derive(Debug, serde::Serialize)] +#[serde(rename_all = "camelCase")] +pub struct GithubDeviceStartResponse { + pub device_code: String, + pub user_code: String, + pub verification_uri: String, + pub interval: u32, + pub expires_in: u32, +} + +#[tauri::command] +pub async fn github_device_flow_start() -> Result { + let client_id = get_github_oauth_client_id()?; + let resp = net::send_with_retry(|| { + net::http() + .post("https://github.com/login/device/code") + .header("Accept", "application/json") + .header("User-Agent", "OpenLess") + .timeout(std::time::Duration::from_secs(15)) + .form(&[("client_id", client_id.as_str()), ("scope", "read:user")]) + }) + .await + .map_err(|e| format!("调用 GitHub /login/device/code 失败：{e}"))?; + let status = resp.status(); + let body: serde_json::Value = resp + .json() + .await + .map_err(|e| format!("解析 device/code 响应失败：{e}"))?; + if !status.is_success() { + let err = body["error"].as_str().unwrap_or("unknown_error"); + let desc = body["error_description"].as_str().unwrap_or(""); + return Err(format!("GitHub device/code {status} {err}: {desc}")); + } + Ok(GithubDeviceStartResponse { + device_code: body["device_code"].as_str().unwrap_or("").to_string(), + user_code: body["user_code"].as_str().unwrap_or("").to_string(), + verification_uri: body["verification_uri"] + .as_str() + .unwrap_or("https://github.com/login/device") + .to_string(), + interval: body["interval"].as_u64().unwrap_or(5) as u32, + expires_in: body["expires_in"].as_u64().unwrap_or(900) as u32, + }) +} + +#[derive(Debug, serde::Serialize)] +#[serde(rename_all = "camelCase", tag = "kind")] +pub enum GithubDevicePollResult { + Authorized { login: String }, + Pending, + SlowDown, + Error { message: String }, +} + +#[tauri::command] +pub async fn github_device_flow_poll( + device_code: String, +) -> Result { + let client_id = get_github_oauth_client_id()?; + let token_resp = net::send_with_retry(|| { + net::http() + .post("https://github.com/login/oauth/access_token") + .header("Accept", "application/json") + .header("User-Agent", "OpenLess") + .timeout(std::time::Duration::from_secs(15)) + .form(&[ + ("client_id", client_id.as_str()), + ("device_code", device_code.as_str()), + ("grant_type", "urn:ietf:params:oauth:grant-type:device_code"), + ]) + }) + .await + .map_err(|e| format!("调用 GitHub /login/oauth/access_token 失败：{e}"))?; + let body: serde_json::Value = token_resp + .json() + .await + .map_err(|e| format!("解析 access_token 响应失败：{e}"))?; + + if let Some(token) = body["access_token"].as_str() { + let user_resp = net::send_with_retry(|| { + net::http() + .get("https://api.github.com/user") + .header("User-Agent", "OpenLess") + .header("Accept", "application/vnd.github+json") + .timeout(std::time::Duration::from_secs(15)) + .bearer_auth(token) + }) + .await + .map_err(|e| format!("调用 GitHub /user 失败：{e}"))?; + let user_body: serde_json::Value = user_resp + .json() + .await + .map_err(|e| format!("解析 /user 响应失败：{e}"))?; + let login = user_body["login"].as_str().unwrap_or("").to_string(); + if login.is_empty() { + return Ok(GithubDevicePollResult::Error { + message: "GitHub /user 返回空 login".to_string(), + }); + } + return Ok(GithubDevicePollResult::Authorized { login }); + } + + let err = body["error"].as_str().unwrap_or(""); + let msg = match err { + "authorization_pending" => return Ok(GithubDevicePollResult::Pending), + "slow_down" => return Ok(GithubDevicePollResult::SlowDown), + "expired_token" => "OAuth 设备码已过期，请重新发起登录".to_string(), + "access_denied" => "你在 GitHub 上拒绝了授权".to_string(), + other if !other.is_empty() => format!("OAuth 错误：{other}"), + _ => "未知 OAuth 错误（access_token 缺失）".to_string(), + }; + Ok(GithubDevicePollResult::Error { message: msg }) +} diff --git a/openless-all/app/src-tauri/src/commands/history.rs b/openless-all/app/src-tauri/src/commands/history.rs new file mode 100644 index 00000000..6f43042b --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/history.rs @@ -0,0 +1,50 @@ +use super::*; + +#[tauri::command] +pub fn list_history(coord: CoordinatorState<'_>) -> Result, String> { + coord.history().list().map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn delete_history_entry(coord: CoordinatorState<'_>, id: String) -> Result<(), String> { + coord.history().delete(&id).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn clear_history(coord: CoordinatorState<'_>) -> Result<(), String> { + coord.history().clear().map_err(|e| e.to_string()) +} + +/// 读取某次会话的原始麦克风 wav 字节流。仅当用户开过 +/// `prefs.record_audio_for_debug` 并且这条 session 是开关打开后录的，才会有文件。 +/// 文件名规约：`/recordings/.wav`，与 DictationSession.id 同名。 +/// +/// 路径校验：session_id **必须**严格匹配 UUID-v4 字面（36 字符 = 8-4-4-4-12 + 4 个 `-`， +/// 内容仅 ASCII 十六进制 + `-`）。白名单胜过黑名单——绝对路径前缀、Windows ADS、 +/// 百分号编码、NUL 字节都不在合法字符集里，挡掉所有 Path::join 越界的可能。 +/// session_id 在仓库内由 `Uuid::new_v4()` 生成 (`dictation.rs:1531`)，前端只会回传 +/// 自己列出的合法 id，但 IPC = boundary，按 boundary 规则严格校验。 +/// +/// async fs：单条 5 分钟 wav 约 9.6MB，同步 `std::fs::read` 会阻塞 Tauri IPC 主循环。 +/// 改 `tokio::fs::read` 后让出线程给其它 IPC。 +#[tauri::command] +pub async fn read_audio_recording(session_id: String) -> Result, String> { + if !is_valid_session_id(&session_id) { + return Err("invalid session id".into()); + } + let path = + crate::persistence::recording_path_for_session(&session_id).map_err(|e| e.to_string())?; + if !path.exists() { + return Err("recording not found".into()); + } + // TOCTOU 兜底：exists() 通过到 read 之间文件可能被 prune（条数 cap / retention + // 清理 / 用户手动删）。把 NotFound 标准化成跟 exists() 失败同样的错误字符串， + // 前端单条 'recording not found' catch 就能稳定隐藏按钮，不依赖本地化 OS 错误。 + tokio::fs::read(&path).await.map_err(|e| { + if e.kind() == std::io::ErrorKind::NotFound { + "recording not found".into() + } else { + format!("read wav failed: {e}") + } + }) +} diff --git a/openless-all/app/src-tauri/src/commands/hotkeys.rs b/openless-all/app/src-tauri/src/commands/hotkeys.rs new file mode 100644 index 00000000..2e4f712f --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/hotkeys.rs @@ -0,0 +1,408 @@ +use super::*; + +#[tauri::command] +pub fn validate_shortcut_binding(binding: ShortcutBinding) -> Result<(), String> { + crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn set_dictation_hotkey( + coord: CoordinatorState<'_>, + binding: ShortcutBinding, +) -> Result<(), String> { + crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string())?; + reject_bare_shift_dictation_shortcut(&binding)?; + let mut prefs = coord.prefs().get(); + if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { + reject_dictation_qa_hotkey_overlap(&binding, qa_hotkey)?; + } + reject_dictation_translation_hotkey_overlap(&binding, &prefs.translation_hotkey)?; + if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { + reject_dictation_switch_style_hotkey_overlap(&binding, switch_style)?; + } + if let Some(open_app) = prefs.open_app_hotkey.as_ref() { + reject_dictation_open_app_hotkey_overlap(&binding, open_app)?; + } + prefs.dictation_hotkey = binding; + sync_dictation_hotkey_legacy_fields(&mut prefs); + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + coord.update_hotkey_binding(); + coord.update_combo_hotkey_binding(); + Ok(()) +} + +#[tauri::command] +pub fn set_translation_hotkey( + coord: CoordinatorState<'_>, + binding: ShortcutBinding, +) -> Result<(), String> { + crate::shortcut_binding::validate_binding(&binding).map_err(|e| e.to_string())?; + let previous = coord.prefs().get(); + reject_dictation_translation_hotkey_overlap(&previous.dictation_hotkey, &binding)?; + if let Some(qa_hotkey) = previous.qa_hotkey.as_ref() { + reject_qa_translation_hotkey_overlap(qa_hotkey, &binding)?; + } + if let Some(switch_style) = previous.switch_style_hotkey.as_ref() { + reject_translation_switch_style_hotkey_overlap(&binding, switch_style)?; + } + if let Some(open_app) = previous.open_app_hotkey.as_ref() { + reject_translation_open_app_hotkey_overlap(&binding, open_app)?; + } + let mut prefs = previous.clone(); + prefs.translation_hotkey = binding; + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + if let Err(e) = coord.try_update_translation_hotkey_binding() { + if let Err(rollback_err) = coord.prefs().set(previous) { + log::warn!("[commands] 回滚翻译快捷键失败: {rollback_err}"); + } + coord.update_translation_hotkey_binding(); + return Err(e); + } + Ok(()) +} + +/// 设置「切换风格」全局快捷键。`binding == None`（前端传 null）= 停用：清空绑定并 +/// 反注册全局键。镜像 `set_qa_hotkey` 的 `Option=None` 停用模式（issue #576）。 +#[tauri::command] +pub fn set_switch_style_hotkey( + coord: CoordinatorState<'_>, + binding: Option, +) -> Result<(), String> { + if let Some(binding) = binding.as_ref() { + crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; + reject_modifier_only_action_shortcut(binding)?; + } + let mut prefs = coord.prefs().get(); + if let Some(binding) = binding.as_ref() { + reject_dictation_switch_style_hotkey_overlap(&prefs.dictation_hotkey, binding)?; + reject_translation_switch_style_hotkey_overlap(&prefs.translation_hotkey, binding)?; + if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { + reject_qa_switch_style_hotkey_overlap(qa_hotkey, binding)?; + } + if let Some(open_app) = prefs.open_app_hotkey.as_ref() { + reject_switch_style_open_app_hotkey_overlap(binding, open_app)?; + } + } + prefs.switch_style_hotkey = binding; + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + coord.update_switch_style_hotkey_binding(); + Ok(()) +} + +/// 设置「唤起 App」全局快捷键。`binding == None`（前端传 null）= 停用（同上）。 +#[tauri::command] +pub fn set_open_app_hotkey( + coord: CoordinatorState<'_>, + binding: Option, +) -> Result<(), String> { + if let Some(binding) = binding.as_ref() { + crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; + reject_modifier_only_action_shortcut(binding)?; + } + let mut prefs = coord.prefs().get(); + if let Some(binding) = binding.as_ref() { + reject_dictation_open_app_hotkey_overlap(&prefs.dictation_hotkey, binding)?; + reject_translation_open_app_hotkey_overlap(&prefs.translation_hotkey, binding)?; + if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { + reject_qa_open_app_hotkey_overlap(qa_hotkey, binding)?; + } + if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { + reject_switch_style_open_app_hotkey_overlap(switch_style, binding)?; + } + } + prefs.open_app_hotkey = binding; + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + coord.update_open_app_hotkey_binding(); + Ok(()) +} + +fn reject_modifier_only_action_shortcut(binding: &ShortcutBinding) -> Result<(), String> { + if binding.modifiers.is_empty() + && (binding.primary.eq_ignore_ascii_case("shift") + || crate::shortcut_binding::legacy_modifier_trigger(binding).is_some()) + { + return Err("该快捷键需要使用组合键或非修饰主键".into()); + } + Ok(()) +} + +#[tauri::command] +pub fn validate_combo_hotkey(binding: ComboBinding) -> Result<(), String> { + let shortcut = ShortcutBinding { + primary: binding.primary, + modifiers: binding.modifiers, + }; + reject_bare_shift_dictation_shortcut(&shortcut)?; + crate::combo_hotkey::validate_binding(&shortcut).map_err(|e| e.to_string()) +} + +/// 设置自定义录音组合键并热更新 monitor。 +#[tauri::command] +pub fn set_combo_hotkey(coord: CoordinatorState<'_>, binding: ComboBinding) -> Result<(), String> { + let mut prefs = coord.prefs().get(); + let shortcut = ShortcutBinding { + primary: binding.primary.clone(), + modifiers: binding.modifiers.clone(), + }; + reject_bare_shift_dictation_shortcut(&shortcut)?; + crate::combo_hotkey::validate_binding(&shortcut).map_err(|e| e.to_string())?; + if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { + reject_dictation_qa_hotkey_overlap(&shortcut, qa_hotkey)?; + } + reject_dictation_translation_hotkey_overlap(&shortcut, &prefs.translation_hotkey)?; + if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { + reject_dictation_switch_style_hotkey_overlap(&shortcut, switch_style)?; + } + if let Some(open_app) = prefs.open_app_hotkey.as_ref() { + reject_dictation_open_app_hotkey_overlap(&shortcut, open_app)?; + } + prefs.custom_combo_hotkey = Some(binding); + prefs.dictation_hotkey = shortcut; + sync_dictation_hotkey_legacy_fields(&mut prefs); + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + coord.update_hotkey_binding(); + coord.update_combo_hotkey_binding(); + Ok(()) +} + +pub(crate) fn reject_bare_shift_dictation_shortcut( + binding: &ShortcutBinding, +) -> Result<(), String> { + if binding.modifiers.is_empty() && binding.primary.eq_ignore_ascii_case("shift") { + return Err("Shift 单键目前只能用于翻译快捷键".into()); + } + Ok(()) +} + +pub(crate) fn sync_dictation_hotkey_legacy_fields(prefs: &mut UserPreferences) { + if let Some(trigger) = crate::shortcut_binding::legacy_modifier_trigger(&prefs.dictation_hotkey) + { + prefs.hotkey.trigger = trigger; + prefs.custom_combo_hotkey = None; + return; + } + prefs.hotkey.trigger = crate::types::HotkeyTrigger::Custom; + prefs.custom_combo_hotkey = if prefs.dictation_hotkey.primary.trim().is_empty() { + None + } else { + Some(ComboBinding { + primary: prefs.dictation_hotkey.primary.clone(), + modifiers: prefs.dictation_hotkey.modifiers.clone(), + }) + }; +} + +pub(crate) fn reject_dictation_qa_hotkey_overlap( + dictation: &ShortcutBinding, + qa: &ShortcutBinding, +) -> Result<(), String> { + if shortcut_bindings_overlap(dictation, qa) { + return Err("QA 快捷键不能和听写快捷键相同".into()); + } + Ok(()) +} + +fn reject_hotkey_overlap( + left: &ShortcutBinding, + right: &ShortcutBinding, + message: &'static str, +) -> Result<(), String> { + if shortcut_bindings_overlap(left, right) { + return Err(message.into()); + } + Ok(()) +} + +pub(crate) fn reject_hotkey_collisions(prefs: &UserPreferences) -> Result<(), String> { + // 停用（None）的 action 快捷键不参与任何冲突检测。 + let switch_style = prefs.switch_style_hotkey.as_ref(); + let open_app = prefs.open_app_hotkey.as_ref(); + let less_computer = prefs.coding_agent_voice_hotkey.as_ref(); + if let Some(qa_hotkey) = prefs.qa_hotkey.as_ref() { + reject_dictation_qa_hotkey_overlap(&prefs.dictation_hotkey, qa_hotkey)?; + reject_qa_translation_hotkey_overlap(qa_hotkey, &prefs.translation_hotkey)?; + if let Some(less_computer) = less_computer { + reject_qa_less_computer_hotkey_overlap(qa_hotkey, less_computer)?; + } + if let Some(switch_style) = switch_style { + reject_qa_switch_style_hotkey_overlap(qa_hotkey, switch_style)?; + } + if let Some(open_app) = open_app { + reject_qa_open_app_hotkey_overlap(qa_hotkey, open_app)?; + } + } + reject_dictation_translation_hotkey_overlap( + &prefs.dictation_hotkey, + &prefs.translation_hotkey, + )?; + if let Some(less_computer) = less_computer { + reject_dictation_less_computer_hotkey_overlap(&prefs.dictation_hotkey, less_computer)?; + reject_translation_less_computer_hotkey_overlap(&prefs.translation_hotkey, less_computer)?; + } + if let Some(switch_style) = switch_style { + reject_dictation_switch_style_hotkey_overlap(&prefs.dictation_hotkey, switch_style)?; + reject_translation_switch_style_hotkey_overlap(&prefs.translation_hotkey, switch_style)?; + if let Some(less_computer) = less_computer { + reject_less_computer_switch_style_hotkey_overlap(less_computer, switch_style)?; + } + } + if let Some(open_app) = open_app { + reject_dictation_open_app_hotkey_overlap(&prefs.dictation_hotkey, open_app)?; + reject_translation_open_app_hotkey_overlap(&prefs.translation_hotkey, open_app)?; + if let Some(less_computer) = less_computer { + reject_less_computer_open_app_hotkey_overlap(less_computer, open_app)?; + } + } + if let (Some(switch_style), Some(open_app)) = (switch_style, open_app) { + reject_switch_style_open_app_hotkey_overlap(switch_style, open_app)?; + } + Ok(()) +} + +pub(crate) fn reject_dictation_translation_hotkey_overlap( + dictation: &ShortcutBinding, + translation: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(dictation, translation, "翻译快捷键不能和听写快捷键相同") +} + +fn reject_dictation_switch_style_hotkey_overlap( + dictation: &ShortcutBinding, + switch_style: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + dictation, + switch_style, + "切换风格快捷键不能和听写快捷键相同", + ) +} + +fn reject_dictation_open_app_hotkey_overlap( + dictation: &ShortcutBinding, + open_app: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(dictation, open_app, "打开应用快捷键不能和听写快捷键相同") +} + +fn reject_dictation_less_computer_hotkey_overlap( + dictation: &ShortcutBinding, + less_computer: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + dictation, + less_computer, + "Less Computer 快捷键不能和听写快捷键相同", + ) +} + +pub(crate) fn reject_qa_translation_hotkey_overlap( + qa: &ShortcutBinding, + translation: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(qa, translation, "翻译快捷键不能和 QA 快捷键相同") +} + +pub(crate) fn reject_qa_switch_style_hotkey_overlap( + qa: &ShortcutBinding, + switch_style: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(qa, switch_style, "切换风格快捷键不能和 QA 快捷键相同") +} + +pub(crate) fn reject_qa_open_app_hotkey_overlap( + qa: &ShortcutBinding, + open_app: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(qa, open_app, "打开应用快捷键不能和 QA 快捷键相同") +} + +fn reject_qa_less_computer_hotkey_overlap( + qa: &ShortcutBinding, + less_computer: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + qa, + less_computer, + "Less Computer 快捷键不能和 QA 快捷键相同", + ) +} + +fn reject_translation_switch_style_hotkey_overlap( + translation: &ShortcutBinding, + switch_style: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + translation, + switch_style, + "切换风格快捷键不能和翻译快捷键相同", + ) +} + +fn reject_translation_open_app_hotkey_overlap( + translation: &ShortcutBinding, + open_app: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap(translation, open_app, "打开应用快捷键不能和翻译快捷键相同") +} + +fn reject_translation_less_computer_hotkey_overlap( + translation: &ShortcutBinding, + less_computer: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + translation, + less_computer, + "Less Computer 快捷键不能和翻译快捷键相同", + ) +} + +fn reject_switch_style_open_app_hotkey_overlap( + switch_style: &ShortcutBinding, + open_app: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + switch_style, + open_app, + "打开应用快捷键不能和切换风格快捷键相同", + ) +} + +fn reject_less_computer_switch_style_hotkey_overlap( + less_computer: &ShortcutBinding, + switch_style: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + less_computer, + switch_style, + "Less Computer 快捷键不能和切换风格快捷键相同", + ) +} + +fn reject_less_computer_open_app_hotkey_overlap( + less_computer: &ShortcutBinding, + open_app: &ShortcutBinding, +) -> Result<(), String> { + reject_hotkey_overlap( + less_computer, + open_app, + "Less Computer 快捷键不能和打开应用快捷键相同", + ) +} + +fn shortcut_bindings_overlap(left: &ShortcutBinding, right: &ShortcutBinding) -> bool { + let left_legacy = crate::shortcut_binding::legacy_modifier_trigger(left); + let right_legacy = crate::shortcut_binding::legacy_modifier_trigger(right); + match (left_legacy, right_legacy) { + (Some(left), Some(right)) => left == right, + (Some(_), None) | (None, Some(_)) => false, + (None, None) => { + let Ok(left) = crate::shortcut_binding::parse_global_hotkey(left) else { + return false; + }; + let Ok(right) = crate::shortcut_binding::parse_global_hotkey(right) else { + return false; + }; + left == right + } + } +} diff --git a/openless-all/app/src-tauri/src/commands/local_asr.rs b/openless-all/app/src-tauri/src/commands/local_asr.rs new file mode 100644 index 00000000..c89dc960 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/local_asr.rs @@ -0,0 +1,309 @@ +use super::*; + +use crate::asr::local::{ + download::{fetch_remote_info, RemoteInfo}, + DownloadManager, ModelId, ModelStatus, PROVIDER_ID as LOCAL_PROVIDER_ID, +}; + +#[derive(Serialize)] +#[serde(rename_all = "camelCase")] +pub struct LocalAsrSettings { + pub provider_id: String, + pub active_model: String, + pub mirror: String, + pub models_base_dir: Option, + pub models_root_dir: String, + /// macOS 才编入引擎；Windows 端 UI 需要据此把"开始下载"按钮灰掉。 + pub engine_available: bool, +} + +#[tauri::command] +pub fn local_asr_get_settings(coord: CoordinatorState<'_>) -> LocalAsrSettings { + let prefs = coord.prefs().get(); + let models_base_dir = non_empty_string(prefs.local_asr_models_base_dir.clone()); + let models_root_dir = crate::persistence::models_root_for_base_dir(models_base_dir.as_deref()) + .map(|path| path.display().to_string()) + .unwrap_or_default(); + LocalAsrSettings { + provider_id: LOCAL_PROVIDER_ID.into(), + active_model: prefs.local_asr_active_model, + mirror: prefs.local_asr_mirror, + models_base_dir, + models_root_dir, + engine_available: cfg!(target_os = "macos"), + } +} + +fn non_empty_string(value: String) -> Option { + let trimmed = value.trim(); + if trimmed.is_empty() { + None + } else { + Some(trimmed.to_string()) + } +} + +#[derive(Debug, Clone, Serialize)] +#[serde(rename_all = "camelCase")] +pub struct LocalAsrStorageSettings { + pub models_base_dir: Option, + pub models_root_dir: String, + pub is_default: bool, +} + +#[tauri::command] +pub fn local_asr_storage_settings( + coord: CoordinatorState<'_>, +) -> Result { + let prefs = coord.prefs().get(); + let models_base_dir = non_empty_string(prefs.local_asr_models_base_dir); + let models_root_dir = crate::persistence::models_root_for_base_dir(models_base_dir.as_deref()) + .map_err(|e| format!("{e:#}"))? + .display() + .to_string(); + Ok(LocalAsrStorageSettings { + is_default: models_base_dir.is_none(), + models_base_dir, + models_root_dir, + }) +} + +#[tauri::command] +pub async fn local_asr_set_models_base_dir( + coord: CoordinatorState<'_>, + qwen_manager: State<'_, Arc>, + foundry_runtime: State<'_, Arc>, + sherpa_manager: State<'_, Arc>, + sherpa_runtime: State<'_, Arc>, + models_base_dir: Option, +) -> Result { + let next_base_dir = models_base_dir + .as_deref() + .map(str::trim) + .filter(|value| !value.is_empty()) + .map(str::to_string); + let new_root = crate::persistence::validate_models_base_dir(next_base_dir.as_deref()) + .map_err(|e| format!("{e:#}"))?; + + let prefs = coord.prefs().get(); + let current_base_dir = non_empty_string(prefs.local_asr_models_base_dir.clone()); + let old_root = crate::persistence::models_root_for_base_dir(current_base_dir.as_deref()) + .map_err(|e| format!("{e:#}"))?; + let same_root = same_path_for_command(&old_root, &new_root); + if same_root && current_base_dir == next_base_dir { + return local_asr_storage_settings(coord); + } + if !same_root && foundry_runtime.storage_configuration_locked() { + return Err( + "Foundry Local has already been initialized in this app session; restart OpenLess before changing the model storage location" + .to_string(), + ); + } + + quiesce_local_asr_storage_users( + coord.inner(), + qwen_manager.inner(), + foundry_runtime.inner(), + sherpa_manager.inner(), + sherpa_runtime.inner(), + ) + .await?; + crate::persistence::migrate_models_root(&old_root, &new_root).map_err(|e| format!("{e:#}"))?; + + let mut prefs = prefs; + prefs.local_asr_models_base_dir = next_base_dir.clone().unwrap_or_default(); + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + local_asr_storage_settings(coord) +} + +fn same_path_for_command(left: &std::path::Path, right: &std::path::Path) -> bool { + match (left.canonicalize(), right.canonicalize()) { + (Ok(left), Ok(right)) => left == right, + _ => false, + } +} + +async fn quiesce_local_asr_storage_users( + coord: &Arc, + qwen_manager: &Arc, + foundry_runtime: &Arc, + sherpa_manager: &Arc, + sherpa_runtime: &Arc, +) -> Result<(), String> { + for model_id in ModelId::all() { + qwen_manager.cancel(*model_id); + } + for model in crate::asr::local::sherpa::MODELS { + sherpa_manager.cancel(model.alias); + } + foundry_runtime.request_cancel_prepare(); + sherpa_runtime.request_cancel_prepare(); + coord.release_local_asr_engine(); + foundry_runtime + .release_now() + .await + .map_err(|e| format!("{e:#}"))?; + sherpa_runtime + .release_now() + .await + .map_err(|e| format!("{e:#}"))?; + + for _ in 0..50 { + let qwen_active = ModelId::all() + .iter() + .any(|model_id| qwen_manager.is_active(*model_id)); + let sherpa_active = crate::asr::local::sherpa::MODELS + .iter() + .any(|model| sherpa_manager.is_active(model.alias)); + if !qwen_active && !sherpa_active { + return Ok(()); + } + tokio::time::sleep(std::time::Duration::from_millis(100)).await; + } + Err("local ASR downloads are still stopping; retry after cancellation finishes".to_string()) +} + +#[tauri::command] +pub fn local_asr_set_active_model( + coord: CoordinatorState<'_>, + model_id: String, +) -> Result<(), String> { + if ModelId::from_str(&model_id).is_none() { + return Err(format!("unknown model id: {model_id}")); + } + let mut prefs = coord.prefs().get(); + prefs.local_asr_active_model = model_id; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn local_asr_set_mirror(coord: CoordinatorState<'_>, mirror: String) -> Result<(), String> { + let _normalized = Mirror::from_str(&mirror); + let mut prefs = coord.prefs().get(); + prefs.local_asr_mirror = mirror; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn local_asr_list_models() -> Vec { + crate::asr::local::models::list_status() +} + +/// 实时去 HuggingFace API 拉某个模型的真实文件清单 + 总尺寸； +/// 前端在显示模型卡时调一次，避免硬编码尺寸过期。 +#[tauri::command] +pub async fn local_asr_fetch_remote_info( + model_id: String, + mirror: Option, +) -> Result { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + let m = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); + fetch_remote_info(id, m).await.map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn local_asr_download_model( + app: AppHandle, + manager: State<'_, Arc>, + model_id: String, + mirror: Option, +) -> Result<(), String> { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + let m = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); + manager.start(app, id, m); + Ok(()) +} + +#[tauri::command] +pub fn local_asr_cancel_download( + manager: State<'_, Arc>, + model_id: String, +) -> Result<(), String> { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + manager.cancel(id); + Ok(()) +} + +#[tauri::command] +pub fn local_asr_delete_model(coord: CoordinatorState<'_>, model_id: String) -> Result<(), String> { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + // 如果内存里加载的就是要删的这个模型，先释放：否则 mmap 残留指向已 unlink 的文件， + // 且 RAM 直到下次切模型 / 用户手动按"释放"才回收。 + if coord.local_asr_loaded_model().as_deref() == Some(id.as_str()) { + coord.release_local_asr_engine(); + } + crate::asr::local::models::delete_model(id).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn local_asr_model_dir(model_id: String) -> Result { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + crate::asr::local::models::model_dir(id) + .map(|path| path.display().to_string()) + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn local_asr_reveal_model_dir(model_id: String) -> Result<(), String> { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + let dir = crate::asr::local::models::model_dir(id).map_err(|e| format!("{e:#}"))?; + std::fs::create_dir_all(&dir).map_err(|e| format!("create {} failed: {e}", dir.display()))?; + open_path_in_file_manager(&dir) +} + +#[tauri::command] +pub fn local_asr_reveal_models_root(coord: CoordinatorState<'_>) -> Result<(), String> { + let prefs = coord.prefs().get(); + let base_dir = non_empty_string(prefs.local_asr_models_base_dir); + let dir = crate::persistence::models_root_for_base_dir(base_dir.as_deref()) + .map_err(|e| format!("{e:#}"))?; + open_path_in_file_manager(&dir) +} + +#[tauri::command] +pub async fn local_asr_test_model( + model_id: String, +) -> Result { + let id = ModelId::from_str(&model_id).ok_or_else(|| format!("unknown model id: {model_id}"))?; + crate::asr::local::test_run::run_test(id) + .await + .map_err(|e| format!("{e:#}")) +} + +#[derive(Serialize)] +#[serde(rename_all = "camelCase")] +pub struct LocalAsrEngineStatus { + pub loaded: bool, + pub model_id: Option, + pub keep_loaded_secs: u32, +} + +#[tauri::command] +pub fn local_asr_engine_status(coord: CoordinatorState<'_>) -> LocalAsrEngineStatus { + let prefs = coord.prefs().get(); + LocalAsrEngineStatus { + loaded: coord.local_asr_loaded_model().is_some(), + model_id: coord.local_asr_loaded_model(), + keep_loaded_secs: prefs.local_asr_keep_loaded_secs, + } +} + +#[tauri::command] +pub fn local_asr_release_engine(coord: CoordinatorState<'_>) { + coord.release_local_asr_engine(); +} + +#[tauri::command] +pub fn local_asr_preload(coord: tauri::State<'_, std::sync::Arc>) { + coord.preload_local_asr_in_background(); +} + +#[tauri::command] +pub fn local_asr_set_keep_loaded_secs( + coord: CoordinatorState<'_>, + seconds: u32, +) -> Result<(), String> { + let mut prefs = coord.prefs().get(); + prefs.local_asr_keep_loaded_secs = seconds; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} diff --git a/openless-all/app/src-tauri/src/commands/marketplace.rs b/openless-all/app/src-tauri/src/commands/marketplace.rs new file mode 100644 index 00000000..52c558cf --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/marketplace.rs @@ -0,0 +1,404 @@ +use super::*; + +// ─────────────────────────── marketplace (Phase A) ─────────────────────────── +// +// 客户端跟 marketplace backend 的 HTTP 客户端封装。Backend URL 走 prefs +// `marketplace_base_url`（默认 http://127.0.0.1:8090 开发；生产用户填 https://api.）。 +// dev-mode auth：用户在 Settings 填 `marketplace_dev_login`（GitHub 风格 username）， +// 后续 OAuth 接入时换成 token 字段。 +// +// 5 个 IPC： +// - marketplace_list 列表 + 搜索 + 排序 +// - marketplace_detail 详情（含完整 prompt） +// - marketplace_install 下载 ZIP + 直接调 import_from_zip 装到本地 +// - marketplace_upload 把本地某个 style pack export ZIP → multipart 上传 +// - marketplace_like 点赞 + +#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] +#[serde(rename_all = "camelCase")] +pub struct MarketplaceListItem { + pub id: String, + pub slug: String, + pub name: String, + pub description: String, + #[serde(default)] + pub author_login: String, + pub version: String, + pub base_mode: String, + #[serde(default)] + pub tags: Vec, + pub like_count: i64, + pub download_count: i64, + pub published_at: String, + pub updated_at: String, + pub origin_pack_id: Option, + pub origin_author_login: Option, +} + +#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] +#[serde(rename_all = "camelCase")] +pub struct MarketplaceDetail { + #[serde(flatten)] + pub summary: MarketplaceListItem, + pub prompt: String, + pub state: String, +} + +#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)] +#[serde(rename_all = "camelCase")] +pub struct MarketplaceMyPackItem { + #[serde(flatten)] + pub summary: MarketplaceListItem, + pub state: String, +} + +/// 风格市场 backend URL —— 硬编码到生产云端，不再读 prefs。 +/// +/// 历史上这里读 `prefs.marketplace_base_url`（dev 本地可填 127.0.0.1:8090）， +/// 现在风格市场已经稳定部署在 apic.openless.top，把 URL 锁死避免用户误改 / 写错。 +/// 参数 `_prefs` 保留是为不动调用点签名；将来需要白名单 / 多 endpoint 时再开口。 +pub(crate) const MARKETPLACE_BASE_URL: &str = "https://apic.openless.top"; + +fn marketplace_url_from_prefs(_prefs: &UserPreferences) -> String { + MARKETPLACE_BASE_URL.to_string() +} + +fn marketplace_dev_user(prefs: &UserPreferences) -> String { + prefs.marketplace_dev_login.trim().to_string() +} + +#[tauri::command] +pub async fn marketplace_list( + coord: CoordinatorState<'_>, + query: Option, + sort: Option, + limit: Option, +) -> Result, String> { + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let mut url = reqwest::Url::parse(&format!("{base}/packs")) + .map_err(|e| format!("invalid marketplace url: {e}"))?; + if let Some(q) = query.as_deref() { + if !q.trim().is_empty() { + url.query_pairs_mut().append_pair("q", q.trim()); + } + } + if let Some(s) = sort.as_deref() { + if !s.trim().is_empty() { + url.query_pairs_mut().append_pair("sort", s.trim()); + } + } + if let Some(n) = limit { + url.query_pairs_mut().append_pair("limit", &n.to_string()); + } + let resp = net::send_with_retry(|| { + net::http() + .get(url.clone()) + .timeout(std::time::Duration::from_secs(10)) + }) + .await + .map_err(|e| format!("marketplace request failed: {e}"))?; + if !resp.status().is_success() { + let status = resp.status(); + let body = resp.text().await.unwrap_or_default(); + return Err(format!("marketplace HTTP {status}: {body}")); + } + let items: Vec = resp + .json() + .await + .map_err(|e| format!("parse failed: {e}"))?; + Ok(items) +} + +#[tauri::command] +pub async fn marketplace_detail( + coord: CoordinatorState<'_>, + pack_id: String, +) -> Result { + if !is_valid_session_id(&pack_id) { + return Err("invalid pack id".into()); + } + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let url = format!("{base}/packs/{pack_id}"); + let resp = net::send_with_retry(|| { + net::http() + .get(&url) + .timeout(std::time::Duration::from_secs(10)) + }) + .await + .map_err(|e| format!("marketplace request failed: {e}"))?; + if !resp.status().is_success() { + let status = resp.status(); + return Err(format!("marketplace HTTP {status}")); + } + resp.json::() + .await + .map_err(|e| format!("parse failed: {e}")) +} + +#[tauri::command] +pub async fn marketplace_install( + coord: CoordinatorState<'_>, + pack_id: String, +) -> Result { + // 安全校验：pack_id 来自远端 backend，可能含路径遍历 segment。 + // 用跟 read_audio_recording 同样的 UUID-v4 白名单挡住 ../ / 绝对路径等。 + // backend 当前用 Uuid::new_v4 生成所有 id，合法 id 必然匹配。 + if !is_valid_session_id(&pack_id) { + return Err("invalid pack id".into()); + } + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + + // 先拉 detail 拿 authorLogin —— 装好后本地写 originAuthorLogin， + // 后续编辑+发布时 backend 据此判 supersede（原作者）vs derivative（他人 fork）。 + let detail_url = format!("{base}/packs/{pack_id}"); + let detail: serde_json::Value = net::send_with_retry(|| { + net::http() + .get(&detail_url) + .timeout(std::time::Duration::from_secs(15)) + }) + .await + .map_err(|e| format!("marketplace detail failed: {e}"))? + .error_for_status() + .map_err(|e| format!("marketplace detail HTTP error: {e}"))? + .json() + .await + .map_err(|e| format!("parse detail failed: {e}"))?; + let origin_author_login = detail + .get("authorLogin") + .and_then(|v| v.as_str()) + .map(|s| s.to_string()); + + let download_url = format!("{base}/packs/{pack_id}/download"); + let bytes = net::send_with_retry(|| { + net::http() + .get(&download_url) + .timeout(std::time::Duration::from_secs(30)) + }) + .await + .map_err(|e| format!("marketplace download failed: {e}"))? + .error_for_status() + .map_err(|e| format!("marketplace HTTP error: {e}"))? + .bytes() + .await + .map_err(|e| format!("read body failed: {e}"))?; + + // pack_id 已经过 UUID 白名单，拼临时文件路径安全。 + let tmp = std::env::temp_dir().join(format!("openless-marketplace-{pack_id}.zip")); + std::fs::write(&tmp, &bytes).map_err(|e| format!("write tmp zip: {e}"))?; + let imported_result = coord + .style_packs() + .import_from_zip(&tmp) + .map_err(|e| e.to_string()); + let _ = std::fs::remove_file(&tmp); + let imported = imported_result?; + + // 绑定 origin —— 后续编辑+发布走 derivative / supersede 分支。 + coord + .style_packs() + .set_origin(&imported.id, Some(pack_id), origin_author_login) + .map_err(|e| format!("set origin failed: {e}")) +} + +#[tauri::command] +pub async fn marketplace_upload( + coord: CoordinatorState<'_>, + pack_id: String, + origin_pack_id: Option, +) -> Result { + // 本地 pack id 形态：`builtin.light` / 用户 slug / Uuid。用 local 白名单挡 `..` / `/` / `\`。 + if !is_valid_local_pack_id(&pack_id) { + return Err("invalid pack id".into()); + } + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let dev_user = marketplace_dev_user(&prefs); + if dev_user.is_empty() { + return Err("未登录：先在 Settings 填发布者名字".into()); + } + + // 拉本地 pack 拿 origin_pack_id —— 装过的 pack 这里有值， + // backend 据此判同作者就 supersede 原行（新版本），他人就 derivative（独立新 row）。 + let local_pack = coord + .style_packs() + .get(&pack_id) + .map_err(|e| format!("local pack not found: {e}"))?; + let origin_pack_id = origin_pack_id + .filter(|id| is_valid_session_id(id)) + .or_else(|| local_pack.origin_pack_id.clone()); + + // 先 export 本地 pack → 临时 ZIP + let tmp = std::env::temp_dir().join(format!("openless-marketplace-upload-{pack_id}.zip")); + coord + .style_packs() + .export_to_zip(&pack_id, &tmp) + .map_err(|e| format!("export local pack failed: {e}"))?; + let bytes = std::fs::read(&tmp).map_err(|e| format!("read exported zip: {e}"))?; + let _ = std::fs::remove_file(&tmp); + + // multipart 上传：表单是流式 body，不走 send_with_retry 的闭包重试；改用共享 + // 客户端 —— 之前 list/detail 命令若已打开过连接，这里直接复用连接池里的连接。 + let part = reqwest::multipart::Part::bytes(bytes) + .file_name(format!("{pack_id}.zip")) + .mime_str("application/zip") + .map_err(|e| format!("multipart build failed: {e}"))?; + let mut form = reqwest::multipart::Form::new().part("file", part); + if let Some(ref oid) = origin_pack_id { + form = form.text("origin_pack_id", oid.clone()); + } + let resp = net::http() + .post(format!("{base}/packs")) + .header("X-Dev-User", dev_user) + .timeout(std::time::Duration::from_secs(30)) + .multipart(form) + .send() + .await + .map_err(|e| format!("upload request failed: {e}"))?; + let status = resp.status(); + let body = resp + .text() + .await + .unwrap_or_else(|e| format!("read body failed: {e}")) + .clone(); + if !status.is_success() { + return Err(format!("upload HTTP {status}: {body}")); + } + let parsed = serde_json::from_str::(&body) + .map_err(|e| format!("parse upload response failed: {e}"))?; + + // 本地从未绑定 origin（首次上传一个本地原创 pack）→ 把 backend 分配的 pack id 写回本地， + // 让用户在同设备上后续编辑能继续走「同作者 supersede」分支，更新自己原创的包。 + if origin_pack_id.is_none() { + if let Some(remote_id) = parsed.get("id").and_then(|v| v.as_str()) { + let prefs2 = coord.prefs().get(); + let dev_user2 = marketplace_dev_user(&prefs2); + let _ = coord.style_packs().set_origin( + &pack_id, + Some(remote_id.to_string()), + Some(dev_user2), + ); + } + } + + Ok(parsed) +} + +#[tauri::command] +pub async fn marketplace_like( + coord: CoordinatorState<'_>, + pack_id: String, +) -> Result { + if !is_valid_session_id(&pack_id) { + return Err("invalid pack id".into()); + } + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let dev_user = marketplace_dev_user(&prefs); + if dev_user.is_empty() { + return Err("未登录：先在 Settings 填发布者名字".into()); + } + let like_url = format!("{base}/packs/{pack_id}/like"); + let resp = net::send_with_retry(|| { + net::http() + .post(&like_url) + .header("X-Dev-User", dev_user.as_str()) + .timeout(std::time::Duration::from_secs(10)) + }) + .await + .map_err(|e| format!("like request failed: {e}"))?; + if !resp.status().is_success() { + return Err(format!("like HTTP {}", resp.status())); + } + resp.json::() + .await + .map_err(|e| format!("parse failed: {e}")) +} + +/// 撤回自己发布的 pack（后端软删 state='withdrawn'，前端列表不再可见）。 +/// pack_id 来自远端，必须是 UUID-v4。 +#[tauri::command] +pub async fn marketplace_delete( + coord: CoordinatorState<'_>, + pack_id: String, +) -> Result<(), String> { + if !is_valid_session_id(&pack_id) { + return Err("invalid pack id".into()); + } + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let dev_user = marketplace_dev_user(&prefs); + if dev_user.is_empty() { + return Err("未登录：先在 Settings 填发布者名字".into()); + } + let delete_url = format!("{base}/packs/{pack_id}"); + let resp = net::send_with_retry(|| { + net::http() + .delete(&delete_url) + .header("X-Dev-User", dev_user.as_str()) + .timeout(std::time::Duration::from_secs(15)) + }) + .await + .map_err(|e| format!("delete request failed: {e}"))?; + let status = resp.status(); + if !status.is_success() { + let body = resp.text().await.unwrap_or_default(); + return Err(format!("delete HTTP {status}: {body}")); + } + Ok(()) +} + +/// 拉当前用户赞过的所有 pack id，用于客户端市场页面渲染红心 + 「我赞过的」过滤。 +#[tauri::command] +pub async fn marketplace_my_likes(coord: CoordinatorState<'_>) -> Result, String> { + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let dev_user = marketplace_dev_user(&prefs); + if dev_user.is_empty() { + return Ok(Vec::new()); // 未登录就空集合，UI 渲染无红心 + } + let likes_url = format!("{base}/me/likes"); + let resp = net::send_with_retry(|| { + net::http() + .get(&likes_url) + .header("X-Dev-User", dev_user.as_str()) + .timeout(std::time::Duration::from_secs(10)) + }) + .await + .map_err(|e| format!("my-likes request failed: {e}"))?; + if !resp.status().is_success() { + return Err(format!("my-likes HTTP {}", resp.status())); + } + resp.json::>() + .await + .map_err(|e| format!("parse my-likes failed: {e}")) +} + +/// 拉当前用户发布过的 pack（含审核中/已通过/已拒绝/已撤回），用于「我的发布」页面。 +#[tauri::command] +pub async fn marketplace_my_packs( + coord: CoordinatorState<'_>, +) -> Result, String> { + let prefs = coord.prefs().get(); + let base = marketplace_url_from_prefs(&prefs); + let dev_user = marketplace_dev_user(&prefs); + if dev_user.is_empty() { + return Ok(Vec::new()); + } + let packs_url = format!("{base}/me/packs"); + let resp = net::send_with_retry(|| { + net::http() + .get(&packs_url) + .header("X-Dev-User", dev_user.as_str()) + .timeout(std::time::Duration::from_secs(10)) + }) + .await + .map_err(|e| format!("my-packs request failed: {e}"))?; + if !resp.status().is_success() { + return Err(format!("my-packs HTTP {}", resp.status())); + } + resp.json::>() + .await + .map_err(|e| format!("parse my-packs failed: {e}")) +} diff --git a/openless-all/app/src-tauri/src/commands/misc.rs b/openless-all/app/src-tauri/src/commands/misc.rs new file mode 100644 index 00000000..bb2bfc9c --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/misc.rs @@ -0,0 +1,123 @@ +use super::*; + +#[derive(Debug, Clone, Serialize)] +#[serde(rename_all = "camelCase")] +pub struct NetworkCheckResult { + pub online: bool, + pub latency_ms: Option, +} + +#[tauri::command] +pub async fn check_network() -> NetworkCheckResult { + // 探一个真实存在的接口。旧逻辑探 `/health` —— 实测返回 404，链路正常也永远判 + // 离线；且用 HEAD（后端只挂 GET）。改成 GET `/packs`，拿到任意 HTTP 响应即算通。 + // + // 单发、不走 send_with_retry：这是每 30s 跑一次的状态探针，要的是「快」。10 次 + // 退避重试会让被过滤 / 黑洞的网络下探测拖到近一分钟、状态灯像卡死。偶发的瞬时 + // 误判由下一个 30s 周期自动纠正。仍用 net::http() 共享连接池。 + let url = format!("{MARKETPLACE_BASE_URL}/packs?limit=1"); + let start = std::time::Instant::now(); + match net::http() + .get(&url) + .timeout(std::time::Duration::from_secs(8)) + .send() + .await + { + Ok(_) => NetworkCheckResult { + online: true, + latency_ms: Some(start.elapsed().as_millis() as u64), + }, + Err(_) => NetworkCheckResult { + online: false, + latency_ms: None, + }, + } +} + +#[tauri::command] +pub fn get_hotkey_status(coord: CoordinatorState<'_>) -> HotkeyStatus { + coord.hotkey_status() +} + +#[tauri::command] +pub fn get_hotkey_capability(coord: CoordinatorState<'_>) -> HotkeyCapability { + coord.hotkey_capability() +} + +#[tauri::command] +pub fn set_shortcut_recording_active(coord: CoordinatorState<'_>, active: bool) { + coord.set_shortcut_recording_active(active); +} + +#[tauri::command] +pub fn get_windows_ime_status() -> WindowsImeStatus { + crate::windows_ime_profile::get_windows_ime_status() +} + +#[tauri::command] +pub fn list_microphone_devices() -> Result, String> { + crate::recorder::list_input_devices().map_err(|e| e.to_string()) +} + +#[tauri::command] +pub async fn start_microphone_level_monitor( + app: AppHandle, + device_name: String, +) -> Result<(), String> { + tauri::async_runtime::spawn_blocking(move || { + let state = app.state::(); + if let Some(existing) = state.lock().take() { + existing.stop(); + } + + let selected = device_name.trim().to_string(); + let microphone_device_name = if selected.is_empty() { + None + } else { + Some(selected) + }; + let consumer: Arc = Arc::new(LevelProbeConsumer); + let level_app = app.clone(); + let level_handler: Arc = Arc::new(move |level| { + let _ = level_app.emit("microphone:level", serde_json::json!({ "level": level })); + }); + let (recorder, _runtime_errors, _archive_active) = + Recorder::start(microphone_device_name, consumer, level_handler, None) + .map_err(|e| e.to_string())?; + *state.lock() = Some(recorder); + Ok(()) + }) + .await + .map_err(|e| format!("start microphone monitor task failed: {e}"))? +} + +#[tauri::command] +pub async fn stop_microphone_level_monitor(app: AppHandle) { + let _ = tauri::async_runtime::spawn_blocking(move || { + let state = app.state::(); + let recorder = state.lock().take(); + if let Some(recorder) = recorder { + recorder.stop(); + } + }) + .await; +} + +/// 把当前会话的 openless.log 复制到用户选择的位置（前端用 plugin-dialog 拿 target_path）。 +/// 路径来自 lib::log_dir_path() —— mac: ~/Library/Logs/OpenLess/openless.log， +/// windows: %LOCALAPPDATA%\OpenLess\Logs\openless.log。 +#[tauri::command] +pub fn export_error_log(target_path: String) -> Result<(), String> { + let src = crate::log_dir_path().join("openless.log"); + if !src.exists() { + return Err(format!("日志文件不存在：{}", src.display())); + } + std::fs::copy(&src, std::path::Path::new(&target_path)) + .map(|_| ()) + .map_err(|e| format!("复制日志失败：{}", e)) +} + +// ─────────────────────────── unused but exported (silences dead_code) ─────────────────────────── + +#[allow(dead_code)] +fn _ensure_snapshot_used(_: CredentialsSnapshot) {} diff --git a/openless-all/app/src-tauri/src/commands/mod.rs b/openless-all/app/src-tauri/src/commands/mod.rs new file mode 100644 index 00000000..46ee988f --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/mod.rs @@ -0,0 +1,1364 @@ +#![cfg_attr(target_os = "linux", allow(dead_code, unused_variables))] +//! Tauri command surface — every IPC entry the React UI invokes lives here. +//! +//! issue: 历史上整个 IPC 表（89 个 `#[tauri::command]` + 跨域 helper）挤在单个 +//! 4800 行的 `commands.rs` 里。按单一职责拆成 `commands/` 下的域模块（settings / +//! credentials / providers / history / …），每个文件聚焦一个领域。对外路径保持不变： +//! 本模块用 `pub use ::*` glob 重导出每个子模块，`commands::` 仍然解析， +//! `lib.rs` 的 `generate_handler!` 清单与类型引用零改动。`#[tauri::command]` 生成的 +//! `__cmd__` 伴生项也随 glob 一并重导出——这是 Tauri 拆分命令文件的标准做法。 + +use std::sync::Arc; + +use parking_lot::Mutex; +use tauri::{Manager, State}; + +// 跨域共享的 crate 级导入：以 `pub(crate) use` 重导出，子模块用 `use super::*;` +// 即可拿到，避免在 16 个文件里重复同一组 import。 +pub(crate) use serde::Serialize; +pub(crate) use serde_json::Value; +pub(crate) use tauri::{AppHandle, Emitter, Window}; + +pub(crate) use crate::asr::local::foundry::{ + model_alias_is_known, FoundryCatalogModel, FoundryPrepareProgressPayload, FoundryRuntimeStatus, + DEFAULT_MODEL_ALIAS, PROVIDER_ID as FOUNDRY_LOCAL_PROVIDER_ID, +}; +pub(crate) use crate::asr::local::sherpa::{ + model_alias_is_known as sherpa_model_alias_is_known, SherpaCatalogModel, + SherpaPrepareProgressPayload, SherpaRuntimeStatus, + DEFAULT_MODEL_ALIAS as SHERPA_DEFAULT_MODEL_ALIAS, +}; +pub(crate) use crate::asr::local::sherpa_download::{ + fetch_remote_info as fetch_sherpa_remote_info, SherpaDownloadManager, SherpaRemoteInfo, +}; +pub(crate) use crate::asr::local::{FoundryLocalRuntime, Mirror, SherpaOnnxRuntime}; +pub(crate) use crate::coordinator::Coordinator; +pub(crate) use crate::net; +pub(crate) use crate::permissions::{self, PermissionStatus}; +pub(crate) use crate::persistence::{ + sync_style_pack_preferences, CredentialAccount, CredentialsSnapshot, CredentialsVault, + PreferencesStore, +}; +pub(crate) use crate::polish::{ + http_client_builder, CodexOAuthConfig, CodexOAuthCredentials, CodexOAuthLLMProvider, LLMError, + OpenAICompatibleConfig, OpenAICompatibleLLMProvider, CODEX_DEFAULT_MODEL, + CODEX_OAUTH_PROVIDER_ID, +}; +pub(crate) use crate::recorder::{AudioConsumer, Recorder}; +pub(crate) use crate::types::{ + builtin_style_pack_id, default_active_style_pack_id, ChineseScriptPreference, ComboBinding, + CorrectionRule, CredentialsStatus, DictationSession, DictionaryEntry, HotkeyCapability, + HotkeyStatus, OutputLanguagePreference, PolishMode, ShortcutBinding, StylePack, StylePackKind, + StylePackRuntimeDiagnostics, StyleSystemPrompts, UpdateChannel, UserPreferences, + VocabPresetStore, WindowsImeStatus, +}; + +mod credentials; +mod dictation; +mod dictionary; +mod foundry_asr; +mod github_oauth; +mod history; +mod hotkeys; +mod local_asr; +mod marketplace; +mod misc; +mod permissions_cmds; +mod providers; +mod qa; +mod settings; +mod sherpa_asr; +mod style_packs; + +pub use credentials::*; +pub use dictation::*; +pub use dictionary::*; +pub use foundry_asr::*; +pub use github_oauth::*; +pub use history::*; +pub use hotkeys::*; +pub use local_asr::*; +pub use marketplace::*; +pub use misc::*; +pub use permissions_cmds::*; +pub use providers::*; +pub use qa::*; +pub use settings::*; +// sherpa_onnx_asr_* 命令整组 `#[cfg(target_os = "windows")]`（见 lib.rs 的 +// generate_handler! 清单）。非 Windows 平台这组 glob 重导出无人引用，会触发 +// unused_imports；这是平台 cfg 的正常结果，不是真正的死代码。 +#[allow(unused_imports)] +pub use sherpa_asr::*; +pub use style_packs::*; + +pub(crate) type CoordinatorState<'a> = State<'a, Arc>; +pub type MicrophoneMonitorState = Mutex>; +pub type TrayMicrophoneMenuState = Mutex>; + +pub struct TrayMicrophoneMenuItem { + pub id: String, + pub device_name: String, + pub item: tauri::menu::CheckMenuItem, +} + +pub fn sync_tray_microphone_selection(items: &[TrayMicrophoneMenuItem], device_name: &str) { + for item in items { + let _ = item.item.set_checked(item.device_name == device_name); + } +} + +pub(crate) struct LevelProbeConsumer; + +impl AudioConsumer for LevelProbeConsumer { + fn consume_pcm_chunk(&self, _pcm: &[u8]) {} +} + +// ─────────────────────────── 跨域共享校验 helper ─────────────────────────── + +/// UUID-v4 字面校验：36 字符 + 5 段 `-` 分隔（8-4-4-4-12）+ 仅 ASCII 十六进制。 +/// 用于 install/detail/like —— pack_id 来自远端服务器，必须是它发的 UUID。 +pub(crate) fn is_valid_session_id(s: &str) -> bool { + if s.len() != 36 { + return false; + } + let bytes = s.as_bytes(); + for (i, b) in bytes.iter().enumerate() { + let is_dash_position = matches!(i, 8 | 13 | 18 | 23); + if is_dash_position { + if *b != b'-' { + return false; + } + } else if !b.is_ascii_hexdigit() { + return false; + } + } + true +} + +/// 本地 style pack id 白名单：`[A-Za-z0-9._-]`、长度 1..=128。 +/// 上传走本地 id（`builtin.light` / 用户自取 slug / UUID 都可），不是远端 UUID。 +/// 仍阻断 `..` / `/` / `\` / 控制字符，避免 path traversal 进临时 zip 文件名。 +pub(crate) fn is_valid_local_pack_id(s: &str) -> bool { + if s.is_empty() || s.len() > 128 { + return false; + } + s.bytes() + .all(|b| b.is_ascii_alphanumeric() || b == b'.' || b == b'-' || b == b'_') +} + +// ─── 在系统文件管理器中打开路径（三个 ASR 模块共用，cfg 分平台）─── + +#[cfg(target_os = "windows")] +pub(crate) fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { + use windows::core::PCWSTR; + use windows::Win32::UI::Shell::ShellExecuteW; + use windows::Win32::UI::WindowsAndMessaging::SW_SHOWNORMAL; + + fn wide_null(value: &str) -> Vec { + value.encode_utf16().chain(std::iter::once(0)).collect() + } + + let operation = wide_null("open"); + let target = wide_null(&path.display().to_string()); + let result = unsafe { + ShellExecuteW( + None, + PCWSTR(operation.as_ptr()), + PCWSTR(target.as_ptr()), + PCWSTR::null(), + PCWSTR::null(), + SW_SHOWNORMAL, + ) + }; + if result.0 as isize <= 32 { + Err(format!("ShellExecuteW failed: {}", result.0 as isize)) + } else { + Ok(()) + } +} + +#[cfg(target_os = "macos")] +pub(crate) fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { + std::process::Command::new("open") + .arg(path) + .spawn() + .map(|_| ()) + .map_err(|e| e.to_string()) +} + +#[cfg(all(unix, not(target_os = "macos")))] +pub(crate) fn open_path_in_file_manager(path: &std::path::Path) -> Result<(), String> { + std::process::Command::new("xdg-open") + .arg(path) + .spawn() + .map(|_| ()) + .map_err(|e| e.to_string()) +} + +#[cfg(test)] +mod tests { + use super::{is_valid_local_pack_id, is_valid_session_id}; + use crate::commands::credentials::{ + asr_configured_for_provider, llm_configured_for_provider, + local_asr_release_plan_for_provider, release_foundry_runtime_if_inactive, + release_sherpa_runtime_if_inactive, + }; + use crate::commands::foundry_asr::{ + active_foundry_model_from_prefs, normalize_foundry_language_hint, + validate_foundry_model_alias, + }; + use crate::commands::providers::{ + active_asr_is_keyless_for_validation, asr_transcriptions_url, fetch_provider_models, + is_gemini_base_url, models_url, parse_gemini_model_ids, parse_model_ids, ProviderConfig, + }; + use crate::commands::settings::{ + parse_latest_beta_from_atom, persist_settings, SettingsWriter, + }; + use crate::commands::sherpa_asr::{ + active_sherpa_model_from_prefs, normalize_sherpa_language_hint, validate_sherpa_model_alias, + }; + use crate::persistence::CredentialsSnapshot; + use crate::types::{ + ComboBinding, HotkeyBinding, HotkeyMode, HotkeyTrigger, ShortcutBinding, UserPreferences, + }; + use std::io::{Read, Write}; + use std::net::TcpListener; + use std::sync::Mutex; + use std::thread; + + #[derive(Default)] + struct FakeSettingsWriter { + saved: Mutex>, + active_asr_provider_syncs: Mutex>, + write_settings_error: Mutex>, + write_settings_errors: Mutex>>, + active_asr_provider_sync_error: Mutex>, + active_asr_provider_sync_errors: Mutex>>, + dictation_refreshes: Mutex, + qa_refreshes: Mutex, + combo_refreshes: Mutex, + translation_refreshes: Mutex, + switch_style_refreshes: Mutex, + open_app_refreshes: Mutex, + coding_agent_refreshes: Mutex, + } + + fn snapshot() -> CredentialsSnapshot { + CredentialsSnapshot::default() + } + + #[test] + fn credentials_status_follows_active_asr_provider_requirements() { + let volcengine = CredentialsSnapshot { + volcengine_app_key: Some("app".into()), + volcengine_access_key: Some("access".into()), + volcengine_resource_id: Some("resource".into()), + ..snapshot() + }; + assert!(asr_configured_for_provider("volcengine", &volcengine)); + + let whisper_key_only = CredentialsSnapshot { + asr_api_key: Some("key".into()), + ..snapshot() + }; + assert!(!asr_configured_for_provider("whisper", &whisper_key_only)); + assert!(asr_configured_for_provider( + crate::asr::bailian::PROVIDER_ID, + &whisper_key_only + )); + + let whisper_keyless_ready = CredentialsSnapshot { + asr_endpoint: Some("https://api.openai.com/v1".into()), + asr_model: Some("whisper-1".into()), + ..snapshot() + }; + assert!(asr_configured_for_provider( + "whisper", + &whisper_keyless_ready + )); + assert!(!asr_configured_for_provider( + crate::asr::bailian::PROVIDER_ID, + &whisper_keyless_ready + )); + + assert!(asr_configured_for_provider( + crate::asr::local::PROVIDER_ID, + &snapshot() + )); + #[cfg(target_os = "windows")] + assert!(asr_configured_for_provider( + crate::asr::local::foundry::PROVIDER_ID, + &snapshot() + )); + #[cfg(target_os = "windows")] + assert!(asr_configured_for_provider( + crate::asr::local::sherpa::PROVIDER_ID, + &snapshot() + )); + #[cfg(not(target_os = "windows"))] + assert!(!asr_configured_for_provider( + crate::asr::local::foundry::PROVIDER_ID, + &snapshot() + )); + #[cfg(not(target_os = "windows"))] + assert!(!asr_configured_for_provider( + crate::asr::local::sherpa::PROVIDER_ID, + &snapshot() + )); + } + + #[test] + fn credentials_status_treats_foundry_local_asr_as_configured() { + #[cfg(target_os = "windows")] + { + assert!(asr_configured_for_provider( + crate::asr::local::foundry::PROVIDER_ID, + &CredentialsSnapshot::default() + )); + } + #[cfg(not(target_os = "windows"))] + { + assert!(!asr_configured_for_provider( + crate::asr::local::foundry::PROVIDER_ID, + &CredentialsSnapshot::default() + )); + } + } + + #[test] + fn local_asr_providers_skip_external_validation() { + assert!(active_asr_is_keyless_for_validation( + crate::asr::local::PROVIDER_ID + )); + #[cfg(target_os = "windows")] + assert!(active_asr_is_keyless_for_validation( + crate::asr::local::foundry::PROVIDER_ID + )); + #[cfg(target_os = "windows")] + assert!(active_asr_is_keyless_for_validation( + crate::asr::local::sherpa::PROVIDER_ID + )); + #[cfg(not(target_os = "windows"))] + assert!(!active_asr_is_keyless_for_validation( + crate::asr::local::foundry::PROVIDER_ID + )); + #[cfg(not(target_os = "windows"))] + assert!(!active_asr_is_keyless_for_validation( + crate::asr::local::sherpa::PROVIDER_ID + )); + assert!(!active_asr_is_keyless_for_validation("volcengine")); + assert!(!active_asr_is_keyless_for_validation("whisper")); + } + + #[test] + fn provider_switch_release_plan_covers_inactive_local_runtimes() { + let qwen = local_asr_release_plan_for_provider(crate::asr::local::PROVIDER_ID); + assert!(!qwen.qwen); + assert!(qwen.foundry); + assert!(qwen.sherpa); + + let foundry = local_asr_release_plan_for_provider(crate::asr::local::foundry::PROVIDER_ID); + assert!(foundry.qwen); + assert!(!foundry.foundry); + assert!(foundry.sherpa); + + let sherpa = local_asr_release_plan_for_provider(crate::asr::local::sherpa::PROVIDER_ID); + assert!(sherpa.qwen); + assert!(sherpa.foundry); + assert!(!sherpa.sherpa); + + let cloud = local_asr_release_plan_for_provider("volcengine"); + assert!(cloud.qwen); + assert!(cloud.foundry); + assert!(cloud.sherpa); + } + + #[cfg(target_os = "windows")] + #[tokio::test] + async fn provider_switch_release_requests_foundry_prepare_cancel_first() { + let runtime = std::sync::Arc::new(crate::asr::local::FoundryLocalRuntime::new()); + + release_foundry_runtime_if_inactive(&runtime, true).await; + + assert!(runtime.cancel_prepare_requested_for_tests()); + } + + #[tokio::test] + async fn provider_switch_release_requests_sherpa_prepare_cancel_first() { + let runtime = std::sync::Arc::new(crate::asr::local::SherpaOnnxRuntime::new()); + + release_sherpa_runtime_if_inactive(&runtime, true).await; + + assert!(runtime.cancel_prepare_requested_for_tests()); + let status = runtime.status_snapshot("sense-voice-small-zh").await; + assert!(!status.runtime_ready); + } + + #[test] + fn foundry_language_hint_accepts_empty_and_lowercase_iso_639_1() { + assert_eq!(normalize_foundry_language_hint("").unwrap(), ""); + assert_eq!(normalize_foundry_language_hint(" ").unwrap(), ""); + assert_eq!(normalize_foundry_language_hint("zh").unwrap(), "zh"); + assert_eq!(normalize_foundry_language_hint(" en ").unwrap(), "en"); + } + + #[test] + fn foundry_language_hint_rejects_non_lowercase_iso_639_1() { + assert!(normalize_foundry_language_hint("ZH").is_err()); + assert!(normalize_foundry_language_hint("zho").is_err()); + assert!(normalize_foundry_language_hint("z1").is_err()); + } + + #[test] + fn foundry_model_alias_validation_rejects_unknown_alias() { + assert!( + validate_foundry_model_alias(crate::asr::local::foundry::DEFAULT_MODEL_ALIAS).is_ok() + ); + assert!(validate_foundry_model_alias("whisper-medium").is_ok()); + assert!(validate_foundry_model_alias("whisper-large-v3-turbo").is_ok()); + assert!(validate_foundry_model_alias("whisper-large").is_err()); + } + + #[test] + fn foundry_active_model_pref_falls_back_to_default_for_unknown_alias() { + let prefs = UserPreferences { + foundry_local_asr_model: "whisper-large".to_string(), + ..Default::default() + }; + + assert_eq!( + active_foundry_model_from_prefs(&prefs), + crate::asr::local::foundry::DEFAULT_MODEL_ALIAS + ); + } + + #[test] + fn foundry_active_model_pref_preserves_large_model_aliases() { + for alias in ["whisper-medium", "whisper-large-v3-turbo"] { + let prefs = UserPreferences { + foundry_local_asr_model: alias.to_string(), + ..Default::default() + }; + + assert_eq!(active_foundry_model_from_prefs(&prefs), alias); + } + } + + #[test] + fn sherpa_language_hint_accepts_empty_and_supported_lowercase_tags() { + assert_eq!(normalize_sherpa_language_hint("").unwrap(), ""); + assert_eq!(normalize_sherpa_language_hint(" ").unwrap(), ""); + assert_eq!(normalize_sherpa_language_hint("zh").unwrap(), "zh"); + assert_eq!(normalize_sherpa_language_hint(" en ").unwrap(), "en"); + assert_eq!(normalize_sherpa_language_hint("zh-cn").unwrap(), "zh-cn"); + assert_eq!(normalize_sherpa_language_hint("yue").unwrap(), "yue"); + } + + #[test] + fn sherpa_language_hint_normalizes_uppercase_and_rejects_digits() { + assert_eq!(normalize_sherpa_language_hint("ZH").unwrap(), "zh"); + assert!(normalize_sherpa_language_hint("zh-1").is_err()); + assert!(normalize_sherpa_language_hint("zh_CN").is_err()); + } + + #[test] + fn sherpa_model_alias_validation_matches_catalog() { + assert!( + validate_sherpa_model_alias(crate::asr::local::sherpa::DEFAULT_MODEL_ALIAS).is_ok() + ); + assert!(validate_sherpa_model_alias("qwen3-asr-0.6b-int8").is_ok()); + assert!( + validate_sherpa_model_alias(crate::asr::local::sherpa::DEFAULT_ONLINE_MODEL_ALIAS) + .is_ok() + ); + assert!(validate_sherpa_model_alias("zipformer-zh-streaming").is_err()); + } + + #[test] + fn sherpa_active_model_pref_falls_back_to_default_for_unknown_alias() { + let prefs = UserPreferences { + sherpa_onnx_model: "zipformer-zh-streaming".to_string(), + ..Default::default() + }; + + assert_eq!( + active_sherpa_model_from_prefs(&prefs), + crate::asr::local::sherpa::DEFAULT_MODEL_ALIAS + ); + } + + #[test] + fn credentials_status_accepts_keyless_custom_llm_only() { + let keyless_ready = CredentialsSnapshot { + ark_endpoint: Some("http://localhost:11434/v1".into()), + ark_model_id: Some("qwen".into()), + ..snapshot() + }; + assert!(llm_configured_for_provider("custom", &keyless_ready)); + assert!(llm_configured_for_provider("self-hosted", &keyless_ready)); + assert!(llm_configured_for_provider( + "openrouterFree", + &keyless_ready + )); + + let hosted_keyless = CredentialsSnapshot { + ark_endpoint: Some("https://openrouter.ai/api/v1".into()), + ark_model_id: Some("qwen/qwen3-coder:free".into()), + ..snapshot() + }; + assert!(!llm_configured_for_provider( + "openrouterFree", + &hosted_keyless + )); + + let hosted_ready = CredentialsSnapshot { + ark_api_key: Some("key".into()), + ark_endpoint: Some("https://openrouter.ai/api/v1/chat/completions".into()), + ark_model_id: Some("qwen/qwen3-coder:free".into()), + ..snapshot() + }; + assert!(llm_configured_for_provider("openrouterFree", &hosted_ready)); + + let key_without_endpoint = CredentialsSnapshot { + ark_api_key: Some("key".into()), + ark_model_id: Some("qwen".into()), + ..snapshot() + }; + assert!(!llm_configured_for_provider( + "custom", + &key_without_endpoint + )); + + let endpoint_without_model = CredentialsSnapshot { + ark_endpoint: Some("http://localhost:11434/v1".into()), + ..snapshot() + }; + assert!(!llm_configured_for_provider( + "custom", + &endpoint_without_model + )); + } + + impl SettingsWriter for FakeSettingsWriter { + fn read_settings(&self) -> UserPreferences { + self.saved.lock().unwrap().clone().unwrap_or_default() + } + + fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { + if let Some(error) = { + let mut errors = self.write_settings_errors.lock().unwrap(); + if errors.is_empty() { + None + } else { + errors.remove(0) + } + } { + return Err(error); + } + if let Some(error) = self.write_settings_error.lock().unwrap().clone() { + return Err(error); + } + *self.saved.lock().unwrap() = Some(prefs); + Ok(()) + } + + fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { + self.active_asr_provider_syncs + .lock() + .unwrap() + .push(provider.to_string()); + if let Some(error) = { + let mut errors = self.active_asr_provider_sync_errors.lock().unwrap(); + if errors.is_empty() { + None + } else { + errors.remove(0) + } + } { + return Err(error); + } + if let Some(error) = self.active_asr_provider_sync_error.lock().unwrap().clone() { + return Err(error); + } + Ok(()) + } + + fn refresh_dictation_hotkey(&self) { + *self.dictation_refreshes.lock().unwrap() += 1; + } + + fn refresh_qa_hotkey(&self) { + *self.qa_refreshes.lock().unwrap() += 1; + } + + fn refresh_combo_hotkey(&self) { + *self.combo_refreshes.lock().unwrap() += 1; + } + + fn refresh_translation_hotkey(&self) { + *self.translation_refreshes.lock().unwrap() += 1; + } + + fn refresh_switch_style_hotkey(&self) { + *self.switch_style_refreshes.lock().unwrap() += 1; + } + + fn refresh_open_app_hotkey(&self) { + *self.open_app_refreshes.lock().unwrap() += 1; + } + + fn refresh_coding_agent_hotkey(&self) { + *self.coding_agent_refreshes.lock().unwrap() += 1; + } + } + + #[test] + fn models_url_accepts_base_or_chat_endpoint() { + assert_eq!( + models_url("https://api.openai.com/v1"), + "https://api.openai.com/v1/models" + ); + assert_eq!( + models_url("https://api.openai.com/v1/chat/completions"), + "https://api.openai.com/v1/models" + ); + } + + #[test] + fn asr_transcriptions_url_accepts_base_or_transcriptions_endpoint() { + assert_eq!( + asr_transcriptions_url("https://api.openai.com/v1").unwrap(), + "https://api.openai.com/v1/audio/transcriptions" + ); + assert_eq!( + asr_transcriptions_url("https://api.openai.com/v1/chat/completions").unwrap(), + "https://api.openai.com/v1/audio/transcriptions" + ); + assert_eq!( + asr_transcriptions_url("https://api.openai.com/v1/audio").unwrap(), + "https://api.openai.com/v1/audio/transcriptions" + ); + assert_eq!( + asr_transcriptions_url("https://api.openai.com/v1/audio/transcriptions").unwrap(), + "https://api.openai.com/v1/audio/transcriptions" + ); + assert_eq!( + asr_transcriptions_url("https://api.openai.com/v1?api-version=2024-12-01").unwrap(), + "https://api.openai.com/v1/audio/transcriptions?api-version=2024-12-01" + ); + assert_eq!( + asr_transcriptions_url("http://192.168.1.10:8000/v1").unwrap(), + "http://192.168.1.10:8000/v1/audio/transcriptions" + ); + } + + #[test] + fn parse_model_ids_sorts_and_deduplicates() { + let models = + parse_model_ids(r#"{ "data": [{ "id": "b" }, { "id": "a" }, { "id": "b" }] }"#) + .unwrap(); + assert_eq!(models, vec!["a".to_string(), "b".to_string()]); + } + + #[test] + fn parse_gemini_model_ids_strips_models_prefix_and_dedups() { + // Google v1beta/models 真实响应的子集——name 字段带 `models/` 前缀， + // ProviderTools 选中后写入 ark.model_id 时不能带这个前缀（generateContent + // URL 拼接已经会加 `models/`，不去前缀就会变成 `models/models/...`）。 + // 字段缺失时保守保留（视为支持 generateContent）。 + let body = r#"{"models":[ + {"name":"models/gemini-2.5-pro"}, + {"name":"models/gemini-2.5-flash"}, + {"name":"models/gemini-2.5-flash"}, + {"name":"models/gemini-3-flash-preview"} + ]}"#; + let ids = parse_gemini_model_ids(body).unwrap(); + assert_eq!( + ids, + vec![ + "gemini-2.5-flash".to_string(), + "gemini-2.5-pro".to_string(), + "gemini-3-flash-preview".to_string(), + ] + ); + } + + #[test] + fn parse_gemini_model_ids_filters_out_non_generate_content_families() { + // 真实 Google v1beta/models 响应里同时有 generateContent / embedContent / + // generateMessage 等多种家族。用户选中 embedding/TTS/image 模型写入 + // ark.model_id → polish 必败。这里是 PR #398 pr_agent advisory 的回归用例： + // 只把 supportedGenerationMethods 里含 generateContent 的过滤出来。 + let body = r#"{"models":[ + {"name":"models/gemini-2.5-flash","supportedGenerationMethods":["generateContent","streamGenerateContent","countTokens"]}, + {"name":"models/gemini-embedding-2","supportedGenerationMethods":["embedContent"]}, + {"name":"models/text-embedding-004","supportedGenerationMethods":["embedContent","countTextTokens"]}, + {"name":"models/gemini-2.5-pro-preview-tts","supportedGenerationMethods":["generateContent"]}, + {"name":"models/gemini-2.5-flash-image","supportedGenerationMethods":["predict"]} + ]}"#; + let ids = parse_gemini_model_ids(body).unwrap(); + // 只剩两条声明 generateContent 的；embedding 与 image (predict-only) 必须被过滤。 + assert_eq!( + ids, + vec![ + "gemini-2.5-flash".to_string(), + "gemini-2.5-pro-preview-tts".to_string(), + ] + ); + } + + #[test] + fn is_gemini_base_url_matches_official_domain() { + assert!(is_gemini_base_url( + "https://generativelanguage.googleapis.com/v1beta" + )); + assert!(is_gemini_base_url( + "https://generativelanguage.googleapis.com/v1beta/" + )); + assert!(!is_gemini_base_url("https://api.openai.com/v1")); + assert!(!is_gemini_base_url( + "https://ark.cn-beijing.volces.com/api/v3" + )); + } + + #[test] + fn persist_settings_refreshes_changed_hotkey_pipelines() { + let writer = FakeSettingsWriter::default(); + let previous = UserPreferences::default(); + *writer.saved.lock().unwrap() = Some(previous); + let prefs = UserPreferences { + dictation_hotkey: ShortcutBinding { + primary: "D".to_string(), + modifiers: vec!["ctrl".to_string()], + }, + qa_hotkey: Some(ShortcutBinding { + primary: "Q".to_string(), + modifiers: vec!["ctrl".to_string(), "alt".to_string()], + }), + translation_hotkey: ShortcutBinding { + primary: "T".to_string(), + modifiers: vec!["ctrl".to_string(), "alt".to_string()], + }, + switch_style_hotkey: Some(ShortcutBinding { + primary: "S".to_string(), + modifiers: vec!["ctrl".to_string(), "alt".to_string()], + }), + open_app_hotkey: Some(ShortcutBinding { + primary: "O".to_string(), + modifiers: vec!["ctrl".to_string(), "alt".to_string()], + }), + coding_agent_voice_hotkey: Some(ShortcutBinding { + primary: "RightControl".to_string(), + modifiers: vec![], + }), + hotkey: HotkeyBinding { + trigger: HotkeyTrigger::Custom, + mode: HotkeyMode::Hold, + ..Default::default() + }, + ..Default::default() + }; + + persist_settings(&writer, prefs.clone()).unwrap(); + + let saved = writer + .saved + .lock() + .unwrap() + .clone() + .expect("settings saved"); + assert_eq!(saved.hotkey.trigger, HotkeyTrigger::Custom); + assert_eq!(saved.hotkey.mode, prefs.hotkey.mode); + assert_eq!( + saved.dictation_hotkey.primary, + prefs.dictation_hotkey.primary + ); + assert_eq!(saved.qa_hotkey.unwrap().primary, "Q"); + assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.combo_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.qa_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.translation_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 1); + assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 1); + } + + #[test] + fn persist_settings_rejects_less_computer_dictation_overlap() { + let writer = FakeSettingsWriter::default(); + let binding = ShortcutBinding { + primary: "LeftControl".into(), + modifiers: vec![], + }; + let prefs = UserPreferences { + dictation_hotkey: binding.clone(), + coding_agent_voice_hotkey: Some(binding), + ..Default::default() + }; + + assert_eq!( + persist_settings(&writer, prefs), + Err("Less Computer 快捷键不能和听写快捷键相同".into()) + ); + assert!(writer.saved.lock().unwrap().is_none()); + } + + #[test] + fn persist_settings_syncs_active_asr_provider_without_hotkey_refresh() { + let writer = FakeSettingsWriter::default(); + let previous = UserPreferences::default(); + *writer.saved.lock().unwrap() = Some(previous.clone()); + let prefs = UserPreferences { + active_asr_provider: "whisper".to_string(), + microphone_device_name: "External Mic".to_string(), + hotkey: previous.hotkey, + dictation_hotkey: previous.dictation_hotkey, + qa_hotkey: previous.qa_hotkey, + translation_hotkey: previous.translation_hotkey, + switch_style_hotkey: previous.switch_style_hotkey, + open_app_hotkey: previous.open_app_hotkey, + ..Default::default() + }; + + persist_settings(&writer, prefs.clone()).unwrap(); + + let saved = writer + .saved + .lock() + .unwrap() + .clone() + .expect("settings saved"); + assert_eq!(saved.active_asr_provider, prefs.active_asr_provider); + assert_eq!(saved.microphone_device_name, prefs.microphone_device_name); + assert_eq!( + writer.active_asr_provider_syncs.lock().unwrap().clone(), + vec![prefs.active_asr_provider.clone()] + ); + assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); + } + + #[test] + fn persist_settings_does_not_save_when_active_asr_sync_fails() { + let writer = FakeSettingsWriter::default(); + let previous = UserPreferences::default(); + *writer.saved.lock().unwrap() = Some(previous.clone()); + *writer.active_asr_provider_sync_error.lock().unwrap() = Some("sync failed".to_string()); + let prefs = UserPreferences { + active_asr_provider: "whisper".to_string(), + microphone_device_name: "External Mic".to_string(), + hotkey: previous.hotkey, + dictation_hotkey: previous.dictation_hotkey, + qa_hotkey: previous.qa_hotkey, + translation_hotkey: previous.translation_hotkey, + switch_style_hotkey: previous.switch_style_hotkey, + open_app_hotkey: previous.open_app_hotkey, + ..Default::default() + }; + + let error = persist_settings(&writer, prefs).unwrap_err(); + + assert_eq!(error, "sync failed"); + let saved = writer + .saved + .lock() + .unwrap() + .clone() + .expect("previous settings remain saved"); + assert_eq!(saved.active_asr_provider, previous.active_asr_provider); + assert_eq!( + saved.microphone_device_name, + previous.microphone_device_name + ); + assert_eq!( + writer.active_asr_provider_syncs.lock().unwrap().clone(), + vec!["whisper".to_string()] + ); + assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); + } + + #[test] + fn persist_settings_restores_active_asr_provider_when_save_fails_after_sync() { + let writer = FakeSettingsWriter::default(); + let previous = UserPreferences::default(); + *writer.saved.lock().unwrap() = Some(previous.clone()); + *writer.write_settings_error.lock().unwrap() = Some("save failed".to_string()); + let prefs = UserPreferences { + active_asr_provider: "whisper".to_string(), + microphone_device_name: "External Mic".to_string(), + hotkey: previous.hotkey, + dictation_hotkey: previous.dictation_hotkey, + qa_hotkey: previous.qa_hotkey, + translation_hotkey: previous.translation_hotkey, + switch_style_hotkey: previous.switch_style_hotkey, + open_app_hotkey: previous.open_app_hotkey, + ..Default::default() + }; + + let error = persist_settings(&writer, prefs).unwrap_err(); + + assert_eq!(error, "save failed"); + let saved = writer + .saved + .lock() + .unwrap() + .clone() + .expect("previous settings remain saved"); + assert_eq!(saved.active_asr_provider, previous.active_asr_provider); + assert_eq!( + saved.microphone_device_name, + previous.microphone_device_name + ); + assert_eq!( + writer.active_asr_provider_syncs.lock().unwrap().clone(), + vec!["whisper".to_string(), previous.active_asr_provider.clone()] + ); + assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); + } + + #[test] + fn persist_settings_keeps_new_active_asr_provider_when_rollback_fails() { + let writer = FakeSettingsWriter::default(); + let previous = UserPreferences::default(); + *writer.saved.lock().unwrap() = Some(previous.clone()); + *writer.write_settings_errors.lock().unwrap() = vec![Some("save failed".to_string()), None]; + *writer.active_asr_provider_sync_errors.lock().unwrap() = + vec![None, Some("rollback failed".to_string())]; + let prefs = UserPreferences { + active_asr_provider: "whisper".to_string(), + microphone_device_name: "External Mic".to_string(), + hotkey: previous.hotkey, + dictation_hotkey: previous.dictation_hotkey, + qa_hotkey: previous.qa_hotkey, + translation_hotkey: previous.translation_hotkey, + switch_style_hotkey: previous.switch_style_hotkey, + open_app_hotkey: previous.open_app_hotkey, + ..Default::default() + }; + + persist_settings(&writer, prefs.clone()).expect("settings remain consistent"); + + let saved = writer + .saved + .lock() + .unwrap() + .clone() + .expect("new settings saved"); + assert_eq!(saved.active_asr_provider, prefs.active_asr_provider); + assert_eq!(saved.microphone_device_name, prefs.microphone_device_name); + assert_eq!( + writer.active_asr_provider_syncs.lock().unwrap().clone(), + vec!["whisper".to_string(), previous.active_asr_provider.clone()] + ); + assert_eq!(*writer.dictation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.combo_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.qa_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.translation_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.switch_style_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.open_app_refreshes.lock().unwrap(), 0); + assert_eq!(*writer.coding_agent_refreshes.lock().unwrap(), 0); + } + + #[test] + fn sync_dictation_hotkey_sets_modifier_trigger_and_clears_combo() { + let mut prefs = UserPreferences { + hotkey: HotkeyBinding { + trigger: HotkeyTrigger::Custom, + mode: HotkeyMode::Toggle, + keys: None, + }, + custom_combo_hotkey: Some(ComboBinding { + primary: "D".into(), + modifiers: vec!["cmd".into(), "shift".into()], + }), + dictation_hotkey: ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }, + ..Default::default() + }; + + super::sync_dictation_hotkey_legacy_fields(&mut prefs); + + assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::RightControl); + assert!(prefs.custom_combo_hotkey.is_none()); + } + + #[test] + fn sync_dictation_hotkey_sets_custom_trigger_and_combo_binding() { + let mut prefs = UserPreferences { + hotkey: HotkeyBinding { + trigger: HotkeyTrigger::RightControl, + mode: HotkeyMode::Toggle, + keys: None, + }, + dictation_hotkey: ShortcutBinding { + primary: "D".into(), + modifiers: vec!["cmd".into(), "shift".into()], + }, + ..Default::default() + }; + + super::sync_dictation_hotkey_legacy_fields(&mut prefs); + + assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::Custom); + let combo = prefs.custom_combo_hotkey.expect("combo binding saved"); + assert_eq!(combo.primary, "D"); + assert_eq!( + combo.modifiers, + vec!["cmd".to_string(), "shift".to_string()] + ); + } + + #[test] + fn sync_dictation_hotkey_clears_empty_custom_binding() { + let mut prefs = UserPreferences { + hotkey: HotkeyBinding { + trigger: HotkeyTrigger::RightControl, + mode: HotkeyMode::Toggle, + keys: None, + }, + custom_combo_hotkey: Some(ComboBinding { + primary: "D".into(), + modifiers: vec!["cmd".into(), "shift".into()], + }), + dictation_hotkey: ShortcutBinding { + primary: " ".into(), + modifiers: vec!["cmd".into()], + }, + ..Default::default() + }; + + super::sync_dictation_hotkey_legacy_fields(&mut prefs); + + assert_eq!(prefs.hotkey.trigger, HotkeyTrigger::Custom); + assert!(prefs.custom_combo_hotkey.is_none()); + } + + #[test] + fn validate_combo_hotkey_rejects_bare_shift() { + let result = super::validate_combo_hotkey(ComboBinding { + primary: "Shift".into(), + modifiers: vec![], + }); + + assert!(result.is_err()); + } + + #[test] + fn combo_hotkey_bare_shift_rejection_matches_dictation_setter() { + let binding = ShortcutBinding { + primary: "Shift".into(), + modifiers: vec![], + }; + + assert_eq!( + super::reject_bare_shift_dictation_shortcut(&binding), + Err("Shift 单键目前只能用于翻译快捷键".into()) + ); + } + + #[test] + fn dictation_qa_overlap_rejects_same_modifier_only_binding() { + let binding = ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }; + + assert_eq!( + super::reject_dictation_qa_hotkey_overlap(&binding, &binding), + Err("QA 快捷键不能和听写快捷键相同".into()) + ); + } + + #[test] + fn dictation_qa_overlap_rejects_same_combo_binding() { + let dictation = ShortcutBinding { + primary: ";".into(), + modifiers: vec!["ctrl".into(), "shift".into()], + }; + let qa = ShortcutBinding { + primary: ";".into(), + modifiers: vec!["control".into(), "shift".into()], + }; + + assert_eq!( + super::reject_dictation_qa_hotkey_overlap(&dictation, &qa), + Err("QA 快捷键不能和听写快捷键相同".into()) + ); + } + + #[test] + fn dictation_qa_overlap_allows_distinct_bindings() { + let dictation = ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }; + let qa = ShortcutBinding { + primary: ";".into(), + modifiers: vec!["ctrl".into(), "shift".into()], + }; + + assert!(super::reject_dictation_qa_hotkey_overlap(&dictation, &qa).is_ok()); + } + + #[test] + fn dictation_translation_overlap_rejects_same_modifier_only_binding() { + let binding = ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }; + + assert_eq!( + super::reject_dictation_translation_hotkey_overlap(&binding, &binding), + Err("翻译快捷键不能和听写快捷键相同".into()) + ); + } + + #[test] + fn dictation_translation_overlap_rejects_same_combo_binding() { + let dictation = ShortcutBinding { + primary: "T".into(), + modifiers: vec!["ctrl".into(), "shift".into()], + }; + let translation = ShortcutBinding { + primary: "T".into(), + modifiers: vec!["control".into(), "shift".into()], + }; + + assert_eq!( + super::reject_dictation_translation_hotkey_overlap(&dictation, &translation), + Err("翻译快捷键不能和听写快捷键相同".into()) + ); + } + + #[test] + fn dictation_translation_overlap_allows_distinct_bindings() { + let dictation = ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }; + let translation = ShortcutBinding { + primary: "Shift".into(), + modifiers: vec![], + }; + + assert!( + super::reject_dictation_translation_hotkey_overlap(&dictation, &translation).is_ok() + ); + } + + #[test] + fn persist_settings_rejects_dictation_translation_overlap() { + let writer = FakeSettingsWriter::default(); + let binding = ShortcutBinding { + primary: "RightControl".into(), + modifiers: vec![], + }; + let prefs = UserPreferences { + dictation_hotkey: binding.clone(), + translation_hotkey: binding, + ..Default::default() + }; + + assert_eq!( + persist_settings(&writer, prefs), + Err("翻译快捷键不能和听写快捷键相同".into()) + ); + assert!(writer.saved.lock().unwrap().is_none()); + } + + #[test] + fn persist_settings_rejects_translation_switch_style_overlap() { + let writer = FakeSettingsWriter::default(); + let binding = ShortcutBinding { + primary: "T".into(), + modifiers: vec!["cmd".into(), "shift".into()], + }; + let prefs = UserPreferences { + translation_hotkey: binding.clone(), + switch_style_hotkey: Some(binding), + ..Default::default() + }; + + assert_eq!( + persist_settings(&writer, prefs), + Err("切换风格快捷键不能和翻译快捷键相同".into()) + ); + assert!(writer.saved.lock().unwrap().is_none()); + } + + #[test] + fn persist_settings_rejects_switch_style_open_app_overlap() { + let writer = FakeSettingsWriter::default(); + let binding = ShortcutBinding { + primary: "K".into(), + modifiers: vec!["cmd".into(), "shift".into()], + }; + let prefs = UserPreferences { + switch_style_hotkey: Some(binding.clone()), + open_app_hotkey: Some(binding), + ..Default::default() + }; + + assert_eq!( + persist_settings(&writer, prefs), + Err("打开应用快捷键不能和切换风格快捷键相同".into()) + ); + assert!(writer.saved.lock().unwrap().is_none()); + } + + #[test] + fn parse_latest_beta_from_atom_picks_first_beta_tagged_entry() { + // Fixture trimmed from real `releases.atom`：包含一条 stable + 一条 Beta。 + // 解析必须跳过 stable（tag 不以 -beta-tauri 结尾），返回 Beta。 + let body = r#" + + + tag:github.com,2008:Repository/X/v1.2.23-tauri + 2026-05-07T09:05:00Z + + OpenLess v1.2.23-tauri + + + tag:github.com,2008:Repository/X/v1.2.24-2-beta-tauri + 2026-05-08T01:27:23Z + + OpenLess v1.2.24-2-beta-tauri + +"#; + let got = parse_latest_beta_from_atom(body).expect("must find a Beta entry"); + assert_eq!(got.tag_name, "v1.2.24-2-beta-tauri"); + assert_eq!( + got.html_url, + "https://github.com/appergb/openless/releases/tag/v1.2.24-2-beta-tauri" + ); + assert_eq!(got.published_at, "2026-05-08T01:27:23Z"); + } + + #[test] + fn parse_latest_beta_from_atom_returns_none_when_only_stable_releases() { + let body = r#" + + + 2026-05-07T09:05:00Z + +"#; + assert!(parse_latest_beta_from_atom(body).is_none()); + } + + #[tokio::test] + async fn fetch_provider_models_omits_authorization_when_api_key_is_empty() { + let listener = TcpListener::bind("127.0.0.1:0").unwrap(); + let addr = listener.local_addr().unwrap(); + + let server = thread::spawn(move || { + let (mut stream, _) = listener.accept().unwrap(); + let mut buf = [0u8; 8192]; + let mut request = Vec::new(); + loop { + let n = stream.read(&mut buf).unwrap(); + if n == 0 { + break; + } + request.extend_from_slice(&buf[..n]); + if request.windows(4).any(|w| w == b"\r\n\r\n") { + break; + } + } + let request_text = String::from_utf8_lossy(&request); + assert!(!request_text.contains("Authorization: Bearer")); + + let body = r#"{"data":[{"id":"m1"},{"id":"m2"}]}"#; + let response = format!( + "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\nContent-Length: {}\r\nConnection: close\r\n\r\n{}", + body.len(), + body + ); + stream.write_all(response.as_bytes()).unwrap(); + }); + + let models = fetch_provider_models(&ProviderConfig { + base_url: format!("http://{}", addr), + api_key: String::new(), + }) + .await + .unwrap(); + + assert_eq!(models, vec!["m1".to_string(), "m2".to_string()]); + server.join().unwrap(); + } + + #[test] + fn is_valid_session_id_accepts_canonical_uuid_v4() { + // canonical UUID-v4 字面：8-4-4-4-12，全小写、全大写、混合都接受。 + assert!(is_valid_session_id("550e8400-e29b-41d4-a716-446655440000")); + assert!(is_valid_session_id("550E8400-E29B-41D4-A716-446655440000")); + assert!(is_valid_session_id("Abc12345-6789-abcd-EF01-234567890abc")); + } + + #[test] + fn is_valid_session_id_rejects_path_traversal_and_garbage() { + assert!(!is_valid_session_id("")); + assert!(!is_valid_session_id("../../etc/passwd")); + assert!(!is_valid_session_id("..\\..\\windows\\system32")); + // 长度对但含 `/`：dash 位置错或非 hex 字符都不通过 + assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544/000")); + assert!(!is_valid_session_id("550e8400_e29b_41d4_a716_446655440000")); // 用 _ 代 - + // 非 hex 字符 + assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544000g")); + // 长度不对（35 / 37） + assert!(!is_valid_session_id("550e8400-e29b-41d4-a716-44665544000")); + assert!(!is_valid_session_id( + "550e8400-e29b-41d4-a716-4466554400000" + )); + // NUL 字节 + assert!(!is_valid_session_id( + "550e8400-e29b-41d4-a716-44665544\x00000" + )); + // 百分号编码与绝对路径 + assert!(!is_valid_session_id("%2e%2e/recordings/x")); + assert!(!is_valid_session_id("/Users/attacker/secret.wav")); + } + + #[test] + fn is_valid_local_pack_id_accepts_realistic_ids() { + assert!(is_valid_local_pack_id("builtin.light")); + assert!(is_valid_local_pack_id("builtin.structured")); + assert!(is_valid_local_pack_id("custom.meeting")); + assert!(is_valid_local_pack_id( + "550e8400-e29b-41d4-a716-446655440000" + )); + assert!(is_valid_local_pack_id("my_pack_v2")); + assert!(is_valid_local_pack_id("Pack-2026.05")); + } + + #[test] + fn is_valid_local_pack_id_rejects_path_traversal() { + assert!(!is_valid_local_pack_id("")); + assert!(!is_valid_local_pack_id("../etc/passwd")); + assert!(!is_valid_local_pack_id("..\\windows\\system32")); + assert!(!is_valid_local_pack_id("pack/../../etc")); + assert!(!is_valid_local_pack_id("/abs/path")); + assert!(!is_valid_local_pack_id("with space")); + assert!(!is_valid_local_pack_id("with\x00null")); + assert!(!is_valid_local_pack_id(&"a".repeat(129))); + } +} diff --git a/openless-all/app/src-tauri/src/commands/permissions_cmds.rs b/openless-all/app/src-tauri/src/commands/permissions_cmds.rs new file mode 100644 index 00000000..d9351636 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/permissions_cmds.rs @@ -0,0 +1,100 @@ +use super::*; + +#[tauri::command] +pub fn check_accessibility_permission() -> PermissionStatus { + permissions::check_accessibility() +} + +#[tauri::command] +pub fn request_accessibility_permission() -> PermissionStatus { + permissions::request_accessibility() +} + +#[tauri::command] +pub fn check_microphone_permission() -> PermissionStatus { + permissions::check_microphone() +} + +#[tauri::command] +pub fn request_microphone_permission(app: AppHandle) -> PermissionStatus { + crate::request_microphone_from_foreground(&app) +} + +/// 跳到 macOS 系统设置的指定隐私面板。pane: "accessibility" | "microphone". +#[tauri::command] +pub fn open_system_settings(pane: String) -> Result<(), String> { + #[cfg(target_os = "macos")] + { + let url = match pane.as_str() { + "accessibility" => { + "x-apple.systempreferences:com.apple.preference.security?Privacy_Accessibility" + } + "microphone" => { + "x-apple.systempreferences:com.apple.preference.security?Privacy_Microphone" + } + _ => "x-apple.systempreferences:com.apple.preference.security?Privacy", + }; + std::process::Command::new("open") + .arg(url) + .spawn() + .map(|_| ()) + .map_err(|e| e.to_string()) + } + #[cfg(target_os = "windows")] + { + use windows::core::PCWSTR; + use windows::Win32::UI::Shell::ShellExecuteW; + use windows::Win32::UI::WindowsAndMessaging::SW_SHOWNORMAL; + + fn wide_null(value: &str) -> Vec { + value.encode_utf16().chain(std::iter::once(0)).collect() + } + + let uri = match pane.as_str() { + "microphone" => "ms-settings:privacy-microphone", + "sound" => "ms-settings:sound", + "accessibility" => "ms-settings:easeofaccess", + _ => "ms-settings:", + }; + + let operation = wide_null("open"); + let target = wide_null(uri); + let result = unsafe { + ShellExecuteW( + None, + PCWSTR(operation.as_ptr()), + PCWSTR(target.as_ptr()), + PCWSTR::null(), + PCWSTR::null(), + SW_SHOWNORMAL, + ) + }; + + if result.0 as isize <= 32 { + Err(format!("ShellExecuteW failed: {}", result.0 as isize)) + } else { + Ok(()) + } + } + #[cfg(all(not(target_os = "macos"), not(target_os = "windows")))] + { + let _ = pane; + Err("open_system_settings is only supported on macOS and Windows".to_string()) + } +} + +/// 触发 macOS 系统弹"是否允许 OpenLess 访问麦克风"对话框。 +/// 与 Swift `MicrophonePermission.request()` 同语义：只信系统权限回调， +/// 不用 cpal stream 成功与否伪造授权状态。 +#[tauri::command] +pub fn trigger_microphone_prompt(app: AppHandle) -> Result<(), String> { + let status = crate::request_microphone_from_foreground(&app); + if matches!( + status, + PermissionStatus::Granted | PermissionStatus::NotApplicable + ) { + Ok(()) + } else { + Err(format!("microphone permission is {status:?}")) + } +} diff --git a/openless-all/app/src-tauri/src/commands/providers.rs b/openless-all/app/src-tauri/src/commands/providers.rs new file mode 100644 index 00000000..d7f03e19 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/providers.rs @@ -0,0 +1,496 @@ +use super::*; + +#[derive(Serialize)] +#[serde(rename_all = "camelCase")] +pub struct ProviderCheckResult { + ok: bool, +} + +#[derive(Serialize)] +pub struct ProviderModelsResult { + models: Vec, +} + +#[tauri::command] +pub async fn validate_provider_credentials(kind: String) -> Result { + match kind.as_str() { + "llm" => validate_llm_provider() + .await + .map(|()| ProviderCheckResult { ok: true }), + "asr" => validate_asr_provider() + .await + .map(|()| ProviderCheckResult { ok: true }), + _ => Err(format!("unknown provider kind: {kind}")), + } +} + +#[tauri::command] +pub async fn list_provider_models(kind: String) -> Result { + if kind == "asr" && CredentialsVault::get_active_asr() == crate::asr::bailian::PROVIDER_ID { + return Ok(ProviderModelsResult { + models: vec![crate::asr::bailian::DEFAULT_MODEL.to_string()], + }); + } + if kind == "asr" && CredentialsVault::get_active_asr() == crate::asr::mimo::PROVIDER_ID { + return Ok(ProviderModelsResult { + models: vec![crate::asr::mimo::DEFAULT_MODEL.to_string()], + }); + } + if kind == "llm" && CredentialsVault::get_active_llm() == CODEX_OAUTH_PROVIDER_ID { + return Ok(ProviderModelsResult { + models: vec![ + CODEX_DEFAULT_MODEL.to_string(), + "gpt-5.3-codex".to_string(), + "gpt-5.4".to_string(), + "gpt-5.5".to_string(), + ], + }); + } + let config = read_openai_provider_config(&kind)?; + fetch_provider_models(&config) + .await + .map(|models| ProviderModelsResult { models }) +} + +pub(crate) struct ProviderConfig { + pub(crate) base_url: String, + pub(crate) api_key: String, +} + +fn read_openai_provider_config(kind: &str) -> Result { + let (api_key_account, endpoint_account, api_key_required) = match kind { + "llm" => ( + CredentialAccount::ArkApiKey, + CredentialAccount::ArkEndpoint, + false, + ), + "asr" => ( + CredentialAccount::AsrApiKey, + CredentialAccount::AsrEndpoint, + true, + ), + _ => return Err(format!("unknown provider kind: {kind}")), + }; + let api_key = CredentialsVault::get(api_key_account) + .map_err(|e| e.to_string())? + .unwrap_or_default(); + let base_url = CredentialsVault::get(endpoint_account) + .map_err(|e| e.to_string())? + .unwrap_or_default(); + if api_key_required && api_key.trim().is_empty() { + return Err("API Key 为空".to_string()); + } + if base_url.trim().is_empty() { + return Err("Endpoint 为空".to_string()); + } + Ok(ProviderConfig { base_url, api_key }) +} + +async fn validate_llm_provider() -> Result<(), String> { + let llm_thinking_enabled = PreferencesStore::new() + .map_err(|e| e.to_string())? + .get() + .llm_thinking_enabled; + if CredentialsVault::get_active_llm() == CODEX_OAUTH_PROVIDER_ID { + let model = CredentialsVault::get(CredentialAccount::ArkModelId) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()) + .unwrap_or_else(|| CODEX_DEFAULT_MODEL.to_string()); + let provider = CodexOAuthLLMProvider::new( + CodexOAuthConfig::new(model).with_thinking_enabled(llm_thinking_enabled), + ); + return provider + .polish( + "验证连接", + PolishMode::Raw, + &[], + "", + &[], + ChineseScriptPreference::Auto, + OutputLanguagePreference::Auto, + None, + &[], + ) + .await + .map(|_| ()) + .map_err(|e| match e { + LLMError::InvalidResponse { status, .. } => { + format!("providerHttpStatus:{status}") + } + other => other.to_string(), + }); + } + + let config = read_openai_provider_config("llm")?; + let active_llm = CredentialsVault::get_active_llm(); + let model = CredentialsVault::get(CredentialAccount::ArkModelId) + .map_err(|e| e.to_string())? + .filter(|s| !s.is_empty()) + .ok_or_else(|| "llmModelMissing".to_string())?; + let provider = OpenAICompatibleLLMProvider::new( + OpenAICompatibleConfig::new( + active_llm.clone(), + active_llm, + config.base_url, + config.api_key, + model, + ) + .with_thinking_enabled(llm_thinking_enabled), + ); + provider + .polish( + "验证连接", + PolishMode::Raw, + &[], + "", + &[], + ChineseScriptPreference::Auto, + OutputLanguagePreference::Auto, + None, + &[], + ) + .await + .map(|_| ()) + .map_err(|e| match e { + LLMError::InvalidResponse { status, .. } => { + format!("providerHttpStatus:{status}") + } + other => other.to_string(), + }) +} + +async fn validate_asr_provider() -> Result<(), String> { + let active_asr = CredentialsVault::get_active_asr(); + if active_asr_is_keyless_for_validation(&active_asr) { + return Ok(()); + } + + if active_asr == crate::asr::bailian::PROVIDER_ID { + return validate_bailian_asr_provider().await; + } + if active_asr == crate::asr::mimo::PROVIDER_ID { + return validate_mimo_asr_provider().await; + } + + let config = read_openai_provider_config("asr")?; + let model = CredentialsVault::get(CredentialAccount::AsrModel) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()) + .ok_or_else(|| "asrModelMissing".to_string())?; + validate_asr_transcription(&config, model.trim()).await +} + +async fn validate_mimo_asr_provider() -> Result<(), String> { + let config = read_openai_provider_config("asr")?; + let model = CredentialsVault::get(CredentialAccount::AsrModel) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()) + .unwrap_or_else(|| crate::asr::mimo::DEFAULT_MODEL.to_string()); + let asr = crate::asr::MimoBatchASR::new(config.api_key, config.base_url, model); + crate::recorder::AudioConsumer::consume_pcm_chunk( + &asr, + &encode_wav_16k_mono_silence(250)[44..], + ); + asr.transcribe() + .await + .map(|_| ()) + .map_err(|e| e.to_string()) +} + +async fn validate_bailian_asr_provider() -> Result<(), String> { + let api_key = CredentialsVault::get(CredentialAccount::AsrApiKey) + .map_err(|e| e.to_string())? + .unwrap_or_default(); + if api_key.trim().is_empty() { + return Err("API Key 为空".to_string()); + } + let endpoint = CredentialsVault::get(CredentialAccount::AsrEndpoint) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()) + .unwrap_or_else(|| crate::asr::bailian::DEFAULT_ENDPOINT.to_string()); + let model = CredentialsVault::get(CredentialAccount::AsrModel) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()) + .unwrap_or_else(|| crate::asr::bailian::DEFAULT_MODEL.to_string()); + let vocabulary_id = CredentialsVault::get(CredentialAccount::AsrVocabularyId) + .map_err(|e| e.to_string())? + .filter(|s| !s.trim().is_empty()); + let asr = std::sync::Arc::new(crate::asr::BailianRealtimeASR::new( + crate::asr::BailianCredentials { + api_key, + endpoint, + model, + vocabulary_id, + }, + )); + asr.open_session().await.map_err(|e| e.to_string())?; + crate::asr::AudioConsumer::consume_pcm_chunk( + &*asr, + &vec![0u8; crate::asr::bailian::TARGET_AUDIO_CHUNK_BYTES], + ); + asr.send_last_frame().await.map_err(|e| e.to_string())?; + asr.await_final_result() + .await + .map(|_| ()) + .map_err(|e| e.to_string()) +} + +pub(crate) fn active_asr_is_keyless_for_validation(provider: &str) -> bool { + provider == crate::asr::local::PROVIDER_ID + || active_foundry_asr_is_supported(provider) + || active_sherpa_asr_is_supported(provider) +} + +pub(crate) fn active_foundry_asr_is_supported(provider: &str) -> bool { + #[cfg(target_os = "windows")] + { + provider == FOUNDRY_LOCAL_PROVIDER_ID + } + #[cfg(not(target_os = "windows"))] + { + let _ = provider; + false + } +} + +pub(crate) fn active_sherpa_asr_is_supported(provider: &str) -> bool { + #[cfg(target_os = "windows")] + { + provider == crate::asr::local::sherpa::PROVIDER_ID + } + #[cfg(not(target_os = "windows"))] + { + let _ = provider; + false + } +} + +async fn validate_asr_transcription(config: &ProviderConfig, model: &str) -> Result<(), String> { + const MAX_ASR_VALIDATE_BODY_BYTES: usize = 1024 * 1024; + const MAX_ATTEMPTS: u32 = 6; + let url = asr_transcriptions_url(&config.base_url)?; + let wav = encode_wav_16k_mono_silence(250); + let client = http_client_builder(&url, 20) + .build() + .map_err(|_| "providerClientInitFailed".to_string())?; + // 连接 / 请求未送出类失败做指数退避重试 —— 这类失败请求尚未送达服务端，重试 + // 安全。超时不重试（服务端可能已在处理）。multipart 是流式 body，每次重建。 + let mut attempt: u32 = 0; + let response = loop { + attempt += 1; + let wav_part = reqwest::multipart::Part::bytes(wav.clone()) + .file_name("openless-asr-check.wav") + .mime_str("audio/wav") + .map_err(|e| format!("请求体构建失败: {e}"))?; + let form = reqwest::multipart::Form::new() + .part("file", wav_part) + .text("model", model.to_string()); + match client + .post(&url) + .header("Authorization", format!("Bearer {}", config.api_key)) + .multipart(form) + .send() + .await + { + Ok(resp) => break resp, + Err(e) if e.is_timeout() => return Err("providerRequestTimeout".to_string()), + Err(e) if (e.is_connect() || e.is_request()) && attempt < MAX_ATTEMPTS => { + let backoff = (200u64 * 2u64.pow((attempt - 1).min(3))).min(900); + tokio::time::sleep(std::time::Duration::from_millis(backoff)).await; + continue; + } + Err(_) => return Err("providerNetworkError".to_string()), + } + }; + let status = response.status(); + if !status.is_success() { + return Err(format!("providerHttpStatus:{}", status.as_u16())); + } + if let Some(len) = response.content_length() { + if len as usize > MAX_ASR_VALIDATE_BODY_BYTES { + return Err("providerResponseTooLarge".to_string()); + } + } + use futures_util::StreamExt; + let mut body = Vec::::new(); + let mut stream = response.bytes_stream(); + while let Some(chunk) = stream.next().await { + let chunk = chunk.map_err(|_| "providerReadResponseFailed".to_string())?; + if body.len().saturating_add(chunk.len()) > MAX_ASR_VALIDATE_BODY_BYTES { + return Err("providerResponseTooLarge".to_string()); + } + body.extend_from_slice(&chunk); + } + let json: Value = serde_json::from_slice(&body).map_err(|_| "asrInvalidJson".to_string())?; + if !json.is_object() || json.get("text").is_none() { + return Err("asrMissingTextField".to_string()); + } + Ok(()) +} + +pub(crate) fn asr_transcriptions_url(base_url: &str) -> Result { + let parsed = reqwest::Url::parse(base_url.trim()).map_err(|_| "endpointInvalid".to_string())?; + + // Work on the URL path only so we don't corrupt query parameters. + let mut url = parsed.clone(); + let path = parsed.path().trim_end_matches('/'); + let next_path = if path.ends_with("/audio/transcriptions") { + path.to_string() + } else if path.ends_with("/audio") { + format!("{path}/transcriptions") + } else if let Some(prefix) = path.strip_suffix("/chat/completions") { + format!("{prefix}/audio/transcriptions") + } else { + format!("{path}/audio/transcriptions") + }; + url.set_path(&next_path); + Ok(url.to_string()) +} + +fn encode_wav_16k_mono_silence(duration_ms: u32) -> Vec { + let sample_rate: u32 = 16_000; + let num_channels: u16 = 1; + let bits_per_sample: u16 = 16; + let bytes_per_sample = (bits_per_sample / 8) as usize; + let samples = (sample_rate as usize * duration_ms as usize) / 1000; + let pcm_len = samples * bytes_per_sample; + let data_size = pcm_len as u32; + let byte_rate = sample_rate * num_channels as u32 * bits_per_sample as u32 / 8; + let block_align = num_channels * bits_per_sample / 8; + let chunk_size = 36 + data_size; + + let mut wav = Vec::with_capacity(44 + pcm_len); + wav.extend_from_slice(b"RIFF"); + wav.extend_from_slice(&chunk_size.to_le_bytes()); + wav.extend_from_slice(b"WAVE"); + wav.extend_from_slice(b"fmt "); + wav.extend_from_slice(&16u32.to_le_bytes()); + wav.extend_from_slice(&1u16.to_le_bytes()); + wav.extend_from_slice(&num_channels.to_le_bytes()); + wav.extend_from_slice(&sample_rate.to_le_bytes()); + wav.extend_from_slice(&byte_rate.to_le_bytes()); + wav.extend_from_slice(&block_align.to_le_bytes()); + wav.extend_from_slice(&bits_per_sample.to_le_bytes()); + wav.extend_from_slice(b"data"); + wav.extend_from_slice(&data_size.to_le_bytes()); + wav.resize(44 + pcm_len, 0); + wav +} + +pub(crate) async fn fetch_provider_models(config: &ProviderConfig) -> Result, String> { + let url = models_url(&config.base_url); + let is_gemini = is_gemini_base_url(&config.base_url); + log::info!("[provider-check] GET {url} (gemini={is_gemini})"); + let client = http_client_builder(&config.base_url, 15) + .build() + .map_err(|e| format!("HTTP client 初始化失败: {e}"))?; + let mut request = client.get(&url); + if !config.api_key.trim().is_empty() { + // 谷歌原生 generativelanguage.googleapis.com 不识别 Bearer Authorization, + // 必须用 x-goog-api-key 头。其它 OpenAI 兼容 provider 仍走 Bearer。 + if is_gemini { + request = request.header("x-goog-api-key", config.api_key.as_str()); + } else { + request = request.header("Authorization", format!("Bearer {}", config.api_key)); + } + } + let response = request.send().await.map_err(|e| { + if e.is_timeout() { + "请求超时".to_string() + } else { + format!("网络错误: {e}") + } + })?; + let status = response.status(); + let body = response + .text() + .await + .map_err(|e| format!("读取响应失败: {e}"))?; + if !status.is_success() { + return Err(format!("providerHttpStatus:{}", status.as_u16())); + } + if is_gemini { + parse_gemini_model_ids(&body) + } else { + parse_model_ids(&body) + } +} + +pub(crate) fn is_gemini_base_url(base_url: &str) -> bool { + base_url.contains("generativelanguage.googleapis.com") +} + +pub(crate) fn models_url(base_url: &str) -> String { + let trimmed = base_url.trim().trim_end_matches('/'); + if trimmed.ends_with("/models") { + return trimmed.to_string(); + } + if let Some(prefix) = trimmed.strip_suffix("/chat/completions") { + return format!("{prefix}/models"); + } + format!("{trimmed}/models") +} + +pub(crate) fn parse_model_ids(body: &str) -> Result, String> { + let json: Value = + serde_json::from_str(body).map_err(|e| format!("模型列表不是有效 JSON: {e}"))?; + let data = json + .get("data") + .and_then(|v| v.as_array()) + .ok_or_else(|| "模型列表缺少 data 数组".to_string())?; + let mut models = data + .iter() + .filter_map(|item| item.get("id").and_then(|id| id.as_str())) + .map(str::trim) + .filter(|id| !id.is_empty()) + .map(ToOwned::to_owned) + .collect::>(); + models.sort(); + models.dedup(); + Ok(models) +} + +/// 谷歌 v1beta/models 响应形状：`{models: [{name: "models/gemini-2.5-flash", +/// supportedGenerationMethods: ["generateContent", ...], ...}, ...]}`。 +/// 与 OpenAI `{data: [{id: "..."}]}` 不兼容，所以单独解析；name 字段去掉 +/// "models/" 前缀后即是 ProviderTools「拉取模型」按钮可直接写入 ark.model_id +/// 的字符串。 +/// +/// 过滤：只保留声明支持 `generateContent` 的模型——Google 的 model list 同时 +/// 暴露 embedding (`gemini-embedding-2`)、TTS、image 等不支持 +/// generateContent 的家族；用户选中那种 ID 后 polish 必失败（PR #398 pr_agent +/// 漏洞反馈）。`supportedGenerationMethods` 字段缺失时保守保留——某些 preview +/// 模型可能未暴露这个字段，宁误显示也不要把新模型挡在外面。 +pub(crate) fn parse_gemini_model_ids(body: &str) -> Result, String> { + let json: Value = + serde_json::from_str(body).map_err(|e| format!("模型列表不是有效 JSON: {e}"))?; + let models = json + .get("models") + .and_then(|v| v.as_array()) + .ok_or_else(|| "Gemini 模型列表缺少 models 数组".to_string())?; + let mut ids = models + .iter() + .filter(|item| { + match item + .get("supportedGenerationMethods") + .and_then(|v| v.as_array()) + { + Some(methods) => methods + .iter() + .any(|m| m.as_str() == Some("generateContent")), + None => true, // 字段缺失：保守包含 + } + }) + .filter_map(|item| item.get("name").and_then(|n| n.as_str())) + .map(|name| { + name.strip_prefix("models/") + .unwrap_or(name) + .trim() + .to_string() + }) + .filter(|id| !id.is_empty()) + .collect::>(); + ids.sort(); + ids.dedup(); + Ok(ids) +} diff --git a/openless-all/app/src-tauri/src/commands/qa.rs b/openless-all/app/src-tauri/src/commands/qa.rs new file mode 100644 index 00000000..8d7a626b --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/qa.rs @@ -0,0 +1,67 @@ +use super::*; + +#[tauri::command] +pub fn get_qa_hotkey_label(coord: CoordinatorState<'_>) -> String { + coord.qa_hotkey_label() +} + +/// 设置 QA 快捷键并热更新 monitor。 +/// 传入 `None` 形式的字段不在这里支持——前端用 `binding == null` 时调下面的 +/// "disable" 写法（写 prefs.qa_hotkey = None）即可。 +#[tauri::command] +pub fn set_qa_hotkey( + coord: CoordinatorState<'_>, + binding: Option, +) -> Result<(), String> { + if let Some(binding) = binding.as_ref() { + crate::shortcut_binding::validate_binding(binding).map_err(|e| e.to_string())?; + if binding.modifiers.is_empty() && binding.primary.eq_ignore_ascii_case("shift") { + return Err("Shift 单键目前只能用于翻译快捷键".into()); + } + } + let mut prefs = coord.prefs().get(); + if let Some(binding) = binding.as_ref() { + reject_dictation_qa_hotkey_overlap(&prefs.dictation_hotkey, binding)?; + reject_qa_translation_hotkey_overlap(binding, &prefs.translation_hotkey)?; + if let Some(switch_style) = prefs.switch_style_hotkey.as_ref() { + reject_qa_switch_style_hotkey_overlap(binding, switch_style)?; + } + if let Some(open_app) = prefs.open_app_hotkey.as_ref() { + reject_qa_open_app_hotkey_overlap(binding, open_app)?; + } + } + prefs.qa_hotkey = binding; + coord.prefs().set(prefs).map_err(|e| e.to_string())?; + coord.update_qa_hotkey_binding(); + Ok(()) +} + +/// 用户点 ✕ / 按 Esc 关 QA 浮窗。 +#[tauri::command] +pub fn qa_window_dismiss(coord: CoordinatorState<'_>) { + coord.qa_window_dismiss(); +} + +/// 用户点 📌 / 取消 📌。pinned=true 时浮窗不会自动隐藏。 +#[tauri::command] +pub fn qa_window_pin(coord: CoordinatorState<'_>, pinned: bool) { + coord.qa_window_pin(pinned); +} + +/// 用户点 ✕ / 按 Esc 关 Less Computer 浮窗。 +#[tauri::command] +pub fn less_computer_window_dismiss(coord: CoordinatorState<'_>) { + coord.less_computer_window_dismiss(); +} + +/// 前端按内容测高后回传高度，后端 clamp + bottom-anchored 重新摆放浮窗。 +#[tauri::command] +pub fn less_computer_window_resize(coord: CoordinatorState<'_>, height: f64) { + coord.less_computer_window_resize(height); +} + +/// 内联审批卡的 Approve / Deny 回执。token 关联到等待中的拦截动作。 +#[tauri::command] +pub fn less_computer_approve(coord: CoordinatorState<'_>, token: String, approved: bool) { + coord.less_computer_approve(&token, approved); +} diff --git a/openless-all/app/src-tauri/src/commands/settings.rs b/openless-all/app/src-tauri/src/commands/settings.rs new file mode 100644 index 00000000..9df40bbb --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/settings.rs @@ -0,0 +1,412 @@ +use super::*; + +#[tauri::command] +pub fn get_settings(coord: CoordinatorState<'_>) -> UserPreferences { + coord.prefs().get() +} + +#[tauri::command] +pub fn get_default_style_system_prompts() -> StyleSystemPrompts { + StyleSystemPrompts::default() +} + +pub(crate) trait SettingsWriter { + fn read_settings(&self) -> UserPreferences; + fn write_settings(&self, prefs: UserPreferences) -> Result<(), String>; + fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String>; + fn refresh_dictation_hotkey(&self); + fn refresh_qa_hotkey(&self); + fn refresh_combo_hotkey(&self); + fn refresh_translation_hotkey(&self); + fn refresh_switch_style_hotkey(&self); + fn refresh_open_app_hotkey(&self); + fn refresh_coding_agent_hotkey(&self); +} + +impl SettingsWriter for Coordinator { + fn read_settings(&self) -> UserPreferences { + self.prefs().get() + } + + fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { + self.prefs().set(prefs).map_err(|e| e.to_string()) + } + + fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { + self.sync_active_asr_provider_to_vault(provider) + } + + fn refresh_dictation_hotkey(&self) { + self.update_hotkey_binding(); + } + + fn refresh_qa_hotkey(&self) { + self.update_qa_hotkey_binding(); + } + + fn refresh_combo_hotkey(&self) { + self.update_combo_hotkey_binding(); + } + + fn refresh_translation_hotkey(&self) { + self.update_translation_hotkey_binding(); + } + + fn refresh_switch_style_hotkey(&self) { + self.update_switch_style_hotkey_binding(); + } + + fn refresh_open_app_hotkey(&self) { + self.update_open_app_hotkey_binding(); + } + + fn refresh_coding_agent_hotkey(&self) { + self.update_coding_agent_hotkey_binding(); + } +} + +impl SettingsWriter for Arc { + fn read_settings(&self) -> UserPreferences { + (**self).read_settings() + } + + fn write_settings(&self, prefs: UserPreferences) -> Result<(), String> { + (**self).write_settings(prefs) + } + + fn sync_active_asr_provider(&self, provider: &str) -> Result<(), String> { + (**self).sync_active_asr_provider(provider) + } + + fn refresh_dictation_hotkey(&self) { + (**self).refresh_dictation_hotkey(); + } + + fn refresh_qa_hotkey(&self) { + (**self).refresh_qa_hotkey(); + } + + fn refresh_combo_hotkey(&self) { + (**self).refresh_combo_hotkey(); + } + + fn refresh_translation_hotkey(&self) { + (**self).refresh_translation_hotkey(); + } + + fn refresh_switch_style_hotkey(&self) { + (**self).refresh_switch_style_hotkey(); + } + + fn refresh_open_app_hotkey(&self) { + (**self).refresh_open_app_hotkey(); + } + + fn refresh_coding_agent_hotkey(&self) { + (**self).refresh_coding_agent_hotkey(); + } +} + +pub(crate) fn persist_settings( + coord: &T, + mut prefs: UserPreferences, +) -> Result<(), String> { + let mut previous = coord.read_settings(); + sync_dictation_hotkey_legacy_fields(&mut previous); + sync_dictation_hotkey_legacy_fields(&mut prefs); + reject_hotkey_collisions(&prefs)?; + let dictation_shortcut_changed = previous.dictation_hotkey != prefs.dictation_hotkey; + let dictation_mode_changed = previous.hotkey.mode != prefs.hotkey.mode; + let qa_changed = previous.qa_hotkey != prefs.qa_hotkey; + let translation_changed = previous.translation_hotkey != prefs.translation_hotkey; + let switch_style_changed = previous.switch_style_hotkey != prefs.switch_style_hotkey; + let open_app_changed = previous.open_app_hotkey != prefs.open_app_hotkey; + let coding_agent_changed = previous.coding_agent_enabled != prefs.coding_agent_enabled + || previous.coding_agent_voice_hotkey != prefs.coding_agent_voice_hotkey; + let active_asr_provider_changed = previous.active_asr_provider != prefs.active_asr_provider; + let active_asr_provider = prefs.active_asr_provider.clone(); + if active_asr_provider_changed { + coord.sync_active_asr_provider(&active_asr_provider)?; + } + if let Err(error) = coord.write_settings(prefs.clone()) { + if active_asr_provider_changed { + if let Err(rollback_error) = + coord.sync_active_asr_provider(&previous.active_asr_provider) + { + coord.write_settings(prefs).map_err(|roll_forward_error| { + format!( + "{error}; additionally failed to restore active ASR provider: {rollback_error}; additionally failed to preserve active ASR provider consistency: {roll_forward_error}" + ) + })?; + } else { + return Err(error); + } + } else { + return Err(error); + } + } + if dictation_shortcut_changed || dictation_mode_changed { + coord.refresh_dictation_hotkey(); + } + if dictation_shortcut_changed { + coord.refresh_combo_hotkey(); + } + if qa_changed { + coord.refresh_qa_hotkey(); + } + if translation_changed { + coord.refresh_translation_hotkey(); + } + if switch_style_changed { + coord.refresh_switch_style_hotkey(); + } + if open_app_changed { + coord.refresh_open_app_hotkey(); + } + if coding_agent_changed { + coord.refresh_coding_agent_hotkey(); + } + Ok(()) +} + +#[tauri::command] +pub fn set_settings( + coord: CoordinatorState<'_>, + app: AppHandle, + tray_microphones: State<'_, TrayMicrophoneMenuState>, + mut prefs: UserPreferences, +) -> Result<(), String> { + let packs = coord.style_packs().list().map_err(|e| e.to_string())?; + sync_style_pack_preferences(&mut prefs, &packs); + // 广播给所有 webview。issue #205：QaPanel 跑在独立 webview， + // 没有 HotkeySettingsContext，必须靠事件感知录音键变化，否则面板可见时 + // 用户改键会让浮窗里的 "{recordHotkey}" 文案一直停留在旧值。 + persist_settings(&*coord, prefs.clone())?; + // refresh_tray_microphone_menu 内部会调用 NSStatusItem.set_menu，必须在主线程上跑。 + // set_settings 本身是同步 Tauri command，在 IPC handler 线程上执行；从这里直接调 + // 会触发 macOS 主线程断言或在 dispatch 队列上死锁，导致整个 UI 无响应（用户改 + // 偏好后所有按键都没反应即此根因）。dispatch 到主线程后立即返回，IPC 线程不阻塞。 + let app_for_main = app.clone(); + let prefs_for_main = prefs.clone(); + let _ = app.run_on_main_thread(move || { + if let Err(err) = crate::refresh_tray_microphone_menu(&app_for_main) { + log::warn!("[tray] refresh microphone menu after settings save failed: {err}"); + let tray_state = app_for_main.state::(); + sync_tray_microphone_selection( + &tray_state.lock(), + &prefs_for_main.microphone_device_name, + ); + } + }); + // 抑制 unused 警告：tray_microphones 现在改在闭包里通过 app.state 取， + // 但函数签名保留 State 入参，以便 Tauri 在调用前注入。 + let _ = tray_microphones; + let _ = app.emit("prefs:changed", &prefs); + Ok(()) +} + +// ─────────────────────────── release channel (Beta opt-in) ─────────────────────────── +// +// 渠道偏好的写入路径跟 set_settings 复用 persist_settings：保持热键兜底归一化 +// 跟其他 prefs 写入一致，且写完后 emit "prefs:changed"，让前端跨 webview 同步。 +// +// 更新：plugin-updater 2.10.1 的 Builder 现在暴露 .endpoints() runtime API（CLAUDE.md +// 当年记的"不支持"已不成立）。本节配合 `app_check_update_with_channel` 命令实现 +// Beta auto-update：Stable 渠道 → 走 tauri.conf 的默认 endpoints；Beta 渠道 → +// fetch_latest_beta_release 拿最新 prerelease tag → 拼成 -beta manifest URL → +// builder.endpoints(vec![url]).build().check()。Stable 用户绝对不会撞到 Beta 包 +// （Beta tag 的 manifest 文件名带 `-beta` 后缀，跟 Stable manifest 在 GitHub +// Release assets 里物理分离）。 + +#[tauri::command] +pub fn get_update_channel(coord: CoordinatorState<'_>) -> UpdateChannel { + coord.prefs().get().update_channel +} + +#[tauri::command] +pub fn set_update_channel( + coord: CoordinatorState<'_>, + app: AppHandle, + channel: UpdateChannel, +) -> Result<(), String> { + let mut prefs = coord.prefs().get(); + if prefs.update_channel == channel { + return Ok(()); + } + prefs.update_channel = channel; + persist_settings(&*coord, prefs.clone())?; + let _ = app.emit("prefs:changed", &prefs); + Ok(()) +} + +#[derive(Debug, Clone, Serialize)] +#[serde(rename_all = "camelCase")] +pub struct LatestBetaRelease { + pub tag_name: String, + pub html_url: String, + pub published_at: String, +} + +/// 拉 GitHub Releases atom feed 找最新 Beta release（tag 以 `-beta-tauri` 结尾）。 +/// +/// 历史：之前用 `api.github.com/repos/.../releases` REST 端点，**未认证 60 req/h/IP**， +/// 多人多次切 Beta toggle 很容易撞 403 rate limit（用户报"获取 Beta 版本信息失败" +/// 即是这个）。换成 `releases.atom` 后是公开页面 + CDN cache，没有同等 rate 限制。 +/// Atom feed 不显式标 prerelease，但项目约定 tag 后缀 `-beta-tauri` 必为 Beta， +/// 所以只用 tag 后缀过滤就够了。 +/// +/// 返回 `Ok(None)` = 当前没发过 Beta 版；`Err(String)` = 网络/解析故障。 +#[tauri::command] +pub async fn fetch_latest_beta_release() -> Result, String> { + let resp = net::send_with_retry(|| { + net::http() + .get("https://github.com/appergb/openless/releases.atom") + .timeout(std::time::Duration::from_secs(15)) + }) + .await + .map_err(|e| format!("fetch releases.atom: {e}"))?; + if !resp.status().is_success() { + return Err(format!("releases.atom status {}", resp.status())); + } + let body = resp + .text() + .await + .map_err(|e| format!("read atom body: {e}"))?; + Ok(parse_latest_beta_from_atom(&body)) +} + +/// 简单字符串解析 atom feed，避免引 XML 库。每个 `...` 内含一行 +/// ``， +/// 用 `/releases/tag/` 这个唯一锚点抓 tag。 +pub(crate) fn parse_latest_beta_from_atom(body: &str) -> Option { + for entry in body.split("").skip(1) { + let entry_body = entry + .split_once("") + .map(|(b, _)| b) + .unwrap_or(entry); + let needle = "/releases/tag/"; + let tag_start = match entry_body.find(needle) { + Some(i) => i + needle.len(), + None => continue, + }; + let tag_after = &entry_body[tag_start..]; + let tag_end = tag_after + .find(|c: char| c == '"' || c == '<' || c == ' ' || c == '/') + .unwrap_or(tag_after.len()); + let tag_name = tag_after[..tag_end].to_string(); + if !tag_name.ends_with("-beta-tauri") { + continue; + } + let html_url = format!("https://github.com/appergb/openless/releases/tag/{tag_name}"); + let published_at = + extract_between(entry_body, "", "").unwrap_or_default(); + return Some(LatestBetaRelease { + tag_name, + html_url, + published_at, + }); + } + None +} + +fn extract_between(haystack: &str, open: &str, close: &str) -> Option { + let start = haystack.find(open)? + open.len(); + let end = haystack[start..].find(close)?; + Some(haystack[start..start + end].to_string()) +} + +// ─────────────────────── Channel-aware updater check ──────────────────────── +// +// 替换前端原来直接 import('@tauri-apps/plugin-updater').check() 的路径： +// - Stable 渠道：builder 不动 endpoints，沿用 tauri.conf 配的 stable manifest URL。 +// - Beta 渠道：先 fetch_latest_beta_release 拿最新 prerelease tag，拼成 -beta manifest +// URL（同时给一对 mirror + direct），再 builder.endpoints(vec![url])?.build()?.check()。 +// +// 返回的 Metadata 形状与 plugin-updater 的 JS UpdateMetadata 完全一致（rid + +// currentVersion 等驼峰字段），前端可以直接 `new Update(metadata)` 复用 plugin +// 的 download / install / close 实现，无需我们自己写下载和签名校验。 +// +// 物理隔离：Beta tag 推出来的 manifest 文件名带 `-beta` 后缀（参见 release-tauri.yml +// 第 382 行注释），跟 Stable 的 `latest-{tgt}-{arch}.json` 在 GitHub Release assets +// 里是分开的两份文件 —— 即使代码逻辑写错把 Beta URL 传给 Stable 用户，HTTP 也是 +// 直接 404，绝不会拿到错档。 + +#[derive(Debug, Clone, Serialize)] +#[serde(rename_all = "camelCase")] +pub struct AppUpdateMetadata { + pub rid: tauri::ResourceId, + pub current_version: String, + pub version: String, + pub date: Option, + pub body: Option, + /// 原始 manifest JSON——`new Update(metadata)` 在 JS 那边会校验它存在； + /// 我们透传 plugin 自己 check 时拿到的字段。 + pub raw_json: serde_json::Value, +} + +/// 决定 manifest 来源后走 plugin-updater 的标准 check 流程。 +/// 渠道：显式传入 `channel` 时用它（关于页固定查 Stable、高级页 Beta 区查 Beta）； +/// 不传则回落到 `prefs.update_channel`（后台 AutoUpdateGate 自动检查走这条）。 +/// 返回 None = 当前是最新；Some(metadata) = 有新版可装。 +#[tauri::command] +pub async fn app_check_update_with_channel( + coord: CoordinatorState<'_>, + webview: tauri::Webview, + timeout_ms: Option, + channel: Option, +) -> Result, String> { + use tauri_plugin_updater::UpdaterExt; + + let channel = channel.unwrap_or_else(|| coord.prefs().get().update_channel); + let mut builder = webview.updater_builder(); + if let Some(ms) = timeout_ms { + builder = builder.timeout(std::time::Duration::from_millis(ms)); + } + if matches!(channel, UpdateChannel::Beta) { + let urls = resolve_beta_manifest_endpoints().await?; + builder = builder + .endpoints(urls) + .map_err(|e| format!("set beta endpoints: {e}"))?; + } + let updater = builder.build().map_err(|e| format!("build updater: {e}"))?; + let update = updater + .check() + .await + .map_err(|e| format!("check update failed: {e}"))?; + + let Some(update) = update else { + return Ok(None); + }; + // date 字段透传需要引 time crate；前端 AutoUpdate.tsx 实际并不用 date，所以这里 + // 直接置 None，避免拉一个新 dep 进 src-tauri/Cargo.toml。 + let metadata = AppUpdateMetadata { + current_version: update.current_version.clone(), + version: update.version.clone(), + date: None, + body: update.body.clone(), + raw_json: update.raw_json.clone(), + rid: webview.resources_table().add(update), + }; + Ok(Some(metadata)) +} + +/// 把 fetch_latest_beta_release 找到的最新 prerelease tag 拼成 -beta manifest URL 对。 +/// 顺序：先镜像（fastgit.cc 代理 GitHub），后直连 —— 跟 tauri.conf 现有 Stable +/// endpoints 一致，让国内访问优先打到 CDN。 +async fn resolve_beta_manifest_endpoints() -> Result, String> { + let Some(latest) = fetch_latest_beta_release().await? else { + return Err("尚未发布过 Beta 版本".to_string()); + }; + let tag = latest.tag_name; + // {{target}} / {{arch}} 占位符由 plugin 在 check 时替换。Rust raw string 用 r#""# + // 不需要转义双花括号，比 format! 干净。 + let mirror = format!( + "https://fastgit.cc/https://github.com/appergb/openless/releases/download/{tag}/latest-{{{{target}}}}-{{{{arch}}}}-beta-mirror.json" + ); + let direct = format!( + "https://github.com/appergb/openless/releases/download/{tag}/latest-{{{{target}}}}-{{{{arch}}}}-beta.json" + ); + let mirror_url = url::Url::parse(&mirror).map_err(|e| format!("parse beta mirror url: {e}"))?; + let direct_url = url::Url::parse(&direct).map_err(|e| format!("parse beta direct url: {e}"))?; + Ok(vec![mirror_url, direct_url]) +} diff --git a/openless-all/app/src-tauri/src/commands/sherpa_asr.rs b/openless-all/app/src-tauri/src/commands/sherpa_asr.rs new file mode 100644 index 00000000..0dd8c9b0 --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/sherpa_asr.rs @@ -0,0 +1,192 @@ +use super::*; + +pub(crate) fn active_sherpa_model_from_prefs(prefs: &UserPreferences) -> String { + if sherpa_model_alias_is_known(&prefs.sherpa_onnx_model) { + prefs.sherpa_onnx_model.clone() + } else { + SHERPA_DEFAULT_MODEL_ALIAS.to_string() + } +} + +pub(crate) fn validate_sherpa_model_alias(model_alias: &str) -> Result<(), String> { + if sherpa_model_alias_is_known(model_alias) { + Ok(()) + } else { + Err(format!("unknown sherpa-onnx model alias: {model_alias}")) + } +} + +pub(crate) fn normalize_sherpa_language_hint(language_hint: &str) -> Result { + let normalized = language_hint.trim().to_lowercase(); + if normalized.is_empty() + || normalized + .chars() + .all(|c| c.is_ascii_lowercase() || c == '-') + { + Ok(normalized) + } else { + Err("language hint must be empty or BCP-47 lowercase code".to_string()) + } +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_status( + coord: CoordinatorState<'_>, + runtime: State<'_, Arc>, +) -> Result { + let prefs = coord.prefs().get(); + let active_model = active_sherpa_model_from_prefs(&prefs); + Ok(runtime.status_snapshot(&active_model).await) +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_catalog( + runtime: State<'_, Arc>, +) -> Result, String> { + runtime + .catalog_snapshot() + .await + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_fetch_remote_info( + model_alias: String, + mirror: Option, +) -> Result { + validate_sherpa_model_alias(&model_alias)?; + let mirror = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); + fetch_sherpa_remote_info(&model_alias, mirror) + .await + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_download_model( + app: AppHandle, + manager: State<'_, Arc>, + model_alias: String, + mirror: Option, +) -> Result<(), String> { + validate_sherpa_model_alias(&model_alias)?; + let mirror = mirror.as_deref().map(Mirror::from_str).unwrap_or_default(); + manager.start(app, model_alias, mirror); + Ok(()) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_cancel_download( + manager: State<'_, Arc>, + model_alias: String, +) -> Result<(), String> { + validate_sherpa_model_alias(&model_alias)?; + manager.cancel(&model_alias); + Ok(()) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_set_model( + coord: CoordinatorState<'_>, + model_alias: String, +) -> Result<(), String> { + validate_sherpa_model_alias(&model_alias)?; + let mut prefs = coord.prefs().get(); + if prefs.sherpa_onnx_model == model_alias { + return Ok(()); + } + prefs.sherpa_onnx_model = model_alias; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_set_language_hint( + coord: CoordinatorState<'_>, + language_hint: String, +) -> Result<(), String> { + let normalized = normalize_sherpa_language_hint(&language_hint)?; + let mut prefs = coord.prefs().get(); + if prefs.sherpa_onnx_language_hint == normalized { + return Ok(()); + } + prefs.sherpa_onnx_language_hint = normalized; + coord.prefs().set(prefs).map_err(|e| e.to_string()) +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_prepare( + app: AppHandle, + runtime: State<'_, Arc>, + model_alias: String, +) -> Result { + validate_sherpa_model_alias(&model_alias)?; + let progress_app = app.clone(); + let result = runtime + .ensure_loaded_with_progress(&model_alias, move |payload| { + emit_sherpa_prepare_progress(&progress_app, payload); + }) + .await; + match result { + Ok(loaded) => Ok(loaded), + Err(error) => { + let message = format!("{error:#}"); + emit_sherpa_prepare_progress( + &app, + SherpaPrepareProgressPayload::failed( + model_alias, + "sherpa-onnx prepare failed", + message.clone(), + ), + ); + Err(message) + } + } +} + +#[tauri::command] +pub fn sherpa_onnx_asr_cancel_prepare( + runtime: State<'_, Arc>, +) -> Result<(), String> { + runtime.request_cancel_prepare(); + Ok(()) +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_release( + runtime: State<'_, Arc>, +) -> Result<(), String> { + runtime.release_now().await.map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_model_dir(model_alias: String) -> Result { + validate_sherpa_model_alias(&model_alias)?; + SherpaOnnxRuntime::model_dir_for_alias(&model_alias) + .map(|path| path.display().to_string()) + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub async fn sherpa_onnx_asr_delete_model( + runtime: State<'_, Arc>, + model_alias: String, +) -> Result<(), String> { + validate_sherpa_model_alias(&model_alias)?; + runtime + .delete_model(&model_alias) + .await + .map_err(|e| format!("{e:#}")) +} + +#[tauri::command] +pub fn sherpa_onnx_asr_reveal_model_dir(model_alias: String) -> Result<(), String> { + validate_sherpa_model_alias(&model_alias)?; + let dir = SherpaOnnxRuntime::model_dir_for_alias(&model_alias).map_err(|e| format!("{e:#}"))?; + std::fs::create_dir_all(&dir).map_err(|e| format!("create {} failed: {e}", dir.display()))?; + open_path_in_file_manager(&dir) +} + +fn emit_sherpa_prepare_progress(app: &AppHandle, payload: SherpaPrepareProgressPayload) { + if let Err(error) = app.emit("sherpa-onnx-asr-prepare-progress", payload) { + log::warn!("[sherpa-asr] emit prepare progress failed: {error}"); + } +} diff --git a/openless-all/app/src-tauri/src/commands/style_packs.rs b/openless-all/app/src-tauri/src/commands/style_packs.rs new file mode 100644 index 00000000..9afb8e8e --- /dev/null +++ b/openless-all/app/src-tauri/src/commands/style_packs.rs @@ -0,0 +1,287 @@ +use super::*; + +fn refresh_tray_menu_async(app: &AppHandle) { + let app_for_main = app.clone(); + let _ = app.run_on_main_thread(move || { + if let Err(err) = crate::refresh_tray_microphone_menu(&app_for_main) { + log::warn!("[tray] refresh after style change failed: {err}"); + } + }); +} + +fn emit_prefs_changed(app: &AppHandle, prefs: &UserPreferences) { + let _ = app.emit("prefs:changed", prefs); + let _ = app.emit_to("main", "prefs:changed", prefs); +} + +pub(crate) fn sync_style_pack_prefs_and_persist( + coord: &Coordinator, + app: &AppHandle, + mut prefs: UserPreferences, +) -> Result { + let packs = coord.style_packs().list().map_err(|e| e.to_string())?; + sync_style_pack_preferences(&mut prefs, &packs); + coord + .prefs() + .set(prefs.clone()) + .map_err(|e| e.to_string())?; + emit_prefs_changed(app, &prefs); + refresh_tray_menu_async(app); + Ok(prefs) +} + +pub(crate) fn activate_style_pack_by_id( + coord: &Coordinator, + app: &AppHandle, + id: &str, +) -> Result { + let mut prefs = coord.prefs().get(); + let pack = coord.style_packs().get(id).map_err(|e| e.to_string())?; + log::info!( + "[style-pack] activate helper requested id={} kind={:?} base_mode={:?} enabled={}", + pack.id, + pack.kind, + pack.base_mode, + pack.enabled + ); + if !pack.enabled { + coord + .style_packs() + .set_enabled(id, true) + .map_err(|e| e.to_string())?; + } + prefs.active_style_pack_id = id.to_string(); + sync_style_pack_prefs_and_persist(coord, app, prefs)?; + log::info!("[style-pack] activate helper applied id={id}"); + coord + .style_packs() + .get(id) + .map(|mut pack| { + pack.active = true; + pack + }) + .map_err(|e| e.to_string()) +} + +pub(crate) fn activate_builtin_style_mode( + coord: &Coordinator, + app: &AppHandle, + mode: PolishMode, +) -> Result<(), String> { + let pack_id = builtin_style_pack_id(mode).to_string(); + log::info!( + "[style-pack] activate builtin mode helper mode={:?} pack_id={}", + mode, + pack_id + ); + let _ = activate_style_pack_by_id(coord, app, &pack_id)?; + Ok(()) +} + +// ─────────────────────────── style packs ─────────────────────────── + +#[tauri::command] +pub fn list_style_packs(coord: CoordinatorState<'_>) -> Result, String> { + let prefs = coord.prefs().get(); + coord + .style_packs() + .list_with_active(&prefs.active_style_pack_id) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn create_style_pack_from_template( + coord: CoordinatorState<'_>, + app: AppHandle, + template: StylePack, +) -> Result { + log::info!( + "[style-pack] command create_from_template name={} base_mode={:?}", + template.name, + template.base_mode + ); + let created = coord + .style_packs() + .create_from_template(template) + .map_err(|e| e.to_string())?; + let prefs = coord.prefs().get(); + let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + Ok(created) +} + +#[tauri::command] +pub fn save_style_pack( + coord: CoordinatorState<'_>, + app: AppHandle, + style_pack: StylePack, +) -> Result { + log::info!( + "[style-pack] command save id={} kind={:?} base_mode={:?}", + style_pack.id, + style_pack.kind, + style_pack.base_mode + ); + let saved = coord + .style_packs() + .upsert(style_pack) + .map_err(|e| e.to_string())?; + if saved.kind == StylePackKind::Builtin { + let prefs = coord.prefs().get(); + let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + } + Ok(saved) +} + +#[tauri::command] +pub fn preview_style_pack_runtime( + coord: CoordinatorState<'_>, + style_pack: StylePack, +) -> Result { + log::info!( + "[style-pack] command preview_runtime id={} base_mode={:?} prompt_chars={}", + style_pack.id, + style_pack.base_mode, + style_pack.prompt.chars().count() + ); + Ok(coord.preview_style_pack_runtime(&style_pack)) +} + +#[tauri::command] +pub fn set_active_style_pack( + coord: CoordinatorState<'_>, + app: AppHandle, + id: String, +) -> Result { + activate_style_pack_by_id(&coord, &app, &id) +} + +#[tauri::command] +pub fn set_style_pack_enabled( + coord: CoordinatorState<'_>, + app: AppHandle, + id: String, + enabled: bool, +) -> Result, String> { + log::info!( + "[style-pack] command set_enabled requested id={} enabled={}", + id, + enabled + ); + coord + .style_packs() + .set_enabled(&id, enabled) + .map_err(|e| e.to_string())?; + let mut prefs = coord.prefs().get(); + if !enabled && prefs.active_style_pack_id == id { + prefs.active_style_pack_id = default_active_style_pack_id(); + } + let prefs = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + coord + .style_packs() + .list_with_active(&prefs.active_style_pack_id) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn reset_builtin_style_pack( + coord: CoordinatorState<'_>, + app: AppHandle, + id: String, +) -> Result { + log::info!("[style-pack] command reset_builtin requested id={id}"); + let saved = coord + .style_packs() + .reset_builtin(&id) + .map_err(|e| e.to_string())?; + let prefs = coord.prefs().get(); + let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + Ok(saved) +} + +#[tauri::command] +pub fn delete_style_pack( + coord: CoordinatorState<'_>, + app: AppHandle, + id: String, +) -> Result<(), String> { + let mut prefs = coord.prefs().get(); + log::info!("[style-pack] command delete requested id={id}"); + coord + .style_packs() + .remove_imported(&id) + .map_err(|e| e.to_string())?; + if prefs.active_style_pack_id == id { + prefs.active_style_pack_id = default_active_style_pack_id(); + let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + } else { + refresh_tray_menu_async(&app); + } + Ok(()) +} + +#[tauri::command] +pub fn import_style_pack_from_zip( + coord: CoordinatorState<'_>, + zip_path: String, +) -> Result { + log::info!("[style-pack] command import requested zip_path={zip_path}"); + coord + .style_packs() + .import_from_zip(std::path::Path::new(&zip_path)) + .map_err(|e| e.to_string()) +} + +#[tauri::command] +pub fn export_style_pack_to_zip( + coord: CoordinatorState<'_>, + id: String, + target_path: String, +) -> Result { + log::info!( + "[style-pack] command export requested id={} target_path={}", + id, + target_path + ); + coord + .style_packs() + .export_to_zip(&id, std::path::Path::new(&target_path)) + .map_err(|e| e.to_string())?; + Ok(target_path) +} + +// ─────────────────────────── style toggles (compat) ─────────────────────────── + +#[tauri::command] +pub fn set_default_polish_mode( + coord: CoordinatorState<'_>, + app: AppHandle, + mode: PolishMode, +) -> Result<(), String> { + activate_builtin_style_mode(&coord, &app, mode) +} + +#[tauri::command] +pub fn set_style_enabled( + coord: CoordinatorState<'_>, + app: AppHandle, + mode: PolishMode, + enabled: bool, +) -> Result<(), String> { + let pack_id = builtin_style_pack_id(mode).to_string(); + log::info!( + "[style-pack] compat set_style_enabled mode={:?} pack_id={} enabled={}", + mode, + pack_id, + enabled + ); + coord + .style_packs() + .set_enabled(&pack_id, enabled) + .map_err(|e| e.to_string())?; + let mut prefs = coord.prefs().get(); + if !enabled && prefs.active_style_pack_id == pack_id { + prefs.active_style_pack_id = default_active_style_pack_id(); + } + let _ = sync_style_pack_prefs_and_persist(&*coord, &app, prefs)?; + Ok(()) +}