buzz/crates/buzz-agent/src/auth.rs

//! Token sources for the LLM transport layer.
//!
//! [`TokenSource`] decouples request auth from `Config::api_key`: providers
//! can supply a static string ([`StaticTokenSource`]) or a refreshable OAuth
//! 2.0 PKCE engine ([`PkceOAuthTokenSource`]). Engines own their own cache
//! and refresh logic; the [`Llm`] just asks for a bearer per request.
//!
//! The PKCE engine implements RFC 6749 + RFC 7636 with on-disk token
//! caching keyed by `sha256(discovery_url|client_id|scopes)`. It's the
//! same shape goose uses for Databricks, but we own the wire format and
//! cache directory so the two are independently upgradable.
//!
//! First-use (cache empty) requires a browser: the engine opens
//! `authorization_endpoint` in `webbrowser`, listens on `127.0.0.1:0`,
//! captures the redirect, and exchanges the code for a token. Subsequent
//! calls hit the cache and silently refresh when expired.

use std::collections::HashMap;
use std::fs;
use std::io::{self, Write};
use std::path::{Path, PathBuf};
use std::sync::{Arc, LazyLock};
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use async_trait::async_trait;
use base64::Engine;
use fs2::FileExt;
use reqwest::Client;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use sha2::Digest;
use tokio::sync::{watch, Mutex};

use crate::{auth_http::read_oauth_json, databricks::StrictWorkspace, types::AgentError};

/// Buffer before `expires_at` to consider a cached token "still good".
/// Keeps us off the cliff if the clock or the server's clock drifts.
const TOKEN_REFRESH_LEEWAY: Duration = Duration::from_secs(60);

/// Wall-clock budget for the interactive browser dance. Goose uses 60s.
/// We match: any longer and the user has gone to lunch.
const BROWSER_AUTH_TIMEOUT: Duration = Duration::from_secs(60);

/// Per-request network timeout for every OAuth HTTP call (discovery, refresh
/// grant, code exchange). Without this, a hung provider connection would stall
/// the caller — and, worse, stall every same-key caller waiting on the
/// cross-process lock this holder owns.
const HTTP_REQUEST_TIMEOUT: Duration = Duration::from_secs(30);

/// Longest an in-flight auth attempt can legitimately run: cold discovery
/// (`30s`) + browser wait (`60s`) + code exchange (`30s`), plus a failed
/// refresh (`30s`) ahead of the browser. Rounded to `150s`. A waiter derives
/// its lock-wait bound from this so it never times out ahead of a healthy
/// holder.
const AUTH_ATTEMPT_DEADLINE: Duration = Duration::from_secs(150);

/// How long a same-key caller waits to acquire the cross-process lock before
/// giving up with [`AuthError::LockTimeout`]. Deliberately longer than
/// [`AUTH_ATTEMPT_DEADLINE`] so a waiter outlasts any legitimate holder rather
/// than timing out mid-flow.
const LOCK_WAIT_TIMEOUT: Duration = Duration::from_secs(165);

/// Poll interval for deadline-aware lock acquisition. `try_lock` is
/// non-blocking, so we sleep between attempts rather than blocking a worker.
const LOCK_POLL_INTERVAL: Duration = Duration::from_millis(100);

/// How long a failed interactive (browser) attempt suppresses automatic
/// re-launch for the same key. Long enough that a spurned dropdown does not
/// re-pop a browser on the next debounced refresh, short enough that a user
/// who fixes the problem is not locked out.
const COOLDOWN_DURATION: Duration = Duration::from_secs(300);

/// Why an auth acquisition wants a token, which decides whether it may open a
/// browser and whether it honors a cooldown.
///
/// - [`Auto`](Self::Auto): passive Desktop discovery (create/edit/defaults/
///   onboarding). May open a browser, but honors an unexpired cooldown and
///   returns its recorded outcome instead of re-launching.
/// - [`UserInitiated`](Self::UserInitiated): an explicit human action — the
///   saved-agent model picker or `buzz-agent auth databricks`. May open a
///   browser and *bypasses* the cooldown (the user asked for it now).
/// - [`Headless`](Self::Headless): managed-runtime inference and provider
///   preflight. Never opens a browser; may consume another attempt's cached
///   success but never becomes the initiator.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum AuthIntent {
    Auto,
    UserInitiated,
    Headless,
}

impl AuthIntent {
    /// `true` for the intents permitted to open a browser.
    fn may_open_browser(self) -> bool {
        matches!(self, Self::Auto | Self::UserInitiated)
    }

    /// `true` for the one intent that honors a recorded cooldown on read.
    fn honors_cooldown(self) -> bool {
        matches!(self, Self::Auto)
    }

    /// Stable discriminant for the cross-process attempt sidecar. A queued
    /// caller adopts a completed attempt's failure only when the recorded
    /// intent matches its own — the durable mirror of the in-process
    /// [`INFLIGHT`] registry's `(path, intent)` keying, so a `UserInitiated`
    /// caller never inherits an `Auto` attempt's suppressed result across
    /// processes any more than it does within one.
    fn as_str(self) -> &'static str {
        match self {
            Self::Auto => "auto",
            Self::UserInitiated => "user_initiated",
            Self::Headless => "headless",
        }
    }
}

/// Typed result of an auth acquisition. `Ok` carries the bearer; the error
/// arm classifies *why* no token was produced so callers (and, in Phase 2, the
/// Tauri boundary) can branch on a stable code instead of matching display
/// text.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum AuthError {
    /// No cached token, no refresh grant, and the caller may not open a
    /// browser (`Headless`).
    NoCredential,
    /// The user (or provider) rejected the browser authorization.
    Denied,
    /// The browser flow was not completed within [`BROWSER_AUTH_TIMEOUT`].
    TimedOut,
    /// Every browser-launch strategy failed, so the flow never started.
    BrowserOpenFailed,
    /// An OAuth network call (discovery/refresh/exchange) could not reach the
    /// provider or timed out.
    NetworkUnavailable,
    /// A refresh-token grant was rejected (dead/rotated refresh token) and the
    /// caller may not fall back to a browser.
    RefreshRejected,
    /// The authorization-code exchange itself was rejected by the token
    /// endpoint (distinct from a refresh rejection).
    ExchangeFailed,
    /// Could not acquire the cross-process auth lock within
    /// [`LOCK_WAIT_TIMEOUT`].
    LockTimeout,
}

impl AuthError {
    /// Stable machine-readable code. Phase 2 serializes this across the Tauri
    /// boundary (the `project_git_merge_error` `{code, message}` precedent) so
    /// the Desktop formatter switches on the code, never on display text.
    pub fn code(&self) -> &'static str {
        match self {
            Self::NoCredential => "no_credential",
            Self::Denied => "denied",
            Self::TimedOut => "timed_out",
            Self::BrowserOpenFailed => "browser_open_failed",
            Self::NetworkUnavailable => "network_unavailable",
            Self::RefreshRejected => "refresh_rejected",
            Self::ExchangeFailed => "exchange_failed",
            Self::LockTimeout => "lock_timeout",
        }
    }

    /// `true` for the browser-attempt outcomes worth recording in the cooldown
    /// sidecar — the failures that would otherwise re-pop a browser on the
    /// next automatic attempt. Non-browser failures (no credential, refresh
    /// rejection, lock timeout, network) are not recorded.
    fn is_cooldown_worthy(&self) -> bool {
        matches!(
            self,
            Self::Denied | Self::TimedOut | Self::BrowserOpenFailed | Self::ExchangeFailed
        )
    }

    /// Reconstruct a recorded outcome from its [`code`](Self::code). The
    /// cooldown-worthy variants always round-trip; `RefreshRejected` and
    /// `NoCredential` are also reconstructed for the cross-process attempt
    /// adoption path. Any other code (a forward-compat sidecar written by a
    /// newer buzz-agent) yields `None`, treated as "no active record" rather
    /// than a hard failure.
    fn from_code(code: &str) -> Option<Self> {
        match code {
            "denied" => Some(Self::Denied),
            "timed_out" => Some(Self::TimedOut),
            "browser_open_failed" => Some(Self::BrowserOpenFailed),
            "exchange_failed" => Some(Self::ExchangeFailed),
            "refresh_rejected" => Some(Self::RefreshRejected),
            "no_credential" => Some(Self::NoCredential),
            _ => None,
        }
    }

    fn message(&self) -> String {
        match self {
            Self::NoCredential => {
                "no cached Databricks token; run `buzz-agent auth databricks` first".into()
            }
            Self::Denied => "Databricks authorization was denied".into(),
            Self::TimedOut => "Databricks browser authorization timed out".into(),
            Self::BrowserOpenFailed => "could not open a browser for Databricks sign-in".into(),
            Self::NetworkUnavailable => "could not reach Databricks to authenticate".into(),
            Self::RefreshRejected => "Databricks rejected the refresh token; sign in again".into(),
            Self::ExchangeFailed => "Databricks rejected the authorization code".into(),
            Self::LockTimeout => "timed out waiting for a concurrent Databricks sign-in".into(),
        }
    }
}

impl From<AuthError> for AgentError {
    /// Map a typed auth failure onto the crate error the [`TokenSource`] trait
    /// returns. Auth-decision failures become [`AgentError::LlmAuth`] so the
    /// caller's retry loop stops instead of hammering a rejected credential;
    /// purely infrastructural failures (network, lock contention) become
    /// [`AgentError::Llm`], matching the pre-coordinator classification of a
    /// discovery/network error.
    fn from(e: AuthError) -> Self {
        match e {
            AuthError::NetworkUnavailable | AuthError::LockTimeout => AgentError::Llm(e.message()),
            AuthError::NoCredential
            | AuthError::Denied
            | AuthError::TimedOut
            | AuthError::BrowserOpenFailed
            | AuthError::RefreshRejected
            | AuthError::ExchangeFailed => AgentError::LlmAuth(e.message()),
        }
    }
}

/// Opens a URL for the interactive browser step. Injected so the PKCE
/// continuation (callback listener, verifier, timeout) stays alive across the
/// launch: the coordinator calls this *while* the localhost listener is
/// bound, so a launch failure never leaves a returned URL pointing at a torn
/// down listener. Desktop (Phase 2) supplies the Tauri opener; the CLI uses
/// [`DefaultBrowserOpener`], which prints the URL and opens the system
/// browser.
pub trait BrowserOpener: Send + Sync {
    /// Attempt to present `url` to the user. Returning `Err` means every
    /// launch strategy for this opener failed; the coordinator then reports
    /// [`AuthError::BrowserOpenFailed`] without waiting on a listener nobody
    /// will reach.
    fn open(&self, url: &str) -> Result<(), String>;
}

/// Default opener: print the URL (so a user on a headless box can copy it)
/// and open the system browser. Printing is itself a launch strategy, so this
/// never reports failure — the URL is always visible to the waiting user.
pub struct DefaultBrowserOpener;

impl BrowserOpener for DefaultBrowserOpener {
    fn open(&self, url: &str) -> Result<(), String> {
        eprintln!("Opening browser for authentication. If it doesn't open, visit:\n  {url}");
        let _ = webbrowser::open(url);
        Ok(())
    }
}

/// Asynchronous source of a bearer token. The [`Llm`] calls this per
/// request, so impls are expected to be cheap on the cache-hit path.
#[async_trait]
pub trait TokenSource: Send + Sync {
    async fn bearer(&self) -> Result<String, AgentError>;

    /// Return a bearer token from cache or refresh, **never** opening a browser.
    ///
    /// The default delegates to [`bearer`](Self::bearer) — correct for token
    /// sources (e.g. static API keys) that can never trigger a browser flow.
    /// [`PkceOAuthTokenSource`] overrides this to stop before the browser step.
    async fn bearer_no_browser(&self) -> Result<String, AgentError> {
        self.bearer().await
    }

    /// Force a fresh bearer after the server rejected the current one (401).
    ///
    /// `rejected` is the exact access token that just got the 401. Unlike
    /// [`bearer`](Self::bearer), which trusts the local expiry clock, this is
    /// driven by the server's verdict: the cached token looked valid to us
    /// (well within its local expiry) but the provider rejected it — clock
    /// skew, server-side revocation, or a node that never saw it. The clock
    /// therefore can't decide whether to refresh; the caller passes the
    /// rejected token so the impl can refresh unless a concurrent caller has
    /// *already* replaced it. Implementations must obtain a new token without
    /// any interactive step, so a headless harness never hangs. The default
    /// returns the existing bearer — correct for sources that can't refresh
    /// (a static key); the caller's retry then fails terminally rather than
    /// looping.
    async fn refresh_now(&self, _rejected: &str) -> Result<String, AgentError> {
        self.bearer().await
    }
}

/// A token that never changes for the life of the process.
pub struct StaticTokenSource(String);

impl StaticTokenSource {
    pub fn new(token: impl Into<String>) -> Self {
        Self(token.into())
    }
}

#[async_trait]
impl TokenSource for StaticTokenSource {
    async fn bearer(&self) -> Result<String, AgentError> {
        Ok(self.0.clone())
    }
}

/// Static config for an OAuth 2.0 Authorization Code + PKCE provider.
///
/// The `discovery_url` must return a JSON document with at least
/// `authorization_endpoint` and `token_endpoint` (RFC 8414). The
/// `cache_namespace` is the directory under the platform config directory's
/// `buzz-agent/oauth/` root where the token JSON lives — separates providers'
/// caches cleanly.
#[derive(Debug, Clone)]
pub struct PkceOAuthConfig {
    pub discovery_url: String,
    pub client_id: String,
    pub scopes: Vec<String>,
    pub cache_namespace: String,
    /// When `Some`, the engine writes tokens here instead of
    /// `<platform config dir>/buzz-agent/oauth/<cache_namespace>/`. Production code
    /// leaves this `None`. Integration tests use it to avoid stomping on
    /// a shared `$HOME` when running in parallel.
    pub cache_dir_override: Option<PathBuf>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
struct CachedToken {
    access_token: String,
    refresh_token: Option<String>,
    /// Unix seconds. `None` means the server didn't advertise an expiry;
    /// we use it without checking and rely on refresh on 401.
    expires_at: Option<u64>,
}

#[derive(Debug, Clone)]
struct OidcEndpoints {
    authorization_endpoint: String,
    token_endpoint: String,
}

/// Typed result of a refresh-token grant, so the coordinator can separate an
/// actual credential rejection from a transient fault.
///
/// - [`Refreshed`](Self::Refreshed): a fresh token — success.
/// - [`Rejected`](Self::Rejected): the token endpoint returned an
///   `invalid_grant` error (dead/rotated refresh token). This is the only
///   outcome that becomes [`AuthError::RefreshRejected`] for `Headless` or
///   drives a browser fallback for interactive intents.
/// - [`Network`](Self::Network): transport error, timeout, 5xx, any 4xx that
///   is not `invalid_grant` (e.g. `invalid_request`, `invalid_client`, 429),
///   an unparseable error body, or an undecodable/malformed success body —
///   infrastructural or misconfiguration, never a credential decision, so it
///   surfaces as [`AuthError::NetworkUnavailable`] and never pops a browser.
enum RefreshOutcome {
    Refreshed(CachedToken),
    Rejected,
    Network,
}

/// PKCE OAuth token source with on-disk refresh cache.
///
/// First call:
///   1. Loads from cache if present and unexpired.
///   2. Otherwise tries `refresh_token` if cached.
///   3. Otherwise runs the full browser flow.
///
/// Subsequent calls hit an in-memory copy of the cached token and only
/// touch disk/network if the access token is past `expires_at`.
pub struct PkceOAuthTokenSource {
    cfg: PkceOAuthConfig,
    http: Client,
    workspace: Option<StrictWorkspace>,
    cache_path: PathBuf,
    /// Injected browser launcher, called inside [`browser_pkce_flow`] while the
    /// localhost listener is live. Production uses [`DefaultBrowserOpener`];
    /// Phase 2 supplies the Tauri opener.
    opener: Arc<dyn BrowserOpener>,
    /// In-memory single-flight *and* fast-path cache. The cross-process file
    /// lock serializes slow-path work; this cell keeps the fast path off disk
    /// during a turn and off the lock entirely.
    state: Mutex<Option<CachedToken>>,
}

impl PkceOAuthTokenSource {
    /// Construct with the default browser opener (prints the URL and opens the
    /// system browser). This is the signature every production call site uses.
    pub fn new(cfg: PkceOAuthConfig) -> Result<Arc<Self>, AgentError> {
        Self::new_with(cfg, Arc::new(DefaultBrowserOpener))
    }

