codex_helper_core/

codex_integration.rs

use std::collections::BTreeSet;
use std::fs;
use std::io::{Read, Seek, SeekFrom};
use std::path::{Path, PathBuf};

use crate::client_config::{
    CLAUDE_ABSENT_BACKUP_SENTINEL, claude_settings_backup_path_for as claude_settings_backup_path,
    claude_settings_path, codex_app_db_path, codex_auth_path, codex_config_path,
    codex_switch_state_path,
};
use crate::codex_patch_plan::{
    CodexAuthPatchPlan, CodexPatchPlan, CodexSwitchOnEffectOrder, CodexTomlBoolPatch,
};
pub use crate::codex_patch_plan::{CodexPatchMode, CodexSwitchOptions};
use crate::config::{ProxyConfig, ProxyConfigV2, ProxyConfigV4, RoutingAffinityPolicyV5};
use crate::file_replace::write_text_file;
use anyhow::{Context, Result, anyhow};
use toml::Value;
use toml_edit::{
    Document as EditableTomlDocument, Item as EditableTomlItem, Table as EditableTomlTable,
    Value as EditableTomlValue, value as editable_toml_value,
};

fn read_config_text(path: &Path) -> Result<String> {
    if !path.exists() {
        return Ok(String::new());
    }
    let mut file = fs::File::open(path).with_context(|| format!("open {:?}", path))?;
    let mut buf = String::new();
    file.read_to_string(&mut buf)
        .with_context(|| format!("read {:?}", path))?;
    Ok(buf)
}

fn atomic_write(path: &Path, data: &str) -> Result<()> {
    write_text_file(path, data)
}

fn set_toml_value_preserving_decor(item: &mut EditableTomlItem, mut value: EditableTomlValue) {
    if let Some(current) = item.as_value_mut() {
        let decor = current.decor().clone();
        *value.decor_mut() = decor;
        *current = value;
    } else {
        *item = EditableTomlItem::Value(value);
    }
}

fn set_toml_string(table: &mut EditableTomlTable, key: &str, value: impl Into<String>) {
    let item = table.entry(key).or_insert(EditableTomlItem::None);
    set_toml_value_preserving_decor(item, EditableTomlValue::from(value.into()));
}

fn toml_string(table: &EditableTomlTable, key: &str) -> Option<String> {
    table
        .get(key)
        .and_then(EditableTomlItem::as_value)
        .and_then(EditableTomlValue::as_str)
        .map(ToOwned::to_owned)
}

fn local_helper_proxy_item(item: Option<&EditableTomlItem>) -> bool {
    let Some(table) = item.and_then(EditableTomlItem::as_table) else {
        return false;
    };
    let name_is_helper = toml_string(table, "name").as_deref() == Some("codex-helper");
    let base_url_is_local = toml_string(table, "base_url")
        .as_deref()
        .is_some_and(|url| url.contains("127.0.0.1") || url.contains("localhost"));
    name_is_helper || base_url_is_local
}

fn codex_text_points_to_local_helper(text: &str) -> Result<bool> {
    if text.trim().is_empty() {
        return Ok(false);
    }
    let doc = text.parse::<EditableTomlDocument>()?;
    let root = doc.as_table();
    if toml_string(root, "model_provider").as_deref() != Some("codex_proxy") {
        return Ok(false);
    }
    Ok(root
        .get("model_providers")
        .and_then(EditableTomlItem::as_table)
        .and_then(|table| table.get("codex_proxy"))
        .is_some_and(|item| local_helper_proxy_item(Some(item))))
}

#[derive(Debug, Clone, PartialEq, serde::Deserialize, serde::Serialize)]
struct CodexSwitchState {
    version: u32,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    patch_mode: Option<CodexPatchMode>,
    #[serde(default, skip_serializing_if = "bool_is_false")]
    responses_websocket: bool,
    original_config_absent: bool,
    original_model_provider: Option<String>,
    original_codex_proxy: Option<Value>,
    had_model_providers: bool,
    #[serde(default)]
    original_auth_json_absent: bool,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    original_auth_json: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    patched_auth_json: Option<String>,
}

impl CodexSwitchState {
    fn from_codex_config_text(text: &str, original_config_absent: bool) -> Result<Self> {
        let doc = if text.trim().is_empty() {
            EditableTomlDocument::new()
        } else {
            text.parse::<EditableTomlDocument>()?
        };
        let root = doc.as_table();
        let providers_table = root
            .get("model_providers")
            .and_then(EditableTomlItem::as_table);

        Ok(Self {
            version: 2,
            patch_mode: None,
            responses_websocket: false,
            original_config_absent,
            original_model_provider: toml_string(root, "model_provider"),
            original_codex_proxy: original_codex_proxy_value(text)?,
            had_model_providers: providers_table.is_some(),
            original_auth_json_absent: false,
            original_auth_json: None,
            patched_auth_json: None,
        })
    }

    fn set_auth_patch(&mut self, patch: &CodexAuthPatch) {
        self.original_auth_json_absent = patch.original_absent;
        self.original_auth_json = patch.original_text.clone();
        self.patched_auth_json = Some(patch.patched_text.clone());
    }

    fn clear_auth_patch(&mut self) {
        self.original_auth_json_absent = false;
        self.original_auth_json = None;
        self.patched_auth_json = None;
    }
}

fn original_codex_proxy_value(text: &str) -> Result<Option<Value>> {
    if text.trim().is_empty() {
        return Ok(None);
    }
    let value = text.parse::<Value>()?;
    Ok(value
        .as_table()
        .and_then(|root| root.get("model_providers"))
        .and_then(Value::as_table)
        .and_then(|providers| providers.get("codex_proxy"))
        .cloned())
}

fn editable_item_from_toml_value(value: &Value) -> Result<EditableTomlItem> {
    match value {
        Value::Table(table) => {
            let body = toml::to_string(table)?;
            let doc = format!("[codex_proxy]\n{body}").parse::<EditableTomlDocument>()?;
            doc.as_table()
                .get("codex_proxy")
                .cloned()
                .ok_or_else(|| anyhow!("failed to parse stored codex_proxy state"))
        }
        _ => Err(anyhow!("stored codex_proxy state must be a TOML table")),
    }
}

fn read_codex_switch_state() -> Result<Option<CodexSwitchState>> {
    let path = codex_switch_state_path();
    if !path.exists() {
        return Ok(None);
    }
    let text = read_config_text(&path)?;
    let state = serde_json::from_str::<CodexSwitchState>(&text)
        .with_context(|| format!("parse {:?}", path))?;
    Ok(Some(state))
}

fn write_codex_switch_state(state: &CodexSwitchState) -> Result<()> {
    let path = codex_switch_state_path();
    let text = serde_json::to_string_pretty(state)?;
    atomic_write(&path, &text)
}

pub fn codex_switch_state_exists() -> bool {
    codex_switch_state_path().exists()
}

fn bool_is_false(value: &bool) -> bool {
    !*value
}

enum CodexSwitchOffEdit {
    Write(String),
    RemoveFile,
}

struct CodexAuthPatch {
    original_absent: bool,
    original_text: Option<String>,
    patched_text: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
enum CodexAuthEdit {
    None,
    Write(String),
    RemoveFile,
}

fn auth_json_matches_helper_patch(current_text: Option<&str>, patched_auth_json: &str) -> bool {
    let Some(current_text) = current_text else {
        return false;
    };
    if current_text == patched_auth_json {
        return true;
    }

    let Ok(current_value) = serde_json::from_str::<serde_json::Value>(current_text) else {
        return false;
    };
    let Ok(patched_value) = serde_json::from_str::<serde_json::Value>(patched_auth_json) else {
        return false;
    };
    current_value == patched_value
}

fn apply_codex_auth_edit(edit: CodexAuthEdit) -> Result<()> {
    match edit {
        CodexAuthEdit::None => Ok(()),
        CodexAuthEdit::Write(text) => atomic_write(&codex_auth_path(), &text)
            .with_context(|| format!("patch {:?}", codex_auth_path())),
        CodexAuthEdit::RemoveFile => {
            let path = codex_auth_path();
            if path.exists() {
                fs::remove_file(&path).with_context(|| format!("remove {:?}", path))?;
            }
            Ok(())
        }
    }
}

fn read_current_codex_auth_text() -> Result<Option<String>> {
    let path = codex_auth_path();
    if path.exists() {
        read_config_text(&path).map(Some)
    } else {
        Ok(None)
    }
}

fn auth_restore_edit_from_state(state: &mut CodexSwitchState) -> Result<CodexAuthEdit> {
    let current_text = read_current_codex_auth_text()?;
    Ok(auth_restore_edit_from_state_and_current(
        state,
        current_text.as_deref(),
    ))
}

fn auth_restore_edit_from_state_and_current(
    state: &mut CodexSwitchState,
    current_text: Option<&str>,
) -> CodexAuthEdit {
    let Some(patched_auth_json) = state.patched_auth_json.as_deref() else {
        return CodexAuthEdit::None;
    };

    let edit = if auth_json_matches_helper_patch(current_text, patched_auth_json) {
        if state.original_auth_json_absent {
            CodexAuthEdit::RemoveFile
        } else if let Some(original) = state.original_auth_json.clone() {
            CodexAuthEdit::Write(original)
        } else {
            CodexAuthEdit::None
        }
    } else {
        CodexAuthEdit::None
    };

    state.clear_auth_patch();
    edit
}

fn auth_baseline_for_patch_from_current(
    state: &CodexSwitchState,
    current_text: Option<String>,
) -> (bool, Option<String>) {
    if let Some(patched_auth_json) = state.patched_auth_json.as_deref()
        && auth_json_matches_helper_patch(current_text.as_deref(), patched_auth_json)
    {
        return (
            state.original_auth_json_absent,
            state.original_auth_json.clone(),
        );
    }

    (current_text.is_none(), current_text)
}

fn switch_off_codex_toml(
    current_text: &str,
    original: &CodexSwitchState,
) -> Result<CodexSwitchOffEdit> {
    let mut doc = if current_text.trim().is_empty() {
        EditableTomlDocument::new()
    } else {
        current_text.parse::<EditableTomlDocument>()?
    };
    let root = doc.as_table_mut();

    let current_model_provider = toml_string(root, "model_provider");
    let proxy_is_helper = root
        .get("model_providers")
        .and_then(EditableTomlItem::as_table)
        .and_then(|table| table.get("codex_proxy"))
        .map(|item| local_helper_proxy_item(Some(item)))
        .unwrap_or(current_model_provider.as_deref() == Some("codex_proxy"));

    if current_model_provider.as_deref() == Some("codex_proxy") && proxy_is_helper {
        if let Some(provider) = original.original_model_provider.as_deref() {
            set_toml_string(root, "model_provider", provider);
        } else {
            root.remove("model_provider");
        }
    }

    let mut remove_model_providers = false;
    if let Some(providers_table) = root
        .get_mut("model_providers")
        .and_then(EditableTomlItem::as_table_mut)
    {
        let proxy_is_helper = local_helper_proxy_item(providers_table.get("codex_proxy"));
        if proxy_is_helper {
            if let Some(original_proxy) = original.original_codex_proxy.as_ref() {
                providers_table.insert(
                    "codex_proxy",
                    editable_item_from_toml_value(original_proxy)?,
                );
            } else {
                providers_table.remove("codex_proxy");
            }
        }
        remove_model_providers = !original.had_model_providers && providers_table.is_empty();
    }
    if remove_model_providers {
        root.remove("model_providers");
    }

    if original.original_config_absent && root.is_empty() {
        Ok(CodexSwitchOffEdit::RemoveFile)
    } else {
        Ok(CodexSwitchOffEdit::Write(doc.to_string()))
    }
}

fn codex_config_text_with_switch_state(
    current_text: &str,
    state: &CodexSwitchState,
) -> Result<String> {
    let mut doc = if current_text.trim().is_empty() {
        EditableTomlDocument::new()
    } else {
        current_text.parse::<EditableTomlDocument>()?
    };
    let root = doc.as_table_mut();
    let current_model_provider = toml_string(root, "model_provider");
    let proxy_is_helper = root
        .get("model_providers")
        .and_then(EditableTomlItem::as_table)
        .and_then(|table| table.get("codex_proxy"))
        .map(|item| local_helper_proxy_item(Some(item)))
        .unwrap_or(current_model_provider.as_deref() == Some("codex_proxy"));

    if current_model_provider.as_deref() != Some("codex_proxy") || !proxy_is_helper {
        return Ok(current_text.to_string());
    }

    if let Some(provider) = state.original_model_provider.as_deref() {
        set_toml_string(root, "model_provider", provider);
    } else {
        root.remove("model_provider");
    }

    let mut remove_model_providers = false;
    if let Some(providers_table) = root
        .get_mut("model_providers")
        .and_then(EditableTomlItem::as_table_mut)
    {
        if let Some(original_proxy) = state.original_codex_proxy.as_ref() {
            providers_table.insert(
                "codex_proxy",
                editable_item_from_toml_value(original_proxy)?,
            );
        } else {
            providers_table.remove("codex_proxy");
        }
        remove_model_providers = !state.had_model_providers && providers_table.is_empty();
    }
    if remove_model_providers {
        root.remove("model_providers");
    }

    Ok(doc.to_string())
}

pub fn codex_config_text_for_import() -> Result<Option<String>> {
    let cfg_path = codex_config_path();
    if !cfg_path.exists() {
        return Ok(None);
    }
    let current_text = read_config_text(&cfg_path)?;
    let Some(state) = read_codex_switch_state()? else {
        return Ok(Some(current_text));
    };
    codex_config_text_with_switch_state(&current_text, &state).map(Some)
}

#[cfg(test)]
fn switch_on_codex_toml_with_mode(text: &str, port: u16, mode: CodexPatchMode) -> Result<String> {
    switch_on_codex_toml_with_options(text, port, mode, CodexSwitchOptions::default())
}

#[cfg(test)]
fn switch_on_codex_toml_with_options(
    text: &str,
    port: u16,
    mode: CodexPatchMode,
    options: CodexSwitchOptions,
) -> Result<String> {
    let plan = CodexPatchPlan::for_switch_on(mode, options)?;
    switch_on_codex_toml_with_plan(text, port, plan)
}

#[cfg(test)]
fn switch_on_codex_toml_with_plan(text: &str, port: u16, plan: CodexPatchPlan) -> Result<String> {
    switch_on_codex_toml_with_base_url_and_plan(text, &format!("http://127.0.0.1:{port}"), plan)
}

fn switch_on_codex_toml_with_base_url_and_plan(
    text: &str,
    base_url: &str,
    plan: CodexPatchPlan,
) -> Result<String> {
    let mut doc = if text.trim().is_empty() {
        EditableTomlDocument::new()
    } else {
        text.parse::<EditableTomlDocument>()?
    };
    let root = doc.as_table_mut();

    if !root.contains_key("model_providers") {
        root.insert(
            "model_providers",
            EditableTomlItem::Table(EditableTomlTable::new()),
        );
    }
    let providers_table = root
        .get_mut("model_providers")
        .and_then(EditableTomlItem::as_table_mut)
        .ok_or_else(|| anyhow!("model_providers must be a table"))?;

    if !providers_table.contains_key("codex_proxy") {
        providers_table.insert(
            "codex_proxy",
            EditableTomlItem::Table(EditableTomlTable::new()),
        );
    }
    let proxy_table = providers_table
        .get_mut("codex_proxy")
        .and_then(EditableTomlItem::as_table_mut)
        .ok_or_else(|| anyhow!("model_providers.codex_proxy must be a table"))?;

    let provider = plan.provider();
    set_toml_string(proxy_table, "name", provider.provider_name());
    set_toml_string(proxy_table, "base_url", base_url);
    set_toml_string(proxy_table, "wire_api", "responses");
    if !proxy_table.contains_key("request_max_retries") {
        proxy_table.insert("request_max_retries", editable_toml_value(0));
    }
    match provider.requires_openai_auth() {
        CodexTomlBoolPatch::Remove => {
            proxy_table.remove("requires_openai_auth");
        }
        CodexTomlBoolPatch::Set(value) => {
            proxy_table.insert("requires_openai_auth", editable_toml_value(value));
        }
    }
    match provider.supports_websockets() {
        CodexTomlBoolPatch::Remove => {
            proxy_table.remove("supports_websockets");
        }
        CodexTomlBoolPatch::Set(value) => {
            proxy_table.insert("supports_websockets", editable_toml_value(value));
        }
    }

    set_toml_string(root, "model_provider", "codex_proxy");
    Ok(doc.to_string())
}

fn ensure_codex_remote_connections_feature_in_toml(text: &str) -> Result<String> {
    let mut doc = if text.trim().is_empty() {
        EditableTomlDocument::new()
    } else {
        text.parse::<EditableTomlDocument>()?
    };
    let root = doc.as_table_mut();

    if !root.contains_key("features") {
        root.insert(
            "features",
            EditableTomlItem::Table(EditableTomlTable::new()),
        );
    }
    let features = root
        .get_mut("features")
        .and_then(EditableTomlItem::as_table_mut)
        .ok_or_else(|| anyhow!("features must be a table"))?;
    features.insert("remote_connections", editable_toml_value(true));
    features.remove("remote_control");

    Ok(doc.to_string())
}

fn codex_remote_connections_feature_enabled_from_toml(text: &str) -> Result<bool> {
    if text.trim().is_empty() {
        return Ok(false);
    }
    let value = text.parse::<Value>()?;
    Ok(value
        .as_table()
        .and_then(|root| root.get("features"))
        .and_then(Value::as_table)
        .and_then(|features| features.get("remote_connections"))
        .and_then(Value::as_bool)
        .unwrap_or(false))
}

fn codex_remote_control_feature_present_in_toml(text: &str) -> Result<bool> {
    if text.trim().is_empty() {
        return Ok(false);
    }
    let value = text.parse::<Value>()?;
    Ok(value
        .as_table()
        .and_then(|root| root.get("features"))
        .and_then(Value::as_table)
        .is_some_and(|features| features.contains_key("remote_control")))
}

fn codex_remote_compaction_v2_feature_enabled_from_toml(text: &str) -> Result<bool> {
    if text.trim().is_empty() {
        return Ok(false);
    }
    let value = text.parse::<Value>()?;
    Ok(value
        .as_table()
        .and_then(|root| root.get("features"))
        .and_then(Value::as_table)
        .and_then(|features| features.get("remote_compaction_v2"))
        .and_then(Value::as_bool)
        .unwrap_or(false))
}

fn json_string_at_path<'a>(value: &'a serde_json::Value, path: &[&str]) -> Option<&'a str> {
    let mut current = value;
    for key in path {
        current = current.get(*key)?;
    }
    current.as_str().filter(|text| !text.trim().is_empty())
}

