use crate::http::admin;
use crate::http::handlers::{catalog, file, info, manifest, publisher, search, shard, ui, versions};
use crate::repo::DepotRepo;
use axum::{
    Router,
    response::Redirect,
    routing::{get, post},
};
use std::sync::Arc;
use tower_http::trace::TraceLayer;

pub fn app_router(state: Arc<DepotRepo>) -> Router {
    Router::new()
        .route("/", get(|| async { Redirect::permanent("/ui/") }))
        .route("/versions/0", get(versions::get_versions))
        .route("/versions/0/", get(versions::get_versions))
        .route(
            "/{publisher}/catalog/1/{filename}",
            get(catalog::get_catalog_v1).head(catalog::get_catalog_v1),
        )
        .route(
            "/{publisher}/catalog/2/catalog.attrs",
            get(shard::get_shard_index).head(shard::get_shard_index),
        )
        .route(
            "/{publisher}/catalog/2/{sha256}",
            get(shard::get_shard_blob).head(shard::get_shard_blob),
        )
        .route(
            "/{publisher}/manifest/0/{fmri}",
            get(manifest::get_manifest).head(manifest::get_manifest),
        )
        .route(
            "/{publisher}/manifest/1/{fmri}",
            get(manifest::get_manifest).head(manifest::get_manifest),
        )
        .route(
            "/{publisher}/file/0/{algo}/{digest}",
            get(file::get_file).head(file::get_file),
        )
        .route(
            "/{publisher}/file/1/{algo}/{digest}",
            get(file::get_file).head(file::get_file),
        )
        .route(
            "/{publisher}/file/1/{digest}",
            get(file::get_file_no_algo).head(file::get_file_no_algo),
        )
        .route("/{publisher}/info/0/{fmri}", get(info::get_info))
        .route("/{publisher}/publisher/0", get(publisher::get_publisher_v0))
        .route(
            "/{publisher}/publisher/0/",
            get(publisher::get_publisher_v0),
        )
        .route("/{publisher}/publisher/1", get(publisher::get_publisher_v1))
        .route(
            "/{publisher}/publisher/1/",
            get(publisher::get_publisher_v1),
        )
        .route("/publisher/0", get(publisher::get_default_publisher_v0))
        .route("/publisher/0/", get(publisher::get_default_publisher_v0))
        .route("/publisher/1", get(publisher::get_default_publisher_v1))
        .route("/publisher/1/", get(publisher::get_default_publisher_v1))
        .route("/{publisher}/search/0/{token}", get(search::get_search_v0))
        .route("/{publisher}/search/1/{token}", get(search::get_search_v1))
        // Admin API over HTTP
        .route("/admin/health", get(admin::health))
        .route("/admin/auth/check", post(admin::auth_check))
        // Web UI routes
        .route("/ui/", get(ui::ui_index))
        .route("/ui/packages/{publisher}", get(ui::ui_packages))
        .route("/ui/search", get(ui::ui_search))
        .route("/ui/search/results", get(ui::ui_search_results))
        .route("/ui/package/{publisher}/{*fmri}", get(ui::ui_package_detail))
        .route("/ui/p5i", get(ui::ui_p5i_generate))
        // Embedded static assets
        .route("/ui/static/css/style.css", get(ui::static_css))
        .route("/ui/static/js/htmx.min.js", get(ui::static_htmx_js))
        .layer(TraceLayer::new_for_http())
        .with_state(state)
}