    /// Construct with an injected [`BrowserOpener`]. Tests substitute a
    /// recording/failing opener to exercise the browser branch without a real
    /// window; Phase 2 Desktop injects the Tauri opener.
    pub fn new_with(
        cfg: PkceOAuthConfig,
        opener: Arc<dyn BrowserOpener>,
    ) -> Result<Arc<Self>, AgentError> {
        Self::new_with_http_timeout(cfg, opener, HTTP_REQUEST_TIMEOUT)
    }

    /// Construct with an injected opener *and* an explicit per-request HTTP
    /// timeout. Only the refresh-timeout integration test passes the timeout
    /// argument: it drives a hung token endpoint against a short bound so the
    /// per-request timeout classification (`NetworkUnavailable`, never
    /// `RefreshRejected`) is exercised in real time. A paused-clock test can't
    /// do this — tokio auto-advances into the timer while the real loopback
    /// discovery call is still in flight, tripping the timeout on the wrong
    /// request. Every production and other-test path goes through
    /// [`new`](Self::new) or [`new_with`](Self::new_with) at the default
    /// [`HTTP_REQUEST_TIMEOUT`].
    pub fn new_with_http_timeout(
        cfg: PkceOAuthConfig,
        opener: Arc<dyn BrowserOpener>,
        http_timeout: Duration,
    ) -> Result<Arc<Self>, AgentError> {
        let http = Client::builder()
            .timeout(http_timeout)
            .build()
            .map_err(|_| AgentError::Llm("oauth http client construction failed".into()))?;
        Self::from_parts(cfg, opener, http, None)
    }

    pub(crate) fn for_workspace(
        workspace: StrictWorkspace,
        cache_root: &Path,
        opener: Arc<dyn BrowserOpener>,
        http: Client,
    ) -> Result<Arc<Self>, AgentError> {
        let mut cfg =
            crate::llm::databricks_pkce_config(workspace.as_str(), Some(cache_root.to_path_buf()));
        // Strict and legacy sources must not share single-flight/cache state,
        // even if a caller accidentally supplies the same root to both APIs.
        cfg.cache_namespace = "databricks-strict".into();
        Self::from_parts(cfg, opener, http, Some(workspace))
    }

    fn from_parts(
        cfg: PkceOAuthConfig,
        opener: Arc<dyn BrowserOpener>,
        http: Client,
        workspace: Option<StrictWorkspace>,
    ) -> Result<Arc<Self>, AgentError> {
        let cache_path = cache_path_for(&cfg)?;
        if let Some(parent) = cache_path.parent() {
            fs::create_dir_all(parent)
                .map_err(|e| AgentError::Llm(format!("oauth cache dir {parent:?}: {e}")))?;
        }
        let initial = read_cache(&cache_path);
        Ok(Arc::new(Self {
            cfg,
            http,
            workspace,
            cache_path,
            opener,
            state: Mutex::new(initial),
        }))
    }

    /// Path of the cross-process advisory lock file guarding slow-path auth
    /// for this cache key. Co-located with the cache so it shares the
    /// per-key directory and `$HOME` override.
    fn lock_path(&self) -> PathBuf {
        append_ext(&self.cache_path, "lock")
    }

    /// Path of the cooldown sidecar recording the last browser-attempt
    /// failure for this cache key.
    fn cooldown_path(&self) -> PathBuf {
        append_ext(&self.cache_path, "cooldown")
    }

    /// Path of the attempt sidecar recording the generation and outcome of the
    /// last completed slow-path acquisition for this cache key. Drives the
    /// cross-process single-flight of *failures* (see [`AttemptRecord`]).
    fn attempt_path(&self) -> PathBuf {
        append_ext(&self.cache_path, "attempt")
    }

    /// Discover authorization + token endpoints from the well-known URL.
    async fn endpoints(&self) -> Result<OidcEndpoints, AgentError> {
        let response = self
            .http
            .get(&self.cfg.discovery_url)
            .send()
            .await
            .map_err(|_| AgentError::Llm("oauth discovery request failed".into()))?;
        if !response.status().is_success() {
            return Err(AgentError::Llm("oauth discovery status failed".into()));
        }
        let v = read_oauth_json(response)
            .await
            .map_err(|_| AgentError::Llm("oauth discovery response invalid or too large".into()))?;
        let auth = v
            .get("authorization_endpoint")
            .and_then(Value::as_str)
            .ok_or_else(|| {
                AgentError::Llm("oauth discovery: authorization_endpoint missing".into())
            })?
            .to_string();
        let token = v
            .get("token_endpoint")
            .and_then(Value::as_str)
            .ok_or_else(|| AgentError::Llm("oauth discovery: token_endpoint missing".into()))?
            .to_string();
        if let Some(workspace) = &self.workspace {
            workspace.validate_endpoint(&auth)?;
            workspace.validate_endpoint(&token)?;
        }
        Ok(OidcEndpoints {
            authorization_endpoint: auth,
            token_endpoint: token,
        })
    }

    /// Persist a token to disk and the in-memory cell.
    ///
    /// The cache holds both the access and refresh tokens, so the on-disk
    /// file is written owner-only (`0o600` on Unix) via an atomic
    /// inode-swapping rename — see [`write_private_cache`].
    ///
    /// On non-Unix platforms the token is stored in-memory only: the
    /// `write_private_cache` path creates files with default ACLs, which do
    /// not enforce owner-only access. Disk persistence is intentionally
    /// disabled until a Windows-specific owner-only DACL is implemented (see
    /// the `create_private_temp_file` non-Unix branch). The cost is that each
    /// process performs its own acquisition on non-Unix — cross-process
    /// *success* handoff requires the shared on-disk cache, so processes
    /// serialize through the lock but the loser repeats the flow rather than
    /// reading the winner's token. Cross-process *failure* adoption still works
    /// because it uses the attempt sidecar (no token bytes). Correct and
    /// safe until owner-only DACL persistence exists.
    fn save(&self, state: &mut Option<CachedToken>, token: CachedToken) -> Result<(), AgentError> {
        self.persist(&token)?;
        *state = Some(token);
        Ok(())
    }

    /// Write `token` to the on-disk cache. Split out of [`save`](Self::save) so
    /// the 401 neutralization path can rewrite the disk layer without clobbering
    /// a distinct in-memory entry. No-op on non-Unix (see [`save`](Self::save)).
    fn persist(&self, token: &CachedToken) -> Result<(), AgentError> {
        #[cfg(unix)]
        {
            let body = serde_json::to_vec_pretty(token)
                .map_err(|e| AgentError::Llm(format!("oauth cache serialize: {e}")))?;
            write_private_cache(&self.cache_path, &body).map_err(|e| {
                AgentError::Llm(format!("oauth cache write {:?}: {e}", self.cache_path))
            })?;
        }
        #[cfg(not(unix))]
        {
            // Disk persistence disabled on non-Unix: owner-only file
            // permissions require a DACL that is not yet implemented.
            let _ = token;
        }
        Ok(())
    }

    /// Neutralize the matching rejected credential in B's own in-memory `state`
    /// only — no disk I/O. The joiner matching-failure path calls this rather
    /// than `expire_rejected`: the leader already ran the durable disk
    /// invalidation under the cross-process file lock, and re-running disk
    /// mutations from the lockless joiner can race with a concurrent process C
    /// that persisted a valid replacement under the same lock (C's rename can
    /// be overwritten by B's unfenced rename).
    ///
    /// Contract: only the access-token identity is checked — the refresh token
    /// is left intact so callers reaching the recovery disk-read path below can
    /// still attempt a fresh token exchange with the un-revoked refresh secret.
    ///
    /// Limitation: the joiner's match arm triggers on a same-digest leader
    /// error regardless of error code (see `acquire`'s `Err` match arm). A
    /// pre-lock failure (e.g. `LockTimeout`) with a matching rejected digest
    /// therefore also reaches this helper, even though the leader never
    /// durably invalidated the disk copy. In that case B's in-memory entry is
    /// neutralized and B returns the shared error; the disk copy survives
    /// intact. A subsequent plain `bearer()` (`rejected = None`) can re-read
    /// the disk entry. This is a known bounded limitation: in-memory
    /// neutralization is applied without a guarantee that the durable copy is
    /// also gone.
    fn expire_rejected_memory(&self, state: &mut Option<CachedToken>, rejected: Option<&str>) {
        let Some(rej) = rejected else { return };
        if let Some(tok) = state.as_mut() {
            if tok.access_token == rej {
                tok.expires_at = Some(0);
            }
        }
    }

    /// Neutralize a cached token the caller just reported 401-rejected.
    ///
    /// A 401 means the cached access token is dead even though its local expiry
    /// clock still looks fresh. [`cached_hit`](Self::cached_hit) and
    /// [`usable_from_disk`](Self::usable_from_disk) already exclude it for a
    /// caller carrying `rejected`, but a *later* plain `bearer()`
    /// (`rejected = None`) trusts the clock and would serve it, and a freshly
    /// constructed source would restore it from disk. Force it expired in both
    /// layers so [`is_expired`] excludes it for every future caller and every
    /// fresh process, while the refresh token — which was *not* rejected and
    /// drives this very recovery — stays intact. Each layer is neutralized only
    /// when its access token byte-equals `rejected`, so a sibling's
    /// concurrently-written distinct replacement is preserved.
    ///
    /// Disk neutralization is a bounded three-stage process: on atomic-rewrite
    /// failure (e.g. non-writable parent directory), the implementation falls
    /// back to an in-place truncating overwrite of the existing file (no
    /// parent-dir perms required), and finally to `remove_file`. If all three
    /// fail the file survives; `cached_hit`'s `rejected`-aware filter protects
    /// this caller's path, but a later plain `bearer()` could re-read the
    /// unexpired file. That residual corner is outside the normal threat model
    /// (owner actively hardening their own cache file to 0400 against their own
    /// process).
    fn expire_rejected(&self, state: &mut Option<CachedToken>, rejected: Option<&str>) {
        let Some(rej) = rejected else { return };
        // Neutralize the in-memory entry: force-expire so `is_expired` excludes
        // it for every subsequent in-process caller, while the refresh token
        // (which was not rejected) stays intact for the recovery below.
        if let Some(tok) = state.as_mut() {
            if tok.access_token == rej {
                tok.expires_at = Some(0);
            }
        }
        // Neutralize the on-disk copy. Prefer atomic rewrite via `persist()`
        // (temp-file + rename, owner-only permissions). If the atomic rewrite
        // fails (e.g. the parent directory denies temp-file creation), fall back
        // to in-place truncating overwrite: `OpenOptions::write().truncate(true)`
        // on the existing file does not require parent-directory write permission,
        // only that the file itself is owner-writable (0600, which our cache files
        // always are). As a last resort, attempt `remove_file`. The two-stage
        // fallback covers the proven hostile case: a 0600 token file under a
        // 0500 parent — the atomic path cannot create the temp file (EACCES), but
        // the in-place write succeeds because the file's own mode permits it.
        // Residual out of threat model: if the owner explicitly chmodded their own
        // cache file to 0400 before this runs, the in-place write also fails and
        // we fall through to `remove_file`; if that too fails, the file survives
        // with `expires_at = 0` still NOT written — `cached_hit`'s
        // `rejected`-aware filter still protects the calling 401-recovery path,
        // but a later plain `bearer()` could re-adopt the file. That corner is
        // not in the normal threat model (a user actively hardening their own
        // cache file against their own process).
        if let Some(mut disk) = read_cache(&self.cache_path) {
            if disk.access_token == rej {
                disk.expires_at = Some(0);
                if self.persist(&disk).is_err() {
                    // Atomic rewrite failed. Try in-place truncating overwrite —
                    // does not need parent-dir write permission, only the file's
                    // own mode.
                    let inplace_ok = serde_json::to_vec_pretty(&disk).ok().is_some_and(|body| {
                        use std::io::Write as _;
                        fs::OpenOptions::new()
                            .write(true)
                            .truncate(true)
                            .open(&self.cache_path)
                            .and_then(|mut f| f.write_all(&body))
                            .is_ok()
                    });
                    if !inplace_ok {
                        let _ = fs::remove_file(&self.cache_path);
                    }
                }
            }
        }
    }

    /// Exchange a refresh token for a fresh access token.
    ///
    /// The outcome is typed so the caller can tell an actual credential
    /// rejection apart from a transient fault. Only a token-endpoint rejection
    /// of the grant itself (a 4xx `invalid_grant`-class response) is a dead
    /// refresh token; a transport failure, timeout, 5xx, or an
    /// undecodable/malformed response is infrastructural and must never be
    /// mistaken for a credential decision (it would otherwise pop a browser or
    /// return `RefreshRejected` when nothing was actually rejected).
    async fn refresh(&self, endpoints: &OidcEndpoints, refresh_token: &str) -> RefreshOutcome {
        let params = [
            ("grant_type", "refresh_token"),
            ("refresh_token", refresh_token),
            ("client_id", &self.cfg.client_id),
        ];
        let resp = match self
            .http
            .post(&endpoints.token_endpoint)
            .form(&params)
            .send()
            .await
        {
            Ok(resp) => resp,
            // Transport error or the per-request timeout elapsed: no verdict
            // from the provider, so this is infrastructural, not a rejection.
            Err(_) => {
                tracing::warn!("oauth refresh transport failure");
                return RefreshOutcome::Network;
            }
        };
        let status = resp.status();
        if !status.is_success() {
            let body = read_oauth_json(resp).await.ok();
            // Per RFC 6749 §5.2 only `error == "invalid_grant"` means the
            // refresh token itself is dead (expired/revoked) — the one failure
            // a browser sign-in can repair. Every other 4xx (`invalid_request`,
            // `invalid_client`, `unsupported_grant_type`, `invalid_scope`, 408,
            // 429, …), an unparseable error body, and all 5xx are
            // infrastructural or misconfiguration: a browser can't fix them, so
            // they stay in the non-credential bucket and surface as
            // `NetworkUnavailable` without ever popping a browser.
            if status.is_client_error()
                && body
                    .and_then(|v| v.get("error").and_then(Value::as_str).map(str::to_owned))
                    .as_deref()
                    == Some("invalid_grant")
            {
                tracing::warn!(status = %status, "oauth refresh grant rejected");
                return RefreshOutcome::Rejected;
            }
            tracing::warn!(status = %status, "oauth refresh not repairable by browser");
            return RefreshOutcome::Network;
        }
        let v: Value = match read_oauth_json(resp).await {
            Ok(v) => v,
            Err(_) => {
                tracing::warn!("oauth refresh response decode failure");
                return RefreshOutcome::Network;
            }
        };
        match token_from_response(&v, Some(refresh_token)) {
            Ok(token) => RefreshOutcome::Refreshed(token),
            Err(e) => {
                tracing::warn!(error = %e, "oauth refresh response missing access_token");
                RefreshOutcome::Network
            }
        }
    }

    /// Run the full browser-mediated Authorization Code + PKCE flow and cache
    /// the result. Routes through the coordinator as a [`UserInitiated`]
    /// acquisition: it may open a browser, bypasses (and clears) any cooldown,
    /// and single-flights with concurrent callers on the cross-process lock. A
    /// still-valid cached token short-circuits to success without re-prompting.
    ///
    /// This is the no-rejected convenience: it trusts the local expiry clock,
    /// so a not-yet-expired cached token is accepted. When the caller already
    /// knows the cached bearer was rejected by the server (a 401), it must use
    /// [`acquire_with_intent`](Self::acquire_with_intent) with `rejected` set
    /// so the stale-but-fresh token can't short-circuit the sign-in.
    ///
    /// [`UserInitiated`]: AuthIntent::UserInitiated
    pub async fn interactive_login(&self) -> Result<(), AgentError> {
        self.acquire(AuthIntent::UserInitiated, None).await?;
        Ok(())
    }