fn decode_jwt_payload(jwt: &str) -> Result<serde_json::Value> {
    use base64::Engine as _;

    let mut parts = jwt.split('.');
    let (_header, payload, _signature) = match (parts.next(), parts.next(), parts.next()) {
        (Some(header), Some(payload), Some(signature))
            if !header.is_empty() && !payload.is_empty() && !signature.is_empty() =>
        {
            (header, payload, signature)
        }
        _ => return Err(anyhow!("invalid JWT format")),
    };
    let bytes = base64::engine::general_purpose::URL_SAFE_NO_PAD
        .decode(payload)
        .or_else(|_| base64::engine::general_purpose::URL_SAFE.decode(payload))
        .context("decode JWT payload")?;
    serde_json::from_slice(&bytes).context("parse JWT payload JSON")
}

fn chatgpt_bridge_auth_requirements_missing(
    value: &serde_json::Value,
) -> Result<Vec<&'static str>> {
    let obj = value
        .as_object()
        .ok_or_else(|| anyhow!("Codex auth.json root must be a JSON object"))?;
    let tokens = obj.get("tokens").and_then(serde_json::Value::as_object);
    let mut missing = Vec::new();

    let id_token = match tokens.and_then(|tokens| tokens.get("id_token")) {
        Some(value) => value.as_str().filter(|text| !text.trim().is_empty()),
        None => None,
    };
    let id_token_payload = match id_token {
        Some(id_token) => Some(decode_jwt_payload(id_token).context("decode tokens.id_token")?),
        None => {
            missing.push("tokens.id_token");
            None
        }
    };

    if tokens
        .and_then(|tokens| tokens.get("access_token"))
        .and_then(serde_json::Value::as_str)
        .is_none_or(|text| text.trim().is_empty())
    {
        missing.push("tokens.access_token");
    }
    if tokens
        .and_then(|tokens| tokens.get("refresh_token"))
        .and_then(serde_json::Value::as_str)
        .is_none_or(|text| text.trim().is_empty())
    {
        missing.push("tokens.refresh_token");
    }
    if obj
        .get("last_refresh")
        .is_none_or(serde_json::Value::is_null)
    {
        missing.push("last_refresh");
    }

    if let Some(payload) = id_token_payload.as_ref() {
        let has_email = json_string_at_path(payload, &["email"])
            .or_else(|| json_string_at_path(payload, &["https://api.openai.com/profile", "email"]))
            .is_some();
        if !has_email {
            missing.push("tokens.id_token.email");
        }

        let has_account_id = tokens
            .and_then(|tokens| tokens.get("account_id"))
            .and_then(serde_json::Value::as_str)
            .is_some_and(|text| !text.trim().is_empty())
            || json_string_at_path(
                payload,
                &["https://api.openai.com/auth", "chatgpt_account_id"],
            )
            .is_some();
        if !has_account_id {
            missing.push("tokens.account_id or tokens.id_token.chatgpt_account_id");
        }
    }

    Ok(missing)
}

fn ensure_chatgpt_bridge_auth_ready(value: &serde_json::Value) -> Result<()> {
    let missing = chatgpt_bridge_auth_requirements_missing(value)?;
    if missing.is_empty() {
        return Ok(());
    }

    Err(anyhow!(
        "Codex auth.json does not contain a complete ChatGPT login state required for chatgpt-bridge (missing: {}). Open Codex and sign in with ChatGPT first, then run `codex-helper switch on --mode chatgpt-bridge` again.",
        missing.join(", ")
    ))
}

fn chatgpt_bridge_auth_json_value(mut value: serde_json::Value) -> Result<serde_json::Value> {
    ensure_chatgpt_bridge_auth_ready(&value)?;
    let obj = value
        .as_object_mut()
        .ok_or_else(|| anyhow!("Codex auth.json root must be a JSON object"))?;
    obj.insert(
        "auth_mode".to_string(),
        serde_json::Value::String("chatgpt".to_string()),
    );
    obj.insert("OPENAI_API_KEY".to_string(), serde_json::Value::Null);
    Ok(value)
}

fn chatgpt_bridge_auth_json_text(text: &str) -> Result<String> {
    let mut value: serde_json::Value =
        serde_json::from_str(text).context("parse Codex auth.json as JSON")?;
    value = chatgpt_bridge_auth_json_value(value)?;
    Ok(serde_json::to_string_pretty(&value)?)
}

fn imagegen_bridge_auth_json_text() -> Result<String> {
    Ok(serde_json::to_string_pretty(&serde_json::json!({}))?)
}

fn auth_json_is_empty_chatgpt_facade_text(text: Option<&str>) -> bool {
    let Some(text) = text else {
        return false;
    };
    serde_json::from_str::<serde_json::Value>(text)
        .ok()
        .and_then(|value| value.as_object().cloned())
        .is_some_and(|object| object.is_empty())
}

fn current_auth_json_is_empty_chatgpt_facade() -> bool {
    let path = codex_auth_path();
    if !path.exists() {
        return false;
    }
    let Ok(text) = read_config_text(&path) else {
        return false;
    };
    auth_json_is_empty_chatgpt_facade_text(Some(&text))
}

fn current_auth_json_facade_state() -> Result<Option<bool>> {
    let path = codex_auth_path();
    if !path.exists() {
        return Ok(None);
    }
    let text = read_config_text(&path).with_context(|| format!("read {:?}", path))?;
    Ok(Some(auth_json_is_empty_chatgpt_facade_text(Some(&text))))
}

fn prepare_chatgpt_bridge_auth_patch_from_baseline(
    original_absent: bool,
    original_text: Option<String>,
) -> Result<CodexAuthPatch> {
    let auth_path = codex_auth_path();
    let Some(original_text) = original_text else {
        return Err(anyhow!(
            "Codex auth.json not found at {:?}; run `codex login` first, then enable chatgpt-bridge preset.",
            auth_path
        ));
    };
    if original_absent {
        return Err(anyhow!(
            "Codex auth.json not found at {:?}; run `codex login` first, then enable chatgpt-bridge preset.",
            auth_path
        ));
    }
    let patched_text = chatgpt_bridge_auth_json_text(&original_text)?;
    Ok(CodexAuthPatch {
        original_absent: false,
        original_text: Some(original_text),
        patched_text,
    })
}

fn prepare_imagegen_bridge_auth_patch_from_baseline(
    original_absent: bool,
    original_text: Option<String>,
) -> Result<CodexAuthPatch> {
    Ok(CodexAuthPatch {
        original_absent,
        original_text,
        patched_text: imagegen_bridge_auth_json_text()?,
    })
}

fn auth_edit_for_switch_on_plan(
    plan: CodexPatchPlan,
    state: &mut CodexSwitchState,
) -> Result<CodexAuthEdit> {
    match plan.auth() {
        CodexAuthPatchPlan::RestoreOriginalIfHelperPatched => auth_restore_edit_from_state(state),
        CodexAuthPatchPlan::PatchChatGptBridge | CodexAuthPatchPlan::PatchImagegenFacade => {
            let current_auth = read_current_codex_auth_text()?;
            let (original_absent, original_text) =
                auth_baseline_for_patch_from_current(state, current_auth.clone());
            let patch = match plan.auth() {
                CodexAuthPatchPlan::PatchChatGptBridge => {
                    prepare_chatgpt_bridge_auth_patch_from_baseline(original_absent, original_text)?
                }
                CodexAuthPatchPlan::PatchImagegenFacade => {
                    prepare_imagegen_bridge_auth_patch_from_baseline(
                        original_absent,
                        original_text,
                    )?
                }
                CodexAuthPatchPlan::RestoreOriginalIfHelperPatched => {
                    unreachable!("handled above")
                }
            };
            let auth_edit = if auth_json_matches_helper_patch(
                current_auth.as_deref(),
                patch.patched_text.as_str(),
            ) {
                CodexAuthEdit::None
            } else {
                CodexAuthEdit::Write(patch.patched_text.clone())
            };
            state.set_auth_patch(&patch);
            Ok(auth_edit)
        }
    }
}

fn apply_switch_on_effects(
    plan: CodexPatchPlan,
    cfg_path: &Path,
    new_text: &str,
    state: &CodexSwitchState,
    auth_edit: CodexAuthEdit,
) -> Result<()> {
    match plan.effect_order() {
        CodexSwitchOnEffectOrder::ConfigAuthState => {
            atomic_write(cfg_path, new_text)?;
            apply_codex_auth_edit(auth_edit)?;
            write_codex_switch_state(state)?;
        }
        CodexSwitchOnEffectOrder::StateConfigAuth => {
            write_codex_switch_state(state)?;
            atomic_write(cfg_path, new_text)?;
            apply_codex_auth_edit(auth_edit)?;
        }
    }
    Ok(())
}

fn auth_env_is_set(env_name: &str) -> bool {
    let env_name = env_name.trim();
    !env_name.is_empty() && std::env::var(env_name).is_ok_and(|value| !value.trim().is_empty())
}

fn upstream_auth_has_resolved_credential(auth: &crate::config::UpstreamAuth) -> bool {
    auth.auth_token
        .as_deref()
        .is_some_and(|token| !token.trim().is_empty())
        || auth
            .api_key
            .as_deref()
            .is_some_and(|key| !key.trim().is_empty())
        || auth.auth_token_env.as_deref().is_some_and(auth_env_is_set)
        || auth.api_key_env.as_deref().is_some_and(auth_env_is_set)
}

fn upstream_has_resolved_auth(upstream: &crate::config::UpstreamConfig) -> bool {
    upstream_auth_has_resolved_credential(&upstream.auth)
}

fn upstream_auth_env_names(upstream: &crate::config::UpstreamConfig) -> impl Iterator<Item = &str> {
    upstream
        .auth
        .auth_token_env
        .as_deref()
        .into_iter()
        .chain(upstream.auth.api_key_env.as_deref())
        .map(str::trim)
        .filter(|value| !value.is_empty())
}

fn config_toml_schema_version_or_shape(text: &str) -> Option<u32> {
    let value = toml::from_str::<toml::Value>(text).ok()?;
    if let Some(version) = value
        .get("version")
        .and_then(|value| value.as_integer())
        .map(|value| value as u32)
    {
        return Some(version);
    }

    let has_v4_routing = ["codex", "claude"].iter().any(|service| {
        value
            .get(*service)
            .and_then(|service| service.get("routing"))
            .and_then(|routing| routing.get("entry").or_else(|| routing.get("routes")))
            .is_some()
    });
    if has_v4_routing {
        return Some(4);
    }

    let has_legacy_routing = ["codex", "claude"].iter().any(|service| {
        value
            .get(*service)
            .and_then(|service| service.get("routing"))
            .is_some()
    });
    if has_legacy_routing { Some(3) } else { None }
}

fn load_runtime_config_for_bridge_check() -> Result<ProxyConfig> {
    let path = crate::config::config_file_path();
    if !path.exists() {
        return Ok(ProxyConfig::default());
    }

    let text = read_config_text(&path).with_context(|| format!("read {:?}", path))?;
    if path.extension().and_then(|ext| ext.to_str()) == Some("toml") {
        let version = config_toml_schema_version_or_shape(&text);
        if version.is_some_and(crate::config::is_supported_route_graph_config_version) {
            let cfg = toml::from_str::<ProxyConfigV4>(&text)
                .with_context(|| format!("parse {:?} as route graph config", path))?;
            return crate::config::compile_v4_to_runtime(&cfg);
        }
        if version == Some(3) {
            let cfg = toml::from_str::<crate::config::legacy::ProxyConfigV3Legacy>(&text)
                .with_context(|| format!("parse {:?} as legacy route config", path))?;
            let migrated = crate::config::legacy::migrate_v3_legacy_to_v4(&cfg)?;
            return crate::config::compile_v4_to_runtime(&migrated.config);
        }
        if version == Some(2) {
            let cfg = toml::from_str::<ProxyConfigV2>(&text)
                .with_context(|| format!("parse {:?} as v2 config", path))?;
            return crate::config::compile_v2_to_runtime(&cfg);
        }
        return toml::from_str::<ProxyConfig>(&text)
            .with_context(|| format!("parse {:?} as runtime config", path));
    }

    serde_json::from_str::<ProxyConfig>(&text).with_context(|| format!("parse {:?}", path))
}

#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct CodexBridgeRuntimeAuthSnapshot {
    pub routable_upstreams: usize,
    pub authed_upstreams: usize,
    pub missing_env: Vec<String>,
}

fn codex_bridge_runtime_auth_snapshot_from_config(
    cfg: &ProxyConfig,
) -> CodexBridgeRuntimeAuthSnapshot {
    let mut snapshot = CodexBridgeRuntimeAuthSnapshot::default();
    let mut missing_env = BTreeSet::new();
    for station in cfg
        .codex
        .stations()
        .values()
        .filter(|station| station.enabled)
    {
        for upstream in &station.upstreams {
            snapshot.routable_upstreams += 1;
            if upstream_has_resolved_auth(upstream) {
                snapshot.authed_upstreams += 1;
            } else {
                for env_name in upstream_auth_env_names(upstream) {
                    missing_env.insert(env_name.to_string());
                }
            }
        }
    }
    snapshot.missing_env = missing_env.into_iter().collect();
    snapshot
}

pub fn codex_bridge_runtime_auth_snapshot() -> Result<CodexBridgeRuntimeAuthSnapshot> {
    let cfg = load_runtime_config_for_bridge_check()
        .context("load codex-helper config for bridge diagnostics")?;
    Ok(codex_bridge_runtime_auth_snapshot_from_config(&cfg))
}

fn ensure_bridge_runtime_ready(mode: CodexPatchMode) -> Result<()> {
    let cfg = load_runtime_config_for_bridge_check().with_context(|| {
        format!(
            "load codex-helper config before enabling {}",
            mode.as_preset_str()
        )
    })?;
    let snapshot = codex_bridge_runtime_auth_snapshot_from_config(&cfg);

    if snapshot.routable_upstreams == 0 {
        anyhow::bail!(
            "{} requires at least one enabled Codex upstream in codex-helper config; run `codex-helper config init` or add a [codex.providers.*] entry first",
            mode.as_preset_str()
        );
    }
    if snapshot.authed_upstreams == 0 {
        if snapshot.missing_env.is_empty() {
            anyhow::bail!(
                "{} strips Codex client auth, but no enabled Codex upstream has auth_token/auth_token_env/api_key/api_key_env configured; configure an upstream credential before enabling it",
                mode.as_preset_str()
            );
        }
        anyhow::bail!(
            "{} strips Codex client auth, but no enabled Codex upstream credential is available in this process; set one of these env vars first: {}",
            mode.as_preset_str(),
            snapshot.missing_env.join(", ")
        );
    }

    Ok(())
}

#[cfg(test)]
fn ensure_imagegen_bridge_runtime_ready() -> Result<()> {
    ensure_bridge_runtime_ready(CodexPatchMode::ImagegenBridge)
}

pub fn patch_codex_auth_for_chatgpt_bridge() -> Result<()> {
    let auth_path = codex_auth_path();
    if !auth_path.exists() {
        return Err(anyhow!(
            "Codex auth.json not found at {:?}; run `codex login` first, then enable chatgpt-bridge preset.",
            auth_path
        ));
    }
    let text = read_config_text(&auth_path)?;
    let new_text = chatgpt_bridge_auth_json_text(&text)?;
    atomic_write(&auth_path, &new_text)?;
    Ok(())
}

#[derive(Debug, Clone)]
pub struct CodexSwitchStatus {
    /// Whether Codex currently appears to be configured to use the local codex-helper proxy.
    pub enabled: bool,
    /// Current `model_provider` value (if any).
    pub model_provider: Option<String>,
    /// Current `model_providers.codex_proxy.name` value (if any).
    pub provider_name: Option<String>,
    /// Current `model_providers.codex_proxy.base_url` (if any).
    pub base_url: Option<String>,
    /// Current codex-helper Codex preset inferred from `model_providers.codex_proxy`.
    pub patch_mode: Option<CodexPatchMode>,
    /// Current `model_providers.codex_proxy.requires_openai_auth` value (if any).
    pub requires_openai_auth: Option<bool>,
    /// Current `model_providers.codex_proxy.supports_websockets` value (if any).
    pub supports_websockets: Option<bool>,
    /// Whether `[features].remote_compaction_v2 = true` is present in Codex config.
    pub remote_compaction_v2_enabled: bool,
    /// Whether original switch metadata exists for disabling the local proxy patch.
    pub has_switch_state: bool,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize)]
#[serde(rename_all = "snake_case")]
pub enum CodexBridgeDiagnosticStatus {
    Ok,
    Info,
    Warn,
    Fail,
}

impl CodexBridgeDiagnosticStatus {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Ok => "ok",
            Self::Info => "info",
            Self::Warn => "warn",
            Self::Fail => "fail",
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize)]
pub struct CodexBridgeDiagnosticCheck {
    pub id: &'static str,
    pub status: CodexBridgeDiagnosticStatus,
    pub message: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub action: Option<String>,
}

#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize)]
pub struct CodexBridgeDiagnostics {
    pub patch_mode: Option<CodexPatchMode>,
    pub enabled: bool,
    pub remote_compaction_v1_ready: bool,
    pub imagegen_facade_ready: bool,
    pub upstream_auth_ready: bool,
    pub remote_compaction_v2_enabled: bool,
    pub checks: Vec<CodexBridgeDiagnosticCheck>,
}

impl CodexBridgeDiagnostics {
    pub fn worst_status(&self) -> CodexBridgeDiagnosticStatus {
        if self
            .checks
            .iter()
            .any(|check| check.status == CodexBridgeDiagnosticStatus::Fail)
        {
            return CodexBridgeDiagnosticStatus::Fail;
        }
        if self
            .checks
            .iter()
            .any(|check| check.status == CodexBridgeDiagnosticStatus::Warn)
        {
            return CodexBridgeDiagnosticStatus::Warn;
        }
        if self
            .checks
            .iter()
            .any(|check| check.status == CodexBridgeDiagnosticStatus::Info)
        {
            return CodexBridgeDiagnosticStatus::Info;
        }
        CodexBridgeDiagnosticStatus::Ok
    }
}

