link-assistant-router 1.4.6

//! Provider-neutral extraction of stable account-routing signals.

use std::time::Duration;

/// Maximum operational cooldown accepted from an untrusted upstream hint.
pub const MAX_RETRY_AFTER: Duration = Duration::from_secs(24 * 60 * 60);

#[must_use]
pub fn bounded_retry_after(duration: Duration) -> Duration {
    duration.min(MAX_RETRY_AFTER)
}

use axum::http::HeaderMap;

use crate::accounts::RoutingContext;

/// Private carrier retaining the full selected credential for later evidence.
pub struct ResolvedUpstreamCredential {
    pub access_token: String,
    pub account: Option<String>,
    pub evidence_token: Option<crate::subscription::SubscriptionToken>,
}

/// Copy only stable routing signals from a request. Header signals take
/// precedence over JSON metadata and the caller token's account binding is
/// carried separately as a strict pin.
pub fn request_routing_context(
    headers: &HeaderMap,
    body: &serde_json::Value,
    pinned_account: Option<String>,
) -> RoutingContext {
    const SESSION_HEADERS: [&str; 4] = [
        "x-claude-code-session-id",
        "x-codex-session-id",
        "x-session-id",
        "session-id",
    ];
    let header_session = SESSION_HEADERS.iter().find_map(|name| {
        headers
            .get(*name)
            .and_then(|value| value.to_str().ok())
            .filter(|value| !value.is_empty())
            .map(ToString::to_string)
    });
    let body_session = [
        "/context/session_id",
        "/metadata/session_id",
        "/metadata/conversation_id",
        "/session_id",
        "/conversation_id",
        "/prompt_cache_key",
    ]
    .iter()
    .find_map(|pointer| {
        body.pointer(pointer)
            .and_then(serde_json::Value::as_str)
            .filter(|value| !value.is_empty())
            .map(ToString::to_string)
    });
    RoutingContext {
        session_key: header_session.or(body_session),
        pinned_account,
    }
}

/// Parse the standard `Retry-After` delta-seconds or HTTP-date forms.
pub fn retry_after_duration(headers: &HeaderMap) -> Option<Duration> {
    let value = headers.get("retry-after")?.to_str().ok()?.trim();
    if let Ok(seconds) = value.parse::<u64>() {
        return Some(bounded_retry_after(Duration::from_secs(seconds)));
    }
    let retry_at = chrono::DateTime::parse_from_rfc2822(value).ok()?;
    let seconds = retry_at
        .signed_duration_since(chrono::Utc::now())
        .num_seconds()
        .max(0);
    Some(bounded_retry_after(Duration::from_secs(
        u64::try_from(seconds).ok()?,
    )))
}

/// Record what the Anthropic upstream just said about the Claude credential.
///
/// The response status is the only authority on whether a credential still
/// works; see [`crate::refresh::CredentialEvidence`].
pub async fn record_claude_evidence(
    state: &crate::app_state::AppState,
    account: Option<&str>,
    credential: Option<&crate::subscription::SubscriptionToken>,
    status: u16,
) {
    let cache = &state.subscription_cache;
    let account = account.unwrap_or(crate::credential_recovery_store::PRIMARY_ACCOUNT);
    if let Some(credential) = credential {
        cache
            .record_status_for_credential(
                crate::subscription::SubscriptionProvider::Claude,
                account,
                credential,
                status,
            )
            .await;
    } else {
        // Preserve the legacy unregistered OAuth provider exactly. It returns
        // only a bearer string, so there is no authoritative generation to
        // compare and no registered store that could reconcile a replacement.
        cache.record_status_for(
            crate::subscription::SubscriptionProvider::Claude,
            account,
            status,
        );
    }
}