    /// Public entry for passive Desktop discovery and the saved-model picker
    /// (Phase 2): acquire a bearer under an explicit [`AuthIntent`], returning
    /// the typed [`AuthError`] so the caller can branch on a stable `code`
    /// rather than display text. The [`TokenSource`] trait methods wrap this
    /// and flatten the error into [`AgentError`].
    ///
    /// `rejected` carries the exact access token the provider just 401'd, if
    /// any. With `rejected = None` a locally-fresh cached token is a hit (the
    /// normal discovery path). With `rejected = Some(t)` the expiry clock is
    /// untrustworthy — the rejected token looked fresh — so a cached token
    /// equal to `t` is *not* a hit: the acquisition refreshes, and for `Auto`
    /// or `UserInitiated` falls through to a browser when the refresh grant is
    /// dead. This is what lets the saved-picker recovery path say "this
    /// locally-fresh bearer was just rejected — replace it" instead of
    /// re-returning the dead token, which `refresh_now`'s hardcoded
    /// [`Headless`](AuthIntent::Headless) can never escalate to a browser.
    pub async fn acquire_with_intent(
        &self,
        intent: AuthIntent,
        rejected: Option<&str>,
    ) -> Result<String, AuthError> {
        self.acquire(intent, rejected).await
    }

    /// Return a usable cached bearer, applying the identity rule for a
    /// 401-driven acquisition.
    ///
    /// `rejected = None` (normal): a not-yet-expired cached token is a hit.
    /// `rejected = Some(t)`: the expiry clock is untrustworthy — the rejected
    /// token looked locally fresh — so a hit requires the cached token to
    /// *differ* from `t` (a sibling already replaced it) **and** still be
    /// unexpired. Without the expiry check an expired sibling token B could be
    /// returned as A's replacement, skipping the refresh the 401 demanded.
    /// Checks the in-memory cell first, then re-reads disk (a sibling process
    /// may have written a newer token) and adopts it into the cell on a hit.
    fn cached_hit(
        &self,
        state: &mut Option<CachedToken>,
        rejected: Option<&str>,
    ) -> Option<String> {
        let usable =
            |tok: &CachedToken| !is_expired(tok) && rejected != Some(tok.access_token.as_str());
        if let Some(tok) = state.as_ref() {
            if usable(tok) {
                return Some(tok.access_token.clone());
            }
        }
        if let Some(disk) = read_cache(&self.cache_path) {
            if usable(&disk) {
                let bearer = disk.access_token.clone();
                *state = Some(disk);
                return Some(bearer);
            }
        }
        None
    }

    /// Lock-free variant of [`cached_hit`]'s disk branch: read the on-disk
    /// cache and return its bearer if a sibling wrote a usable replacement for
    /// `rejected`. Used by the joiner's shared-failure recheck, where every
    /// waiter wakes at once — taking `self.state` (even with `try_lock`) would
    /// either drop the replacement for `try_lock` losers or serialize the read
    /// behind a new leader holding `state` across its browser flow. The
    /// in-memory memo is intentionally not updated; the next real acquisition
    /// re-reads and adopts under the lock.
    fn usable_from_disk(&self, rejected: Option<&str>) -> Option<String> {
        let disk = read_cache(&self.cache_path)?;
        (!is_expired(&disk) && rejected != Some(disk.access_token.as_str()))
            .then_some(disk.access_token)
    }

    /// Discover OIDC endpoints once per flow, memoizing into `slot` so the
    /// refresh and browser branches share a single discovery call. A discovery
    /// failure (unreachable URL or malformed document) maps to
    /// [`AuthError::NetworkUnavailable`] — the infrastructural bucket, so the
    /// caller's retry loop treats it as transient rather than as an auth
    /// decision.
    async fn discover<'a>(
        &self,
        slot: &'a mut Option<OidcEndpoints>,
    ) -> Result<&'a OidcEndpoints, AuthError> {
        if slot.is_none() {
            let eps = self
                .endpoints()
                .await
                .map_err(|_| AuthError::NetworkUnavailable)?;
            *slot = Some(eps);
        }
        Ok(slot.as_ref().expect("endpoints just populated"))
    }

    /// The single acquisition entry point behind every [`TokenSource`] method.
    ///
    /// `intent` decides browser and cooldown policy; `rejected` (`Some` only on
    /// a 401-driven refresh) switches cache checks from clock-based to
    /// identity-based. The fast path returns a usable cached token without
    /// touching the lock or the network. Otherwise the slow path serializes
    /// every same-key caller — in this process *and* across processes — on the
    /// cross-process advisory lock, so concurrent dialogs coalesce onto one
    /// refresh/browser flow instead of racing browsers.
    async fn acquire(
        &self,
        intent: AuthIntent,
        rejected: Option<&str>,
    ) -> Result<String, AuthError> {
        // Fast path: no lock, no network. `try_lock` rather than `lock().await`
        // so a caller arriving while a leader holds `state` across its browser
        // flow does not block here — it falls through to the in-process
        // registry below and joins the leader instead of waiting out the whole
        // flow and then racing in as a second leader. A cache hit is still
        // served without the file lock; a miss (or contention) coalesces.
        {
            if let Ok(mut state) = self.state.try_lock() {
                if let Some(hit) = self.cached_hit(&mut state, rejected) {
                    return Ok(hit);
                }
            }
        }

        // In-process single-flight (see [`INFLIGHT`]). Keyed by (lock path,
        // intent): callers with the same intent coalesce, so a caller already
        // waiting when the leader's attempt is in flight shares the leader's
        // result instead of taking the lock after it and launching a second
        // browser. Distinct intents key separately: a `Headless` caller never
        // shares a browser-capable slot, and — critically — a `UserInitiated`
        // caller never inherits an `Auto` leader's cooldown-suppressed result,
        // since the two disagree on cooldown and browser policy. Those cases
        // still coordinate through the cross-process file lock.
        let key: InflightKey = (self.lock_path(), intent);
        let (slot, is_leader) = {
            let mut reg = inflight_registry();
            match reg.get(&key) {
                Some(existing) => (existing.clone(), false),
                None => {
                    let slot = Arc::new(InflightSlot::new());
                    reg.insert(key.clone(), slot.clone());
                    (slot, true)
                }
            }
        };
        if !is_leader {
            // Pre-existing joiner: observe the leader's outcome, but do not
            // adopt a result that violates *this* caller's contract. The slot
            // is keyed only by (lock path, intent), so a joiner shares a leader
            // that ran with a *different* `rejected` value — and the leader's
            // result can be wrong for us in two ways:
            //
            //   * It may publish a token equal to THIS caller's `rejected`
            //     bytes — e.g. its cache re-read adopted a sibling write we
            //     just reported 401-rejected. Returning it would retry the
            //     provider with the exact credentials it refused. We instead
            //     run our own acquisition: the slot is evicted before publish
            //     (see [`LeaderGuard::complete`]), so this is a fresh, bounded,
            //     leader-eligible attempt — not a re-join of the dead
            //     generation, and not a loop. Its cache re-read excludes our
            //     `rejected`, and `finish`'s persistence-boundary guard rejects
            //     any refresh- or browser-issued token equal to our `rejected`
            //     with a typed error before caching it — so the rerun never
            //     hands us back our `rejected` on any path.
            //
            //   * It may publish a terminal failure from a *rejection-relative*
            //     cause — e.g. refresh reissued the leader's own `rejected` bytes
            //     and `finish()` returned `RefreshRejected`. That failure is valid
            //     only for the leader's specific rejected token; a joiner with a
            //     *different* `rejected` (or none) should rerun: its refresh may
            //     yield a valid token. The leader publishes its rejected-token
            //     SHA-256 digest so joiners can compare without inspecting the
            //     token bytes directly. A digest mismatch triggers an `acquire_leader`
            //     rerun (the slot is already evicted). A false rerun (non-rejection
            //     failure with digest mismatch) costs one network round-trip and
            //     stays headless — far better than silently adopting a wrong denial.
            //
            //   * It may publish a terminal failure even though a sibling wrote
            //     a valid replacement into the cache while we waited. We
            //     re-check the cache cheaply before adopting the failure — a
            //     lock-free disk read, never a browser or refresh — so a shared
            //     failure can never fan out into an N-way browser storm. The
            //     disk read is lock-free (`usable_from_disk`, not under `state`)
            //     because all waiters wake together and the in-memory memo is
            //     not load-bearing here — the next real acquisition re-reads and
            //     adopts under the lock.
            let (leader_rejected_digest, outcome) = slot.wait().await;
            match outcome {
                Ok(token) if Some(token.access_token.as_str()) != rejected => {
                    // Conditionally reconcile this source's own credential
                    // state so a subsequent plain `bearer()` on this source
                    // returns the newly-acquired token rather than a stale or
                    // absent credential. Adopt when B's state is absent,
                    // expired, or still pointing at B's own rejected token.
                    // Preserve a distinct newer usable credential — if another
                    // task independently installed a valid token into B's state
                    // between B joining and B waking, that token is better than
                    // the shared result and must not be overwritten.
                    //
                    // `lock().await` rather than `try_lock`: the reconciliation
                    // must complete before returning. The joiner holds neither
                    // the INFLIGHT registry mutex nor the cross-process file
                    // lock at this point, so awaiting `state` cannot deadlock
                    // and skipping the write would leave stale or empty state,
                    // recreating the original P1 regression on the next plain
                    // `bearer()` call.
                    {
                        let mut state = self.state.lock().await;
                        let adopt = state.as_ref().is_none_or(|cur| {
                            is_expired(cur) || rejected.is_some_and(|rej| cur.access_token == rej)
                        });
                        if adopt {
                            *state = Some(token.clone());
                        }
                    }
                    return Ok(token.access_token);
                }
                Ok(_) => {
                    return self
                        .acquire_leader(intent, rejected)
                        .await
                        .map(|t| t.access_token);
                }
                Err(shared) => {
                    // Reject-digest mismatch: the leader's failure was
                    // rejection-relative to ITS OWN `rejected` token, not ours.
                    // Rerun so we can pursue our own refresh/browser path.
                    if leader_rejected_digest != digest_of(rejected) {
                        return self
                            .acquire_leader(intent, rejected)
                            .await
                            .map(|t| t.access_token);
                    }
                    // Neutralize B's matching rejected in-memory state so a
                    // subsequent plain `bearer()` on this source does not
                    // resurface the rejected credential.
                    //
                    // `lock().await` rather than `try_lock`: expiry must
                    // complete before returning. The joiner holds neither the
                    // INFLIGHT registry mutex nor the cross-process file lock
                    // here, so awaiting `state` cannot deadlock. Skipping the
                    // expiry would leave matching rejected X live, recreating
                    // the original P1 regression on the next plain `bearer()`.
                    //
                    // In-memory only (`expire_rejected_memory`, not
                    // `expire_rejected`): the leader already ran the durable
                    // disk invalidation under the file lock. Re-running disk
                    // writes here is lockless — process C may have persisted a
                    // valid replacement under the same lock between A's failure
                    // and this rename, and B's unfenced rename would overwrite
                    // it. Note: a subsequent plain `bearer()` (`rejected=None`)
                    // calls `cached_hit` before the cross-process lock and can
                    // therefore re-read the disk copy without acquiring the lock.
                    {
                        let mut state = self.state.lock().await;
                        self.expire_rejected_memory(&mut state, rejected);
                    }
                    if let Some(hit) = self.usable_from_disk(rejected) {
                        return Ok(hit);
                    }
                    return Err(shared);
                }
            }
        }

        // Leader: run the real flow, then evict + publish. The guard makes
        // eviction and joiner wake-up happen even if this future is cancelled
        // or panics, so a dropped leader can never wedge its joiners or leave a
        // dead slot that turns later callers into joiners of nothing.
        let guard = LeaderGuard::new(key, slot);
        let result = self.acquire_leader(intent, rejected).await;
        guard.complete(result, digest_of(rejected))
    }

    /// The leader's slow-path body: take the cross-process lock, then run the
    /// bounded acquisition under it. Split out so [`acquire`] can wrap it in
    /// the in-process single-flight without the lock/deadline logic bleeding
    /// into the joiner path.
    async fn acquire_leader(
        &self,
        intent: AuthIntent,
        rejected: Option<&str>,
    ) -> Result<CachedToken, AuthError> {
        // Snapshot the current attempt generation *before* queueing on the
        // lock. When we acquire the lock, we compare: if the generation
        // advanced, a predecessor completed while we were waiting and we can
        // adopt its outcome instead of re-running the full flow.
        let attempt_path = self.attempt_path();
        let snapshot_gen = read_attempt(&attempt_path)
            .map(|r| r.generation)
            .unwrap_or(0);
        // Observability hook: cross-process tests install a tracing layer that
        // watches for this event to establish deterministic ordering — it fires
        // after the snapshot is taken and before the process queues on the lock.
        tracing::trace!(
            target: "buzz_agent::auth::acquire_leader_snapshot",
            snapshot_gen,
            "snapshot taken"
        );

        // Slow path: one flow at a time per cache key. The waiter's deadline
        // exceeds a healthy holder's attempt deadline, so it never gives up on
        // a live holder.
        let deadline = std::time::Instant::now() + LOCK_WAIT_TIMEOUT;
        let _guard = acquire_auth_lock(&self.lock_path(), deadline).await?;

        // Bound the whole locked attempt so a wedged flow can't hold the lock
        // past the waiters' patience. The deadline is passed *into*
        // `acquire_locked` rather than wrapped around it in a cancelling
        // `tokio::time::timeout`: a cancel drops the future at an arbitrary
        // await point, which would skip the cooldown write for a timed-out
        // interactive attempt and let the next `Auto` caller re-pop a browser.
        // Threading the deadline lets every interactive timeout exit through
        // the common outcome writer while the lock is still held.
        let attempt_deadline = std::time::Instant::now() + AUTH_ATTEMPT_DEADLINE;
        self.acquire_locked(
            intent,
            rejected,
            attempt_deadline,
            &attempt_path,
            snapshot_gen,
        )
        .await
    }

    /// Slow-path body, run while holding the cross-process auth lock.
    ///
    /// `attempt_deadline` bounds the whole locked flow. Discovery and refresh
    /// are each bounded by the HTTP client's per-request timeout; the browser
    /// flow is wrapped in the *remaining* budget so a total-deadline expiry
    /// during the interactive step surfaces as [`AuthError::TimedOut`] through
    /// the same arm that records the cooldown — never as a cancellation that
    /// drops the guard without writing it.
    ///
    /// `attempt_path` + `snapshot_gen` implement cross-process failure
    /// single-flight: the caller snapshotted `snapshot_gen` before queueing on
    /// the lock; if the generation has since advanced, a predecessor completed
    /// while we waited. A caller already queued when the predecessor ran adopts
    /// its same-intent terminal failure rather than re-running — including
    /// `UserInitiated` callers, mirroring what [`INFLIGHT`] does within one
    /// process. A `UserInitiated` caller arriving *after* the failure snapshots
    /// the new generation and naturally does not adopt.
    async fn acquire_locked(
        &self,
        intent: AuthIntent,
        rejected: Option<&str>,
        attempt_deadline: std::time::Instant,
        attempt_path: &Path,
        snapshot_gen: u64,
    ) -> Result<CachedToken, AuthError> {
        let mut state = self.state.lock().await;

        // A 401 (`rejected = Some`) proves the cached access token is dead even
        // though its local expiry clock still looks fresh. Neutralize it now,
        // under the lock, so it can never be served again: cache_hit already
        // excludes it for callers carrying `rejected`, but a later plain
        // `bearer()` (`rejected = None`) or a freshly constructed source would
        // otherwise trust the clock and hand back the proven-dead bytes. The
        // refresh token is untouched — it was not rejected and drives the
        // recovery below.
        self.expire_rejected(&mut state, rejected);

        // Re-check under the lock: a holder we queued behind may have already
        // produced a token (this process or a sibling wrote the cache).
        if self.cached_hit(&mut state, rejected).is_some() {
            // `cached_hit` guarantees state is populated on a hit (memory entry
            // was already there, or disk token was adopted into state).
            return Ok(state.clone().expect("cached_hit confirmed token in state"));
        }

        // Cross-process failure single-flight. A predecessor completed while
        // this caller was waiting on the lock: check whether its outcome was a
        // terminal failure we should adopt rather than re-run. The contract is
        // *temporal*, not intent-based: a caller whose pre-queue snapshot is
        // older than the current generation was already queued while the
        // predecessor ran and may adopt its failure, mirroring how the
        // in-process [`INFLIGHT`] registry coalesces same-intent callers
        // (including `UserInitiated`) within a single process. A `UserInitiated`
        // caller arriving *after* a failure naturally snapshots the new
        // generation and does not adopt, so "later explicit user retry bypasses"
        // falls out without a special case. The conditions are:
        //   (a) the attempt generation advanced past our snapshot — we were
        //       queued while the predecessor ran, not a fresh arrival after it;
        //   (b) the recorded intent matches ours — cross-process adoption
        //       respects the same (path, intent) boundary as INFLIGHT, so a
        //       `UserInitiated` waiter never inherits an `Auto`/`Headless`
        //       failure (different intent, different promise to the user);
        //   (c) the recorded result is a recognized terminal failure — `"ok"`
        //       and unrecognized codes fall through to a normal attempt;
        //   (d) the recorded rejected_digest matches ours — a failure caused by
        //       the predecessor's specific rejected token is not valid for a
        //       caller with a *different* rejected token (both-`None` matches).
        //       A digest mismatch triggers a normal attempt; a false rerun on a
        //       non-rejection-relative failure costs one network round-trip and
        //       stays headless — preferable to silently serving a wrong denial.
        //
        // Adoptors do NOT write a new attempt record: adopting does not
        // represent new work. Writing one would advance the generation so a
        // third caller that arrives after the adoption (snapshot = new gen) sees
        // no advance and tries its own attempt — but a fourth arriving while the
        // third runs would inherit the adopter's re-written record, relaying the
        // original failure indefinitely. The original record already has the
        // correct generation; subsequent waiters with snapshot < original gen
        // still adopt from it directly.
        if let Some(rec) = read_attempt(attempt_path) {
            if rec.generation > snapshot_gen
                && rec.intent == intent.as_str()
                && rec.rejected_digest == digest_of(rejected)
            {
                if let Some(err) = AuthError::from_code(&rec.result) {
                    return Err(err);
                }
            }
        }

        // Refresh-token grant, if we have one. Endpoints are discovered lazily
        // here (and reused by the browser branch) so a no-refresh headless
        // failure never depends on reaching the discovery URL.
        let mut endpoints: Option<OidcEndpoints> = None;
        let mut refresh_failed = false;
        if let Some(rt) = state.as_ref().and_then(|t| t.refresh_token.clone()) {
            let eps = self.discover(&mut endpoints).await?;
            match self.refresh(eps, &rt).await {
                RefreshOutcome::Refreshed(fresh) => {
                    let result = self.finish(&mut state, fresh, intent, rejected);
                    // Record recognized terminal failures (rejected-equal reissuance)
                    // so a cross-process headless waiter can adopt them rather than
                    // re-running the same dead refresh. Successes are shared through
                    // the token cache — a waiter that wins the lock after us finds
                    // the token via `cached_hit` without reaching the adoption check.
                    if let Err(ref e) = result {
                        write_attempt(attempt_path, intent, e.code(), rejected);
                    }
                    return result;
                }
                // A transient fault (transport/timeout/5xx/decode) is not a
                // credential decision: never fall through to a browser or
                // report RefreshRejected. A sibling may have written a fresh
                // token while we ran, so honor that first; otherwise this is
                // infrastructural and surfaces as NetworkUnavailable.
                RefreshOutcome::Network => {
                    if self.cached_hit(&mut state, rejected).is_some() {
                        return Ok(state.clone().expect("cached_hit confirmed token in state"));
                    }
                    return Err(AuthError::NetworkUnavailable);
                }
                // The token endpoint rejected the grant: a dead refresh token.
                // A sibling may still have won the race while we ran; if not,
                // fall through to a browser (interactive) or RefreshRejected
                // (headless).
                RefreshOutcome::Rejected => {
                    if self.cached_hit(&mut state, rejected).is_some() {
                        return Ok(state.clone().expect("cached_hit confirmed token in state"));
                    }
                    refresh_failed = true;
                }
            }
        }

        // No token from cache or refresh. Browser or terminal failure.
        if !intent.may_open_browser() {
            let err = if refresh_failed {
                AuthError::RefreshRejected
            } else {
                AuthError::NoCredential
            };
            write_attempt(attempt_path, intent, err.code(), rejected);
            return Err(err);
        }

        let cooldown_path = self.cooldown_path();
        if intent.honors_cooldown() {
            // A recent browser attempt failed; surface its recorded outcome
            // instead of re-popping a browser on this automatic attempt.
            if let Some(recorded) = read_cooldown(&cooldown_path) {
                return Err(recorded);
            }
        } else {
            // An explicit user retry clears any prior suppression.
            clear_cooldown(&cooldown_path);
        }

        let eps = self.discover(&mut endpoints).await?;
        // Wrap the browser flow in the *remaining* attempt budget so the total
        // locked time never exceeds `attempt_deadline` (and thus never
        // outlasts a waiter's `LOCK_WAIT_TIMEOUT`). A deadline expiry maps to
        // `TimedOut`, which is cooldown-worthy, so it flows through the same
        // writer arm below instead of being dropped by a cancel that would
        // release the lock without recording the cooldown.
        let remaining = attempt_deadline.saturating_duration_since(std::time::Instant::now());
        let flow = browser_pkce_flow(&self.http, &self.cfg, eps, self.opener.as_ref());
        let outcome = match tokio::time::timeout(remaining, flow).await {
            Ok(result) => result,
            Err(_) => Err(AuthError::TimedOut),
        };
        match outcome {
            // `finish` clears the cooldown on success and rejects a re-issued
            // 401'd token before persisting it.
            Ok(fresh) => {
                let result = self.finish(&mut state, fresh, intent, rejected);
                let code = match &result {
                    Ok(_) => "ok",
                    Err(e) => e.code(),
                };
                write_attempt(attempt_path, intent, code, rejected);
                result
            }
            Err(e) => {
                if e.is_cooldown_worthy() {
                    write_cooldown(&cooldown_path, &e);
                }
                write_attempt(attempt_path, intent, e.code(), rejected);
                Err(e)
            }
        }
    }

    /// Persist a freshly-obtained token, clear any cooldown, and return the
    /// full [`CachedToken`] on success. A cache-write failure maps to
    /// [`AuthError::NetworkUnavailable`] (the infrastructural bucket) — the
    /// token was valid but couldn't be persisted, which the caller should treat
    /// as transient, not as a credential rejection.
    ///
    /// The candidate-token persistence boundary for refresh and browser results.
    /// Cache-hit paths bypass this function, but every refresh- or browser-issued
    /// token flows through here before being written to memory or disk. This is
    /// where the 401-recovery invariant is enforced: a token equal to the
    /// caller's `rejected` bytes must never be committed — doing so would cache
    /// the proven-dead token as fresh, so a later plain `bearer()` (`rejected =
    /// None`) or a freshly constructed source reading the same cache would serve
    /// it back. Validating *before* the write keeps the dead token out of the
    /// cache and off disk entirely: we fail typed (`NetworkUnavailable` interactive
    /// / `RefreshRejected` headless) without caching it or clearing the cooldown.
    /// `cached_hit` and `usable_from_disk` already exclude `rejected`, so guarding
    /// the two live-token sites (refresh and browser exchange) here covers every
    /// path that can produce the rejected bytes.
    ///
    /// Returning the full [`CachedToken`] (rather than just the bearer string)
    /// lets `acquire_locked` → `acquire_leader` propagate it all the way to
    /// [`LeaderGuard::complete`], which publishes it through the [`InflightSlot`]
    /// so every joiner can reconcile its own independent `state` cell.
    fn finish(
        &self,
        state: &mut Option<CachedToken>,
        token: CachedToken,
        intent: AuthIntent,
        rejected: Option<&str>,
    ) -> Result<CachedToken, AuthError> {
        if rejected == Some(token.access_token.as_str()) {
            return Err(if intent.may_open_browser() {
                AuthError::NetworkUnavailable
            } else {
                AuthError::RefreshRejected
            });
        }
        self.save(state, token.clone())
            .map_err(|_| AuthError::NetworkUnavailable)?;
        clear_cooldown(&self.cooldown_path());
        Ok(token)
    }
}

