omni_dev/utils/

settings.rs

//! Settings and configuration utilities.
//!
//! This module provides functionality to read settings from $HOME/.omni-dev/settings.json
//! and use them as a fallback for environment variables.
//!
//! It also owns the write side: [`Settings::upsert_env_vars_in`] and
//! [`Settings::remove_env_vars_in`] (plus their base-`env` shorthands
//! [`Settings::upsert_env_vars`] / [`Settings::remove_env_vars`]) are the only
//! production paths that mutate the settings file. Writes target the active
//! profile's `env` when a profile is given, mirroring the read-side isolation
//! of [`Settings::resolve_with`] (issue #1116). Because the `env` maps hold
//! credentials (Atlassian, Datadog), every write is hardened: parent directory
//! `0700`, file `0600`, re-tightened on each write (issue #1128).

use std::collections::{BTreeSet, HashMap};
use std::fmt;
use std::fs;
use std::path::{Path, PathBuf};
use std::sync::Mutex;

use anyhow::{Context, Result};
use serde::Deserialize;

use crate::utils::env::{EnvSource, SystemEnv};

/// Where a resolved environment value came from, for provenance reporting
/// (issue #1143).
///
/// An ambient setting — a shell export or a `settings.json` `env` entry — is
/// sticky across invocations, so warnings about security-sensitive values
/// (e.g. the claude-cli escape hatches) name the source to distinguish a
/// deliberate one-off flag from a forgotten persistent setting.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum EnvValueSource {
    /// Exported into the process environment by a command-line flag during
    /// this invocation (see `Cli::propagate_global_flags`).
    CliFlag,
    /// The process environment (a shell export or inherited variable).
    ProcessEnv,
    /// The base `env` map in `$HOME/.omni-dev/settings.json`.
    SettingsEnv,
    /// The named profile's `env` map in `$HOME/.omni-dev/settings.json`.
    SettingsProfile(String),
}

impl fmt::Display for EnvValueSource {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::CliFlag => write!(f, "command-line flag"),
            Self::ProcessEnv => write!(f, "process environment variable (e.g. a shell export)"),
            Self::SettingsEnv => write!(f, "the env map in $HOME/.omni-dev/settings.json"),
            Self::SettingsProfile(name) => {
                write!(
                    f,
                    "the profile '{name}' env map in $HOME/.omni-dev/settings.json"
                )
            }
        }
    }
}

/// Env-var keys that `Cli::propagate_global_flags` exported from command-line
/// flags this invocation. Additive-only, written once at startup, so readers
/// can attribute a process-env hit to the flag that set it rather than to an
/// ambient shell export. Not an env-mutation seam: tests exercise the sourced
/// resolvers through their injected `from_cli_flag` parameter instead.
static CLI_FLAG_EXPORTS: Mutex<BTreeSet<String>> = Mutex::new(BTreeSet::new());

/// Records that `key` was exported into the process environment by a
/// command-line flag, so [`get_env_var_sourced`] reports
/// [`EnvValueSource::CliFlag`] for it instead of
/// [`EnvValueSource::ProcessEnv`].
pub fn note_cli_flag_export(key: &str) {
    // Recover from poisoning rather than losing provenance: the set is
    // insert-only, so a panicked writer cannot leave it inconsistent.
    let mut set = CLI_FLAG_EXPORTS
        .lock()
        .unwrap_or_else(std::sync::PoisonError::into_inner);
    set.insert(key.to_string());
}

/// Returns whether `key` was exported by a command-line flag this invocation.
#[must_use]
pub fn exported_by_cli_flag(key: &str) -> bool {
    CLI_FLAG_EXPORTS
        .lock()
        .unwrap_or_else(std::sync::PoisonError::into_inner)
        .contains(key)
}

/// Environment variable that selects the active profile, mirroring `AWS_PROFILE`.
///
/// Read from the **raw** process environment only (never through the profile
/// fallback, which would be circular); the `--profile` flag propagates its value
/// here in `Cli::propagate_global_flags`.
pub const PROFILE_ENV_VAR: &str = "OMNI_DEV_PROFILE";

/// A named credential/config bundle inside `settings.json` — its own `env` map,
/// selected per invocation via `--profile` / `OMNI_DEV_PROFILE`.
#[derive(Debug, Default, Deserialize)]
pub struct Profile {
    /// Environment variable overrides applied when this profile is active.
    #[serde(default)]
    pub env: HashMap<String, String>,
}

/// The `mcp` section of `settings.json` — defaults for the `omni-dev-mcp`
/// server (issue #620).
///
/// Every field is optional; an unset field falls back to the built-in default,
/// so an absent `mcp` block preserves the server's behaviour byte-for-byte. See
/// [`Settings::load_mcp`] for the loader and `crate::mcp` for the wiring.
#[derive(Debug, Default, Deserialize)]
pub struct McpSettings {
    /// Default AI model for the `ai_chat` tool, used when the tool's own
    /// `model` parameter is absent. Falls back to the model registry default.
    #[serde(default)]
    pub default_model: Option<String>,

    /// Default tracing directive for the server (e.g. `"info"`,
    /// `"omni_dev::mcp=debug"`). `RUST_LOG` overrides this when set; the
    /// built-in fallback is `"warn"`.
    #[serde(default)]
    pub log_level: Option<String>,

    /// Default cap, in bytes, on an MCP tool response before it is truncated.
    /// Falls back to the server's built-in `DEFAULT_MAX_RESPONSE_BYTES`
    /// (100 KB). A value of `0` disables truncation.
    #[serde(default)]
    pub max_response_bytes: Option<usize>,
}

/// A single named Gmail account's stored OAuth2 credentials, inside the
/// `gmail.accounts` map (issue #1500, [ADR-0066](../../docs/adrs/adr-0066.md)).
///
/// Orthogonal to [`Profile`]: selecting a Gmail account never changes the
/// active `--profile`, and vice versa. See `email_address`'s own doc
/// comment below for how it's populated and used — never for
/// authentication itself.
#[derive(Debug, Default, Deserialize)]
pub struct GmailAccountSettings {
    /// OAuth2 client id from the account's Google Cloud project.
    #[serde(default)]
    pub client_id: Option<String>,
    /// OAuth2 client secret from the account's Google Cloud project.
    #[serde(default)]
    pub client_secret: Option<String>,
    /// The long-lived refresh token obtained by `gmail auth login`.
    #[serde(default)]
    pub refresh_token: Option<String>,
    /// The OAuth2 scope this account was authorized with.
    #[serde(default)]
    pub scope: Option<String>,
    /// Mailbox address for this account. Populated opportunistically by
    /// `gmail auth status` when absent, but may also be set by hand ahead of
    /// the first login — e.g. to opt into `chrome_profile_from_email` below.
    /// An explicit value is never overwritten by the `gmail auth status`
    /// backfill. Never used for authentication itself, only for browser
    /// targeting.
    #[serde(default)]
    pub email_address: Option<String>,

    /// Opt-in (default `false`): resolve which local Chrome profile is
    /// signed into `email_address` and launch `gmail auth login`'s
    /// authorization URL targeting that profile, instead of the OS default
    /// browser. Resolution failure (Chrome not installed, zero or multiple
    /// matching profiles, ...) always falls back to the default browser —
    /// never a hard login failure. Ignored when `browser_command` is set.
    /// (issue #1505, [ADR-0067](../../docs/adrs/adr-0067.md))
    #[serde(default)]
    pub chrome_profile_from_email: bool,

    /// Explicit browser launch command for `gmail auth login`, `{url}`
    /// templated (or appended if no placeholder is present) — the manual
    /// escape hatch, e.g. to target a specific Chrome profile by hand or a
    /// non-Chrome browser entirely. Mirrors `SNOWFLAKE_BROWSER_COMMAND`
    /// (`crate::snowflake`). Takes precedence over `chrome_profile_from_email`.
    /// (issue #1505)
    #[serde(default)]
    pub browser_command: Option<String>,
}

/// The `gmail` section of `settings.json` — named Gmail accounts, selected
/// per invocation via `--account` / `OMNI_DEV_GMAIL_ACCOUNT`
/// (issue #1500, [ADR-0066](../../docs/adrs/adr-0066.md)).
///
/// An absent `gmail` block (or an empty `accounts` map) leaves Gmail
/// credential resolution on today's exact legacy path — see
/// `crate::gmail::account::resolve_account`.
#[derive(Debug, Default, Deserialize)]
pub struct GmailSettings {
    /// The account `gmail account list`/credential resolution falls back to
    /// when `--account`/`OMNI_DEV_GMAIL_ACCOUNT` is unset and more than one
    /// account is configured.
    #[serde(default)]
    pub default_account: Option<String>,

    /// Named accounts, keyed by the name passed to `--account`.
    #[serde(default)]
    pub accounts: HashMap<String, GmailAccountSettings>,
}

