test_codex/
test_codex.rs

1use seher::{BrowserDetector, CookieReader};
2
3#[tokio::main(flavor = "current_thread")]
4async fn main() {
5    let detector = BrowserDetector::new();
6    let browsers = detector.detect_browsers();
7
8    for browser in &browsers {
9        if !browser.is_chromium_based() {
10            continue;
11        }
12        for prof in detector.list_profiles(*browser) {
13            if let Ok(cookies) = CookieReader::read_cookies(&prof, "chatgpt.com") {
14                let has_session = cookies
15                    .iter()
16                    .any(|c| c.name.starts_with("__Secure-next-auth.session-token"));
17
18                if !has_session {
19                    continue;
20                }
21
22                println!(
23                    "Using {} - {} ({} cookies)",
24                    browser.name(),
25                    prof.name,
26                    cookies.len(),
27                );
28
29                match seher::codex::CodexClient::fetch_usage(&cookies).await {
30                    Ok(usage) => {
31                        println!("\nSuccess! Codex usage:");
32                        println!("  plan_type: {}", usage.plan_type);
33                        println!("  rate_limit limited: {}", usage.rate_limit.is_limited());
34                        println!(
35                            "  rate_limit reset_time: {:?}",
36                            usage.rate_limit.next_reset_time()
37                        );
38                        println!(
39                            "  code_review limited: {}",
40                            usage
41                                .code_review_rate_limit
42                                .as_ref()
43                                .is_some_and(seher::CodexRateLimit::is_limited)
44                        );
45                        return;
46                    }
47                    Err(e) => {
48                        println!("\nFailed to fetch Codex usage: {e}");
49                    }
50                }
51            }
52        }
53    }
54
55    println!("No chatgpt.com session with __Secure-next-auth.session-token found");
56}
test_codex/test_codex.rs

test_codex/
test_codex.rs
