codex-switch 0.1.8

Local CLI account switcher for Codex
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use anyhow::{Context, Result};

use crate::auth_json;
use crate::process;
use crate::store;
use crate::token;
use crate::types::StoredAccount;

pub async fn switch_to_account(selector: &str) -> Result<StoredAccount> {
    let store_snapshot = store::load_accounts()?;
    let account_id = store::resolve_account_id(&store_snapshot, selector)?;
    let process_info = process::check_codex_processes()?;
    process::ensure_can_switch_info(&process_info)?;
    switch_to_account_unchecked(&account_id).await
}

pub async fn switch_to_account_unchecked(selector: &str) -> Result<StoredAccount> {
    let store_snapshot = store::load_accounts()?;
    let account_id = store::resolve_account_id(&store_snapshot, selector)?;
    let account = store_snapshot
        .accounts
        .into_iter()
        .find(|account| account.id == account_id)
        .context("Account not found after resolving selector")?;

    let account = token::ensure_chatgpt_tokens_fresh(&account).await?;
    auth_json::write_account_auth(&account)?;
    store::set_active_account(&account.id)?;
    store::touch_account(&account.id)?;

    store::get_account_by_selector(&account.id)
}