/// A single named Google Drive account's stored OAuth2 credentials, inside
/// the `drive.accounts` map (issue #1520,
/// [ADR-0069](../../docs/adrs/adr-0069.md)).
///
/// A field-for-field mirror of [`GmailAccountSettings`]: a Drive account and
/// a Gmail account are different credential axes even when the underlying
/// human is the same person, so the two blocks compose independently rather
/// than one nesting under the other. See `email_address`'s own doc comment
/// below for how it's populated and used — never for authentication itself.
#[derive(Debug, Default, Deserialize)]
pub struct DriveAccountSettings {
    /// OAuth2 client id from the account's Google Cloud project.
    #[serde(default)]
    pub client_id: Option<String>,
    /// OAuth2 client secret from the account's Google Cloud project.
    #[serde(default)]
    pub client_secret: Option<String>,
    /// The long-lived refresh token obtained by `drive auth login`
    /// (issue #1523).
    #[serde(default)]
    pub refresh_token: Option<String>,
    /// The OAuth2 scope this account was authorized with, as the raw string
    /// Google granted. Stays a plain `String` here (unlike Gmail's
    /// equivalent settings field, for the same reason): this is a
    /// status-reporting cache of whatever was actually granted, not the
    /// value driving a live login decision — that's
    /// [`crate::drive::auth::DriveScope`], read from this field via
    /// [`crate::drive::auth::DriveScope::from_granted`]
    /// ([ADR-0070](../../docs/adrs/adr-0070.md), reversing
    /// [ADR-0069](../../docs/adrs/adr-0069.md) §2's "no `DriveScope` enum,
    /// read-only by design").
    #[serde(default)]
    pub scope: Option<String>,
    /// Google account address for this account. Populated opportunistically
    /// by `drive auth status` when absent, but may also be set by hand ahead
    /// of the first login — e.g. to opt into `chrome_profile_from_email`
    /// below. An explicit value is never overwritten by the backfill. Never
    /// used for authentication itself, only for browser targeting.
    #[serde(default)]
    pub email_address: Option<String>,

    /// Opt-in (default `false`): resolve which local Chrome profile is
    /// signed into `email_address` and launch `drive auth login`'s
    /// authorization URL targeting that profile, instead of the OS default
    /// browser. Resolution failure (Chrome not installed, zero or multiple
    /// matching profiles, ...) always falls back to the default browser —
    /// never a hard login failure. Ignored when `browser_command` is set.
    /// (Inherited from Gmail by field —
    /// [ADR-0067](../../docs/adrs/adr-0067.md) via
    /// [ADR-0069](../../docs/adrs/adr-0069.md).)
    #[serde(default)]
    pub chrome_profile_from_email: bool,

    /// Explicit browser launch command for `drive auth login`, `{url}`
    /// templated (or appended if no placeholder is present) — the manual
    /// escape hatch, e.g. to target a specific Chrome profile by hand or a
    /// non-Chrome browser entirely. Mirrors `SNOWFLAKE_BROWSER_COMMAND`
    /// (`crate::snowflake`). Takes precedence over
    /// `chrome_profile_from_email`. (Inherited from Gmail by field —
    /// [ADR-0067](../../docs/adrs/adr-0067.md).)
    #[serde(default)]
    pub browser_command: Option<String>,
}

/// The `drive` section of `settings.json` — named Google Drive accounts,
/// selected per invocation via `--account` / `OMNI_DEV_DRIVE_ACCOUNT`
/// (issue #1520, [ADR-0069](../../docs/adrs/adr-0069.md)).
///
/// An absent `drive` block (or an empty `accounts` map) means Drive is
/// simply unconfigured — unlike [`GmailSettings`], there is no legacy
/// credential path to fall back to, Drive being brand new.
#[derive(Debug, Default, Deserialize)]
pub struct DriveSettings {
    /// The account credential resolution falls back to when
    /// `--account`/`OMNI_DEV_DRIVE_ACCOUNT` is unset and more than one
    /// account is configured.
    #[serde(default)]
    pub default_account: Option<String>,

    /// Named accounts, keyed by the name passed to `--account`.
    #[serde(default)]
    pub accounts: HashMap<String, DriveAccountSettings>,
}

/// Settings loaded from $HOME/.omni-dev/settings.json.
#[derive(Debug, Default, Deserialize)]
pub struct Settings {
    /// Environment variable overrides — the default bundle, consulted only when
    /// **no** profile is active.
    #[serde(default)]
    pub env: HashMap<String, String>,

    /// Named profiles. Selecting one replaces the base `env` in the fallback
    /// chain (isolated / AWS-faithful); see [`Settings::resolve_with`].
    #[serde(default)]
    pub profiles: HashMap<String, Profile>,

    /// MCP server defaults (issue #620); an absent block yields
    /// [`McpSettings::default`].
    #[serde(default)]
    pub mcp: McpSettings,

    /// Named Gmail accounts (issue #1500); an absent block yields
    /// [`GmailSettings::default`], which is an empty account map.
    #[serde(default)]
    pub gmail: GmailSettings,

    /// Named Google Drive accounts (issue #1520); an absent block yields
    /// [`DriveSettings::default`], which is an empty account map.
    #[serde(default)]
    pub drive: DriveSettings,
}

/// Returns the active profile name from `raw` (the process environment), or
/// `None` when `OMNI_DEV_PROFILE` is unset or empty.
///
/// Reads the **raw** env only, so it is pure over the injected source and never
/// resolves through the profile fallback.
pub fn active_profile_from<E: EnvSource>(raw: &E) -> Option<String> {
    raw.var(PROFILE_ENV_VAR).filter(|s| !s.is_empty())
}

/// Renders ` (profile '<name>')` for credential-store CLI messages, or the
/// empty string when no profile is active — so `auth login`/`logout` output
/// names the env map it actually wrote to (issue #1116).
#[must_use]
pub fn profile_suffix(profile: Option<&str>) -> String {
    profile.map_or_else(String::new, |name| format!(" (profile '{name}')"))
}

/// An [`EnvSource`](crate::utils::env::EnvSource) with the settings/profile
/// fallback — the value form of [`get_env_var`].
///
/// Reads the real process environment first, then the active profile's `env`
/// (or the base `env` when no profile is active) in
/// `$HOME/.omni-dev/settings.json`.
///
/// Pass `&SettingsEnv::load()` from a thin production wrapper; tests inject a
/// pure `MapEnv` into the same `*_with(&impl EnvSource, …)` seam instead of
/// mutating the process environment.
#[derive(Debug, Default)]
pub struct SettingsEnv {
    settings: Settings,
    active_profile: Option<String>,
}

impl SettingsEnv {
    /// Loads settings from the default location, falling back to an empty
    /// settings map if they are absent or unreadable (env-only behaviour). The
    /// active profile is read from `OMNI_DEV_PROFILE`.
    pub fn load() -> Self {
        Self::load_with_profile(active_profile_from(&SystemEnv).as_deref())
    }

    /// Like [`load`](Self::load) but with the active profile supplied
    /// explicitly — for tests and embedders that select a profile without
    /// setting `OMNI_DEV_PROFILE` in the process environment.
    pub fn load_with_profile(profile: Option<&str>) -> Self {
        Self {
            settings: Settings::load().unwrap_or_default(),
            active_profile: profile.map(str::to_string),
        }
    }

    /// Wraps an already-loaded [`Settings`], skipping the disk read/parse
    /// [`load`](Self::load)/[`load_with_profile`](Self::load_with_profile)
    /// perform — for callers (e.g. `load_credentials_for`/`status_for` in
    /// `crate::gmail::auth`/`crate::drive::auth`) that already loaded
    /// `Settings` to resolve an account and would otherwise discard it just
    /// to re-read the same file a second time (issue #1533).
    pub fn from_settings(settings: Settings, profile: Option<&str>) -> Self {
        Self {
            settings,
            active_profile: profile.map(str::to_string),
        }
    }
}

impl EnvSource for SettingsEnv {
    fn var(&self, key: &str) -> Option<String> {
        self.settings
            .resolve_with(&SystemEnv, self.active_profile.as_deref(), key)
    }
}

impl Settings {
    /// Loads settings from the default location.
    pub fn load() -> Result<Self> {
        let settings_path = Self::get_settings_path()?;
        Self::load_from_path(&settings_path)
    }

    /// Loads just the [`mcp`](McpSettings) section, falling back to its defaults
    /// when the settings file is absent or unreadable — so the MCP server always
    /// boots even with a malformed `settings.json` (issue #620). Mirrors the
    /// graceful `unwrap_or_default` of [`SettingsEnv::load`].
    pub fn load_mcp() -> McpSettings {
        Self::load().map(|s| s.mcp).unwrap_or_default()
    }

    /// Loads settings from a specific path.
    pub fn load_from_path<P: AsRef<Path>>(path: P) -> Result<Self> {
        let path = path.as_ref();

        // If file doesn't exist, return default settings
        if !path.exists() {
            return Ok(Self::default());
        }

        // Read and parse the settings file
        let content = fs::read_to_string(path)
            .with_context(|| format!("Failed to read settings file: {}", path.display()))?;

        serde_json::from_str::<Self>(&content)
            .with_context(|| format!("Failed to parse settings file: {}", path.display()))
    }

    /// Returns the default settings path.
    pub fn get_settings_path() -> Result<PathBuf> {
        let home_dir = dirs::home_dir().context("Failed to determine home directory")?;

        Ok(home_dir.join(".omni-dev").join("settings.json"))
    }

