use askama::Template;

use askama_web::WebTemplate;

use subtle::ConstantTimeEq;

use axum::{

    Form, Json, Router,

    extract::{Path, Request, State},

    http::{HeaderMap, StatusCode, header},

    middleware::{self, Next},

    response::{Html, IntoResponse, Redirect, Response},

    routing::{delete, get, post},

};

use rust_embed::Embed;

use serde::Deserialize;

use crate::db::{self, Db, Snippet};

use crate::highlight::Highlighter;

use std::collections::HashSet;

use std::sync::Arc;

#[derive(Embed)]

#[folder = "assets/"]

struct Assets;

#[derive(Embed)]

#[folder = "static/"]

struct Static;

#[derive(Clone)]

struct ServerConfig {

    api_key: Option<String>,

    auth_endpoints: HashSet<String>,

}

impl ServerConfig {

    fn from_env() -> Self {

        let api_key = std::env::var("SIPP_API_KEY").ok();

        let auth_endpoints = match std::env::var("SIPP_AUTH_ENDPOINTS") {

            Ok(val) if val.trim().eq_ignore_ascii_case("none") => HashSet::new(),

            Ok(val) => val.split(',').map(|s| s.trim().to_lowercase()).collect(),

            Err(_) => ["api_delete", "api_list"].iter().map(|s| s.to_string()).collect(),

        };

        ServerConfig { api_key, auth_endpoints }

    }

    fn requires_auth(&self, name: &str) -> bool {

        self.auth_endpoints.contains("all") || self.auth_endpoints.contains(name)

    }

}

#[derive(Clone)]

struct AppState {

    db: Db,

    highlighter: Arc<Highlighter>,

    server_config: ServerConfig,

}

#[derive(Template)]

#[template(path = "index.html")]

struct IndexTemplate;

#[derive(Template)]

#[template(path = "snippet.html")]

struct SnippetTemplate {

    name: String,

    content: String,

    highlighted_content: String,

}

#[derive(Deserialize)]

struct CreateSnippetForm {

    name: String,

    content: String,

}

async fn index() -> WebTemplate<IndexTemplate> {

    WebTemplate(IndexTemplate)

}

async fn view_snippet(

    State(state): State<AppState>,

    Path(short_id): Path<String>,

) -> Result<WebTemplate<SnippetTemplate>, (StatusCode, Html<String>)> {

    match db::get_snippet_by_short_id(&state.db, &short_id) {

        Some(snippet) => {

            let highlighted_content = state.highlighter.highlight(&snippet.name, &snippet.content);

            Ok(WebTemplate(SnippetTemplate {

                name: snippet.name,

                content: snippet.content,

                highlighted_content,

            }))

        }

        None => Err((

            StatusCode::NOT_FOUND,

            Html("<h1>Snippet not found</h1>".to_string()),

        )),

    }

}

async fn create_snippet(

    State(state): State<AppState>,

    Form(form): Form<CreateSnippetForm>,

) -> impl IntoResponse {

    let snippet = db::create_snippet(&state.db, &form.name, &form.content);

    Redirect::to(&format!("/s/{}", snippet.short_id))

}

async fn require_api_key(

    State(state): State<AppState>,

    headers: HeaderMap,

    request: Request,

    next: Next,

) -> Result<Response, (StatusCode, Json<serde_json::Value>)> {

    let server_key = match &state.server_config.api_key {

        Some(k) => k,

        None => return Err((

            StatusCode::FORBIDDEN,

            Json(serde_json::json!({"error": "No API key configured on server"})),

        )),

    };

    let provided = headers

        .get("x-api-key")

        .and_then(|v| v.to_str().ok());

    match provided {

        Some(k) if k.as_bytes().ct_eq(server_key.as_bytes()).into() => Ok(next.run(request).await),

        _ => Err((

            StatusCode::UNAUTHORIZED,

            Json(serde_json::json!({"error": "Invalid or missing API key"})),

        )),

    }

}

async fn api_list_snippets(

    State(state): State<AppState>,

) -> Json<Vec<Snippet>> {

    Json(db::get_all_snippets(&state.db))

}

async fn api_get_snippet(

    State(state): State<AppState>,

    Path(short_id): Path<String>,

) -> Result<Json<Snippet>, (StatusCode, Json<serde_json::Value>)> {

    match db::get_snippet_by_short_id(&state.db, &short_id) {

        Some(snippet) => Ok(Json(snippet)),

        None => Err((StatusCode::NOT_FOUND, Json(serde_json::json!({"error": "Snippet not found"})))),

    }

}

#[derive(Deserialize)]

struct ApiCreateSnippet {

    name: String,

    content: String,

}

async fn api_create_snippet(

    State(state): State<AppState>,

    Json(body): Json<ApiCreateSnippet>,

) -> (StatusCode, Json<Snippet>) {

    let snippet = db::create_snippet(&state.db, &body.name, &body.content);

    (StatusCode::CREATED, Json(snippet))

}