fn push_bridge_check(
    checks: &mut Vec<CodexBridgeDiagnosticCheck>,
    id: &'static str,
    status: CodexBridgeDiagnosticStatus,
    message: impl Into<String>,
    action: Option<String>,
) {
    checks.push(CodexBridgeDiagnosticCheck {
        id,
        status,
        message: message.into(),
        action,
    });
}

fn bridge_mode_expects_remote_compaction_v1(mode: Option<CodexPatchMode>) -> bool {
    mode.is_some_and(CodexPatchMode::enables_official_relay_features)
}

fn bridge_mode_expects_imagegen_facade(mode: Option<CodexPatchMode>) -> bool {
    mode.is_some_and(CodexPatchMode::enables_imagegen_facade)
}

pub fn codex_bridge_diagnostics() -> CodexBridgeDiagnostics {
    let status_result = codex_switch_status();
    let mut checks = Vec::new();
    let mut enabled = false;
    let mut patch_mode = None;
    let mut remote_compaction_v2_enabled = false;
    let mut remote_compaction_v1_ready = false;
    let mut imagegen_facade_ready = false;

    match status_result {
        Ok(status) => {
            enabled = status.enabled;
            patch_mode = status.patch_mode;
            remote_compaction_v2_enabled = status.remote_compaction_v2_enabled;

            if !status.enabled {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.switch",
                    CodexBridgeDiagnosticStatus::Info,
                    format!(
                        "Codex is not currently routed through codex-helper (model_provider={}).",
                        status.model_provider.as_deref().unwrap_or("<unset>")
                    ),
                    Some(
                        "Run `codex-helper switch on --preset official-imagegen` after starting helper if you want relay + remote compact + imagegen.".to_string(),
                    ),
                );
            } else {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.switch",
                    CodexBridgeDiagnosticStatus::Ok,
                    format!(
                        "Codex is routed through codex-helper on {} with preset={}.",
                        status.base_url.as_deref().unwrap_or("<missing base_url>"),
                        patch_mode
                            .map(|mode| mode.as_preset_str())
                            .unwrap_or("<unknown>")
                    ),
                    None,
                );
            }

            if bridge_mode_expects_remote_compaction_v1(patch_mode) {
                let provider_ok = status.provider_name.as_deref() == Some("OpenAI");
                remote_compaction_v1_ready = status.enabled && provider_ok;
                let status_label = if remote_compaction_v1_ready {
                    CodexBridgeDiagnosticStatus::Ok
                } else {
                    CodexBridgeDiagnosticStatus::Fail
                };
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.remote_compaction_v1",
                    status_label,
                    format!(
                        "Remote compaction v1 requires provider name OpenAI; current name={}, supports_websockets={}.",
                        status.provider_name.as_deref().unwrap_or("<missing>"),
                        status
                            .supports_websockets
                            .map(|value| value.to_string())
                            .unwrap_or_else(|| "<missing>".to_string())
                    ),
                    (!remote_compaction_v1_ready).then(|| {
                        "Run `codex-helper switch on --preset official-imagegen` or `--preset official-relay`, then fully restart Codex clients.".to_string()
                    }),
                );
            } else {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.remote_compaction_v1",
                    CodexBridgeDiagnosticStatus::Info,
                    format!(
                        "Current preset {} does not advertise the local relay as the official OpenAI provider.",
                        patch_mode
                            .map(|mode| mode.as_preset_str())
                            .unwrap_or("<unknown>")
                    ),
                    Some(
                        "Use official-relay or official-imagegen preset when you need Codex remote compaction v1 through relay.".to_string(),
                    ),
                );
            }

            if status.supports_websockets == Some(true) {
                let provider_ok = status.provider_name.as_deref() == Some("OpenAI");
                let ws_ready = status.enabled && provider_ok;
                let status_label = if ws_ready {
                    CodexBridgeDiagnosticStatus::Ok
                } else {
                    CodexBridgeDiagnosticStatus::Fail
                };
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.responses_websocket",
                    status_label,
                    format!(
                        "Responses WebSocket requires provider name OpenAI and supports_websockets=true; current name={}, supports_websockets={}.",
                        status.provider_name.as_deref().unwrap_or("<missing>"),
                        status
                            .supports_websockets
                            .map(|value| value.to_string())
                            .unwrap_or_else(|| "<missing>".to_string())
                    ),
                    (!ws_ready).then(|| {
                        "Run `codex-helper switch on --preset official-relay --responses-websocket` or `--preset official-imagegen --responses-websocket`, then fully restart Codex clients.".to_string()
                    }),
                );
            } else {
                let ws_status =
                    if patch_mode.is_some_and(CodexPatchMode::enables_official_relay_features) {
                        CodexBridgeDiagnosticStatus::Ok
                    } else {
                        CodexBridgeDiagnosticStatus::Info
                    };
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.responses_websocket",
                    ws_status,
                    "Current preset does not advertise Responses WebSocket transport."
                        .to_string(),
                    Some(
                        "Use `--responses-websocket` only when helper and the selected relay both support Responses WebSocket v2.".to_string(),
                    ),
                );
            }

            if bridge_mode_expects_imagegen_facade(patch_mode) {
                match current_auth_json_facade_state() {
                    Ok(Some(true)) => {
                        imagegen_facade_ready = status.enabled;
                        push_bridge_check(
                            &mut checks,
                            "codex_bridge.imagegen_facade",
                            CodexBridgeDiagnosticStatus::Ok,
                            "Codex auth.json is the empty ChatGPT facade used to expose hosted image generation.".to_string(),
                            None,
                        );
                    }
                    Ok(Some(false)) => {
                        push_bridge_check(
                            &mut checks,
                            "codex_bridge.imagegen_facade",
                            CodexBridgeDiagnosticStatus::Fail,
                            "Codex auth.json is not the empty ChatGPT facade expected by imagegen bridge preset.".to_string(),
                            Some(
                                "Run `codex-helper switch on --preset official-imagegen`, then fully restart Codex clients.".to_string(),
                            ),
                        );
                    }
                    Ok(None) => {
                        push_bridge_check(
                            &mut checks,
                            "codex_bridge.imagegen_facade",
                            CodexBridgeDiagnosticStatus::Fail,
                            format!("Codex auth.json is missing at {:?}.", codex_auth_path()),
                            Some(
                                "Run `codex-helper switch on --preset official-imagegen` so helper can write the temporary facade.".to_string(),
                            ),
                        );
                    }
                    Err(err) => {
                        push_bridge_check(
                            &mut checks,
                            "codex_bridge.imagegen_facade",
                            CodexBridgeDiagnosticStatus::Warn,
                            format!("Could not inspect Codex auth.json facade state: {err}"),
                            Some("Inspect ~/.codex/auth.json and rerun `codex-helper switch status`.".to_string()),
                        );
                    }
                }
            } else {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.imagegen_facade",
                    CodexBridgeDiagnosticStatus::Info,
                    format!(
                        "Current preset {} does not install the imagegen auth facade.",
                        patch_mode
                            .map(|mode| mode.as_preset_str())
                            .unwrap_or("<unknown>")
                    ),
                    Some("Use official-imagegen preset when you need relay + remote compact + hosted image generation.".to_string()),
                );
            }

            if status.remote_compaction_v2_enabled {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.remote_compaction_v2",
                    CodexBridgeDiagnosticStatus::Warn,
                    "Codex remote_compaction_v2 is enabled; helper will detect compaction_trigger requests, but relay compatibility is still less stable than v1 /responses/compact.".to_string(),
                    Some(
                        "Only keep [features].remote_compaction_v2 = true when your relay explicitly supports compaction_trigger requests and compaction response items.".to_string(),
                    ),
                );
            } else {
                push_bridge_check(
                    &mut checks,
                    "codex_bridge.remote_compaction_v2",
                    CodexBridgeDiagnosticStatus::Ok,
                    "Codex remote_compaction_v2 is not enabled; remote compaction stays on the v1 /responses/compact path for official bridge presets.".to_string(),
                    None,
                );
            }
        }
        Err(err) => {
            push_bridge_check(
                &mut checks,
                "codex_bridge.switch",
                CodexBridgeDiagnosticStatus::Fail,
                format!("Could not inspect Codex switch status: {err}"),
                Some(
                    "Run `codex-helper switch status` and fix ~/.codex/config.toml parsing issues."
                        .to_string(),
                ),
            );
        }
    }

    let runtime_auth = match codex_bridge_runtime_auth_snapshot() {
        Ok(snapshot) => {
            let auth_ready = snapshot.authed_upstreams > 0;
            let status = if auth_ready {
                CodexBridgeDiagnosticStatus::Ok
            } else {
                CodexBridgeDiagnosticStatus::Fail
            };
            let action = if auth_ready {
                None
            } else if snapshot.missing_env.is_empty() {
                Some(
                    "Configure auth_token/auth_token_env/api_key/api_key_env for an enabled Codex upstream in codex-helper config.".to_string(),
                )
            } else {
                Some(format!(
                    "Set one of these env vars before starting codex-helper: {}.",
                    snapshot.missing_env.join(", ")
                ))
            };
            push_bridge_check(
                &mut checks,
                "codex_bridge.upstream_auth",
                status,
                format!(
                    "codex-helper has {} enabled Codex upstream(s), {} with usable credentials in this process.",
                    snapshot.routable_upstreams, snapshot.authed_upstreams
                ),
                action,
            );
            auth_ready
        }
        Err(err) => {
            push_bridge_check(
                &mut checks,
                "codex_bridge.upstream_auth",
                CodexBridgeDiagnosticStatus::Warn,
                format!("Could not inspect codex-helper upstream credentials: {err}"),
                Some("Check ~/.codex-helper/config.toml and rerun doctor.".to_string()),
            );
            false
        }
    };

    CodexBridgeDiagnostics {
        patch_mode,
        enabled,
        remote_compaction_v1_ready,
        imagegen_facade_ready,
        upstream_auth_ready: runtime_auth,
        remote_compaction_v2_enabled,
        checks,
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum CodexStartupReadinessSeverity {
    Info,
    Warning,
}

impl CodexStartupReadinessSeverity {
    pub fn label(self) -> &'static str {
        match self {
            Self::Info => "info",
            Self::Warning => "warning",
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum CodexStartupReadinessIssueKind {
    ClientStateChanged,
    SwitchFailed,
    SwitchDisabled,
    SwitchPortMismatch,
    PatchModeMismatch,
    MissingSwitchState,
    RemoteControlRemovedKeyPresent,
    RemoteControlIncomplete,
    RemoteControlLogUnconfirmed,
    OfficialRelayAffinityPolicy,
    DiagnosticError,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CodexStartupReadinessIssue {
    pub kind: CodexStartupReadinessIssueKind,
    pub severity: CodexStartupReadinessSeverity,
    pub title: String,
    pub detail: String,
    pub action: String,
}

#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct CodexStartupReadiness {
    pub issues: Vec<CodexStartupReadinessIssue>,
}

impl CodexStartupReadiness {
    pub fn has_issues(&self) -> bool {
        !self.issues.is_empty()
    }

    fn push(
        &mut self,
        kind: CodexStartupReadinessIssueKind,
        severity: CodexStartupReadinessSeverity,
        title: impl Into<String>,
        detail: impl Into<String>,
        action: impl Into<String>,
    ) {
        self.issues.push(CodexStartupReadinessIssue {
            kind,
            severity,
            title: title.into(),
            detail: detail.into(),
            action: action.into(),
        });
    }
}

#[derive(Debug, Clone)]
pub struct CodexStartupReadinessInput {
    pub expected_port: u16,
    pub expected_patch_mode: CodexPatchMode,
    pub expected_responses_websocket: bool,
    pub client_state_changed_this_startup: bool,
    pub switch_error: Option<String>,
}

pub fn codex_tui_startup_readiness(input: CodexStartupReadinessInput) -> CodexStartupReadiness {
    let mut report = CodexStartupReadiness::default();

    if input.client_state_changed_this_startup {
        report.push(
            CodexStartupReadinessIssueKind::ClientStateChanged,
            CodexStartupReadinessSeverity::Warning,
            "Codex client config changed on startup",
            "codex-helper updated ~/.codex/config.toml or ~/.codex/auth.json for the local bridge.",
            "Fully restart any already running Codex App, Codex TUI, or codex exec session so it rereads the client config.",
        );
    }

    if let Some(err) = input
        .switch_error
        .as_deref()
        .filter(|err| !err.trim().is_empty())
    {
        report.push(
            CodexStartupReadinessIssueKind::SwitchFailed,
            CodexStartupReadinessSeverity::Warning,
            "Codex local proxy patch failed",
            err,
            "Run `codex-helper switch status` and fix the reported Codex client config issue before relying on the bridge.",
        );
    }

    match codex_switch_status() {
        Ok(status) => collect_switch_startup_issues(&mut report, &input, &status),
        Err(err) => report.push(
            CodexStartupReadinessIssueKind::DiagnosticError,
            CodexStartupReadinessSeverity::Warning,
            "Could not inspect Codex switch status",
            err.to_string(),
            "Run `codex-helper switch status` from a normal shell to inspect the client config.",
        ),
    }

    match codex_remote_control_status() {
        Ok(status) => collect_remote_control_startup_issues(&mut report, &status),
        Err(err) => report.push(
            CodexStartupReadinessIssueKind::DiagnosticError,
            CodexStartupReadinessSeverity::Warning,
            "Could not inspect Codex remote-control status",
            err.to_string(),
            "Run `codex-helper switch remote-control status` from a normal shell to inspect the desktop state.",
        ),
    }

    report
}

fn collect_switch_startup_issues(
    report: &mut CodexStartupReadiness,
    input: &CodexStartupReadinessInput,
    status: &CodexSwitchStatus,
) {
    if !status.enabled {
        report.push(
            CodexStartupReadinessIssueKind::SwitchDisabled,
            CodexStartupReadinessSeverity::Warning,
            "Codex is not using the local helper",
            format!(
                "Current model_provider is {}.",
                status.model_provider.as_deref().unwrap_or("<unset>")
            ),
            format!(
                "Run `codex-helper switch on --port {}` or restart codex-helper so the client patch can be applied.",
                input.expected_port
            ),
        );
        return;
    }

    if !status.has_switch_state {
        report.push(
            CodexStartupReadinessIssueKind::MissingSwitchState,
            CodexStartupReadinessSeverity::Warning,
            "Codex local proxy patch has no switch state",
            "Codex points at the local helper, but codex-helper cannot find its restore metadata.",
            "Inspect ~/.codex/config.toml before running switch-off operations.",
        );
    }

    if !base_url_points_to_expected_local_port(status.base_url.as_deref(), input.expected_port) {
        report.push(
            CodexStartupReadinessIssueKind::SwitchPortMismatch,
            CodexStartupReadinessSeverity::Warning,
            "Codex local proxy port does not match this TUI",
            format!(
                "codex_proxy.base_url is {}; this TUI is serving port {}.",
                status.base_url.as_deref().unwrap_or("<missing>"),
                input.expected_port
            ),
            format!(
                "Run `codex-helper switch on --port {}` or restart this helper instance on the configured port.",
                input.expected_port
            ),
        );
    }

    if status.patch_mode != Some(input.expected_patch_mode) {
        let actual = status
            .patch_mode
            .map(|mode| mode.as_preset_str())
            .unwrap_or("<unknown>");
        report.push(
            CodexStartupReadinessIssueKind::PatchModeMismatch,
            CodexStartupReadinessSeverity::Warning,
            "Codex bridge preset does not match helper config",
            format!(
                "Expected preset {}, but Codex currently reports {}.",
                input.expected_patch_mode.as_preset_str(),
                actual
            ),
            "Run `codex-helper switch status`; if this changed recently, fully restart Codex clients after switching.",
        );
    }

    if status.supports_websockets.unwrap_or(false) != input.expected_responses_websocket {
        report.push(
            CodexStartupReadinessIssueKind::PatchModeMismatch,
            CodexStartupReadinessSeverity::Warning,
            "Codex WebSocket transport flag does not match helper config",
            format!(
                "Expected supports_websockets={}, but Codex currently reports {}.",
                input.expected_responses_websocket,
                status
                    .supports_websockets
                    .map(|value| value.to_string())
                    .unwrap_or_else(|| "<missing>".to_string())
            ),
            "Run `codex-helper switch status`; if this changed recently, fully restart Codex clients after switching.",
        );
    }

    if status
        .patch_mode
        .is_some_and(CodexPatchMode::enables_official_relay_features)
    {
        match official_relay_affinity_policy_warning() {
            Ok(Some(detail)) => report.push(
                CodexStartupReadinessIssueKind::OfficialRelayAffinityPolicy,
                CodexStartupReadinessSeverity::Warning,
                "Official relay preset can route a session across providers",
                detail,
                "For the most official-like remote compaction behavior, set [codex.routing].affinity_policy = \"fallback-sticky\" or \"hard\" when using multiple authenticated upstreams.",
            ),
            Ok(None) => {}
            Err(err) => report.push(
                CodexStartupReadinessIssueKind::DiagnosticError,
                CodexStartupReadinessSeverity::Warning,
                "Could not inspect codex-helper routing affinity",
                err.to_string(),
                "Inspect ~/.codex-helper/config.toml and choose an affinity policy appropriate for official relay features.",
            ),
        }
    }
}

fn base_url_points_to_expected_local_port(base_url: Option<&str>, port: u16) -> bool {
    let Some(base_url) = base_url else {
        return false;
    };
    let port_marker = format!(":{port}");
    (base_url.contains("127.0.0.1") || base_url.contains("localhost") || base_url.contains("[::1]"))
        && base_url.contains(&port_marker)
}

fn official_relay_affinity_policy_warning() -> Result<Option<String>> {
    let path = crate::config::config_file_path();
    if !path.exists() {
        return Ok(None);
    }
    let text = read_config_text(&path).with_context(|| format!("read {:?}", path))?;
    if text.trim().is_empty() {
        return Ok(None);
    }
    if path.extension().and_then(|ext| ext.to_str()) != Some("toml") {
        return Ok(None);
    }

    let version = config_toml_schema_version_or_shape(&text);
    if !version.is_some_and(crate::config::is_supported_route_graph_config_version) {
        return Ok(None);
    }

    let cfg = toml::from_str::<ProxyConfigV4>(&text)
        .with_context(|| format!("parse {:?} as route graph config", path))?;
    let Some(routing) = cfg.codex.routing.as_ref() else {
        return Ok(None);
    };
    if routing.affinity_policy != RoutingAffinityPolicyV5::PreferredGroup {
        return Ok(None);
    }

    let authed_provider_count = cfg
        .codex
        .providers
        .values()
        .filter(|provider| provider.enabled && provider_v4_has_resolved_auth(provider))
        .count();
    if authed_provider_count <= 1 {
        return Ok(None);
    }

    Ok(Some(format!(
        "codex-helper has {authed_provider_count} authenticated Codex providers and [codex.routing].affinity_policy is \"preferred-group\". Remote compaction v1 may include encrypted conversation state, so /responses and /responses/compact should stay on the same upstream account when a session has failed over."
    )))
}

fn provider_v4_has_resolved_auth(provider: &crate::config::ProviderConfigV4) -> bool {
    upstream_auth_has_resolved_credential(&provider.auth)
        || upstream_auth_has_resolved_credential(&provider.inline_auth)
}

fn collect_remote_control_startup_issues(
    report: &mut CodexStartupReadiness,
    status: &CodexRemoteControlStatus,
) {
    if status.remote_control_config_present {
        report.push(
            CodexStartupReadinessIssueKind::RemoteControlRemovedKeyPresent,
            CodexStartupReadinessSeverity::Warning,
            "Removed remote_control config key is present",
            format!(
                "{:?} contains [features].remote_control, which current Codex builds do not use for this enablement path.",
                status.config_path
            ),
            "Remove remote_control and keep [features].remote_connections = true instead.",
        );
    }

    let remote_requested = status.remote_connections_enabled
        || status.remote_control_config_present
        || status.db_enabled == Some(true);
    if !remote_requested {
        return;
    }

    if !remote_control_status_is_fully_enabled(status) {
        report.push(
            CodexStartupReadinessIssueKind::RemoteControlIncomplete,
            CodexStartupReadinessSeverity::Warning,
            "Codex App remote-control state is incomplete",
            format!(
                "remote_connections={}, db_exists={}, table_exists={}, db_enabled={}.",
                status.remote_connections_enabled,
                status.db_exists,
                status.db_table_exists,
                status
                    .db_enabled
                    .map(|value| value.to_string())
                    .unwrap_or_else(|| "<missing>".to_string())
            ),
            "Run `codex-helper switch remote-control enable`, then fully restart Codex App.",
        );
        return;
    }

    match codex_remote_control_successful_enablement_log_seen() {
        Ok(true) => {}
        Ok(false) => report.push(
            CodexStartupReadinessIssueKind::RemoteControlLogUnconfirmed,
            CodexStartupReadinessSeverity::Warning,
            "Remote-control enablement is not confirmed in Codex logs",
            "The config and SQLite state look enabled, but no experimentalFeature/enablement/set success log was found.",
            "Fully restart Codex App, then run `codex-helper switch remote-control check-logs`.",
        ),
        Err(err) => report.push(
            CodexStartupReadinessIssueKind::DiagnosticError,
            CodexStartupReadinessSeverity::Warning,
            "Could not inspect Codex remote-control logs",
            err.to_string(),
            "Run `codex-helper switch remote-control check-logs` from a normal shell after restarting Codex App.",
        ),
    }
}

fn remote_control_status_is_fully_enabled(status: &CodexRemoteControlStatus) -> bool {
    status.remote_connections_enabled
        && !status.remote_control_config_present
        && status.db_exists
        && status.db_table_exists
        && status.db_enabled == Some(true)
}

pub fn codex_switch_status() -> Result<CodexSwitchStatus> {
    let cfg_path = codex_config_path();
    let state_path = codex_switch_state_path();

    if !cfg_path.exists() {
        return Ok(CodexSwitchStatus {
            enabled: false,
            model_provider: None,
            provider_name: None,
            base_url: None,
            patch_mode: None,
            requires_openai_auth: None,
            supports_websockets: None,
            remote_compaction_v2_enabled: false,
            has_switch_state: state_path.exists(),
        });
    }

    let text = read_config_text(&cfg_path)?;
    let remote_compaction_v2_enabled =
        codex_remote_compaction_v2_feature_enabled_from_toml(&text).unwrap_or(false);
    if text.trim().is_empty() {
        return Ok(CodexSwitchStatus {
            enabled: false,
            model_provider: None,
            provider_name: None,
            base_url: None,
            patch_mode: None,
            requires_openai_auth: None,
            supports_websockets: None,
            remote_compaction_v2_enabled,
            has_switch_state: state_path.exists(),
        });
    }

    let value: Value = match text.parse() {
        Ok(v) => v,
        Err(_) => {
            return Ok(CodexSwitchStatus {
                enabled: false,
                model_provider: None,
                provider_name: None,
                base_url: None,
                patch_mode: None,
                requires_openai_auth: None,
                supports_websockets: None,
                remote_compaction_v2_enabled,
                has_switch_state: state_path.exists(),
            });
        }
    };
    let table = match value.as_table() {
        Some(t) => t,
        None => {
            return Ok(CodexSwitchStatus {
                enabled: false,
                model_provider: None,
                provider_name: None,
                base_url: None,
                patch_mode: None,
                requires_openai_auth: None,
                supports_websockets: None,
                remote_compaction_v2_enabled,
                has_switch_state: state_path.exists(),
            });
        }
    };

    let model_provider = table
        .get("model_provider")
        .and_then(|v| v.as_str())
        .map(|s| s.to_string());

    if model_provider.as_deref() != Some("codex_proxy") {
        return Ok(CodexSwitchStatus {
            enabled: false,
            model_provider,
            provider_name: None,
            base_url: None,
            patch_mode: None,
            requires_openai_auth: None,
            supports_websockets: None,
            remote_compaction_v2_enabled,
            has_switch_state: state_path.exists(),
        });
    }

    let empty_map = toml::map::Map::new();
    let providers_table = table
        .get("model_providers")
        .and_then(|v| v.as_table())
        .unwrap_or(&empty_map);
    let empty_provider = toml::map::Map::new();
    let proxy_table = providers_table
        .get("codex_proxy")
        .and_then(|v| v.as_table())
        .unwrap_or(&empty_provider);

    let base_url = proxy_table
        .get("base_url")
        .and_then(|v| v.as_str())
        .map(|s| s.to_string());
    let name = proxy_table
        .get("name")
        .and_then(|v| v.as_str())
        .unwrap_or_default();
    let provider_name = (!name.is_empty()).then(|| name.to_string());
    let requires_openai_auth = proxy_table
        .get("requires_openai_auth")
        .and_then(|v| v.as_bool());
    let supports_websockets = proxy_table
        .get("supports_websockets")
        .and_then(|v| v.as_bool());

    let is_local = base_url
        .as_deref()
        .is_some_and(|u| u.contains("127.0.0.1") || u.contains("localhost"));
    let is_helper_name = name == "codex-helper";
    let enabled = is_local || is_helper_name;

    let stored_patch_mode = read_codex_switch_state()?.and_then(|state| state.patch_mode);
    let inferred_patch_mode = if requires_openai_auth == Some(true) {
        CodexPatchMode::ChatGptBridge
    } else if name == "OpenAI" {
        if current_auth_json_is_empty_chatgpt_facade() {
            CodexPatchMode::OfficialImagegenBridge
        } else {
            CodexPatchMode::OfficialRelayBridge
        }
    } else {
        CodexPatchMode::Default
    };

    Ok(CodexSwitchStatus {
        enabled,
        model_provider,
        provider_name,
        base_url,
        patch_mode: enabled.then_some(stored_patch_mode.unwrap_or(inferred_patch_mode)),
        requires_openai_auth,
        supports_websockets,
        remote_compaction_v2_enabled,
        has_switch_state: state_path.exists(),
    })
}

/// Switch Codex to use the local codex-helper model provider.
pub fn switch_on(port: u16) -> Result<()> {
    switch_on_with_mode(port, CodexPatchMode::Default)
}

/// Switch Codex to use the local codex-helper model provider using the preset configured in
/// `~/.codex-helper/config.toml`.
pub fn switch_on_with_configured_preset(port: u16) -> Result<()> {
    let client_patch = crate::config::codex_client_patch_config_from_config_file()
        .context("read codex.client_patch from codex-helper config before switch on")?;
    switch_on_with_options(port, client_patch.preset, client_patch.options)
}

/// Switch Codex to use the local codex-helper model provider with an explicit client preset.
pub fn switch_on_with_mode(port: u16, mode: CodexPatchMode) -> Result<()> {
    switch_on_with_options(port, mode, CodexSwitchOptions::default())
}

/// Switch Codex to use the local codex-helper model provider with an explicit client preset and
/// transport options.
pub fn switch_on_with_options(
    port: u16,
    mode: CodexPatchMode,
    options: CodexSwitchOptions,
) -> Result<()> {
    switch_on_with_base_url(&format!("http://127.0.0.1:{port}"), mode, options)
}

/// Switch Codex to use a codex-helper-compatible model provider at an explicit base URL.
pub fn switch_on_with_base_url(
    base_url: &str,
    mode: CodexPatchMode,
    options: CodexSwitchOptions,
) -> Result<()> {
    let base_url = crate::control_plane_client::normalize_base_url(base_url)
        .ok_or_else(|| anyhow!("Codex proxy base URL must start with http:// or https://"))?;
    let plan = CodexPatchPlan::for_switch_on(mode, options)?;
    if plan.requires_bridge_runtime_ready() {
        ensure_bridge_runtime_ready(plan.mode())?;
    }

    let cfg_path = codex_config_path();
    let state_path = codex_switch_state_path();
    let text = read_config_text(&cfg_path)?;
    if !state_path.exists() && codex_text_points_to_local_helper(&text)? {
        return Err(anyhow!(
            "Codex already points to the local codex-helper proxy, but no switch state was found at {:?}; refusing to treat the local proxy as the original provider. Please inspect ~/.codex/config.toml manually or run `codex-helper switch off` only if a switch state exists.",
            state_path
        ));
    }
    let mut state = if state_path.exists() {
        read_codex_switch_state()?.ok_or_else(|| {
            anyhow!(
                "missing Codex switch state at {:?}",
                codex_switch_state_path()
            )
        })?
    } else {
        CodexSwitchState::from_codex_config_text(&text, !cfg_path.exists())?
    };
    state.patch_mode = Some(plan.mode());
    state.responses_websocket = plan.options().responses_websocket;

    let auth_edit = auth_edit_for_switch_on_plan(plan, &mut state)?;
    let new_text = switch_on_codex_toml_with_base_url_and_plan(&text, &base_url, plan)?;
    apply_switch_on_effects(plan, &cfg_path, &new_text, &state, auth_edit)?;
    Ok(())
}

/// Undo the local Codex proxy patch while preserving config edits made during the run.
pub fn switch_off() -> Result<()> {
    let cfg_path = codex_config_path();
    let state_path = codex_switch_state_path();
    if state_path.exists() {
        let mut state = read_codex_switch_state()?.ok_or_else(|| {
            anyhow!(
                "missing Codex switch state at {:?}",
                codex_switch_state_path()
            )
        })?;
        let auth_edit = auth_restore_edit_from_state(&mut state)?;
        if !cfg_path.exists() {
            apply_codex_auth_edit(auth_edit)?;
            fs::remove_file(&state_path)
                .with_context(|| format!("remove stale switch state {:?}", state_path))?;
            return Ok(());
        }
        let current_text = read_config_text(&cfg_path)?;
        match switch_off_codex_toml(&current_text, &state)? {
            CodexSwitchOffEdit::RemoveFile => {
                if cfg_path.exists() {
                    fs::remove_file(&cfg_path)
                        .with_context(|| format!("remove {:?} (restore absent)", cfg_path))?;
                }
            }
            CodexSwitchOffEdit::Write(text) => {
                atomic_write(&cfg_path, &text)
                    .with_context(|| format!("patch {:?} to disable local proxy", cfg_path))?;
            }
        }
        apply_codex_auth_edit(auth_edit)?;
        fs::remove_file(&state_path)
            .with_context(|| format!("remove stale switch state {:?}", state_path))?;
    }
    Ok(())
}

#[derive(Debug, Clone)]
pub struct CodexRemoteControlStatus {
    pub config_path: PathBuf,
    pub remote_connections_enabled: bool,
    pub remote_control_config_present: bool,
    pub db_path: PathBuf,
    pub db_exists: bool,
    pub db_table_exists: bool,
    pub db_enabled: Option<bool>,
    pub db_updated_at: Option<i64>,
}

#[derive(Debug, Clone)]
pub struct CodexRemoteControlEnablement {
    pub status: CodexRemoteControlStatus,
    pub backup_path: PathBuf,
}

pub fn codex_remote_control_enable() -> Result<CodexRemoteControlEnablement> {
    let cfg_path = codex_config_path();
    let config_text = read_config_text(&cfg_path)?;
    let updated_config = ensure_codex_remote_connections_feature_in_toml(&config_text)?;
    let db_path = codex_app_db_path();

    validate_codex_remote_control_feature_enablement_schema(&db_path)?;
    let backup_path = backup_codex_app_db(&db_path)?;

    if updated_config != config_text {
        atomic_write(&cfg_path, &updated_config)
            .with_context(|| format!("patch {:?} for Codex remote connections", cfg_path))?;
    }

    upsert_codex_remote_control_feature_enablement(&db_path)?;
    let status = codex_remote_control_status()?;

    Ok(CodexRemoteControlEnablement {
        status,
        backup_path,
    })
}

pub fn codex_remote_control_status() -> Result<CodexRemoteControlStatus> {
    let config_path = codex_config_path();
    let config_text = read_config_text(&config_path)?;
    let remote_connections_enabled =
        codex_remote_connections_feature_enabled_from_toml(&config_text)?;
    let remote_control_config_present = codex_remote_control_feature_present_in_toml(&config_text)?;

    let db_path = codex_app_db_path();
    let db_exists = db_path.exists();
    let db_status = if db_exists {
        read_codex_remote_control_db_status(&db_path)?
    } else {
        CodexRemoteControlDbStatus {
            table_exists: false,
            enabled: None,
            updated_at: None,
        }
    };

    Ok(CodexRemoteControlStatus {
        config_path,
        remote_connections_enabled,
        remote_control_config_present,
        db_path,
        db_exists,
        db_table_exists: db_status.table_exists,
        db_enabled: db_status.enabled,
        db_updated_at: db_status.updated_at,
    })
}

fn backup_codex_app_db(db_path: &Path) -> Result<PathBuf> {
    if !db_path.exists() {
        return Err(anyhow!(
            "Codex App SQLite database not found at {:?}; open Codex App once so it creates the local database, then retry",
            db_path
        ));
    }

    let timestamp = timestamp_for_backup_filename();
    let file_name = db_path
        .file_name()
        .and_then(|name| name.to_str())
        .unwrap_or("codex-dev.db");
    let backup_path = db_path.with_file_name(format!("{file_name}.{timestamp}.bak"));
    fs::copy(db_path, &backup_path)
        .with_context(|| format!("backup {:?} -> {:?}", db_path, backup_path))?;
    Ok(backup_path)
}

fn validate_codex_remote_control_feature_enablement_schema(db_path: &Path) -> Result<()> {
    if !db_path.exists() {
        return Err(anyhow!(
            "Codex App SQLite database not found at {:?}; open Codex App once so it creates the local database, then retry",
            db_path
        ));
    }

    let conn = rusqlite::Connection::open(db_path)
        .with_context(|| format!("open Codex App SQLite database {:?}", db_path))?;
    let columns = local_app_server_feature_enablement_columns(&conn)?;
    if columns.is_empty() {
        return Err(anyhow!(
            "SQLite table local_app_server_feature_enablement not found in {:?}; this Codex App build may use a different desktop state schema",
            db_path
        ));
    }
    ensure_required_enablement_columns(&columns)
}

fn timestamp_for_backup_filename() -> String {
    use std::time::{SystemTime, UNIX_EPOCH};

    let millis = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|duration| duration.as_millis())
        .unwrap_or(0);
    millis.to_string()
}

fn current_unix_millis_i64() -> i64 {
    use std::time::{SystemTime, UNIX_EPOCH};

    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|duration| duration.as_millis().min(i64::MAX as u128) as i64)
        .unwrap_or(0)
}

fn upsert_codex_remote_control_feature_enablement(db_path: &Path) -> Result<()> {
    let conn = rusqlite::Connection::open(db_path)
        .with_context(|| format!("open Codex App SQLite database {:?}", db_path))?;
    let columns = local_app_server_feature_enablement_columns(&conn)?;
    if columns.is_empty() {
        return Err(anyhow!(
            "SQLite table local_app_server_feature_enablement not found in {:?}; this Codex App build may use a different desktop state schema",
            db_path
        ));
    }
    ensure_required_enablement_columns(&columns)?;

    let updated_at = current_unix_millis_i64();
    let column_names = columns
        .iter()
        .map(|column| column.name.as_str())
        .collect::<Vec<_>>();
    let has_created_at = column_names.contains(&"created_at");

    let updated_rows = conn
        .execute(
            "UPDATE local_app_server_feature_enablement \
             SET enabled = ?1, updated_at = ?2 \
             WHERE feature_name = ?3",
            rusqlite::params![1_i64, updated_at, "remote_control"],
        )
        .with_context(|| {
            format!(
                "update remote_control in local_app_server_feature_enablement in {:?}",
                db_path
            )
        })?;
    if updated_rows > 0 {
        return Ok(());
    }

    let mut insert_columns = vec!["feature_name", "enabled", "updated_at"];
    let mut insert_values = vec!["?1", "?2", "?3"];
    if has_created_at {
        insert_columns.push("created_at");
        insert_values.push("?3");
    }

    let sql = format!(
        "INSERT INTO local_app_server_feature_enablement ({}) VALUES ({})",
        insert_columns.join(", "),
        insert_values.join(", ")
    );
    conn.execute(
        sql.as_str(),
        rusqlite::params!["remote_control", 1_i64, updated_at],
    )
    .with_context(|| {
        format!(
            "upsert remote_control into local_app_server_feature_enablement in {:?}",
            db_path
        )
    })?;

    Ok(())
}

#[derive(Debug)]
struct SqliteColumnInfo {
    name: String,
    not_null: bool,
    pk: i32,
}

fn local_app_server_feature_enablement_columns(
    conn: &rusqlite::Connection,
) -> Result<Vec<SqliteColumnInfo>> {
    let mut stmt = conn
        .prepare("PRAGMA table_info(local_app_server_feature_enablement)")
        .context("prepare table_info(local_app_server_feature_enablement)")?;
    let rows = stmt
        .query_map([], |row| {
            Ok(SqliteColumnInfo {
                name: row.get(1)?,
                not_null: row.get::<_, i64>(3)? != 0,
                pk: row.get(5)?,
            })
        })
        .context("query table_info(local_app_server_feature_enablement)")?;

    let mut columns = Vec::new();
    for row in rows {
        columns.push(row?);
    }
    Ok(columns)
}

fn ensure_required_enablement_columns(columns: &[SqliteColumnInfo]) -> Result<()> {
    for required in ["feature_name", "enabled", "updated_at"] {
        if !columns.iter().any(|column| column.name == required) {
            return Err(anyhow!(
                "SQLite table local_app_server_feature_enablement is missing required column `{required}`"
            ));
        }
    }

    let optional_supported = ["id", "created_at", "feature_name", "enabled", "updated_at"];
    let unsupported_required = columns
        .iter()
        .filter(|column| column.not_null && column.pk == 0)
        .filter(|column| {
            !optional_supported
                .iter()
                .any(|supported| *supported == column.name)
        })
        .map(|column| column.name.as_str())
        .collect::<Vec<_>>();
    if !unsupported_required.is_empty() {
        return Err(anyhow!(
            "SQLite table local_app_server_feature_enablement has unsupported NOT NULL columns without known values: {}",
            unsupported_required.join(", ")
        ));
    }

    Ok(())
}

#[derive(Debug)]
struct CodexRemoteControlDbStatus {
    table_exists: bool,
    enabled: Option<bool>,
    updated_at: Option<i64>,
}

fn read_codex_remote_control_db_status(db_path: &Path) -> Result<CodexRemoteControlDbStatus> {
    let conn = rusqlite::Connection::open(db_path)
        .with_context(|| format!("open Codex App SQLite database {:?}", db_path))?;
    let columns = local_app_server_feature_enablement_columns(&conn)?;
    if columns.is_empty() {
        return Ok(CodexRemoteControlDbStatus {
            table_exists: false,
            enabled: None,
            updated_at: None,
        });
    }

    ensure_required_enablement_columns(&columns)?;
    let mut stmt = conn
        .prepare(
            "SELECT enabled, updated_at \
             FROM local_app_server_feature_enablement \
             WHERE feature_name = ?1 \
             ORDER BY updated_at DESC \
             LIMIT 1",
        )
        .context("prepare local_app_server_feature_enablement status query")?;
    let mut rows = stmt
        .query(rusqlite::params!["remote_control"])
        .context("query local_app_server_feature_enablement status")?;

    if let Some(row) = rows.next()? {
        Ok(CodexRemoteControlDbStatus {
            table_exists: true,
            enabled: Some(row.get::<_, i64>(0)? != 0),
            updated_at: row.get(1)?,
        })
    } else {
        Ok(CodexRemoteControlDbStatus {
            table_exists: true,
            enabled: None,
            updated_at: None,
        })
    }
}

pub fn codex_remote_control_successful_enablement_log_seen() -> Result<bool> {
    let base_dir = codex_config_path()
        .parent()
        .map(Path::to_path_buf)
        .unwrap_or_else(|| PathBuf::from("."));
    for candidate in [base_dir.join("log"), base_dir.join("logs")] {
        if codex_remote_control_successful_enablement_log_seen_in_dir(candidate.as_path())? {
            return Ok(true);
        }
    }
    Ok(false)
}

fn codex_remote_control_successful_enablement_log_seen_in_dir(log_dir: &Path) -> Result<bool> {
    if !log_dir.exists() {
        return Ok(false);
    }

    let mut files = Vec::new();
    collect_regular_files(log_dir, &mut files)?;
    files.sort_by_key(|path| {
        fs::metadata(path)
            .and_then(|metadata| metadata.modified())
            .ok()
    });

    for path in files.into_iter().rev().take(20) {
        if file_contains_remote_control_success(path.as_path())? {
            return Ok(true);
        }
    }
    Ok(false)
}

fn collect_regular_files(dir: &Path, out: &mut Vec<PathBuf>) -> Result<()> {
    for entry in fs::read_dir(dir).with_context(|| format!("read_dir {:?}", dir))? {
        let entry = entry.with_context(|| format!("read entry in {:?}", dir))?;
        let path = entry.path();
        let file_type = entry
            .file_type()
            .with_context(|| format!("read file type for {:?}", path))?;
        if file_type.is_dir() {
            collect_regular_files(&path, out)?;
        } else if file_type.is_file() {
            out.push(path);
        }
    }
    Ok(())
}

fn file_contains_remote_control_success(path: &Path) -> Result<bool> {
    const MAX_BYTES: u64 = 2 * 1024 * 1024;
    let mut file = fs::File::open(path).with_context(|| format!("open log {:?}", path))?;
    let len = file.metadata()?.len();
    if len > MAX_BYTES {
        file.seek(SeekFrom::End(-(MAX_BYTES as i64)))
            .with_context(|| format!("seek log {:?}", path))?;
    }
    let mut bytes = Vec::new();
    file.read_to_end(&mut bytes)
        .with_context(|| format!("read log {:?}", path))?;
    let text = String::from_utf8_lossy(&bytes);
    Ok(text.contains("experimentalFeature/enablement/set")
        && (text.contains("errorCode=null")
            || text.contains("\"errorCode\":null")
            || text.contains("'errorCode':null")))
}

#[derive(Debug, Clone)]
pub struct ClaudeSwitchStatus {
    /// Whether Claude Code currently appears to be configured to use the local codex-helper proxy.
    pub enabled: bool,
    /// Current `env.ANTHROPIC_BASE_URL` value (if any).
    pub base_url: Option<String>,
    /// Whether a backup file exists for safe restore.
    pub has_backup: bool,
    /// The resolved settings file path (settings.json or legacy claude.json).
    pub settings_path: PathBuf,
}

pub fn claude_switch_status() -> Result<ClaudeSwitchStatus> {
    let settings_path = claude_settings_path();
    let backup_path = claude_settings_backup_path(&settings_path);

    if !settings_path.exists() {
        return Ok(ClaudeSwitchStatus {
            enabled: false,
            base_url: None,
            has_backup: backup_path.exists(),
            settings_path,
        });
    }

    let text = read_settings_text(&settings_path)?;
    if text.trim().is_empty() {
        return Ok(ClaudeSwitchStatus {
            enabled: false,
            base_url: None,
            has_backup: backup_path.exists(),
            settings_path,
        });
    }

    let value: serde_json::Value = match serde_json::from_str(&text) {
        Ok(v) => v,
        Err(_) => {
            return Ok(ClaudeSwitchStatus {
                enabled: false,
                base_url: None,
                has_backup: backup_path.exists(),
                settings_path,
            });
        }
    };

    let env_obj = value
        .as_object()
        .and_then(|o| o.get("env"))
        .and_then(|v| v.as_object());

    let base_url = env_obj
        .and_then(|e| e.get("ANTHROPIC_BASE_URL"))
        .and_then(|v| v.as_str())
        .map(|s| s.to_string());

    let enabled = base_url
        .as_deref()
        .is_some_and(|u| u.contains("127.0.0.1") || u.contains("localhost"));

    Ok(ClaudeSwitchStatus {
        enabled,
        base_url,
        has_backup: backup_path.exists(),
        settings_path,
    })
}

/// Warn before replacing an existing local Codex proxy patch.
pub fn guard_codex_config_before_switch_on_interactive() -> Result<()> {
    use std::io::{self, Write};

    let cfg_path = codex_config_path();
    let state_path = codex_switch_state_path();

    if !cfg_path.exists() {
        return Ok(());
    }

    let text = read_config_text(&cfg_path)?;
    if text.trim().is_empty() {
        return Ok(());
    }

    let value: Value = match text.parse() {
        Ok(value) => value,
        Err(_) => return Ok(()),
    };
    let table = match value.as_table() {
        Some(table) => table,
        None => return Ok(()),
    };

    let current_provider = table
        .get("model_provider")
        .and_then(|value| value.as_str())
        .unwrap_or_default();
    if current_provider != "codex_proxy" {
        return Ok(());
    }

    let empty_map = toml::map::Map::new();
    let providers_table = table
        .get("model_providers")
        .and_then(|value| value.as_table())
        .unwrap_or(&empty_map);
    let empty_provider = toml::map::Map::new();
    let proxy_table = providers_table
        .get("codex_proxy")
        .and_then(|value| value.as_table())
        .unwrap_or(&empty_provider);

    let base_url = proxy_table
        .get("base_url")
        .and_then(|value| value.as_str())
        .unwrap_or_default();
    let name = proxy_table
        .get("name")
        .and_then(|value| value.as_str())
        .unwrap_or_default();

    let is_local = base_url.contains("127.0.0.1") || base_url.contains("localhost");
    let is_helper_name = name == "codex-helper";
    if !is_local && !is_helper_name {
        return Ok(());
    }

    if !state_path.exists() {
        eprintln!(
            "Warning: Codex currently points to the local proxy ({base_url}), but no codex-helper switch state {:?} was found; please inspect ~/.codex/config.toml manually if this is unexpected.",
            state_path
        );
        return Ok(());
    }

    let is_tty = atty::is(atty::Stream::Stdin) && atty::is(atty::Stream::Stdout);
    if !is_tty {
        eprintln!(
            "Notice: Codex currently points to local codex-helper ({base_url}) and switch state {:?} exists; run `codex-helper switch off` to disable the local proxy patch while preserving other config edits.",
            state_path
        );
        return Ok(());
    }

    eprintln!(
        "Codex currently points to local codex-helper ({base_url}), and switch state {:?} exists.\nThis usually means the previous run did not switch off cleanly.\nDisable the local proxy patch now while preserving other config edits? [Y/n] ",
        state_path
    );
    eprint!("> ");
    io::stdout().flush().ok();

    let mut input = String::new();
    if let Err(err) = io::stdin().read_line(&mut input) {
        eprintln!("Failed to read input: {err}");
        return Ok(());
    }
    let answer = input.trim();
    let yes =
        answer.is_empty() || answer.eq_ignore_ascii_case("y") || answer.eq_ignore_ascii_case("yes");

    if yes {
        if let Err(err) = switch_off() {
            eprintln!("Failed to disable local Codex proxy patch: {err}");
        } else {
            eprintln!("Disabled local Codex proxy patch.");
        }
    } else {
        eprintln!("Keeping current Codex config unchanged.");
    }

    Ok(())
}

fn read_settings_text(path: &Path) -> Result<String> {
    if !path.exists() {
        return Ok(String::new());
    }
    let mut file = fs::File::open(path).with_context(|| format!("open {:?}", path))?;
    let mut buf = String::new();
    file.read_to_string(&mut buf)
        .with_context(|| format!("read {:?}", path))?;
    Ok(buf)
}

pub fn claude_switch_on(port: u16) -> Result<()> {
    claude_switch_on_base_url(&format!("http://127.0.0.1:{port}"))
}

pub fn claude_switch_on_base_url(base_url: &str) -> Result<()> {
    let base_url = crate::control_plane_client::normalize_base_url(base_url)
        .ok_or_else(|| anyhow!("Claude proxy base URL must start with http:// or https://"))?;
    let settings_path = claude_settings_path();
    let backup_path = claude_settings_backup_path(&settings_path);

    if settings_path.exists() && !backup_path.exists() {
        fs::copy(&settings_path, &backup_path).with_context(|| {
            format!(
                "backup Claude settings {:?} -> {:?}",
                settings_path, backup_path
            )
        })?;
    } else if !settings_path.exists() && !backup_path.exists() {
        // If Claude Code has no settings yet, create a sentinel backup so we can restore
        // to the "absent" state on claude_switch_off.
        if let Some(parent) = backup_path.parent() {
            fs::create_dir_all(parent).with_context(|| format!("create_dir_all {:?}", parent))?;
        }
        fs::write(&backup_path, CLAUDE_ABSENT_BACKUP_SENTINEL)
            .with_context(|| format!("write {:?}", backup_path))?;
    }

    let text = read_settings_text(&settings_path)?;
    let mut value: serde_json::Value = if text.trim().is_empty() {
        serde_json::json!({})
    } else {
        serde_json::from_str(&text).with_context(|| format!("parse {:?} as JSON", settings_path))?
    };

    let obj = value
        .as_object_mut()
        .ok_or_else(|| anyhow!("Claude settings root must be an object"))?;

    let env_val = obj
        .entry("env".to_string())
        .or_insert_with(|| serde_json::json!({}));
    let env_obj = env_val
        .as_object_mut()
        .ok_or_else(|| anyhow!("Claude settings env must be an object"))?;

    env_obj.insert(
        "ANTHROPIC_BASE_URL".to_string(),
        serde_json::Value::String(base_url),
    );

    let new_text = serde_json::to_string_pretty(&value)?;
    write_text_file(&settings_path, &new_text)
        .with_context(|| format!("write {:?}", settings_path))?;

    eprintln!(
        "[EXPERIMENTAL] Updated {:?} to use Claude proxy via codex-helper",
        settings_path
    );
    Ok(())
}

pub fn claude_switch_off() -> Result<()> {
    let settings_path = claude_settings_path();
    let backup_path = claude_settings_backup_path(&settings_path);
    if backup_path.exists() {
        let text = read_settings_text(&backup_path)?;
        if text.trim() == CLAUDE_ABSENT_BACKUP_SENTINEL {
            if settings_path.exists() {
                fs::remove_file(&settings_path)
                    .with_context(|| format!("remove {:?} (restore absent)", settings_path))?;
            }
        } else {
            atomic_write(&settings_path, &text)
                .with_context(|| format!("restore {:?} -> {:?}", backup_path, settings_path))?;
            eprintln!(
                "[EXPERIMENTAL] Restored Claude settings from backup {:?}",
                backup_path
            );
        }
        fs::remove_file(&backup_path)
            .with_context(|| format!("remove stale backup {:?}", backup_path))?;
    }
    Ok(())
}

#[cfg(test)]
#[allow(clippy::items_after_test_module)]
mod tests {
    use super::*;
    use base64::Engine as _;
    use serde_json::json;
    use std::path::Path;
    use std::sync::{Mutex, OnceLock};

    struct ScopedEnv {
        saved: Vec<(String, Option<String>)>,
    }

    impl ScopedEnv {
        fn new() -> Self {
            Self { saved: Vec::new() }
        }

        unsafe fn set_path(&mut self, key: &str, value: &Path) {
            self.saved.push((key.to_string(), std::env::var(key).ok()));
            unsafe { std::env::set_var(key, value) };
        }

        unsafe fn set_str(&mut self, key: &str, value: &str) {
            self.saved.push((key.to_string(), std::env::var(key).ok()));
            unsafe { std::env::set_var(key, value) };
        }
    }

    impl Drop for ScopedEnv {
        fn drop(&mut self) {
            for (key, old) in self.saved.drain(..).rev() {
                unsafe {
                    match old {
                        Some(value) => std::env::set_var(&key, value),
                        None => std::env::remove_var(&key),
                    }
                }
            }
        }
    }

    fn env_lock() -> std::sync::MutexGuard<'static, ()> {
        static LOCK: OnceLock<Mutex<()>> = OnceLock::new();
        match LOCK.get_or_init(|| Mutex::new(())).lock() {
            Ok(guard) => guard,
            Err(err) => err.into_inner(),
        }
    }

    struct TestEnv {
        _lock: std::sync::MutexGuard<'static, ()>,
        _env: ScopedEnv,
        codex_home: PathBuf,
        claude_home: PathBuf,
    }

    fn setup_temp_env() -> TestEnv {
        let lock = env_lock();
        let root =
            std::env::temp_dir().join(format!("codex-helper-switch-test-{}", uuid::Uuid::new_v4()));
        std::fs::create_dir_all(&root).expect("create temp root");

        let codex_home = root.join(".codex");
        let claude_home = root.join(".claude");
        let proxy_home = root.join(".codex-helper");
        std::fs::create_dir_all(&codex_home).expect("create temp codex home");
        std::fs::create_dir_all(&claude_home).expect("create temp claude home");
        std::fs::create_dir_all(&proxy_home).expect("create temp proxy home");

        let mut scoped = ScopedEnv::new();
        unsafe {
            scoped.set_path("CODEX_HOME", &codex_home);
            scoped.set_path("CLAUDE_HOME", &claude_home);
            scoped.set_path("CODEX_HELPER_HOME", &proxy_home);
            scoped.set_path("HOME", &root);
            scoped.set_path("USERPROFILE", &root);
            scoped.set_str("CODEX_HELPER_IMAGEGEN_TEST_KEY", "sk-relay-test");
            scoped.set_str("CODEX_HELPER_MISSING_IMAGEGEN_KEY", "");
        }

        TestEnv {
            _lock: lock,
            _env: scoped,
            codex_home,
            claude_home,
        }
    }

    fn write_file(path: &Path, content: &str) {
        if let Some(parent) = path.parent() {
            std::fs::create_dir_all(parent).expect("create parent directories");
        }
        std::fs::write(path, content).expect("write test file");
    }

    fn read_file(path: &Path) -> String {
        std::fs::read_to_string(path).expect("read test file")
    }

    fn write_helper_codex_config(env: &TestEnv, content: &str) {
        let proxy_home = env.codex_home.parent().unwrap().join(".codex-helper");
        write_file(&proxy_home.join("config.toml"), content.trim_start());
    }

    fn write_helper_codex_config_with_env_auth(env: &TestEnv) {
        write_helper_codex_config(
            env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );
    }

    fn fake_chatgpt_jwt(email: &str, account_id: &str, plan_type: &str) -> String {
        let header = json!({
            "alg": "none",
            "typ": "JWT",
        });
        let payload = json!({
            "email": email,
            "https://api.openai.com/auth": {
                "chatgpt_account_id": account_id,
                "chatgpt_plan_type": plan_type,
            },
        });
        let encode = |value: serde_json::Value| {
            let bytes = serde_json::to_vec(&value).expect("serialize JWT part");
            base64::engine::general_purpose::URL_SAFE_NO_PAD.encode(bytes)
        };
        format!("{}.{}.sig", encode(header), encode(payload))
    }

    fn chatgpt_auth_json(email: &str, account_id: &str, plan_type: &str) -> String {
        let id_token = fake_chatgpt_jwt(email, account_id, plan_type);
        serde_json::to_string_pretty(&json!({
            "auth_mode": "chatgpt",
            "OPENAI_API_KEY": "sk-platform-onboarding",
            "tokens": {
                "id_token": id_token,
                "access_token": "chatgpt-access-token",
                "refresh_token": "chatgpt-refresh-token",
                "account_id": account_id,
            },
            "last_refresh": "2026-05-17T00:00:00Z",
        }))
        .expect("serialize chatgpt auth fixture")
    }

    fn chatgpt_auth_json_without_plan(email: &str, account_id: &str) -> String {
        let header = json!({
            "alg": "none",
            "typ": "JWT",
        });
        let payload = json!({
            "email": email,
            "https://api.openai.com/auth": {
                "chatgpt_account_id": account_id,
            },
        });
        let encode = |value: serde_json::Value| {
            let bytes = serde_json::to_vec(&value).expect("serialize JWT part");
            base64::engine::general_purpose::URL_SAFE_NO_PAD.encode(bytes)
        };
        let id_token = format!("{}.{}.sig", encode(header), encode(payload));
        serde_json::to_string_pretty(&json!({
            "auth_mode": "chatgpt",
            "OPENAI_API_KEY": null,
            "tokens": {
                "id_token": id_token,
                "access_token": "chatgpt-access-token",
                "refresh_token": "chatgpt-refresh-token",
                "account_id": account_id,
            },
            "last_refresh": "2026-05-17T00:00:00Z",
        }))
        .expect("serialize chatgpt auth fixture")
    }

    #[test]
    fn codex_switch_on_preserves_unrelated_toml_comments_and_fields() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");

        let original = r#"# top comment
model_provider = "openai"

[model_providers.openai]
# keep this comment
name = "OpenAI"
base_url = "https://api.openai.com/v1"
request_max_retries = 3

[projects."D:\\Work"]
trust_level = "trusted"
"#;

        write_file(&cfg_path, original);
        switch_on(3211).expect("switch_on should preserve editable TOML structure");

        let updated = read_file(&cfg_path);
        assert!(updated.contains("# top comment"));
        assert!(updated.contains("# keep this comment"));
        assert!(updated.contains("[model_providers.openai]"));
        assert!(updated.contains("[projects."));
        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("[model_providers.codex_proxy]"));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3211\""));
    }

    #[test]
    fn codex_switch_on_keeps_existing_proxy_retry_setting() {
        let text = r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "custom"
base_url = "http://127.0.0.1:1111"
request_max_retries = 5
"#;

        let updated = switch_on_codex_toml_with_mode(text, 3333, CodexPatchMode::Default)
            .expect("switch_on should update the local proxy provider in place");

        assert!(updated.contains("request_max_retries = 5"));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("name = \"codex-helper\""));
    }

    #[test]
    fn codex_patch_plan_chatgpt_bridge_keeps_account_auth_shape_and_safe_write_order() {
        let plan = CodexPatchPlan::for_switch_on(
            CodexPatchMode::ChatGptBridge,
            CodexSwitchOptions::default(),
        )
        .expect("chatgpt bridge patch plan");

        assert_eq!(plan.provider().provider_name(), "codex-helper");
        assert_eq!(plan.provider().requires_openai_auth().value(), Some(true));
        assert_eq!(plan.provider().supports_websockets().value(), Some(false));
        assert_eq!(plan.auth(), CodexAuthPatchPlan::PatchChatGptBridge);
        assert_eq!(
            plan.effect_order(),
            CodexSwitchOnEffectOrder::StateConfigAuth
        );
        assert!(
            !plan.requires_bridge_runtime_ready(),
            "chatgpt bridge validates the existing Codex auth.json instead of helper upstream auth"
        );
    }

    #[test]
    fn codex_patch_plan_official_relay_uses_openai_identity_without_auth_facade() {
        let plan = CodexPatchPlan::for_switch_on(
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("official relay patch plan");

        assert_eq!(plan.provider().provider_name(), "OpenAI");
        assert_eq!(plan.provider().requires_openai_auth().value(), None);
        assert_eq!(plan.provider().supports_websockets().value(), Some(true));
        assert_eq!(
            plan.auth(),
            CodexAuthPatchPlan::RestoreOriginalIfHelperPatched
        );
        assert_eq!(
            plan.effect_order(),
            CodexSwitchOnEffectOrder::ConfigAuthState
        );
        assert!(plan.requires_bridge_runtime_ready());
    }

    #[test]
    fn codex_patch_plan_official_imagegen_combines_openai_identity_and_auth_facade() {
        let plan = CodexPatchPlan::for_switch_on(
            CodexPatchMode::OfficialImagegenBridge,
            CodexSwitchOptions::default(),
        )
        .expect("official imagegen patch plan");

        assert_eq!(plan.provider().provider_name(), "OpenAI");
        assert_eq!(plan.provider().requires_openai_auth().value(), None);
        assert_eq!(plan.provider().supports_websockets().value(), Some(false));
        assert_eq!(plan.auth(), CodexAuthPatchPlan::PatchImagegenFacade);
        assert_eq!(
            plan.effect_order(),
            CodexSwitchOnEffectOrder::StateConfigAuth
        );
        assert!(plan.requires_bridge_runtime_ready());
    }

    #[test]
    fn codex_patch_plan_rejects_websocket_transport_without_official_identity() {
        let err = CodexPatchPlan::for_switch_on(
            CodexPatchMode::ImagegenBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect_err("websocket transport should be official identity only");

        assert!(
            err.to_string()
                .contains("requires --preset official-relay or --preset official-imagegen")
        );
    }

    #[test]
    fn codex_switch_on_toml_is_driven_by_patch_plan() {
        let text = r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "codex-helper"
base_url = "http://127.0.0.1:1111"
wire_api = "responses"
requires_openai_auth = true
supports_websockets = false
"#;
        let plan = CodexPatchPlan::for_switch_on(
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("official relay patch plan");

        let updated =
            switch_on_codex_toml_with_plan(text, 3333, plan).expect("plan-driven TOML patch");

        assert!(updated.contains("name = \"OpenAI\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("supports_websockets = true"));
        assert!(!updated.contains("requires_openai_auth"));
    }

    #[test]
    fn codex_auth_edit_for_switch_on_plan_restores_prior_facade_from_current_text() {
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");
        let mut state = CodexSwitchState {
            version: 2,
            patch_mode: Some(CodexPatchMode::OfficialImagegenBridge),
            responses_websocket: false,
            original_config_absent: false,
            original_model_provider: None,
            original_codex_proxy: None,
            had_model_providers: false,
            original_auth_json_absent: false,
            original_auth_json: Some(original_auth.clone()),
            patched_auth_json: Some("{}".to_string()),
        };
        let plan = CodexPatchPlan::for_switch_on(
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions::default(),
        )
        .expect("official relay patch plan");

        let edit = auth_restore_edit_from_state_and_current(&mut state, Some("{}"));

        assert_eq!(
            plan.auth(),
            CodexAuthPatchPlan::RestoreOriginalIfHelperPatched
        );
        assert_eq!(edit, CodexAuthEdit::Write(original_auth));
        assert_eq!(state.patched_auth_json, None);
    }

    #[test]
    fn codex_switch_on_chatgpt_bridge_sets_openai_auth_flags() {
        let updated = switch_on_codex_toml_with_mode("", 3333, CodexPatchMode::ChatGptBridge)
            .expect("switch_on should write chatgpt bridge fields");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("requires_openai_auth = true"));
        assert!(updated.contains("supports_websockets = false"));
    }

    #[test]
    fn codex_switch_on_imagegen_bridge_uses_default_proxy_flags() {
        let updated = switch_on_codex_toml_with_mode("", 3333, CodexPatchMode::ImagegenBridge)
            .expect("switch_on should write imagegen bridge fields");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("name = \"codex-helper\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(!updated.contains("requires_openai_auth"));
        assert!(!updated.contains("supports_websockets"));
    }

    #[test]
    fn codex_switch_on_official_relay_bridge_sets_openai_name_and_disables_websockets() {
        let updated = switch_on_codex_toml_with_mode("", 3333, CodexPatchMode::OfficialRelayBridge)
            .expect("switch_on should write official relay preset fields");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("name = \"OpenAI\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("wire_api = \"responses\""));
        assert!(updated.contains("supports_websockets = false"));
        assert!(!updated.contains("requires_openai_auth"));
    }

    #[test]
    fn codex_switch_on_official_relay_bridge_can_enable_responses_websocket() {
        let updated = switch_on_codex_toml_with_options(
            "",
            3333,
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("switch_on should write official relay preset fields with websocket transport");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("name = \"OpenAI\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("wire_api = \"responses\""));
        assert!(updated.contains("supports_websockets = true"));
        assert!(!updated.contains("requires_openai_auth"));
    }

    #[test]
    fn codex_switch_on_official_imagegen_bridge_sets_openai_name_and_disables_websockets() {
        let updated =
            switch_on_codex_toml_with_mode("", 3333, CodexPatchMode::OfficialImagegenBridge)
                .expect("switch_on should write official imagegen preset fields");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("name = \"OpenAI\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("wire_api = \"responses\""));
        assert!(updated.contains("supports_websockets = false"));
        assert!(!updated.contains("requires_openai_auth"));
    }

    #[test]
    fn codex_switch_on_official_imagegen_bridge_can_enable_responses_websocket() {
        let updated = switch_on_codex_toml_with_options(
            "",
            3333,
            CodexPatchMode::OfficialImagegenBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("switch_on should write official imagegen preset fields with websocket transport");

        assert!(updated.contains("model_provider = \"codex_proxy\""));
        assert!(updated.contains("name = \"OpenAI\""));
        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(updated.contains("wire_api = \"responses\""));
        assert!(updated.contains("supports_websockets = true"));
        assert!(!updated.contains("requires_openai_auth"));
    }

    #[test]
    fn empty_auth_json_facade_detection_uses_json_semantics() {
        assert!(auth_json_is_empty_chatgpt_facade_text(Some("{}")));
        assert!(auth_json_is_empty_chatgpt_facade_text(Some("{\n}\n")));
        assert!(!auth_json_is_empty_chatgpt_facade_text(Some(
            r#"{"auth_mode":"chatgpt"}"#
        )));
        assert!(!auth_json_is_empty_chatgpt_facade_text(None));
        assert!(!auth_json_is_empty_chatgpt_facade_text(Some("not-json")));
    }

    #[test]
    fn codex_switch_on_default_removes_bridge_only_flags() {
        let text = r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "codex-helper"
base_url = "http://127.0.0.1:1111"
wire_api = "responses"
requires_openai_auth = true
supports_websockets = false
"#;

        let updated = switch_on_codex_toml_with_mode(text, 3333, CodexPatchMode::Default)
            .expect("switch_on should switch local proxy back to default mode");

        assert!(updated.contains("base_url = \"http://127.0.0.1:3333\""));
        assert!(!updated.contains("requires_openai_auth"));
        assert!(!updated.contains("supports_websockets"));
    }

    #[test]
    fn remote_control_config_patch_sets_remote_connections_and_removes_removed_key() {
        let text = r#"
[features]
remote_control = true
apps = true
"#;

        let updated = ensure_codex_remote_connections_feature_in_toml(text)
            .expect("remote-control config patch should parse");

        assert!(updated.contains("remote_connections = true"));
        assert!(updated.contains("apps = true"));
        assert!(!updated.contains("remote_control = true"));
        assert!(
            codex_remote_connections_feature_enabled_from_toml(&updated)
                .expect("remote_connections should parse")
        );
        assert!(
            !codex_remote_control_feature_present_in_toml(&updated)
                .expect("remote_control should parse")
        );
    }

    #[test]
    fn codex_remote_control_enable_patches_config_backs_up_db_and_writes_sqlite() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let db_dir = env.codex_home.join("sqlite");
        let db_path = db_dir.join("codex-dev.db");
        std::fs::create_dir_all(&db_dir).expect("create sqlite dir");
        write_file(
            &cfg_path,
            r#"
[features]
remote_control = true
"#
            .trim_start(),
        );

        {
            let conn = rusqlite::Connection::open(&db_path).expect("open test sqlite");
            conn.execute(
                "CREATE TABLE local_app_server_feature_enablement (
                    feature_name TEXT PRIMARY KEY,
                    enabled INTEGER NOT NULL,
                    updated_at INTEGER NOT NULL
                )",
                [],
            )
            .expect("create feature table");
            conn.execute(
                "INSERT INTO local_app_server_feature_enablement (feature_name, enabled, updated_at)
                 VALUES (?1, ?2, ?3)",
                rusqlite::params!["remote_control", 0_i64, 7_i64],
            )
            .expect("seed feature row");
        }

        let result =
            codex_remote_control_enable().expect("remote-control enablement should succeed");

        assert!(result.backup_path.exists(), "backup should be created");
        assert!(result.status.remote_connections_enabled);
        assert!(!result.status.remote_control_config_present);
        assert_eq!(result.status.db_enabled, Some(true));
        assert!(result.status.db_updated_at.unwrap_or_default() > 7);

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("remote_connections = true"));
        assert!(!updated_cfg.contains("remote_control = true"));
    }

    #[test]
    fn codex_remote_control_enable_does_not_patch_config_when_db_schema_is_missing() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let db_dir = env.codex_home.join("sqlite");
        let db_path = db_dir.join("codex-dev.db");
        std::fs::create_dir_all(&db_dir).expect("create sqlite dir");
        write_file(
            &cfg_path,
            r#"
[features]
apps = true
"#
            .trim_start(),
        );
        {
            let _conn = rusqlite::Connection::open(&db_path).expect("open empty test sqlite");
        }
        let original = read_file(&cfg_path);

        let err = codex_remote_control_enable()
            .expect_err("remote-control enablement should reject missing feature table");

        assert!(
            err.to_string()
                .contains("local_app_server_feature_enablement")
        );
        assert_eq!(read_file(&cfg_path), original);
    }

    #[test]
    fn remote_control_log_scan_detects_successful_enablement_set() {
        let env = setup_temp_env();
        let log_dir = env.codex_home.join("log");
        std::fs::create_dir_all(&log_dir).expect("create log dir");
        write_file(
            &log_dir.join("codex-app.log"),
            r#"{"method":"experimentalFeature/enablement/set","errorCode":null}"#,
        );

        assert!(
            codex_remote_control_successful_enablement_log_seen_in_dir(&log_dir)
                .expect("scan logs")
        );
    }

    fn startup_readiness_input(changed: bool) -> CodexStartupReadinessInput {
        CodexStartupReadinessInput {
            expected_port: 3211,
            expected_patch_mode: CodexPatchMode::Default,
            expected_responses_websocket: false,
            client_state_changed_this_startup: changed,
            switch_error: None,
        }
    }

    fn has_startup_issue(
        report: &CodexStartupReadiness,
        kind: CodexStartupReadinessIssueKind,
    ) -> bool {
        report.issues.iter().any(|issue| issue.kind == kind)
    }

    #[test]
    fn codex_tui_startup_readiness_is_quiet_when_switch_is_ready() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        switch_on(3211).expect("switch_on should create a valid local proxy patch");

        let report = codex_tui_startup_readiness(startup_readiness_input(false));

        assert_eq!(report.issues, Vec::new());
    }

    #[test]
    fn codex_tui_startup_readiness_reports_client_state_changed() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        switch_on(3211).expect("switch_on should create a valid local proxy patch");

        let report = codex_tui_startup_readiness(startup_readiness_input(true));

        assert!(has_startup_issue(
            &report,
            CodexStartupReadinessIssueKind::ClientStateChanged
        ));
        assert_eq!(report.issues.len(), 1);
    }

    #[test]
    fn codex_tui_startup_readiness_warns_for_local_proxy_without_switch_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "codex-helper"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