    /// Returns an environment variable with fallback to settings, honouring the
    /// active profile from `OMNI_DEV_PROFILE`.
    pub fn get_env_var(&self, key: &str) -> Option<String> {
        self.resolve_with(&SystemEnv, active_profile_from(&SystemEnv).as_deref(), key)
    }

    /// Isolated / AWS-faithful resolution: `raw` (the process environment) wins;
    /// then the active profile's `env` if `active` is set, else the base `env`.
    /// The base map is **not** consulted when a profile is active, so a missing
    /// key fails loud rather than silently reusing a default credential against
    /// the wrong tenant.
    ///
    /// This is the pure seam: production wrappers pass `&SystemEnv`; tests pass
    /// a `MapEnv` and an explicit `active`, mutating no process-global state.
    pub fn resolve_with<E: EnvSource>(
        &self,
        raw: &E,
        active: Option<&str>,
        key: &str,
    ) -> Option<String> {
        self.resolve_with_source(raw, active, key)
            .map(|(value, _)| value)
    }

    /// Like [`Settings::resolve_with`], but also reports which layer supplied
    /// the value: the raw process environment, the active profile's `env`, or
    /// the base `env` (issue #1143). Same precedence, same profile isolation.
    ///
    /// A [`EnvValueSource::CliFlag`] attribution is layered on top by
    /// [`get_env_var_sourced`], which knows about flag exports; this resolver
    /// only distinguishes what it can see.
    pub fn resolve_with_source<E: EnvSource>(
        &self,
        raw: &E,
        active: Option<&str>,
        key: &str,
    ) -> Option<(String, EnvValueSource)> {
        if let Some(value) = raw.var(key) {
            return Some((value, EnvValueSource::ProcessEnv));
        }
        match active {
            Some(name) => self
                .profiles
                .get(name)
                .and_then(|p| p.env.get(key).cloned())
                .map(|value| (value, EnvValueSource::SettingsProfile(name.to_string()))),
            None => self
                .env
                .get(key)
                .cloned()
                .map(|value| (value, EnvValueSource::SettingsEnv)),
        }
    }

    /// Merges the given key/value pairs into the base `env` object of the
    /// settings file at `path` — [`Settings::upsert_env_vars_in`] with no
    /// profile.
    pub fn upsert_env_vars(path: &Path, vars: &[(&str, &str)]) -> Result<()> {
        Self::upsert_env_vars_in(path, None, vars)
    }

    /// Merges the given key/value pairs into the `env` object targeted by
    /// `profile` — `profiles.<name>.env` when `Some`, the base `env` when
    /// `None` — creating the file, its parent directory, and any missing
    /// intermediate objects as needed. Writes therefore land where
    /// [`Settings::resolve_with`] will look for them (issue #1116).
    ///
    /// A `profile` absent from the file is created rather than rejected; the
    /// CLI validates the active profile before dispatch, so this only affects
    /// library callers.
    ///
    /// The file is read and written as a generic JSON value, so every other
    /// field (other profiles, unknown keys) is preserved verbatim. Because the
    /// `env` maps hold credentials, the write is hardened: parent directory
    /// `0700`, file `0600` (see [`write_settings`]).
    pub fn upsert_env_vars_in(
        path: &Path,
        profile: Option<&str>,
        vars: &[(&str, &str)],
    ) -> Result<()> {
        let mut settings_value = read_or_default_settings(path)?;

        let env = ensure_env_object(&mut settings_value, profile)?;
        for (key, value) in vars {
            env.insert(
                (*key).to_string(),
                serde_json::Value::String((*value).to_string()),
            );
        }

        write_settings(path, &settings_value)
    }

    /// Removes the given keys from the base `env` object of the settings file
    /// at `path` — [`Settings::remove_env_vars_in`] with no profile.
    pub fn remove_env_vars(path: &Path, keys: &[&str]) -> Result<bool> {
        Self::remove_env_vars_in(path, None, keys)
    }

    /// Removes the given keys from the `env` object targeted by `profile`
    /// (`profiles.<name>.env` when `Some`, the base `env` when `None`),
    /// leaving all other settings — including the same keys in other env
    /// maps — intact.
    ///
    /// Returns `true` if any key was present in the targeted map and removed
    /// (the file is rewritten, hardened as in
    /// [`Settings::upsert_env_vars_in`]), `false` when the file did not
    /// exist, the targeted map was absent, or it contained none of the keys
    /// (the file is left untouched).
    pub fn remove_env_vars_in(path: &Path, profile: Option<&str>, keys: &[&str]) -> Result<bool> {
        if !path.exists() {
            return Ok(false);
        }
        let mut settings_value = read_or_default_settings(path)?;

        let mut removed = false;
        if let Some(env) = env_object_mut(&mut settings_value, profile) {
            for key in keys {
                if env.remove(*key).is_some() {
                    removed = true;
                }
            }
        }

        if removed {
            write_settings(path, &settings_value)?;
        }
        Ok(removed)
    }

    /// Validates that `name` is a known profile, returning a hard error that
    /// lists the known profiles (sorted) otherwise. Called once at the CLI
    /// boundary so a typo never silently falls back to base credentials.
    pub fn validate_profile(&self, name: &str) -> Result<()> {
        if self.profiles.contains_key(name) {
            return Ok(());
        }
        let known = if self.profiles.is_empty() {
            "(none)".to_string()
        } else {
            let mut names: Vec<&str> = self.profiles.keys().map(String::as_str).collect();
            names.sort_unstable();
            names.join(", ")
        };
        Err(anyhow::anyhow!(
            "unknown profile '{name}'; known profiles: {known}"
        ))
    }

    /// Merges the given key/value pairs into `gmail.accounts.<account>`,
    /// creating the file, its parent directory, and any missing intermediate
    /// objects as needed (issue #1500,
    /// [ADR-0066](../../docs/adrs/adr-0066.md)). Same hardening and
    /// unknown-field preservation as [`Settings::upsert_env_vars_in`] — no
    /// new file-handling code.
    ///
    /// `vars` takes [`serde_json::Value`] rather than `&str` (widened in
    /// #1523, PR #1528 review) because `GmailAccountSettings` has non-string
    /// fields (`chrome_profile_from_email: bool`) — a hard-coded
    /// `Value::String` wrap would write the JSON string `"true"` into a
    /// `bool` field, and since `Settings` deserializes as one unit, the next
    /// `Settings::load()` would hard-fail parsing the *entire* file. Callers
    /// writing a string field pass `serde_json::Value::String(...)`
    /// explicitly.
    pub fn upsert_gmail_account(
        path: &Path,
        account: &str,
        vars: &[(&str, serde_json::Value)],
    ) -> Result<()> {
        let mut settings_value = read_or_default_settings(path)?;

        let entry = ensure_object_at(&mut settings_value, &["gmail", "accounts", account])?;
        for (key, value) in vars {
            entry.insert((*key).to_string(), value.clone());
        }

        write_settings(path, &settings_value)
    }

    /// Removes `gmail.accounts.<account>` entirely — an account is coherent
    /// as a unit, unlike the key-by-key removal
    /// [`Settings::remove_env_vars_in`] does. Also clears
    /// `gmail.default_account` if it named the removed account, so
    /// resolution (`src/gmail/account.rs::resolve_default_account`) falls
    /// back to the sole-remaining-account rule instead of hard-erroring on
    /// a dangling pointer (issue #1529). Returns `true` if the account was
    /// present and removed, `false` when the file, `gmail`,
    /// `gmail.accounts`, or the named account did not exist (the file is
    /// left untouched in that case).
    pub fn remove_gmail_account(path: &Path, account: &str) -> Result<bool> {
        if !path.exists() {
            return Ok(false);
        }
        let mut settings_value = read_or_default_settings(path)?;

        let removed = object_at_mut(&mut settings_value, &["gmail", "accounts"])
            .is_some_and(|accounts| accounts.remove(account).is_some());

        if removed {
            if let Some(gmail) = object_at_mut(&mut settings_value, &["gmail"]) {
                if gmail.get("default_account").and_then(|v| v.as_str()) == Some(account) {
                    gmail.remove("default_account");
                }
            }
            write_settings(path, &settings_value)?;
        }
        Ok(removed)
    }

    /// Sets (`Some`) or clears (`None`) `gmail.default_account`. Always
    /// writes, mirroring [`Settings::upsert_env_vars_in`]'s unconditional-write
    /// semantics rather than [`Settings::remove_env_vars_in`]'s
    /// changed-only one, since this is fundamentally an upsert of a single
    /// scalar rather than a set of keys.
    pub fn set_gmail_default_account(path: &Path, account: Option<&str>) -> Result<()> {
        let mut settings_value = read_or_default_settings(path)?;

        match account {
            Some(name) => {
                let gmail = ensure_object_at(&mut settings_value, &["gmail"])?;
                gmail.insert(
                    "default_account".to_string(),
                    serde_json::Value::String(name.to_string()),
                );
            }
            None => {
                if let Some(gmail) = object_at_mut(&mut settings_value, &["gmail"]) {
                    gmail.remove("default_account");
                }
            }
        }

        write_settings(path, &settings_value)
    }