#[async_trait]
impl TokenSource for PkceOAuthTokenSource {
    /// Acquire a bearer for a request. Routes through the coordinator as a
    /// [`Headless`](AuthIntent::Headless) acquisition: it serves a cached or
    /// refreshed token but never opens a browser, so a managed runtime with no
    /// interactive display can never hang on inference. First-use auth is the
    /// job of `buzz-agent auth databricks` ([`interactive_login`]).
    ///
    /// [`interactive_login`]: PkceOAuthTokenSource::interactive_login
    async fn bearer(&self) -> Result<String, AgentError> {
        self.acquire(AuthIntent::Headless, None)
            .await
            .map_err(Into::into)
    }

    /// Identical to [`bearer`](Self::bearer) for this source — both are
    /// headless. Retained as a distinct method so callers can state the
    /// no-browser requirement at the call site (and so other [`TokenSource`]
    /// impls that *would* browse in `bearer` can still expose a safe path).
    async fn bearer_no_browser(&self) -> Result<String, AgentError> {
        self.acquire(AuthIntent::Headless, None)
            .await
            .map_err(Into::into)
    }

    /// Force a fresh bearer after the server rejected `rejected` with a 401.
    ///
    /// A [`Headless`](AuthIntent::Headless) acquisition keyed by token
    /// *identity* rather than the expiry clock: a 401 means the cached token
    /// was rejected while still locally fresh, so [`is_expired`] would wrongly
    /// keep it. Passing `rejected` makes the coordinator run the refresh-token
    /// grant unless a concurrent caller already replaced the token, in which
    /// case that newer token is returned without a second grant. Never opens a
    /// browser; a dead refresh token surfaces terminally so the retry loop
    /// stops instead of hanging.
    async fn refresh_now(&self, rejected: &str) -> Result<String, AgentError> {
        self.acquire(AuthIntent::Headless, Some(rejected))
            .await
            .map_err(Into::into)
    }
}

// ---- helpers -------------------------------------------------------------

/// SHA-256 hex digest of `rejected` token bytes, or `None` when there is no
/// rejected token. Used to scope in-process and cross-process failure adoption
/// to the specific token that was rejected — a joiner carrying a *different*
/// rejected token (or none) must not inherit a rejection-relative failure.
fn digest_of(rejected: Option<&str>) -> Option<String> {
    rejected.map(|r| hex::encode(sha2::Sha256::digest(r.as_bytes())))
}

/// Aborts a spawned task when dropped. Used to guarantee the localhost
/// callback server doesn't outlive a failed/abandoned PKCE attempt.
struct AbortOnDrop(tokio::task::JoinHandle<()>);

impl Drop for AbortOnDrop {
    fn drop(&mut self) {
        self.0.abort();
    }
}

fn is_expired(t: &CachedToken) -> bool {
    let Some(exp) = t.expires_at else {
        return false;
    };
    now_secs() + TOKEN_REFRESH_LEEWAY.as_secs() >= exp
}

const BUZZ_AGENT_CONFIG_DIR_ENV: &str = "BUZZ_AGENT_CONFIG_DIR";

fn oauth_cache_root_for(
    config_override: Option<PathBuf>,
    home_dir: Option<PathBuf>,
) -> Result<PathBuf, AgentError> {
    if let Some(root) = config_override {
        return Ok(root.join("buzz-agent").join("oauth"));
    }
    Ok(home_dir
        .ok_or_else(|| AgentError::Llm("oauth cache: home directory not found".into()))?
        .join(".config")
        .join("buzz-agent")
        .join("oauth"))
}

fn default_oauth_cache_root() -> Result<PathBuf, AgentError> {
    oauth_cache_root_for(
        std::env::var_os(BUZZ_AGENT_CONFIG_DIR_ENV).map(PathBuf::from),
        dirs::home_dir(),
    )
}

fn cache_path_for(cfg: &PkceOAuthConfig) -> Result<PathBuf, AgentError> {
    let mut h = sha2::Sha256::new();
    h.update(cfg.discovery_url.as_bytes());
    h.update(b"|");
    h.update(cfg.client_id.as_bytes());
    h.update(b"|");
    h.update(cfg.scopes.join(",").as_bytes());
    let hash = hex::encode(h.finalize());

    let dir = match &cfg.cache_dir_override {
        Some(p) => p.join(&cfg.cache_namespace),
        None => default_oauth_cache_root()?.join(&cfg.cache_namespace),
    };
    Ok(dir.join(format!("{hash}.json")))
}

/// Append `ext` as an extra extension onto `base` (e.g. `<hash>.json` →
/// `<hash>.json.lock`). Keeps the lock and cooldown sidecars in the same
/// per-key directory as the cache, so they inherit its `$HOME` override and
/// owner-only parent without a second key derivation.
fn append_ext(base: &Path, ext: &str) -> PathBuf {
    let mut name = base.as_os_str().to_owned();
    name.push(".");
    name.push(ext);
    PathBuf::from(name)
}

/// Durable record of the last browser-attempt failure for a cache key. Written
/// while holding the auth lock so concurrent writers can't interleave, read by
/// `Auto` callers to decide whether to suppress an automatic browser re-launch.
#[derive(Debug, Serialize, Deserialize)]
struct CooldownRecord {
    /// [`AuthError::code`] of the failure being cooled down.
    code: String,
    /// Unix seconds after which the cooldown lapses and an `Auto` caller may
    /// launch a browser again.
    until: u64,
}

/// Durable record of the generation and outcome of the most recently completed
/// slow-path acquisition attempt for a cache key.
///
/// Cross-process single-flight for *failures*: the in-process [`INFLIGHT`]
/// registry coalesces same-key callers within one process, but two separate
/// processes both waiting on the OS file lock do NOT share the registry. When
/// process A holds the lock and fails (e.g. browser denial or dead refresh),
/// process B's queued caller acquires the lock after A releases it and — under
/// the old protocol — would re-run the full flow from scratch. This record lets
/// B detect that it was already queued while A ran and adopt A's failure
/// instead of hammering the provider again.
///
/// Protocol:
/// - A caller **snapshots** the current generation from the sidecar *before*
///   queueing on the file lock.
/// - A caller that **acquires** the lock compares the current generation to its
///   snapshot: if it advanced, a predecessor completed while it was waiting.
///   If the recorded intent matches this caller's intent and the outcome is a
///   recognized terminal failure, adopt it rather than re-running.
/// - Completing attempts **write** a fresh record under the lock. Write
///   coverage: the headless no-browser arm (`RefreshRejected`/`NoCredential`),
///   the refresh arm when `finish()` fails typed (rejected-equal reissuance),
///   and the browser arm (all outcomes including `"ok"`). Omissions that are
///   intentionally not adoption-worthy: transient `Network` errors, discovery
///   failures (both non-terminal; next caller retries), and cache/refresh-
///   success paths (a waiting caller finds the token via `cached_hit` without
///   reaching the adoption check).
///
/// The generation counter is read fresh from disk at write time so each
/// completed attempt strictly advances the value regardless of when the
/// caller's pre-queue snapshot was taken.
///
/// Intent matching is same-intent only, mirroring the in-process `(path,
/// intent)` key. The temporal condition handles "later explicit retry bypasses":
/// a `UserInitiated` caller arriving after the failure snapshots the new
/// generation and sees no advance, so it always runs its own attempt and never
/// inherits a prior failure — regardless of intent.
#[derive(Debug, Serialize, Deserialize)]
struct AttemptRecord {
    /// Strictly increasing counter: read from disk at write time and incremented
    /// by one so each attempt advances from the actual current value regardless
    /// of when the writing caller's snapshot was taken.
    generation: u64,
    /// Intent of the attempt that completed, as [`AuthIntent::as_str`].
    intent: String,
    /// Error code of the terminal failure, or `"ok"` on success. Matches
    /// [`AuthError::code`] / the `"ok"` sentinel.
    result: String,
    /// SHA-256 hex digest of the token bytes that the completing caller had
    /// marked as `rejected`, or `None` when the caller carried no rejected
    /// token. A waiter adopts only when its own digest matches: a failure caused
    /// by the leader's specific rejected token is not valid for a waiter with a
    /// *different* rejected token (or none) — its refresh may yield a live
    /// token. Both-`None` is a match. A mismatched digest triggers a normal
    /// attempt; a false rerun on a non-rejection failure costs one network round-
    /// trip and stays headless — preferable to silently adopting a wrong denial.
    #[serde(default)]
    rejected_digest: Option<String>,
}

/// Read the attempt sidecar at `path`, if any. Returns `None` when absent,
/// unparseable, or the generation is 0 (no attempt has completed yet).
fn read_attempt(path: &Path) -> Option<AttemptRecord> {
    let body = fs::read(path).ok()?;
    let record: AttemptRecord = serde_json::from_slice(&body).ok()?;
    Some(record)
}

