use std::collections::HashSet; use std::path::{Path, PathBuf}; use crate::config::{ atomic_write, delete_file, get_home_dir, read_json_file, sanitize_provider_name, write_json_file, write_text_file, }; use crate::error::AppError; use crate::model_capabilities::{image_input_capability_from_modalities, ImageInputCapability}; use once_cell::sync::OnceCell; use serde_json::{json, Value}; use std::fs; use std::process::{Command, Stdio}; use toml_edit::DocumentMut; pub const CC_SWITCH_CODEX_MODEL_PROVIDER_ID: &str = "custom"; /// Temporary model-provider id used while the built-in `codex-official` /// provider is routed through CC Switch. A dedicated id is an ownership /// marker: unlike a generic localhost `base_url`, it can be detected and /// cleaned up without mistaking a user's own local provider for takeover. pub const CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID: &str = "cc-switch-official"; pub const CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME: &str = "cc-switch-model-catalog.json"; const CODEX_PROXY_AUTH_PLACEHOLDER: &str = "PROXY_MANAGED"; #[cfg(target_os = "windows")] const CREATE_NO_WINDOW: u32 = 0x08000000; // Generating a ProxyChat catalog only needs one stable Codex model template per // process. Without this cache every provider switch/takeover can start the // Codex CLI again, which is especially expensive for npm-installed `codex.cmd` // on Windows. Tests deliberately bypass the global cache because they isolate // CODEX_HOME and seed different model templates. #[cfg(not(test))] static CODEX_MODEL_CATALOG_TEMPLATE_CACHE: OnceCell = OnceCell::new(); /// Top-level `config.toml` key that controls Codex's built-in web-search tool. pub(crate) const CODEX_WEB_SEARCH_FIELD: &str = "web_search"; /// Value that disables the web-search tool. Some native `/responses` gateways /// reject a `web_search` tool with `responses_feature_not_supported` ("tool type /// 'web_search' is not supported by this gateway phase"), so for those we write /// this per the vendors' official Codex docs. Also doubles as cc-switch's /// ownership sentinel: we only ever remove a `web_search` key whose value equals /// this string, never a user's own setting. pub(crate) const CODEX_WEB_SEARCH_DISABLED: &str = "disabled"; /// Native `/responses` gateways whose first-party models do NOT support the Codex /// `web_search` hosted tool. A BLACKLIST (default-on): everything not listed keeps /// Codex's default, so relays/aggregators fronting real GPT — and any unknown /// provider — are never touched. This avoids a whitelist's dangerous failure mode /// (a fragile "is this GPT?" heuristic wrongly keeping web_search ON → hard 400); /// the blacklist's failure mode is the safe, recoverable one (a not-yet-listed /// broken gateway errors once → add it here). /// /// Matched two ways so an aggregator (e.g. SiliconFlow) fronting these vendors' /// models is also caught: /// - `base_url` host substring, and /// - the model id's brand prefix (after stripping any `vendor/` path segment). /// /// Verified 2026-06-28 doc audit — reject: MiMo (hard 400), LongCat (official /// config ships `web_search = "disabled"`), MiniMax (tool-type enum `['function']` /// only), and Qwen3-Coder models (百炼 marks built-in tools unsupported for /// the coder series). Deliberately NOT listed by host: 火山方舟豆包, general /// 阿里百炼 Qwen models that support built-in web_search, and GPT-native relays. const CODEX_WEB_SEARCH_REJECT_HOSTS: &[&str] = &[ "xiaomimimo.com", // Xiaomi MiMo (api.xiaomimimo.com, token-plan-cn.xiaomimimo.com) "longcat.chat", // Meituan LongCat (api.longcat.chat) "minimax.io", // MiniMax global (api.minimax.io) "minimaxi.com", // MiniMax CN (api.minimaxi.com) ]; /// Brand prefixes of models whose native gateways reject `web_search`, matched /// against the model id's last `/`-segment so aggregator ids like /// `MiniMaxAI/MiniMax-M3` are caught. Exact brand names (not a fuzzy heuristic), /// so a supporting gateway is never wrongly matched. const CODEX_WEB_SEARCH_REJECT_MODEL_PREFIXES: &[&str] = &["mimo", "longcat", "minimax", "qwen3-coder"]; /// Top-level `model` id from a Codex `config.toml`. fn codex_top_level_model(config_text: &str) -> Option { let doc = config_text.parse::().ok()?; doc.get("model") .and_then(|value| value.as_str()) .map(|value| value.trim().to_string()) .filter(|value| !value.is_empty()) } /// Whether a native `/responses` provider's gateway is known to reject the Codex /// `web_search` hosted tool — by `base_url` host OR by the active model's brand /// (so an aggregator fronting a reject vendor's model is caught too). Driven by /// the live `config.toml`, so it applies to existing providers without a re-save. fn codex_native_gateway_rejects_web_search(config_text: &str) -> bool { if let Some(base_url) = extract_codex_base_url(config_text) { let base_url = base_url.to_ascii_lowercase(); if CODEX_WEB_SEARCH_REJECT_HOSTS .iter() .any(|host| base_url.contains(host)) { return true; } } if let Some(model) = codex_top_level_model(config_text) { let model = model.to_ascii_lowercase(); // Strip any aggregator "vendor/" prefix, e.g. "MiniMaxAI/MiniMax-M3" // or "qwen/qwen3-coder-plus". let model = model.rsplit('/').next().unwrap_or(model.as_str()); if CODEX_WEB_SEARCH_REJECT_MODEL_PREFIXES .iter() .any(|prefix| model.starts_with(prefix)) { return true; } } false } const CODEX_MODEL_CATALOG_TEMPLATE_SLUG: &str = "gpt-5.5"; /// Which Codex tool surface the generated model catalog should target. /// /// - `ProxyChat`: cc-switch's proxy takes over and converts Responses<->Chat, /// so the catalog keeps Codex's default tool set (incl. the freeform /// `apply_patch` custom tool, which the proxy rewrites to a function tool). /// - `NativeResponses`: Codex talks directly to a provider's native /// `/responses` endpoint (no proxy). Such gateways (e.g. Xiaomi MiMo, /// MiniMax) reject `type=="custom"` tools, so the catalog must suppress the /// freeform `apply_patch` and rely on `shell_type="shell_command"` for edits. #[derive(Debug, Clone, Copy, PartialEq, Eq)] pub enum CodexCatalogToolProfile { ProxyChat, NativeResponses, /// Codex talks (through cc-switch's proxy) to a native Anthropic Messages /// gateway. Like `NativeResponses` it must suppress Codex's freeform custom /// tools — the Responses→Anthropic transform keeps only `function` tools. /// Additionally the Codex `web_search` hosted tool is unusable on this path /// (the transform drops it), so it is always disabled — see /// `prepare_codex_config_text_with_model_catalog`. Anthropic, } impl CodexCatalogToolProfile { /// Pick the catalog tool profile from a provider's `apiFormat` meta value. /// /// Prefer [`crate::proxy::providers::codex::resolve_codex_catalog_tool_profile`], /// which also honors settings-level `apiFormat` and the TOML `wire_api` (matching /// the proxy router). This string-only mapping is the fallback for non-Anthropic /// cases. pub fn from_api_format(api_format: Option<&str>) -> Self { match api_format { Some("anthropic") => CodexCatalogToolProfile::Anthropic, // Native (direct) Responses gateways reject Codex's freeform custom // tools (apply_patch, etc.); strip them via the NativeResponses profile. Some("openai_responses") => CodexCatalogToolProfile::NativeResponses, _ => CodexCatalogToolProfile::ProxyChat, } } } /// Reserved built-in provider IDs from OpenAI Codex's config/model-provider /// catalog. Keep in sync with Codex `RESERVED_MODEL_PROVIDER_IDS` and legacy /// removed provider aliases. const CODEX_RESERVED_MODEL_PROVIDER_IDS: &[&str] = &[ "amazon-bedrock", "openai", "ollama", "lmstudio", "oss", "ollama-chat", ]; /// 获取 Codex 配置目录路径 pub fn get_codex_config_dir() -> PathBuf { if let Some(custom) = crate::settings::get_codex_override_dir() { return custom; } get_home_dir().join(".codex") } /// 获取 Codex auth.json 路径 pub fn get_codex_auth_path() -> PathBuf { get_codex_config_dir().join("auth.json") } /// 获取 Codex config.toml 路径 pub fn get_codex_config_path() -> PathBuf { get_codex_config_dir().join("config.toml") } pub fn get_codex_model_catalog_path() -> PathBuf { get_codex_config_dir().join(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME) } /// 获取 Codex 供应商配置文件路径 #[allow(dead_code)] pub fn get_codex_provider_paths( provider_id: &str, provider_name: Option<&str>, ) -> (PathBuf, PathBuf) { let base_name = provider_name .map(sanitize_provider_name) .unwrap_or_else(|| sanitize_provider_name(provider_id)); let auth_path = get_codex_config_dir().join(format!("auth-{base_name}.json")); let config_path = get_codex_config_dir().join(format!("config-{base_name}.toml")); (auth_path, config_path) } /// 删除 Codex 供应商配置文件 #[allow(dead_code)] pub fn delete_codex_provider_config( provider_id: &str, provider_name: &str, ) -> Result<(), AppError> { let (auth_path, config_path) = get_codex_provider_paths(provider_id, Some(provider_name)); delete_file(&auth_path).ok(); delete_file(&config_path).ok(); Ok(()) } /// 原子写 Codex 的 `auth.json` 与 `config.toml`,在第二步失败时回滚第一步 pub fn write_codex_live_atomic( auth: &Value, config_text_opt: Option<&str>, ) -> Result<(), AppError> { let auth_path = get_codex_auth_path(); let config_path = get_codex_config_path(); if let Some(parent) = auth_path.parent() { std::fs::create_dir_all(parent).map_err(|e| AppError::io(parent, e))?; } // 读取旧内容用于回滚 let old_auth = if auth_path.exists() { Some(fs::read(&auth_path).map_err(|e| AppError::io(&auth_path, e))?) } else { None }; let _old_config = if config_path.exists() { Some(fs::read(&config_path).map_err(|e| AppError::io(&config_path, e))?) } else { None }; // 准备写入内容 let cfg_text = match config_text_opt { Some(s) => s.to_string(), None => String::new(), }; if !cfg_text.trim().is_empty() { toml::from_str::(&cfg_text).map_err(|e| AppError::toml(&config_path, e))?; } // 第一步:写 auth.json write_json_file(&auth_path, auth)?; // 第二步:写 config.toml(失败则回滚 auth.json) if let Err(e) = write_text_file(&config_path, &cfg_text) { // 回滚 auth.json if let Some(bytes) = old_auth { let _ = atomic_write(&auth_path, &bytes); } else { let _ = delete_file(&auth_path); } return Err(e); } Ok(()) } /// 读取 `~/.codex/config.toml`,若不存在返回空字符串 pub fn read_codex_config_text() -> Result { let path = get_codex_config_path(); if path.exists() { std::fs::read_to_string(&path).map_err(|e| AppError::io(&path, e)) } else { Ok(String::new()) } } /// 对非空的 TOML 文本进行语法校验 pub fn validate_config_toml(text: &str) -> Result<(), AppError> { if text.trim().is_empty() { return Ok(()); } toml::from_str::(text) .map(|_| ()) .map_err(|e| AppError::toml(Path::new("config.toml"), e)) } /// 读取并校验 `~/.codex/config.toml`,返回文本(可能为空) pub fn read_and_validate_codex_config_text() -> Result { let s = read_codex_config_text()?; validate_config_toml(&s)?; Ok(s) } fn active_codex_model_provider_id(doc: &DocumentMut) -> Option { doc.get("model_provider") .and_then(|item| item.as_str()) .map(str::trim) .filter(|id| !id.is_empty()) .map(str::to_string) } pub(crate) fn is_custom_codex_model_provider_id(id: &str) -> bool { let id = id.trim(); !id.is_empty() && !CODEX_RESERVED_MODEL_PROVIDER_IDS .iter() .any(|reserved| reserved.eq_ignore_ascii_case(id)) } /// Write only Codex `config.toml` for provider switching. /// /// Codex login state lives in `auth.json`; provider routing, endpoint, model, /// and provider-scoped bearer tokens live in `config.toml`. Provider switches /// should not overwrite the user's ChatGPT login cache. pub fn write_codex_live_config_atomic(config_text_opt: Option<&str>) -> Result<(), AppError> { let config_path = get_codex_config_path(); let cfg_text = match config_text_opt { Some(config_text) => config_text.to_string(), None => String::new(), }; if !cfg_text.trim().is_empty() { toml::from_str::(&cfg_text).map_err(|e| AppError::toml(&config_path, e))?; } write_text_file(&config_path, &cfg_text) } pub fn extract_codex_auth_api_key(auth: &Value) -> Option { auth.get("OPENAI_API_KEY") .and_then(|value| value.as_str()) .map(str::trim) .filter(|key| !key.is_empty()) .map(str::to_string) } pub fn extract_codex_api_key(auth: Option<&Value>, config_text: Option<&str>) -> Option { auth.and_then(extract_codex_auth_api_key) .or_else(|| config_text.and_then(extract_codex_experimental_bearer_token)) } /// Extract the upstream base URL from a Codex `config.toml` string. /// /// Prefers the active `[model_providers.].base_url`, falling /// back to a top-level `base_url`. Deliberately never reads a non-active /// `[model_providers.*]` section — the frontend `extractCodexBaseUrl` /// (`getRecoverableBaseUrlAssignments`) excludes those too, and a leftover /// section unrelated to the active provider must not leak into `{{baseUrl}}`. pub fn extract_codex_base_url(config_text: &str) -> Option { let doc = config_text.parse::().ok()?; if let Some(active_provider) = doc.get("model_provider").and_then(|v| v.as_str()) { if let Some(base_url) = doc .get("model_providers") .and_then(|providers| providers.get(active_provider)) .and_then(|provider| provider.get("base_url")) .and_then(|v| v.as_str()) { return Some(base_url.to_string()); } } doc.get("base_url") .and_then(|v| v.as_str()) .map(ToString::to_string) } pub fn codex_auth_has_login_material(auth: &Value) -> bool { let Some(obj) = auth.as_object() else { return false; }; obj.iter().any(|(key, value)| { if key == "auth_mode" { return false; } if key == "OPENAI_API_KEY" { return value .as_str() .map(str::trim) .is_some_and(|token| !token.is_empty()); } match value { Value::Null => false, Value::String(text) => !text.trim().is_empty(), Value::Array(items) => !items.is_empty(), Value::Object(map) => !map.is_empty(), _ => true, } }) } pub fn codex_auth_has_oauth_login_material(auth: &Value) -> bool { let Some(obj) = auth.as_object() else { return false; }; obj.iter().any(|(key, value)| { if key == "auth_mode" || key == "OPENAI_API_KEY" { return false; } match value { Value::Null => false, Value::String(text) => !text.trim().is_empty(), Value::Array(items) => !items.is_empty(), Value::Object(map) => !map.is_empty(), _ => true, } }) } pub fn should_restore_codex_provider_token_for_backfill( category: Option<&str>, template_settings: &Value, ) -> bool { if category == Some("official") { return false; } let Some(auth) = template_settings.get("auth") else { return true; }; let has_provider_api_key = extract_codex_auth_api_key(auth).is_some(); let has_oauth_login = codex_auth_has_oauth_login_material(auth); !has_oauth_login || has_provider_api_key } fn parse_codex_positive_u64(value: Option<&Value>) -> Option { match value { Some(Value::Number(n)) => n.as_u64().filter(|v| *v > 0), Some(Value::String(s)) => s.trim().parse::().ok().filter(|v| *v > 0), _ => None, } } fn extract_codex_top_level_u64(config_text: &str, field: &str) -> Option { let doc = config_text.parse::().ok()?; doc.get(field) .and_then(|value| value.as_integer()) .and_then(|value| u64::try_from(value).ok()) .filter(|value| *value > 0) } fn codex_catalog_input_modalities( model: &str, declared_modalities: Option<&[String]>, ) -> Vec { let modalities = match image_input_capability_from_modalities(model, declared_modalities) { ImageInputCapability::Unsupported => &["text"][..], ImageInputCapability::Supported | ImageInputCapability::Unknown => &["text", "image"][..], }; modalities.iter().map(|item| (*item).to_string()).collect() } fn codex_catalog_model_entry( template: &Value, spec: &CodexCatalogModelSpec, priority: usize, profile: CodexCatalogToolProfile, ) -> Value { let mut entry = template.clone(); let Some(entry_obj) = entry.as_object_mut() else { return json!({}); }; entry_obj.insert("slug".to_string(), json!(spec.model)); entry_obj.insert("display_name".to_string(), json!(spec.display_name)); entry_obj.insert("description".to_string(), json!(spec.display_name)); entry_obj.insert("context_window".to_string(), json!(spec.context_window)); entry_obj.insert("max_context_window".to_string(), json!(spec.context_window)); entry_obj.insert("priority".to_string(), json!(1000 + priority)); entry_obj.insert("additional_speed_tiers".to_string(), json!([])); entry_obj.insert("service_tiers".to_string(), json!([])); entry_obj.insert("availability_nux".to_string(), Value::Null); entry_obj.insert("upgrade".to_string(), Value::Null); // Image support is a model capability, not a tool-profile capability. // Trust hidden preset metadata first, then the confirmed text-only registry; // every unknown model fails open so GPT/relay aliases are never declared // text-only merely because a template had a conservative default. entry_obj.insert( "input_modalities".to_string(), json!(codex_catalog_input_modalities( &spec.model, spec.input_modalities.as_deref(), )), ); if profile != CodexCatalogToolProfile::ProxyChat { // Native `/responses` and Anthropic gateways reject / drop Codex's freeform // `apply_patch` (type=="custom") tool. Strip any key that would make Codex // emit a custom/freeform tool, and rely on shell_type="shell_command" for // edits. Defensive even though the native template is already clean // (guards against template drift / an accidental gpt-5.5 clone). // // NOTE: `base_instructions` is NOT stripped — Codex's catalog parser // treats it as a REQUIRED field and refuses to load the file without // it ("missing field `base_instructions`"). The template carries a // neutral identity default; per-vendor official text overrides below. for key in [ "apply_patch_tool_type", "web_search_tool_type", "tools", "model_messages", ] { entry_obj.remove(key); } entry_obj.insert("shell_type".to_string(), json!("shell_command")); if let Some(base_instructions) = spec .base_instructions .as_deref() .map(str::trim) .filter(|s| !s.is_empty()) { entry_obj.insert("base_instructions".to_string(), json!(base_instructions)); } if let Some(parallel) = spec.supports_parallel_tool_calls { entry_obj.insert("supports_parallel_tool_calls".to_string(), json!(parallel)); } } entry } #[derive(Debug, Clone, PartialEq, Eq)] struct CodexCatalogModelSpec { model: String, display_name: String, context_window: u64, /// Per-row override for the native template's `supports_parallel_tool_calls` /// (e.g. MiniMax=true, MiMo=false). Only consulted for `NativeResponses`. supports_parallel_tool_calls: Option, /// Hidden per-row capability declaration from built-in provider metadata. /// When omitted, all catalog profiles consult the shared text-only model /// registry and otherwise default to `["text", "image"]`. input_modalities: Option>, /// Per-row override for the native template's `base_instructions` (the /// model identity / system preamble). Carries each vendor's OFFICIAL value /// (e.g. MiMo "developed by Xiaomi", MiniMax "based on MiniMax-M3"); falls /// back to the template default when absent. Only consulted for /// `NativeResponses`. base_instructions: Option, } fn codex_catalog_model_specs(settings: &Value, config_text: &str) -> Vec { let Some(models) = settings .get("modelCatalog") .and_then(|catalog| catalog.get("models")) .and_then(|models| models.as_array()) else { return Vec::new(); }; let default_context_window = extract_codex_top_level_u64(config_text, "model_context_window").unwrap_or(128_000); let mut seen = std::collections::HashSet::new(); let mut specs = Vec::new(); for model_config in models { let Some(model) = model_config .get("model") .and_then(|value| value.as_str()) .map(str::trim) .filter(|model| !model.is_empty()) else { continue; }; if !seen.insert(model.to_string()) { continue; } let display_name = model_config .get("displayName") .or_else(|| model_config.get("display_name")) .and_then(|value| value.as_str()) .map(str::trim) .filter(|name| !name.is_empty()) .unwrap_or(model); let context_window = parse_codex_positive_u64( model_config .get("contextWindow") .or_else(|| model_config.get("context_window")), ) .unwrap_or(default_context_window); let supports_parallel_tool_calls = model_config .get("supportsParallelToolCalls") .or_else(|| model_config.get("supports_parallel_tool_calls")) .and_then(|value| value.as_bool()); let input_modalities = model_config .get("inputModalities") .or_else(|| model_config.get("input_modalities")) .and_then(|value| value.as_array()) .map(|items| { items .iter() .filter_map(|item| item.as_str()) .map(str::to_string) .collect::>() }) .filter(|items| !items.is_empty()); let base_instructions = model_config .get("baseInstructions") .or_else(|| model_config.get("base_instructions")) .and_then(|value| value.as_str()) .map(str::trim) .filter(|text| !text.is_empty()) .map(str::to_string); specs.push(CodexCatalogModelSpec { model: model.to_string(), display_name: display_name.to_string(), context_window, supports_parallel_tool_calls, input_modalities, base_instructions, }); } specs } fn find_codex_model_template(catalog: &Value) -> Option { catalog .get("models") .and_then(|models| models.as_array()) .and_then(|models| { models.iter().find(|model| { model.get("slug").and_then(|slug| slug.as_str()) == Some(CODEX_MODEL_CATALOG_TEMPLATE_SLUG) }) }) .cloned() } fn load_codex_model_template_from_cache() -> Result, AppError> { let path = get_codex_config_dir().join("models_cache.json"); if !path.exists() { return Ok(None); } let text = fs::read_to_string(&path).map_err(|e| AppError::io(&path, e))?; let catalog: Value = serde_json::from_str(&text).map_err(|e| AppError::json(&path, e))?; Ok(find_codex_model_template(&catalog)) } /// Fixed candidates for locating the `codex` CLI when it is not on the process /// PATH (common in GUI apps launched outside a terminal). const CODEX_CLI_FIXED_CANDIDATES: &[&str] = &[ "codex", // PATH (all platforms) "/opt/homebrew/bin/codex", // macOS Apple Silicon Homebrew "/usr/local/bin/codex", // macOS Intel Homebrew / Linux "/home/linuxbrew/.linuxbrew/bin/codex", // Linux Homebrew ]; fn push_codex_cli_candidate( candidates: &mut Vec, seen: &mut HashSet, candidate: PathBuf, ) { let key = candidate.to_string_lossy().into_owned(); if seen.insert(key) { candidates.push(candidate); } } fn push_existing_codex_cli_candidate( candidates: &mut Vec, seen: &mut HashSet, candidate: PathBuf, ) { if candidate.exists() { push_codex_cli_candidate(candidates, seen, candidate); } } fn push_codex_cli_candidates_from_version_dirs( candidates: &mut Vec, seen: &mut HashSet, versions_dir: PathBuf, suffix: &[&str], ) { let Ok(entries) = fs::read_dir(versions_dir) else { return; }; let mut discovered = entries .filter_map(Result::ok) .map(|entry| { let mut candidate = entry.path(); for component in suffix { candidate.push(component); } candidate }) .filter(|candidate| candidate.exists()) .collect::>(); // Prefer newer-looking version directories before older global installs. discovered.sort_by(|a, b| b.cmp(a)); for candidate in discovered { push_codex_cli_candidate(candidates, seen, candidate); } } fn push_home_codex_cli_candidates( candidates: &mut Vec, seen: &mut HashSet, home: &Path, ) { for relative in [ ".nvm/current/bin/codex", ".volta/bin/codex", ".asdf/shims/codex", ".local/share/mise/shims/codex", ".config/mise/shims/codex", ".local/bin/codex", ".npm-global/bin/codex", ".npm-packages/bin/codex", ".local/share/pnpm/codex", "Library/pnpm/codex", ] { push_existing_codex_cli_candidate(candidates, seen, home.join(relative)); } push_codex_cli_candidates_from_version_dirs( candidates, seen, home.join(".nvm/versions/node"), &["bin", "codex"], ); push_codex_cli_candidates_from_version_dirs( candidates, seen, home.join(".local/share/fnm/node-versions"), &["installation", "bin", "codex"], ); push_codex_cli_candidates_from_version_dirs( candidates, seen, home.join("Library/Application Support/fnm/node-versions"), &["installation", "bin", "codex"], ); } fn push_env_codex_cli_candidates(candidates: &mut Vec, seen: &mut HashSet) { for (env_key, suffix) in [ ("NPM_CONFIG_PREFIX", &["bin", "codex"][..]), ("VOLTA_HOME", &["bin", "codex"][..]), ("ASDF_DATA_DIR", &["shims", "codex"][..]), ("MISE_DATA_DIR", &["shims", "codex"][..]), ("PNPM_HOME", &["codex"][..]), ] { let Some(prefix) = std::env::var_os(env_key) else { continue; }; let mut candidate = PathBuf::from(prefix); for component in suffix { candidate.push(component); } push_existing_codex_cli_candidate(candidates, seen, candidate); } if let Some(nvm_dir) = std::env::var_os("NVM_DIR") { push_codex_cli_candidates_from_version_dirs( candidates, seen, PathBuf::from(nvm_dir).join("versions/node"), &["bin", "codex"], ); } if let Some(fnm_dir) = std::env::var_os("FNM_DIR") { push_codex_cli_candidates_from_version_dirs( candidates, seen, PathBuf::from(fnm_dir).join("node-versions"), &["installation", "bin", "codex"], ); } #[cfg(windows)] { if let Some(appdata) = std::env::var_os("APPDATA") { let npm_dir = PathBuf::from(appdata).join("npm"); for name in ["codex.cmd", "codex.exe", "codex"] { push_existing_codex_cli_candidate(candidates, seen, npm_dir.join(name)); } } } } fn codex_cli_candidates() -> Vec { let mut candidates = Vec::new(); let mut seen = HashSet::new(); for candidate in CODEX_CLI_FIXED_CANDIDATES { push_codex_cli_candidate(&mut candidates, &mut seen, PathBuf::from(candidate)); } push_env_codex_cli_candidates(&mut candidates, &mut seen); push_home_codex_cli_candidates(&mut candidates, &mut seen, &get_home_dir()); candidates } fn codex_bundled_models_command(candidate: &Path) -> Command { let mut command = Command::new(candidate); command .args(["debug", "models", "--bundled"]) .stdin(Stdio::null()); // A release build uses the Windows GUI subsystem, so a console child that // is created without this flag gets its own transient console window. npm // installs Codex as `codex.cmd`, which Windows launches through cmd.exe. #[cfg(target_os = "windows")] { use std::os::windows::process::CommandExt; command.creation_flags(CREATE_NO_WINDOW); } command } fn load_codex_model_template_from_bundled() -> Result, AppError> { for candidate in codex_cli_candidates() { let candidate_label = candidate.to_string_lossy(); let output = match codex_bundled_models_command(&candidate).output() { Ok(output) => output, Err(err) => { log::debug!("failed to run `{candidate_label} debug models --bundled`: {err}"); continue; } }; if !output.status.success() { let stderr = String::from_utf8_lossy(&output.stderr); log::debug!("`{candidate_label} debug models --bundled` failed: {stderr}"); continue; } let catalog: Value = match serde_json::from_slice(&output.stdout) { Ok(catalog) => catalog, Err(e) => { log::debug!( "Failed to parse `{candidate_label} debug models --bundled` output: {e}" ); continue; } }; if let Some(template) = find_codex_model_template(&catalog) { return Ok(Some(template)); } } Ok(None) } fn load_codex_model_template_static() -> Option { let text = include_str!("resources/gpt5_5_template.json"); match serde_json::from_str(text) { Ok(template) => Some(template), Err(e) => { log::warn!("Failed to parse bundled gpt-5.5 template: {e}"); None } } } /// Bundled clean template for native `/responses` providers. Unlike the /// gpt-5.5 template it carries NO freeform `apply_patch` / `web_search` tool /// declarations and no GPT-5 base_instructions, so Codex never emits a /// `type=="custom"` tool that native gateways (MiMo/MiniMax/…) reject. Edits /// flow through `shell_type="shell_command"` instead. We deliberately do NOT /// fall back to `models_cache.json` here (that would reintroduce gpt-5.5's /// freeform apply_patch). fn load_codex_native_responses_template() -> Value { let text = include_str!("resources/codex_native_responses_template.json"); serde_json::from_str(text).expect("bundled codex native responses template must be valid JSON") } /// Fields Codex's external-catalog parser REQUIRES (no serde default): when /// one is missing Codex rejects the whole catalog file at startup ("missing /// field ..."). `base_instructions` is the other known required field; the /// templates always carry it and `codex_catalog_model_entry` handles it. /// When Codex requires a new field, add it here AND to the static templates. const CODEX_CATALOG_PARSER_REQUIRED_FIELDS: &[&str] = &["supports_reasoning_summaries"]; /// `models_cache.json` is shared by every Codex install on the machine (npm /// CLI, desktop-bundled binary, ...), and each version serializes its own /// `ModelInfo` shape — the cache's field set follows whichever process wrote /// it last, so it cannot be assumed to satisfy the current external-catalog /// schema (observed live: 0.144.5 requires `supports_reasoning_summaries` /// while a coexisting build kept rewriting the cache without it). Backfill /// ONLY parser-required fields from the bundled static template: optional /// capability fields keep their missing-means-default semantics, and existing /// values always win. fn fill_template_fields_from_static(template: &mut Value) { let Some(static_template) = load_codex_model_template_static() else { return; }; let (Some(template_obj), Some(static_obj)) = (template.as_object_mut(), static_template.as_object()) else { return; }; for key in CODEX_CATALOG_PARSER_REQUIRED_FIELDS { if !template_obj.contains_key(*key) { if let Some(value) = static_obj.get(*key) { template_obj.insert((*key).to_string(), value.clone()); } } } } fn load_codex_model_catalog_template_uncached() -> Result { // ① models_cache.json (created by Codex when it connects to OpenAI) if let Some(mut template) = load_codex_model_template_from_cache()? { fill_template_fields_from_static(&mut template); return Ok(template); } // ② codex CLI (PATH + platform-specific common paths) if let Some(mut template) = load_codex_model_template_from_bundled()? { fill_template_fields_from_static(&mut template); return Ok(template); } // ③ Static fallback bundled at compile time if let Some(template) = load_codex_model_template_static() { return Ok(template); } Err(AppError::Message(format!( "Codex model catalog template `{CODEX_MODEL_CATALOG_TEMPLATE_SLUG}` not found. Please start Codex once so models_cache.json is available, or ensure the `codex` CLI is on PATH." ))) } fn get_or_load_codex_model_catalog_template( cache: &OnceCell, loader: F, ) -> Result where F: FnOnce() -> Result, { cache.get_or_try_init(loader).cloned() } #[cfg(not(test))] fn load_codex_model_catalog_template() -> Result { get_or_load_codex_model_catalog_template( &CODEX_MODEL_CATALOG_TEMPLATE_CACHE, load_codex_model_catalog_template_uncached, ) } #[cfg(test)] fn load_codex_model_catalog_template() -> Result { load_codex_model_catalog_template_uncached() } fn codex_model_catalog_from_specs( specs: &[CodexCatalogModelSpec], template: &Value, profile: CodexCatalogToolProfile, ) -> Value { let entries: Vec = specs .iter() .enumerate() .map(|(index, spec)| codex_catalog_model_entry(template, spec, index, profile)) .collect(); json!({ "models": entries }) } fn codex_model_catalog_from_settings( settings: &Value, config_text: &str, profile: CodexCatalogToolProfile, ) -> Result, AppError> { let specs = codex_catalog_model_specs(settings, config_text); if specs.is_empty() { return Ok(None); } // Native providers use the bundled clean template (no freeform apply_patch, // no cache dependency); proxy-chat providers keep cloning Codex's gpt-5.5 // entry so the proxy can rewrite custom<->function tools as before. let template = match profile { CodexCatalogToolProfile::NativeResponses | CodexCatalogToolProfile::Anthropic => { load_codex_native_responses_template() } CodexCatalogToolProfile::ProxyChat => load_codex_model_catalog_template()?, }; Ok(Some(codex_model_catalog_from_specs( &specs, &template, profile, ))) } fn set_codex_model_catalog_json_field( config_text: &str, catalog_path: Option<&Path>, ) -> Result { let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; match catalog_path { Some(_) => { doc["model_catalog_json"] = toml_edit::value(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME); } None => { let should_remove = doc .get("model_catalog_json") .and_then(|item| item.as_str()) .map(|path| { Path::new(path).file_name().and_then(|name| name.to_str()) == Some(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME) }) .unwrap_or(false); if should_remove { doc.as_table_mut().remove("model_catalog_json"); } } } Ok(doc.to_string()) } /// Pure toggle for the top-level `web_search` field that turns Codex's built-in /// web-search tool off. When `disable` is true we write `web_search = "disabled"` /// (the catalog's `supports_search_tool` does NOT gate this — the request-time /// tool comes from the config, defaulting on). When false we *remove* the field, /// but only when it carries cc-switch's own `"disabled"` sentinel, so switching /// back to a web-search-capable provider re-enables it without clobbering a /// user's manual setting. /// /// The caller decides `disable` (see `codex_native_gateway_rejects_web_search`); /// lifecycle is bound to the cc-switch catalog pointer so the field is set/cleaned /// up wherever the native catalog is written/removed. fn set_codex_native_web_search_field(config_text: &str, disable: bool) -> Result { let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; if disable { doc[CODEX_WEB_SEARCH_FIELD] = toml_edit::value(CODEX_WEB_SEARCH_DISABLED); } else { let owned = doc .get(CODEX_WEB_SEARCH_FIELD) .and_then(|item| item.as_str()) == Some(CODEX_WEB_SEARCH_DISABLED); if owned { doc.as_table_mut().remove(CODEX_WEB_SEARCH_FIELD); } } Ok(doc.to_string()) } /// Generate Codex `model_catalog_json` from provider settings and inject/remove /// the top-level TOML field that points Codex to the generated file. pub fn prepare_codex_config_text_with_model_catalog( settings: &Value, config_text: &str, profile: CodexCatalogToolProfile, ) -> Result { let catalog_path = get_codex_model_catalog_path(); if let Some(catalog) = codex_model_catalog_from_settings(settings, config_text, profile)? { let config_text = set_codex_model_catalog_json_field(config_text, Some(&catalog_path))?; // Disable web_search only for native gateways on the reject blacklist // (MiMo/LongCat/MiniMax by host or model brand; Qwen3-Coder by model). // Everything else — relays, DouBao, web-search-capable Qwen models, // unknown providers — keeps Codex's default. let disable_web_search = match profile { // The Responses→Anthropic transform silently drops the Codex web_search // hosted tool, so always disable it here rather than present a dead tool. CodexCatalogToolProfile::Anthropic => true, CodexCatalogToolProfile::NativeResponses => { codex_native_gateway_rejects_web_search(&config_text) } CodexCatalogToolProfile::ProxyChat => false, }; let config_text = set_codex_native_web_search_field(&config_text, disable_web_search)?; write_json_file(&catalog_path, &catalog)?; Ok(config_text) } else { let config_text = set_codex_model_catalog_json_field(config_text, None)?; // Even without a generated catalog, the Responses→Anthropic transform drops the // Codex web_search hosted tool, so keep the invariant that an Anthropic provider // never presents it as a dead tool. let disable_web_search = profile == CodexCatalogToolProfile::Anthropic; set_codex_native_web_search_field(&config_text, disable_web_search) } } /// Reverse of `prepare_codex_config_text_with_model_catalog`: read the /// cc-switch–maintained catalog file referenced by `~/.codex/config.toml` and /// convert it back into the simplified shape the frontend table uses: /// `{ "models": [{ "model", "displayName"?, "contextWindow"?, hidden overrides... }, ...] }`. /// /// We only reverse-parse catalogs whose `model_catalog_json` path is the /// cc-switch–generated file (identified by filename /// `cc-switch-model-catalog.json`). A user-managed external catalog file is /// left alone — surfacing its richer structure as the simplified table would /// be a downgrade we can't safely round-trip. /// /// `displayName`, `contextWindow`, and `inputModalities` are omitted from the /// returned entry when the on-disk value matches the fallback that /// `codex_model_catalog_from_settings` injects for unset inputs (slug for /// display_name, `model_context_window` or 128_000 for context_window, and the /// shared confirmed-text-only inference for input modalities). This preserves /// the "user left it blank" intent across round-trip; an unavoidable edge case /// is that a user-typed value that happens to equal the fallback also collapses /// to blank, but the next save writes the same fallback so behavior is stable. /// /// All failure modes (missing file, parse error, no `model_catalog_json`, /// entries without `slug`) collapse to `Ok(None)` so callers can treat this /// as best-effort enrichment without making `read_live_settings` brittle. pub fn read_codex_model_catalog_simplified_from_live() -> Result, AppError> { let config_text = read_codex_config_text()?; let generated_path = get_codex_model_catalog_path(); let Some(catalog_path) = resolve_cc_switch_catalog_path(&config_text, &generated_path) else { return Ok(None); }; if !catalog_path.exists() { return Ok(None); } let Ok(catalog_text) = fs::read_to_string(&catalog_path) else { return Ok(None); }; Ok(build_simplified_catalog_from_texts( &config_text, &catalog_text, )) } /// Given `config.toml` text, resolve the on-disk path of the cc-switch–owned /// catalog file (returns `None` if `model_catalog_json` is absent or points at /// a file we don't own). Relative paths fall back to `generated_path`. pub(crate) fn resolve_cc_switch_catalog_path( config_text: &str, generated_path: &Path, ) -> Option { if config_text.trim().is_empty() { return None; } let doc = config_text.parse::().ok()?; let catalog_path_str = doc .get("model_catalog_json") .and_then(|item| item.as_str()) .map(str::trim) .filter(|s| !s.is_empty())?; let referenced_path = Path::new(catalog_path_str); let is_cc_switch_owned = referenced_path.file_name().and_then(|name| name.to_str()) == Some(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME); if !is_cc_switch_owned { return None; } if referenced_path.is_absolute() { Some(referenced_path.to_path_buf()) } else { Some(generated_path.to_path_buf()) } } /// Pure reverse-parsing core: convert Codex catalog JSON text back into the /// frontend's simplified model-mapping shape. Returns `None` when the catalog /// is unparseable, has no `models` array, or yields zero valid entries. fn build_simplified_catalog_from_texts(config_text: &str, catalog_text: &str) -> Option { let catalog: Value = serde_json::from_str(catalog_text).ok()?; let models = catalog.get("models").and_then(|m| m.as_array())?; let default_context_window = extract_codex_top_level_u64(config_text, "model_context_window").unwrap_or(128_000); let mut entries = Vec::with_capacity(models.len()); for entry in models { let Some(model) = entry .get("slug") .and_then(|v| v.as_str()) .map(str::trim) .filter(|s| !s.is_empty()) else { continue; }; let mut obj = serde_json::Map::new(); obj.insert("model".to_string(), json!(model)); if let Some(display_name) = entry .get("display_name") .and_then(|v| v.as_str()) .map(str::trim) .filter(|s| !s.is_empty() && *s != model) { obj.insert("displayName".to_string(), json!(display_name)); } if let Some(context_window) = entry .get("context_window") .and_then(|v| v.as_u64()) .filter(|v| *v > 0 && *v != default_context_window) { obj.insert("contextWindow".to_string(), json!(context_window)); } // Preserve native-profile per-row overrides so a DB-SSOT-missing // fallback round-trip doesn't silently drop them. if let Some(parallel) = entry .get("supports_parallel_tool_calls") .and_then(|v| v.as_bool()) { obj.insert("supportsParallelToolCalls".to_string(), json!(parallel)); } if let Some(modalities) = entry.get("input_modalities").and_then(|v| v.as_array()) { let mods: Vec = modalities .iter() .filter_map(|m| m.as_str()) .map(str::to_string) .collect(); let inferred = codex_catalog_input_modalities(model, None); if !mods.is_empty() && mods != inferred { obj.insert("inputModalities".to_string(), json!(mods)); } } entries.push(Value::Object(obj)); } if entries.is_empty() { return None; } Some(json!({ "models": entries })) } /// Decide the `config.toml` text to write during a takeover-off restore, /// projecting the model catalog **only when `settings` carries an inline /// `modelCatalog`**. /// /// Restore feeds back a stored backup, and Codex backups come in two shapes that /// need opposite handling: /// /// - **Snapshot backup** (`read_codex_live_settings`): `{ auth, config }` with no /// inline `modelCatalog`. Its `config.toml` text already carries whatever /// `model_catalog_json` pointer existed at backup time, and the generated /// catalog file on disk is untouched. Here we must keep the config **raw** — /// running catalog projection would see "no specs" and strip the live pointer. /// - **Provider-rebuilt backup** (`update_live_backup_from_provider`): the DB /// provider's settings, i.e. `{ auth, config (no pointer), modelCatalog /// (inline DB SSOT) }`. Here the pointer/catalog file must be (re)generated /// from the inline `modelCatalog`, or the mapping is lost on restore. /// /// Gating on the presence of the inline `modelCatalog` key routes each shape /// correctly; an empty inline catalog still projects (and so correctly drops a /// now-stale pointer), while an absent key leaves the text untouched. This is /// **orthogonal to auth** — a provider-rebuilt backup can pair an inline /// `modelCatalog` with empty `auth.json` (the API key living in the config's /// `experimental_bearer_token`), so the caller must decide config projection /// independently of whether it writes or deletes `auth.json`. pub fn prepare_codex_live_config_text_with_optional_catalog( settings: &Value, config_text: &str, profile: CodexCatalogToolProfile, ) -> Result { if settings.get("modelCatalog").is_some() { prepare_codex_config_text_with_model_catalog(settings, config_text, profile) } else { Ok(config_text.to_string()) } } pub fn write_codex_provider_live_with_catalog( settings: &Value, category: Option<&str>, auth: &Value, config_text: Option<&str>, profile: CodexCatalogToolProfile, ) -> Result<(), AppError> { let prepared_config = config_text .map(|text| prepare_codex_config_text_with_model_catalog(settings, text, profile)) .transpose()?; write_codex_live_for_provider(category, auth, prepared_config.as_deref()) } /// Extract a provider-scoped `experimental_bearer_token` from Codex `config.toml`. /// /// Mobile compat: third-party providers may store the API key inside /// `[model_providers.].experimental_bearer_token` while keeping the /// user's ChatGPT login cache intact in `auth.json`. Falls back to the /// top-level `experimental_bearer_token` when no active model provider is set. pub fn extract_codex_experimental_bearer_token(config_text: &str) -> Option { if !config_text.contains("experimental_bearer_token") { return None; } let doc = config_text.parse::().ok()?; let provider_id = active_codex_model_provider_id(&doc); let top_level_token = || { doc.get("experimental_bearer_token") .and_then(|item| item.as_str()) }; let token = match provider_id.as_deref() { Some(id) if is_custom_codex_model_provider_id(id) => doc .get("model_providers") .and_then(|item| item.as_table()) .and_then(|table| table.get(id)) .and_then(|item| item.as_table()) .and_then(|table| table.get("experimental_bearer_token")) .and_then(|item| item.as_str()) .or_else(top_level_token), Some(_) => top_level_token(), None => top_level_token(), }; token .map(str::trim) .filter(|token| !token.is_empty()) .map(str::to_string) } fn set_codex_experimental_bearer_token(config_text: &str, token: &str) -> Result { if config_text.trim().is_empty() { return Err(AppError::localized( "provider.codex.config.missing", "Codex 第三方供应商缺少 config.toml 配置,无法写入 bearer token", "Codex third-party provider is missing config.toml, cannot write bearer token", )); } let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; let Some(provider_id) = active_codex_model_provider_id(&doc) else { doc["experimental_bearer_token"] = toml_edit::value(token); return Ok(doc.to_string()); }; if !is_custom_codex_model_provider_id(&provider_id) { // Reserved Codex provider IDs are owned by the CLI. Keep third-party // bearer tokens at the top level so we do not shadow built-in tables. doc["experimental_bearer_token"] = toml_edit::value(token); return Ok(doc.to_string()); } if let Some(model_providers) = doc .get_mut("model_providers") .and_then(|item| item.as_table_mut()) { if let Some(provider_table) = model_providers .get_mut(provider_id.as_str()) .and_then(|item| item.as_table_mut()) { provider_table["experimental_bearer_token"] = toml_edit::value(token); return Ok(doc.to_string()); } } doc["experimental_bearer_token"] = toml_edit::value(token); Ok(doc.to_string()) } pub fn remove_codex_experimental_bearer_token_if( config_text: &str, predicate: impl Fn(&str) -> bool, ) -> Result { if config_text.trim().is_empty() || !config_text.contains("experimental_bearer_token") { return Ok(config_text.to_string()); } let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; if let Some(provider_id) = active_codex_model_provider_id(&doc) { if let Some(provider_table) = doc .get_mut("model_providers") .and_then(|item| item.as_table_mut()) .and_then(|table| table.get_mut(provider_id.as_str())) .and_then(|item| item.as_table_mut()) { let should_remove = provider_table .get("experimental_bearer_token") .and_then(|item| item.as_str()) .map(str::trim) .is_some_and(&predicate); if should_remove { provider_table.remove("experimental_bearer_token"); } } } let should_remove_top_level = doc .get("experimental_bearer_token") .and_then(|item| item.as_str()) .map(str::trim) .is_some_and(&predicate); if should_remove_top_level { doc.as_table_mut().remove("experimental_bearer_token"); } Ok(doc.to_string()) } fn remove_codex_experimental_bearer_token(config_text: &str) -> Result { remove_codex_experimental_bearer_token_if(config_text, |_| true) } /// Read the current Codex live settings as a `{ auth, config }` object. /// /// Missing `auth.json` collapses to `{}` so a config-only third-party install /// is still importable; both files empty is treated as "no live install". pub fn read_codex_live_settings() -> Result { let auth_path = get_codex_auth_path(); let auth_present = auth_path.exists(); let auth: Value = if auth_present { read_json_file(&auth_path)? } else { json!({}) }; let cfg_text = read_and_validate_codex_config_text()?; if !auth_present && cfg_text.trim().is_empty() { return Err(AppError::localized( "codex.live.missing", "Codex 配置文件不存在", "Codex configuration is missing", )); } Ok(json!({ "auth": auth, "config": cfg_text })) } /// `[model_providers.custom]` entry that makes an official (ChatGPT OAuth) /// provider behave like Codex's built-in `openai` entry while running under /// the shared custom id: `requires_openai_auth` routes auth to the ChatGPT /// login in `auth.json` (base_url then defaults to the official Codex /// backend), `name = "OpenAI"` keeps Codex's `is_openai()` feature gates /// (web search, remote compaction), and `supports_websockets` restores the /// built-in default that custom entries otherwise lose. fn codex_official_provider_table( base_url: Option<&str>, supports_websockets: bool, ) -> toml_edit::Table { let mut table = toml_edit::Table::new(); table["name"] = toml_edit::value("OpenAI"); table["requires_openai_auth"] = toml_edit::value(true); table["supports_websockets"] = toml_edit::value(supports_websockets); table["wire_api"] = toml_edit::value("responses"); if let Some(base_url) = base_url { table["base_url"] = toml_edit::value(base_url.trim_end_matches('/')); } table } fn codex_unified_official_provider_table() -> toml_edit::Table { codex_official_provider_table(None, true) } fn remove_codex_proxy_placeholders_from_providers(providers: &mut toml_edit::Table) { for (_, item) in providers.iter_mut() { if let Some(table) = item.as_table_mut() { let should_remove = table .get("experimental_bearer_token") .and_then(|item| item.as_str()) == Some(CODEX_PROXY_AUTH_PLACEHOLDER); if should_remove { table.remove("experimental_bearer_token"); } } else if let Some(table) = item.as_inline_table_mut() { let should_remove = table .get("experimental_bearer_token") .and_then(|value| value.as_str()) == Some(CODEX_PROXY_AUTH_PLACEHOLDER); if should_remove { table.remove("experimental_bearer_token"); } } } } /// Project the built-in Codex official provider through the local proxy while /// keeping authentication owned by Codex itself. /// /// The resulting custom provider explicitly opts into OpenAI authentication, /// so Codex forwards its existing ChatGPT login to the local `/responses` /// endpoint. No API key or bearer placeholder is written to `auth.json`. pub fn apply_codex_official_proxy_route( config_text: &str, proxy_base_url: &str, ) -> Result { let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; // A third-party takeover may have left the proxy placeholder in config.toml. // The official route must use Codex's native OpenAI login instead. doc.as_table_mut().remove("experimental_bearer_token"); doc["model_provider"] = toml_edit::value(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID); let mut providers = match doc.as_table_mut().remove("model_providers") { Some(item) => item.into_table().map_err(|_| { AppError::Message( "Invalid Codex config.toml: model_providers must be a table".to_string(), ) })?, None => { let mut table = toml_edit::Table::new(); table.set_implicit(true); table } }; // Clean only CC Switch's placeholder from every stale provider table. Real // user bearer tokens are preserved, as are all unrelated provider fields. remove_codex_proxy_placeholders_from_providers(&mut providers); // The local proxy currently exposes HTTP/SSE, not Codex websocket routes. let table = codex_official_provider_table(Some(proxy_base_url), false); providers.insert( CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID, toml_edit::Item::Table(table), ); doc["model_providers"] = toml_edit::Item::Table(providers); Ok(doc.to_string()) } /// Whether a live Codex config is the official route projected by CC Switch. pub fn codex_config_has_official_proxy_route(config_text: &str) -> bool { if !config_text.contains(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) { return false; } config_text .parse::() .ok() .and_then(|doc| { doc.get("model_provider") .and_then(|item| item.as_str()) .map(str::to_string) }) .as_deref() == Some(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) } /// Remove only the official takeover route owned by CC Switch. This is a /// last-resort crash cleanup when no live backup or provider SSOT is usable. pub fn remove_codex_official_proxy_route(config_text: &str) -> Result { let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; if doc.get("model_provider").and_then(|item| item.as_str()) != Some(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) { return Ok(config_text.to_string()); } doc.as_table_mut().remove("model_provider"); if let Some(item) = doc.as_table_mut().remove("model_providers") { let mut providers = item.into_table().map_err(|_| { AppError::Message( "Invalid Codex config.toml: model_providers must be a table".to_string(), ) })?; providers.remove(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID); remove_codex_proxy_placeholders_from_providers(&mut providers); if !providers.is_empty() { doc["model_providers"] = toml_edit::Item::Table(providers); } } Ok(doc.to_string()) } fn table_matches_codex_unified_official_provider(table: &toml_edit::Table) -> bool { table.len() == 4 && table.get("name").and_then(|item| item.as_str()) == Some("OpenAI") && table .get("requires_openai_auth") .and_then(|item| item.as_bool()) == Some(true) && table .get("supports_websockets") .and_then(|item| item.as_bool()) == Some(true) && table.get("wire_api").and_then(|item| item.as_str()) == Some("responses") } /// 统一 Codex 会话历史:把官方供应商的 live 配置改写为以共享的 /// `custom` model_provider 标识运行(认证仍走 `auth.json` 的 ChatGPT 登录), /// 使开关开启后创建的官方会话与第三方会话共用同一个 resume 历史桶。 /// /// 两种情况拒绝注入、原样返回: /// - 配置已有显式 `model_provider`:用户手工指定的路由不被覆盖; /// - 配置已有形态不同的 `[model_providers.custom]` 表:设置 `model_provider` /// 会激活这张我们不认识的表(可能带第三方 base_url/token,会把 ChatGPT /// OAuth 流量路由到错误后端),宁可让开关对该配置不生效。 pub fn inject_codex_unified_session_bucket(config_text: &str) -> Result { let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; if doc.get("model_provider").is_some() { return Ok(config_text.to_string()); } let existing_custom_conflicts = doc .get("model_providers") .and_then(|item| item.as_table()) .and_then(|providers| providers.get(CC_SWITCH_CODEX_MODEL_PROVIDER_ID)) .and_then(|item| item.as_table()) .is_some_and(|table| !table_matches_codex_unified_official_provider(table)); if existing_custom_conflicts { log::warn!( "官方 Codex 配置已存在自定义 [model_providers.custom],跳过统一会话路由注入以避免激活未知路由" ); return Ok(config_text.to_string()); } doc["model_provider"] = toml_edit::value(CC_SWITCH_CODEX_MODEL_PROVIDER_ID); if doc.get("model_providers").is_none() { let mut parent = toml_edit::Table::new(); parent.set_implicit(true); doc["model_providers"] = toml_edit::Item::Table(parent); } if let Some(providers) = doc["model_providers"].as_table_mut() { if !providers.contains_key(CC_SWITCH_CODEX_MODEL_PROVIDER_ID) { providers.insert( CC_SWITCH_CODEX_MODEL_PROVIDER_ID, toml_edit::Item::Table(codex_unified_official_provider_table()), ); } } Ok(doc.to_string()) } /// `inject_codex_unified_session_bucket` 的反向操作:从配置文本里剥掉注入的 /// 统一会话路由,保证切换回填不会把它带进数据库的存储配置(关闭开关后 /// 切换即可完全还原)。仅当形态与注入产物完全一致时才剥离;第三方模板和 /// 用户自定义的 `custom` 条目(带 base_url 等差异字段)原样保留。 pub fn strip_codex_unified_session_bucket(config_text: &str) -> Result { if !config_text.contains("model_provider") { return Ok(config_text.to_string()); } let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; if doc.get("model_provider").and_then(|item| item.as_str()) != Some(CC_SWITCH_CODEX_MODEL_PROVIDER_ID) { return Ok(config_text.to_string()); } let matches_injected = doc .get("model_providers") .and_then(|item| item.as_table()) .and_then(|providers| providers.get(CC_SWITCH_CODEX_MODEL_PROVIDER_ID)) .and_then(|item| item.as_table()) .is_some_and(table_matches_codex_unified_official_provider); if !matches_injected { return Ok(config_text.to_string()); } doc.as_table_mut().remove("model_provider"); let providers_empty = doc["model_providers"] .as_table_mut() .map(|providers| { providers.remove(CC_SWITCH_CODEX_MODEL_PROVIDER_ID); providers.is_empty() }) .unwrap_or(false); if providers_empty { doc.as_table_mut().remove("model_providers"); } Ok(doc.to_string()) } /// 统一会话开关开启时,把官方供应商 `{ auth, config }` 设置对象中的 /// config 文本注入共享 custom 路由;开关关闭或非官方供应商时不做改动。 /// /// 普通 live 写入(`write_codex_live_for_provider`)与代理接管备份 /// (`update_live_backup_from_provider`)两条落盘路径共用:接管期间 /// live 归代理所有,注入必须进备份,接管释放恢复的 live 才带统一路由。 pub fn apply_codex_unified_session_bucket_to_settings( category: Option<&str>, settings: &mut Value, ) -> Result<(), AppError> { if category != Some("official") || !crate::settings::unify_codex_session_history() { return Ok(()); } let config_text = settings .get("config") .and_then(|value| value.as_str()) .unwrap_or("") .to_string(); let injected = inject_codex_unified_session_bucket(&config_text)?; if injected != config_text { if let Some(obj) = settings.as_object_mut() { obj.insert("config".to_string(), Value::String(injected)); } } Ok(()) } /// Backfill helper: strip the unified-session injection from a live /// `{ auth, config }` settings object before it is stored back to the DB. pub fn strip_codex_unified_session_bucket_from_settings( settings: &mut Value, ) -> Result<(), AppError> { let Some(config_text) = settings .get("config") .and_then(|value| value.as_str()) .map(str::to_string) else { return Ok(()); }; let stripped = strip_codex_unified_session_bucket(&config_text)?; if stripped != config_text { if let Some(obj) = settings.as_object_mut() { obj.insert("config".to_string(), Value::String(stripped)); } } Ok(()) } /// Backfill helper: strip `[mcp_servers]` from a live `{ auth, config }` /// settings object before it is stored back to the DB. /// /// MCP 服务器的 SSOT 是 DB 的 mcp_servers 表,live `config.toml` 里的 /// `[mcp_servers]` 只是每次写 live 之后由 MCP 同步重新投影的产物。若回填时 /// 烙进供应商存储配置,已在应用里删除的服务器会随下次激活该供应商被写回 /// live,而逐条 reconcile 只认识 DB 现存条目、永远清不掉这种孤儿。 pub fn strip_codex_mcp_servers_from_settings(settings: &mut Value) -> Result<(), AppError> { let Some(config_text) = settings .get("config") .and_then(|value| value.as_str()) .map(str::to_string) else { return Ok(()); }; if !config_text.contains("mcp") { return Ok(()); } let mut doc = config_text .parse::() .map_err(|e| AppError::Message(format!("Invalid Codex config.toml: {e}")))?; let mut changed = doc.as_table_mut().remove("mcp_servers").is_some(); // 历史错误格式 [mcp.servers] 一并清理(live 侧 MCP 同步也做同样迁移) if let Some(mcp_tbl) = doc.get_mut("mcp").and_then(|item| item.as_table_like_mut()) { if mcp_tbl.remove("servers").is_some() { changed = true; } if mcp_tbl.is_empty() { doc.as_table_mut().remove("mcp"); } } if changed { if let Some(obj) = settings.as_object_mut() { obj.insert("config".to_string(), Value::String(doc.to_string())); } } Ok(()) } /// Route a Codex live write between full auth+config or config-only. /// /// Official providers with usable login material own `auth.json`. Third-party /// providers only touch `config.toml` when the compatibility setting is enabled /// so the user's ChatGPT login cache survives provider switches. /// /// 统一会话开关开启时,官方配置在落盘前注入共享的 `custom` 路由 /// (见 `inject_codex_unified_session_bucket`)。 pub fn write_codex_live_for_provider( category: Option<&str>, auth: &Value, config_text: Option<&str>, ) -> Result<(), AppError> { let unified_official_config = if category == Some("official") && crate::settings::unify_codex_session_history() { Some(inject_codex_unified_session_bucket( config_text.unwrap_or(""), )?) } else { None }; let config_text = unified_official_config.as_deref().or(config_text); let should_write_auth = (category == Some("official") && codex_auth_has_login_material(auth)) || (category != Some("official") && !crate::settings::preserve_codex_official_auth_on_switch()); if should_write_auth { write_codex_live_atomic(auth, config_text) } else { let live_config = prepare_codex_provider_live_config(auth, config_text.unwrap_or(""))?; write_codex_live_config_atomic(Some(&live_config)) } } /// Build the live Codex config for provider switching. /// /// The stored provider keeps its API key in `auth.OPENAI_API_KEY`. Live Codex /// requests can use a provider-scoped `experimental_bearer_token`, so switching /// providers only needs to update `config.toml`; `auth.json` stays as the user's /// long-lived ChatGPT login cache. pub fn prepare_codex_provider_live_config( auth: &Value, config_text: &str, ) -> Result { let token = extract_codex_auth_api_key(auth) .or_else(|| extract_codex_experimental_bearer_token(config_text)); Ok(match token { Some(token) => set_codex_experimental_bearer_token(config_text, &token)?, None => config_text.to_string(), }) } /// During DB backfill, lift a live `experimental_bearer_token` back into /// `auth.OPENAI_API_KEY` so the stored provider keeps its canonical shape /// and generated live tokens don't leak into stored provider TOML. /// /// Only intervenes when the live config actually carries a bearer token — /// otherwise the function is a no-op so the caller's normal backfill path /// (which keeps live `auth` as the authoritative source) is unaffected. pub fn restore_codex_provider_token_for_backfill( settings: &mut Value, template_settings: &Value, ) -> Result<(), AppError> { let Some(config_text) = settings .get("config") .and_then(|value| value.as_str()) .map(str::to_string) else { return Ok(()); }; let Some(token) = extract_codex_experimental_bearer_token(&config_text) else { return Ok(()); }; let cleaned_config = remove_codex_experimental_bearer_token(&config_text)?; if let Some(obj) = settings.as_object_mut() { obj.insert("config".to_string(), Value::String(cleaned_config)); let mut auth = template_settings .get("auth") .filter(|value| value.is_object()) .cloned() .unwrap_or_else(|| Value::Object(serde_json::Map::new())); if let Some(auth_obj) = auth.as_object_mut() { auth_obj.insert("OPENAI_API_KEY".to_string(), Value::String(token)); } obj.insert("auth".to_string(), auth); } Ok(()) } pub fn restore_codex_settings_for_backfill( settings: &mut Value, template_settings: &Value, restore_provider_token: bool, ) -> Result<(), AppError> { if restore_provider_token { restore_codex_provider_token_for_backfill(settings, template_settings)?; } Ok(()) } /// Update a field in Codex config.toml using toml_edit (syntax-preserving). /// /// Supported fields: /// - `"base_url"`: writes to `[model_providers.].base_url` if `model_provider` exists, /// otherwise falls back to top-level `base_url`. /// - `"wire_api"`: writes to `[model_providers.].wire_api` if `model_provider` exists, /// otherwise falls back to top-level `wire_api`. /// - `"model"` / `"model_catalog_json"`: writes to top-level field. /// /// Empty value removes the field. pub fn update_codex_toml_field(toml_str: &str, field: &str, value: &str) -> Result { let mut doc = toml_str .parse::() .map_err(|e| format!("TOML parse error: {e}"))?; let trimmed = value.trim(); match field { "base_url" | "wire_api" => { let model_provider = doc .get("model_provider") .and_then(|item| item.as_str()) .map(str::to_string); if let Some(provider_key) = model_provider { // Ensure [model_providers] table exists if doc.get("model_providers").is_none() { doc["model_providers"] = toml_edit::table(); } if let Some(model_providers) = doc["model_providers"].as_table_mut() { // Ensure [model_providers.] table exists if !model_providers.contains_key(&provider_key) { model_providers[&provider_key] = toml_edit::table(); } if let Some(provider_table) = model_providers[&provider_key].as_table_mut() { if trimmed.is_empty() { provider_table.remove(field); } else { provider_table[field] = toml_edit::value(trimmed); } return Ok(doc.to_string()); } } } // Fallback: no model_provider or structure mismatch → top-level field if trimmed.is_empty() { doc.as_table_mut().remove(field); } else { doc[field] = toml_edit::value(trimmed); } } "model" | "model_catalog_json" => { if trimmed.is_empty() { doc.as_table_mut().remove(field); } else { doc[field] = toml_edit::value(trimmed); } } _ => return Err(format!("unsupported field: {field}")), } Ok(doc.to_string()) } /// Remove `base_url` from the active model_provider section only if it matches `predicate`. /// Also removes top-level `base_url` if it matches. /// Used by proxy cleanup to strip local proxy URLs without touching user-configured URLs. pub fn remove_codex_toml_base_url_if(toml_str: &str, predicate: impl Fn(&str) -> bool) -> String { let mut doc = match toml_str.parse::() { Ok(doc) => doc, Err(_) => return toml_str.to_string(), }; let model_provider = doc .get("model_provider") .and_then(|item| item.as_str()) .map(str::to_string); if let Some(provider_key) = model_provider { if let Some(model_providers) = doc .get_mut("model_providers") .and_then(|v| v.as_table_mut()) { if let Some(provider_table) = model_providers .get_mut(provider_key.as_str()) .and_then(|v| v.as_table_mut()) { let should_remove = provider_table .get("base_url") .and_then(|item| item.as_str()) .map(&predicate) .unwrap_or(false); if should_remove { provider_table.remove("base_url"); } } } } // Fallback: also clean up top-level base_url if it matches let should_remove_root = doc .get("base_url") .and_then(|item| item.as_str()) .map(&predicate) .unwrap_or(false); if should_remove_root { doc.as_table_mut().remove("base_url"); } doc.to_string() } #[cfg(test)] mod tests { use super::*; use serde_json::json; #[test] fn catalog_tool_profile_from_api_format() { assert_eq!( CodexCatalogToolProfile::from_api_format(Some("anthropic")), CodexCatalogToolProfile::Anthropic ); assert_eq!( CodexCatalogToolProfile::from_api_format(Some("openai_responses")), CodexCatalogToolProfile::NativeResponses ); assert_eq!( CodexCatalogToolProfile::from_api_format(Some("openai_chat")), CodexCatalogToolProfile::ProxyChat ); assert_eq!( CodexCatalogToolProfile::from_api_format(None), CodexCatalogToolProfile::ProxyChat ); } #[test] fn unified_session_bucket_injects_for_empty_official_config() { let injected = inject_codex_unified_session_bucket("").expect("inject"); let doc: toml::Table = toml::from_str(&injected).expect("parse injected config"); assert_eq!( doc.get("model_provider").and_then(|v| v.as_str()), Some(CC_SWITCH_CODEX_MODEL_PROVIDER_ID) ); let custom = doc["model_providers"][CC_SWITCH_CODEX_MODEL_PROVIDER_ID] .as_table() .expect("custom provider table"); assert_eq!(custom.get("name").and_then(|v| v.as_str()), Some("OpenAI")); assert_eq!( custom.get("requires_openai_auth").and_then(|v| v.as_bool()), Some(true) ); assert_eq!( custom.get("supports_websockets").and_then(|v| v.as_bool()), Some(true) ); assert_eq!( custom.get("wire_api").and_then(|v| v.as_str()), Some("responses") ); } #[test] fn official_proxy_route_uses_native_auth_and_local_responses_provider() { let input = r#"model = "gpt-5.4" experimental_bearer_token = "PROXY_MANAGED" [mcp_servers.example] command = "example" "#; let output = apply_codex_official_proxy_route(input, "http://127.0.0.1:15721/v1") .expect("apply official proxy route"); let doc: toml::Value = toml::from_str(&output).expect("parse output"); assert_eq!( doc.get("model_provider").and_then(toml::Value::as_str), Some(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) ); assert!(doc.get("experimental_bearer_token").is_none()); assert!( doc.get("mcp_servers").is_some(), "unrelated config survives" ); let provider = &doc["model_providers"][CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID]; assert_eq!( provider.get("base_url").and_then(toml::Value::as_str), Some("http://127.0.0.1:15721/v1") ); assert_eq!( provider .get("requires_openai_auth") .and_then(toml::Value::as_bool), Some(true) ); assert_eq!( provider .get("supports_websockets") .and_then(toml::Value::as_bool), Some(false) ); assert!(codex_config_has_official_proxy_route(&output)); } #[test] fn official_proxy_route_cleanup_only_removes_owned_provider() { let projected = apply_codex_official_proxy_route("model = \"gpt-5.4\"\n", "http://127.0.0.1:15721/v1") .expect("project"); let cleaned = remove_codex_official_proxy_route(&projected).expect("clean"); let doc: toml::Value = toml::from_str(&cleaned).expect("parse cleaned"); assert!(doc.get("model_provider").is_none()); assert!(doc.get("model_providers").is_none()); assert_eq!( doc.get("model").and_then(toml::Value::as_str), Some("gpt-5.4") ); } #[test] fn official_proxy_route_rejects_non_table_model_providers_without_panicking() { for input in [ "model_providers = 3\n", "[[model_providers]]\nname = \"broken\"\n", ] { let result = apply_codex_official_proxy_route(input, "http://127.0.0.1:15721/v1"); assert!(result.is_err()); } } #[test] fn official_proxy_route_normalizes_inline_tables_and_cleans_stale_placeholder() { let input = r#"model_provider = "rightcode" model_providers = { rightcode = { name = "RightCode", experimental_bearer_token = "PROXY_MANAGED" } } "#; let projected = apply_codex_official_proxy_route(input, "http://127.0.0.1:15721/v1") .expect("project inline provider table"); let projected_doc: toml::Value = toml::from_str(&projected).expect("parse projected"); assert!(projected_doc["model_providers"]["rightcode"] .get("experimental_bearer_token") .is_none()); assert!(projected_doc["model_providers"] .get(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) .is_some()); let cleaned = remove_codex_official_proxy_route(&projected).expect("clean projected"); let cleaned_doc: toml::Value = toml::from_str(&cleaned).expect("parse cleaned"); assert!(cleaned_doc.get("model_provider").is_none()); assert!(cleaned_doc["model_providers"].get("rightcode").is_some()); assert!(cleaned_doc["model_providers"] .get(CC_SWITCH_CODEX_OFFICIAL_PROXY_PROVIDER_ID) .is_none()); } #[test] fn unified_session_bucket_preserves_other_keys_and_explicit_routing() { let with_catalog = "model_catalog_json = \"cc-switch-model-catalog.json\"\n"; let injected = inject_codex_unified_session_bucket(with_catalog).expect("inject"); assert!(injected.contains("model_catalog_json")); assert!(injected.contains("model_provider = \"custom\"")); // 用户显式指定过 model_provider 的官方配置不被覆盖 let explicit = "model_provider = \"openai_https\"\n"; let unchanged = inject_codex_unified_session_bucket(explicit).expect("inject"); assert_eq!(unchanged, explicit); } #[test] fn unified_session_bucket_skips_conflicting_custom_table() { // 残留的非注入形态 custom 表:设置 model_provider 会把官方流量 // 路由到表里的第三方端点,必须整体拒绝注入。 let stale = r#"[model_providers.custom] name = "Relay" base_url = "https://relay.example/v1" "#; let unchanged = inject_codex_unified_session_bucket(stale).expect("inject"); assert_eq!(unchanged, stale); // 已是注入形态的 custom 表(如重复注入)则照常补上 model_provider let injected_once = inject_codex_unified_session_bucket("").expect("inject"); let reinjected = inject_codex_unified_session_bucket(&injected_once).expect("re-inject"); assert_eq!(reinjected, injected_once); } #[test] fn unified_session_bucket_strip_round_trips_injection() { let injected = inject_codex_unified_session_bucket("").expect("inject"); let stripped = strip_codex_unified_session_bucket(&injected).expect("strip"); assert_eq!(stripped.trim(), ""); let with_catalog = "model_catalog_json = \"cc-switch-model-catalog.json\"\n"; let injected = inject_codex_unified_session_bucket(with_catalog).expect("inject"); let stripped = strip_codex_unified_session_bucket(&injected).expect("strip"); assert_eq!(stripped, with_catalog); } #[test] fn unified_session_bucket_strip_keeps_third_party_custom_entry() { // 第三方模板同样用 custom 路由,但条目带 base_url 等差异字段, // 形态不等于注入产物,必须原样保留。 let third_party = r#"model_provider = "custom" [model_providers.custom] name = "Relay" base_url = "https://relay.example/v1" wire_api = "responses" requires_openai_auth = true "#; let untouched = strip_codex_unified_session_bucket(third_party).expect("strip"); assert_eq!(untouched, third_party); } #[test] fn unified_session_bucket_strip_from_settings_only_touches_config() { let injected = inject_codex_unified_session_bucket("").expect("inject"); let mut settings = json!({ "auth": { "tokens": { "access_token": "secret" } }, "config": injected, }); strip_codex_unified_session_bucket_from_settings(&mut settings).expect("strip settings"); assert_eq!( settings .get("config") .and_then(|v| v.as_str()) .map(str::trim), Some("") ); assert!(settings.pointer("/auth/tokens/access_token").is_some()); } #[test] fn strip_mcp_servers_from_settings_removes_table_and_legacy_form() { let mut settings = json!({ "auth": { "OPENAI_API_KEY": "sk-test" }, "config": "# user comment\nmodel = \"gpt-5.5\"\n\n[mcp_servers.echo]\ntype = \"stdio\"\ncommand = \"echo\"\n\n[mcp.servers.legacy]\ncommand = \"noop\"\n", }); strip_codex_mcp_servers_from_settings(&mut settings).expect("strip mcp"); let config = settings .get("config") .and_then(|v| v.as_str()) .expect("config text"); assert!(!config.contains("mcp_servers"), "got: {config}"); assert!( !config.contains("[mcp"), "legacy [mcp.servers] gone: {config}" ); assert!(config.contains("# user comment"), "comments preserved"); assert!(config.contains("model = \"gpt-5.5\"")); } #[test] fn strip_mcp_servers_from_settings_is_noop_without_mcp() { let original = "# comment\nmodel = \"gpt-5.5\"\n"; let mut settings = json!({ "auth": {}, "config": original, }); strip_codex_mcp_servers_from_settings(&mut settings).expect("strip mcp"); assert_eq!( settings.get("config").and_then(|v| v.as_str()), Some(original), "config text must be byte-identical when nothing is stripped" ); } #[test] fn extract_base_url_prefers_active_provider_section() { let input = r#"model_provider = "azure" [model_providers.azure] base_url = "https://azure.example.com/v1" [model_providers.other] base_url = "https://other.example.com/v1" "#; assert_eq!( extract_codex_base_url(input).as_deref(), Some("https://azure.example.com/v1") ); } #[test] fn extract_base_url_falls_back_to_top_level_only() { let top_level = r#"base_url = "https://top-level.example.com/v1""#; assert_eq!( extract_codex_base_url(top_level).as_deref(), Some("https://top-level.example.com/v1") ); } // Mirrors the frontend extractCodexBaseUrl: a non-active provider section // is never a credential source, whether the active provider points // elsewhere (e.g. the built-in "openai") or none is selected at all. #[test] fn extract_base_url_ignores_non_active_provider_sections() { let mismatched = r#"model_provider = "openai" [model_providers.custom] base_url = "https://leftover.example.com/v1" "#; assert_eq!(extract_codex_base_url(mismatched), None); let no_active = r#"[model_providers.any] base_url = "https://single.example.com/v1" "#; assert_eq!(extract_codex_base_url(no_active), None); } #[test] fn prepare_provider_live_config_rejects_key_without_config() { let err = prepare_codex_provider_live_config(&json!({"OPENAI_API_KEY": "sk-test"}), "") .expect_err("empty config with API key should not truncate live config"); assert!( err.to_string().contains("config.toml"), "error should explain missing config.toml, got: {err}" ); } #[test] fn prepare_provider_live_config_uses_top_level_token_for_reserved_provider() { let input = r#"model_provider = "openai" model = "gpt-5" "#; let output = prepare_codex_provider_live_config(&json!({"OPENAI_API_KEY": "sk-test"}), input) .expect("prepare live config"); let parsed: toml::Value = toml::from_str(&output).expect("parse output"); assert_eq!( parsed .get("experimental_bearer_token") .and_then(|v| v.as_str()), Some("sk-test") ); assert!( parsed.get("model_providers").is_none(), "reserved provider tables should not be synthesized" ); } #[test] fn extract_bearer_uses_top_level_token_for_reserved_provider() { let input = r#"model_provider = "openai" experimental_bearer_token = "top-level-key" [model_providers.openai] experimental_bearer_token = "stale-table-key" "#; assert_eq!( extract_codex_experimental_bearer_token(input).as_deref(), Some("top-level-key") ); } #[test] fn should_not_restore_provider_token_for_oauth_only_template() { let oauth_template = json!({ "auth": { "auth_mode": "chatgpt", "tokens": { "access_token": "oauth-access" } } }); let api_key_template = json!({ "auth": { "OPENAI_API_KEY": "sk-test" } }); assert!( !should_restore_codex_provider_token_for_backfill(Some("custom"), &oauth_template), "OAuth-only templates should not backfill bearer tokens into OPENAI_API_KEY" ); assert!( should_restore_codex_provider_token_for_backfill(Some("custom"), &api_key_template), "custom API-key providers should still restore provider bearer tokens" ); assert!( !should_restore_codex_provider_token_for_backfill(Some("official"), &api_key_template), "official providers should never restore third-party bearer tokens" ); } #[test] fn prepare_provider_live_config_does_not_create_incomplete_provider_table() { let input = r#"model_provider = "vendor_x" model = "gpt-5" "#; let output = prepare_codex_provider_live_config(&json!({"OPENAI_API_KEY": "sk-test"}), input) .expect("prepare live config"); let parsed: toml::Value = toml::from_str(&output).expect("parse output"); assert_eq!( parsed .get("experimental_bearer_token") .and_then(|v| v.as_str()), Some("sk-test") ); assert!( parsed.get("model_providers").is_none(), "missing provider tables should not be synthesized without endpoint fields" ); } #[test] fn prepare_provider_live_config_preserves_custom_provider_id() { let input = r#"model_provider = "vendor_alpha" model = "gpt-5.4" profile = "work" [model_providers.vendor_alpha] name = "Vendor Alpha" base_url = "https://alpha.example/v1" wire_api = "responses" [profiles.work] model_provider = "vendor_alpha" model = "gpt-5.4" "#; let result = prepare_codex_provider_live_config(&json!({"OPENAI_API_KEY": "sk-test"}), input) .expect("prepare live config"); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed.get("model_provider").and_then(|v| v.as_str()), Some("vendor_alpha") ); assert!( parsed .get("model_providers") .and_then(|v| v.get("custom")) .is_none(), "provider writes should not force custom provider ids" ); assert_eq!( parsed .get("model_providers") .and_then(|v| v.get("vendor_alpha")) .and_then(|v| v.get("experimental_bearer_token")) .and_then(|v| v.as_str()), Some("sk-test") ); assert_eq!( parsed .get("profiles") .and_then(|v| v.get("work")) .and_then(|v| v.get("model_provider")) .and_then(|v| v.as_str()), Some("vendor_alpha"), "profile provider references should be preserved" ); } #[test] fn backfill_preserves_live_model_provider_id() { let mut live_settings = json!({ "auth": {}, "config": r#"model_provider = "vendor_beta" [model_providers.vendor_beta] name = "Vendor Beta" base_url = "https://beta.example/v1" wire_api = "responses" "#, }); let template_settings = json!({ "auth": {}, "config": r#"model_provider = "custom" [model_providers.custom] name = "Custom" base_url = "https://custom.example/v1" wire_api = "responses" "#, }); restore_codex_settings_for_backfill(&mut live_settings, &template_settings, false).unwrap(); let config = live_settings.get("config").and_then(Value::as_str).unwrap(); let parsed: toml::Value = toml::from_str(config).unwrap(); assert_eq!( parsed.get("model_provider").and_then(|v| v.as_str()), Some("vendor_beta") ); assert!( parsed .get("model_providers") .and_then(|v| v.get("vendor_beta")) .is_some(), "backfill should not rewrite user-selected provider tables" ); } #[test] fn base_url_writes_into_correct_model_provider_section() { let input = r#"model_provider = "any" model = "gpt-5.1-codex" [model_providers.any] name = "any" wire_api = "responses" "#; let result = update_codex_toml_field(input, "base_url", "https://example.com/v1").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); let base_url = parsed .get("model_providers") .and_then(|v| v.get("any")) .and_then(|v| v.get("base_url")) .and_then(|v| v.as_str()) .expect("base_url should be in model_providers.any"); assert_eq!(base_url, "https://example.com/v1"); // Should NOT have top-level base_url assert!(parsed.get("base_url").is_none()); // wire_api preserved let wire_api = parsed .get("model_providers") .and_then(|v| v.get("any")) .and_then(|v| v.get("wire_api")) .and_then(|v| v.as_str()); assert_eq!(wire_api, Some("responses")); } #[test] fn wire_api_writes_into_correct_model_provider_section() { let input = r#"model_provider = "chat_only" model = "gpt-5.1-codex" [model_providers.chat_only] name = "Chat Only" base_url = "https://example.com/v1" wire_api = "chat" "#; let result = update_codex_toml_field(input, "wire_api", "responses").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); let provider = parsed .get("model_providers") .and_then(|v| v.get("chat_only")) .expect("model_providers.chat_only should exist"); assert_eq!( provider.get("wire_api").and_then(|v| v.as_str()), Some("responses") ); assert_eq!( provider.get("base_url").and_then(|v| v.as_str()), Some("https://example.com/v1") ); assert!(parsed.get("wire_api").is_none()); } #[test] fn base_url_creates_section_when_missing() { let input = r#"model_provider = "custom" model = "gpt-4" "#; let result = update_codex_toml_field(input, "base_url", "https://custom.api/v1").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); let base_url = parsed .get("model_providers") .and_then(|v| v.get("custom")) .and_then(|v| v.get("base_url")) .and_then(|v| v.as_str()) .expect("should create section and set base_url"); assert_eq!(base_url, "https://custom.api/v1"); } #[test] fn base_url_falls_back_to_top_level_without_model_provider() { let input = r#"model = "gpt-4" "#; let result = update_codex_toml_field(input, "base_url", "https://fallback.api/v1").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); let base_url = parsed .get("base_url") .and_then(|v| v.as_str()) .expect("should set top-level base_url"); assert_eq!(base_url, "https://fallback.api/v1"); } #[test] fn clearing_base_url_removes_only_from_correct_section() { let input = r#"model_provider = "any" [model_providers.any] name = "any" base_url = "https://old.api/v1" wire_api = "responses" [mcp_servers.context7] command = "npx" "#; let result = update_codex_toml_field(input, "base_url", "").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); // base_url removed from model_providers.any let any_section = parsed .get("model_providers") .and_then(|v| v.get("any")) .expect("model_providers.any should exist"); assert!(any_section.get("base_url").is_none()); // wire_api preserved assert_eq!( any_section.get("wire_api").and_then(|v| v.as_str()), Some("responses") ); // mcp_servers untouched assert!(parsed.get("mcp_servers").is_some()); } #[test] fn model_field_operates_on_top_level() { let input = r#"model_provider = "any" model = "gpt-4" [model_providers.any] name = "any" "#; let result = update_codex_toml_field(input, "model", "gpt-5").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!(parsed.get("model").and_then(|v| v.as_str()), Some("gpt-5")); // Clear model let result2 = update_codex_toml_field(&result, "model", "").unwrap(); let parsed2: toml::Value = toml::from_str(&result2).unwrap(); assert!(parsed2.get("model").is_none()); } #[test] fn preserves_comments_and_whitespace() { let input = r#"# My Codex config model_provider = "any" model = "gpt-4" # Provider section [model_providers.any] name = "any" base_url = "https://old.api/v1" "#; let result = update_codex_toml_field(input, "base_url", "https://new.api/v1").unwrap(); // Comments should be preserved assert!(result.contains("# My Codex config")); assert!(result.contains("# Provider section")); } #[test] fn does_not_misplace_when_profiles_section_follows() { let input = r#"model_provider = "any" [model_providers.any] name = "any" base_url = "https://old.api/v1" [profiles.default] model = "gpt-4" "#; let result = update_codex_toml_field(input, "base_url", "https://new.api/v1").unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); // base_url in correct section let base_url = parsed .get("model_providers") .and_then(|v| v.get("any")) .and_then(|v| v.get("base_url")) .and_then(|v| v.as_str()); assert_eq!(base_url, Some("https://new.api/v1")); // profiles section untouched let profile_model = parsed .get("profiles") .and_then(|v| v.get("default")) .and_then(|v| v.get("model")) .and_then(|v| v.as_str()); assert_eq!(profile_model, Some("gpt-4")); } #[test] fn remove_base_url_if_predicate() { let input = r#"model_provider = "any" [model_providers.any] name = "any" base_url = "http://127.0.0.1:5000/v1" wire_api = "responses" "#; let result = remove_codex_toml_base_url_if(input, |url| url.starts_with("http://127.0.0.1")); let parsed: toml::Value = toml::from_str(&result).unwrap(); let any_section = parsed .get("model_providers") .and_then(|v| v.get("any")) .unwrap(); assert!(any_section.get("base_url").is_none()); assert_eq!( any_section.get("wire_api").and_then(|v| v.as_str()), Some("responses") ); } #[test] fn remove_base_url_if_keeps_non_matching() { let input = r#"model_provider = "any" [model_providers.any] base_url = "https://production.api/v1" "#; let result = remove_codex_toml_base_url_if(input, |url| url.starts_with("http://127.0.0.1")); let parsed: toml::Value = toml::from_str(&result).unwrap(); let base_url = parsed .get("model_providers") .and_then(|v| v.get("any")) .and_then(|v| v.get("base_url")) .and_then(|v| v.as_str()); assert_eq!(base_url, Some("https://production.api/v1")); } #[test] fn dynamic_template_backfills_parser_required_fields_from_static() { // Simulate a template cloned from a models_cache.json written by a // Codex build whose ModelInfo lacks parser-side required fields such // as `supports_reasoning_summaries` (codex >= 0.144.5 rejects the // whole catalog file without it). let mut template = json!({ "slug": "gpt-5.5", "context_window": 272_000, "supports_parallel_tool_calls": false }); fill_template_fields_from_static(&mut template); assert_eq!( template .get("supports_reasoning_summaries") .and_then(Value::as_bool), Some(true) ); // Keys already present in the dynamic template are never overwritten. assert_eq!( template .get("supports_parallel_tool_calls") .and_then(Value::as_bool), Some(false) ); assert_eq!( template.get("context_window").and_then(Value::as_u64), Some(272_000) ); // Optional capability fields must NOT be backfilled: for the catalog // parser "missing" means the parser default, not the static // template's value. assert!(template.get("supports_search_tool").is_none()); assert!(template.get("supports_image_detail_original").is_none()); assert!(template.get("web_search_tool_type").is_none()); } #[test] fn proxy_chat_catalog_entries_carry_reasoning_summaries_flag() { // End to end: a stale dynamic template, once backfilled, must yield // catalog entries codex 0.144.5+ can parse. let mut template = json!({ "slug": "gpt-5.5" }); fill_template_fields_from_static(&mut template); let specs = vec![CodexCatalogModelSpec { model: "k3".to_string(), display_name: "Kimi K3".to_string(), context_window: 262_144, supports_parallel_tool_calls: None, input_modalities: None, base_instructions: None, }]; let catalog = codex_model_catalog_from_specs(&specs, &template, CodexCatalogToolProfile::ProxyChat); assert_eq!( catalog["models"][0] .get("supports_reasoning_summaries") .and_then(Value::as_bool), Some(true) ); } #[test] fn codex_model_catalog_uses_provider_models_and_context() { let template = json!({ "slug": "gpt-5.5", "display_name": "GPT-5.5", "description": "Frontier model", "base_instructions": "gpt-5.5 base instructions", "model_messages": { "instructions_template": "gpt-5.5 instructions template", "instructions_variables": { "personality_default": "", "personality_friendly": "", "personality_pragmatic": "" } }, "additional_speed_tiers": ["fast"], "service_tiers": [ { "id": "priority", "name": "Fast", "description": "1.5x speed, increased usage" } ], "availability_nux": { "message": "GPT-5.5 is now available." }, "upgrade": { "target": "gpt-5.5" }, "context_window": 272000, "max_context_window": 272000 }); let settings = json!({ "modelCatalog": { "models": [ { "model": "deepseek-v4-flash", "displayName": "DeepSeek V4 Flash", "contextWindow": "64000" }, { "model": "kimi-k2", "display_name": "Kimi K2" } ] } }); let specs = codex_catalog_model_specs(&settings, r#"model_context_window = 128000"#); let catalog = codex_model_catalog_from_specs(&specs, &template, CodexCatalogToolProfile::ProxyChat); let models = catalog .get("models") .and_then(|value| value.as_array()) .expect("models should be an array"); assert_eq!(models.len(), 2); assert_eq!( models[0].get("slug").and_then(|value| value.as_str()), Some("deepseek-v4-flash") ); assert_eq!( models[0] .get("context_window") .and_then(|value| value.as_u64()), Some(64_000) ); assert_eq!( models[1] .get("context_window") .and_then(|value| value.as_u64()), Some(128_000) ); assert!( models[0].get("model_messages").is_some(), "Codex requires model_messages in custom catalogs" ); assert_eq!( models[0] .get("base_instructions") .and_then(|value| value.as_str()), Some("gpt-5.5 base instructions") ); assert_eq!( models[0].get("model_messages"), template.get("model_messages"), "custom catalog entries should keep the gpt-5.5 agent template" ); assert_eq!( models[0].get("additional_speed_tiers"), Some(&json!([])), "generated third-party entries should not inherit OpenAI speed tiers" ); assert!( models[0] .get("availability_nux") .is_some_and(|value| value.is_null()), "generated third-party entries should not inherit GPT-5.5 launch messaging" ); } #[test] fn native_responses_profile_suppresses_apply_patch_and_keeps_shell() { // Native (direct) /responses providers must NOT emit a freeform // apply_patch (type=="custom") tool — gateways like MiMo reject it. // The native profile uses the bundled clean template and relies on // shell_type="shell_command" for edits, plus per-row overrides. let settings = json!({ "modelCatalog": { "models": [ { "model": "MiniMax-M3", "displayName": "MiniMax-M3", "contextWindow": 1_000_000, "supportsParallelToolCalls": true, "inputModalities": ["text", "image"], "baseInstructions": "You are Codex, a coding agent based on MiniMax-M3." } ] } }); let catalog = codex_model_catalog_from_settings( &settings, "", CodexCatalogToolProfile::NativeResponses, ) .expect("native catalog generation should not error") .expect("non-empty modelCatalog must yield a catalog"); let entry = &catalog["models"][0]; assert_eq!( entry.get("slug").and_then(|v| v.as_str()), Some("MiniMax-M3") ); assert_eq!( entry.get("shell_type").and_then(|v| v.as_str()), Some("shell_command"), "native entries edit via shell, not the custom apply_patch tool" ); assert!( entry.get("apply_patch_tool_type").is_none(), "native entries must NOT declare a freeform apply_patch tool" ); // `base_instructions` is REQUIRED by Codex's catalog parser, so it must // be present — and the per-row official override must win over the // template default. assert_eq!( entry.get("base_instructions").and_then(|v| v.as_str()), Some("You are Codex, a coding agent based on MiniMax-M3."), "per-row baseInstructions override must apply (and field must exist)" ); assert!( entry.get("model_messages").is_none(), "native entries must not carry the gpt-5.5 model_messages persona text" ); assert_eq!( entry.get("supports_parallel_tool_calls"), Some(&json!(true)), "per-row supportsParallelToolCalls override must apply" ); assert_eq!( entry.get("input_modalities"), Some(&json!(["text", "image"])), "per-row inputModalities override must apply" ); assert_eq!( entry.get("context_window").and_then(|v| v.as_u64()), Some(1_000_000) ); } #[test] fn catalog_infers_image_input_independently_of_tool_profile() { // Start from a deliberately text-only template to prove that every // profile overwrites template defaults with shared capability logic. let template = json!({ "input_modalities": ["text"], "apply_patch_tool_type": "freeform" }); let specs = vec![ CodexCatalogModelSpec { model: "gpt-5.4".to_string(), display_name: "GPT 5.4".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: None, base_instructions: None, }, CodexCatalogModelSpec { model: "deepseek/deepseek-v4-pro".to_string(), display_name: "DeepSeek V4 Pro".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: None, base_instructions: None, }, CodexCatalogModelSpec { model: "glm-5.2v".to_string(), display_name: "GLM 5.2V".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: None, base_instructions: None, }, CodexCatalogModelSpec { model: "deepseek-v4-flash".to_string(), display_name: "Explicit Visual Override".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: Some(vec!["text".to_string(), "image".to_string()]), base_instructions: None, }, CodexCatalogModelSpec { model: "custom-text-alias".to_string(), display_name: "Explicit Text Override".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: Some(vec!["text".to_string()]), base_instructions: None, }, ]; for profile in [ CodexCatalogToolProfile::ProxyChat, CodexCatalogToolProfile::NativeResponses, CodexCatalogToolProfile::Anthropic, ] { let catalog = codex_model_catalog_from_specs(&specs, &template, profile); let models = catalog["models"].as_array().expect("models array"); let modalities = |slug: &str| { models .iter() .find(|entry| entry["slug"] == slug) .and_then(|entry| entry.get("input_modalities")) .cloned() .unwrap_or(Value::Null) }; assert_eq!(modalities("gpt-5.4"), json!(["text", "image"])); assert_eq!(modalities("deepseek/deepseek-v4-pro"), json!(["text"])); assert_eq!(modalities("glm-5.2v"), json!(["text", "image"])); assert_eq!( modalities("deepseek-v4-flash"), json!(["text", "image"]), "explicit provider metadata must override the text-only registry" ); assert_eq!(modalities("custom-text-alias"), json!(["text"])); } } #[test] fn native_responses_catalog_always_carries_base_instructions() { // Regression guard for the "missing field `base_instructions`" parse // error: Codex refuses to load a model catalog whose entries lack // base_instructions. Synthesized presets carry no per-row override, so // the entry MUST inherit the template's neutral default rather than // dropping the field entirely. let settings = json!({ "modelCatalog": { "models": [{ "model": "qwen3-coder-plus" }] } }); let catalog = codex_model_catalog_from_settings( &settings, "", CodexCatalogToolProfile::NativeResponses, ) .expect("native catalog generation should not error") .expect("non-empty modelCatalog must yield a catalog"); let base = catalog["models"][0] .get("base_instructions") .and_then(|v| v.as_str()); assert!( base.is_some_and(|s| !s.trim().is_empty()), "every native entry must carry a non-empty base_instructions (Codex requires it)" ); } #[test] fn proxy_chat_profile_still_keeps_apply_patch() { // Regression guard for Mode A: the proxy-chat profile must keep the // freeform apply_patch tool (the proxy rewrites custom<->function). let template = load_codex_native_responses_template(); let specs = vec![CodexCatalogModelSpec { model: "x".to_string(), display_name: "x".to_string(), context_window: 128_000, supports_parallel_tool_calls: None, input_modalities: None, base_instructions: None, }]; // Using a gpt-5.5-shaped template under ProxyChat must NOT strip // apply_patch_tool_type. (The native template lacks it, so synthesize // one with the field present to prove ProxyChat leaves it intact.) let mut proxy_template = template.clone(); proxy_template["apply_patch_tool_type"] = json!("freeform"); let catalog = codex_model_catalog_from_specs( &specs, &proxy_template, CodexCatalogToolProfile::ProxyChat, ); assert_eq!( catalog["models"][0] .get("apply_patch_tool_type") .and_then(|v| v.as_str()), Some("freeform"), "ProxyChat must preserve apply_patch_tool_type (no native stripping)" ); } #[test] fn model_catalog_json_field_writes_relative_filename() { let input = r#"model_provider = "any" [model_providers.any] name = "any" "#; let catalog_path = Path::new("/tmp/cc-switch-model-catalog.json"); let result = set_codex_model_catalog_json_field(input, Some(catalog_path)).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed .get("model_catalog_json") .and_then(|value| value.as_str()), Some(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME) ); assert!( parsed .get("model_providers") .and_then(|value| value.get("any")) .and_then(|value| value.get("model_catalog_json")) .is_none(), "model_catalog_json should stay top-level" ); } #[test] fn native_web_search_field_disables_at_top_level() { // Native `/responses` gateways reject the web_search tool, so the // NativeResponses profile must write the top-level disable line even // when sections are present (it must NOT land inside a section). let input = r#"model_provider = "custom" [model_providers.custom] name = "xiaomi_mimo" "#; let result = set_codex_native_web_search_field(input, true).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed.get("web_search").and_then(|value| value.as_str()), Some("disabled") ); assert!( parsed .get("model_providers") .and_then(|value| value.get("custom")) .and_then(|value| value.get("web_search")) .is_none(), "web_search should stay top-level" ); } #[test] fn native_web_search_field_removes_own_sentinel_when_not_disabled() { // Switching away from a native provider must re-enable web search by // removing cc-switch's own "disabled" sentinel. let input = r#"model = "gpt-5.5" web_search = "disabled" "#; let result = set_codex_native_web_search_field(input, false).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert!( parsed.get("web_search").is_none(), "cc-switch's disabled sentinel should be removed when not native" ); } #[test] fn native_web_search_field_preserves_user_value() { // A user's own web_search value must never be clobbered by cleanup, // only cc-switch's "disabled" sentinel is owned/removable. let input = r#"web_search = "enabled" "#; let result = set_codex_native_web_search_field(input, false).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed.get("web_search").and_then(|value| value.as_str()), Some("enabled"), "a user-set web_search value must be preserved" ); } #[test] fn anthropic_profile_disables_web_search_without_catalog() { // Regression: even when no model catalog is generated (empty/absent // modelCatalog), an Anthropic provider must still disable web_search — the // Responses→Anthropic transform drops the hosted tool, so leaving it on // exposes a dead tool. The None-catalog branch previously always left it on. let config = "model = \"claude-sonnet-4-6\"\n"; let settings = serde_json::json!({}); let anthropic = prepare_codex_config_text_with_model_catalog( &settings, config, CodexCatalogToolProfile::Anthropic, ) .unwrap(); let parsed: toml::Value = toml::from_str(&anthropic).unwrap(); assert_eq!( parsed.get("web_search").and_then(|v| v.as_str()), Some("disabled"), "Anthropic profile must disable web_search even with no catalog" ); // ProxyChat on the same no-catalog path must NOT add a disable line. let proxy = prepare_codex_config_text_with_model_catalog( &settings, config, CodexCatalogToolProfile::ProxyChat, ) .unwrap(); let parsed: toml::Value = toml::from_str(&proxy).unwrap(); assert!( parsed.get("web_search").is_none(), "ProxyChat profile must not disable web_search on the no-catalog path" ); } #[test] fn web_search_blacklist_disables_only_known_reject_gateways() { let cfg = |model: &str, base_url: &str| { format!( "model_provider = \"custom\"\nmodel = \"{model}\"\n\n[model_providers.custom]\nname = \"x\"\nbase_url = \"{base_url}\"\nwire_api = \"responses\"\n" ) }; // Blacklisted by host (first-party reject gateways) → disable. for (model, host) in [ ("mimo-v2.5-pro", "https://api.xiaomimimo.com/v1"), ("mimo-v2.5", "https://token-plan-cn.xiaomimimo.com/v1"), ("LongCat-2.0", "https://api.longcat.chat/openai/v1"), ("MiniMax-M3", "https://api.minimax.io/v1"), ("MiniMax-M3", "https://api.minimaxi.com/v1"), ] { assert!( codex_native_gateway_rejects_web_search(&cfg(model, host)), "{host} should be blacklisted" ); } // Blacklisted by MODEL brand even on an aggregator host (SiliconFlow // fronting a reject vendor's model) → disable. for (model, host) in [ ("MiniMax-M3", "https://api.siliconflow.cn/v1"), ("MiniMaxAI/MiniMax-M3", "https://api.siliconflow.cn/v1"), ("mimo-v2.5-pro", "https://some-aggregator.example/v1"), ( "qwen/qwen3-coder-plus", "https://some-aggregator.example/v1", ), ] { assert!( codex_native_gateway_rejects_web_search(&cfg(model, host)), "{model} @ {host} should be blacklisted by model brand" ); } // Qwen3-Coder is blacklisted by model, not by DashScope host. This keeps // general Qwen models that support built-in web_search on the same host // enabled while protecting the native qwen3-coder-plus preset. assert!(codex_native_gateway_rejects_web_search(&cfg( "qwen3-coder-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1", ))); assert!(!codex_native_gateway_rejects_web_search(&cfg( "qwen3.7-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1", ))); // NOT blacklisted → keep Codex default (relays/GPT, DouBao, general Qwen, // and any unknown provider incl. an aggregator serving a non-reject model). for (model, host) in [ ("gpt-5.5", "https://www.packyapi.com/v1"), ("gpt-5-codex", "https://aihubmix.com/v1"), ( "doubao-seed-2-1-pro-260628", "https://ark.cn-beijing.volces.com/api/v3", ), ("Pro/moonshotai/Kimi-K2.6", "https://api.siliconflow.cn/v1"), ] { assert!( !codex_native_gateway_rejects_web_search(&cfg(model, host)), "{model} @ {host} should NOT be blacklisted" ); } } #[test] fn resolve_catalog_path_returns_none_when_config_missing_field() { let generated = PathBuf::from("/tmp/.codex/cc-switch-model-catalog.json"); assert!(resolve_cc_switch_catalog_path("", &generated).is_none()); assert!( resolve_cc_switch_catalog_path("model = \"gpt-5\"", &generated).is_none(), "no model_catalog_json field should yield None" ); } #[test] fn resolve_catalog_path_accepts_cc_switch_owned_file() { let generated = PathBuf::from("/tmp/.codex/cc-switch-model-catalog.json"); let config = r#"model_catalog_json = "/tmp/.codex/cc-switch-model-catalog.json" "#; let resolved = resolve_cc_switch_catalog_path(config, &generated).expect("path resolves"); assert_eq!(resolved, generated); } #[test] fn resolve_catalog_path_rejects_user_owned_external_file() { let generated = PathBuf::from("/tmp/.codex/cc-switch-model-catalog.json"); let config = r#"model_catalog_json = "/Users/me/.codex/my-handwritten-catalog.json" "#; assert!( resolve_cc_switch_catalog_path(config, &generated).is_none(), "external catalog files should be left alone" ); } #[test] fn build_simplified_catalog_round_trips_user_input() { let config = ""; let catalog = r#"{ "models": [ { "slug": "deepseek-v4-pro", "display_name": "deepseek-v4-pro", "context_window": 1000000 }, { "slug": "deepseek-v4-flash", "display_name": "DeepSeek Flash", "context_window": 1000000 } ] }"#; let result = build_simplified_catalog_from_texts(config, catalog).expect("entries found"); let models = result .get("models") .and_then(|m| m.as_array()) .expect("models array"); assert_eq!(models.len(), 2); // First entry: display_name == slug → displayName squashed; explicit // context_window != default 128_000 → preserved. assert_eq!( models[0].get("model").and_then(|v| v.as_str()), Some("deepseek-v4-pro") ); assert!(models[0].get("displayName").is_none()); assert_eq!( models[0].get("contextWindow").and_then(|v| v.as_u64()), Some(1_000_000) ); // Second entry: display_name distinct from slug → preserved. assert_eq!( models[1].get("displayName").and_then(|v| v.as_str()), Some("DeepSeek Flash") ); } #[test] fn build_simplified_catalog_squashes_default_context_window() { // Default fallback is 128_000 when config.toml has no model_context_window. let catalog = r#"{ "models": [{ "slug": "kimi", "display_name": "kimi", "context_window": 128000 }] }"#; let result = build_simplified_catalog_from_texts("", catalog).expect("entry"); let entry = &result.get("models").unwrap().as_array().unwrap()[0]; assert!( entry.get("contextWindow").is_none(), "default 128_000 should be squashed so the form shows blank, matching the user's blank input" ); } #[test] fn build_simplified_catalog_respects_explicit_model_context_window() { // When config.toml sets model_context_window, that becomes the default fallback. let config = r#"model_context_window = 200000 "#; let catalog = r#"{ "models": [ { "slug": "a", "display_name": "a", "context_window": 200000 }, { "slug": "b", "display_name": "b", "context_window": 500000 } ] }"#; let result = build_simplified_catalog_from_texts(config, catalog).expect("entries"); let models = result.get("models").unwrap().as_array().unwrap(); // Matches default → squashed. assert!(models[0].get("contextWindow").is_none()); // Different from default → preserved. assert_eq!( models[1].get("contextWindow").and_then(|v| v.as_u64()), Some(500_000) ); } #[test] fn build_simplified_catalog_squashes_inferred_modalities_and_keeps_overrides() { let catalog = r#"{ "models": [ { "slug": "gpt-5.4", "input_modalities": ["text", "image"] }, { "slug": "deepseek-v4-pro", "input_modalities": ["text"] }, { "slug": "gpt-text-override", "input_modalities": ["text"] }, { "slug": "deepseek-v4-flash", "input_modalities": ["text", "image"] } ] }"#; let result = build_simplified_catalog_from_texts("", catalog).expect("entries"); let models = result.get("models").unwrap().as_array().unwrap(); assert!( models[0].get("inputModalities").is_none(), "GPT text+image is inferred and must not become a sticky hidden override" ); assert!( models[1].get("inputModalities").is_none(), "confirmed text-only capability is inferred and must remain registry-driven" ); assert_eq!( models[2].get("inputModalities"), Some(&json!(["text"])), "an unknown model explicitly forced to text-only must round-trip" ); assert_eq!( models[3].get("inputModalities"), Some(&json!(["text", "image"])), "an explicit image override for a registered text-only model must round-trip" ); } #[test] fn build_simplified_catalog_returns_none_when_unparseable() { assert!(build_simplified_catalog_from_texts("", "not json").is_none()); assert!(build_simplified_catalog_from_texts("", "{}").is_none()); assert!( build_simplified_catalog_from_texts("", r#"{"models": []}"#).is_none(), "empty models array should yield None so the field is not inserted at all" ); assert!( build_simplified_catalog_from_texts( "", r#"{"models": [{"display_name": "no slug"}]}"#, ) .is_none(), "entries lacking slug are skipped; a fully-skipped catalog yields None" ); } #[test] fn codex_cli_candidates_are_non_empty() { let candidates = codex_cli_candidates(); assert!( candidates .iter() .any(|candidate| candidate == Path::new("codex")), "codex CLI candidates must include the PATH entry" ); } #[test] fn codex_bundled_models_command_uses_expected_program_and_args() { let command = codex_bundled_models_command(Path::new("codex")); assert_eq!(command.get_program(), "codex"); assert_eq!( command .get_args() .map(|arg| arg.to_string_lossy().into_owned()) .collect::>(), ["debug", "models", "--bundled"] ); } #[test] fn successful_model_catalog_template_load_is_cached() { use std::cell::Cell; let cache = OnceCell::new(); let calls = Cell::new(0); let first = get_or_load_codex_model_catalog_template(&cache, || { calls.set(calls.get() + 1); Ok(json!({ "slug": "first" })) }) .expect("first template load"); let second = get_or_load_codex_model_catalog_template(&cache, || { calls.set(calls.get() + 1); Ok(json!({ "slug": "second" })) }) .expect("cached template load"); assert_eq!(first, json!({ "slug": "first" })); assert_eq!(second, first); assert_eq!(calls.get(), 1, "successful template should load only once"); } #[test] fn failed_model_catalog_template_load_can_retry() { use std::cell::Cell; let cache = OnceCell::new(); let calls = Cell::new(0); let first = get_or_load_codex_model_catalog_template(&cache, || { calls.set(calls.get() + 1); Err(AppError::Message("temporary failure".to_string())) }); assert!(first.is_err()); let second = get_or_load_codex_model_catalog_template(&cache, || { calls.set(calls.get() + 1); Ok(json!({ "slug": "recovered" })) }) .expect("retry template load"); assert_eq!(second, json!({ "slug": "recovered" })); assert_eq!(calls.get(), 2, "failed loads must not poison the cache"); } #[test] fn codex_cli_candidates_include_user_node_manager_bins() { let temp_home = tempfile::tempdir().expect("create temp home"); let home = temp_home.path(); let expected = [ home.join(".nvm/versions/node/v22.14.0/bin/codex"), home.join(".volta/bin/codex"), home.join(".asdf/shims/codex"), home.join(".local/share/mise/shims/codex"), home.join(".local/share/fnm/node-versions/v22.14.0/installation/bin/codex"), ]; for candidate in &expected { std::fs::create_dir_all(candidate.parent().expect("candidate parent")) .expect("create candidate parent"); std::fs::write(candidate, "").expect("create candidate"); } let mut candidates = Vec::new(); let mut seen = HashSet::new(); push_home_codex_cli_candidates(&mut candidates, &mut seen, home); for candidate in expected { assert!( candidates.contains(&candidate), "user-level Codex CLI candidate should be discovered: {}", candidate.display() ); } } #[test] fn codex_cli_candidates_deduplicate_entries() { let temp_home = tempfile::tempdir().expect("create temp home"); let home = temp_home.path(); let candidate = home.join(".volta/bin/codex"); std::fs::create_dir_all(candidate.parent().expect("candidate parent")) .expect("create candidate parent"); std::fs::write(&candidate, "").expect("create candidate"); let mut candidates = Vec::new(); let mut seen = HashSet::new(); push_existing_codex_cli_candidate(&mut candidates, &mut seen, candidate.clone()); push_home_codex_cli_candidates(&mut candidates, &mut seen, home); assert_eq!( candidates.iter().filter(|path| **path == candidate).count(), 1, "duplicate candidates should be removed" ); } #[test] fn static_template_is_valid_json_with_slug() { let template = load_codex_model_template_static().expect("static template must parse as valid JSON"); assert_eq!( template.get("slug").and_then(|v| v.as_str()), Some("gpt-5.5"), "static template slug must be gpt-5.5" ); } #[test] fn static_template_has_required_keys() { let template = load_codex_model_template_static().expect("static template must parse as valid JSON"); for key in &[ "model_messages", "base_instructions", "context_window", "display_name", ] { assert!( template.get(key).is_some(), "static template must contain key '{key}'" ); } } #[test] #[cfg(target_os = "windows")] fn set_catalog_json_field_writes_filename_ignoring_unc_path() { let input = r#"model_provider = "custom" model = "glm-5" "#; // Simulate a WSL UNC path as cc-switch would see it on Windows; // the function now writes just the relative filename. let unc_path = Path::new(r"\\wsl.localhost\Ubuntu\home\user\.codex\cc-switch-model-catalog.json"); let result = set_codex_model_catalog_json_field(input, Some(unc_path)).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); let written_path = parsed .get("model_catalog_json") .and_then(|v| v.as_str()) .expect("model_catalog_json should be set"); assert_eq!( written_path, CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME, "should write only the relative filename, not the UNC path" ); } #[test] fn set_catalog_json_field_writes_filename_for_any_path() { let input = r#"model_provider = "custom" model = "glm-5" "#; let regular_path = Path::new("/home/user/.codex/cc-switch-model-catalog.json"); let result = set_codex_model_catalog_json_field(input, Some(regular_path)).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed.get("model_catalog_json").and_then(|v| v.as_str()), Some(CC_SWITCH_CODEX_MODEL_CATALOG_FILENAME), "should write only the relative filename, not the full path" ); } #[test] fn set_catalog_json_none_removes_cc_switch_owned_by_filename() { // After the WSL fix, TOML may contain a Linux-style path. // The None arm must still remove it (file_name match catches any format). let input = r#"model_catalog_json = "/home/user/.codex/cc-switch-model-catalog.json" "#; let result = set_codex_model_catalog_json_field(input, None).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert!( parsed.get("model_catalog_json").is_none(), "None arm should remove cc-switch-owned field regardless of path format" ); } #[test] fn set_catalog_json_none_preserves_user_owned_catalog() { let input = r#"model_catalog_json = "/Users/me/.codex/my-custom-catalog.json" "#; let result = set_codex_model_catalog_json_field(input, None).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert_eq!( parsed.get("model_catalog_json").and_then(|v| v.as_str()), Some("/Users/me/.codex/my-custom-catalog.json"), "None arm should NOT remove user-owned catalog" ); } #[test] fn resolve_catalog_finds_relative_filename() { let config_text = r#"model_provider = "custom" model_catalog_json = "cc-switch-model-catalog.json" "#; let generated_path = PathBuf::from("/home/user/.codex/cc-switch-model-catalog.json"); let result = resolve_cc_switch_catalog_path(config_text, &generated_path); assert_eq!( result, Some(generated_path), "relative filename should resolve to generated_path for file I/O" ); } #[test] fn resolve_catalog_ignores_user_owned_relative() { let config_text = r#"model_catalog_json = "my-custom-catalog.json" "#; let generated_path = PathBuf::from("/home/user/.codex/cc-switch-model-catalog.json"); let result = resolve_cc_switch_catalog_path(config_text, &generated_path); assert_eq!( result, None, "user-owned catalog should not be claimed by cc-switch" ); } #[test] fn set_catalog_json_none_removes_relative_path() { let input = r#"model_catalog_json = "cc-switch-model-catalog.json" "#; let result = set_codex_model_catalog_json_field(input, None).unwrap(); let parsed: toml::Value = toml::from_str(&result).unwrap(); assert!( parsed.get("model_catalog_json").is_none(), "None arm should remove relative cc-switch-owned field" ); } }