mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-05-07 22:28:26 +08:00
* feat: add Rust-side write-through usage cache Introduce an in-memory UsageCache on AppState that the existing usage query commands populate on success. The cache is read-only to the rest of the app today; the next commit consumes it from the tray menu. - New services::usage_cache module with split maps: subscription keyed by AppType, script keyed by (AppType, provider_id). - AppType gains Eq + Hash so it can be used as a HashMap key. - commands::subscription::get_subscription_quota now takes State<AppState> and writes through on success (signature change is invisible to the frontend — Tauri injects State automatically). - commands::provider::queryProviderUsage body extracted into an inner async fn; the public command wraps it with write-through, covering Copilot, coding-plan, balance, and generic script paths uniformly. Cache is in-memory only; auto-query interval and the upcoming tray refresh action rebuild it after restarts. * feat(tray): surface cached usage in the system tray menu Read UsageCache populated by the previous commit and render it in three places, scoped to whatever TRAY_SECTIONS covers (Claude/Codex/Gemini): 1. Inline suffix on each provider submenu item "AnyProvider · 🟢 5h 18% / 7d 23%" 2. Disabled summary row per visible app under "Show Main" "Claude · Anthropic Official · 🟢 5h 18% / 7d 23%" 3. "Refresh all usage" menu item that triggers get_subscription_quota + queryProviderUsage for every applicable provider, then rebuilds the tray menu via the existing refresh_tray_menu path. Color encoding uses emoji (🟢 <70% / 🟠 70-89% / 🔴 ≥90%) since Tauri 2 tray labels are plain text. Missing cache entry leaves the label unchanged — tray never issues network requests when opened. Three new i18n-ready strings live in TrayTexts (en/zh/ja), following the existing pattern for tray text. Closes #2178. * feat(usage): bridge tray UsageCache writes to frontend React Query Why: tray hover triggers backend-only refresh that wrote to UsageCache but never notified the frontend, leaving main UI stale while tray showed fresh numbers. Emit a payload-carrying event after each cache write so React Query can setQueryData directly, keeping both views in sync without duplicate fetches. * fix(tray): skip hidden apps on hover refresh and drop stale disabled-script cache Address P2 findings from automated review on #2184: 1. refresh_all_usage_in_tray now filters TRAY_SECTIONS by settings.visible_apps before scheduling subscription/script queries, matching create_tray_menu and preventing wasted external API calls (and rate-limit/auth-error log noise) for apps the user has hidden. 2. format_usage_suffix only trusts the script cache when provider.meta.usage_script is still enabled; when a script is disabled/removed the cached suffix is now invalidated so the tray label no longer shows stale data indefinitely. * refactor: consolidate codex provider helpers and fix test semantics - Add Provider::is_codex_oauth() and Provider::codex_fast_mode_enabled() to eliminate duplicated meta extraction in claude.rs and stream_check.rs - Fix non-codex-oauth tests to pass codex_fast_mode=false (was true, harmless but semantically misleading) - Remove redundant is_dir() guard after resolve_skill_source_dir already guarantees the returned path is a directory * style: apply cargo fmt * fix(tray): reflect failed refreshes in cache and support Gemini flash-lite Follow-up to the tray usage-display feature addressing review feedback: - Write snapshots for both Ok(success:false) and Err paths in queryProviderUsage / get_subscription_quota so stale success data no longer persists across failed refreshes; the original Err is still returned to the frontend onError handler. - Include gemini_flash_lite tier in the tray summary with label "l". Matches the frontend SubscriptionQuotaFooter and keeps the worst emoji correct when lite is the highest utilization. - Add TIER_GEMINI_PRO / _FLASH / _FLASH_LITE constants in services/subscription.rs and reuse them in classify_gemini_model and sort_order. - Extract Provider::has_usage_script_enabled() to remove the duplicated meta.usage_script chain at two call sites. - Use db.get_provider_by_id in refresh_all_usage_in_tray instead of materialising the full provider map, and parallelise subscription and script futures via futures::future::join. - Narrow refresh_all_usage_in_tray to each section's effective current provider (script if enabled, else subscription when the provider is official). Hover refreshes now issue at most TRAY_SECTIONS.len() outbound requests. - Add 10 unit tests in tray::tests covering Claude/Codex h/w dispatch, Gemini p/f/l dispatch (including lite-only and lite-worst cases), and success/failure guards. --------- Co-authored-by: Jason <farion1231@gmail.com>
87 lines
3.4 KiB
Rust
87 lines
3.4 KiB
Rust
use std::sync::Arc;
|
|
|
|
use cc_switch_lib::{import_provider_from_deeplink, parse_deeplink_url, AppState, Database};
|
|
|
|
#[path = "support.rs"]
|
|
mod support;
|
|
use support::{ensure_test_home, reset_test_fs, test_mutex};
|
|
|
|
#[test]
|
|
fn deeplink_import_claude_provider_persists_to_db() {
|
|
let _guard = test_mutex().lock().expect("acquire test mutex");
|
|
reset_test_fs();
|
|
let _home = ensure_test_home();
|
|
|
|
let url = "ccswitch://v1/import?resource=provider&app=claude&name=DeepLink%20Claude&homepage=https%3A%2F%2Fexample.com&endpoint=https%3A%2F%2Fapi.example.com%2Fv1&apiKey=sk-test-claude-key&model=claude-sonnet-4&icon=claude";
|
|
let request = parse_deeplink_url(url).expect("parse deeplink url");
|
|
|
|
let db = Arc::new(Database::memory().expect("create memory db"));
|
|
let state = AppState::new(db.clone());
|
|
|
|
let provider_id = import_provider_from_deeplink(&state, request.clone())
|
|
.expect("import provider from deeplink");
|
|
|
|
// Verify DB state
|
|
let providers = db.get_all_providers("claude").expect("get providers");
|
|
let provider = providers
|
|
.get(&provider_id)
|
|
.expect("provider created via deeplink");
|
|
|
|
assert_eq!(provider.name, request.name.clone().unwrap());
|
|
assert_eq!(provider.website_url.as_deref(), request.homepage.as_deref());
|
|
assert_eq!(provider.icon.as_deref(), Some("claude"));
|
|
let auth_token = provider
|
|
.settings_config
|
|
.pointer("/env/ANTHROPIC_AUTH_TOKEN")
|
|
.and_then(|v| v.as_str());
|
|
let base_url = provider
|
|
.settings_config
|
|
.pointer("/env/ANTHROPIC_BASE_URL")
|
|
.and_then(|v| v.as_str());
|
|
assert_eq!(auth_token, request.api_key.as_deref());
|
|
assert_eq!(base_url, request.endpoint.as_deref());
|
|
}
|
|
|
|
#[test]
|
|
fn deeplink_import_codex_provider_builds_auth_and_config() {
|
|
let _guard = test_mutex().lock().expect("acquire test mutex");
|
|
reset_test_fs();
|
|
let _home = ensure_test_home();
|
|
|
|
let url = "ccswitch://v1/import?resource=provider&app=codex&name=DeepLink%20Codex&homepage=https%3A%2F%2Fopenai.example&endpoint=https%3A%2F%2Fapi.openai.example%2Fv1&apiKey=sk-test-codex-key&model=gpt-4o&icon=openai";
|
|
let request = parse_deeplink_url(url).expect("parse deeplink url");
|
|
|
|
let db = Arc::new(Database::memory().expect("create memory db"));
|
|
let state = AppState::new(db.clone());
|
|
|
|
let provider_id = import_provider_from_deeplink(&state, request.clone())
|
|
.expect("import provider from deeplink");
|
|
|
|
let providers = db.get_all_providers("codex").expect("get providers");
|
|
let provider = providers
|
|
.get(&provider_id)
|
|
.expect("provider created via deeplink");
|
|
|
|
assert_eq!(provider.name, request.name.clone().unwrap());
|
|
assert_eq!(provider.website_url.as_deref(), request.homepage.as_deref());
|
|
assert_eq!(provider.icon.as_deref(), Some("openai"));
|
|
let auth_value = provider
|
|
.settings_config
|
|
.pointer("/auth/OPENAI_API_KEY")
|
|
.and_then(|v| v.as_str());
|
|
let config_text = provider
|
|
.settings_config
|
|
.get("config")
|
|
.and_then(|v| v.as_str())
|
|
.unwrap_or_default();
|
|
assert_eq!(auth_value, request.api_key.as_deref());
|
|
assert!(
|
|
config_text.contains(request.endpoint.as_deref().unwrap()),
|
|
"config.toml content should contain endpoint"
|
|
);
|
|
assert!(
|
|
config_text.contains("model = \"gpt-4o\""),
|
|
"config.toml content should contain model setting"
|
|
);
|
|
}
|