use serde::{Deserialize, Serialize};
use std::fmt;
use std::time::Duration;

use crate::AccountStatus;
use crate::api::ApiError;
use crate::cache::Cache;
use crate::oauth::scopes::ScopePermissions;
use crate::types::Did;
use tranquil_db_traits::{OAuthRepository, UserRepository};

pub mod account_verified;
pub mod email_token;
pub mod extractor;
pub mod legacy_2fa;
pub mod login_identifier;
pub mod mfa_verified;
pub mod reauth;
pub mod scope_check;
pub mod scope_verified;
pub mod service;
pub mod verification_token;
pub mod webauthn;

pub use login_identifier::{BareLoginIdentifier, NormalizedLoginIdentifier};

pub use account_verified::{AccountVerified, require_not_migrated, require_verified_or_delegated};
pub use extractor::{
    Active, Admin, AnyUser, Auth, AuthAny, AuthError, AuthPolicy, AuthScheme, ExtractedToken,
    NotTakendown, Permissive, ServiceAuth, extract_auth_token_from_header,
    extract_bearer_token_from_header, extract_jti_from_headers,
};
pub use mfa_verified::{
    MfaMethod, MfaVerified, require_legacy_session_mfa, require_reauth_window,
    require_reauth_window_if_available, verify_password_mfa, verify_totp_mfa,
};
pub use scope_verified::{
    AccountManage, AccountRead, BatchWriteScopes, BlobScopeAction, BlobUpload, ControllerDid,
    IdentityAccess, PrincipalDid, RepoCreate, RepoDelete, RepoScopeAction, RepoUpdate, RepoUpsert,
    RpcCall, ScopeAction, ScopeVerificationError, ScopeVerified, VerifyScope, WriteOpKind,
    verify_batch_write_scopes,
};
pub use service::{ServiceTokenClaims, ServiceTokenError, ServiceTokenVerifier, is_service_token};

pub use tranquil_auth::{
    ActClaim, Claims, Header, SigningAlgorithm, TokenData, TokenDecodeError, TokenScope, TokenType,
    TokenVerifyError, TokenWithMetadata, TotpError, UnsafeClaims, create_access_token,
    create_access_token_hs256, create_access_token_hs256_with_metadata,
    create_access_token_with_delegation, create_access_token_with_jti,
    create_access_token_with_metadata, create_access_token_with_scope_metadata,
    create_refresh_token, create_refresh_token_hs256, create_refresh_token_hs256_with_metadata,
    create_refresh_token_with_jti, create_refresh_token_with_metadata, create_service_token,
    create_service_token_hs256, generate_backup_codes, generate_qr_png_base64,
    generate_totp_secret, generate_totp_uri, get_algorithm_from_token, get_did_from_token,
    get_jti_from_token, hash_backup_code, is_backup_code_format, verify_access_token,
    verify_access_token_hs256, verify_backup_code, verify_refresh_token,
    verify_refresh_token_hs256, verify_token, verify_token_es256k, verify_totp_code,
};

pub fn lxm_permits(lxm: &str, expected: &crate::types::Nsid) -> bool {
    lxm == "*" || lxm == expected.as_str()
}

pub fn try_decrypt_user_key(
    key_bytes: Option<&[u8]>,
    encryption_version: Option<i32>,
) -> Option<Vec<u8>> {
    match (key_bytes, encryption_version) {
        (Some(kb), Some(ev)) => crate::config::decrypt_key(kb, Some(ev)).ok(),
        _ => None,
    }
}

pub fn encrypt_totp_secret(secret: &[u8]) -> Result<Vec<u8>, crate::config::CryptoError> {
    crate::config::encrypt_key(secret)
}

pub fn decrypt_totp_secret(
    encrypted: &[u8],
    version: i32,
) -> Result<Vec<u8>, crate::config::CryptoError> {
    crate::config::decrypt_key(encrypted, Some(version))
}

pub fn generate_app_password() -> crate::types::PlainPassword {
    use rand::Rng;
    let chars: &[u8] = b"abcdefghijklmnopqrstuvwxyz234567";
    let mut rng = rand::thread_rng();
    let segments: Vec<String> = (0..4)
        .map(|_| {
            (0..4)
                .map(|_| chars[rng.gen_range(0..chars.len())] as char)
                .collect()
        })
        .collect();
    crate::types::PlainPassword::new(segments.join("-"))
}