"#
            .trim_start(),
        );

        let report = codex_tui_startup_readiness(startup_readiness_input(false));

        assert!(has_startup_issue(
            &report,
            CodexStartupReadinessIssueKind::MissingSwitchState
        ));
    }

    #[test]
    fn codex_tui_startup_readiness_warns_official_bridge_with_preferred_group_multi_provider() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.a]
base_url = "https://a.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.providers.b]
base_url = "https://b.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.routing]
entry = "main"
affinity_policy = "preferred-group"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["a", "b"]
"#,
        );
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );

        let mut input = startup_readiness_input(false);
        input.expected_patch_mode = CodexPatchMode::OfficialRelayBridge;
        let report = codex_tui_startup_readiness(input);

        assert!(has_startup_issue(
            &report,
            CodexStartupReadinessIssueKind::OfficialRelayAffinityPolicy
        ));
    }

    #[test]
    fn codex_tui_startup_readiness_accepts_official_bridge_with_fallback_sticky() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.a]
base_url = "https://a.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.providers.b]
base_url = "https://b.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.routing]
entry = "main"
affinity_policy = "fallback-sticky"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["a", "b"]
"#,
        );
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );

        let mut input = startup_readiness_input(false);
        input.expected_patch_mode = CodexPatchMode::OfficialRelayBridge;
        let report = codex_tui_startup_readiness(input);

        assert!(!has_startup_issue(
            &report,
            CodexStartupReadinessIssueKind::OfficialRelayAffinityPolicy
        ));
    }

    #[test]
    fn codex_tui_startup_readiness_warns_when_remote_control_log_is_unconfirmed() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let db_dir = env.codex_home.join("sqlite");
        let db_path = db_dir.join("codex-dev.db");
        std::fs::create_dir_all(&db_dir).expect("create sqlite dir");
        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[features]
