allowthem-core 0.0.9

use base64ct::{Base64UrlUnpadded, Encoding};
use chrono::{DateTime, Utc};
use rand::TryRngCore;
use rand::rngs::OsRng;
use serde::Serialize;
use url::Url;

use crate::db::Db;
use crate::error::AuthError;
use crate::types::{
    AccentInk, ApplicationId, ClientId, ClientSecret, ClientType, Mode, PasswordHash,
    SplashPrimitive, UserId,
};

/// An OIDC client application registered with allowthem.
///
/// `client_secret_hash` is skipped during serialization — the raw secret
/// is returned once at creation and is never retrievable again.
#[derive(Debug, Clone, Serialize, sqlx::FromRow)]
pub struct Application {
    pub id: ApplicationId,
    pub name: String,
    pub client_id: ClientId,
    pub client_type: ClientType,
    #[serde(skip_serializing)]
    pub client_secret_hash: Option<PasswordHash>,
    pub redirect_uris: String, // JSON array, parsed at the call site
    pub logo_url: Option<String>,
    pub primary_color: Option<String>,
    // Wave Funk branding fields (all optional).
    pub accent_hex: Option<String>,
    pub accent_ink: Option<AccentInk>,
    pub forced_mode: Option<Mode>,
    pub font_css_url: Option<String>,
    pub font_family: Option<String>,
    pub splash_text: Option<String>,
    pub splash_image_url: Option<String>,
    pub splash_primitive: Option<SplashPrimitive>,
    pub splash_url: Option<String>,
    pub shader_cell_scale: Option<i64>,
    pub is_trusted: bool,
    pub created_by: Option<UserId>,
    pub is_active: bool,
    pub created_at: DateTime<Utc>,
    pub updated_at: DateTime<Utc>,
}

/// Branding configuration for an application's hosted auth pages.
///
/// Extracted from `Application` — contains only the fields needed to
/// theme login, register, consent, and other OIDC-flow pages.
///
/// Derives `sqlx::FromRow` for use with `query_as` in
/// `get_branding_by_client_id`. The SQL query aliases `name` to
/// `application_name` to match the struct field.
#[derive(Debug, Clone, Serialize, sqlx::FromRow)]
pub struct BrandingConfig {
    pub application_name: String,
    pub logo_url: Option<String>,
    pub primary_color: Option<String>,
    pub accent_hex: Option<String>,
    pub accent_ink: Option<AccentInk>,
    pub forced_mode: Option<Mode>,
    pub font_css_url: Option<String>,
    pub font_family: Option<String>,
    pub splash_text: Option<String>,
    pub splash_image_url: Option<String>,
    pub splash_primitive: Option<SplashPrimitive>,
    pub splash_url: Option<String>,
    pub shader_cell_scale: Option<i64>,
    /// Override the `— allowthem` suffix in page titles.
    /// Defaults to `application_name` when `None`.
    #[sqlx(skip)]
    pub title_brand: Option<String>,
}

/// Generate a new `client_id`: `ath_` + 24 random bytes base64url-encoded.
///
/// Produces a 36-character string (`ath_` + 32 base64url chars). 192 bits of
/// entropy from `OsRng` makes collision effectively impossible.
pub fn generate_client_id() -> ClientId {
    let mut bytes = [0u8; 24];
    OsRng
        .try_fill_bytes(&mut bytes)
        .expect("OS RNG unavailable");
    let encoded = Base64UrlUnpadded::encode_string(&bytes);
    ClientId::new_unchecked(format!("ath_{encoded}"))
}

/// Generate a new client secret and its Argon2 hash.
///
/// Returns `(raw_secret, hash)`. The raw secret is shown once to the admin
/// and must never be stored. The hash is stored as `client_secret_hash`.
/// Reuses `password::hash_password` — a client secret is functionally a
/// high-entropy password and the security requirements are identical.
pub fn generate_client_secret() -> Result<(ClientSecret, PasswordHash), AuthError> {
    let mut bytes = [0u8; 32];
    OsRng
        .try_fill_bytes(&mut bytes)
        .expect("OS RNG unavailable");
    let raw = Base64UrlUnpadded::encode_string(&bytes);
    let hash = crate::password::hash_password(&raw)?;
    Ok((ClientSecret::new_unchecked(raw), hash))
}

impl Application {
    /// Parse the stored JSON `redirect_uris` string into a `Vec<String>`.
    ///
    /// Returns `AuthError::Database` if the stored value is malformed JSON.
    /// This indicates a data integrity error — the core layer always validates
    /// and serializes URIs correctly on write.
    pub fn redirect_uri_list(&self) -> Result<Vec<String>, AuthError> {
        serde_json::from_str(&self.redirect_uris)
            .map_err(|e| AuthError::Database(sqlx::Error::Decode(Box::new(e))))
    }

    /// Extract the branding configuration for use in themed auth pages.
    pub fn branding(&self) -> BrandingConfig {
        BrandingConfig {
            application_name: self.name.clone(),
            logo_url: self.logo_url.clone(),
            primary_color: self.primary_color.clone(),
            accent_hex: self.accent_hex.clone(),
            accent_ink: self.accent_ink,
            forced_mode: self.forced_mode,
            font_css_url: self.font_css_url.clone(),
            font_family: self.font_family.clone(),
            splash_text: self.splash_text.clone(),
            splash_image_url: self.splash_image_url.clone(),
            splash_primitive: self.splash_primitive,
            splash_url: self.splash_url.clone(),
            shader_cell_scale: self.shader_cell_scale,
            title_brand: None,
        }
    }
}

impl BrandingConfig {
    /// Construct an all-defaults `BrandingConfig` with only the required
    /// `application_name` set. Embedders use this as a starting point:
    /// `BrandingConfig::new("Transfer These Files").with_accent("#ff7a1a", AccentInk::Black)`.
    pub fn new(application_name: impl Into<String>) -> Self {
        Self {
            application_name: application_name.into(),
            logo_url: None,
            primary_color: None,
            accent_hex: None,
            accent_ink: None,
            forced_mode: None,
            font_css_url: None,
            font_family: None,
            splash_text: None,
            splash_image_url: None,
            splash_primitive: None,
            splash_url: None,
            shader_cell_scale: None,
            title_brand: None,
        }
    }

