mod app_config; mod app_store; mod auto_launch; mod claude_mcp; mod claude_plugin; mod codex_config; mod commands; mod config; mod database; mod deeplink; mod error; mod gemini_config; mod gemini_mcp; mod init_status; mod mcp; mod prompt; mod prompt_files; mod provider; mod provider_defaults; mod proxy; mod services; mod settings; mod store; mod tray; mod usage_script; pub use app_config::{AppType, McpApps, McpServer, MultiAppConfig}; pub use codex_config::{get_codex_auth_path, get_codex_config_path, write_codex_live_atomic}; pub use commands::*; pub use config::{get_claude_mcp_path, get_claude_settings_path, read_json_file}; pub use database::Database; pub use deeplink::{import_provider_from_deeplink, parse_deeplink_url, DeepLinkImportRequest}; pub use error::AppError; pub use mcp::{ import_from_claude, import_from_codex, import_from_gemini, remove_server_from_claude, remove_server_from_codex, remove_server_from_gemini, sync_enabled_to_claude, sync_enabled_to_codex, sync_enabled_to_gemini, sync_single_server_to_claude, sync_single_server_to_codex, sync_single_server_to_gemini, }; pub use provider::{Provider, ProviderMeta}; pub use services::{ ConfigService, EndpointLatency, McpService, PromptService, ProviderService, SkillService, SpeedtestService, }; pub use settings::{update_settings, AppSettings}; pub use store::AppState; use tauri_plugin_deep_link::DeepLinkExt; use tauri_plugin_dialog::{DialogExt, MessageDialogButtons, MessageDialogKind}; use std::sync::Arc; use tauri::tray::{TrayIconBuilder, TrayIconEvent}; #[cfg(target_os = "macos")] use tauri::RunEvent; use tauri::{Emitter, Manager}; /// 统一处理 ccswitch:// 深链接 URL /// /// - 解析 URL /// - 向前端发射 `deeplink-import` / `deeplink-error` 事件 /// - 可选:在成功时聚焦主窗口 fn handle_deeplink_url( app: &tauri::AppHandle, url_str: &str, focus_main_window: bool, source: &str, ) -> bool { if !url_str.starts_with("ccswitch://") { return false; } log::info!("✓ Deep link URL detected from {source}: {url_str}"); match crate::deeplink::parse_deeplink_url(url_str) { Ok(request) => { log::info!( "✓ Successfully parsed deep link: resource={}, app={:?}, name={:?}", request.resource, request.app, request.name ); if let Err(e) = app.emit("deeplink-import", &request) { log::error!("✗ Failed to emit deeplink-import event: {e}"); } else { log::info!("✓ Emitted deeplink-import event to frontend"); } if focus_main_window { if let Some(window) = app.get_webview_window("main") { let _ = window.unminimize(); let _ = window.show(); let _ = window.set_focus(); log::info!("✓ Window shown and focused"); } } } Err(e) => { log::error!("✗ Failed to parse deep link URL: {e}"); if let Err(emit_err) = app.emit( "deeplink-error", serde_json::json!({ "url": url_str, "error": e.to_string() }), ) { log::error!("✗ Failed to emit deeplink-error event: {emit_err}"); } } } true } /// 更新托盘菜单的Tauri命令 #[tauri::command] async fn update_tray_menu( app: tauri::AppHandle, state: tauri::State<'_, AppState>, ) -> Result { match tray::create_tray_menu(&app, state.inner()) { Ok(new_menu) => { if let Some(tray) = app.tray_by_id("main") { tray.set_menu(Some(new_menu)) .map_err(|e| format!("更新托盘菜单失败: {e}"))?; return Ok(true); } Ok(false) } Err(err) => { log::error!("创建托盘菜单失败: {err}"); Ok(false) } } } #[cfg_attr(mobile, tauri::mobile_entry_point)] pub fn run() { let mut builder = tauri::Builder::default(); #[cfg(any(target_os = "macos", target_os = "windows", target_os = "linux"))] { builder = builder.plugin(tauri_plugin_single_instance::init(|app, args, _cwd| { log::info!("=== Single Instance Callback Triggered ==="); log::info!("Args count: {}", args.len()); for (i, arg) in args.iter().enumerate() { log::info!(" arg[{i}]: {arg}"); } // Check for deep link URL in args (mainly for Windows/Linux command line) let mut found_deeplink = false; for arg in &args { if handle_deeplink_url(app, arg, false, "single_instance args") { found_deeplink = true; break; } } if !found_deeplink { log::info!("ℹ No deep link URL found in args (this is expected on macOS when launched via system)"); } // Show and focus window regardless if let Some(window) = app.get_webview_window("main") { let _ = window.unminimize(); let _ = window.show(); let _ = window.set_focus(); } })); } let builder = builder // 注册 deep-link 插件(处理 macOS AppleEvent 和其他平台的深链接) .plugin(tauri_plugin_deep_link::init()) // 拦截窗口关闭:根据设置决定是否最小化到托盘 .on_window_event(|window, event| { if let tauri::WindowEvent::CloseRequested { api, .. } = event { let settings = crate::settings::get_settings(); if settings.minimize_to_tray_on_close { api.prevent_close(); let _ = window.hide(); #[cfg(target_os = "windows")] { let _ = window.set_skip_taskbar(true); } #[cfg(target_os = "macos")] { tray::apply_tray_policy(window.app_handle(), false); } } else { window.app_handle().exit(0); } } }) .plugin(tauri_plugin_process::init()) .plugin(tauri_plugin_dialog::init()) .plugin(tauri_plugin_opener::init()) .plugin(tauri_plugin_store::Builder::new().build()) .setup(|app| { // 注册 Updater 插件(桌面端) #[cfg(desktop)] { if let Err(e) = app .handle() .plugin(tauri_plugin_updater::Builder::new().build()) { // 若配置不完整(如缺少 pubkey),跳过 Updater 而不中断应用 log::warn!("初始化 Updater 插件失败,已跳过:{e}"); } } #[cfg(target_os = "macos")] { // 设置 macOS 标题栏背景色为主界面蓝色 if let Some(window) = app.get_webview_window("main") { use objc2::rc::Retained; use objc2::runtime::AnyObject; use objc2_app_kit::NSColor; match window.ns_window() { Ok(ns_window_ptr) => { if let Some(ns_window) = unsafe { Retained::retain(ns_window_ptr as *mut AnyObject) } { // 使用与主界面 banner 相同的蓝色 #3498db // #3498db = RGB(52, 152, 219) let bg_color = unsafe { NSColor::colorWithRed_green_blue_alpha( 52.0 / 255.0, // R: 52 152.0 / 255.0, // G: 152 219.0 / 255.0, // B: 219 1.0, // Alpha: 1.0 ) }; unsafe { use objc2::msg_send; let _: () = msg_send![&*ns_window, setBackgroundColor: &*bg_color]; } } else { log::warn!("Failed to retain NSWindow reference"); } } Err(e) => log::warn!("Failed to get NSWindow pointer: {e}"), } } } // 初始化日志 if cfg!(debug_assertions) { app.handle().plugin( tauri_plugin_log::Builder::default() .level(log::LevelFilter::Info) .build(), )?; } // 预先刷新 Store 覆盖配置,确保 AppState 初始化时可读取到最新路径 app_store::refresh_app_config_dir_override(app.handle()); // 初始化数据库 let app_config_dir = crate::config::get_app_config_dir(); let db_path = app_config_dir.join("cc-switch.db"); let json_path = app_config_dir.join("config.json"); // 检查是否需要从 config.json 迁移到 SQLite let has_json = json_path.exists(); let has_db = db_path.exists(); // 如果需要迁移,先验证 config.json 是否可以加载(在创建数据库之前) // 这样如果加载失败用户选择退出,数据库文件还没被创建,下次可以正常重试 let migration_config = if !has_db && has_json { log::info!("检测到旧版配置文件,验证配置文件..."); // 循环:支持用户重试加载配置文件 loop { match crate::app_config::MultiAppConfig::load() { Ok(config) => { log::info!("✓ 配置文件加载成功"); break Some(config); } Err(e) => { log::error!("加载旧配置文件失败: {e}"); // 弹出系统对话框让用户选择 if !show_migration_error_dialog(app.handle(), &e.to_string()) { // 用户选择退出(此时数据库还没创建,下次启动可以重试) log::info!("用户选择退出程序"); std::process::exit(1); } // 用户选择重试,继续循环 log::info!("用户选择重试加载配置文件"); } } } } else { None }; // 现在创建数据库 let db = match crate::database::Database::init() { Ok(db) => Arc::new(db), Err(e) => { log::error!("Failed to init database: {e}"); return Err(Box::new(e)); } }; // 如果有预加载的配置,执行迁移 if let Some(config) = migration_config { log::info!("开始执行数据迁移..."); match db.migrate_from_json(&config) { Ok(_) => { log::info!("✓ 配置迁移成功"); // 标记迁移成功,供前端显示 Toast crate::init_status::set_migration_success(); // 归档旧配置文件(重命名而非删除,便于用户恢复) let archive_path = json_path.with_extension("json.migrated"); if let Err(e) = std::fs::rename(&json_path, &archive_path) { log::warn!("归档旧配置文件失败: {e}"); } else { log::info!("✓ 旧配置已归档为 config.json.migrated"); } } Err(e) => { // 配置加载成功但迁移失败的情况极少(磁盘满等),仅记录日志 log::error!("配置迁移失败: {e},将从现有配置导入"); } } } let app_state = AppState::new(db); // ============================================================ // 按表独立判断的导入逻辑(各类数据独立检查,互不影响) // ============================================================ // 1. 初始化默认 Skills 仓库(已有内置检查:表非空则跳过) match app_state.db.init_default_skill_repos() { Ok(count) if count > 0 => { log::info!("✓ Initialized {count} default skill repositories"); } Ok(_) => {} // 表非空,静默跳过 Err(e) => log::warn!("✗ Failed to initialize default skill repos: {e}"), } // 2. 导入供应商配置(已有内置检查:该应用已有供应商则跳过) for app in [ crate::app_config::AppType::Claude, crate::app_config::AppType::Codex, crate::app_config::AppType::Gemini, ] { match crate::services::provider::ProviderService::import_default_config( &app_state, app.clone(), ) { Ok(true) => { log::info!("✓ Imported default provider for {}", app.as_str()); } Ok(false) => {} // 已有供应商,静默跳过 Err(e) => { log::debug!( "○ No default provider to import for {}: {}", app.as_str(), e ); } } } // 3. 导入 MCP 服务器配置(表空时触发) if app_state.db.is_mcp_table_empty().unwrap_or(false) { log::info!("MCP table empty, importing from live configurations..."); match crate::services::mcp::McpService::import_from_claude(&app_state) { Ok(count) if count > 0 => { log::info!("✓ Imported {count} MCP server(s) from Claude"); } Ok(_) => log::debug!("○ No Claude MCP servers found to import"), Err(e) => log::warn!("✗ Failed to import Claude MCP: {e}"), } match crate::services::mcp::McpService::import_from_codex(&app_state) { Ok(count) if count > 0 => { log::info!("✓ Imported {count} MCP server(s) from Codex"); } Ok(_) => log::debug!("○ No Codex MCP servers found to import"), Err(e) => log::warn!("✗ Failed to import Codex MCP: {e}"), } match crate::services::mcp::McpService::import_from_gemini(&app_state) { Ok(count) if count > 0 => { log::info!("✓ Imported {count} MCP server(s) from Gemini"); } Ok(_) => log::debug!("○ No Gemini MCP servers found to import"), Err(e) => log::warn!("✗ Failed to import Gemini MCP: {e}"), } } // 4. 导入提示词文件(表空时触发) if app_state.db.is_prompts_table_empty().unwrap_or(false) { log::info!("Prompts table empty, importing from live configurations..."); for app in [ crate::app_config::AppType::Claude, crate::app_config::AppType::Codex, crate::app_config::AppType::Gemini, ] { match crate::services::prompt::PromptService::import_from_file_on_first_launch( &app_state, app.clone(), ) { Ok(count) if count > 0 => { log::info!("✓ Imported {count} prompt(s) for {}", app.as_str()); } Ok(_) => log::debug!("○ No prompt file found for {}", app.as_str()), Err(e) => log::warn!("✗ Failed to import prompt for {}: {e}", app.as_str()), } } } // 迁移旧的 app_config_dir 配置到 Store if let Err(e) = app_store::migrate_app_config_dir_from_settings(app.handle()) { log::warn!("迁移 app_config_dir 失败: {e}"); } // 启动阶段不再无条件保存,避免意外覆盖用户配置。 // 注册 deep-link URL 处理器(使用正确的 DeepLinkExt API) log::info!("=== Registering deep-link URL handler ==="); // Linux 和 Windows 调试模式需要显式注册 #[cfg(any(target_os = "linux", all(debug_assertions, windows)))] { #[cfg(target_os = "linux")] { // Use Tauri's path API to get correct path (includes app identifier) // tauri-plugin-deep-link writes to: ~/.local/share/com.ccswitch.desktop/applications/cc-switch-handler.desktop // Only register if .desktop file doesn't exist to avoid overwriting user customizations let should_register = app .path() .data_dir() .map(|d| !d.join("applications/cc-switch-handler.desktop").exists()) .unwrap_or(true); if should_register { if let Err(e) = app.deep_link().register_all() { log::error!("✗ Failed to register deep link schemes: {}", e); } else { log::info!("✓ Deep link schemes registered (Linux)"); } } else { log::info!("⊘ Deep link handler already exists, skipping registration"); } } #[cfg(all(debug_assertions, windows))] { if let Err(e) = app.deep_link().register_all() { log::error!("✗ Failed to register deep link schemes: {}", e); } else { log::info!("✓ Deep link schemes registered (Windows debug)"); } } } // 注册 URL 处理回调(所有平台通用) app.deep_link().on_open_url({ let app_handle = app.handle().clone(); move |event| { log::info!("=== Deep Link Event Received (on_open_url) ==="); let urls = event.urls(); log::info!("Received {} URL(s)", urls.len()); for (i, url) in urls.iter().enumerate() { let url_str = url.as_str(); log::info!(" URL[{i}]: {url_str}"); if handle_deeplink_url(&app_handle, url_str, true, "on_open_url") { break; // Process only first ccswitch:// URL } } } }); log::info!("✓ Deep-link URL handler registered"); // 创建动态托盘菜单 let menu = tray::create_tray_menu(app.handle(), &app_state)?; // 构建托盘 let mut tray_builder = TrayIconBuilder::with_id("main") .on_tray_icon_event(|_tray, event| match event { // 左键点击已通过 show_menu_on_left_click(true) 打开菜单,这里不再额外处理 TrayIconEvent::Click { .. } => {} _ => log::debug!("unhandled event {event:?}"), }) .menu(&menu) .on_menu_event(|app, event| { tray::handle_tray_menu_event(app, &event.id.0); }) .show_menu_on_left_click(true); // 统一使用应用默认图标;待托盘模板图标就绪后再启用 if let Some(icon) = app.default_window_icon() { tray_builder = tray_builder.icon(icon.clone()); } else { log::warn!("Failed to get default window icon for tray"); } let _tray = tray_builder.build(app)?; // 将同一个实例注入到全局状态,避免重复创建导致的不一致 app.manage(app_state); // 初始化 SkillService match SkillService::new() { Ok(skill_service) => { app.manage(commands::skill::SkillServiceState(Arc::new(skill_service))); } Err(e) => { log::warn!("初始化 SkillService 失败: {e}"); } } // 自动启动代理服务器 let app_handle = app.handle().clone(); tauri::async_runtime::spawn(async move { let state = app_handle.state::(); match state.db.get_proxy_config().await { Ok(config) => { if config.enabled { log::info!("代理服务配置为启用,正在启动..."); match state.proxy_service.start().await { Ok(info) => log::info!( "代理服务器自动启动成功: {}:{}", info.address, info.port ), Err(e) => log::error!("代理服务器自动启动失败: {e}"), } } } Err(e) => log::error!("启动时获取代理配置失败: {e}"), } }); Ok(()) }) .invoke_handler(tauri::generate_handler![ commands::get_providers, commands::get_current_provider, commands::add_provider, commands::update_provider, commands::delete_provider, commands::switch_provider, commands::set_proxy_target_provider, commands::import_default_config, commands::get_claude_config_status, commands::get_config_status, commands::get_claude_code_config_path, commands::get_config_dir, commands::open_config_folder, commands::pick_directory, commands::open_external, commands::get_init_error, commands::get_migration_result, commands::get_app_config_path, commands::open_app_config_folder, commands::get_claude_common_config_snippet, commands::set_claude_common_config_snippet, commands::get_common_config_snippet, commands::set_common_config_snippet, commands::read_live_provider_settings, commands::get_settings, commands::save_settings, commands::restart_app, commands::check_for_updates, commands::is_portable_mode, commands::get_claude_plugin_status, commands::read_claude_plugin_config, commands::apply_claude_plugin_config, commands::is_claude_plugin_applied, // Claude MCP management commands::get_claude_mcp_status, commands::read_claude_mcp_config, commands::upsert_claude_mcp_server, commands::delete_claude_mcp_server, commands::validate_mcp_command, // usage query commands::queryProviderUsage, commands::testUsageScript, // New MCP via config.json (SSOT) commands::get_mcp_config, commands::upsert_mcp_server_in_config, commands::delete_mcp_server_in_config, commands::set_mcp_enabled, // v3.7.0: Unified MCP management commands::get_mcp_servers, commands::upsert_mcp_server, commands::delete_mcp_server, commands::toggle_mcp_app, // Prompt management commands::get_prompts, commands::upsert_prompt, commands::delete_prompt, commands::enable_prompt, commands::import_prompt_from_file, commands::get_current_prompt_file_content, // ours: endpoint speed test + custom endpoint management commands::test_api_endpoints, commands::get_custom_endpoints, commands::add_custom_endpoint, commands::remove_custom_endpoint, commands::update_endpoint_last_used, // app_config_dir override via Store commands::get_app_config_dir_override, commands::set_app_config_dir_override, // provider sort order management commands::update_providers_sort_order, // theirs: config import/export and dialogs commands::export_config_to_file, commands::import_config_from_file, commands::save_file_dialog, commands::open_file_dialog, commands::sync_current_providers_live, // Deep link import commands::parse_deeplink, commands::merge_deeplink_config, commands::import_from_deeplink, commands::import_from_deeplink_unified, update_tray_menu, // Environment variable management commands::check_env_conflicts, commands::delete_env_vars, commands::restore_env_backup, // Skill management commands::get_skills, commands::install_skill, commands::uninstall_skill, commands::get_skill_repos, commands::add_skill_repo, commands::remove_skill_repo, // Auto launch commands::set_auto_launch, commands::get_auto_launch_status, // Proxy server management commands::start_proxy_server, commands::stop_proxy_server, commands::get_proxy_status, commands::get_proxy_config, commands::update_proxy_config, commands::is_proxy_running, // Usage statistics commands::get_usage_summary, commands::get_usage_trends, commands::get_provider_stats, commands::get_model_stats, commands::get_request_logs, commands::get_request_detail, commands::get_model_pricing, commands::update_model_pricing, commands::delete_model_pricing, commands::check_provider_limits, // Model testing commands::test_provider_model, commands::test_all_providers_model, commands::get_model_test_config, commands::save_model_test_config, commands::get_model_test_logs, commands::cleanup_model_test_logs, ]); let app = builder .build(tauri::generate_context!()) .expect("error while running tauri application"); app.run(|app_handle, event| { #[cfg(target_os = "macos")] { match event { // macOS 在 Dock 图标被点击并重新激活应用时会触发 Reopen 事件,这里手动恢复主窗口 RunEvent::Reopen { .. } => { if let Some(window) = app_handle.get_webview_window("main") { #[cfg(target_os = "windows")] { let _ = window.set_skip_taskbar(false); } let _ = window.unminimize(); let _ = window.show(); let _ = window.set_focus(); tray::apply_tray_policy(app_handle, true); } } // 处理通过自定义 URL 协议触发的打开事件(例如 ccswitch://...) RunEvent::Opened { urls } => { if let Some(url) = urls.first() { let url_str = url.to_string(); log::info!("RunEvent::Opened with URL: {url_str}"); if url_str.starts_with("ccswitch://") { // 解析并广播深链接事件,复用与 single_instance 相同的逻辑 match crate::deeplink::parse_deeplink_url(&url_str) { Ok(request) => { log::info!( "Successfully parsed deep link from RunEvent::Opened: resource={}, app={:?}", request.resource, request.app ); if let Err(e) = app_handle.emit("deeplink-import", &request) { log::error!( "Failed to emit deep link event from RunEvent::Opened: {e}" ); } } Err(e) => { log::error!( "Failed to parse deep link URL from RunEvent::Opened: {e}" ); if let Err(emit_err) = app_handle.emit( "deeplink-error", serde_json::json!({ "url": url_str, "error": e.to_string() }), ) { log::error!( "Failed to emit deep link error event from RunEvent::Opened: {emit_err}" ); } } } // 确保主窗口可见 if let Some(window) = app_handle.get_webview_window("main") { let _ = window.unminimize(); let _ = window.show(); let _ = window.set_focus(); } } } } _ => {} } } #[cfg(not(target_os = "macos"))] { let _ = (app_handle, event); } }); } // ============================================================ // 迁移错误对话框辅助函数 // ============================================================ /// 检测是否为中文环境 fn is_chinese_locale() -> bool { std::env::var("LANG") .or_else(|_| std::env::var("LC_ALL")) .or_else(|_| std::env::var("LC_MESSAGES")) .map(|lang| lang.starts_with("zh")) .unwrap_or(false) } /// 显示迁移错误对话框 /// 返回 true 表示用户选择重试,false 表示用户选择退出 fn show_migration_error_dialog(app: &tauri::AppHandle, error: &str) -> bool { let title = if is_chinese_locale() { "配置迁移失败" } else { "Migration Failed" }; let message = if is_chinese_locale() { format!( "从旧版本迁移配置时发生错误:\n\n{error}\n\n\ 您的数据尚未丢失,旧配置文件仍然保留。\n\ 建议回退到旧版本 CC Switch 以保护数据。\n\n\ 点击「重试」重新尝试迁移\n\ 点击「退出」关闭程序(可回退版本后重新打开)" ) } else { format!( "An error occurred while migrating configuration:\n\n{error}\n\n\ Your data is NOT lost - the old config file is still preserved.\n\ Consider rolling back to an older CC Switch version.\n\n\ Click 'Retry' to attempt migration again\n\ Click 'Exit' to close the program" ) }; let retry_text = if is_chinese_locale() { "重试" } else { "Retry" }; let exit_text = if is_chinese_locale() { "退出" } else { "Exit" }; // 使用 blocking_show 同步等待用户响应 // OkCancelCustom: 第一个按钮(重试)返回 true,第二个按钮(退出)返回 false app.dialog() .message(&message) .title(title) .kind(MessageDialogKind::Error) .buttons(MessageDialogButtons::OkCancelCustom( retry_text.to_string(), exit_text.to_string(), )) .blocking_show() }