remote_connections = true

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        {
            let conn = rusqlite::Connection::open(&db_path).expect("open test sqlite");
            conn.execute(
                "CREATE TABLE local_app_server_feature_enablement (
                    feature_name TEXT PRIMARY KEY,
                    enabled INTEGER NOT NULL,
                    updated_at INTEGER NOT NULL
                )",
                [],
            )
            .expect("create feature table");
            conn.execute(
                "INSERT INTO local_app_server_feature_enablement (feature_name, enabled, updated_at)
                 VALUES (?1, ?2, ?3)",
                rusqlite::params!["remote_control", 1_i64, 7_i64],
            )
            .expect("seed feature row");
        }
        switch_on(3211).expect("switch_on should preserve features");

        let report = codex_tui_startup_readiness(startup_readiness_input(false));

        assert!(has_startup_issue(
            &report,
            CodexStartupReadinessIssueKind::RemoteControlLogUnconfirmed
        ));
    }

    #[test]
    fn chatgpt_bridge_auth_patch_preserves_other_auth_json_fields() {
        let mut input: serde_json::Value =
            serde_json::from_str(&chatgpt_auth_json("user@example.com", "account-1", "plus"))
                .expect("valid fixture");
        input["last_refresh"] = json!(123);
        input["unrelated"] = json!("keep");

        let updated = chatgpt_bridge_auth_json_text(&serde_json::to_string_pretty(&input).unwrap())
            .expect("auth json patch should preserve unrelated fields");
        let value: serde_json::Value = serde_json::from_str(&updated).expect("valid json");
        let object = value.as_object().expect("root object");

        assert_eq!(
            object.get("auth_mode").and_then(|value| value.as_str()),
            Some("chatgpt")
        );
        assert!(
            object
                .get("OPENAI_API_KEY")
                .is_some_and(|value| value.is_null())
        );
        assert_eq!(
            object
                .get("tokens")
                .and_then(|value| value.get("access_token"))
                .and_then(|value| value.as_str()),
            Some("chatgpt-access-token")
        );
        assert_eq!(
            object
                .get("tokens")
                .and_then(|value| value.get("account_id"))
                .and_then(|value| value.as_str()),
            Some("account-1")
        );
        assert_eq!(
            object.get("last_refresh").and_then(|value| value.as_i64()),
            Some(123)
        );
        assert_eq!(
            object.get("unrelated").and_then(|value| value.as_str()),
            Some("keep")
        );
    }

    #[test]
    fn chatgpt_bridge_auth_patch_rejects_incomplete_login_state() {
        let err = chatgpt_bridge_auth_json_text(r#"{"auth_mode":"chatgpt","OPENAI_API_KEY":null}"#)
            .expect_err("empty ChatGPT auth state should be rejected");

        let message = err.to_string();
        assert!(message.contains("complete ChatGPT login state"));
        assert!(message.contains("tokens.id_token"));
        assert!(message.contains("tokens.access_token"));
        assert!(message.contains("last_refresh"));
    }

    #[test]
    fn chatgpt_bridge_auth_patch_rejects_api_key_only_auth() {
        let err = chatgpt_bridge_auth_json_text(
            r#"{"auth_mode":"apikey","OPENAI_API_KEY":"sk-old","account_id":"acct_1"}"#,
        )
        .expect_err("API key auth should not be converted into fake ChatGPT auth");

        assert!(
            err.to_string()
                .contains("Open Codex and sign in with ChatGPT first")
        );
    }

    #[test]
    fn chatgpt_bridge_auth_patch_accepts_chatgpt_auth_without_plan_claim() {
        let input = chatgpt_auth_json_without_plan("user@example.com", "acct_1");

        let updated = chatgpt_bridge_auth_json_text(&input)
            .expect("Codex maps missing ChatGPT plan claims to unknown");
        let value: serde_json::Value = serde_json::from_str(&updated).expect("valid json");

        assert_eq!(
            value.get("auth_mode").and_then(|value| value.as_str()),
            Some("chatgpt")
        );
        assert!(
            value
                .get("OPENAI_API_KEY")
                .is_some_and(|value| value.is_null())
        );
    }

    #[test]
    fn imagegen_bridge_auth_patch_writes_empty_chatgpt_facade() {
        let updated = imagegen_bridge_auth_json_text().expect("serialize facade auth");
        let value: serde_json::Value = serde_json::from_str(&updated).expect("valid json");

        assert!(
            value.as_object().is_some_and(serde_json::Map::is_empty),
            "imagegen bridge must rely on Codex's default ChatGPT mode fallback, not an explicit auth_mode field"
        );
    }

    #[test]
    fn helper_auth_patch_match_uses_json_semantics() {
        assert!(auth_json_matches_helper_patch(
            Some(r#"{"auth_mode":"chatgpt"}"#),
            "{\n  \"auth_mode\": \"chatgpt\"\n}"
        ));
        assert!(!auth_json_matches_helper_patch(
            Some(r#"{"auth_mode":"apikey"}"#),
            "{\n  \"auth_mode\": \"chatgpt\"\n}"
        ));
    }

    #[test]
    fn imagegen_bridge_ready_check_rejects_missing_codex_upstreams() {
        let _env = setup_temp_env();

        let err = ensure_imagegen_bridge_runtime_ready()
            .expect_err("imagegen bridge should require codex-helper upstreams");

        assert!(
            err.to_string()
                .contains("requires at least one enabled Codex upstream")
        );
    }

    #[test]
    fn imagegen_bridge_ready_check_rejects_unresolved_upstream_env() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_MISSING_IMAGEGEN_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );

        let err = ensure_imagegen_bridge_runtime_ready()
            .expect_err("imagegen bridge should require resolved upstream auth");

        let message = err.to_string();
        assert!(message.contains("no enabled Codex upstream credential is available"));
        assert!(message.contains("CODEX_HELPER_MISSING_IMAGEGEN_KEY"));
    }

    #[test]
    fn official_relay_bridge_ready_check_rejects_unresolved_upstream_env() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_MISSING_IMAGEGEN_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );

        let err = ensure_bridge_runtime_ready(CodexPatchMode::OfficialRelayBridge)
            .expect_err("official relay preset should require resolved upstream auth");

        let message = err.to_string();
        assert!(message.contains("official-relay"));
        assert!(message.contains("no enabled Codex upstream credential is available"));
        assert!(message.contains("CODEX_HELPER_MISSING_IMAGEGEN_KEY"));
    }

    #[test]
    fn official_imagegen_bridge_ready_check_rejects_unresolved_upstream_env() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_MISSING_IMAGEGEN_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );

        let err = ensure_bridge_runtime_ready(CodexPatchMode::OfficialImagegenBridge)
            .expect_err("official imagegen preset should require resolved upstream auth");

        let message = err.to_string();
        assert!(message.contains("official-imagegen"));
        assert!(message.contains("no enabled Codex upstream credential is available"));
        assert!(message.contains("CODEX_HELPER_MISSING_IMAGEGEN_KEY"));
    }

    #[test]
    fn official_relay_bridge_with_responses_websocket_ready_check_rejects_unresolved_upstream_env()
    {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_MISSING_IMAGEGEN_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );

        let err = switch_on_with_options(
            3211,
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect_err("official relay preset with websocket should require resolved upstream auth");

        let message = err.to_string();
        assert!(message.contains("official-relay"));
        assert!(message.contains("no enabled Codex upstream credential is available"));
        assert!(message.contains("CODEX_HELPER_MISSING_IMAGEGEN_KEY"));
    }

    #[test]
    fn codex_switch_on_chatgpt_bridge_patches_auth_json() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(
            &auth_path,
            &chatgpt_auth_json("user@example.com", "acct_1", "plus"),
        );

        switch_on_with_mode(3211, CodexPatchMode::ChatGptBridge)
            .expect("switch_on bridge should patch config and auth");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("requires_openai_auth = true"));
        assert!(updated_cfg.contains("supports_websockets = false"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert_eq!(
            updated_auth
                .get("auth_mode")
                .and_then(|value| value.as_str()),
            Some("chatgpt")
        );
        assert!(
            updated_auth
                .get("OPENAI_API_KEY")
                .is_some_and(|value| value.is_null())
        );
        assert_eq!(
            updated_auth
                .get("tokens")
                .and_then(|value| value.get("account_id"))
                .and_then(|value| value.as_str()),
            Some("acct_1")
        );
    }

    #[test]
    fn codex_switch_on_imagegen_bridge_patches_auth_json_and_records_mode() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::ImagegenBridge)
            .expect("switch_on imagegen bridge should patch config and auth");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(!updated_cfg.contains("requires_openai_auth"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert!(
            updated_auth
                .as_object()
                .is_some_and(serde_json::Map::is_empty)
        );

        let state_text = read_file(&state_path);
        let state: serde_json::Value = serde_json::from_str(&state_text).expect("valid state");
        assert_eq!(
            state.get("patch_mode").and_then(|value| value.as_str()),
            Some("imagegen-bridge")
        );
        assert_eq!(
            state
                .get("original_auth_json")
                .and_then(|value| value.as_str()),
            Some(original_auth.as_str())
        );
        assert_eq!(
            state
                .get("patched_auth_json")
                .and_then(|value| value.as_str()),
            Some("{}")
        );
    }

    #[test]
    fn codex_switch_on_official_relay_bridge_records_mode_without_auth_json_patch() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::OfficialRelayBridge)
            .expect("switch_on official relay preset should patch config");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(updated_cfg.contains("name = \"OpenAI\""));
        assert!(updated_cfg.contains("supports_websockets = false"));
        assert!(!updated_cfg.contains("requires_openai_auth"));
        assert_eq!(read_file(&auth_path), original_auth);

        let state_text = read_file(&state_path);
        let state: serde_json::Value = serde_json::from_str(&state_text).expect("valid state");
        assert_eq!(
            state.get("patch_mode").and_then(|value| value.as_str()),
            Some("official-relay-bridge")
        );
        assert!(state.get("patched_auth_json").is_none());

        let status = codex_switch_status().expect("status should load");
        assert_eq!(status.patch_mode, Some(CodexPatchMode::OfficialRelayBridge));
        assert_eq!(status.requires_openai_auth, None);
        assert_eq!(status.supports_websockets, Some(false));
    }

    #[test]
    fn codex_switch_on_official_relay_bridge_records_transport_option_without_auth_json_patch() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_options(
            3211,
            CodexPatchMode::OfficialRelayBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("switch_on official relay preset should patch config with websocket transport");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(updated_cfg.contains("name = \"OpenAI\""));
        assert!(updated_cfg.contains("supports_websockets = true"));
        assert!(!updated_cfg.contains("requires_openai_auth"));
        assert_eq!(read_file(&auth_path), original_auth);

        let state_text = read_file(&state_path);
        let state: serde_json::Value = serde_json::from_str(&state_text).expect("valid state");
        assert_eq!(
            state.get("patch_mode").and_then(|value| value.as_str()),
            Some("official-relay-bridge")
        );
        assert_eq!(
            state
                .get("responses_websocket")
                .and_then(|value| value.as_bool()),
            Some(true)
        );
        assert!(state.get("patched_auth_json").is_none());

        let status = codex_switch_status().expect("status should load");
        assert_eq!(status.patch_mode, Some(CodexPatchMode::OfficialRelayBridge));
        assert_eq!(status.requires_openai_auth, None);
        assert_eq!(status.supports_websockets, Some(true));
    }

    #[test]
    fn codex_switch_on_official_imagegen_bridge_records_mode_and_patches_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::OfficialImagegenBridge)
            .expect("switch_on official imagegen preset should patch config and auth");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(updated_cfg.contains("name = \"OpenAI\""));
        assert!(updated_cfg.contains("supports_websockets = false"));
        assert!(!updated_cfg.contains("requires_openai_auth"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert!(
            updated_auth
                .as_object()
                .is_some_and(serde_json::Map::is_empty)
        );

        let state_text = read_file(&state_path);
        let state: serde_json::Value = serde_json::from_str(&state_text).expect("valid state");
        assert_eq!(
            state.get("patch_mode").and_then(|value| value.as_str()),
            Some("official-imagegen-bridge")
        );
        assert_eq!(
            state
                .get("original_auth_json")
                .and_then(|value| value.as_str()),
            Some(original_auth.as_str())
        );
        assert_eq!(
            state
                .get("patched_auth_json")
                .and_then(|value| value.as_str()),
            Some("{}")
        );

        let status = codex_switch_status().expect("status should load");
        assert_eq!(
            status.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert_eq!(status.requires_openai_auth, None);
        assert_eq!(status.supports_websockets, Some(false));
    }

    #[test]
    fn codex_switch_on_official_imagegen_bridge_records_transport_option_and_patches_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_options(
            3211,
            CodexPatchMode::OfficialImagegenBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("switch_on official imagegen preset should patch config and auth with websocket transport");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(updated_cfg.contains("name = \"OpenAI\""));
        assert!(updated_cfg.contains("supports_websockets = true"));
        assert!(!updated_cfg.contains("requires_openai_auth"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert!(
            updated_auth
                .as_object()
                .is_some_and(serde_json::Map::is_empty)
        );

        let state_text = read_file(&state_path);
        let state: serde_json::Value = serde_json::from_str(&state_text).expect("valid state");
        assert_eq!(
            state.get("patch_mode").and_then(|value| value.as_str()),
            Some("official-imagegen-bridge")
        );
        assert_eq!(
            state
                .get("responses_websocket")
                .and_then(|value| value.as_bool()),
            Some(true)
        );
        assert_eq!(
            state
                .get("original_auth_json")
                .and_then(|value| value.as_str()),
            Some(original_auth.as_str())
        );
        assert_eq!(
            state
                .get("patched_auth_json")
                .and_then(|value| value.as_str()),
            Some("{}")
        );

        let status = codex_switch_status().expect("status should load");
        assert_eq!(
            status.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert_eq!(status.requires_openai_auth, None);
        assert_eq!(status.supports_websockets, Some(true));
    }

    #[test]
    fn codex_switch_on_with_configured_preset_uses_helper_client_patch() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.client_patch]
preset = "official-imagegen"
responses_websocket = true

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_IMAGEGEN_TEST_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_configured_preset(3211)
            .expect("configured official-imagegen preset should patch config and auth");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("model_provider = \"codex_proxy\""));
        assert!(updated_cfg.contains("name = \"OpenAI\""));
        assert!(updated_cfg.contains("base_url = \"http://127.0.0.1:3211\""));
        assert!(updated_cfg.contains("supports_websockets = true"));
        assert!(!updated_cfg.contains("requires_openai_auth"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert!(
            updated_auth
                .as_object()
                .is_some_and(serde_json::Map::is_empty)
        );

        let status = codex_switch_status().expect("status should load");
        assert_eq!(
            status.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert_eq!(status.provider_name.as_deref(), Some("OpenAI"));
        assert_eq!(status.supports_websockets, Some(true));
    }

    #[test]
    fn codex_switch_status_infers_official_relay_bridge_without_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );

        let status = codex_switch_status().expect("status should load");

        assert!(status.enabled);
        assert_eq!(status.patch_mode, Some(CodexPatchMode::OfficialRelayBridge));
        assert_eq!(status.supports_websockets, Some(false));
        assert_eq!(status.requires_openai_auth, None);
        assert!(!status.has_switch_state);
    }

    #[test]
    fn codex_switch_status_keeps_websocket_as_transport_for_official_relay_without_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = true
"#
            .trim_start(),
        );

        let status = codex_switch_status().expect("status should load");

        assert!(status.enabled);
        assert_eq!(status.patch_mode, Some(CodexPatchMode::OfficialRelayBridge));
        assert_eq!(status.supports_websockets, Some(true));
        assert_eq!(status.requires_openai_auth, None);
        assert!(!status.has_switch_state);
    }

    #[test]
    fn codex_switch_status_infers_official_imagegen_bridge_from_empty_auth_facade_without_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );
        write_file(&auth_path, "{}");

        let status = codex_switch_status().expect("status should load");

        assert!(status.enabled);
        assert_eq!(
            status.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert_eq!(status.supports_websockets, Some(false));
        assert_eq!(status.requires_openai_auth, None);
        assert!(!status.has_switch_state);
    }

    #[test]
    fn codex_switch_status_keeps_websocket_as_transport_for_official_imagegen_without_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = true