/// Write a fresh attempt record at `path`. Called under the auth lock.
/// Best-effort — a write failure only means the next cross-process waiter
/// cannot adopt this attempt's outcome, so errors are swallowed.
///
/// Always reads the current on-disk generation before writing so the new
/// record strictly advances from the actual last-recorded value, not from
/// any caller's pre-queue snapshot. An intervening different-intent attempt
/// that advanced the sidecar between snapshot and lock-acquire is reflected
/// correctly: the next waiter's comparison still sees a real advance.
fn write_attempt(path: &Path, intent: AuthIntent, result: &str, rejected: Option<&str>) {
    let current_gen = read_attempt(path).map_or(0, |r| r.generation);
    let record = AttemptRecord {
        generation: current_gen.wrapping_add(1),
        intent: intent.as_str().to_owned(),
        result: result.to_owned(),
        rejected_digest: digest_of(rejected),
    };
    if let Ok(body) = serde_json::to_vec(&record) {
        let _ = write_private_cache(path, &body);
    }
}

fn now_secs() -> u64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|d| d.as_secs())
        .unwrap_or(0)
}

/// Return the still-active cooldown outcome for `path`, if any.
///
/// `None` when the sidecar is absent, unparseable, expired, or records a code
/// this build doesn't recognize — every one of those means "no active
/// cooldown", so the caller proceeds to a normal attempt. An expired record is
/// removed opportunistically so the directory doesn't accumulate stale files.
fn read_cooldown(path: &Path) -> Option<AuthError> {
    let body = fs::read(path).ok()?;
    let record: CooldownRecord = serde_json::from_slice(&body).ok()?;
    if record.until > now_secs() {
        AuthError::from_code(&record.code)
    } else {
        let _ = fs::remove_file(path);
        None
    }
}

/// Record `err` as a fresh cooldown at `path`, expiring [`COOLDOWN_DURATION`]
/// from now. Best-effort: a write failure only means the next automatic
/// attempt may re-pop a browser, never a hard auth failure, so errors are
/// swallowed. Called while holding the auth lock.
fn write_cooldown(path: &Path, err: &AuthError) {
    let record = CooldownRecord {
        code: err.code().to_string(),
        until: now_secs() + COOLDOWN_DURATION.as_secs(),
    };
    if let Ok(body) = serde_json::to_vec(&record) {
        let _ = write_private_cache(path, &body);
    }
}

/// Remove any cooldown sidecar at `path`. Called on a successful acquisition
/// (the problem is resolved) and by `UserInitiated` callers that bypass the
/// cooldown (an explicit retry clears the suppression). Best-effort.
fn clear_cooldown(path: &Path) {
    let _ = fs::remove_file(path);
}

/// Hold on the cross-process auth lock. Dropping it (or the owning process
/// dying) releases the OS advisory lock — no PID files, no manual break.
#[derive(Debug)]
struct AuthLockGuard(fs::File);

impl Drop for AuthLockGuard {
    fn drop(&mut self) {
        // Explicit for intent; closing the fd would release it regardless.
        let _ = FileExt::unlock(&self.0);
    }
}

/// Acquire the cross-process auth lock at `path`, polling until `deadline`.
///
/// `fs2::FileExt::try_lock_exclusive` maps to `flock(LOCK_EX | LOCK_NB)` on
/// Unix and `LockFileEx` on Windows — advisory, per–open-file-description, so
/// a lock taken on one handle blocks every other handle (same process or not),
/// which is exactly the cross-process single-flight guarantee we want. The
/// try-lock is non-blocking, so we poll on [`LOCK_POLL_INTERVAL`] rather than
/// parking a worker thread in a blocking `lock_exclusive()`. Contention is
/// reported as [`fs2::lock_contended_error`] (`EWOULDBLOCK`/`EACCES` on Unix,
/// `ERROR_LOCK_VIOLATION` on Windows); we match its `raw_os_error` and retry.
/// Any other error is a real fault and returns [`AuthError::LockTimeout`]. A
/// waiter whose `deadline` lapses also returns [`AuthError::LockTimeout`];
/// because the caller sets that deadline longer than [`AUTH_ATTEMPT_DEADLINE`],
/// a healthy holder always finishes first.
async fn acquire_auth_lock(
    path: &Path,
    deadline: std::time::Instant,
) -> Result<AuthLockGuard, AuthError> {
    if let Some(parent) = path.parent() {
        fs::create_dir_all(parent).map_err(|_| AuthError::LockTimeout)?;
    }
    let file = fs::OpenOptions::new()
        .create(true)
        .truncate(false)
        .write(true)
        .open(path)
        .map_err(|_| AuthError::LockTimeout)?;
    let contended = fs2::lock_contended_error().raw_os_error();
    loop {
        match file.try_lock_exclusive() {
            Ok(()) => return Ok(AuthLockGuard(file)),
            Err(e) if e.raw_os_error() == contended => {
                if std::time::Instant::now() >= deadline {
                    return Err(AuthError::LockTimeout);
                }
                tokio::time::sleep(LOCK_POLL_INTERVAL).await;
            }
            Err(_) => return Err(AuthError::LockTimeout),
        }
    }
}

/// Key for the in-process single-flight registry: the cross-process lock path
/// (one per cache key) paired with the caller's [`AuthIntent`]. Keying by the
/// full intent — not merely browser capability — keeps callers with *different*
/// outcome policy from coalescing: an `Auto` leader honors a live cooldown and
/// returns its recorded `Denied`/`TimedOut`, but a `UserInitiated` caller is
/// promised a cooldown bypass and a fresh browser, so it must never inherit an
/// `Auto` leader's suppressed result. Each intent still coalesces with itself
/// (two concurrent `UserInitiated` sign-ins share one browser), and all intents
/// on the same key still serialize through the cross-process file lock.
type InflightKey = (PathBuf, AuthIntent);

/// Process-global registry of in-flight auth attempts, the in-process
/// counterpart to [`acquire_auth_lock`]'s cross-process file lock. The file
/// lock serializes work across processes and shares *success* via a cache
/// re-read, but a queued caller that acquires the lock after a browser denial
/// would clear the sidecar and pop a second browser. This registry closes that
/// gap: a caller that arrives while a leader's attempt is in flight joins the
/// leader's [`InflightSlot`] and receives the *same* result — success or
/// failure — instead of taking the lock afterward and launching again. Guarded
/// by a `std::sync::Mutex` because every critical section is a cheap map lookup
/// with no `.await` held.
static INFLIGHT: LazyLock<std::sync::Mutex<HashMap<InflightKey, Arc<InflightSlot>>>> =
    LazyLock::new(|| std::sync::Mutex::new(HashMap::new()));

/// Lock the in-flight registry, recovering from a poisoned mutex rather than
/// panicking: the only work done under this lock is map lookups that can't
/// leave inconsistent state, so a poison from an unrelated panic must not wedge
/// every future auth attempt.
fn inflight_registry() -> std::sync::MutexGuard<'static, HashMap<InflightKey, Arc<InflightSlot>>> {
    INFLIGHT.lock().unwrap_or_else(|e| e.into_inner())
}

/// The value published by a leader to its joiners: the leader's rejected-token
/// SHA-256 digest (non-secret identity, `None` when the leader carried no
/// `rejected`) paired with the attempt result. Joiners use the digest to detect
/// a mismatch — the leader's rejection-relative failure is not valid for a
/// joiner that carried a *different* rejected token.
///
/// On success the full [`CachedToken`] is published so each joiner can
/// conditionally reconcile its own independent [`PkceOAuthTokenSource::state`]
/// cell. Publishing the full credential (not just the bearer string) prevents
/// a joining source's state from remaining stale or empty after the coalesced
/// flow, which would otherwise cause a subsequent plain `bearer()` on that
/// source to resurface a rejected or absent credential rather than the
/// newly-acquired one.
type SlotPublish = (Option<String>, Result<CachedToken, AuthError>);

/// The shared result of one leader's auth attempt, awaited by any joiner that
/// arrived while the leader was in flight. A `watch` channel gives us
/// publish-once plus wait-for-publish in one primitive: the leader publishes
/// exactly once through [`LeaderGuard`]; joiners clone the published result.
struct InflightSlot {
    tx: watch::Sender<Option<SlotPublish>>,
    rx: watch::Receiver<Option<SlotPublish>>,
}

impl InflightSlot {
    fn new() -> Self {
        let (tx, rx) = watch::channel(None);
        Self { tx, rx }
    }

    /// Block until the leader publishes, then clone out `(rejected_digest, result)`.
    ///
    /// `borrow_and_update` marks the current value seen before awaiting, so a
    /// publish that lands between the read and the `changed()` await is not a
    /// lost wakeup — the version has advanced, so `changed()` returns at once.
    /// A closed channel (leader dropped without publishing — which
    /// [`LeaderGuard`]'s `Drop` prevents) surfaces as a transient so the caller
    /// retries rather than hangs.
    async fn wait(&self) -> SlotPublish {
        let mut rx = self.rx.clone();
        loop {
            if let Some(publish) = rx.borrow_and_update().clone() {
                return publish;
            }
            if rx.changed().await.is_err() {
                return (None, Err(AuthError::NetworkUnavailable));
            }
        }
    }

    /// Publish `(rejected_digest, result)` to every waiting joiner. A send
    /// error means no joiners remain, which is fine.
    fn publish(&self, rejected_digest: Option<String>, result: Result<CachedToken, AuthError>) {
        let _ = self.tx.send(Some((rejected_digest, result)));
    }
}

/// RAII owner of a leader's in-flight slot. Guarantees the slot is evicted from
/// [`INFLIGHT`] and a result published to joiners even if the leader future is
/// cancelled or panics: a leader that skipped this would leave a dead slot that
/// turns every later caller into a joiner of an attempt that never publishes,
/// wedging them until `LOCK_WAIT_TIMEOUT`.
struct LeaderGuard {
    key: InflightKey,
    slot: Arc<InflightSlot>,
    done: bool,
}

impl LeaderGuard {
    fn new(key: InflightKey, slot: Arc<InflightSlot>) -> Self {
        Self {
            key,
            slot,
            done: false,
        }
    }

    /// Normal completion: evict the slot, publish `(rejected_digest, result)`
    /// to joiners, and return the bearer to the leader. The full
    /// [`CachedToken`] is published so joiners can reconcile their own
    /// [`PkceOAuthTokenSource::state`] before returning. Evicting *before*
    /// publishing means a caller arriving after this point starts a fresh
    /// attempt (a later explicit retry may launch), while joiners already
    /// holding the slot still receive the result. `Drop` covers the cancel/panic
    /// path.
    fn complete(
        mut self,
        result: Result<CachedToken, AuthError>,
        rejected_digest: Option<String>,
    ) -> Result<String, AuthError> {
        self.done = true;
        Self::evict(&self.key, &self.slot);
        // Clone the error before moving `result` into the slot publish so we
        // can return the original error to the leader on failure.
        let leader_return = result
            .as_ref()
            .map(|t| t.access_token.clone())
            .map_err(|e| e.clone());
        self.slot.publish(rejected_digest, result);
        leader_return
    }

    /// Remove this leader's slot from the registry, but only if it is still the
    /// same slot — defends against evicting a successor a later attempt may
    /// have installed under the same key.
    fn evict(key: &InflightKey, slot: &Arc<InflightSlot>) {
        let mut reg = inflight_registry();
        if reg
            .get(key)
            .is_some_and(|existing| Arc::ptr_eq(existing, slot))
        {
            reg.remove(key);
        }
    }
}

impl Drop for LeaderGuard {
    fn drop(&mut self) {
        if self.done {
            return;
        }
        // Cancelled or panicked before `complete`: evict so later callers start
        // fresh, and wake joiners with a transient error so they retry rather
        // than hang on a leader that will never publish.
        Self::evict(&self.key, &self.slot);
        self.slot.publish(None, Err(AuthError::NetworkUnavailable));
    }
}

/// Load a cached token, enforcing the owner-only invariant on load.
///
/// Owner-only permissions are a cache *lifecycle* invariant, not just a
/// write-path property: a world-readable cache left by an older buzz-agent
/// (or any tampering) must be tightened the moment we touch it, before the
/// tokens are used — otherwise a file that never expires stays exposed until
/// some future refresh happens to rewrite it. Every load path (initial and
/// cross-process re-reads) funnels through here, so the repair covers them
/// all. Returns `None` when the cache is absent, unreadable, unparseable, or
/// cannot be secured; the caller then falls through to refresh/browser.
fn read_cache(path: &Path) -> Option<CachedToken> {
    let body = read_private_cache(path).ok()?;
    serde_json::from_slice(&body).ok()
}

/// Open the cache, reject symlinks, tighten loose permissions to `0o600`, and
/// return its bytes.
///
/// On Unix `O_NOFOLLOW` rejects a symlinked cache path at the kernel level
/// (no stat/open TOCTOU), and `fchmod` on the already-open handle repairs a
/// loose mode against the pinned inode rather than re-resolving the path.
/// A cache that exists but cannot be secured is an error, so the caller fails
/// closed instead of using an exposed file.
#[cfg(unix)]
fn read_private_cache(path: &Path) -> io::Result<Vec<u8>> {
    use std::io::Read;
    use std::os::unix::fs::{OpenOptionsExt, PermissionsExt};

    let mut file = fs::OpenOptions::new()
        .read(true)
        .custom_flags(nix::libc::O_NOFOLLOW)
        .open(path)?;

    let meta = file.metadata()?;
    if !meta.file_type().is_file() {
        return Err(io::Error::new(
            io::ErrorKind::InvalidData,
            "oauth cache is not a regular file",
        ));
    }
    // Tighten in place on the open fd if any group/other bit is set. fchmod
    // targets the inode we already hold, so no attacker can swap the path
    // between the check and the repair.
    if meta.permissions().mode() & 0o077 != 0 {
        file.set_permissions(fs::Permissions::from_mode(0o600))?;
    }

    let mut body = Vec::new();
    file.read_to_end(&mut body)?;
    Ok(body)
}

/// Non-Unix: token persistence and reading are both disabled until a
/// Windows-specific owner-only DACL is implemented. Any legacy token file
/// left by an older build (written with default ACLs) is deleted
/// opportunistically so the exposed artifact cannot be served by new builds.
/// Returns an error so [`read_cache`] yields `None`, giving a consistent
/// memory-only cache on non-Unix.
#[cfg(not(unix))]
fn read_private_cache(path: &Path) -> io::Result<Vec<u8>> {
    // Best-effort removal of any legacy file. Errors are ignored — either the
    // file does not exist (normal case) or it cannot be removed (no worse
    // than before — the DACL story is still broken, but that is the pre-fix
    // state we are trying to retire).
    let _ = fs::remove_file(path);
    Err(io::Error::new(
        io::ErrorKind::Unsupported,
        "token disk cache disabled on non-Unix (no owner-only DACL)",
    ))
}

/// Removes a temp file on drop unless it was already renamed away. Keeps a
/// failed/partial write from leaving a stray token file behind.
struct TmpFileGuard<'a>(&'a Path);

impl Drop for TmpFileGuard<'_> {
    fn drop(&mut self) {
        let _ = fs::remove_file(self.0);
    }
}

/// A per-write-unique temp suffix so concurrent savers — sibling threads or
/// separate processes sharing `$HOME` — never collide on one temp path.
/// Falls back to a timestamp if the RNG is unavailable rather than panicking
/// mid-auth.
fn unique_suffix() -> String {
    let mut bytes = [0u8; 8];
    if getrandom::fill(&mut bytes).is_ok() {
        return hex::encode(bytes);
    }
    let nanos = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|d| d.as_nanos())
        .unwrap_or(0);
    format!("{nanos:x}")
}