    pub fn with_accent(mut self, hex: impl Into<String>, ink: AccentInk) -> Self {
        self.accent_hex = Some(hex.into());
        self.accent_ink = Some(ink);
        self
    }

    pub fn with_primary_color(mut self, hex: impl Into<String>) -> Self {
        self.primary_color = Some(hex.into());
        self
    }

    pub fn with_logo_url(mut self, url: impl Into<String>) -> Self {
        self.logo_url = Some(url.into());
        self
    }

    pub fn with_splash_text(mut self, text: impl Into<String>) -> Self {
        self.splash_text = Some(text.into());
        self
    }

    pub fn with_splash_image_url(mut self, url: impl Into<String>) -> Self {
        self.splash_image_url = Some(url.into());
        self
    }

    pub fn with_splash_primitive(mut self, primitive: SplashPrimitive) -> Self {
        self.splash_primitive = Some(primitive);
        self
    }

    pub fn with_shader_cell_scale(mut self, scale: i64) -> Self {
        self.shader_cell_scale = Some(scale);
        self
    }

    pub fn with_title_brand(mut self, brand: impl Into<String>) -> Self {
        self.title_brand = Some(brand.into());
        self
    }
}

fn map_unique_violation(err: sqlx::Error) -> AuthError {
    if let sqlx::Error::Database(ref db_err) = err {
        let msg = db_err.message();
        if msg.contains("UNIQUE constraint failed") && msg.contains("client_id") {
            return AuthError::Conflict("client_id already exists".into());
        }
    }
    AuthError::Database(err)
}

/// Opaque keyset cursor for paginating `list_applications_paginated`.
///
/// Encodes `(created_at, id)` as a base64url-encoded JSON blob. Callers
/// treat the encoded string as opaque and pass it back verbatim.
pub struct ApplicationCursor {
    pub created_at: DateTime<Utc>,
    pub id: ApplicationId,
}

#[derive(serde::Serialize, serde::Deserialize)]
struct RawCursor {
    ca: String,
    id: String,
}

impl ApplicationCursor {
    pub fn from_app(app: &Application) -> Self {
        Self {
            created_at: app.created_at,
            id: app.id,
        }
    }

    pub fn encode(&self) -> String {
        let raw = RawCursor {
            ca: self.created_at.to_rfc3339(),
            id: self.id.to_string(),
        };
        let json = serde_json::to_string(&raw).expect("RawCursor serializes");
        Base64UrlUnpadded::encode_string(json.as_bytes())
    }

    pub fn decode(s: &str) -> Option<Self> {
        let bytes = Base64UrlUnpadded::decode_vec(s).ok()?;
        let raw: RawCursor = serde_json::from_slice(&bytes).ok()?;
        let created_at = chrono::DateTime::parse_from_rfc3339(&raw.ca)
            .ok()?
            .with_timezone(&Utc);
        let id = raw
            .id
            .parse::<uuid::Uuid>()
            .ok()
            .map(ApplicationId::from_uuid)?;
        Some(Self { created_at, id })
    }
}

/// Parameters for registering a new OIDC application via [`Db::create_application`].
pub struct CreateApplicationParams {
    pub name: String,
    pub client_type: ClientType,
    pub redirect_uris: Vec<String>,
    pub is_trusted: bool,
    pub created_by: Option<UserId>,
    pub logo_url: Option<String>,
    pub primary_color: Option<String>,
    pub accent_hex: Option<String>,
    pub accent_ink: Option<AccentInk>,
    pub forced_mode: Option<Mode>,
    pub font_css_url: Option<String>,
    pub font_family: Option<String>,
    pub splash_text: Option<String>,
    pub splash_image_url: Option<String>,
    pub splash_primitive: Option<SplashPrimitive>,
    pub splash_url: Option<String>,
    pub shader_cell_scale: Option<i64>,
}

/// Parameters for updating an application's mutable fields.
///
/// All fields are required. Fetch the current application first
/// to populate fields that should remain unchanged.
pub struct UpdateApplication {
    pub name: String,
    pub redirect_uris: Vec<String>,
    pub is_trusted: bool,
    pub is_active: bool,
    pub logo_url: Option<String>,
    pub primary_color: Option<String>,
    pub accent_hex: Option<String>,
    pub accent_ink: Option<AccentInk>,
    pub forced_mode: Option<Mode>,
    pub font_css_url: Option<String>,
    pub font_family: Option<String>,
    pub splash_text: Option<String>,
    pub splash_image_url: Option<String>,
    pub splash_primitive: Option<SplashPrimitive>,
    pub splash_url: Option<String>,
    pub shader_cell_scale: Option<i64>,
}

/// Validate a list of redirect URIs for registration (create or update).
///
/// Rules (per RFC 6749 and RFC 8252):
/// - List must not be empty.
/// - Each URI must parse as an absolute URL (has a scheme).
/// - No fragment component — prohibited by RFC 6749 Section 3.1.2.
/// - HTTPS required, except `http://localhost` and `http://127.0.0.1`
///   (loopback URIs permitted per RFC 8252 Section 8.3).
///
/// Returns `AuthError::InvalidRedirectUri` with the offending URI on first failure.
pub fn validate_redirect_uris(uris: &[String]) -> Result<(), AuthError> {
    if uris.is_empty() {
        return Err(AuthError::InvalidRedirectUri(
            "redirect_uris must not be empty".into(),
        ));
    }
    for uri in uris {
        let parsed = Url::parse(uri).map_err(|_| AuthError::InvalidRedirectUri(uri.clone()))?;
        if parsed.fragment().is_some() {
            return Err(AuthError::InvalidRedirectUri(uri.clone()));
        }
        let scheme = parsed.scheme();
        if scheme == "https" {
            continue;
        }
        if scheme == "http" {
            let host = parsed.host_str().unwrap_or("");
            if host == "localhost" || host == "127.0.0.1" {
                continue;
            }
        }
        return Err(AuthError::InvalidRedirectUri(uri.clone()));
    }
    Ok(())
}

