1use std::collections::HashSet;
4use std::time::Duration;
5
6use chrono::Utc;
7use ratatui::layout::Rect;
8use reqwest::Client;
9
10use crate::cache::DEFAULT_TTL;
11use crate::config::{Config, CustomProviderConfig};
12use crate::error::{AppError, Result};
13use crate::theme::Theme;
14use crate::tui::settings::SettingsRow;
15use crate::vendor::{VendorId, VendorOutcome};
16
17#[derive(Debug, Clone)]
23pub enum TabState {
24 Loading,
25 Ready(Box<ReadyTab>),
26 Error {
27 message: String,
28 plan: Option<String>,
32 },
33}
34
35#[derive(Debug, Clone)]
36pub struct ReadyTab {
37 pub snapshot: crate::usage::VendorSnapshot,
38 pub stale: bool,
39 pub last_error: Option<(u16, String)>,
40 pub fetched_at: Option<chrono::DateTime<chrono::Utc>>,
45 pub display: crate::balance::DisplayPrefs,
51}
52
53#[derive(Debug, Clone, PartialEq, Eq, Hash)]
57pub enum TabSource {
58 Builtin(VendorId),
59 Custom {
60 id: String,
61 name: String,
62 short_name: String,
63 },
64}
65
66impl TabState {
67 pub fn error(message: impl AsRef<str>) -> Self {
68 Self::error_with_plan(message, None)
69 }
70
71 pub fn error_with_plan(message: impl AsRef<str>, plan: Option<String>) -> Self {
72 let plan = plan.and_then(|plan| {
73 let cleaned = crate::display::sanitize_untrusted_field(plan.trim());
74 if cleaned.is_empty() || cleaned.eq_ignore_ascii_case("unknown") {
75 None
76 } else {
77 Some(cleaned)
78 }
79 });
80 Self::Error {
81 message: message.as_ref().to_string(),
82 plan,
83 }
84 }
85}
86
87#[derive(Debug, Clone, PartialEq, Eq, Hash)]
94pub struct TabId {
95 pub source: TabSource,
96 pub account: Option<String>,
97 pub desktop: bool,
98}
99
100impl TabId {
101 pub fn vendor(vendor: VendorId) -> Self {
103 Self {
104 source: TabSource::Builtin(vendor),
105 account: None,
106 desktop: false,
107 }
108 }
109
110 pub fn account(label: impl Into<String>) -> Self {
112 Self::account_for(VendorId::Anthropic, label)
113 }
114
115 pub fn account_for(vendor: VendorId, label: impl Into<String>) -> Self {
117 Self {
118 source: TabSource::Builtin(vendor),
119 account: Some(label.into()),
120 desktop: false,
121 }
122 }
123
124 pub fn desktop_account(label: impl Into<String>) -> Self {
127 Self {
128 source: TabSource::Builtin(VendorId::Anthropic),
129 account: Some(label.into()),
130 desktop: true,
131 }
132 }
133
134 pub fn custom(spec: &CustomProviderConfig) -> Self {
136 Self {
137 source: TabSource::Custom {
138 id: spec.id.clone(),
139 name: spec.name.clone(),
140 short_name: spec.short_name.clone(),
141 },
142 account: None,
143 desktop: false,
144 }
145 }
146
147 pub fn vendor_id(&self) -> Option<VendorId> {
150 match &self.source {
151 TabSource::Builtin(vendor) => Some(*vendor),
152 TabSource::Custom { .. } => None,
153 }
154 }
155}
156
157pub fn tabs_from_config(config: &Config) -> Vec<TabId> {
166 build_tabs(config, &[])
167}
168
169pub fn tabs_with_desktop(config: &Config) -> Vec<TabId> {
173 build_tabs(config, &desktop_profile_labels(config))
174}
175
176fn build_tabs(config: &Config, desktop_labels: &[String]) -> Vec<TabId> {
189 let desktop_set: HashSet<&str> = desktop_labels.iter().map(String::as_str).collect();
190 let mut tabs = Vec::new();
191 for vendor in config.enabled_vendors() {
192 if vendor == VendorId::Anthropic {
193 let accounts: Vec<_> = config
194 .anthropic
195 .all_accounts()
196 .into_iter()
197 .filter(|a| !desktop_set.contains(a.label.as_str()))
198 .collect();
199 if config.anthropic.show_default_account
203 || (accounts.is_empty() && desktop_labels.is_empty())
204 {
205 tabs.push(TabId::vendor(vendor));
206 }
207 for acct in accounts {
208 tabs.push(TabId::account(acct.label));
209 }
210 for label in desktop_labels {
211 tabs.push(TabId::desktop_account(label.clone()));
212 }
213 } else if let Some(accounts) = config.api_key_accounts(vendor) {
214 if config.show_default_api_key_account(vendor) || accounts.is_empty() {
215 tabs.push(TabId::vendor(vendor));
216 }
217 for account in accounts {
218 tabs.push(TabId::account_for(vendor, account.label.clone()));
219 }
220 } else if vendor == VendorId::Openai {
221 if config.openai.show_default_account || config.openai.accounts.is_empty() {
222 tabs.push(TabId::vendor(vendor));
223 }
224 for account in &config.openai.accounts {
225 tabs.push(TabId::account_for(vendor, account.label.clone()));
226 }
227 } else {
228 tabs.push(TabId::vendor(vendor));
229 }
230 }
231 for spec in config.enabled_custom() {
233 tabs.push(TabId::custom(spec));
234 }
235 tabs
236}
237
238#[cfg(target_os = "macos")]
242fn desktop_profile_labels(config: &Config) -> Vec<String> {
243 let Ok(paths) = crate::claude_desktop::Paths::resolve(&config.anthropic) else {
244 return Vec::new();
245 };
246 if !paths.available() {
247 return Vec::new();
248 }
249 crate::claude_desktop::load_profiles(&paths.profiles_dir)
250 .into_iter()
251 .filter(|p| p.has_credentials)
252 .map(|p| p.label)
253 .collect()
254}
255
256#[cfg(not(target_os = "macos"))]
257fn desktop_profile_labels(_config: &Config) -> Vec<String> {
258 Vec::new()
259}
260
261#[derive(Debug, Default, Clone)]
266pub struct HitTargets {
267 pub nav_entries: Vec<(NavTarget, Rect)>,
269 pub footer_actions: Vec<(FooterAction, Rect)>,
271 pub settings_rows: Vec<(SettingsRow, Rect)>,
274}
275
276#[derive(Debug, Clone, Copy, PartialEq, Eq)]
278pub enum NavTarget {
279 Overview,
280 Tab(usize),
281}
282
283#[derive(Debug, Clone, Copy, PartialEq, Eq)]
285pub enum FooterAction {
286 Refresh,
287 RefreshAll,
288 Settings,
289 Quit,
290}
291
292#[derive(Debug)]
293pub struct App {
294 pub tabs_meta: Vec<TabId>,
295 pub active: usize,
296 pub tabs: Vec<TabState>,
297 refreshing_tabs: HashSet<TabId>,
300 pub tab_generation: u64,
304 pub overview: bool,
307 pub overview_vendors: Option<Vec<VendorId>>,
309 pub theme: Theme,
310 pub quit: bool,
311 pub settings: Option<crate::tui::settings::SettingsState>,
313 pub context_enabled: bool,
316 pub context_generation: u64,
319 pub context: Option<crate::tui::context::ContextState>,
321 pub vendor_box: crate::config::VendorBoxStyle,
323 pub hit: std::rc::Rc<std::cell::RefCell<HitTargets>>,
325}
326
327impl App {
328 pub fn new(tabs_meta: Vec<TabId>) -> Self {
329 Self::with_theme(tabs_meta, Theme::default().merged_with_omarchy())
332 }
333
334 pub fn with_theme(tabs_meta: Vec<TabId>, theme: Theme) -> Self {
340 let n = tabs_meta.len();
341 Self {
342 tabs_meta,
343 active: 0,
344 tabs: vec![TabState::Loading; n],
345 refreshing_tabs: HashSet::new(),
346 tab_generation: 0,
347 overview: false,
348 overview_vendors: None,
349 theme,
350 quit: false,
351 settings: None,
352 context_enabled: false,
353 context_generation: 0,
354 context: None,
355 vendor_box: crate::config::VendorBoxStyle::Sidebar,
356 hit: std::rc::Rc::new(std::cell::RefCell::new(HitTargets::default())),
357 }
358 }
359
360 pub fn new_with_primary(tabs_meta: Vec<TabId>, primary: Option<VendorId>) -> Self {
364 let mut app = Self::new(tabs_meta);
365 if primary.is_some() {
368 app.select_primary(primary);
369 } else {
370 app.overview = true;
371 }
372 app
373 }
374
375 pub fn active_tab_id(&self) -> Option<&TabId> {
376 self.tabs_meta.get(self.active)
377 }
378
379 pub fn active_vendor(&self) -> Option<VendorId> {
380 self.tabs_meta.get(self.active).and_then(TabId::vendor_id)
381 }
382
383 pub fn set_tabs(&mut self, tabs_meta: Vec<TabId>) {
390 let selected = self.active_tab_id().cloned();
391 let fallback = self.active.min(tabs_meta.len().saturating_sub(1));
392 self.tab_generation = self.tab_generation.wrapping_add(1);
393 self.active = selected
394 .as_ref()
395 .and_then(|tab| tabs_meta.iter().position(|candidate| candidate == tab))
396 .unwrap_or(fallback);
397 self.tabs = vec![TabState::Loading; tabs_meta.len()];
398 self.tabs_meta = tabs_meta;
399 self.refreshing_tabs.clear();
400 }
401
402 pub fn begin_refresh(&mut self, tab: &TabId) -> bool {
406 let Some(index) = self.tabs_meta.iter().position(|current| current == tab) else {
407 return false;
408 };
409 if !self.refreshing_tabs.insert(tab.clone()) {
410 return false;
411 }
412 if !matches!(self.tabs[index], TabState::Ready(_)) {
413 self.tabs[index] = TabState::Loading;
414 }
415 true
416 }
417
418 pub fn is_refreshing(&self, tab: &TabId) -> bool {
419 self.refreshing_tabs.contains(tab)
420 }
421
422 pub fn tab_is_refreshing(&self, index: usize) -> bool {
423 self.tabs_meta
424 .get(index)
425 .is_some_and(|tab| self.is_refreshing(tab))
426 }
427
428 pub fn apply_refresh(&mut self, generation: u64, tab: &TabId, state: TabState) -> bool {
433 if generation != self.tab_generation {
434 return false;
435 }
436 let Some(index) = self.tabs_meta.iter().position(|current| current == tab) else {
437 return false;
438 };
439 let was_refreshing = self.refreshing_tabs.remove(tab);
440 if was_refreshing
444 && let TabState::Ready(ready) = &mut self.tabs[index]
445 && let TabState::Error { message, .. } = state
446 {
447 ready.stale = true;
448 ready.last_error = Some((0, message));
449 } else {
450 self.tabs[index] = state;
451 }
452 true
453 }
454
455 pub fn select_primary(&mut self, primary: Option<VendorId>) {
458 if let Some(p) = primary
459 && let Some(idx) = self.tabs_meta.iter().position(|t| t.vendor_id() == Some(p))
460 {
461 self.active = idx;
462 self.overview = false;
463 }
464 }
465
466 pub fn next_tab(&mut self) {
469 if self.overview {
470 if !self.tabs_meta.is_empty() {
471 self.overview = false;
472 self.active = 0;
473 }
474 } else if self.active + 1 < self.tabs_meta.len() {
475 self.active += 1;
476 } else {
477 self.overview = true;
478 }
479 }
480
481 pub fn prev_tab(&mut self) {
482 if self.overview {
483 if !self.tabs_meta.is_empty() {
484 self.overview = false;
485 self.active = self.tabs_meta.len() - 1;
486 }
487 } else if self.active > 0 {
488 self.active -= 1;
489 } else {
490 self.overview = true;
491 }
492 }
493
494 pub fn select_overview(&mut self) {
496 self.overview = true;
497 }
498
499 pub fn select_tab(&mut self, index: usize) {
501 if index < self.tabs_meta.len() {
502 self.active = index;
503 self.overview = false;
504 }
505 }
506
507 pub fn nav_from_target(&mut self, target: NavTarget) {
509 match target {
510 NavTarget::Overview => self.select_overview(),
511 NavTarget::Tab(index) => self.select_tab(index),
512 }
513 }
514
515 pub fn overview_tabs(&self) -> Vec<usize> {
520 match &self.overview_vendors {
521 None => (0..self.tabs_meta.len()).collect(),
522 Some(wanted) => wanted
523 .iter()
524 .flat_map(|v| {
525 self.tabs_meta
526 .iter()
527 .enumerate()
528 .filter(move |(_, t)| t.vendor_id() == Some(*v))
529 .map(|(i, _)| i)
530 })
531 .collect(),
532 }
533 }
534}
535
536pub async fn refresh_one(client: &Client, config: &Config, tab: &TabId) -> TabState {
538 match build_outcome(client, config, tab).await {
539 Ok(outcome) => {
540 let now = Utc::now();
545 let off_the_wire = outcome.off_the_wire();
546 let fetched_at = outcome
547 .cache_age
548 .map(|age| now - chrono::Duration::from_std(age).unwrap_or_default());
549 let state = TabState::Ready(Box::new(ReadyTab {
550 snapshot: outcome.snapshot,
551 stale: outcome.stale,
552 last_error: outcome.last_error.map(|(code, message)| {
553 (code, crate::display::sanitize_untrusted_field(&message))
554 }),
555 fetched_at,
556 display: match &tab.source {
557 TabSource::Builtin(vendor) => config.display_prefs(*vendor),
558 TabSource::Custom { .. } => crate::balance::DisplayPrefs::default(),
561 },
562 }));
563 if off_the_wire
569 && config.notifications.enabled
570 && let Some(input) = crate::notify::RefreshInput::from_tab(tab, &state, now)
571 {
572 let _ = crate::notify::run(input, config.notifications.threshold).await;
573 }
574 state
575 }
576 Err(e) => TabState::error_with_plan(
577 crate::display::sanitize_untrusted_field(&e.user_message()),
578 e.plan().map(str::to_string),
579 ),
580 }
581}
582
583fn api_key_and_cache(
587 config: &Config,
588 vendor: VendorId,
589 account: Option<&str>,
590) -> Result<(String, crate::cache::Cache)> {
591 let api_key = config.resolve_account_api_key_for(vendor, account)?;
592 let cache = match account {
593 Some(label) => crate::cache::Cache::for_vendor_account(vendor.slug(), label)?,
594 None => crate::cache::Cache::for_vendor(vendor.slug())?,
595 };
596 Ok((api_key, cache))
597}
598
599async fn build_outcome(client: &Client, config: &Config, tab: &TabId) -> Result<VendorOutcome> {
600 let vendor = match &tab.source {
601 TabSource::Builtin(vendor) => *vendor,
602 TabSource::Custom { id, .. } => {
603 let spec = config.custom_by_id(id).ok_or_else(|| {
606 AppError::Other(format!("custom provider {id} is not configured"))
607 })?;
608 let api_key = spec.resolve_api_key()?;
609 let cache = crate::cache::Cache::for_vendor_account("custom", id)?;
610 let outcome =
611 crate::custom::fetch_snapshot(client, spec, &api_key, &cache, spec.cache_ttl())
612 .await?;
613 return Ok(outcome.into());
614 }
615 };
616 match vendor {
617 VendorId::Anthropic => {
618 let (creds_target, cache) = match tab.account.as_deref() {
624 Some(label) if tab.desktop => {
625 crate::anthropic::desktop_creds::account_target(config, label)?
626 }
627 Some(label) => config.anthropic.account_target(label)?,
628 None => {
629 let target = match config.anthropic.credentials_path.clone() {
630 Some(p) => crate::anthropic::creds::CredsTarget::Explicit(p),
631 None => crate::anthropic::creds::CredsTarget::Default(
632 crate::anthropic::creds::default_path().unwrap_or_default(),
633 ),
634 };
635 (target, crate::cache::Cache::for_vendor("anthropic")?)
636 }
637 };
638 let endpoints = crate::anthropic::fetch::Endpoints::default();
639 let outcome = crate::anthropic::fetch_snapshot(
640 client,
641 &creds_target,
642 &cache,
643 &endpoints,
644 DEFAULT_TTL,
645 )
646 .await?;
647 Ok(outcome.map(crate::usage::VendorSnapshot::Anthropic))
648 }
649 VendorId::AnthropicApi => {
650 let key = crate::config::resolve_api_key(
651 "Anthropic_API",
652 &config.anthropic_api.api_key_env,
653 config.anthropic_api.api_key.as_deref(),
654 )?;
655 let cache = crate::cache::Cache::for_vendor("anthropic_api")?;
656 let endpoints = crate::anthropic_api::fetch::Endpoints::default();
657 let outcome = crate::anthropic_api::fetch_snapshot(
658 client,
659 &key,
660 &cache,
661 &endpoints,
662 DEFAULT_TTL,
663 config.anthropic_api.monthly_limit,
664 )
665 .await?;
666 Ok(outcome.into())
667 }
668 VendorId::Openrouter => {
669 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
670 let management_key = config.openrouter_management_key(tab.account.as_deref());
671 let endpoints = crate::openrouter::fetch::Endpoints::default();
672 let outcome = crate::openrouter::fetch_snapshot(
673 client,
674 &api_key,
675 management_key.as_deref(),
676 &cache,
677 &endpoints,
678 DEFAULT_TTL,
679 )
680 .await?;
681 Ok(outcome.into())
682 }
683 VendorId::Zai => {
684 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
685 let endpoints = crate::zai::fetch::Endpoints::default();
686 let outcome = crate::zai::fetch_snapshot(
687 client,
688 &api_key,
689 &cache,
690 &endpoints,
691 DEFAULT_TTL,
692 config.zai.plan_tier.as_deref(),
693 )
694 .await?;
695 Ok(outcome.into())
696 }
697 VendorId::Openai => {
698 let label = tab.account.as_deref();
699 let cache = match label {
700 Some(label) => crate::cache::Cache::for_vendor_account("openai", label)?,
701 None => crate::cache::Cache::for_vendor("openai")?,
702 };
703 let route = || config.openai.fetch_auth_path(label);
704 let endpoints = crate::openai::fetch::Endpoints::default();
705 let outcome = crate::openai::fetch_snapshot_routed(
706 client,
707 route,
708 &cache,
709 &endpoints,
710 DEFAULT_TTL,
711 )
712 .await?;
713 Ok(outcome.into())
714 }
715 VendorId::Copilot => {
716 let token = config.copilot.resolve_token()?;
717 let cache = crate::cache::Cache::for_vendor("copilot")?;
718 let endpoints = crate::copilot::fetch::Endpoints::default();
719 let outcome =
720 crate::copilot::fetch_snapshot(client, &token, &cache, &endpoints, DEFAULT_TTL)
721 .await?;
722 Ok(outcome.into())
723 }
724 VendorId::Deepseek => {
725 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
726 let endpoints = crate::deepseek::fetch::Endpoints::default();
727 let outcome =
728 crate::deepseek::fetch_snapshot(client, &api_key, &cache, &endpoints, DEFAULT_TTL)
729 .await?;
730 Ok(outcome.into())
731 }
732 VendorId::Deepinfra => {
733 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
734 let endpoints = crate::deepinfra::fetch::Endpoints::default();
735 let outcome = crate::deepinfra::fetch::fetch_snapshot(
736 client,
737 &api_key,
738 &cache,
739 &endpoints,
740 DEFAULT_TTL,
741 )
742 .await?;
743 Ok(outcome.into())
744 }
745 VendorId::Kimi => {
746 let (auth, endpoints) = crate::kimi::resolve_auth(&config.kimi)?;
747 let cache = crate::cache::Cache::for_vendor("kimi")?;
748 let outcome = crate::kimi::fetch::fetch_snapshot_with_auth(
749 client,
750 &auth,
751 &cache,
752 &endpoints,
753 DEFAULT_TTL,
754 )
755 .await?;
756 Ok(outcome.into())
757 }
758 VendorId::Kilo => {
759 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
760 let endpoints = crate::kilo::fetch::Endpoints::default();
761 let outcome = crate::kilo::fetch_snapshot(
762 client,
763 &api_key,
764 &cache,
765 &endpoints,
766 DEFAULT_TTL,
767 config.kilo.organization_id.as_deref(),
768 )
769 .await?;
770 Ok(outcome.into())
771 }
772 VendorId::Novita => {
773 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
774 let endpoints = crate::novita::fetch::Endpoints::default();
775 let outcome =
776 crate::novita::fetch_snapshot(client, &api_key, &cache, &endpoints, DEFAULT_TTL)
777 .await?;
778 Ok(outcome.into())
779 }
780 VendorId::Moonshot => {
781 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
782 let (endpoints, currency) =
783 crate::moonshot::fetch::Endpoints::for_region(&config.moonshot.region);
784 let outcome = crate::moonshot::fetch_snapshot(
785 client,
786 &api_key,
787 &cache,
788 &endpoints,
789 DEFAULT_TTL,
790 currency,
791 )
792 .await?;
793 Ok(outcome.into())
794 }
795 VendorId::Grok => {
796 let (key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
797 let endpoints = crate::grok::fetch::Endpoints::default();
798 let outcome = crate::grok::fetch_snapshot(
799 client,
800 &key,
801 &cache,
802 &endpoints,
803 DEFAULT_TTL,
804 config.grok.team_id.as_deref(),
805 )
806 .await?;
807 Ok(outcome.into())
808 }
809 VendorId::Supergrok => {
810 let cache = crate::cache::Cache::for_vendor("supergrok")?;
811 let scope_paths = crate::supergrok::scope::ScopePaths::with_overrides(
812 config.supergrok.auth_path.as_deref(),
813 config.supergrok.config_path.as_deref(),
814 )?;
815 let outcome = crate::supergrok::fetch_snapshot(
816 &config.supergrok.grok_binary,
817 &scope_paths,
818 &cache,
819 DEFAULT_TTL,
820 )
821 .await?;
822 Ok(outcome.into())
823 }
824 VendorId::Antigravity => {
825 let cache = crate::cache::Cache::for_vendor("antigravity")?;
828 let oauth = crate::antigravity::cloud::OauthClient::from_config(
829 config.antigravity.oauth_client_id.as_deref(),
830 config.antigravity.oauth_client_secret.as_deref(),
831 );
832 let outcome =
833 crate::antigravity::fetch_snapshot(client, &cache, DEFAULT_TTL, oauth.as_ref())
834 .await?;
835 Ok(outcome.into())
836 }
837 VendorId::Grokbot => {
838 let creds = crate::grokbot::resolve_credentials(&config.grokbot)?;
841 let cache = crate::cache::Cache::for_vendor("grokbot")?;
842 let endpoints = crate::grokbot::fetch::Endpoints::default();
843 let outcome = crate::grokbot::fetch::fetch_snapshot_with(
844 client,
845 &creds,
846 &cache,
847 &endpoints,
848 DEFAULT_TTL,
849 )
850 .await?;
851 Ok(outcome.into())
852 }
853 VendorId::ModelStudio => {
854 let creds = crate::modelstudio::resolve_credentials(&config.modelstudio)?;
857 let cache = crate::cache::Cache::for_vendor("modelstudio")?;
858 let endpoints =
859 crate::modelstudio::fetch::Endpoints::for_gateway(creds.region, creds.site);
860 let outcome = crate::modelstudio::fetch_snapshot_with(
861 client,
862 &creds,
863 &cache,
864 &endpoints,
865 DEFAULT_TTL,
866 )
867 .await?;
868 Ok(outcome.into())
869 }
870 VendorId::Minimax => {
871 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
872 let endpoints = crate::minimax::fetch::Endpoints::for_region(&config.minimax.region);
873 let outcome =
874 crate::minimax::fetch_snapshot(client, &api_key, &cache, &endpoints, DEFAULT_TTL)
875 .await?;
876 Ok(outcome.into())
877 }
878 VendorId::Cursor => {
879 let cache = crate::cache::Cache::for_vendor("cursor")?;
880 let db_path = config
881 .cursor
882 .db_path
883 .clone()
884 .map(Ok)
885 .unwrap_or_else(crate::cursor::db::default_db_path)?;
886 let agent_auth_path = config
887 .cursor
888 .agent_auth_path
889 .clone()
890 .map(Ok)
891 .unwrap_or_else(crate::cursor::db::default_agent_auth_path)?;
892 let endpoints = crate::cursor::fetch::Endpoints::default();
893 let outcome = crate::cursor::fetch_snapshot(
894 client,
895 &db_path,
896 &agent_auth_path,
897 &cache,
898 &endpoints,
899 DEFAULT_TTL,
900 )
901 .await?;
902 Ok(outcome.into())
903 }
904 VendorId::Kiro => {
905 let cache = crate::cache::Cache::for_vendor("kiro")?;
906 let db_path = config
907 .kiro
908 .db_path
909 .clone()
910 .map(Ok)
911 .unwrap_or_else(crate::kiro::db::default_db_path)?;
912 let outcome =
913 crate::kiro::fetch_snapshot(client, &db_path, &cache, DEFAULT_TTL).await?;
914 Ok(outcome.into())
915 }
916 VendorId::NousResearch => {
917 let store = crate::nous::credentials::CredentialStore::default();
918 let endpoints = crate::nous::fetch::Endpoints::default();
919 let account = crate::nous::fetch::fetch_account_with_refresh(
920 client,
921 &store,
922 &endpoints,
923 Utc::now(),
924 )
925 .await?;
926 Ok(crate::outcome::Outcome::fresh(
928 crate::usage::VendorSnapshot::NousResearch(account),
929 ))
930 }
931 VendorId::OpenCodeGo => {
932 let api_key = crate::config::resolve_api_key(
933 "OpenCode Go",
934 &config.opencode_go.api_key_env,
935 config.opencode_go.api_key.as_deref(),
936 )?;
937 let cache = crate::cache::Cache::for_vendor("opencode-go")?;
938 let endpoints = crate::opencode_go::fetch::Endpoints::default();
939 let outcome = crate::opencode_go::fetch::fetch_snapshot(
940 client,
941 &api_key,
942 &cache,
943 &endpoints,
944 DEFAULT_TTL,
945 )
946 .await?;
947 Ok(outcome.into())
948 }
949 VendorId::CommandCode => {
950 let credential =
951 crate::commandcode::creds::resolve(config.commandcode.auth_paths.as_deref())?;
952 let cache = crate::cache::Cache::for_vendor("commandcode")?;
953 let endpoints = crate::commandcode::fetch::Endpoints::default();
954 let outcome = crate::commandcode::fetch::fetch_snapshot(
955 client,
956 &credential.token,
957 &cache,
958 &endpoints,
959 DEFAULT_TTL,
960 )
961 .await?;
962 Ok(outcome.into())
963 }
964 VendorId::Ollama => {
965 let api_key = crate::config::resolve_api_key(
966 "Ollama",
967 &config.ollama.api_key_env,
968 config.ollama.api_key.as_deref(),
969 )?;
970 let cache = crate::cache::Cache::for_vendor("ollama")?;
971 let endpoints = crate::ollama::fetch::Endpoints::default();
972 let outcome = crate::ollama::fetch_snapshot(
973 client,
974 &api_key,
975 &config.ollama.plan,
976 &cache,
977 &endpoints,
978 DEFAULT_TTL,
979 )
980 .await?;
981 Ok(outcome.into())
982 }
983 VendorId::OrcaRouter => {
984 let (api_key, cache) = api_key_and_cache(config, vendor, tab.account.as_deref())?;
985 let endpoints = crate::orcarouter::fetch::Endpoints::default();
986 let outcome = crate::orcarouter::fetch_snapshot(
987 client,
988 &api_key,
989 &cache,
990 &endpoints,
991 DEFAULT_TTL,
992 )
993 .await?;
994 Ok(outcome.into())
995 }
996 }
997}
998
999pub const REFRESH_INTERVAL: Duration = Duration::from_secs(60);
1002
1003pub const ANTHROPIC_REFRESH_STAGGER: Duration = Duration::from_millis(800);
1010
1011pub fn refresh_stagger(tabs: &[TabId], step: Duration) -> Vec<Duration> {
1017 let mut anthropic_seen: u32 = 0;
1018 tabs.iter()
1019 .map(|tab| {
1020 if tab.vendor_id() == Some(VendorId::Anthropic) {
1021 let delay = step * anthropic_seen;
1022 anthropic_seen += 1;
1023 delay
1024 } else {
1025 Duration::ZERO
1026 }
1027 })
1028 .collect()
1029}
1030
1031#[cfg(test)]
1032mod tests {
1033 use super::*;
1034 use chrono::TimeZone;
1035
1036 #[test]
1040 fn refresh_stagger_spaces_out_anthropic_tabs_only() {
1041 let step = Duration::from_millis(800);
1042 let tabs = vec![
1043 TabId::vendor(VendorId::Anthropic), TabId::account("work"),
1045 TabId::account("personal"),
1046 TabId::vendor(VendorId::Openai),
1047 TabId::vendor(VendorId::Zai),
1048 ];
1049 let delays = refresh_stagger(&tabs, step);
1050 assert_eq!(
1051 delays,
1052 vec![
1053 Duration::ZERO, step, step * 2, Duration::ZERO, Duration::ZERO, ]
1059 );
1060 }
1061
1062 #[test]
1063 fn refresh_stagger_is_a_noop_without_anthropic_accounts() {
1064 let tabs = vec![
1066 TabId::vendor(VendorId::Anthropic),
1067 TabId::vendor(VendorId::Openrouter),
1068 ];
1069 assert!(
1070 refresh_stagger(&tabs, Duration::from_millis(800))
1071 .iter()
1072 .all(|d| d.is_zero())
1073 );
1074 }
1075
1076 #[test]
1077 fn select_primary_moves_to_enabled_vendor() {
1078 let mut app = App::with_theme(
1079 vec![
1080 TabId::vendor(VendorId::Anthropic),
1081 TabId::vendor(VendorId::Openrouter),
1082 ],
1083 Theme::default(),
1084 );
1085 app.select_primary(Some(VendorId::Openrouter));
1086 assert_eq!(app.active_vendor(), Some(VendorId::Openrouter));
1087 }
1088
1089 #[test]
1090 fn select_primary_ignores_disabled_vendor() {
1091 let mut app = App::with_theme(vec![TabId::vendor(VendorId::Anthropic)], Theme::default());
1092 app.select_primary(Some(VendorId::Openai));
1093 assert_eq!(app.active_vendor(), Some(VendorId::Anthropic));
1094 }
1095
1096 #[test]
1097 fn nav_ring_wraps_through_the_overview_at_both_ends() {
1098 let mut app = App::with_theme(
1099 vec![
1100 TabId::vendor(VendorId::Anthropic),
1101 TabId::vendor(VendorId::Openai),
1102 ],
1103 Theme::default(),
1104 );
1105 app.overview = true;
1106
1107 app.next_tab(); assert!(!app.overview);
1109 assert_eq!(app.active, 0);
1110 app.next_tab();
1111 assert_eq!(app.active, 1);
1112 app.next_tab(); assert!(app.overview);
1114
1115 app.prev_tab(); assert!(!app.overview);
1117 assert_eq!(app.active, 1);
1118 app.prev_tab();
1119 assert_eq!(app.active, 0);
1120 app.prev_tab(); assert!(app.overview);
1122 }
1123
1124 #[test]
1125 fn mouse_nav_targets_select_overview_and_tabs() {
1126 let mut app = App::with_theme(
1127 vec![
1128 TabId::vendor(VendorId::Anthropic),
1129 TabId::vendor(VendorId::Openai),
1130 ],
1131 Theme::default(),
1132 );
1133 app.overview = true;
1134 app.nav_from_target(NavTarget::Tab(1));
1135 assert!(!app.overview);
1136 assert_eq!(app.active, 1);
1137 app.nav_from_target(NavTarget::Overview);
1138 assert!(app.overview);
1139 app.nav_from_target(NavTarget::Tab(99));
1141 assert!(app.overview);
1142 }
1143
1144 #[test]
1145 fn hit_targets_default_to_empty() {
1146 let app = App::with_theme(Vec::new(), Theme::default());
1147 assert!(app.hit.borrow().nav_entries.is_empty());
1148 assert!(app.hit.borrow().footer_actions.is_empty());
1149 assert!(app.hit.borrow().settings_rows.is_empty());
1150 }
1151
1152 #[test]
1153 fn overview_tabs_defaults_to_all_and_honors_the_config_filter() {
1154 let mut app = App::with_theme(
1155 vec![
1156 TabId::vendor(VendorId::Anthropic),
1157 TabId::vendor(VendorId::Openai),
1158 TabId::vendor(VendorId::Zai),
1159 ],
1160 Theme::default(),
1161 );
1162 assert_eq!(app.overview_tabs(), vec![0, 1, 2]);
1163
1164 app.overview_vendors = Some(vec![VendorId::Zai, VendorId::Anthropic]);
1166 assert_eq!(app.overview_tabs(), vec![2, 0]);
1167
1168 app.overview_vendors = Some(vec![VendorId::Grok, VendorId::Openai]);
1170 assert_eq!(app.overview_tabs(), vec![1]);
1171 }
1172
1173 fn config_with_accounts(labels: &[&str]) -> Config {
1174 let mut config = Config::default();
1175 config.openai.enabled = false;
1178 config.zai.enabled = false;
1179 config.openrouter.enabled = false;
1180 config.commandcode.enabled = false;
1181 config.anthropic.accounts = labels
1182 .iter()
1183 .map(|l| crate::config::AnthropicAccount {
1184 label: (*l).to_string(),
1185 credentials_path: format!("/creds/{l}.json").into(),
1186 })
1187 .collect();
1188 config
1189 }
1190
1191 #[test]
1192 fn show_default_account_false_hides_the_unnamed_claude_tab() {
1193 let mut config = config_with_accounts(&["work", "personal"]);
1196 config.anthropic.show_default_account = false;
1197 assert_eq!(
1198 tabs_from_config(&config),
1199 vec![TabId::account("work"), TabId::account("personal")]
1200 );
1201
1202 let mut empty = Config::default();
1205 empty.openai.enabled = false;
1206 empty.zai.enabled = false;
1207 empty.openrouter.enabled = false;
1208 empty.commandcode.enabled = false;
1209 empty.anthropic.show_default_account = false;
1210 assert_eq!(
1211 tabs_from_config(&empty),
1212 vec![TabId::vendor(VendorId::Anthropic)]
1213 );
1214 }
1215
1216 #[test]
1217 fn tabs_expand_anthropic_accounts_after_default() {
1218 let tabs = tabs_from_config(&config_with_accounts(&["work", "personal"]));
1220 assert_eq!(
1221 tabs,
1222 vec![
1223 TabId::vendor(VendorId::Anthropic),
1224 TabId::account("work"),
1225 TabId::account("personal"),
1226 ]
1227 );
1228 }
1229
1230 #[test]
1231 fn tabs_without_accounts_are_just_enabled_vendors() {
1232 let config = Config::default();
1234 let tabs = tabs_from_config(&config);
1235 let vendors: Vec<VendorId> = tabs.iter().filter_map(TabId::vendor_id).collect();
1236 assert_eq!(vendors, config.enabled_vendors());
1237 assert!(tabs.iter().all(|t| t.account.is_none()));
1238 }
1239
1240 #[test]
1241 fn tabs_expand_openrouter_accounts_without_changing_other_vendors() {
1242 let mut config = Config::default();
1243 config.anthropic.enabled = false;
1244 config.openai.enabled = false;
1245 config.zai.enabled = false;
1246 config.commandcode.enabled = false;
1247 config.openrouter.accounts = vec![
1248 crate::config::ApiKeyAccount {
1249 label: "work".into(),
1250 api_key_env: Some("OPENROUTER_WORK_API_KEY".into()),
1251 api_key: None,
1252 management_api_key_env: None,
1253 },
1254 crate::config::ApiKeyAccount {
1255 label: "personal".into(),
1256 api_key_env: None,
1257 api_key: Some("personal-key".into()),
1258 management_api_key_env: None,
1259 },
1260 ];
1261 assert_eq!(
1262 tabs_from_config(&config),
1263 vec![
1264 TabId::vendor(VendorId::Openrouter),
1265 TabId::account_for(VendorId::Openrouter, "work"),
1266 TabId::account_for(VendorId::Openrouter, "personal"),
1267 ]
1268 );
1269 }
1270
1271 #[test]
1272 fn openai_named_accounts_get_their_own_tabs_after_the_default() {
1273 let mut config = Config::default();
1274 config.anthropic.enabled = false;
1275 config.zai.enabled = false;
1276 config.openrouter.enabled = false;
1277 config.commandcode.enabled = false;
1278 config.openai.accounts.push(crate::config::OpenAiAccount {
1279 label: "work".into(),
1280 codex_auth_path: "/tmp/codex-work/auth.json".into(),
1281 });
1282 assert_eq!(
1283 tabs_from_config(&config),
1284 vec![
1285 TabId::vendor(VendorId::Openai),
1286 TabId::account_for(VendorId::Openai, "work"),
1287 ]
1288 );
1289 }
1290
1291 #[test]
1292 fn openrouter_can_hide_default_only_when_named_accounts_exist() {
1293 let mut config = Config::default();
1294 config.anthropic.enabled = false;
1295 config.openai.enabled = false;
1296 config.zai.enabled = false;
1297 config.commandcode.enabled = false;
1298 config.openrouter.show_default_account = false;
1299 assert_eq!(
1300 tabs_from_config(&config),
1301 vec![TabId::vendor(VendorId::Openrouter)]
1302 );
1303
1304 config
1305 .openrouter
1306 .accounts
1307 .push(crate::config::ApiKeyAccount {
1308 label: "work".into(),
1309 api_key_env: Some("OPENROUTER_WORK_API_KEY".into()),
1310 api_key: None,
1311 management_api_key_env: None,
1312 });
1313 assert_eq!(
1314 tabs_from_config(&config),
1315 vec![TabId::account_for(VendorId::Openrouter, "work")]
1316 );
1317 }
1318
1319 #[test]
1320 fn every_api_key_account_vendor_fans_out_like_openrouter() {
1321 let config_for = |vendor: VendorId, show_default: bool| -> Config {
1322 let section = vendor.config_section();
1323 let mut text: String = ["anthropic", "openai", "commandcode", "zai", "openrouter"]
1326 .into_iter()
1327 .filter(|other| *other != section)
1328 .map(|other| format!("[{other}]\nenabled = false\n"))
1329 .collect();
1330 text.push_str(&format!(
1331 "[{section}]\nenabled = true\nshow_default_account = {show_default}\n\
1332 [[{section}.accounts]]\nlabel = \"work\"\napi_key = \"k1\"\n\
1333 [[{section}.accounts]]\nlabel = \"personal\"\napi_key = \"k2\"\n"
1334 ));
1335 let config: Config = toml::from_str(&text).unwrap();
1336 config.validate().unwrap();
1337 config
1338 };
1339 for vendor in Config::API_KEY_ACCOUNT_VENDORS {
1340 let named = vec![
1341 TabId::account_for(vendor, "work"),
1342 TabId::account_for(vendor, "personal"),
1343 ];
1344 let mut with_default = vec![TabId::vendor(vendor)];
1345 with_default.extend(named.clone());
1346 assert_eq!(
1347 tabs_from_config(&config_for(vendor, true)),
1348 with_default,
1349 "{vendor:?}"
1350 );
1351 assert_eq!(
1352 tabs_from_config(&config_for(vendor, false)),
1353 named,
1354 "{vendor:?}"
1355 );
1356 }
1357 }
1358
1359 #[test]
1360 fn codex_can_hide_default_only_when_named_accounts_exist() {
1361 let mut config = Config::default();
1362 config.anthropic.enabled = false;
1363 config.zai.enabled = false;
1364 config.openrouter.enabled = false;
1365 config.commandcode.enabled = false;
1366 config.openai.show_default_account = false;
1367 assert_eq!(
1368 tabs_from_config(&config),
1369 vec![TabId::vendor(VendorId::Openai)]
1370 );
1371
1372 config.openai.accounts.push(crate::config::OpenAiAccount {
1373 label: "work".into(),
1374 codex_auth_path: "/tmp/codex-work/auth.json".into(),
1375 });
1376 assert_eq!(
1377 tabs_from_config(&config),
1378 vec![TabId::account_for(VendorId::Openai, "work")]
1379 );
1380 }
1381
1382 #[test]
1383 fn tabs_include_accounts_auto_discovered_from_accounts_dir() {
1384 let td = tempfile::tempdir().unwrap();
1387 for label in ["work", "personal"] {
1388 let dir = td.path().join(label);
1389 std::fs::create_dir_all(&dir).unwrap();
1390 std::fs::write(dir.join(".credentials.json"), "{}").unwrap();
1391 }
1392 let mut config = Config::default();
1393 config.openai.enabled = false;
1394 config.zai.enabled = false;
1395 config.openrouter.enabled = false;
1396 config.commandcode.enabled = false;
1397 config.anthropic.accounts_dir = Some(td.path().to_path_buf());
1398
1399 let tabs = tabs_from_config(&config);
1400 assert_eq!(
1401 tabs,
1402 vec![
1403 TabId::vendor(VendorId::Anthropic),
1404 TabId::account("personal"), TabId::account("work"),
1406 ]
1407 );
1408 }
1409
1410 #[test]
1411 fn desktop_labels_become_account_tabs_after_cli_accounts() {
1412 let config = config_with_accounts(&["work"]);
1414 let tabs = build_tabs(&config, &["gmail".into(), "hotmail".into()]);
1415 assert_eq!(
1416 tabs,
1417 vec![
1418 TabId::vendor(VendorId::Anthropic),
1419 TabId::account("work"),
1420 TabId::desktop_account("gmail"),
1421 TabId::desktop_account("hotmail"),
1422 ]
1423 );
1424 }
1425
1426 #[test]
1427 fn a_desktop_profile_wins_a_label_collision_with_a_cli_account() {
1428 let config = config_with_accounts(&["gmail", "work"]);
1433 let tabs = build_tabs(&config, &["gmail".into(), "hotmail".into()]);
1434 assert_eq!(
1435 tabs,
1436 vec![
1437 TabId::vendor(VendorId::Anthropic),
1438 TabId::account("work"),
1439 TabId::desktop_account("gmail"),
1440 TabId::desktop_account("hotmail"),
1441 ]
1442 );
1443 }
1444
1445 #[test]
1446 fn desktop_accounts_suppress_the_default_tab_like_named_ones() {
1447 let mut config = config_with_accounts(&[]);
1450 config.cursor.enabled = false;
1451 config.anthropic.show_default_account = false;
1452
1453 assert_eq!(
1456 build_tabs(&config, &[]),
1457 vec![TabId::vendor(VendorId::Anthropic)]
1458 );
1459 assert_eq!(
1461 build_tabs(&config, &["gmail".into()]),
1462 vec![TabId::desktop_account("gmail")]
1463 );
1464 }
1465
1466 #[test]
1467 fn set_tabs_resets_states_and_clamps_selection() {
1468 let mut app = App::with_theme(
1472 tabs_from_config(&config_with_accounts(&["work", "personal"])),
1473 Theme::default(),
1474 );
1475 app.active = 2; app.tabs[0] = TabState::error("old");
1477 let old_tab = app.tabs_meta[0].clone();
1478 assert!(app.begin_refresh(&old_tab));
1479
1480 app.set_tabs(tabs_from_config(&config_with_accounts(&[])));
1481 assert_eq!(app.tabs_meta, vec![TabId::vendor(VendorId::Anthropic)]);
1482 assert_eq!(app.active, 0, "selection clamped after shrink");
1483 assert!(matches!(app.tabs[0], TabState::Loading));
1484 assert!(!app.is_refreshing(&old_tab));
1485 }
1486
1487 #[test]
1488 fn set_tabs_preserves_selected_identity_when_entries_are_inserted() {
1489 let mut app = App::with_theme(
1490 vec![
1491 TabId::vendor(VendorId::Anthropic),
1492 TabId::vendor(VendorId::Openai),
1493 ],
1494 Theme::default(),
1495 );
1496 app.active = 1;
1497
1498 app.set_tabs(vec![
1499 TabId::vendor(VendorId::Anthropic),
1500 TabId::account("work"),
1501 TabId::vendor(VendorId::Openai),
1502 ]);
1503
1504 assert_eq!(app.active, 2);
1505 assert_eq!(app.active_tab_id(), Some(&TabId::vendor(VendorId::Openai)));
1506 }
1507
1508 #[test]
1509 fn refresh_from_old_generation_is_discarded() {
1510 let mut app = App::with_theme(vec![TabId::vendor(VendorId::Anthropic)], Theme::default());
1511 let old_generation = app.tab_generation;
1512 app.set_tabs(vec![TabId::vendor(VendorId::Openai)]);
1513
1514 assert!(!app.apply_refresh(
1515 old_generation,
1516 &TabId::vendor(VendorId::Anthropic),
1517 TabState::error("old result"),
1518 ));
1519 assert!(matches!(app.tabs[0], TabState::Loading));
1520 }
1521
1522 #[test]
1523 fn refresh_identity_mismatch_is_discarded() {
1524 let mut app = App::with_theme(vec![TabId::vendor(VendorId::Anthropic)], Theme::default());
1525 let generation = app.tab_generation;
1526
1527 assert!(!app.apply_refresh(
1528 generation,
1529 &TabId::vendor(VendorId::Openai),
1530 TabState::error("wrong tab"),
1531 ));
1532 assert!(matches!(app.tabs[0], TabState::Loading));
1533 }
1534
1535 #[test]
1536 fn refresh_identity_lands_at_new_index_after_same_generation_reorder() {
1537 let anthropic = TabId::vendor(VendorId::Anthropic);
1538 let openai = TabId::vendor(VendorId::Openai);
1539 let mut app = App::with_theme(vec![anthropic.clone(), openai.clone()], Theme::default());
1540 let generation = app.tab_generation;
1541 assert!(app.begin_refresh(&anthropic));
1542
1543 app.tabs_meta.swap(0, 1);
1546 app.tabs.swap(0, 1);
1547 assert!(app.apply_refresh(generation, &anthropic, TabState::error("ready")));
1548 assert!(matches!(app.tabs[0], TabState::Loading));
1549 assert!(matches!(&app.tabs[1], TabState::Error { message, .. } if message == "ready"));
1550 assert!(!app.is_refreshing(&anthropic));
1551 }
1552
1553 fn ready_at(fetched_at: chrono::DateTime<Utc>) -> TabState {
1554 TabState::Ready(Box::new(ReadyTab {
1555 snapshot: crate::usage::VendorSnapshot::Openrouter(crate::usage::OpenRouterSnapshot {
1556 label: "test".into(),
1557 total_credits: 0.0,
1558 total_usage: 0.0,
1559 usage_daily: 0.0,
1560 usage_weekly: 0.0,
1561 usage_monthly: 0.0,
1562 is_free_tier: false,
1563 limit: None,
1564 limit_remaining: None,
1565 recent_models: Vec::new(),
1566 }),
1567 stale: false,
1568 last_error: None,
1569 fetched_at: Some(fetched_at),
1570 display: Default::default(),
1571 }))
1572 }
1573
1574 #[test]
1575 fn refresh_keeps_ready_snapshot_visible_and_suppresses_duplicates() {
1576 let tab = TabId::vendor(VendorId::Openrouter);
1577 let fetched_at = Utc.with_ymd_and_hms(2026, 5, 23, 12, 0, 0).unwrap();
1578 let mut app = App::with_theme(vec![tab.clone()], Theme::default());
1579 app.tabs[0] = ready_at(fetched_at);
1580
1581 assert!(app.begin_refresh(&tab));
1582 assert!(
1583 !app.begin_refresh(&tab),
1584 "duplicate request must be suppressed"
1585 );
1586 assert!(app.is_refreshing(&tab));
1587 match &app.tabs[0] {
1588 TabState::Ready(ready) => assert_eq!(ready.fetched_at, Some(fetched_at)),
1589 other => panic!("ready snapshot disappeared during refresh: {other:?}"),
1590 }
1591 }
1592
1593 #[test]
1594 fn first_refresh_still_uses_loading_until_data_arrives() {
1595 let tab = TabId::vendor(VendorId::Openrouter);
1596 let mut app = App::with_theme(vec![tab.clone()], Theme::default());
1597
1598 assert!(app.begin_refresh(&tab));
1599 assert!(app.is_refreshing(&tab));
1600 assert!(matches!(app.tabs[0], TabState::Loading));
1601
1602 assert!(app.apply_refresh(app.tab_generation, &tab, TabState::error("not signed in"),));
1603 assert!(!app.is_refreshing(&tab));
1604 assert!(
1605 matches!(&app.tabs[0], TabState::Error { message, .. } if message == "not signed in")
1606 );
1607 }
1608
1609 #[test]
1610 fn successful_revalidation_replaces_snapshot_and_clears_indicator() {
1611 let tab = TabId::vendor(VendorId::Openrouter);
1612 let old_at = Utc.with_ymd_and_hms(2026, 5, 23, 12, 0, 0).unwrap();
1613 let new_at = Utc.with_ymd_and_hms(2026, 5, 23, 12, 1, 0).unwrap();
1614 let mut app = App::with_theme(vec![tab.clone()], Theme::default());
1615 app.tabs[0] = ready_at(old_at);
1616
1617 assert!(app.begin_refresh(&tab));
1618 assert!(app.apply_refresh(app.tab_generation, &tab, ready_at(new_at)));
1619 assert!(!app.is_refreshing(&tab));
1620 match &app.tabs[0] {
1621 TabState::Ready(ready) => assert_eq!(ready.fetched_at, Some(new_at)),
1622 other => panic!("expected replacement snapshot, got {other:?}"),
1623 }
1624 }
1625
1626 #[test]
1627 fn failed_revalidation_preserves_snapshot_with_visible_warning() {
1628 let tab = TabId::vendor(VendorId::Openrouter);
1629 let fetched_at = Utc.with_ymd_and_hms(2026, 5, 23, 12, 0, 0).unwrap();
1630 let mut app = App::with_theme(vec![tab.clone()], Theme::default());
1631 app.tabs[0] = ready_at(fetched_at);
1632
1633 assert!(app.begin_refresh(&tab));
1634 assert!(app.apply_refresh(app.tab_generation, &tab, TabState::error("refresh failed"),));
1635 assert!(!app.is_refreshing(&tab));
1636 match &app.tabs[0] {
1637 TabState::Ready(ready) => {
1638 assert_eq!(ready.fetched_at, Some(fetched_at));
1639 assert!(ready.stale);
1640 assert_eq!(ready.last_error, Some((0, "refresh failed".into())));
1641 }
1642 other => panic!("last successful snapshot was lost: {other:?}"),
1643 }
1644 let sections = crate::tui::panels::sections_for(&app.tabs[0], Utc::now(), 5);
1645 assert!(sections.iter().any(|section| matches!(
1646 section,
1647 crate::tui::panels::Section::Text { label, value }
1648 if label == "Warning" && value == "refresh failed"
1649 )));
1650 }
1651
1652 #[test]
1653 fn old_generation_result_does_not_clear_current_refresh() {
1654 let tab = TabId::vendor(VendorId::Openrouter);
1655 let mut app = App::with_theme(vec![tab.clone()], Theme::default());
1656 let old_generation = app.tab_generation;
1657 app.set_tabs(vec![tab.clone()]);
1658 assert!(app.begin_refresh(&tab));
1659
1660 assert!(!app.apply_refresh(old_generation, &tab, TabState::error("old result"),));
1661 assert!(app.is_refreshing(&tab));
1662 assert!(matches!(app.tabs[0], TabState::Loading));
1663 }
1664
1665 #[test]
1666 fn apply_refresh_stamps_fetched_at_on_only_the_matching_tab() {
1667 let anthropic = TabId::vendor(VendorId::Anthropic);
1673 let openai = TabId::vendor(VendorId::Openai);
1674 let mut app = App::with_theme(vec![anthropic.clone(), openai], Theme::default());
1675 let generation = app.tab_generation;
1676 let fetched_at = Utc.with_ymd_and_hms(2026, 5, 23, 12, 0, 0).unwrap();
1677
1678 assert!(app.apply_refresh(generation, &anthropic, ready_at(fetched_at)));
1679 match &app.tabs[0] {
1680 TabState::Ready(ready) => assert_eq!(ready.fetched_at, Some(fetched_at)),
1681 other => panic!("expected Anthropic tab Ready, got {other:?}"),
1682 }
1683 assert!(matches!(app.tabs[1], TabState::Loading));
1684 }
1685
1686 #[test]
1687 fn select_primary_lands_on_default_account_tab() {
1688 let app = {
1691 let tabs = tabs_from_config(&config_with_accounts(&["work"]));
1692 let mut a = App::with_theme(tabs, Theme::default());
1693 a.select_primary(Some(VendorId::Anthropic));
1694 a
1695 };
1696 assert_eq!(app.active, 0);
1697 assert_eq!(
1698 app.active_tab_id(),
1699 Some(&TabId::vendor(VendorId::Anthropic))
1700 );
1701 }
1702
1703 fn custom_spec(id: &str, enabled: bool) -> CustomProviderConfig {
1704 CustomProviderConfig {
1705 id: id.into(),
1706 name: "My Tool".into(),
1707 short_name: "myt".into(),
1708 enabled,
1709 ..Default::default()
1710 }
1711 }
1712
1713 #[test]
1714 fn custom_providers_get_tabs_after_every_builtin() {
1715 let config = Config {
1716 custom: vec![custom_spec("mytool", true), custom_spec("other", true)],
1717 ..Default::default()
1718 };
1719 let tabs = tabs_from_config(&config);
1720 let builtin_count = config.enabled_vendors().len();
1721 assert_eq!(tabs.len(), builtin_count + 2);
1722 assert!(
1723 tabs[..builtin_count]
1724 .iter()
1725 .all(|t| matches!(t.source, TabSource::Builtin(_)))
1726 );
1727 assert_eq!(tabs[builtin_count], TabId::custom(&config.custom[0]));
1728 assert_eq!(tabs[builtin_count + 1], TabId::custom(&config.custom[1]));
1729 assert_eq!(
1730 tabs[builtin_count].source,
1731 TabSource::Custom {
1732 id: "mytool".into(),
1733 name: "My Tool".into(),
1734 short_name: "myt".into(),
1735 }
1736 );
1737 assert!(tabs[builtin_count].account.is_none());
1738 assert!(!tabs[builtin_count].desktop);
1739 assert_eq!(tabs[builtin_count].vendor_id(), None);
1740 }
1741
1742 #[test]
1743 fn disabled_custom_provider_has_no_tab() {
1744 let config = Config {
1745 custom: vec![custom_spec("mytool", false)],
1746 ..Default::default()
1747 };
1748 let tabs = tabs_from_config(&config);
1749 assert!(
1750 tabs.iter()
1751 .all(|t| matches!(t.source, TabSource::Builtin(_)))
1752 );
1753 assert_eq!(tabs.len(), config.enabled_vendors().len());
1754 }
1755
1756 #[test]
1757 fn custom_tabs_are_listed_by_the_default_overview_but_not_by_a_vendor_filter() {
1758 let spec = custom_spec("mytool", true);
1759 let mut app = App::with_theme(
1760 vec![
1761 TabId::vendor(VendorId::Anthropic),
1762 TabId::custom(&spec),
1763 TabId::vendor(VendorId::Openai),
1764 ],
1765 Theme::default(),
1766 );
1767 assert_eq!(app.overview_tabs(), vec![0, 1, 2]);
1768
1769 app.overview_vendors = Some(vec![VendorId::Openai, VendorId::Anthropic]);
1771 assert_eq!(app.overview_tabs(), vec![2, 0]);
1772 }
1773
1774 #[test]
1775 fn a_custom_active_tab_has_no_vendor_and_never_staggers() {
1776 let spec = custom_spec("mytool", true);
1777 let tabs = vec![
1778 TabId::vendor(VendorId::Anthropic),
1779 TabId::custom(&spec),
1780 TabId::account("work"),
1781 ];
1782 let mut app = App::with_theme(tabs.clone(), Theme::default());
1783 app.active = 1;
1784 assert_eq!(app.active_vendor(), None);
1785 assert_eq!(app.active_tab_id(), Some(&TabId::custom(&spec)));
1786
1787 app.select_primary(Some(VendorId::Anthropic));
1789 assert_eq!(app.active, 0);
1790
1791 let step = Duration::from_millis(800);
1792 assert_eq!(
1793 refresh_stagger(&tabs, step),
1794 vec![Duration::ZERO, Duration::ZERO, step]
1795 );
1796 }
1797}