/// Write `body` to `path` as an owner-only file via an atomic rename.
///
/// The cache holds both the refresh and access tokens, so it must never be
/// readable by other users. We create a uniquely-named temp file in the same
/// directory with owner-only protection at creation time — mode `0o600` on
/// Unix (see [`create_private_temp_file`]) — so it is never briefly
/// world/other readable, write and fsync it, then rename over the
/// destination. The rename swaps the inode/entry wholesale, so a pre-existing
/// cache file with loose permissions is *replaced* by the new private one;
/// its old mode never survives. `fs::rename` maps to
/// `MOVEFILE_REPLACE_EXISTING` on Windows, so the atomic replace holds on
/// both platforms; the Windows owner-only DACL is pending the unsafe-FFI
/// decision noted at the seam.
fn write_private_cache(path: &Path, body: &[u8]) -> io::Result<()> {
    let parent = path.parent().ok_or_else(|| {
        io::Error::new(
            io::ErrorKind::InvalidInput,
            "oauth cache path has no parent directory",
        )
    })?;
    fs::create_dir_all(parent)?;

    let file_name = path
        .file_name()
        .and_then(|n| n.to_str())
        .unwrap_or("oauth-cache");
    let tmp = parent.join(format!(".{file_name}.{}.tmp", unique_suffix()));
    let guard = TmpFileGuard(&tmp);

    let mut f = create_private_temp_file(&tmp)?;
    f.write_all(body)?;
    f.sync_all()?;
    drop(f);

    fs::rename(&tmp, path)?;
    // The rename consumed the temp path; nothing left to clean up.
    std::mem::forget(guard);
    Ok(())
}

/// Create `tmp` for writing with owner-only permissions from the moment it
/// exists. Fails if the file already exists (`create_new`), which the
/// per-write-unique suffix makes effectively impossible.
#[cfg(unix)]
fn create_private_temp_file(tmp: &Path) -> io::Result<fs::File> {
    use std::os::unix::fs::OpenOptionsExt;
    fs::OpenOptions::new()
        .write(true)
        .create_new(true)
        .mode(0o600)
        .open(tmp)
}

/// Non-Unix fallback: create the temp file if it does not already exist.
///
/// On Windows the owner-only equivalent is an explicit DACL set at creation
/// (`CreateFileW` with SDDL `D:P(A;;FA;;;OW)`, matching goose's
/// `private_file.rs`), but that FFI needs `unsafe`, which this crate forbids.
/// Reconciling the two — an isolated helper crate, a vetted safe dependency,
/// or descoping Windows — is an open decision escalated to the maintainer, so
/// this interim relies on the default per-user ACLs and drops the owner-only
/// implementation in behind this seam once the decision lands. `create_new`
/// fails if the file already exists.
#[cfg(not(unix))]
fn create_private_temp_file(tmp: &Path) -> io::Result<fs::File> {
    fs::OpenOptions::new()
        .write(true)
        .create_new(true)
        .open(tmp)
}

/// Parse a token-endpoint JSON response. Fails loudly when `access_token`
/// is missing or empty — without this, a malformed server response would
/// be cached and `bearer()` would silently return `""` until the entry
/// expires or is deleted by hand.
fn token_from_response(
    v: &Value,
    fallback_refresh: Option<&str>,
) -> Result<CachedToken, AgentError> {
    let access_token = v
        .get("access_token")
        .and_then(Value::as_str)
        .filter(|s| !s.is_empty())
        .ok_or_else(|| AgentError::Llm("oauth: token response missing/empty access_token".into()))?
        .to_string();
    let refresh_token = v
        .get("refresh_token")
        .and_then(Value::as_str)
        .map(str::to_string)
        .or_else(|| fallback_refresh.map(str::to_string));
    let expires_at = v
        .get("expires_in")
        .and_then(Value::as_u64)
        .map(|secs| {
            SystemTime::now()
                .duration_since(UNIX_EPOCH)
                .map(|d| d.as_secs())
                .unwrap_or(0)
                .checked_add(secs)
                .ok_or_else(|| AgentError::Llm("oauth: token expiry overflow".into()))
        })
        .transpose()?;
    Ok(CachedToken {
        access_token,
        refresh_token,
        expires_at,
    })
}

/// PKCE pieces: URL-safe random verifier (~64 chars) and its SHA-256
/// challenge (RFC 7636 §4.2).
fn pkce_pair() -> Result<(String, String), AgentError> {
    let mut bytes = [0u8; 48];
    getrandom::fill(&mut bytes).map_err(|e| AgentError::Llm(format!("pkce rng: {e}")))?;
    let verifier = base64::engine::general_purpose::URL_SAFE_NO_PAD.encode(bytes);
    let challenge = base64::engine::general_purpose::URL_SAFE_NO_PAD
        .encode(sha2::Sha256::digest(verifier.as_bytes()));
    Ok((verifier, challenge))
}

fn random_state() -> Result<String, AgentError> {
    let mut bytes = [0u8; 16];
    getrandom::fill(&mut bytes).map_err(|e| AgentError::Llm(format!("state rng: {e}")))?;
    Ok(base64::engine::general_purpose::URL_SAFE_NO_PAD.encode(bytes))
}

/// Decide the OAuth callback result and the HTML page to serve.
///
/// Returns `(result, page)`: `result` carries the auth code (or a detail
/// string on failure) to the waiting flow via the oneshot channel; `page` is
/// the *static* HTML shown in the browser. The page never embeds any request
/// parameter — the `error` query value is attacker-influenceable, so
/// reflecting it would be an XSS sink on the localhost callback. Failure
/// detail travels only through `result`; the waiting flow discards it and
/// reports a typed error and fixed diagnostic. The page stays static.
fn callback_outcome(
    params: &std::collections::HashMap<String, String>,
    expected_state: &str,
) -> (Result<String, String>, String) {
    let result = match (params.get("code"), params.get("state")) {
        (Some(code), Some(st)) if st == expected_state => Ok(code.clone()),
        (Some(_), Some(_)) => Err("state mismatch".to_string()),
        _ => Err(params
            .get("error")
            .map(|e| sanitize_callback_detail(e))
            .unwrap_or_else(|| "missing code".into())),
    };
    let page = match result {
        Ok(_) => "<h2>Buzz: signed in</h2><p>You can close this window.</p>",
        Err(_) => "<h2>Buzz auth failed</h2><p>You can close this window and try again.</p>",
    }
    .to_string();
    (result, page)
}

/// Bound and normalize an attacker-controllable OAuth `error` value carried
/// internally through the callback result. The waiting flow discards this detail;
/// it must not be reflected in browser markup, outward errors or logs.
fn sanitize_callback_detail(raw: &str) -> String {
    const MAX: usize = 200;
    raw.chars()
        .map(|c| if c.is_control() { ' ' } else { c })
        .take(MAX)
        .collect()
}