    /// Merges the given key/value pairs into `drive.accounts.<account>`,
    /// creating the file, its parent directory, and any missing intermediate
    /// objects as needed (issue #1522,
    /// [ADR-0069](../../docs/adrs/adr-0069.md)). Same hardening and
    /// unknown-field preservation as [`Settings::upsert_env_vars_in`] — no
    /// new file-handling code.
    ///
    /// `vars` takes [`serde_json::Value`], not `&str` — see
    /// [`Settings::upsert_gmail_account`]'s doc comment for why (this
    /// helper's twin, and the write path the PR #1528 review comment
    /// flagged the bug against).
    pub fn upsert_drive_account(
        path: &Path,
        account: &str,
        vars: &[(&str, serde_json::Value)],
    ) -> Result<()> {
        let mut settings_value = read_or_default_settings(path)?;

        let entry = ensure_object_at(&mut settings_value, &["drive", "accounts", account])?;
        for (key, value) in vars {
            entry.insert((*key).to_string(), value.clone());
        }

        write_settings(path, &settings_value)
    }

    /// Removes `drive.accounts.<account>` entirely — an account is coherent
    /// as a unit, exactly like [`Settings::remove_gmail_account`]. Also
    /// clears `drive.default_account` if it named the removed account, for
    /// the same dangling-pointer reason (issue #1529) — Drive has no
    /// legacy-credential fallback to soften a stale default the way
    /// Gmail's resolution table can (ADR-0069 §3), so the failure would be
    /// total. Returns `true` if the account was present and removed,
    /// `false` when the file, `drive`, `drive.accounts`, or the named
    /// account did not exist (the file is left untouched in that case).
    pub fn remove_drive_account(path: &Path, account: &str) -> Result<bool> {
        if !path.exists() {
            return Ok(false);
        }
        let mut settings_value = read_or_default_settings(path)?;

        let removed = object_at_mut(&mut settings_value, &["drive", "accounts"])
            .is_some_and(|accounts| accounts.remove(account).is_some());

        if removed {
            if let Some(drive) = object_at_mut(&mut settings_value, &["drive"]) {
                if drive.get("default_account").and_then(|v| v.as_str()) == Some(account) {
                    drive.remove("default_account");
                }
            }
            write_settings(path, &settings_value)?;
        }
        Ok(removed)
    }

    /// Sets (`Some`) or clears (`None`) `drive.default_account`. Always
    /// writes, mirroring [`Settings::set_gmail_default_account`]'s
    /// unconditional-write semantics — fundamentally an upsert of a single
    /// scalar rather than a set of keys.
    pub fn set_drive_default_account(path: &Path, account: Option<&str>) -> Result<()> {
        let mut settings_value = read_or_default_settings(path)?;

        match account {
            Some(name) => {
                let drive = ensure_object_at(&mut settings_value, &["drive"])?;
                drive.insert(
                    "default_account".to_string(),
                    serde_json::Value::String(name.to_string()),
                );
            }
            None => {
                if let Some(drive) = object_at_mut(&mut settings_value, &["drive"]) {
                    drive.remove("default_account");
                }
            }
        }

        write_settings(path, &settings_value)
    }
}

/// Navigates `root` to the env object targeted by `profile` — the base `env`
/// when `None`, `profiles.<name>.env` when `Some`. Thin specialization of
/// [`ensure_object_at`] for the two-level `env`/`profiles.<name>.env` shape.
fn ensure_env_object<'a>(
    root: &'a mut serde_json::Value,
    profile: Option<&str>,
) -> Result<&'a mut serde_json::Map<String, serde_json::Value>> {
    match profile {
        Some(name) => ensure_object_at(root, &["profiles", name, "env"]),
        None => ensure_object_at(root, &["env"]),
    }
}

/// Navigates `root` to the env object targeted by `profile`, or `None` when
/// any node on the way is absent or not an object. Thin specialization of
/// [`object_at_mut`] for the two-level `env`/`profiles.<name>.env` shape.
fn env_object_mut<'a>(
    root: &'a mut serde_json::Value,
    profile: Option<&str>,
) -> Option<&'a mut serde_json::Map<String, serde_json::Value>> {
    match profile {
        Some(name) => object_at_mut(root, &["profiles", name, "env"]),
        None => object_at_mut(root, &["env"]),
    }
}

/// Navigates `root` through each of `segments` in turn, creating missing
/// intermediate objects and replacing non-object nodes along the way, and
/// returns the object at the end of the path. The creating counterpart of
/// [`object_at_mut`], for upserts. Generalizes what were previously two
/// hardcoded two-level walks (`env`, `profiles.<name>.env`) to an arbitrary
/// depth, so a third settings substructure (`gmail.accounts.<name>`, issue
/// #1500) reuses the same file-handling code rather than duplicating it.
fn ensure_object_at<'a>(
    root: &'a mut serde_json::Value,
    segments: &[&str],
) -> Result<&'a mut serde_json::Map<String, serde_json::Value>> {
    let mut current = root;
    for segment in segments {
        if !current
            .get(*segment)
            .is_some_and(serde_json::Value::is_object)
        {
            current[*segment] = serde_json::json!({});
        }
        current = current
            .get_mut(*segment)
            .context("Internal error: target key missing immediately after being created")?;
    }
    current
        .as_object_mut()
        .context("Internal error: target key is not an object after initialization")
}

/// Navigates `root` through each of `segments` in turn, or returns `None`
/// when any node on the way is absent or not an object. The non-creating
/// counterpart of [`ensure_object_at`], for removals.
fn object_at_mut<'a>(
    root: &'a mut serde_json::Value,
    segments: &[&str],
) -> Option<&'a mut serde_json::Map<String, serde_json::Value>> {
    let mut current = root;
    for segment in segments {
        current = current.get_mut(*segment)?;
    }
    current.as_object_mut()
}

/// Reads and parses the settings file at `path` as a generic JSON value
/// (preserving unknown fields), or returns `{}` when the file does not exist.
fn read_or_default_settings(path: &Path) -> Result<serde_json::Value> {
    if path.exists() {
        let content = fs::read_to_string(path)
            .with_context(|| format!("Failed to read {}", path.display()))?;
        serde_json::from_str(&content)
            .with_context(|| format!("Failed to parse {}", path.display()))
    } else {
        Ok(serde_json::json!({}))
    }
}

/// The single hardened write site for the settings file: creates the parent
/// directory `0700`, writes the pretty-printed JSON through a `0600` handle
/// (no window where a fresh file is world-readable), and re-tightens a
/// pre-existing looser-permission file on every write (issue #1128).
fn write_settings(path: &Path, value: &serde_json::Value) -> Result<()> {
    if let Some(parent) = path.parent() {
        if !parent.as_os_str().is_empty() {
            crate::daemon::paths::ensure_dir_0700(parent)?;
        }
    }
    let formatted =
        serde_json::to_string_pretty(value).context("Failed to serialize settings JSON")?;
    write_file_0600(path, &formatted)
        .with_context(|| format!("Failed to write {}", path.display()))?;
    crate::daemon::paths::set_file_0600(path)?;
    Ok(())
}

/// Creates/truncates `path` with owner-only (`0600`) permissions on Unix.
#[cfg(unix)]
fn write_file_0600(path: &Path, contents: &str) -> std::io::Result<()> {
    use std::io::Write;
    use std::os::unix::fs::OpenOptionsExt;

    let mut file = fs::OpenOptions::new()
        .write(true)
        .create(true)
        .truncate(true)
        .mode(0o600)
        .open(path)?;
    file.write_all(contents.as_bytes())
}

/// Non-Unix fallback: a plain write ([`set_file_0600`](crate::daemon::paths::set_file_0600)
/// is a no-op there too).
#[cfg(not(unix))]
fn write_file_0600(path: &Path, contents: &str) -> std::io::Result<()> {
    fs::write(path, contents)
}

/// Returns an environment variable with fallback to settings, honouring the
/// active profile from `OMNI_DEV_PROFILE`.
pub fn get_env_var(key: &str) -> Result<String> {
    get_env_var_with(&SystemEnv, Settings::load, key)
}

/// Like [`get_env_var`], but also reports where the value came from.
///
/// The source is a command-line flag export, the process environment, or a
/// settings.json `env` map (issue #1143) — for warnings about
/// security-sensitive values (e.g. the claude-cli escape hatches) that
/// should name their source.
pub fn get_env_var_sourced(key: &str) -> Result<(String, EnvValueSource)> {
    get_env_var_sourced_with(&SystemEnv, Settings::load, exported_by_cli_flag(key), key)
}

/// Pure core of [`get_env_var`]: [`get_env_var_sourced_with`] with the source
/// dropped.
fn get_env_var_with<E, F>(env: &E, load: F, key: &str) -> Result<String>
where
    E: EnvSource,
    F: FnOnce() -> Result<Settings>,
{
    get_env_var_sourced_with(env, load, false, key).map(|(value, _)| value)
}