"#
            .trim_start(),
        );
        write_file(&auth_path, "{}");

        let status = codex_switch_status().expect("status should load");

        assert!(status.enabled);
        assert_eq!(
            status.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert_eq!(status.supports_websockets, Some(true));
        assert_eq!(status.requires_openai_auth, None);
        assert!(!status.has_switch_state);
    }

    #[test]
    fn codex_bridge_diagnostics_reports_ready_official_imagegen_bridge() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );
        write_file(&auth_path, "{}");

        let diagnostics = codex_bridge_diagnostics();

        assert_eq!(
            diagnostics.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert!(diagnostics.enabled);
        assert!(diagnostics.remote_compaction_v1_ready);
        assert!(diagnostics.imagegen_facade_ready);
        assert!(diagnostics.upstream_auth_ready);
        assert!(!diagnostics.remote_compaction_v2_enabled);
        assert_eq!(diagnostics.worst_status(), CodexBridgeDiagnosticStatus::Ok);
    }

    #[test]
    fn codex_bridge_diagnostics_reports_ready_official_relay_with_responses_websocket() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = true
"#
            .trim_start(),
        );

        let diagnostics = codex_bridge_diagnostics();

        assert_eq!(
            diagnostics.patch_mode,
            Some(CodexPatchMode::OfficialRelayBridge)
        );
        assert!(diagnostics.enabled);
        assert!(diagnostics.remote_compaction_v1_ready);
        assert!(!diagnostics.imagegen_facade_ready);
        assert!(diagnostics.upstream_auth_ready);
        let websocket_checks = diagnostics
            .checks
            .iter()
            .filter(|check| check.id == "codex_bridge.responses_websocket")
            .collect::<Vec<_>>();
        assert!(
            websocket_checks
                .iter()
                .any(|check| check.status == CodexBridgeDiagnosticStatus::Ok),
            "expected at least one ready websocket check: {websocket_checks:?}"
        );
        assert_eq!(
            diagnostics.worst_status(),
            CodexBridgeDiagnosticStatus::Info
        );
    }

    #[test]
    fn codex_bridge_diagnostics_warns_when_remote_compaction_v2_is_enabled() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[features]