/// Validate that `redirect_uri` exactly matches one of the registered URIs.
///
/// Used by the authorization endpoint (M39) to reject unregistered redirect targets.
/// Exact string match — no normalization, no wildcard expansion.
///
/// Returns `AuthError::InvalidRedirectUri` if `redirect_uri` is not in `registered`.
pub fn validate_redirect_uri(redirect_uri: &str, registered: &[String]) -> Result<(), AuthError> {
    if registered.iter().any(|r| r == redirect_uri) {
        Ok(())
    } else {
        Err(AuthError::InvalidRedirectUri(redirect_uri.to_owned()))
    }
}

/// Validate a logo URL for branding.
///
/// Must be an absolute URL with HTTPS scheme. HTTP is permitted for
/// localhost and 127.0.0.1 (development loopback exception, matching
/// redirect URI validation).
pub fn validate_logo_url(url: &str) -> Result<(), AuthError> {
    let parsed = Url::parse(url)
        .map_err(|_| AuthError::Validation("logo_url must be a valid absolute URL".into()))?;
    let scheme = parsed.scheme();
    if scheme == "https" {
        return Ok(());
    }
    if scheme == "http" {
        let host = parsed.host_str().unwrap_or("");
        if host == "localhost" || host == "127.0.0.1" {
            return Ok(());
        }
    }
    Err(AuthError::Validation(
        "logo_url must be an HTTPS URL".into(),
    ))
}

/// Validate a font CSS URL. Must be an HTTPS URL (no loopback exception —
/// production asset URL).
pub fn validate_font_css_url(url: &str) -> Result<(), AuthError> {
    validate_https_url(url, "font_css_url")
}

/// Validate a splash image URL. Must be an HTTPS URL (no loopback exception).
pub fn validate_splash_image_url(url: &str) -> Result<(), AuthError> {
    validate_https_url(url, "splash_image_url")
}

/// Validate a splash URL. Must be an HTTPS URL (no loopback exception).
pub fn validate_splash_url(url: &str) -> Result<(), AuthError> {
    validate_https_url(url, "splash_url")
}

/// Shared HTTPS-only URL validator used by branding asset URL fields.
///
/// Unlike `validate_logo_url`, no loopback exception — these fields are
/// intended for production assets only.
fn validate_https_url(url: &str, field: &str) -> Result<(), AuthError> {
    let parsed = Url::parse(url)
        .map_err(|_| AuthError::Validation(format!("{field} must be a valid absolute URL")))?;
    if parsed.scheme() != "https" {
        return Err(AuthError::Validation(format!(
            "{field} must be an HTTPS URL"
        )));
    }
    Ok(())
}

/// Shared hex color validator used by branding color fields.
///
/// Accepts `#RRGGBB` (7 chars: `#` + 6 hex digits). Safe for injection
/// into CSS `color` / `background` declarations without escaping.
fn validate_hex_color(color: &str, field: &str) -> Result<(), AuthError> {
    let bytes = color.as_bytes();
    if bytes.len() != 7 || bytes[0] != b'#' || !bytes[1..].iter().all(|b| b.is_ascii_hexdigit()) {
        return Err(AuthError::Validation(format!(
            "{field} must be a hex color (#RRGGBB)"
        )));
    }
    Ok(())
}

/// Validate a primary color for Wave Funk branding.
///
/// Must be a 7-character CSS hex color: `#` followed by exactly 6 hex
/// digits (e.g., `#3B82F6`). This format is safe for injection into
/// HTML `style` attributes without escaping.
pub fn validate_primary_color(color: &str) -> Result<(), AuthError> {
    validate_hex_color(color, "primary_color")
}

/// Validate an accent color for Wave Funk branding.
///
/// Same format as `validate_primary_color` — `#RRGGBB` (7 chars, `#` + 6 hex
/// digits). Safe for injection into CSS `color` / `background` declarations
/// without escaping.
pub fn validate_accent_hex(color: &str) -> Result<(), AuthError> {
    validate_hex_color(color, "accent_hex")
}

impl Db {
    /// Register a new OIDC application.
    ///
    /// Generates a `client_id` and `client_secret`, hashes the secret, and inserts
    /// the row. Returns the persisted `Application` and the raw `ClientSecret`.
    /// The raw secret is shown once and is not recoverable — the caller must present
    /// it to the admin immediately.
    ///
    /// Validates `redirect_uris` before inserting. Returns `AuthError::InvalidRedirectUri`
    /// if any URI fails validation.
    pub async fn create_application(
        &self,
        params: CreateApplicationParams,
    ) -> Result<(Application, Option<ClientSecret>), AuthError> {
        let CreateApplicationParams {
            name,
            client_type,
            redirect_uris,
            is_trusted,
            created_by,
            logo_url,
            primary_color,
            accent_hex,
            accent_ink,
            forced_mode,
            font_css_url,
            font_family,
            splash_text,
            splash_image_url,
            splash_primitive,
            splash_url,
            shader_cell_scale,
        } = params;
        validate_redirect_uris(&redirect_uris)?;
        if let Some(ref url) = logo_url {
            validate_logo_url(url)?;
        }
        if let Some(ref color) = primary_color {
            validate_primary_color(color)?;
        }
        if let Some(ref hex) = accent_hex {
            validate_accent_hex(hex)?;
        }
        if let Some(ref url) = font_css_url {
            validate_font_css_url(url)?;
        }
        if let Some(ref url) = splash_image_url {
            validate_splash_image_url(url)?;
        }
        if let Some(ref url) = splash_url {
            validate_splash_url(url)?;
        }
        let id = ApplicationId::new();
        let client_id = generate_client_id();
        let (raw_secret, hash) = match client_type {
            ClientType::Confidential => {
                let (secret, hash) = generate_client_secret()?;
                (Some(secret), Some(hash))
            }
            ClientType::Public => (None, None),
        };
        let redirect_uris_json =
            serde_json::to_string(&redirect_uris).expect("Vec<String> serializes to JSON");
        let now = Utc::now().format("%Y-%m-%dT%H:%M:%S%.3fZ").to_string();

        sqlx::query(
            "INSERT INTO allowthem_applications \
             (id, name, client_id, client_type, client_secret_hash, redirect_uris, logo_url, \
              primary_color, \
              accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
              splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
              is_trusted, created_by, is_active, created_at, updated_at) \
             VALUES \
             (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, \
              ?9, ?10, ?11, ?12, ?13, \
              ?14, ?15, ?16, ?17, ?18, \
              ?19, ?20, 1, ?21, ?21)",
        )
        .bind(id)
        .bind(&name)
        .bind(&client_id)
        .bind(client_type)
        .bind(&hash)
        .bind(&redirect_uris_json)
        .bind(&logo_url)
        .bind(&primary_color)
        .bind(&accent_hex)
        .bind(accent_ink.map(|v| v.as_str()))
        .bind(forced_mode.map(|v| v.as_str()))
        .bind(&font_css_url)
        .bind(&font_family)
        .bind(&splash_text)
        .bind(&splash_image_url)
        .bind(splash_primitive.map(|v| v.as_str()))
        .bind(&splash_url)
        .bind(shader_cell_scale)
        .bind(is_trusted)
        .bind(created_by)
        .bind(&now)
        .execute(self.pool())
        .await
        .map_err(map_unique_violation)?;