async fn api_delete_snippet(

    State(state): State<AppState>,

    Path(short_id): Path<String>,

) -> Result<Json<serde_json::Value>, (StatusCode, Json<serde_json::Value>)> {

    if db::delete_snippet_by_short_id(&state.db, &short_id) {

        Ok(Json(serde_json::json!({"deleted": true})))

    } else {

        Err((StatusCode::NOT_FOUND, Json(serde_json::json!({"error": "Snippet not found"}))))

    }

}

fn build_api_routes(state: &AppState) -> Router<AppState> {

    let config = &state.server_config;

    let auth_layer = middleware::from_fn_with_state(state.clone(), require_api_key);

    // /api/snippets — GET (api_list) and POST (api_create)

    let list_authed = config.requires_auth("api_list");

    let create_authed = config.requires_auth("api_create");

    // /api/snippets/{short_id} — GET (api_get) and DELETE (api_delete)

    let get_authed = config.requires_auth("api_get");

    let delete_authed = config.requires_auth("api_delete");

    // Build authed router

    let mut authed = Router::new();

    if list_authed {

        authed = authed.route("/api/snippets", get(api_list_snippets));

    }

    if create_authed {

        authed = authed.route("/api/snippets", post(api_create_snippet));

    }

    if get_authed {

        authed = authed.route("/api/snippets/{short_id}", get(api_get_snippet));

    }

    if delete_authed {

        authed = authed.route("/api/snippets/{short_id}", delete(api_delete_snippet));

    }

    let authed = authed.route_layer(auth_layer);

    // Build open router

    let mut open = Router::new();

    if !list_authed {

        open = open.route("/api/snippets", get(api_list_snippets));

    }

    if !create_authed {

        open = open.route("/api/snippets", post(api_create_snippet));

    }

    if !get_authed {

        open = open.route("/api/snippets/{short_id}", get(api_get_snippet));

    }

    if !delete_authed {

        open = open.route("/api/snippets/{short_id}", delete(api_delete_snippet));

    }

    authed.merge(open)

}

fn mime_from_path(path: &str) -> &'static str {

    match path.rsplit('.').next().unwrap_or("") {

        "css" => "text/css",

        "js" => "application/javascript",

        "html" => "text/html",

        "png" => "image/png",

        "ico" => "image/x-icon",

        "svg" => "image/svg+xml",

        "woff" => "font/woff",

        "woff2" => "font/woff2",

        "ttf" => "font/ttf",

        "otf" => "font/otf",

        "json" | "webmanifest" => "application/json",

        "jpg" | "jpeg" => "image/jpeg",

        _ => "application/octet-stream",

    }

}

async fn serve_assets(Path(path): Path<String>) -> Response {

    match Assets::get(&path) {

        Some(file) => {

            let mime = mime_from_path(&path);

            ([(header::CONTENT_TYPE, mime)], file.data).into_response()

        }

        None => StatusCode::NOT_FOUND.into_response(),

    }

}

async fn serve_static(Path(path): Path<String>) -> Response {

    match Static::get(&path) {

        Some(file) => {

            let mime = mime_from_path(&path);

            ([(header::CONTENT_TYPE, mime)], file.data).into_response()

        }

        None => StatusCode::NOT_FOUND.into_response(),

    }

}

pub async fn run(host: String, port: u16) {

    dotenvy::dotenv().ok();

    let server_config = ServerConfig::from_env();

    // Validate endpoint names

    let known = ["api_list", "api_create", "api_get", "api_delete", "all", "none"];

    for name in &server_config.auth_endpoints {

        if !known.contains(&name.as_str()) {

            eprintln!("Warning: unknown auth endpoint name '{}' in SIPP_AUTH_ENDPOINTS", name);

        }

    }

    if !server_config.auth_endpoints.is_empty() && server_config.api_key.is_none() {

        eprintln!("Warning: SIPP_AUTH_ENDPOINTS is set but SIPP_API_KEY is not configured");

    }

    if server_config.auth_endpoints.is_empty() {

        println!("Auth: disabled (no endpoints require authentication)");

    } else {

        let names: Vec<&str> = server_config.auth_endpoints.iter().map(|s| s.as_str()).collect();

        println!("Auth: enabled for endpoints: {}", names.join(", "));

    }

    let state = AppState {

        db: db::init_db(),

        highlighter: Arc::new(Highlighter::new()),

        server_config,

    };

    let api_routes = build_api_routes(&state);

    let app = Router::new()

        .route("/", get(index))

        .route("/s/{short_id}", get(view_snippet))

        .route("/snippets", post(create_snippet))

        .merge(api_routes)

        .route("/assets/{*path}", get(serve_assets))

        .route("/static/{*path}", get(serve_static))

        .with_state(state);

    let addr = format!("{}:{}", host, port);

    let listener = tokio::net::TcpListener::bind(&addr)

        .await

        .unwrap_or_else(|_| panic!("Failed to bind to {}", addr));

    println!("Server running at http://{}:{}", host, port);

    axum::serve(listener, app)

        .await

        .expect("Failed to start server");

}