const KEY_CACHE_TTL_SECS: u64 = 300;
const SESSION_CACHE_TTL_SECS: u64 = 60;
const USER_STATUS_CACHE_TTL_SECS: u64 = 60;

#[derive(Serialize, Deserialize)]
struct CachedUserStatus {
    deactivated: bool,
    takendown: bool,
    is_admin: bool,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum TokenValidationError {
    AccountDeactivated,
    AccountTakedown,
    KeyDecryptionFailed,
    AuthenticationFailed,
    TokenExpired,
    OAuthTokenExpired,
    InvalidToken,
    UseDpopNonce(String),
    InvalidDpopProof(String),
}

impl fmt::Display for TokenValidationError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::AccountDeactivated => write!(f, "AccountDeactivated"),
            Self::AccountTakedown => write!(f, "AccountTakedown"),
            Self::KeyDecryptionFailed => write!(f, "KeyDecryptionFailed"),
            Self::AuthenticationFailed => write!(f, "AuthenticationFailed"),
            Self::TokenExpired | Self::OAuthTokenExpired => write!(f, "ExpiredToken"),
            Self::InvalidToken => write!(f, "InvalidToken"),
            Self::UseDpopNonce(_) => write!(f, "use_dpop_nonce"),
            Self::InvalidDpopProof(_) => write!(f, "invalid_dpop_proof"),
        }
    }
}

#[derive(Debug, Clone)]
pub enum AuthSource {
    Session,
    OAuth,
    Service { claims: ServiceTokenClaims },
}

impl AuthSource {
    pub fn is_oauth(&self) -> bool {
        matches!(self, Self::OAuth)
    }

    pub fn is_service(&self) -> bool {
        matches!(self, Self::Service { .. })
    }

    pub fn service_claims(&self) -> Option<&ServiceTokenClaims> {
        match self {
            Self::Service { claims } => Some(claims),
            _ => None,
        }
    }
}

pub struct AuthenticatedUser {
    pub did: Did,
    pub key_bytes: Option<Vec<u8>>,
    pub is_admin: bool,
    pub status: AccountStatus,
    pub scope: Option<String>,
    pub controller_did: Option<Did>,
    pub auth_source: AuthSource,
}

impl AuthenticatedUser {
    pub fn is_oauth(&self) -> bool {
        self.auth_source.is_oauth()
    }

    pub fn is_service(&self) -> bool {
        self.auth_source.is_service()
    }

    pub fn service_claims(&self) -> Option<&ServiceTokenClaims> {
        self.auth_source.service_claims()
    }

    pub fn require_lxm(&self, expected_lxm: &crate::types::Nsid) -> Result<(), ApiError> {
        match self.auth_source.service_claims() {
            Some(claims) => match &claims.lxm {
                Some(lxm) if lxm_permits(lxm, expected_lxm) => Ok(()),
                Some(lxm) => Err(ApiError::AuthorizationError(format!(
                    "Token lxm '{}' does not permit '{}'",
                    lxm, expected_lxm
                ))),
                None => Err(ApiError::AuthorizationError(
                    "Token missing lxm claim".to_string(),
                )),
            },
            None => Ok(()),
        }
    }

    pub fn require_user(&self) -> Result<&Self, ApiError> {
        if self.is_service() {
            return Err(ApiError::AuthenticationFailed(Some(
                "User authentication required".to_string(),
            )));
        }
        Ok(self)
    }

    pub fn as_user(&self) -> Option<&Self> {
        if self.is_service() { None } else { Some(self) }
    }
}

impl AuthenticatedUser {
    pub fn permissions(&self) -> ScopePermissions {
        if let Some(ref scope) = self.scope
            && scope != TokenScope::Access.as_str()
        {
            return ScopePermissions::from_scope_string(Some(scope));
        }
        if !self.is_oauth() {
            return ScopePermissions::from_scope_string(Some(
                "transition:generic transition:chat.bsky",
            ));
        }
        ScopePermissions::from_scope_string(self.scope.as_deref())
    }

    pub fn is_takendown(&self) -> bool {
        self.status.is_takendown()
    }
}