        let app = self.get_application(id).await?;
        Ok((app, raw_secret))
    }

    /// Get an application by internal ID.
    pub async fn get_application(&self, id: ApplicationId) -> Result<Application, AuthError> {
        sqlx::query_as::<_, Application>(
            "SELECT id, name, client_id, client_type, client_secret_hash, redirect_uris, \
             logo_url, primary_color, \
             accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
             splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
             is_trusted, created_by, is_active, created_at, updated_at \
             FROM allowthem_applications WHERE id = ?",
        )
        .bind(id)
        .fetch_optional(self.pool())
        .await?
        .ok_or(AuthError::NotFound)
    }

    /// Get an application by its public client_id.
    ///
    /// Used by OAuth endpoints that receive client_id in request parameters.
    pub async fn get_application_by_client_id(
        &self,
        client_id: &ClientId,
    ) -> Result<Application, AuthError> {
        sqlx::query_as::<_, Application>(
            "SELECT id, name, client_id, client_type, client_secret_hash, redirect_uris, \
             logo_url, primary_color, \
             accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
             splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
             is_trusted, created_by, is_active, created_at, updated_at \
             FROM allowthem_applications WHERE client_id = ?",
        )
        .bind(client_id)
        .fetch_optional(self.pool())
        .await?
        .ok_or(AuthError::NotFound)
    }

    /// Get branding configuration for an application by client_id.
    ///
    /// Returns `None` if no application with the given `client_id` exists
    /// or if the application is inactive. Branded pages fall back to
    /// default allowthem styling when this returns `None`.
    pub async fn get_branding_by_client_id(
        &self,
        client_id: &ClientId,
    ) -> Result<Option<BrandingConfig>, AuthError> {
        sqlx::query_as::<_, BrandingConfig>(
            "SELECT name AS application_name, logo_url, primary_color, \
             accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
             splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale \
             FROM allowthem_applications \
             WHERE client_id = ? AND is_active = 1",
        )
        .bind(client_id)
        .fetch_optional(self.pool())
        .await
        .map_err(AuthError::Database)
    }

    /// Count of all applications in the tenant DB. Used by the SaaS
    /// super-admin tenant detail panel (99c.6 §6.1).
    pub async fn count_applications(&self) -> Result<u64, AuthError> {
        let n: i64 = sqlx::query_scalar("SELECT COUNT(*) FROM allowthem_applications")
            .fetch_one(self.pool())
            .await
            .map_err(AuthError::Database)?;
        Ok(n as u64)
    }

    /// Count of distinct users who have consented to this application.
    ///
    /// Backed by `allowthem_consents`, which has `UNIQUE(user_id,
    /// application_id)`, so `COUNT(*)` is exactly the distinct-user count.
    /// The semantic is "users who have authorized this app" — which is what
    /// the SaaS dashboard surfaces as "Connected users."
    pub async fn count_users_for_application(&self, id: ApplicationId) -> Result<u64, AuthError> {
        let count: i64 =
            sqlx::query_scalar("SELECT COUNT(*) FROM allowthem_consents WHERE application_id = ?1")
                .bind(id)
                .fetch_one(self.pool())
                .await
                .map_err(AuthError::Database)?;
        Ok(count as u64)
    }

    /// List all applications ordered by `created_at ASC`.
    pub async fn list_applications(&self) -> Result<Vec<Application>, AuthError> {
        sqlx::query_as::<_, Application>(
            "SELECT id, name, client_id, client_type, client_secret_hash, redirect_uris, \
             logo_url, primary_color, \
             accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
             splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
             is_trusted, created_by, is_active, created_at, updated_at \
             FROM allowthem_applications ORDER BY created_at ASC",
        )
        .fetch_all(self.pool())
        .await
        .map_err(AuthError::Database)
    }

    /// Paginated list of applications using a `(created_at, id)` keyset cursor.
    ///
    /// Limits are capped at 200. Pass `None` for cursor to start from the beginning.
    pub async fn list_applications_paginated(
        &self,
        limit: u32,
        cursor: Option<&ApplicationCursor>,
    ) -> Result<Vec<Application>, AuthError> {
        let limit = (limit as i64).min(200);
        match cursor {
            None => sqlx::query_as::<_, Application>(
                "SELECT id, name, client_id, client_type, client_secret_hash, \
                 redirect_uris, logo_url, primary_color, \
                 accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
                 splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
                 is_trusted, created_by, is_active, created_at, updated_at \
                 FROM allowthem_applications \
                 ORDER BY created_at ASC, id ASC LIMIT ?1",
            )
            .bind(limit)
            .fetch_all(self.pool())
            .await
            .map_err(AuthError::Database),
            Some(cur) => {
                // Bind created_at as TEXT matching the schema format so that
                // lexicographic comparison produces the correct ordering.
                let ca = cur.created_at.format("%Y-%m-%dT%H:%M:%S%.3fZ").to_string();
                sqlx::query_as::<_, Application>(
                    "SELECT id, name, client_id, client_type, client_secret_hash, \
                     redirect_uris, logo_url, primary_color, \
                     accent_hex, accent_ink, forced_mode, font_css_url, font_family, \
                     splash_text, splash_image_url, splash_primitive, splash_url, shader_cell_scale, \
                     is_trusted, created_by, is_active, created_at, updated_at \
                     FROM allowthem_applications \
                     WHERE (created_at > ?1 OR (created_at = ?1 AND id > ?2)) \
                     ORDER BY created_at ASC, id ASC LIMIT ?3",
                )
                .bind(&ca)
                .bind(cur.id)
                .bind(limit)
                .fetch_all(self.pool())
                .await
                .map_err(AuthError::Database)
            }
        }
    }

    /// Update an application's mutable fields.
    ///
    /// Validates `redirect_uris`, serializes them to JSON, and writes all
    /// mutable fields atomically. Caller is responsible for fetching the current
    /// application and populating unchanged fields.
    ///
    /// Returns `AuthError::NotFound` if no application with `id` exists.
    /// Returns `AuthError::InvalidRedirectUri` if any URI fails validation.
    pub async fn update_application(
        &self,
        id: ApplicationId,
        params: UpdateApplication,
    ) -> Result<(), AuthError> {
        validate_redirect_uris(&params.redirect_uris)?;
        if let Some(ref url) = params.logo_url {
            validate_logo_url(url)?;
        }
        if let Some(ref color) = params.primary_color {
            validate_primary_color(color)?;
        }
        if let Some(ref hex) = params.accent_hex {
            validate_accent_hex(hex)?;
        }
        if let Some(ref url) = params.font_css_url {
            validate_font_css_url(url)?;
        }
        if let Some(ref url) = params.splash_image_url {
            validate_splash_image_url(url)?;
        }
        if let Some(ref url) = params.splash_url {
            validate_splash_url(url)?;
        }
        let redirect_uris_json =
            serde_json::to_string(&params.redirect_uris).expect("Vec<String> serializes to JSON");
        let now = Utc::now().format("%Y-%m-%dT%H:%M:%S%.3fZ").to_string();

        let result = sqlx::query(
            "UPDATE allowthem_applications \
             SET name = ?1, redirect_uris = ?2, is_trusted = ?3, is_active = ?4, \
                 logo_url = ?5, primary_color = ?6, \
                 accent_hex = ?7, accent_ink = ?8, forced_mode = ?9, \
                 font_css_url = ?10, font_family = ?11, \
                 splash_text = ?12, splash_image_url = ?13, splash_primitive = ?14, \
                 splash_url = ?15, shader_cell_scale = ?16, \
                 updated_at = ?17 \
             WHERE id = ?18",
        )
        .bind(&params.name)
        .bind(&redirect_uris_json)
        .bind(params.is_trusted)
        .bind(params.is_active)
        .bind(&params.logo_url)
        .bind(&params.primary_color)
        .bind(&params.accent_hex)
        .bind(params.accent_ink.map(|v| v.as_str()))
        .bind(params.forced_mode.map(|v| v.as_str()))
        .bind(&params.font_css_url)
        .bind(&params.font_family)
        .bind(&params.splash_text)
        .bind(&params.splash_image_url)
        .bind(params.splash_primitive.map(|v| v.as_str()))
        .bind(&params.splash_url)
        .bind(params.shader_cell_scale)
        .bind(&now)
        .bind(id)
        .execute(self.pool())
        .await?;

        if result.rows_affected() == 0 {
            return Err(AuthError::NotFound);
        }
        Ok(())
    }

    /// Generate a new client secret, invalidating the previous one.
    ///
    /// Returns the updated `Application` and the raw `ClientSecret`.
    /// The new secret is the only opportunity to retrieve it — the old secret
    /// is irrecoverably invalidated on success.
    ///
    /// Returns `AuthError::NotFound` if no application with `id` exists.
    pub async fn regenerate_client_secret(
        &self,
        id: ApplicationId,
    ) -> Result<(Application, ClientSecret), AuthError> {
        let application = self.get_application(id).await?;
        if application.client_type == ClientType::Public {
            return Err(AuthError::InvalidRequest(
                "public clients have no client secret".into(),
            ));
        }
        let (raw_secret, hash) = generate_client_secret()?;
        let now = Utc::now().format("%Y-%m-%dT%H:%M:%S%.3fZ").to_string();

        let result = sqlx::query(
            "UPDATE allowthem_applications \
             SET client_secret_hash = ?1, updated_at = ?2 \
             WHERE id = ?3",
        )
        .bind(&hash)
        .bind(&now)
        .bind(id)
        .execute(self.pool())
        .await?;

        if result.rows_affected() == 0 {
            return Err(AuthError::NotFound);
        }

        let app = self.get_application(id).await?;
        Ok((app, raw_secret))
    }

    /// Permanently delete an application and all associated grants.
    ///
    /// Cascade deletes: authorization_codes, refresh_tokens, consents.
    /// Returns `AuthError::NotFound` if no application with `id` exists.
    pub async fn delete_application(&self, id: ApplicationId) -> Result<(), AuthError> {
        let result = sqlx::query("DELETE FROM allowthem_applications WHERE id = ?")
            .bind(id)
            .execute(self.pool())
            .await?;

        if result.rows_affected() == 0 {
            return Err(AuthError::NotFound);
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::password::verify_password;
    use crate::types::ApplicationId;

    #[test]
    fn client_id_has_ath_prefix() {
        let id = generate_client_id();
        assert!(
            id.as_str().starts_with("ath_"),
            "client_id must start with ath_"
        );
    }

    #[test]
    fn client_id_length_is_36() {
        let id = generate_client_id();
        assert_eq!(id.as_str().len(), 36, "ath_(4) + 32 base64url chars = 36");
    }

    #[test]
    fn client_id_chars_are_url_safe() {
        let id = generate_client_id();
        // base64url uses A-Z, a-z, 0-9, -, _ only (no +, /, =)
        let suffix = &id.as_str()[4..];
        assert!(
            suffix
                .chars()
                .all(|c| c.is_ascii_alphanumeric() || c == '-' || c == '_'),
            "client_id suffix must be URL-safe base64url: got {suffix}"
        );
    }

    #[test]
    fn two_client_ids_differ() {
        let a = generate_client_id();
        let b = generate_client_id();
        assert_ne!(a, b, "each client_id must be unique");
    }

    #[test]
    fn client_secret_verifies_round_trip() {
        let (secret, hash) = generate_client_secret().expect("generate_client_secret");
        let valid = verify_password(secret.as_str(), &hash).expect("verify_password");
        assert!(valid, "generated secret must verify against its own hash");
    }

    #[test]
    fn two_client_secrets_differ() {
        let (s1, _) = generate_client_secret().expect("secret 1");
        let (s2, _) = generate_client_secret().expect("secret 2");
        assert_ne!(s1.as_str(), s2.as_str(), "each secret must be unique");
    }

    #[test]
    fn wrong_secret_does_not_verify() {
        let (_, hash) = generate_client_secret().expect("generate_client_secret");
        let valid = verify_password("wrong-secret", &hash).expect("verify_password");
        assert!(!valid, "wrong secret must not verify");
    }

    // validate_redirect_uris tests

    #[test]
    fn redirect_uri_empty_list_is_rejected() {
        let err = validate_redirect_uris(&[]).unwrap_err();
        assert!(matches!(err, AuthError::InvalidRedirectUri(_)));
    }

    #[test]
    fn redirect_uri_https_is_valid() {
        let uris = vec!["https://example.com/callback".to_string()];
        assert!(validate_redirect_uris(&uris).is_ok());
    }

    #[test]
    fn redirect_uri_http_localhost_is_valid() {
        let uris = vec!["http://localhost/callback".to_string()];
        assert!(validate_redirect_uris(&uris).is_ok());
    }

    #[test]
    fn redirect_uri_http_localhost_with_port_is_valid() {
        let uris = vec!["http://localhost:3000/callback".to_string()];
        assert!(validate_redirect_uris(&uris).is_ok());
    }

    #[test]
    fn redirect_uri_http_127_0_0_1_is_valid() {
        let uris = vec!["http://127.0.0.1:8080/callback".to_string()];
        assert!(validate_redirect_uris(&uris).is_ok());
    }

    #[test]
    fn redirect_uri_http_non_localhost_is_rejected() {
        let uris = vec!["http://example.com/callback".to_string()];
        let err = validate_redirect_uris(&uris).unwrap_err();
        assert!(matches!(err, AuthError::InvalidRedirectUri(_)));
    }

    #[test]
    fn redirect_uri_with_fragment_is_rejected() {
        let uris = vec!["https://example.com/callback#section".to_string()];
        let err = validate_redirect_uris(&uris).unwrap_err();
        assert!(matches!(err, AuthError::InvalidRedirectUri(_)));
    }

    #[test]
    fn redirect_uri_relative_is_rejected() {
        let uris = vec!["/callback".to_string()];
        let err = validate_redirect_uris(&uris).unwrap_err();
        assert!(matches!(err, AuthError::InvalidRedirectUri(_)));
    }

    // validate_redirect_uri tests

    #[test]
    fn redirect_uri_exact_match_passes() {
        let registered = vec!["https://example.com/callback".to_string()];
        assert!(validate_redirect_uri("https://example.com/callback", &registered).is_ok());
    }

    #[test]
    fn redirect_uri_not_in_registered_is_rejected() {
        let registered = vec!["https://example.com/callback".to_string()];
        let err = validate_redirect_uri("https://example.com/other", &registered).unwrap_err();
        assert!(matches!(err, AuthError::InvalidRedirectUri(_)));
    }

    // Application::redirect_uri_list tests

    #[test]
    fn redirect_uri_list_parses_valid_json() {
        let (_, hash) = generate_client_secret().expect("generate_client_secret");
        let app = Application {
            id: ApplicationId::new(),
            name: "Test".to_string(),
            client_id: generate_client_id(),
            client_type: ClientType::Confidential,
            client_secret_hash: Some(hash),
            redirect_uris: r#"["https://example.com/callback","https://example.com/other"]"#
                .to_string(),
            logo_url: None,
            primary_color: None,
            accent_hex: None,
            accent_ink: None,
            forced_mode: None,
            font_css_url: None,
            font_family: None,
            splash_text: None,
            splash_image_url: None,
            splash_primitive: None,
            splash_url: None,
            shader_cell_scale: None,
            is_trusted: false,
            created_by: None,
            is_active: true,
            created_at: chrono::Utc::now(),
            updated_at: chrono::Utc::now(),
        };
        let list = app.redirect_uri_list().expect("redirect_uri_list");
        assert_eq!(
            list,
            vec![
                "https://example.com/callback".to_string(),
                "https://example.com/other".to_string(),
            ]
        );
    }

    #[test]
    fn redirect_uri_list_returns_error_on_malformed_json() {
        let (_, hash) = generate_client_secret().expect("generate_client_secret");
        let app = Application {
            id: ApplicationId::new(),
            name: "Test".to_string(),
            client_id: generate_client_id(),
            client_type: ClientType::Confidential,
            client_secret_hash: Some(hash),
            redirect_uris: "not valid json".to_string(),
            logo_url: None,
            primary_color: None,
            accent_hex: None,
            accent_ink: None,
            forced_mode: None,
            font_css_url: None,
            font_family: None,
            splash_text: None,
            splash_image_url: None,
            splash_primitive: None,
            splash_url: None,
            shader_cell_scale: None,
            is_trusted: false,
            created_by: None,
            is_active: true,
            created_at: chrono::Utc::now(),
            updated_at: chrono::Utc::now(),
        };
        assert!(matches!(
            app.redirect_uri_list(),
            Err(AuthError::Database(_))
        ));
    }

    // validate_logo_url tests

    #[test]
    fn logo_url_https_is_valid() {
        assert!(validate_logo_url("https://example.com/logo.png").is_ok());
    }

    #[test]
    fn logo_url_http_localhost_is_valid() {
        assert!(validate_logo_url("http://localhost:3000/logo.png").is_ok());
    }

    #[test]
    fn logo_url_http_127_is_valid() {
        assert!(validate_logo_url("http://127.0.0.1:8080/logo.png").is_ok());
    }

    #[test]
    fn logo_url_http_non_localhost_is_rejected() {
        let err = validate_logo_url("http://example.com/logo.png").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn logo_url_relative_is_rejected() {
        let err = validate_logo_url("/logo.png").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn logo_url_not_a_url_is_rejected() {
        let err = validate_logo_url("not a url").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    // validate_primary_color tests

    #[test]
    fn primary_color_valid_hex() {
        assert!(validate_primary_color("#3B82F6").is_ok());
    }

    #[test]
    fn primary_color_lowercase_hex() {
        assert!(validate_primary_color("#3b82f6").is_ok());
    }

    #[test]
    fn primary_color_missing_hash() {
        let err = validate_primary_color("3B82F6").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn primary_color_too_short() {
        let err = validate_primary_color("#FFF").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn primary_color_too_long() {
        let err = validate_primary_color("#3B82F6FF").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn primary_color_non_hex_chars() {
        let err = validate_primary_color("#ZZZZZZ").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn primary_color_named_color_rejected() {
        let err = validate_primary_color("red").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    // BrandingConfig extraction test

    #[test]
    fn branding_extracts_correct_fields() {
        let (_, hash) = generate_client_secret().expect("generate");
        let app = Application {
            id: ApplicationId::new(),
            name: "My App".to_string(),
            client_id: generate_client_id(),
            client_type: ClientType::Confidential,
            client_secret_hash: Some(hash),
            redirect_uris: r#"["https://example.com/cb"]"#.to_string(),
            logo_url: Some("https://example.com/logo.png".to_string()),
            primary_color: Some("#3B82F6".to_string()),
            accent_hex: None,
            accent_ink: None,
            forced_mode: None,
            font_css_url: None,
            font_family: None,
            splash_text: None,
            splash_image_url: None,
            splash_primitive: None,
            splash_url: None,
            shader_cell_scale: None,
            is_trusted: false,
            created_by: None,
            is_active: true,
            created_at: chrono::Utc::now(),
            updated_at: chrono::Utc::now(),
        };
        let b = app.branding();
        assert_eq!(b.application_name, "My App");
        assert_eq!(b.logo_url.as_deref(), Some("https://example.com/logo.png"));
        assert_eq!(b.primary_color.as_deref(), Some("#3B82F6"));
    }

    // validate_https_url tests (via public wrappers)

    #[test]
    fn https_url_accepts_https() {
        assert!(validate_font_css_url("https://example.com/x.css").is_ok());
    }

    #[test]
    fn https_url_rejects_http() {
        let err = validate_font_css_url("http://example.com/x.css").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn https_url_rejects_invalid() {
        let err = validate_font_css_url("not a url").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn logo_url_loopback_hostname_accepted() {
        assert!(validate_logo_url("http://localhost/logo.png").is_ok());
    }

    #[test]
    fn logo_url_loopback_ip_accepted() {
        assert!(validate_logo_url("http://127.0.0.1/logo.png").is_ok());
    }

    #[test]
    fn font_css_url_rejects_localhost() {
        let err = validate_font_css_url("http://localhost/font.css").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    // validate_accent_hex tests

    #[test]
    fn accent_hex_valid() {
        assert!(validate_accent_hex("#ff6b35").is_ok());
    }

    #[test]
    fn accent_hex_rejects_named_color() {
        let err = validate_accent_hex("red").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn accent_hex_rejects_shorthand() {
        let err = validate_accent_hex("#fff").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn accent_hex_rejects_non_hex_chars() {
        let err = validate_accent_hex("#gggggg").unwrap_err();
        assert!(matches!(err, AuthError::Validation(_)));
    }

    #[test]
    fn primary_color_still_valid_after_refactor() {
        assert!(validate_primary_color("#3B82F6").is_ok());
    }

    #[test]
    fn application_serialization_omits_secret() {
        let (_, hash) = generate_client_secret().expect("generate_client_secret");
        let app = Application {
            id: ApplicationId::new(),
            name: "Test App".to_string(),
            client_id: generate_client_id(),
            client_type: ClientType::Confidential,
            client_secret_hash: Some(hash),
            redirect_uris: r#"["https://example.com/callback"]"#.to_string(),
            logo_url: None,
            primary_color: None,
            accent_hex: None,
            accent_ink: None,
            forced_mode: None,
            font_css_url: None,
            font_family: None,
            splash_text: None,
            splash_image_url: None,
            splash_primitive: None,
            splash_url: None,
            shader_cell_scale: None,
            is_trusted: false,
            created_by: None,
            is_active: true,
            created_at: chrono::Utc::now(),
            updated_at: chrono::Utc::now(),
        };
        let value = serde_json::to_value(&app).expect("serialize Application");
        assert!(
            value.get("client_secret_hash").is_none(),
            "client_secret_hash must not appear in serialized output"
        );
        assert!(
            value.get("client_id").is_some(),
            "client_id must appear in serialized output"
        );
    }

    // count_users_for_application tests

    #[tokio::test]
    async fn count_users_for_application_returns_consent_count() {
        let db = crate::db::Db::connect("sqlite::memory:")
            .await
            .expect("in-memory db");
        let (app, _secret) = db
            .create_application(CreateApplicationParams {
                name: "Count Test".into(),
                client_type: ClientType::Confidential,
                redirect_uris: vec!["https://example.com/callback".into()],
                is_trusted: false,
                created_by: None,
                logo_url: None,
                primary_color: None,
                accent_hex: None,
                accent_ink: None,
                forced_mode: None,
                font_css_url: None,
                font_family: None,
                splash_text: None,
                splash_image_url: None,
                splash_primitive: None,
                splash_url: None,
                shader_cell_scale: None,
            })
            .await
            .expect("create_application");

        let email1 = crate::Email::new("u1@test.com".into()).expect("email");
        let email2 = crate::Email::new("u2@test.com".into()).expect("email");
        let user1 = db
            .create_user(email1, "pw", None, None)
            .await
            .expect("user1");
        let user2 = db
            .create_user(email2, "pw", None, None)
            .await
            .expect("user2");

        let id1 = uuid::Uuid::new_v4();
        let id2 = uuid::Uuid::new_v4();
        sqlx::query(
            "INSERT OR IGNORE INTO allowthem_consents (id, user_id, application_id) \
             VALUES (?, ?, ?)",
        )
        .bind(id1.to_string())
        .bind(user1.id)
        .bind(app.id)
        .execute(db.pool())
        .await
        .expect("insert consent 1");
        sqlx::query(
            "INSERT OR IGNORE INTO allowthem_consents (id, user_id, application_id) \
             VALUES (?, ?, ?)",
        )
        .bind(id2.to_string())
        .bind(user2.id)
        .bind(app.id)
        .execute(db.pool())
        .await
        .expect("insert consent 2");

        let count = db.count_users_for_application(app.id).await.expect("count");
        assert_eq!(count, 2, "expected 2 consented users");
    }

    #[tokio::test]
    async fn count_users_is_zero_for_unknown_application() {
        let db = crate::db::Db::connect("sqlite::memory:")
            .await
            .expect("in-memory db");
        let unknown_id = ApplicationId::new();
        let count = db
            .count_users_for_application(unknown_id)
            .await
            .expect("count for unknown app");
        assert_eq!(count, 0, "no consents for unknown application");
    }

    // count_applications tests (99c.6 Step 2)

    async fn make_app(db: &crate::db::Db) -> Application {
        let (app, _) = db
            .create_application(CreateApplicationParams {
                name: "Test App".into(),
                client_type: crate::types::ClientType::Confidential,
                redirect_uris: vec!["https://example.com/callback".into()],
                is_trusted: false,
                created_by: None,
                logo_url: None,
                primary_color: None,
                accent_hex: None,
                accent_ink: None,
                forced_mode: None,
                font_css_url: None,
                font_family: None,
                splash_text: None,
                splash_image_url: None,
                splash_primitive: None,
                splash_url: None,
                shader_cell_scale: None,
            })
            .await
            .expect("create_application");
        app
    }

    #[tokio::test]
    async fn count_applications_zero_on_empty_db() {
        let db = crate::db::Db::connect("sqlite::memory:")
            .await
            .expect("in-memory db");
        let n = db.count_applications().await.expect("count_applications");
        assert_eq!(n, 0);
    }

    #[tokio::test]
    async fn count_applications_after_create() {
        let db = crate::db::Db::connect("sqlite::memory:")
            .await
            .expect("in-memory db");
        make_app(&db).await;
        make_app(&db).await;
        make_app(&db).await;
        let n = db.count_applications().await.expect("count_applications");
        assert_eq!(n, 3);
    }

    #[cfg(test)]
    mod branding_config_builder_tests {
        use super::*;
        use crate::types::{AccentInk, SplashPrimitive};

        #[test]
        fn new_sets_application_name_leaves_rest_none() {
            let b = BrandingConfig::new("Fixture Co");
            assert_eq!(b.application_name, "Fixture Co");
            assert!(b.logo_url.is_none());
            assert!(b.primary_color.is_none());
            assert!(b.accent_hex.is_none());
            assert!(b.accent_ink.is_none());
            assert!(b.forced_mode.is_none());
            assert!(b.font_css_url.is_none());
            assert!(b.font_family.is_none());
            assert!(b.splash_text.is_none());
            assert!(b.splash_image_url.is_none());
            assert!(b.splash_primitive.is_none());
            assert!(b.splash_url.is_none());
            assert!(b.shader_cell_scale.is_none());
        }

        #[test]
        fn with_accent_sets_hex_and_ink() {
            let b = BrandingConfig::new("Co").with_accent("#ff7a1a", AccentInk::Black);
            assert_eq!(b.accent_hex.as_deref(), Some("#ff7a1a"));
            assert_eq!(b.accent_ink, Some(AccentInk::Black));
        }

        #[test]
        fn with_splash_text_sets_field() {
            let b = BrandingConfig::new("Co").with_splash_text("TRANSFER");
            assert_eq!(b.splash_text.as_deref(), Some("TRANSFER"));
        }

        #[test]
        fn with_shader_cell_scale_sets_field() {
            let b = BrandingConfig::new("Co").with_shader_cell_scale(18);
            assert_eq!(b.shader_cell_scale, Some(18));
        }

        #[test]
        fn with_splash_primitive_sets_field() {
            let b = BrandingConfig::new("Co").with_splash_primitive(SplashPrimitive::Wave);
            assert_eq!(b.splash_primitive, Some(SplashPrimitive::Wave));
        }

        #[test]
        fn with_logo_url_sets_field() {
            let b = BrandingConfig::new("Co").with_logo_url("https://cdn.example/logo.svg");
            assert_eq!(b.logo_url.as_deref(), Some("https://cdn.example/logo.svg"));
        }

        #[test]
        fn with_primary_color_sets_field() {
            let b = BrandingConfig::new("Co").with_primary_color("#0066ff");
            assert_eq!(b.primary_color.as_deref(), Some("#0066ff"));
        }

        #[test]
        fn with_splash_image_url_sets_field() {
            let b =
                BrandingConfig::new("Co").with_splash_image_url("https://cdn.example/splash.png");
            assert_eq!(
                b.splash_image_url.as_deref(),
                Some("https://cdn.example/splash.png")
            );
        }
    }
}