remote_compaction_v2 = true

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );
        write_file(&auth_path, "{}");

        let diagnostics = codex_bridge_diagnostics();

        assert!(diagnostics.remote_compaction_v2_enabled);
        assert_eq!(
            diagnostics.worst_status(),
            CodexBridgeDiagnosticStatus::Warn
        );
        let v2 = diagnostics
            .checks
            .iter()
            .find(|check| check.id == "codex_bridge.remote_compaction_v2")
            .expect("v2 check");
        assert_eq!(v2.status, CodexBridgeDiagnosticStatus::Warn);
        assert!(v2.message.contains("remote_compaction_v2 is enabled"));
    }

    #[test]
    fn codex_bridge_diagnostics_fails_imagegen_when_auth_facade_is_missing() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "OpenAI"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
supports_websockets = false
"#
            .trim_start(),
        );
        let state = CodexSwitchState {
            patch_mode: Some(CodexPatchMode::OfficialImagegenBridge),
            ..CodexSwitchState::from_codex_config_text("", true).expect("state")
        };
        write_codex_switch_state(&state).expect("write state");

        let diagnostics = codex_bridge_diagnostics();

        assert_eq!(
            diagnostics.patch_mode,
            Some(CodexPatchMode::OfficialImagegenBridge)
        );
        assert!(!diagnostics.imagegen_facade_ready);
        let imagegen = diagnostics
            .checks
            .iter()
            .find(|check| check.id == "codex_bridge.imagegen_facade")
            .expect("imagegen check");
        assert_eq!(imagegen.status, CodexBridgeDiagnosticStatus::Fail);
    }

    #[test]
    fn codex_bridge_runtime_auth_snapshot_reports_missing_env_names() {
        let env = setup_temp_env();
        write_helper_codex_config(
            &env,
            r#"
version = 5

[codex.providers.relay]
base_url = "https://relay.example/v1"
auth_token_env = "CODEX_HELPER_MISSING_IMAGEGEN_KEY"

[codex.routing]
entry = "main"

[codex.routing.routes.main]
strategy = "ordered-failover"
children = ["relay"]
"#,
        );

        let snapshot = codex_bridge_runtime_auth_snapshot().expect("snapshot");

        assert_eq!(snapshot.routable_upstreams, 1);
        assert_eq!(snapshot.authed_upstreams, 0);
        assert_eq!(
            snapshot.missing_env,
            vec!["CODEX_HELPER_MISSING_IMAGEGEN_KEY".to_string()]
        );
    }

    #[test]
    fn codex_switch_default_restores_imagegen_bridge_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::ImagegenBridge)
            .expect("switch_on imagegen bridge should patch auth");
        assert_ne!(read_file(&auth_path), original_auth);

        switch_on_with_mode(3211, CodexPatchMode::Default)
            .expect("switching to default should restore auth");

        assert_eq!(read_file(&auth_path), original_auth);
        let status = codex_switch_status().expect("status should load");
        assert_eq!(status.patch_mode, Some(CodexPatchMode::Default));
    }

    #[test]
    fn codex_switch_default_restores_official_imagegen_bridge_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::OfficialImagegenBridge)
            .expect("switch_on official imagegen preset should patch auth");
        assert_ne!(read_file(&auth_path), original_auth);

        switch_on_with_mode(3211, CodexPatchMode::Default)
            .expect("switching to default should restore auth");

        assert_eq!(read_file(&auth_path), original_auth);
        let status = codex_switch_status().expect("status should load");
        assert_eq!(status.patch_mode, Some(CodexPatchMode::Default));
    }

    #[test]
    fn codex_switch_default_restores_official_imagegen_bridge_auth_json_with_responses_websocket() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_options(
            3211,
            CodexPatchMode::OfficialImagegenBridge,
            CodexSwitchOptions {
                responses_websocket: true,
            },
        )
        .expect("switch_on official imagegen preset should patch auth with websocket transport");
        assert_ne!(read_file(&auth_path), original_auth);

        switch_on_with_mode(3211, CodexPatchMode::Default)
            .expect("switching to default should restore auth");

        assert_eq!(read_file(&auth_path), original_auth);
        let status = codex_switch_status().expect("status should load");
        assert_eq!(status.patch_mode, Some(CodexPatchMode::Default));
    }

    #[test]
    fn codex_switch_off_restores_imagegen_bridge_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::ImagegenBridge)
            .expect("switch_on imagegen bridge should patch auth");
        switch_off().expect("switch_off should restore auth");

        assert_eq!(read_file(&auth_path), original_auth);
        assert!(read_file(&cfg_path).contains("model_provider = \"openai\""));
    }

    #[test]
    fn codex_switch_off_does_not_restore_auth_if_user_changed_it() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let user_auth = r#"{"auth_mode":"apikey","OPENAI_API_KEY":"sk-user"}"#;

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(
            &auth_path,
            &chatgpt_auth_json("user@example.com", "acct_1", "plus"),
        );

        switch_on_with_mode(3211, CodexPatchMode::ImagegenBridge)
            .expect("switch_on imagegen bridge should patch auth");
        write_file(&auth_path, user_auth);
        switch_off().expect("switch_off should leave user auth alone");

        assert_eq!(read_file(&auth_path), user_auth);
    }

    #[test]
    fn codex_switch_imagegen_to_chatgpt_bridge_uses_original_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::ImagegenBridge)
            .expect("switch_on imagegen bridge should patch auth");
        switch_on_with_mode(3211, CodexPatchMode::ChatGptBridge)
            .expect("switching to chatgpt bridge should use original auth snapshot");

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert_eq!(
            updated_auth
                .get("tokens")
                .and_then(|value| value.get("account_id"))
                .and_then(|value| value.as_str()),
            Some("acct_1")
        );
        assert!(
            updated_auth
                .get("OPENAI_API_KEY")
                .is_some_and(|value| value.is_null())
        );
    }

    #[test]
    fn codex_switch_official_imagegen_to_chatgpt_bridge_uses_original_auth_json() {
        let env = setup_temp_env();
        write_helper_codex_config_with_env_auth(&env);
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let original_auth = chatgpt_auth_json("user@example.com", "acct_1", "plus");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(&auth_path, &original_auth);

        switch_on_with_mode(3211, CodexPatchMode::OfficialImagegenBridge)
            .expect("switch_on official imagegen preset should patch auth");
        switch_on_with_mode(3211, CodexPatchMode::ChatGptBridge)
            .expect("switching to chatgpt bridge should use original auth snapshot");

        let updated_cfg = read_file(&cfg_path);
        assert!(updated_cfg.contains("name = \"codex-helper\""));
        assert!(updated_cfg.contains("requires_openai_auth = true"));

        let updated_auth: serde_json::Value =
            serde_json::from_str(&read_file(&auth_path)).expect("valid auth json");
        assert_eq!(
            updated_auth
                .get("tokens")
                .and_then(|value| value.get("account_id"))
                .and_then(|value| value.as_str()),
            Some("acct_1")
        );
        assert!(
            updated_auth
                .get("OPENAI_API_KEY")
                .is_some_and(|value| value.is_null())
        );
    }

    #[test]
    fn codex_switch_on_chatgpt_bridge_does_not_rewrite_already_patched_auth_json() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");

        write_file(
            &cfg_path,
            r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
            .trim_start(),
        );
        write_file(
            &auth_path,
            &chatgpt_bridge_auth_json_text(&chatgpt_auth_json(
                "user@example.com",
                "acct_1",
                "plus",
            ))
            .expect("pre-patch auth fixture"),
        );
        let before = std::fs::metadata(&auth_path)
            .expect("auth metadata")
            .modified()
            .expect("auth modified time");

        switch_on_with_mode(3211, CodexPatchMode::ChatGptBridge)
            .expect("switch_on bridge should patch config without rewriting already-patched auth");

        let after = std::fs::metadata(&auth_path)
            .expect("auth metadata")
            .modified()
            .expect("auth modified time");
        assert_eq!(before, after);
    }

    #[test]
    fn codex_switch_on_chatgpt_bridge_refuses_incomplete_auth_without_writing_config() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let auth_path = env.codex_home.join("auth.json");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");
        let original_config = r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#
        .trim_start();
        let original_auth = r#"{"auth_mode":"chatgpt","OPENAI_API_KEY":null}"#;

        write_file(&cfg_path, original_config);
        write_file(&auth_path, original_auth);

        let err = switch_on_with_mode(3211, CodexPatchMode::ChatGptBridge)
            .expect_err("incomplete ChatGPT auth should be rejected before writing config");

        assert!(err.to_string().contains("complete ChatGPT login state"));
        assert_eq!(read_file(&cfg_path), original_config);
        assert_eq!(read_file(&auth_path), original_auth);
        assert!(
            !state_path.exists(),
            "failed bridge switch must not create switch state"
        );
    }

    #[test]
    fn codex_switch_on_refuses_local_proxy_without_switch_state() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");

        write_file(
            &cfg_path,
            r#"
model_provider = "codex_proxy"

[model_providers.codex_proxy]
name = "codex-helper"
base_url = "http://127.0.0.1:3211"
"#
            .trim_start(),
        );

        let err = switch_on(3211).expect_err("switch_on should not snapshot a local proxy");
        assert!(err.to_string().contains("no switch state was found"));
        assert!(
            !state_path.exists(),
            "switch_on must not create state from an already-patched local proxy"
        );
    }

    #[test]
    fn codex_config_text_for_import_hides_proxy_created_from_absent_config() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");

        switch_on(3211).expect("switch_on should create config");
        assert!(cfg_path.exists());

        let import_text = codex_config_text_for_import()
            .expect("read import view")
            .expect("config exists");
        assert!(
            import_text.trim().is_empty(),
            "import view should not expose helper proxy as a real upstream"
        );
    }

    #[test]
    fn codex_switch_off_clears_switch_state_and_refreshes_next_snapshot() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");

        let original = r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#;
        let updated = r#"