pub async fn validate_bearer_token(
    user_repo: &dyn UserRepository,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, None, token, false, false).await
}

pub async fn validate_bearer_token_allow_deactivated(
    user_repo: &dyn UserRepository,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, None, token, true, false).await
}

pub async fn validate_bearer_token_cached(
    user_repo: &dyn UserRepository,
    cache: &dyn Cache,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, Some(cache), token, false, false).await
}

pub async fn validate_bearer_token_cached_allow_deactivated(
    user_repo: &dyn UserRepository,
    cache: &dyn Cache,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, Some(cache), token, true, false).await
}

pub async fn validate_bearer_token_for_service_auth(
    user_repo: &dyn UserRepository,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, None, token, true, true).await
}

pub async fn validate_bearer_token_allow_takendown(
    user_repo: &dyn UserRepository,
    token: &str,
) -> Result<AuthenticatedUser, TokenValidationError> {
    validate_bearer_token_with_options_internal(user_repo, None, token, false, true).await
}

async fn validate_bearer_token_with_options_internal(
    user_repo: &dyn UserRepository,
    cache: Option<&dyn Cache>,
    token: &str,
    allow_deactivated: bool,
    allow_takendown: bool,
) -> Result<AuthenticatedUser, TokenValidationError> {
    let did_from_token = get_did_from_token(token).ok();

    if let Some(ref did_str) = did_from_token {
        let did: tranquil_types::Did = match did_str.parse() {
            Ok(d) => d,
            Err(_) => return Err(TokenValidationError::InvalidToken),
        };
        let key_cache_key = crate::cache_keys::signing_key_key(did_str);
        let mut cached_key: Option<Vec<u8>> = None;

        if let Some(c) = cache {
            cached_key = c.get_bytes(&key_cache_key).await;
            if cached_key.is_some() {
                crate::metrics::record_auth_cache_hit("key");
            } else {
                crate::metrics::record_auth_cache_miss("key");
            }
        }

        let (decrypted_key, deactivated_at, takedown_ref, is_admin) = if let Some(key) = cached_key
        {
            let status_cache_key = crate::cache_keys::user_status_key(did_str);
            let cached_status: Option<CachedUserStatus> = if let Some(c) = cache {
                c.get(&status_cache_key)
                    .await
                    .and_then(|s| serde_json::from_str(&s).ok())
            } else {
                None
            };

            if let Some(status) = cached_status {
                (
                    Some(key),
                    if status.deactivated {
                        Some(chrono::Utc::now())
                    } else {
                        None
                    },
                    if status.takendown {
                        Some("takendown".to_string())
                    } else {
                        None
                    },
                    status.is_admin,
                )
            } else {
                let user_status = user_repo.get_status_by_did(&did).await.ok().flatten();

                match user_status {
                    Some(status) => {
                        if let Some(c) = cache {
                            let cached = CachedUserStatus {
                                deactivated: status.deactivated_at.is_some(),
                                takendown: status.takedown_ref.is_some(),
                                is_admin: status.is_admin,
                            };
                            if let Ok(json) = serde_json::to_string(&cached) {
                                let _ = c
                                    .set(
                                        &status_cache_key,
                                        &json,
                                        Duration::from_secs(USER_STATUS_CACHE_TTL_SECS),
                                    )
                                    .await;
                            }
                        }
                        (
                            Some(key),
                            status.deactivated_at,
                            status.takedown_ref,
                            status.is_admin,
                        )
                    }
                    None => (None, None, None, false),
                }
            }
        } else if let Some(user) = user_repo.get_with_key_by_did(&did).await.ok().flatten() {
            let key = crate::config::decrypt_key(&user.key_bytes, user.encryption_version)
                .map_err(|_| TokenValidationError::KeyDecryptionFailed)?;

            if let Some(c) = cache {
                let _ = c
                    .set_bytes(
                        &key_cache_key,
                        &key,
                        Duration::from_secs(KEY_CACHE_TTL_SECS),
                    )
                    .await;

                let status_cache_key = crate::cache_keys::user_status_key(&did);
                let cached = CachedUserStatus {
                    deactivated: user.deactivated_at.is_some(),
                    takendown: user.takedown_ref.is_some(),
                    is_admin: user.is_admin,
                };
                if let Ok(json) = serde_json::to_string(&cached) {
                    let _ = c
                        .set(
                            &status_cache_key,
                            &json,
                            Duration::from_secs(USER_STATUS_CACHE_TTL_SECS),
                        )
                        .await;
                }
            }

            (
                Some(key),
                user.deactivated_at,
                user.takedown_ref,
                user.is_admin,
            )
        } else {
            (None, None, None, false)
        };

        if let Some(decrypted_key) = decrypted_key {
            if !allow_deactivated && deactivated_at.is_some() {
                return Err(TokenValidationError::AccountDeactivated);
            }

            if !allow_takendown && takedown_ref.is_some() {
                return Err(TokenValidationError::AccountTakedown);
            }

            match verify_token_es256k(token, &decrypted_key, Some(TokenType::Access), None) {
                Ok(token_data) => {
                    let jti = &token_data.claims.jti;
                    let session_cache_key = crate::cache_keys::session_key(&did, jti);
                    let mut session_valid = false;

                    if let Some(c) = cache {
                        if let Some(cached_value) = c.get(&session_cache_key).await {
                            session_valid = cached_value == "1";
                            crate::metrics::record_auth_cache_hit("session");
                        } else {
                            crate::metrics::record_auth_cache_miss("session");
                        }
                    }

                    if !session_valid {
                        let session_expiry = user_repo
                            .get_session_access_expiry(&did, jti)
                            .await
                            .ok()
                            .flatten();

                        if let Some(expires_at) = session_expiry {
                            if expires_at > chrono::Utc::now() {
                                session_valid = true;
                                if let Some(c) = cache {
                                    let _ = c
                                        .set(
                                            &session_cache_key,
                                            "1",
                                            Duration::from_secs(SESSION_CACHE_TTL_SECS),
                                        )
                                        .await;
                                }
                            } else {
                                return Err(TokenValidationError::TokenExpired);
                            }
                        }
                    }

                    if session_valid {
                        let controller_did: Option<Did> = match &token_data.claims.act {
                            Some(act) => Some(
                                act.sub
                                    .parse()
                                    .map_err(|_| TokenValidationError::InvalidToken)?,
                            ),
                            None => None,
                        };
                        let status =
                            AccountStatus::from_db_fields(takedown_ref.as_deref(), deactivated_at);
                        return Ok(AuthenticatedUser {
                            did: did.clone(),
                            key_bytes: Some(decrypted_key),
                            is_admin,
                            status,
                            scope: token_data.claims.scope.clone(),
                            controller_did,
                            auth_source: AuthSource::Session,
                        });
                    }
                }
                Err(TokenVerifyError::Expired) => {
                    return Err(TokenValidationError::TokenExpired);
                }
                Err(TokenVerifyError::Invalid(_)) => {}
            }
        }
    }

    if let Ok(oauth_info) = crate::oauth::verify::extract_oauth_token_info(token)
        && let Some(oauth_token) = user_repo
            .get_oauth_token_with_user(&oauth_info.token_id)
            .await
            .ok()
            .flatten()
    {
        let status = AccountStatus::from_db_fields(
            oauth_token.takedown_ref.as_deref(),
            oauth_token.deactivated_at,
        );

        if !allow_deactivated && status.is_deactivated() {
            return Err(TokenValidationError::AccountDeactivated);
        }

        if !allow_takendown && status.is_takendown() {
            return Err(TokenValidationError::AccountTakedown);
        }

        let now = chrono::Utc::now();
        if oauth_token.expires_at > now {
            let key_bytes = try_decrypt_user_key(
                oauth_token.key_bytes.as_deref(),
                oauth_token.encryption_version,
            );
            return Ok(AuthenticatedUser {
                did: oauth_token.did,
                key_bytes,
                is_admin: oauth_token.is_admin,
                status,
                scope: oauth_info.scope,
                controller_did: oauth_info.controller_did,
                auth_source: AuthSource::OAuth,
            });
        } else {
            return Err(TokenValidationError::OAuthTokenExpired);
        }
    }

    Err(TokenValidationError::AuthenticationFailed)
}