/// Pure core of [`get_env_var_sourced`]: `env` is the raw source, `load`
/// produces the settings lazily — it is invoked only on a raw-env miss,
/// preserving the no-disk fast path — and `from_cli_flag` says whether a flag
/// exported `key` this invocation (injected so tests never touch the
/// process-global flag registry). Tests inject a `MapEnv` and a closure
/// returning `Ok`/`Err` to cover both the resolved and load-failure branches
/// without touching disk.
fn get_env_var_sourced_with<E, F>(
    env: &E,
    load: F,
    from_cli_flag: bool,
    key: &str,
) -> Result<(String, EnvValueSource)>
where
    E: EnvSource,
    F: FnOnce() -> Result<Settings>,
{
    // A raw process-env hit short-circuits without loading settings from disk.
    // A flag export always lands in the process env, so the flag attribution
    // only ever applies on this branch.
    if let Some(value) = env.var(key) {
        let source = if from_cli_flag {
            EnvValueSource::CliFlag
        } else {
            EnvValueSource::ProcessEnv
        };
        return Ok((value, source));
    }
    match load() {
        Ok(settings) => settings
            .resolve_with_source(env, active_profile_from(env).as_deref(), key)
            .ok_or_else(|| anyhow::anyhow!("Environment variable not found: {key}")),
        Err(err) => {
            // If we couldn't load settings, just return the original env var error
            Err(anyhow::anyhow!("Environment variable not found: {key}").context(err))
        }
    }
}

/// Tries multiple environment variables with fallback to settings.
pub fn get_env_vars(keys: &[&str]) -> Result<String> {
    for key in keys {
        if let Ok(value) = get_env_var(key) {
            return Ok(value);
        }
    }

    Err(anyhow::anyhow!(
        "None of the environment variables found: {keys:?}"
    ))
}

#[cfg(test)]
#[allow(clippy::unwrap_used, clippy::expect_used)]
mod tests {
    use super::*;
    use crate::test_support::env::MapEnv;
    use std::env;
    use std::fs;
    use tempfile::TempDir;

    /// Builds a `Settings` with a base `env` and one profile, for the pure
    /// resolver tests (no disk, no process env).
    fn settings_with_profile() -> Settings {
        let mut base = HashMap::new();
        base.insert("ATLASSIAN_EMAIL".to_string(), "base@x.com".to_string());
        base.insert("SHARED".to_string(), "base-shared".to_string());

        let mut work_env = HashMap::new();
        work_env.insert("ATLASSIAN_EMAIL".to_string(), "me@work.com".to_string());

        let mut profiles = HashMap::new();
        profiles.insert("work".to_string(), Profile { env: work_env });

        Settings {
            env: base,
            profiles,
            ..Settings::default()
        }
    }

    #[test]
    fn settings_load_from_path() {
        // Create a temporary directory (use current dir to avoid TMPDIR issues in tarpaulin)
        let temp_dir = {
            std::fs::create_dir_all("tmp").ok();
            TempDir::new_in("tmp").unwrap()
        };
        let settings_path = temp_dir.path().join("settings.json");

        // Create a test settings file
        let settings_json = r#"{
            "env": {
                "TEST_VAR": "test_value",
                "CLAUDE_API_KEY": "test_api_key"
            }
        }"#;
        fs::write(&settings_path, settings_json).unwrap();

        // Load settings
        let settings = Settings::load_from_path(&settings_path).unwrap();