/// Spin up a localhost callback server, hand the authorize URL to `opener`,
/// wait up to [`BROWSER_AUTH_TIMEOUT`] for the redirect, then exchange the
/// code for a token.
///
/// `opener` is invoked *after* the listener is bound and the abort guard is
/// armed, so a launch failure never returns a URL pointing at a torn-down
/// listener. Every failure is a typed [`AuthError`] so the coordinator can
/// record a cooldown (or not) by category: an open failure is
/// [`BrowserOpenFailed`], a redirect that never arrives is [`TimedOut`], a
/// provider-reported denial is [`Denied`], and a code exchange the provider
/// rejects with `invalid_grant` is [`ExchangeFailed`]; infrastructure faults
/// (bind/exchange transport, 429, 5xx, or a malformed success body) are
/// [`NetworkUnavailable`].
///
/// [`BrowserOpenFailed`]: AuthError::BrowserOpenFailed
/// [`TimedOut`]: AuthError::TimedOut
/// [`Denied`]: AuthError::Denied
/// [`ExchangeFailed`]: AuthError::ExchangeFailed
/// [`NetworkUnavailable`]: AuthError::NetworkUnavailable
async fn browser_pkce_flow(
    http: &Client,
    cfg: &PkceOAuthConfig,
    endpoints: &OidcEndpoints,
    opener: &dyn BrowserOpener,
) -> Result<CachedToken, AuthError> {
    use axum::{extract::Query, response::Html, routing::get, Router};
    use std::collections::HashMap;
    use std::net::SocketAddr;
    use tokio::sync::oneshot;

    let (verifier, challenge) = pkce_pair().map_err(|_| AuthError::NetworkUnavailable)?;
    let state = random_state().map_err(|_| AuthError::NetworkUnavailable)?;

    let (tx, rx) = oneshot::channel::<Result<String, String>>();
    let tx = Arc::new(Mutex::new(Some(tx)));

    let expected_state = state.clone();
    let app = Router::new().route(
        "/",
        get(move |Query(params): Query<HashMap<String, String>>| {
            let tx = Arc::clone(&tx);
            let expected = expected_state.clone();
            async move {
                let (result, page) = callback_outcome(&params, &expected);
                if let Some(sender) = tx.lock().await.take() {
                    let _ = sender.send(result);
                }
                Html(page)
            }
        }),
    );

    let listener = tokio::net::TcpListener::bind(SocketAddr::from(([127, 0, 0, 1], 0)))
        .await
        .map_err(|_| AuthError::NetworkUnavailable)?;
    let port = listener
        .local_addr()
        .map_err(|_| AuthError::NetworkUnavailable)?
        .port();
    let redirect_uri = format!("http://localhost:{port}");

    // `_server` is held until this function returns; the drop guard aborts
    // the axum task on every exit path (timeout, callback error, token
    // exchange failure, or success), so we never leak a listener bound to
    // 127.0.0.1 past the auth attempt.
    let _server = AbortOnDrop(tokio::spawn(async move {
        let _ = axum::serve(listener, app).await;
    }));

    let auth_url = format!(
        "{}?response_type=code&client_id={}&redirect_uri={}&scope={}&state={}&code_challenge={}&code_challenge_method=S256",
        endpoints.authorization_endpoint,
        urlencoding::encode(&cfg.client_id),
        urlencoding::encode(&redirect_uri),
        urlencoding::encode(&cfg.scopes.join(" ")),
        urlencoding::encode(&state),
        urlencoding::encode(&challenge),
    );

    // Launch the browser while the listener is live. A launch failure aborts
    // before we wait on a redirect nobody can send.
    opener.open(&auth_url).map_err(|_| {
        tracing::warn!("oauth browser launch failed");
        AuthError::BrowserOpenFailed
    })?;

    let code = match tokio::time::timeout(BROWSER_AUTH_TIMEOUT, rx).await {
        // Timed out waiting for the redirect.
        Err(_) => return Err(AuthError::TimedOut),
        // Callback task dropped the sender without sending — treat as timeout.
        Ok(Err(_)) => return Err(AuthError::TimedOut),
        // Provider/user reported an error (denial, state mismatch, missing code).
        Ok(Ok(Err(_))) => {
            tracing::warn!("oauth callback reported failure");
            return Err(AuthError::Denied);
        }
        Ok(Ok(Ok(code))) => code,
    };

    // Exchange code for token.
    let params = [
        ("grant_type", "authorization_code"),
        ("code", &code),
        ("redirect_uri", &redirect_uri),
        ("code_verifier", &verifier),
        ("client_id", &cfg.client_id),
    ];
    let resp = http
        .post(&endpoints.token_endpoint)
        .form(&params)
        .send()
        .await
        // Transport error or the per-request timeout elapsed: no verdict from
        // the provider, so this is infrastructural, not a rejected grant.
        .map_err(|_| AuthError::NetworkUnavailable)?;
    let status = resp.status();
    if !status.is_success() {
        let body = read_oauth_json(resp).await.ok();
        // Only a 4xx `invalid_grant` (RFC 6749 §6.4.1) establishes the
        // authorization code itself was rejected — the terminal, cooldown-worthy
        // `ExchangeFailed`. A 429, any 5xx, and any other/unparseable 4xx are a
        // transient provider fault or misconfiguration a cooldown must not
        // suppress, so they surface as `NetworkUnavailable` — mirroring the
        // refresh classifier, which likewise keys on the body `error`, not the
        // bare status class.
        if status.is_client_error()
            && body
                .and_then(|v| v.get("error").and_then(Value::as_str).map(str::to_owned))
                .as_deref()
                == Some("invalid_grant")
        {
            tracing::warn!(status = %status, "oauth code exchange rejected");
            return Err(AuthError::ExchangeFailed);
        }
        tracing::warn!(status = %status, "oauth code exchange not a grant rejection");
        return Err(AuthError::NetworkUnavailable);
    }
    // A 2xx whose body is missing/malformed or lacks an access token is a
    // provider fault, not a rejected grant: it never establishes that the code
    // was refused, so it stays in the transient bucket rather than poisoning a
    // 5-minute cooldown.
    let v = read_oauth_json(resp)
        .await
        .map_err(|_| AuthError::NetworkUnavailable)?;
    token_from_response(&v, None).map_err(|_| AuthError::NetworkUnavailable)
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::time::Instant;

    #[test]
    fn pkce_pair_produces_valid_challenge() {
        let (verifier, challenge) = pkce_pair().unwrap();
        assert!(verifier.len() >= 43);
        let expected = base64::engine::general_purpose::URL_SAFE_NO_PAD
            .encode(sha2::Sha256::digest(verifier.as_bytes()));
        assert_eq!(expected, challenge);
    }

    #[test]
    fn cached_token_no_expiry_is_not_expired() {
        let t = CachedToken {
            access_token: "x".into(),
            refresh_token: None,
            expires_at: None,
        };
        assert!(!is_expired(&t));
    }

    #[test]
    fn cached_token_far_future_is_not_expired() {
        let future = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 3600;
        let t = CachedToken {
            access_token: "x".into(),
            refresh_token: None,
            expires_at: Some(future),
        };
        assert!(!is_expired(&t));
    }

    #[test]
    fn cached_token_within_leeway_is_expired() {
        let near = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 10; // 10s away, leeway is 60s → counts as expired
        let t = CachedToken {
            access_token: "x".into(),
            refresh_token: None,
            expires_at: Some(near),
        };
        assert!(is_expired(&t));
    }

    #[test]
    fn production_and_demo_oauth_roots_are_concrete_and_distinct() {
        let home = PathBuf::from("/Users/demo");
        let production = oauth_cache_root_for(None, Some(home.clone())).unwrap();
        let first_demo_config = home
            .join("Library/Application Support")
            .join("buzz-demo-board-1234567812345678");
        let second_demo_config = home
            .join("Library/Application Support")
            .join("buzz-demo-board-8765432187654321");
        let first_demo = oauth_cache_root_for(Some(first_demo_config), Some(home.clone())).unwrap();
        let second_demo = oauth_cache_root_for(Some(second_demo_config), Some(home)).unwrap();

        assert_eq!(
            production,
            PathBuf::from("/Users/demo/.config/buzz-agent/oauth")
        );
        assert_eq!(
            first_demo,
            PathBuf::from(
                "/Users/demo/Library/Application Support/buzz-demo-board-1234567812345678/buzz-agent/oauth"
            )
        );
        assert_eq!(
            second_demo,
            PathBuf::from(
                "/Users/demo/Library/Application Support/buzz-demo-board-8765432187654321/buzz-agent/oauth"
            )
        );
        assert_ne!(production, first_demo);
        assert_ne!(production, second_demo);
        assert_ne!(first_demo, second_demo);
    }

    #[test]
    fn cache_path_preserves_production_home_config_directory() {
        let cfg = PkceOAuthConfig {
            discovery_url: "https://example.com/.well-known".into(),
            client_id: "abc".into(),
            scopes: vec!["a".into(), "b".into()],
            cache_namespace: "demo".into(),
            cache_dir_override: None,
        };
        let p = cache_path_for(&cfg).unwrap();
        let expected_dir = dirs::home_dir()
            .unwrap()
            .join(".config")
            .join("buzz-agent")
            .join("oauth")
            .join("demo");
        assert_eq!(p.parent(), Some(expected_dir.as_path()));
        assert_eq!(p.extension().and_then(|s| s.to_str()), Some("json"));
    }

    #[test]
    fn token_from_response_uses_fallback_refresh() {
        let v: Value = serde_json::from_str(r#"{"access_token":"abc","expires_in":3600}"#).unwrap();
        let t = token_from_response(&v, Some("old-refresh")).unwrap();
        assert_eq!(t.access_token, "abc");
        assert_eq!(t.refresh_token.as_deref(), Some("old-refresh"));
        assert!(t.expires_at.is_some());
    }

    #[test]
    fn token_from_response_rejects_missing_access_token() {
        let v: Value = serde_json::from_str(r#"{"expires_in":3600}"#).unwrap();
        assert!(token_from_response(&v, None).is_err());
    }

    #[test]
    fn token_from_response_rejects_empty_access_token() {
        let v: Value = serde_json::from_str(r#"{"access_token":""}"#).unwrap();
        assert!(token_from_response(&v, None).is_err());
    }

    #[cfg(unix)] // Disk adoption relies on `write_private_cache`; non-Unix disables disk persistence.
    #[tokio::test]
    async fn test_bearer_reuses_disk_token_after_expiry() {
        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://example.com/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        // Expire the in-memory state.
        {
            let mut state = source.state.lock().await;
            *state = Some(CachedToken {
                access_token: "stale".into(),
                refresh_token: None,
                expires_at: Some(0), // long expired
            });
        }

        // Write a valid token to disk (simulating another process refreshing).
        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;
        let fresh_token = CachedToken {
            access_token: "fresh-from-disk".into(),
            refresh_token: Some("rt".into()),
            expires_at: Some(future_exp),
        };
        let body = serde_json::to_vec_pretty(&fresh_token).unwrap();
        fs::write(&source.cache_path, &body).unwrap();

        // bearer() should pick up the disk token without any network call.
        let result = source.bearer().await.unwrap();
        assert_eq!(result, "fresh-from-disk");
    }

    /// A joiner that wakes to the leader's shared *failure* must still recover
    /// a sibling's valid replacement from disk. The matching-failure path
    /// neutralizes the joiner's own rejected state (under `lock().await`) and
    /// then reads the disk lock-free — so a shared failure never forces an
    /// N-way browser storm when a sibling already wrote a valid cache entry.
    ///
    /// The disk replacement is written AFTER B has deterministically joined the
    /// slot (held state guard forces the joiner path; poll 1 confirms B is
    /// blocked on `state.lock().await`). This ensures the test actually
    /// exercises the joiner recovery branch rather than the initial fast-path
    /// `cached_hit`. Removing the joiner disk-recovery branch must make the
    /// test return Err(RefreshRejected) rather than Ok("sibling-replacement").
    ///
    /// Disk-dependent: the replacement lives on disk, so `write_private_cache`
    /// must be available (i.e. Unix only).
    #[cfg(unix)]
    #[tokio::test]
    async fn test_joiner_shared_failure_recovers_disk_replacement() {
        use std::future::Future as _;
        use std::pin::pin;
        use std::task::{Context, Poll, Waker};

        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://invalid.example.test/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;
        let replacement = CachedToken {
            access_token: "sibling-replacement".into(),
            refresh_token: Some("rt".into()),
            expires_at: Some(future_exp),
        };

        // Pre-install a slot for this key and publish the leader's terminal
        // failure — digest matches "rejected-bytes" so the joiner enters the
        // in-memory neutralization branch.
        let key: InflightKey = (source.lock_path(), AuthIntent::Headless);
        let slot = Arc::new(InflightSlot::new());
        inflight_registry().insert(key.clone(), slot.clone());
        slot.publish(
            digest_of(Some("rejected-bytes")),
            Err(AuthError::RefreshRejected),
        );

        // Hold the state mutex so the fast-path `try_lock` fails and B is
        // forced down the joiner path. The slot is already published, so
        // `slot.wait()` returns immediately; B then calls `state.lock().await`
        // and suspends while we hold the guard.
        let state_guard = source.state.lock().await;

        let mut b_fut = pin!(source.acquire(AuthIntent::Headless, Some("rejected-bytes")));
        let waker = Waker::noop();
        let mut cx = Context::from_waker(waker);

        // Poll 1: B falls through fast-path (try_lock fails), joins the
        // pre-published slot, enters the Err match arm, and blocks on
        // `state.lock().await` — structural proof B is on the joiner path.
        assert!(
            matches!(b_fut.as_mut().poll(&mut cx), Poll::Pending),
            "poll 1 must be Pending: B is blocked at state.lock().await after waking to Err"
        );

        // Now install the disk replacement. B is definitely past the initial
        // fast-path and will only see this token via `usable_from_disk` after
        // reconciliation — the recovery branch we are testing.
        fs::write(
            &source.cache_path,
            serde_json::to_vec(&replacement).unwrap(),
        )
        .unwrap();

        // Release the mutex. B acquires the lock, calls expire_rejected_memory
        // (empty state — no-op), then reads the disk replacement via
        // `usable_from_disk` and returns Ok("sibling-replacement").
        //
        // Mutation check: removing the `usable_from_disk` recovery branch
        // makes B return Err(RefreshRejected) instead — the assertion fails.
        drop(state_guard);

        let result = b_fut.await;
        inflight_registry().remove(&key);

        assert_eq!(
            result,
            Ok("sibling-replacement".to_string()),
            "the joiner must read the disk replacement and not inherit the shared failure — \
             mutation check: removing the usable_from_disk branch returns Err(RefreshRejected)"
        );
    }

    /// **Joiner failure cleanup must not modify the shared disk cache.**
    ///
    /// The matching-failure joiner calls `expire_rejected_memory` (in-process
    /// state only). It must not write, truncate, rename, or remove the disk
    /// cache. An independent process C may have persisted a valid replacement
    /// under the cross-process file lock between A's failure and B's
    /// reconciliation; an unfenced disk write from B would overwrite it.
    ///
    /// This test seeds X on disk, runs B as a joiner that wakes to a matching
    /// failure, and asserts the disk file is byte-for-byte unchanged afterward.
    ///
    /// Mutation check: reverting the joiner arm to call `expire_rejected`
    /// instead of `expire_rejected_memory` makes B read the disk file, see
    /// `access_token == "rejected-X"`, set `expires_at = 0`, and overwrite the
    /// file via `persist` or in-place truncate. The disk bytes change, and the
    /// "disk unchanged" assertion fails — proving the unfenced write is exactly
    /// the race that would overwrite any concurrent C write that landed between
    /// A's failure and B's reconciliation.
    #[cfg(unix)]
    #[tokio::test]
    async fn test_joiner_failure_does_not_write_disk() {
        use std::future::Future as _;
        use std::pin::pin;
        use std::task::{Context, Poll, Waker};

        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://invalid.example.test/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let b = PkceOAuthTokenSource::new(cfg).unwrap();

        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;

        // Seed X on disk. The constructor may not create the parent directory
        // without a pre-existing file, so ensure it exists first.
        let token_x = CachedToken {
            access_token: "rejected-X".into(),
            refresh_token: Some("live-refresh".into()),
            expires_at: Some(future_exp),
        };
        if let Some(parent) = b.cache_path.parent() {
            fs::create_dir_all(parent).unwrap();
        }
        let disk_before = serde_json::to_vec(&token_x).unwrap();
        fs::write(&b.cache_path, &disk_before).unwrap();

        // Pre-install a matching-failure slot (digest matches "rejected-X").
        let key: InflightKey = (b.lock_path(), AuthIntent::Headless);
        let slot = Arc::new(InflightSlot::new());
        inflight_registry().insert(key.clone(), slot.clone());
        slot.publish(
            digest_of(Some("rejected-X")),
            Err(AuthError::RefreshRejected),
        );

        // Hold B's state mutex: fast-path try_lock fails → joiner path;
        // state.lock().await during reconciliation blocks until we drop.
        let state_guard = b.state.lock().await;

        let mut b_fut = pin!(b.acquire(AuthIntent::Headless, Some("rejected-X")));
        let waker = Waker::noop();
        let mut cx = Context::from_waker(waker);

        // Poll 1: B falls through fast-path, joins the pre-published slot,
        // wakes to Err, and parks at state.lock().await.
        assert!(
            matches!(b_fut.as_mut().poll(&mut cx), Poll::Pending),
            "poll 1 must be Pending: B is parked at state.lock().await after waking to Err"
        );

        // Release the state guard. B acquires the lock, calls
        // expire_rejected_memory (in-memory neutralization only — no disk I/O),
        // then checks usable_from_disk. The disk token's access_token is
        // "rejected-X" which equals `rejected`, so usable_from_disk filters it
        // and returns None. B returns Err(RefreshRejected).
        drop(state_guard);

        let result = b_fut.await;
        inflight_registry().remove(&key);

        assert_eq!(
            result,
            Err(AuthError::RefreshRejected),
            "B must propagate the shared failure"
        );

        // The disk file must be byte-for-byte identical to what was seeded.
        // expire_rejected_memory must not have touched it.
        //
        // Mutation check: expire_rejected reads the disk file, finds
        // access_token == "rejected-X", sets expires_at = 0, and rewrites
        // the file. The bytes change and this assertion fails — proving the
        // unfenced write is the exact race that overwrites a concurrent C write
        // landing between A's failure and B's reconciliation.
        let disk_after = fs::read(&b.cache_path).unwrap();
        assert_eq!(
            disk_after, disk_before,
            "joiner failure cleanup must not modify the disk cache — \
             mutation check: expire_rejected rewrites the file (expires_at=0), \
             overwriting any concurrent write from process C"
        );
    }

    #[tokio::test]
    async fn test_bearer_headless_no_credential_is_terminal_without_browser() {
        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            // Unreachable discovery URL: if bearer() ever attempts discovery or
            // a browser flow, this test would hang or error differently. The
            // headless path must not touch either.
            discovery_url: "https://invalid.example.test/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        // Expire the in-memory state with no refresh token.
        {
            let mut state = source.state.lock().await;
            *state = Some(CachedToken {
                access_token: "stale".into(),
                refresh_token: None,
                expires_at: Some(0),
            });
        }

        // Write an expired, refresh-less token to disk too.
        let expired_token = CachedToken {
            access_token: "also-stale".into(),
            refresh_token: None,
            expires_at: Some(0),
        };
        let body = serde_json::to_vec_pretty(&expired_token).unwrap();
        fs::write(&source.cache_path, &body).unwrap();

        // bearer() is a Headless acquisition: past the cache checks with no
        // refresh token, it returns terminally instead of opening a browser.
        // With no refresh token it never even discovers endpoints, so the
        // unreachable URL is never contacted — the error is a graceful
        // LlmAuth, not a hard Llm/discovery error.
        match source.bearer().await.unwrap_err() {
            AgentError::LlmAuth(_) => {} // correct: terminal, no browser
            other => panic!("expected terminal LlmAuth, got: {other:?}"),
        }
    }

    /// `bearer_no_browser` with an empty cache and no refresh token must
    /// return `LlmAuth` immediately — it must NOT attempt OIDC discovery even
    /// when the `discovery_url` is unreachable/invalid, and must never browse.
    /// This guards the regression where `endpoints()` was called
    /// unconditionally before the refresh-token check, causing an `Llm` error
    /// (hard failure) instead of the intended graceful `LlmAuth` fallback.
    #[tokio::test]
    async fn test_bearer_no_browser_empty_cache_no_refresh_returns_llm_auth_without_discovery() {
        let dir = tempfile::tempdir().unwrap();
        // Intentionally invalid/unreachable discovery URL — if endpoints() is
        // called, the test will get an `Llm` error and the assertion below fails.
        let cfg = PkceOAuthConfig {
            discovery_url: "https://invalid.example.test/.well-known/oauth-authorization-server"
                .into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        // Empty in-memory state (no token, no refresh token).
        {
            let mut state = source.state.lock().await;
            *state = None;
        }

        // No disk cache file either — dir is empty.

        let result = source.bearer_no_browser().await;
        assert!(result.is_err(), "expected Err, got Ok");
        match result.unwrap_err() {
            AgentError::LlmAuth(_) => {} // correct: graceful fallback
            other => panic!(
                "expected LlmAuth (no discovery attempted), got: {other:?}\n\
                 This means endpoints() was called before the refresh-token check."
            ),
        }
    }

    // ---- callback HTML must never reflect input --------------------------

    #[test]
    fn test_callback_failure_page_omits_reflected_error_param() {
        // A hostile `error` query value carrying markup must not appear in
        // the served HTML — otherwise the localhost callback is an XSS sink.
        let payload = "<script>alert('xss')</script>";
        let mut params = std::collections::HashMap::new();
        params.insert("error".to_string(), payload.to_string());

        let (result, page) = callback_outcome(&params, "expected-state");

        // The failure detail still reaches the waiting flow via `result`...
        assert_eq!(result.as_ref().err().map(String::as_str), Some(payload));
        // ...but the browser page is static and inert.
        assert!(
            !page.contains(payload),
            "callback page reflected the raw error param: {page}"
        );
        assert!(
            !page.contains("<script>"),
            "callback page contains script tag"
        );
        assert!(
            page.contains("auth failed"),
            "unexpected failure page: {page}"
        );
    }

    #[test]
    fn test_callback_error_detail_strips_control_chars_and_caps_length() {
        // The `error` param feeds an error string that reaches the logs, so
        // CR/LF (log-line injection) must be neutralized and length bounded.
        let payload = format!("bad\r\nInjected: fake-log-line{}", "A".repeat(500));
        let mut params = std::collections::HashMap::new();
        params.insert("error".to_string(), payload);

        let (result, _page) = callback_outcome(&params, "expected-state");
        let detail = result.unwrap_err();

        assert!(
            !detail.contains('\r'),
            "carriage return survived: {detail:?}"
        );
        assert!(!detail.contains('\n'), "newline survived: {detail:?}");
        assert!(
            detail.len() <= 200,
            "detail not length-capped: {}",
            detail.len()
        );
        assert!(
            detail.starts_with("bad  Injected:"),
            "unexpected sanitized detail: {detail:?}"
        );
    }

    #[test]
    fn test_callback_state_mismatch_page_omits_reflected_code() {
        // A returned code with a mismatched state must be rejected, and the
        // page must not echo the (attacker-chosen) state or code values.
        let mut params = std::collections::HashMap::new();
        params.insert(
            "code".to_string(),
            "<img src=x onerror=alert(1)>".to_string(),
        );
        params.insert("state".to_string(), "attacker-state".to_string());

        let (result, page) = callback_outcome(&params, "expected-state");

        assert_eq!(
            result.as_ref().err().map(String::as_str),
            Some("state mismatch")
        );
        assert!(
            !page.contains("<img"),
            "callback page reflected the code param: {page}"
        );
    }

    #[test]
    fn test_callback_success_returns_code_and_static_page() {
        let mut params = std::collections::HashMap::new();
        params.insert("code".to_string(), "auth-code-123".to_string());
        params.insert("state".to_string(), "expected-state".to_string());

        let (result, page) = callback_outcome(&params, "expected-state");

        assert_eq!(
            result.as_ref().ok().map(String::as_str),
            Some("auth-code-123")
        );
        assert!(
            page.contains("signed in"),
            "unexpected success page: {page}"
        );
        // The success page is a fixed literal — no request data in it.
        assert!(!page.contains("auth-code-123"));
    }

    // ---- private atomic cache write --------------------------------------

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

        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://example.com/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        {
            let mut state = source.state.lock().await;
            source
                .save(
                    &mut state,
                    CachedToken {
                        access_token: "secret-access".into(),
                        refresh_token: Some("secret-refresh".into()),
                        expires_at: None,
                    },
                )
                .unwrap();
        }

        let mode = fs::metadata(&source.cache_path)
            .unwrap()
            .permissions()
            .mode();
        assert_eq!(
            mode & 0o777,
            0o600,
            "cache file must be owner-only, got {:o}",
            mode & 0o777
        );
        // No temp file left behind.
        let leftovers: Vec<_> = fs::read_dir(dir.path().join("test"))
            .unwrap()
            .filter_map(|e| e.ok())
            .filter(|e| e.file_name().to_string_lossy().ends_with(".tmp"))
            .collect();
        assert!(leftovers.is_empty(), "temp file leaked: {leftovers:?}");
    }

    #[cfg(unix)]
    #[tokio::test]
    async fn test_save_repairs_preexisting_loose_mode_file() {
        use std::os::unix::fs::{MetadataExt, PermissionsExt};

        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://example.com/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let source = PkceOAuthTokenSource::new(cfg).unwrap();

        // Simulate a world-readable cache written by an older buzz-agent.
        fs::create_dir_all(source.cache_path.parent().unwrap()).unwrap();
        fs::write(&source.cache_path, b"{\"access_token\":\"old\"}").unwrap();
        fs::set_permissions(&source.cache_path, fs::Permissions::from_mode(0o644)).unwrap();
        let old_inode = fs::metadata(&source.cache_path).unwrap().ino();

        {
            let mut state = source.state.lock().await;
            source
                .save(
                    &mut state,
                    CachedToken {
                        access_token: "new-access".into(),
                        refresh_token: Some("new-refresh".into()),
                        expires_at: None,
                    },
                )
                .unwrap();
        }

        let meta = fs::metadata(&source.cache_path).unwrap();
        assert_eq!(
            meta.permissions().mode() & 0o777,
            0o600,
            "loose-mode cache file was not tightened to 0600"
        );
        // The rename swapped the inode — the loose-mode file is gone, not
        // chmod'd in place, so no window where the new tokens inherit 0644.
        assert_ne!(meta.ino(), old_inode, "cache inode was not replaced");
    }

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

        // Unique tmp suffixes must let many concurrent writers to the SAME
        // destination each land a complete, private file — no tmp collision,
        // no failed rename. The fixed `*.json.tmp` name this replaces would
        // race (one writer's rename fails on the other's half-written tmp).
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("cache.json");
        let path = Arc::new(path);

        let handles: Vec<_> = (0..16)
            .map(|i| {
                let path = Arc::clone(&path);
                std::thread::spawn(move || {
                    let body = format!("{{\"n\":{i}}}");
                    write_private_cache(&path, body.as_bytes())
                })
            })
            .collect();

        for h in handles {
            h.join()
                .unwrap()
                .expect("concurrent write_private_cache failed");
        }

        // Destination exists, is valid (one writer's complete body), and 0600.
        let contents = fs::read_to_string(path.as_ref()).unwrap();
        let v: serde_json::Value = serde_json::from_str(&contents).unwrap();
        assert!(v.get("n").is_some(), "cache body was corrupted: {contents}");
        let mode = fs::metadata(path.as_ref()).unwrap().permissions().mode();
        assert_eq!(mode & 0o777, 0o600);

        // No temp files leaked.
        let leftovers: Vec<_> = fs::read_dir(dir.path())
            .unwrap()
            .filter_map(|e| e.ok())
            .filter(|e| e.file_name().to_string_lossy().ends_with(".tmp"))
            .collect();
        assert!(leftovers.is_empty(), "temp files leaked: {leftovers:?}");
    }

    // ---- owner-only is a load-time invariant, not just a save-time one ----

    #[cfg(unix)]
    #[tokio::test]
    async fn test_bearer_cache_hit_tightens_preexisting_loose_mode_file() {
        use std::os::unix::fs::{MetadataExt, PermissionsExt};

        // A world-readable cache left by an older buzz-agent must be tightened
        // the moment we load it — on the plain cache-hit path, with no refresh
        // or save. Otherwise the pre-bug population stays exposed indefinitely.
        let dir = tempfile::tempdir().unwrap();
        let cfg = PkceOAuthConfig {
            discovery_url: "https://example.com/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        };
        let cache_path = cache_path_for(&cfg).unwrap();
        fs::create_dir_all(cache_path.parent().unwrap()).unwrap();

        // A valid, unexpired token so bearer() takes the cache hit and never
        // touches the network or save path.
        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;
        let token = CachedToken {
            access_token: "loose-but-valid".into(),
            refresh_token: Some("rt".into()),
            expires_at: Some(future_exp),
        };
        fs::write(&cache_path, serde_json::to_vec_pretty(&token).unwrap()).unwrap();
        fs::set_permissions(&cache_path, fs::Permissions::from_mode(0o644)).unwrap();
        let old_inode = fs::metadata(&cache_path).unwrap().ino();

        // Constructing the source loads the cache — repair happens here.
        let source = PkceOAuthTokenSource::new(cfg).unwrap();
        let bearer = source.bearer().await.unwrap();
        assert_eq!(bearer, "loose-but-valid");

        let meta = fs::metadata(&cache_path).unwrap();
        assert_eq!(
            meta.permissions().mode() & 0o777,
            0o600,
            "existing loose-mode cache was not tightened on load"
        );
        // Repaired in place on the open fd — same inode, no rewrite/rename.
        assert_eq!(
            meta.ino(),
            old_inode,
            "load-time repair should fchmod in place, not replace the inode"
        );
    }

    #[cfg(unix)]
    #[test]
    fn test_read_cache_refuses_symlinked_cache_path() {
        // A cache path that is a symlink must be rejected, not followed — a
        // hostile symlink could redirect the read (and our chmod) at an
        // arbitrary file. `read_cache` returns None so the caller falls
        // through to a fresh flow instead of trusting the link target.
        let dir = tempfile::tempdir().unwrap();
        let target = dir.path().join("real-token.json");
        fs::write(&target, b"{\"access_token\":\"via-symlink\"}").unwrap();

        let link = dir.path().join("cache.json");
        std::os::unix::fs::symlink(&target, &link).unwrap();

        assert!(
            read_cache(&link).is_none(),
            "read_cache followed a symlinked cache path"
        );
    }

    // ---- cross-process advisory lock primitive --------------------------
    //
    // The full 165s waiter bound (`LOCK_WAIT_TIMEOUT`) is not exercisable in a
    // unit test, so these drive `acquire_auth_lock` with explicit deadlines to
    // pin the three properties the coordinator relies on: a contended waiter
    // times out (never blocks forever), a timeout leaves the *holder*
    // untouched (never cancels the in-flight attempt), and releasing the
    // holder — the RAII stand-in for a crashed process — lets a successor
    // proceed with no wedge and no lock-breaking.

    #[tokio::test]
    async fn test_lock_wait_times_out_and_leaves_holder_untouched() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("cache.json.lock");

        // Holder takes the lock with a generous deadline.
        let holder = acquire_auth_lock(&path, Instant::now() + Duration::from_secs(30))
            .await
            .expect("holder should acquire the free lock");

        // A waiter with an already-lapsed deadline must give up with
        // LockTimeout rather than block — this is the deadline-aware polling
        // that replaces a blocking `lock()`.
        let waiter = acquire_auth_lock(&path, Instant::now()).await;
        assert!(
            matches!(waiter, Err(AuthError::LockTimeout)),
            "contended waiter past its deadline must return LockTimeout, got {waiter:?}"
        );

        // The timeout did not cancel or steal the holder: a second immediate
        // waiter still cannot acquire, proving the holder is intact.
        let still_held = acquire_auth_lock(&path, Instant::now()).await;
        assert!(
            matches!(still_held, Err(AuthError::LockTimeout)),
            "holder must remain intact after a waiter times out, got {still_held:?}"
        );

        drop(holder);
    }

    #[tokio::test]
    async fn test_lock_timeout_leaves_cooldown_sidecar_byte_for_byte_untouched() {
        let dir = tempfile::tempdir().unwrap();
        let lock_path = dir.path().join("cache.json.lock");
        let cooldown_path = dir.path().join("cache.json.cooldown");

        // A pre-existing cooldown sidecar written by an earlier interactive
        // failure. A waiter that can't take the lock must return before any
        // code that reads/clears/writes the cooldown, so these exact bytes
        // survive untouched — otherwise a lock-contended caller could clear a
        // live suppression and let the next Auto caller re-pop a browser.
        let original = br#"{"code":"denied","until":9999999999}"#;
        fs::write(&cooldown_path, original).unwrap();

        // Holder owns the lock (RAII stand-in for another live process).
        let holder = acquire_auth_lock(&lock_path, Instant::now() + Duration::from_secs(30))
            .await
            .expect("holder should acquire the free lock");

        // A waiter past its deadline gives up with LockTimeout — the `?` in
        // `acquire_leader` propagates this before `acquire_locked` (which owns
        // every sidecar mutation) is ever entered.
        let waiter = acquire_auth_lock(&lock_path, Instant::now()).await;
        assert!(
            matches!(waiter, Err(AuthError::LockTimeout)),
            "contended waiter past its deadline must return LockTimeout, got {waiter:?}"
        );

        let after = fs::read(&cooldown_path).unwrap();
        assert_eq!(
            after.as_slice(),
            original.as_slice(),
            "a lock timeout must leave the cooldown sidecar byte-for-byte untouched"
        );

        drop(holder);
    }

    /// **Awaited reconciliation is falsifiable — `lock().await` cannot regress to `try_lock`.**
    ///
    /// Deterministic direct-poll proof: the test task holds B's state mutex and
    /// manually polls a pinned real `acquire()` future at each state transition,
    /// without spawning a task or relying on scheduler ordering.
    ///
    /// Proof sequence:
    /// 1. Seed B's state with stale X; register an unpublished slot.
    /// 2. Hold B's state mutex — blocks the fast-path `try_lock` so B falls
    ///    through to the registry, and will block `lock().await` when B tries
    ///    to reconcile after waking.
    /// 3. Poll B's `acquire()` once: no prior async suspension on the joiner
    ///    path, so B reaches `slot.wait()`'s inner `rx.changed().await` and
    ///    parks — the poll returns `Pending`. This is a structural proof, not a
    ///    scheduler assumption.
    /// 4. Publish Y and poll the same future again while the state mutex is
    ///    still held. `slot.wait()` wakes and returns; B calls
    ///    `state.lock().await`, which must park because we hold the mutex →
    ///    this poll returns `Pending`.
    ///    Mutation check: with `try_lock()` the adopt block is skipped and B
    ///    returns immediately → this poll returns `Ready(Ok("token-Y"))`,
    ///    failing the `Pending` assertion.
    /// 5. Release the state guard; poll to completion (or `await` the future)
    ///    and assert the result is `Ok("token-Y")`.
    /// 6. Assert a subsequent plain `acquire(None)` returns Y from the
    ///    in-memory cache — the P1 contract.
    ///    Mutation check: `try_lock` leaves state == stale X, so this acquire
    ///    returns X — the exact P1 stale-credential regression.
    #[tokio::test]
    async fn test_joiner_reconciliation_blocked_until_state_lock_released() {
        use std::future::Future as _;
        use std::pin::pin;
        use std::task::{Context, Poll, Waker};

        let dir = tempfile::tempdir().unwrap();
        let b = PkceOAuthTokenSource::new(PkceOAuthConfig {
            discovery_url: "https://invalid.example.test/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        })
        .unwrap();

        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;
        let make_token = |access: &str| CachedToken {
            access_token: access.into(),
            refresh_token: Some("rt".into()),
            expires_at: Some(future_exp),
        };

        let token_x = make_token("token-X"); // B's stale/rejected credential.
        let token_y = make_token("token-Y"); // shared leader result — must replace X.

        // Seed B's state with stale X.
        {
            let mut state = b.state.lock().await;
            *state = Some(token_x.clone());
        }

        // Register an unpublished slot so B will join it.
        let key: InflightKey = (b.lock_path(), AuthIntent::Headless);
        let slot = Arc::new(InflightSlot::new());
        inflight_registry().insert(key.clone(), slot.clone());

        // Hold B's state mutex.
        //   (a) The fast-path `try_lock` fails → B falls through to the joiner path.
        //   (b) `state.lock().await` during reconciliation will block until we drop.
        let state_guard = b.state.lock().await;

        // Pin B's acquire() future in this stack frame for manual polling.
        let mut b_fut = pin!(b.acquire(AuthIntent::Headless, Some("token-X")));
        let waker = Waker::noop();
        let mut cx = Context::from_waker(waker);

        // Poll 1: B has no async suspension before `slot.wait()`'s inner
        // `rx.changed().await`. The slot is unpublished, so `changed()` parks.
        // Result must be Pending — structural proof that B reached slot.wait().
        assert!(
            matches!(b_fut.as_mut().poll(&mut cx), Poll::Pending),
            "poll 1 must be Pending: B is parked at slot.wait() awaiting publication"
        );

        // Publish Y. `rx.changed()` wakes; on the next poll B exits slot.wait(),
        // enters reconciliation, and calls `state.lock().await`.
        slot.publish(None, Ok(token_y.clone()));
        inflight_registry().remove(&key);

        // Poll 2: `slot.wait()` returns Y; B calls `state.lock().await`.
        // With `lock().await`: the mutex is held → parks → Pending.
        // Mutation (`try_lock`): try_lock fails → adopt skipped → B returns
        // Ok("token-Y") immediately → Ready, not Pending.
        //
        // This poll is the exact mutation discriminator: Ready here is the
        // bug (B completed without awaited reconciliation).
        assert!(
            matches!(b_fut.as_mut().poll(&mut cx), Poll::Pending),
            "poll 2 must be Pending: B must not return while state mutex is held — \
             mutation check: `try_lock()` returns Ready here, proving early completion \
             without reconciliation (the P1 regression)"
        );

        // Release the mutex. B acquires the lock, evaluates the adoption
        // predicate (state == stale X, matches the rejected token), writes Y,
        // and returns Ok("token-Y").
        drop(state_guard);

        // Await completion (B now owns the mutex).
        let result = b_fut.await;
        assert_eq!(
            result,
            Ok("token-Y".to_string()),
            "B must return the shared token Y after reconciliation completes"
        );

        // Subsequent plain acquire must return Y from the in-memory cache —
        // the P1 contract. With the `try_lock` mutation, state still holds X
        // and this acquire returns X (stale-credential regression).
        let rb_next = b
            .acquire(AuthIntent::Headless, None)
            .await
            .expect("subsequent acquire must return Y from in-memory state");
        assert_eq!(
            rb_next, "token-Y",
            "subsequent in-memory read must return Y, not stale X — \
             mutation check: `try_lock()` leaves state == X, returning X"
        );
    }

    /// **Preserve-distinct-newer — reconciliation must not overwrite B's valid credential.**
    ///
    /// B already holds a valid, usable token Z (distinct from rejected X and from the
    /// leader's shared result Y) in its `state` when the joiner reconciliation runs.
    /// The adoption predicate must evaluate to false for Z and leave it in place.
    ///
    /// Deterministic setup via direct polling: register an unpublished slot; poll
    /// B's `acquire()` once to park it at `slot.wait()`; write Z into B's state;
    /// publish Y and await completion. No scheduler inference or `yield_now()`.
    ///
    /// Mutation check (unconditional adoption): if the reconciliation block writes
    /// `*state = Some(token.clone())` unconditionally, Z is overwritten with Y.
    /// The subsequent state assertion `state == Z` FAILS — proving the predicate
    /// is load-bearing.
    #[tokio::test]
    async fn test_joiner_preserve_distinct_newer_credential() {
        use std::future::Future as _;
        use std::pin::pin;
        use std::task::{Context, Poll, Waker};

        let dir = tempfile::tempdir().unwrap();
        let b = PkceOAuthTokenSource::new(PkceOAuthConfig {
            discovery_url: "https://invalid.example.test/.well-known".into(),
            client_id: "test-client".into(),
            scopes: vec!["offline_access".into()],
            cache_namespace: "test".into(),
            cache_dir_override: Some(dir.path().to_path_buf()),
        })
        .unwrap();

        let future_exp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs()
            + 7200;
        let make_token = |access: &str| CachedToken {
            access_token: access.into(),
            refresh_token: Some("rt".into()),
            expires_at: Some(future_exp),
        };

        let token_z = make_token("token-Z"); // B's distinct, independently acquired credential.
        let token_y = make_token("token-Y"); // leader's shared result — must NOT overwrite Z.

        // Register a not-yet-published slot so B will join it and wait.
        // B starts with empty state so its fast-path cache miss is guaranteed.
        let key: InflightKey = (b.lock_path(), AuthIntent::Headless);
        let slot = Arc::new(InflightSlot::new());
        inflight_registry().insert(key.clone(), slot.clone());

        // Pin B's future and poll once to park it at slot.wait().
        // No async suspension precedes slot.wait() on the joiner path, so the
        // first poll is the structural proof that B is parked there.
        let mut b_fut = pin!(b.acquire(AuthIntent::Headless, Some("token-X")));
        let waker = Waker::noop();
        let mut cx = Context::from_waker(waker);
        assert!(
            matches!(b_fut.as_mut().poll(&mut cx), Poll::Pending),
            "B must park at slot.wait() on the first poll"
        );

        // B is now suspended in slot.wait(). Write Z into B's state — this is an
        // intervening write that B will observe when it evaluates the
        // reconciliation predicate after waking.
        {
            let mut state = b.state.lock().await;
            *state = Some(token_z.clone());
        }

        // Publish Y to wake B. B will call lock().await, see Z (not expired, not
        // matching "token-X"), evaluate the predicate as false, and preserve Z.
        slot.publish(None, Ok(token_y.clone()));
        inflight_registry().remove(&key);

        let result = b_fut.await;

        assert_eq!(
            result,
            Ok("token-Y".to_string()),
            "B must still receive the shared bearer Y"
        );

        // B.state must still hold Z — the adoption predicate correctly skipped
        // the write because Z is usable and distinct from the rejected token.
        {
            let state = b.state.lock().await;
            assert_eq!(
                state.as_ref().map(|t| t.access_token.as_str()),
                Some("token-Z"),
                "B.state must not be overwritten when it holds a distinct usable credential — \
                 mutation check: fails if reconciliation is unconditional \
                 (overwrites Z with Y regardless of predicate)"
            );
        }
    }
}