pub async fn invalidate_auth_cache(cache: &dyn Cache, did: &Did) {
    let key_cache_key = crate::cache_keys::signing_key_key(did);
    let status_cache_key = crate::cache_keys::user_status_key(did);
    let _ = cache.delete(&key_cache_key).await;
    let _ = cache.delete(&status_cache_key).await;
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum AccountRequirement {
    Active,
    NotTakendown,
    AnyStatus,
}

#[allow(clippy::too_many_arguments)]
pub async fn validate_token_with_dpop(
    user_repo: &dyn UserRepository,
    oauth_repo: &dyn OAuthRepository,
    token: &str,
    scheme: AuthScheme,
    dpop_proof: Option<&str>,
    http_method: &str,
    http_uri: &str,
    requirement: AccountRequirement,
) -> Result<AuthenticatedUser, TokenValidationError> {
    if !scheme.is_dpop() {
        return match requirement {
            AccountRequirement::AnyStatus => {
                validate_bearer_token_allow_takendown(user_repo, token).await
            }
            AccountRequirement::NotTakendown => {
                validate_bearer_token_allow_deactivated(user_repo, token).await
            }
            AccountRequirement::Active => validate_bearer_token(user_repo, token).await,
        };
    }
    let (allow_deactivated, allow_takendown) = match requirement {
        AccountRequirement::Active => (false, false),
        AccountRequirement::NotTakendown => (true, false),
        AccountRequirement::AnyStatus => (true, true),
    };
    match crate::oauth::verify::verify_oauth_access_token(
        oauth_repo,
        token,
        dpop_proof,
        http_method,
        http_uri,
    )
    .await
    {
        Ok(result) => {
            let result_did: Did = result
                .did
                .parse()
                .map_err(|_| TokenValidationError::InvalidToken)?;
            let user_info = user_repo
                .get_user_info_by_did(&result_did)
                .await
                .ok()
                .flatten();
            let Some(user_info) = user_info else {
                return Err(TokenValidationError::AuthenticationFailed);
            };
            let status = AccountStatus::from_db_fields(
                user_info.takedown_ref.as_deref(),
                user_info.deactivated_at,
            );
            if !allow_deactivated && status.is_deactivated() {
                return Err(TokenValidationError::AccountDeactivated);
            }
            if !allow_takendown && status.is_takendown() {
                return Err(TokenValidationError::AccountTakedown);
            }
            let key_bytes =
                try_decrypt_user_key(user_info.key_bytes.as_deref(), user_info.encryption_version);
            Ok(AuthenticatedUser {
                did: result_did,
                key_bytes,
                is_admin: user_info.is_admin,
                status,
                scope: result.scope,
                controller_did: None,
                auth_source: AuthSource::OAuth,
            })
        }
        Err(crate::oauth::OAuthError::ExpiredToken(_)) => {
            Err(TokenValidationError::OAuthTokenExpired)
        }
        Err(crate::oauth::OAuthError::UseDpopNonce(nonce)) => {
            Err(TokenValidationError::UseDpopNonce(nonce))
        }
        Err(crate::oauth::OAuthError::InvalidDpopProof(msg)) => {
            Err(TokenValidationError::InvalidDpopProof(msg))
        }
        Err(_) => Err(TokenValidationError::AuthenticationFailed),
    }
}

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

    fn n(s: &str) -> crate::types::Nsid {
        crate::types::Nsid::from(s.to_string())
    }

    #[test]
    fn test_lxm_permits_exact_match() {
        assert!(lxm_permits(
            "com.atproto.repo.uploadBlob",
            &n("com.atproto.repo.uploadBlob")
        ));
    }

    #[test]
    fn test_lxm_permits_wildcard() {
        assert!(lxm_permits("*", &n("com.atproto.repo.uploadBlob")));
        assert!(lxm_permits("*", &n("anything.at.all")));
    }

    #[test]
    fn test_lxm_permits_mismatch() {
        assert!(!lxm_permits(
            "com.atproto.repo.uploadBlob",
            &n("com.atproto.repo.createRecord")
        ));
    }

    #[test]
    fn test_lxm_permits_partial_not_wildcard() {
        assert!(!lxm_permits(
            "com.atproto.*",
            &n("com.atproto.repo.uploadBlob")
        ));
    }
}