model_provider = "packycode"

[model_providers.packycode]
name = "packycode"
base_url = "https://codex-api.packycode.com/v1"
"#;

        write_file(&cfg_path, original.trim_start());
        switch_on(3211).expect("first switch_on should succeed");
        assert!(
            state_path.exists(),
            "switch state should exist while patched"
        );
        let state_text = read_file(&state_path);
        assert!(state_text.contains("\"original_model_provider\": \"openai\""));
        assert!(
            !state_text.contains("api.openai.com"),
            "switch state should not store the full Codex config"
        );

        switch_off().expect("first switch_off should succeed");
        assert_eq!(read_file(&cfg_path), original.trim_start());
        assert!(
            !state_path.exists(),
            "switch state should be removed after patch-off to avoid stale snapshots"
        );

        write_file(&cfg_path, updated.trim_start());
        switch_on(3211).expect("second switch_on should succeed");
        let state_text = read_file(&state_path);
        assert!(state_text.contains("\"original_model_provider\": \"packycode\""));

        switch_off().expect("second switch_off should succeed");
        assert_eq!(read_file(&cfg_path), updated.trim_start());
        assert!(
            !state_path.exists(),
            "switch state should be cleaned up after the second patch-off as well"
        );
    }

    #[test]
    fn codex_switch_off_preserves_codex_runtime_config_edits() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");
        let state_path = env.codex_home.join("codex-helper-switch-state.json");

        let original = r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#;

        write_file(&cfg_path, original.trim_start());
        switch_on(3211).expect("switch_on should succeed");

        let mut during_run = read_file(&cfg_path);
        during_run.push_str(
            r#"
[projects."D:\\Projects\\rust\\codex-helper"]
trust_level = "trusted"
"#,
        );
        write_file(&cfg_path, &during_run);

        switch_off().expect("switch_off should patch rather than restore whole file");

        let updated = read_file(&cfg_path);
        assert!(updated.contains("model_provider = \"openai\""));
        assert!(updated.contains("[model_providers.openai]"));
        assert!(!updated.contains("[model_providers.codex_proxy]"));
        assert!(updated.contains("[projects."));
        assert!(updated.contains("trust_level = \"trusted\""));
        assert!(
            !state_path.exists(),
            "switch state should be removed after successful patch-off"
        );
    }

    #[test]
    fn codex_switch_off_keeps_user_provider_change_made_during_run() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");

        let original = r#"
model_provider = "openai"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"
"#;
        let user_changed = r#"
model_provider = "packycode"

[model_providers.openai]
name = "openai"
base_url = "https://api.openai.com/v1"

[model_providers.codex_proxy]
name = "codex-helper"
base_url = "http://127.0.0.1:3211"
wire_api = "responses"
request_max_retries = 0

[model_providers.packycode]
name = "packycode"
base_url = "https://codex-api.packycode.com/v1"
"#;

        write_file(&cfg_path, original.trim_start());
        switch_on(3211).expect("switch_on should succeed");
        write_file(&cfg_path, user_changed.trim_start());

        switch_off().expect("switch_off should not undo user's model_provider change");

        let updated = read_file(&cfg_path);
        assert!(updated.contains("model_provider = \"packycode\""));
        assert!(updated.contains("[model_providers.packycode]"));
        assert!(!updated.contains("[model_providers.codex_proxy]"));
    }

    #[test]
    fn codex_switch_off_preserves_new_config_when_original_was_absent() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");

        switch_on(3211).expect("switch_on should create config");
        let mut during_run = read_file(&cfg_path);
        during_run.push_str(
            r#"
[projects."D:\\Projects\\rust\\codex-helper"]
trust_level = "trusted"
"#,
        );
        write_file(&cfg_path, &during_run);

        switch_off().expect("switch_off should remove only local proxy fields");

        let updated = read_file(&cfg_path);
        assert!(!updated.contains("model_provider = \"codex_proxy\""));
        assert!(!updated.contains("[model_providers.codex_proxy]"));
        assert!(updated.contains("[projects."));
        assert!(updated.contains("trust_level = \"trusted\""));
    }

    #[test]
    fn codex_switch_off_removes_empty_config_created_by_switch_on() {
        let env = setup_temp_env();
        let cfg_path = env.codex_home.join("config.toml");

        switch_on(3211).expect("switch_on should create config");
        assert!(cfg_path.exists());

        switch_off().expect("switch_off should restore absent config state");

        assert!(
            !cfg_path.exists(),
            "config created only for the local proxy should be removed"
        );
    }

    #[test]
    fn claude_switch_off_clears_backup_and_refreshes_next_snapshot() {
        let env = setup_temp_env();
        let settings_path = env.claude_home.join("settings.json");
        let backup_path = env.claude_home.join("settings.json.codex-helper-backup");

        let original = r#"{
  "env": {
    "ANTHROPIC_BASE_URL": "https://api.anthropic.com/v1",
    "ANTHROPIC_API_KEY": "sk-ant-1"
  }
}"#;
        let updated = r#"{
  "env": {
    "ANTHROPIC_BASE_URL": "https://anthropic-proxy.example/v1",
    "ANTHROPIC_API_KEY": "sk-ant-2"
  }
}"#;

        write_file(&settings_path, original);
        claude_switch_on(3211).expect("first claude_switch_on should succeed");
        assert!(
            backup_path.exists(),
            "backup should exist while switched on"
        );

        claude_switch_off().expect("first claude_switch_off should succeed");
        assert_eq!(read_file(&settings_path), original);
        assert!(
            !backup_path.exists(),
            "backup should be removed after Claude restore to avoid stale snapshots"
        );

        write_file(&settings_path, updated);
        claude_switch_on(3211).expect("second claude_switch_on should succeed");
        assert_eq!(read_file(&backup_path), updated);

        claude_switch_off().expect("second claude_switch_off should succeed");
        assert_eq!(read_file(&settings_path), updated);
        assert!(
            !backup_path.exists(),
            "backup should be cleaned up after the second Claude restore as well"
        );
    }
}

/// Warn before replacing an existing local Claude proxy patch.
pub fn guard_claude_settings_before_switch_on_interactive() -> Result<()> {
    use std::io::{self, Write};

    let settings_path = claude_settings_path();
    if !settings_path.exists() {
        return Ok(());
    }
    let backup_path = claude_settings_backup_path(&settings_path);

    let text = read_settings_text(&settings_path)?;
    if text.trim().is_empty() {
        return Ok(());
    }

    let value: serde_json::Value = match serde_json::from_str(&text) {
        Ok(value) => value,
        Err(_) => return Ok(()),
    };
    let obj = match value.as_object() {
        Some(obj) => obj,
        None => return Ok(()),
    };
    let env_obj = match obj.get("env").and_then(|value| value.as_object()) {
        Some(env_obj) => env_obj,
        None => return Ok(()),
    };

    let base_url = env_obj
        .get("ANTHROPIC_BASE_URL")
        .and_then(|value| value.as_str())
        .unwrap_or_default();

    let is_local = base_url.contains("127.0.0.1") || base_url.contains("localhost");
    if !is_local {
        return Ok(());
    }

    if !backup_path.exists() {
        eprintln!(
            "Warning: Claude settings {:?} points ANTHROPIC_BASE_URL to a local address ({base_url}), but no backup file {:?} was found; please inspect this config file manually if this is unexpected.",
            settings_path, backup_path
        );
        return Ok(());
    }

    let is_tty = atty::is(atty::Stream::Stdin) && atty::is(atty::Stream::Stdout);
    if !is_tty {
        eprintln!(
            "Notice: Claude settings {:?} already points to the local proxy ({base_url}), and backup {:?} exists; run `codex-helper switch off --claude` if you want to restore the original config.",
            settings_path, backup_path
        );
        return Ok(());
    }

    eprintln!(
        "Claude settings {:?} already points ANTHROPIC_BASE_URL to the local proxy ({base_url}), and backup {:?} exists.\nThis usually means the previous run did not switch off cleanly.\nRestore the original Claude settings now? [Y/n] ",
        settings_path, backup_path
    );
    eprint!("> ");
    io::stdout().flush().ok();

    let mut input = String::new();
    if let Err(err) = io::stdin().read_line(&mut input) {
        eprintln!("Failed to read input: {err}");
        return Ok(());
    }
    let answer = input.trim();
    let yes =
        answer.is_empty() || answer.eq_ignore_ascii_case("y") || answer.eq_ignore_ascii_case("yes");

    if yes {
        if let Err(err) = claude_switch_off() {
            eprintln!("Failed to restore Claude settings: {err}");
        } else {
            eprintln!("Restored Claude settings from backup.");
        }
    } else {
        eprintln!("Keeping current Claude settings unchanged.");
    }

    Ok(())
}