        // Check env vars
        assert_eq!(settings.env.get("TEST_VAR").unwrap(), "test_value");
        assert_eq!(settings.env.get("CLAUDE_API_KEY").unwrap(), "test_api_key");
    }

    #[test]
    fn settings_get_env_var() {
        // Create a temporary directory (use current dir to avoid TMPDIR issues in tarpaulin)
        let temp_dir = {
            std::fs::create_dir_all("tmp").ok();
            TempDir::new_in("tmp").unwrap()
        };
        let settings_path = temp_dir.path().join("settings.json");

        // Create a test settings file
        let settings_json = r#"{
            "env": {
                "TEST_VAR": "test_value",
                "CLAUDE_API_KEY": "test_api_key"
            }
        }"#;
        fs::write(&settings_path, settings_json).unwrap();

        // Load settings
        let settings = Settings::load_from_path(&settings_path).unwrap();

        // Set actual environment variable
        env::set_var("TEST_VAR_ENV", "env_value");

        // Test precedence - env var should take precedence
        env::set_var("TEST_VAR", "env_override");
        assert_eq!(settings.get_env_var("TEST_VAR").unwrap(), "env_override");

        // Test fallback to settings
        env::remove_var("TEST_VAR"); // Remove from environment
        assert_eq!(settings.get_env_var("TEST_VAR").unwrap(), "test_value");

        // Test actual env var
        assert_eq!(settings.get_env_var("TEST_VAR_ENV").unwrap(), "env_value");

        // Clean up
        env::remove_var("TEST_VAR_ENV");
    }

    // ── profile resolution (pure: MapEnv raw env, explicit active profile) ──

    #[test]
    fn resolve_no_profile_uses_base_env() {
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(
            settings
                .resolve_with(&raw, None, "ATLASSIAN_EMAIL")
                .as_deref(),
            Some("base@x.com")
        );
    }

    #[test]
    fn resolve_active_profile_uses_profile_env() {
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(
            settings
                .resolve_with(&raw, Some("work"), "ATLASSIAN_EMAIL")
                .as_deref(),
            Some("me@work.com")
        );
    }

    #[test]
    fn resolve_active_profile_does_not_consult_base() {
        // Isolated / AWS-faithful: a key present only in base is invisible while
        // a profile is active — fail loud rather than reuse a default token.
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(settings.resolve_with(&raw, Some("work"), "SHARED"), None);
    }

    #[test]
    fn resolve_process_env_wins_over_profile_and_base() {
        let settings = settings_with_profile();
        let raw = MapEnv::new().with("ATLASSIAN_EMAIL", "cli@x.com");
        assert_eq!(
            settings
                .resolve_with(&raw, Some("work"), "ATLASSIAN_EMAIL")
                .as_deref(),
            Some("cli@x.com")
        );
        assert_eq!(
            settings
                .resolve_with(&raw, None, "ATLASSIAN_EMAIL")
                .as_deref(),
            Some("cli@x.com")
        );
    }

    #[test]
    fn resolve_unknown_active_profile_yields_none() {
        // An unknown name never falls back to base; validation catches it at the
        // CLI boundary, but the resolver itself stays isolated.
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(
            settings.resolve_with(&raw, Some("nope"), "ATLASSIAN_EMAIL"),
            None
        );
    }

    // ── sourced resolution (issue #1143: provenance for warnings) ──

    #[test]
    fn resolve_with_source_process_env_is_process_env() {
        let settings = settings_with_profile();
        let raw = MapEnv::new().with("ATLASSIAN_EMAIL", "cli@x.com");
        assert_eq!(
            settings.resolve_with_source(&raw, None, "ATLASSIAN_EMAIL"),
            Some(("cli@x.com".to_string(), EnvValueSource::ProcessEnv))
        );
    }

    #[test]
    fn resolve_with_source_base_env_is_settings_env() {
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(
            settings.resolve_with_source(&raw, None, "ATLASSIAN_EMAIL"),
            Some(("base@x.com".to_string(), EnvValueSource::SettingsEnv))
        );
    }

    #[test]
    fn resolve_with_source_profile_env_names_profile() {
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(
            settings.resolve_with_source(&raw, Some("work"), "ATLASSIAN_EMAIL"),
            Some((
                "me@work.com".to_string(),
                EnvValueSource::SettingsProfile("work".to_string())
            ))
        );
    }

    #[test]
    fn resolve_with_source_missing_key_is_none() {
        let settings = settings_with_profile();
        let raw = MapEnv::new();
        assert_eq!(settings.resolve_with_source(&raw, None, "MISSING"), None);
    }

    #[test]
    fn env_value_source_display_names_each_layer() {
        assert_eq!(EnvValueSource::CliFlag.to_string(), "command-line flag");
        assert_eq!(
            EnvValueSource::ProcessEnv.to_string(),
            "process environment variable (e.g. a shell export)"
        );
        assert_eq!(
            EnvValueSource::SettingsEnv.to_string(),
            "the env map in $HOME/.omni-dev/settings.json"
        );
        assert_eq!(
            EnvValueSource::SettingsProfile("work".to_string()).to_string(),
            "the profile 'work' env map in $HOME/.omni-dev/settings.json"
        );
    }

    #[test]
    fn active_profile_from_reads_and_trims_empty() {
        assert_eq!(active_profile_from(&MapEnv::new()), None);
        assert_eq!(
            active_profile_from(&MapEnv::new().with(PROFILE_ENV_VAR, "")),
            None
        );
        assert_eq!(
            active_profile_from(&MapEnv::new().with(PROFILE_ENV_VAR, "work")).as_deref(),
            Some("work")
        );
    }

    #[test]
    fn profile_suffix_names_profile_or_is_empty() {
        assert_eq!(profile_suffix(None), "");
        assert_eq!(profile_suffix(Some("work")), " (profile 'work')");
    }

    #[test]
    fn validate_profile_accepts_known() {
        assert!(settings_with_profile().validate_profile("work").is_ok());
    }

    #[test]
    fn validate_profile_rejects_unknown_and_lists_sorted() {
        let mut settings = settings_with_profile();
        settings
            .profiles
            .insert("personal".to_string(), Profile::default());
        let err = settings.validate_profile("wrok").unwrap_err().to_string();
        assert_eq!(
            err,
            "unknown profile 'wrok'; known profiles: personal, work"
        );
    }

    #[test]
    fn validate_profile_reports_none_when_empty() {
        let settings = Settings::default();
        let err = settings.validate_profile("work").unwrap_err().to_string();
        assert_eq!(err, "unknown profile 'work'; known profiles: (none)");
    }

    #[test]
    fn settings_parse_profiles_from_json() {
        let json = r#"{
            "env": { "BASE": "b" },
            "profiles": {
                "work": { "env": { "ATLASSIAN_EMAIL": "me@work.com" } }
            }
        }"#;
        let settings: Settings = serde_json::from_str(json).unwrap();
        assert_eq!(settings.env.get("BASE").unwrap(), "b");
        assert_eq!(
            settings
                .profiles
                .get("work")
                .unwrap()
                .env
                .get("ATLASSIAN_EMAIL")
                .unwrap(),
            "me@work.com"
        );
    }

    #[test]
    fn settings_without_profiles_key_defaults_empty() {
        let settings: Settings = serde_json::from_str(r#"{ "env": {} }"#).unwrap();
        assert!(settings.profiles.is_empty());
    }

    #[test]
    fn settings_parse_mcp_section_from_json() {
        let json = r#"{
            "mcp": {
                "default_model": "claude-sonnet-4-6",
                "log_level": "info",
                "max_response_bytes": 204800
            }
        }"#;
        let settings: Settings = serde_json::from_str(json).unwrap();
        assert_eq!(
            settings.mcp.default_model.as_deref(),
            Some("claude-sonnet-4-6")
        );
        assert_eq!(settings.mcp.log_level.as_deref(), Some("info"));
        assert_eq!(settings.mcp.max_response_bytes, Some(204_800));
    }

    #[test]
    fn settings_without_mcp_key_defaults_all_none() {
        // An absent `mcp` block must leave every field unset so callers fall
        // back to the built-in defaults (byte-for-byte behaviour, issue #620).
        let settings: Settings = serde_json::from_str(r#"{ "env": {} }"#).unwrap();
        assert!(settings.mcp.default_model.is_none());
        assert!(settings.mcp.log_level.is_none());
        assert!(settings.mcp.max_response_bytes.is_none());
    }

    #[test]
    fn settings_mcp_partial_section_leaves_others_none() {
        // A block that sets only one field leaves the rest at their defaults.
        let settings: Settings =
            serde_json::from_str(r#"{ "mcp": { "log_level": "debug" } }"#).unwrap();
        assert_eq!(settings.mcp.log_level.as_deref(), Some("debug"));
        assert!(settings.mcp.default_model.is_none());
        assert!(settings.mcp.max_response_bytes.is_none());
    }

    #[test]
    fn settings_parse_gmail_section_from_json() {
        let json = r#"{
            "gmail": {
                "default_account": "work",
                "accounts": {
                    "work": {
                        "client_id": "id",
                        "client_secret": "secret",
                        "refresh_token": "token",
                        "scope": "https://www.googleapis.com/auth/gmail.modify",
                        "email_address": "alice@work.com"
                    }
                }
            }
        }"#;
        let settings: Settings = serde_json::from_str(json).unwrap();
        assert_eq!(settings.gmail.default_account.as_deref(), Some("work"));
        let account = settings.gmail.accounts.get("work").unwrap();
        assert_eq!(account.client_id.as_deref(), Some("id"));
        assert_eq!(account.client_secret.as_deref(), Some("secret"));
        assert_eq!(account.refresh_token.as_deref(), Some("token"));
        assert_eq!(
            account.scope.as_deref(),
            Some("https://www.googleapis.com/auth/gmail.modify")
        );
        assert_eq!(account.email_address.as_deref(), Some("alice@work.com"));
    }

    #[test]
    fn settings_without_gmail_key_defaults_empty() {
        let settings: Settings = serde_json::from_str(r#"{ "env": {} }"#).unwrap();
        assert!(settings.gmail.default_account.is_none());
        assert!(settings.gmail.accounts.is_empty());
    }

    #[test]
    fn settings_parse_drive_section_from_json() {
        let json = r#"{
            "drive": {
                "default_account": "work",
                "accounts": {
                    "work": {
                        "client_id": "id",
                        "client_secret": "secret",
                        "refresh_token": "token",
                        "scope": "https://www.googleapis.com/auth/drive.readonly",
                        "email_address": "alice@work.com"
                    }
                }
            }
        }"#;
        let settings: Settings = serde_json::from_str(json).unwrap();
        assert_eq!(settings.drive.default_account.as_deref(), Some("work"));
        let account = settings.drive.accounts.get("work").unwrap();
        assert_eq!(account.client_id.as_deref(), Some("id"));
        assert_eq!(account.client_secret.as_deref(), Some("secret"));
        assert_eq!(account.refresh_token.as_deref(), Some("token"));
        assert_eq!(
            account.scope.as_deref(),
            Some("https://www.googleapis.com/auth/drive.readonly")
        );
        assert_eq!(account.email_address.as_deref(), Some("alice@work.com"));
    }

    #[test]
    fn settings_without_drive_key_defaults_empty() {
        let settings: Settings = serde_json::from_str(r#"{ "env": {} }"#).unwrap();
        assert!(settings.drive.default_account.is_none());
        assert!(settings.drive.accounts.is_empty());
    }

    // ── free get_env_var seam (pure: injected raw env + lazy settings loader) ──

    #[test]
    fn get_env_var_with_returns_raw_hit_without_loading() {
        let env = MapEnv::new().with("K", "v");
        let value = get_env_var_with(&env, || panic!("must not load settings"), "K").unwrap();
        assert_eq!(value, "v");
    }

    #[test]
    fn get_env_var_with_falls_back_to_base_settings() {
        let settings = settings_with_profile();
        let env = MapEnv::new();
        let value = get_env_var_with(&env, || Ok(settings), "ATLASSIAN_EMAIL").unwrap();
        assert_eq!(value, "base@x.com");
    }

    #[test]
    fn get_env_var_with_honours_active_profile() {
        let settings = settings_with_profile();
        let env = MapEnv::new().with(PROFILE_ENV_VAR, "work");
        let value = get_env_var_with(&env, || Ok(settings), "ATLASSIAN_EMAIL").unwrap();
        assert_eq!(value, "me@work.com");
    }

    #[test]
    fn get_env_var_with_missing_key_is_not_found() {
        let env = MapEnv::new();
        let err = get_env_var_with(&env, || Ok(Settings::default()), "MISSING")
            .unwrap_err()
            .to_string();
        assert!(err.contains("Environment variable not found: MISSING"));
    }

    #[test]
    fn get_env_var_with_load_error_maps_to_not_found() {
        let env = MapEnv::new();
        let err =
            get_env_var_with(&env, || Err(anyhow::anyhow!("disk boom")), "MISSING").unwrap_err();
        // The load failure is the top-level context; the not-found error is its
        // source. The full chain (`{:#}`) carries both.
        assert_eq!(err.to_string(), "disk boom");
        let chain = format!("{err:#}");
        assert!(chain.contains("Environment variable not found: MISSING"));
    }

    // ── sourced get_env_var seam (issue #1143) ──

    #[test]
    fn get_env_var_sourced_with_raw_hit_is_process_env() {
        let env = MapEnv::new().with("K", "v");
        let resolved =
            get_env_var_sourced_with(&env, || panic!("must not load settings"), false, "K")
                .unwrap();
        assert_eq!(resolved, ("v".to_string(), EnvValueSource::ProcessEnv));
    }

    #[test]
    fn get_env_var_sourced_with_flag_export_is_cli_flag() {
        let env = MapEnv::new().with("K", "true");
        let resolved =
            get_env_var_sourced_with(&env, || panic!("must not load settings"), true, "K").unwrap();
        assert_eq!(resolved, ("true".to_string(), EnvValueSource::CliFlag));
    }

    #[test]
    fn get_env_var_sourced_with_falls_back_to_settings_sources() {
        let settings = settings_with_profile();
        let env = MapEnv::new();
        let resolved =
            get_env_var_sourced_with(&env, || Ok(settings), false, "ATLASSIAN_EMAIL").unwrap();
        assert_eq!(
            resolved,
            ("base@x.com".to_string(), EnvValueSource::SettingsEnv)
        );

        let settings = settings_with_profile();
        let env = MapEnv::new().with(PROFILE_ENV_VAR, "work");
        let resolved =
            get_env_var_sourced_with(&env, || Ok(settings), false, "ATLASSIAN_EMAIL").unwrap();
        assert_eq!(
            resolved,
            (
                "me@work.com".to_string(),
                EnvValueSource::SettingsProfile("work".to_string())
            )
        );
    }

    #[test]
    fn cli_flag_export_registry_roundtrip() {
        // Unique key: the registry is a process-global, additive-only set, so
        // this test must not share keys with other tests (or production code).
        const KEY: &str = "OMNI_DEV_TEST_1143_REGISTRY_ROUNDTRIP";
        assert!(!exported_by_cli_flag(KEY));
        note_cli_flag_export(KEY);
        assert!(exported_by_cli_flag(KEY));
    }

    // ── env-write helpers (injected paths, no HOME mutation — issue #1030) ──

    /// Creates a tempdir under `tmp/` (avoids TMPDIR issues in tarpaulin) and
    /// returns it with a `<dir>/.omni-dev/settings.json` path inside it.
    fn temp_settings_path() -> (TempDir, std::path::PathBuf) {
        let temp_dir = {
            std::fs::create_dir_all("tmp").ok();
            TempDir::new_in("tmp").unwrap()
        };
        let path = temp_dir.path().join(".omni-dev").join("settings.json");
        (temp_dir, path)
    }

    fn read_json(path: &Path) -> serde_json::Value {
        serde_json::from_str(&fs::read_to_string(path).unwrap()).unwrap()
    }

    #[test]
    fn upsert_env_vars_creates_file_and_dir_with_secure_permissions() {
        let (_tmp, path) = temp_settings_path();

        Settings::upsert_env_vars(&path, &[("A_KEY", "a"), ("B_KEY", "b")]).unwrap();

        let val = read_json(&path);
        assert_eq!(val["env"]["A_KEY"], "a");
        assert_eq!(val["env"]["B_KEY"], "b");

        // Credential store hardening (issue #1128): dir 0700, file 0600.
        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt;
            let dir_mode = fs::metadata(path.parent().unwrap())
                .unwrap()
                .permissions()
                .mode();
            assert_eq!(dir_mode & 0o777, 0o700);
            let file_mode = fs::metadata(&path).unwrap().permissions().mode();
            assert_eq!(file_mode & 0o777, 0o600);
        }
    }

    #[test]
    fn upsert_env_vars_merges_and_preserves_unknown_fields() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(&path, r#"{"env": {"OTHER_KEY": "keep_me"}, "extra": true}"#).unwrap();

        Settings::upsert_env_vars(&path, &[("A_KEY", "new")]).unwrap();

        let val = read_json(&path);
        assert_eq!(val["env"]["OTHER_KEY"], "keep_me");
        assert_eq!(val["extra"], true);
        assert_eq!(val["env"]["A_KEY"], "new");
    }

    #[test]
    fn upsert_env_vars_replaces_non_object_env() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(&path, r#"{"env": "not-an-object"}"#).unwrap();

        Settings::upsert_env_vars(&path, &[("A_KEY", "a")]).unwrap();

        assert_eq!(read_json(&path)["env"]["A_KEY"], "a");
    }

    #[cfg(unix)]
    #[test]
    fn upsert_env_vars_retightens_loose_permissions() {
        use std::os::unix::fs::PermissionsExt;

        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(&path, r#"{"env": {}}"#).unwrap();
        fs::set_permissions(&path, fs::Permissions::from_mode(0o644)).unwrap();

        Settings::upsert_env_vars(&path, &[("A_KEY", "a")]).unwrap();

        let file_mode = fs::metadata(&path).unwrap().permissions().mode();
        assert_eq!(file_mode & 0o777, 0o600);
    }

    #[test]
    fn remove_env_vars_removes_listed_keys_and_preserves_rest() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"env": {"A_KEY": "a", "B_KEY": "b", "OTHER_KEY": "keep"}, "extra": true}"#,
        )
        .unwrap();

        let removed = Settings::remove_env_vars(&path, &["A_KEY", "B_KEY", "ABSENT"]).unwrap();
        assert!(removed);

        let val = read_json(&path);
        assert!(val["env"].get("A_KEY").is_none());
        assert!(val["env"].get("B_KEY").is_none());
        assert_eq!(val["env"]["OTHER_KEY"], "keep");
        assert_eq!(val["extra"], true);
    }

    #[test]
    fn remove_env_vars_false_when_file_missing() {
        let (_tmp, path) = temp_settings_path();
        assert!(!Settings::remove_env_vars(&path, &["A_KEY"]).unwrap());
        assert!(!path.exists());
    }

    #[test]
    fn remove_env_vars_false_when_env_missing_or_not_an_object() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();

        // No "env" key at all.
        fs::write(&path, r#"{"extra": true}"#).unwrap();
        assert!(!Settings::remove_env_vars(&path, &["A_KEY"]).unwrap());

        // "env" present but not an object.
        fs::write(&path, r#"{"env": "not-an-object"}"#).unwrap();
        assert!(!Settings::remove_env_vars(&path, &["A_KEY"]).unwrap());
    }

    #[test]
    fn upsert_env_vars_bare_filename_skips_dir_creation() {
        // A bare relative filename has an empty parent — the dir-creation
        // branch must be skipped, not fail on `create_dir_all("")`.
        let name = format!("tmp-upsert-bare-{}.json", std::process::id());
        let path = Path::new(&name);

        Settings::upsert_env_vars(path, &[("A_KEY", "a")]).unwrap();

        assert_eq!(read_json(path)["env"]["A_KEY"], "a");
        fs::remove_file(path).unwrap();
    }

    #[test]
    fn remove_env_vars_false_when_keys_absent_leaves_file_untouched() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        let original = r#"{"env": {"OTHER_KEY": "keep"}}"#;
        fs::write(&path, original).unwrap();

        let removed = Settings::remove_env_vars(&path, &["A_KEY"]).unwrap();
        assert!(!removed);
        // Not rewritten: the raw bytes are exactly as written.
        assert_eq!(fs::read_to_string(&path).unwrap(), original);
    }

    // ── profile-targeted env writes (issue #1116) ────────────────────

    #[test]
    fn upsert_env_vars_in_profile_creates_profile_env() {
        let (_tmp, path) = temp_settings_path();

        Settings::upsert_env_vars_in(&path, Some("work"), &[("A_KEY", "a")]).unwrap();

        let val = read_json(&path);
        assert_eq!(val["profiles"]["work"]["env"]["A_KEY"], "a");
        // The base env map is not touched (read-side isolation mirrored).
        assert!(val.get("env").is_none());

        // Credential store hardening (issue #1128) applies to profile
        // writes too: dir 0700, file 0600.
        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt;
            let dir_mode = fs::metadata(path.parent().unwrap())
                .unwrap()
                .permissions()
                .mode();
            assert_eq!(dir_mode & 0o777, 0o700);
            let file_mode = fs::metadata(&path).unwrap().permissions().mode();
            assert_eq!(file_mode & 0o777, 0o600);
        }
    }

    #[test]
    fn upsert_env_vars_in_profile_preserves_base_and_other_profiles() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{
                "env": {"SHARED": "base"},
                "profiles": {
                    "work": {"env": {"OLD": "keep"}},
                    "home": {"env": {"SHARED": "home"}}
                },
                "extra": true
            }"#,
        )
        .unwrap();

        Settings::upsert_env_vars_in(&path, Some("work"), &[("A_KEY", "a")]).unwrap();

        let val = read_json(&path);
        assert_eq!(val["profiles"]["work"]["env"]["A_KEY"], "a");
        assert_eq!(val["profiles"]["work"]["env"]["OLD"], "keep");
        assert_eq!(val["profiles"]["home"]["env"]["SHARED"], "home");
        assert_eq!(val["env"]["SHARED"], "base");
        assert_eq!(val["extra"], true);
    }

    #[test]
    fn upsert_env_vars_in_profile_replaces_non_object_nodes() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();

        // "profiles" itself is not an object.
        fs::write(&path, r#"{"profiles": "bogus"}"#).unwrap();
        Settings::upsert_env_vars_in(&path, Some("work"), &[("A_KEY", "a")]).unwrap();
        assert_eq!(read_json(&path)["profiles"]["work"]["env"]["A_KEY"], "a");

        // The profile node is not an object.
        fs::write(&path, r#"{"profiles": {"work": []}}"#).unwrap();
        Settings::upsert_env_vars_in(&path, Some("work"), &[("A_KEY", "a")]).unwrap();
        assert_eq!(read_json(&path)["profiles"]["work"]["env"]["A_KEY"], "a");
    }

    #[test]
    fn remove_env_vars_in_profile_removes_only_profile_keys() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{
                "env": {"A_KEY": "base"},
                "profiles": {"work": {"env": {"A_KEY": "work", "OTHER": "keep"}}}
            }"#,
        )
        .unwrap();

        let removed = Settings::remove_env_vars_in(&path, Some("work"), &["A_KEY"]).unwrap();
        assert!(removed);

        let val = read_json(&path);
        assert!(val["profiles"]["work"]["env"].get("A_KEY").is_none());
        assert_eq!(val["profiles"]["work"]["env"]["OTHER"], "keep");
        // The base copy of the same key survives.
        assert_eq!(val["env"]["A_KEY"], "base");
    }

    #[test]
    fn remove_env_vars_in_profile_false_when_profile_missing() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        let original = r#"{"env": {"A_KEY": "base"}}"#;
        fs::write(&path, original).unwrap();

        let removed = Settings::remove_env_vars_in(&path, Some("work"), &["A_KEY"]).unwrap();
        assert!(!removed);
        // Not rewritten: the raw bytes are exactly as written.
        assert_eq!(fs::read_to_string(&path).unwrap(), original);
    }

    #[test]
    fn remove_env_vars_in_none_targets_base_env() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"env": {"A_KEY": "base"}, "profiles": {"work": {"env": {"A_KEY": "work"}}}}"#,
        )
        .unwrap();

        let removed = Settings::remove_env_vars_in(&path, None, &["A_KEY"]).unwrap();
        assert!(removed);

        let val = read_json(&path);
        assert!(val["env"].get("A_KEY").is_none());
        assert_eq!(val["profiles"]["work"]["env"]["A_KEY"], "work");
    }

    // ── generalized path-segment walkers (issue #1500) ────────────────

    #[test]
    fn ensure_object_at_creates_nested_path_at_arbitrary_depth() {
        let mut root = serde_json::json!({});
        {
            let map = ensure_object_at(&mut root, &["gmail", "accounts", "work"]).unwrap();
            map.insert("client_id".to_string(), serde_json::json!("id"));
        }
        assert_eq!(root["gmail"]["accounts"]["work"]["client_id"], "id");
    }

    #[test]
    fn ensure_object_at_replaces_non_object_nodes_along_path() {
        let mut root = serde_json::json!({"gmail": "bogus"});
        {
            let map = ensure_object_at(&mut root, &["gmail", "accounts", "work"]).unwrap();
            map.insert("client_id".to_string(), serde_json::json!("id"));
        }
        assert_eq!(root["gmail"]["accounts"]["work"]["client_id"], "id");
    }

    #[test]
    fn object_at_mut_none_when_any_segment_absent() {
        let mut root = serde_json::json!({"gmail": {"accounts": {}}});
        assert!(object_at_mut(&mut root, &["gmail", "accounts", "work"]).is_none());
        assert!(object_at_mut(&mut root, &["missing", "accounts"]).is_none());
    }

    // ── gmail account writes (issue #1500) ─────────────────────────────

    #[test]
    fn upsert_gmail_account_creates_nested_path_and_preserves_siblings() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"env": {"SHARED": "base"}, "gmail": {"accounts": {"personal": {"client_id": "keep"}}}, "extra": true}"#,
        )
        .unwrap();

        Settings::upsert_gmail_account(
            &path,
            "work",
            &[
                ("client_id", serde_json::Value::String("id".to_string())),
                (
                    "refresh_token",
                    serde_json::Value::String("token".to_string()),
                ),
            ],
        )
        .unwrap();

        let val = read_json(&path);
        assert_eq!(val["gmail"]["accounts"]["work"]["client_id"], "id");
        assert_eq!(val["gmail"]["accounts"]["work"]["refresh_token"], "token");
        assert_eq!(val["gmail"]["accounts"]["personal"]["client_id"], "keep");
        assert_eq!(val["env"]["SHARED"], "base");
        assert_eq!(val["extra"], true);

        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt;
            let file_mode = fs::metadata(&path).unwrap().permissions().mode();
            assert_eq!(file_mode & 0o777, 0o600);
        }
    }

    /// Regression test for the PR #1528 review comment: writing a non-string
    /// field (e.g. `chrome_profile_from_email: bool`) through
    /// `upsert_gmail_account` must round-trip through `Settings::load()` —
    /// before the `vars` type was widened to `serde_json::Value`, this
    /// silently wrote the JSON string `"true"` into a `bool` field and broke
    /// parsing of the entire settings file on next load.
    #[test]
    fn upsert_gmail_account_writes_a_bool_value_that_round_trips_through_settings_load() {
        let (_tmp, path) = temp_settings_path();

        Settings::upsert_gmail_account(
            &path,
            "work",
            &[("chrome_profile_from_email", serde_json::Value::Bool(true))],
        )
        .unwrap();

        let val = read_json(&path);
        assert_eq!(
            val["gmail"]["accounts"]["work"]["chrome_profile_from_email"],
            true
        );

        let settings = Settings::load_from_path(&path).unwrap();
        assert!(
            settings.gmail.accounts["work"].chrome_profile_from_email,
            "the bool field must deserialize back to `true`, not the string \"true\""
        );
    }

    #[test]
    fn remove_gmail_account_true_when_present_false_when_absent() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"gmail": {"accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_gmail_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert!(val["gmail"]["accounts"].get("work").is_none());
        assert_eq!(val["gmail"]["accounts"]["personal"]["client_id"], "keep");

        assert!(!Settings::remove_gmail_account(&path, "work").unwrap());
    }

    #[test]
    fn remove_gmail_account_clears_default_account_when_it_named_the_removed_account() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"gmail": {"default_account": "work", "accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_gmail_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert!(val["gmail"].get("default_account").is_none());
        assert_eq!(val["gmail"]["accounts"]["personal"]["client_id"], "keep");
    }

    #[test]
    fn remove_gmail_account_leaves_default_account_untouched_when_it_names_a_different_account() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"gmail": {"default_account": "personal", "accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_gmail_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert_eq!(val["gmail"]["default_account"], "personal");
    }

    #[test]
    fn remove_gmail_account_false_when_file_missing() {
        let (_tmp, path) = temp_settings_path();
        assert!(!Settings::remove_gmail_account(&path, "work").unwrap());
        assert!(!path.exists());
    }

    #[test]
    fn set_gmail_default_account_sets_and_clears() {
        let (_tmp, path) = temp_settings_path();

        Settings::set_gmail_default_account(&path, Some("work")).unwrap();
        assert_eq!(read_json(&path)["gmail"]["default_account"], "work");

        Settings::set_gmail_default_account(&path, None).unwrap();
        assert!(read_json(&path)["gmail"].get("default_account").is_none());
    }

    // ── drive account writes (issue #1522) ─────────────────────────────

    #[test]
    fn upsert_drive_account_creates_nested_path_and_preserves_siblings() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"env": {"SHARED": "base"}, "drive": {"accounts": {"personal": {"client_id": "keep"}}}, "extra": true}"#,
        )
        .unwrap();

        Settings::upsert_drive_account(
            &path,
            "work",
            &[
                ("client_id", serde_json::Value::String("id".to_string())),
                (
                    "refresh_token",
                    serde_json::Value::String("token".to_string()),
                ),
            ],
        )
        .unwrap();

        let val = read_json(&path);
        assert_eq!(val["drive"]["accounts"]["work"]["client_id"], "id");
        assert_eq!(val["drive"]["accounts"]["work"]["refresh_token"], "token");
        assert_eq!(val["drive"]["accounts"]["personal"]["client_id"], "keep");
        assert_eq!(val["env"]["SHARED"], "base");
        assert_eq!(val["extra"], true);

        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt;
            let file_mode = fs::metadata(&path).unwrap().permissions().mode();
            assert_eq!(file_mode & 0o777, 0o600);
        }
    }

    /// Drive's twin of
    /// `upsert_gmail_account_writes_a_bool_value_that_round_trips_through_settings_load`
    /// — see that test's doc comment for the PR #1528 review comment this
    /// closes.
    #[test]
    fn upsert_drive_account_writes_a_bool_value_that_round_trips_through_settings_load() {
        let (_tmp, path) = temp_settings_path();

        Settings::upsert_drive_account(
            &path,
            "work",
            &[("chrome_profile_from_email", serde_json::Value::Bool(true))],
        )
        .unwrap();

        let val = read_json(&path);
        assert_eq!(
            val["drive"]["accounts"]["work"]["chrome_profile_from_email"],
            true
        );

        let settings = Settings::load_from_path(&path).unwrap();
        assert!(
            settings.drive.accounts["work"].chrome_profile_from_email,
            "the bool field must deserialize back to `true`, not the string \"true\""
        );
    }

    #[test]
    fn remove_drive_account_true_when_present_false_when_absent() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"drive": {"accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_drive_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert!(val["drive"]["accounts"].get("work").is_none());
        assert_eq!(val["drive"]["accounts"]["personal"]["client_id"], "keep");

        assert!(!Settings::remove_drive_account(&path, "work").unwrap());
    }

    #[test]
    fn remove_drive_account_clears_default_account_when_it_named_the_removed_account() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"drive": {"default_account": "work", "accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_drive_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert!(val["drive"].get("default_account").is_none());
        assert_eq!(val["drive"]["accounts"]["personal"]["client_id"], "keep");
    }

    #[test]
    fn remove_drive_account_leaves_default_account_untouched_when_it_names_a_different_account() {
        let (_tmp, path) = temp_settings_path();
        fs::create_dir_all(path.parent().unwrap()).unwrap();
        fs::write(
            &path,
            r#"{"drive": {"default_account": "personal", "accounts": {"work": {"client_id": "id"}, "personal": {"client_id": "keep"}}}}"#,
        )
        .unwrap();

        assert!(Settings::remove_drive_account(&path, "work").unwrap());
        let val = read_json(&path);
        assert_eq!(val["drive"]["default_account"], "personal");
    }

    #[test]
    fn remove_drive_account_false_when_file_missing() {
        let (_tmp, path) = temp_settings_path();
        assert!(!Settings::remove_drive_account(&path, "work").unwrap());
        assert!(!path.exists());
    }

    #[test]
    fn set_drive_default_account_sets_and_clears() {
        let (_tmp, path) = temp_settings_path();

        Settings::set_drive_default_account(&path, Some("work")).unwrap();
        assert_eq!(read_json(&path)["drive"]["default_account"], "work");

        Settings::set_drive_default_account(&path, None).unwrap();
        assert!(read_json(&path)["drive"].get("default_account").is_none());
    }
}