use std::sync::Arc;

use api_types::{
    ADMIN_SETTINGS, ADMIN_USERS, AUTH_LOGIN, AUTH_LOGOUT, AUTH_ME, AUTH_SETUP, FILES, SHARE, SHARES,
};
use axum::Router;
use axum::http::HeaderValue;
use axum::routing::{delete, get, post, put};
use tower_http::set_header::SetResponseHeaderLayer;

use crate::error::AppState;

/// Content-Security-Policy tuned to the built Trunk frontend.
///
/// * `script-src 'unsafe-inline'` — Trunk emits one inline bootstrap module
///   in index.html; every real JS file also carries an SRI integrity hash.
/// * `'wasm-unsafe-eval'` — compiling the same-origin WASM module.
/// * `style-src 'unsafe-inline'` — the context menu sets an inline `style=`.
/// * Everything else locked to the same origin; frames/plugins banned.
const CSP: &str = "default-src 'self'; script-src 'self' 'unsafe-inline' 'wasm-unsafe-eval'; style-src 'self' 'unsafe-inline'; img-src 'self' data: blob:; media-src 'self' blob:; connect-src 'self'; font-src 'self' data:; object-src 'none'; base-uri 'self'; form-action 'self'; frame-ancestors 'none';";

mod admin;
mod auth;
mod common;
mod files;
mod shares;
mod spa;

pub fn router(state: Arc<AppState>) -> Router {
    // Route patterns: the server side of the shared endpoint strings in
    // `api_types` (axum copies them into the route table on insert).
    let files_root = format!("{FILES}/{{root_id}}");
    let files_item = format!("{FILES}/{{root_id}}/{{*path}}");
    let shares_id = format!("{SHARES}/{{id}}");
    let share_token = format!("{SHARE}/{{token}}");
    let admin_user_id = format!("{ADMIN_USERS}/{{id}}");

    Router::new()
        .route(AUTH_LOGIN, post(auth::login))
        .route(AUTH_LOGOUT, post(auth::logout))
        .route(AUTH_ME, get(auth::me))
        .route(AUTH_SETUP, post(auth::setup))
        .route(&files_root, get(files::list_root))
        .route(&files_item, get(files::file_get))
        .route(&files_item, put(files::file_put))
        .route(&files_root, post(files::dispatch_root))
        .route(&files_item, post(files::dispatch))
        .route(&files_item, delete(files::delete))
        .route(SHARES, get(shares::list))
        .route(SHARES, post(shares::create))
        .route(&shares_id, delete(shares::delete))
        .route(&share_token, get(shares::resolve))
        .route(ADMIN_USERS, get(admin::list_users))
        .route(ADMIN_USERS, post(admin::create_user))
        .route(&admin_user_id, put(admin::update_user))
        .route(&admin_user_id, delete(admin::delete_user))
        .route(ADMIN_SETTINGS, get(admin::get_settings))
        .route(ADMIN_SETTINGS, put(admin::update_settings))
        .fallback(spa::fallback)
        .with_state(state)
        // Hard security headers on every response (API and static alike).
        .layer(SetResponseHeaderLayer::overriding(
            "content-security-policy".parse().unwrap(),
            HeaderValue::from_static(CSP),
        ))
        .layer(SetResponseHeaderLayer::overriding(
            "x-content-type-options".parse().unwrap(),
            HeaderValue::from_static("nosniff"),
        ))
        .layer(SetResponseHeaderLayer::overriding(
            "x-frame-options".parse().unwrap(),
            HeaderValue::from_static("DENY"),
        ))
        .layer(SetResponseHeaderLayer::overriding(
            "referrer-policy".parse().unwrap(),
            HeaderValue::from_static("no-referrer"),
        ))
}