use axum::{
    body::Body,
    extract::{Extension, State},
    http::{Request, Response},
    routing::post,
    Router,
};

use crate::{
    cursor::{proxy, CursorSessionRegistry},
    Result,
};

pub const TAB_PATHS: [&str; 17] = [
    "/aiserver.v1.AiService/StreamCpp",
    "/aiserver.v1.AiService/StreamNextCursorPrediction",
    "/aiserver.v1.AiService/GetCppEditClassification",
    "/aiserver.v1.AiService/RefreshTabContext",
    "/aiserver.v1.AiService/CppConfig",
    "/aiserver.v1.AiService/CppEditHistoryStatus",
    "/aiserver.v1.AiService/CppAppend",
    "/aiserver.v1.AiService/CppEditHistoryAppend",
    "/aiserver.v1.AiService/ReportAiCodeChangeMetrics",
    "/aiserver.v1.AiService/WriteGitCommitMessage",
    "/aiserver.v1.AiService/WriteGitBranchName",
    "/aiserver.v1.CppService/AvailableModels",
    "/aiserver.v1.CppService/RecordCppFate",
    "/aiserver.v1.FileSyncService/FSSyncFile",
    "/aiserver.v1.FileSyncService/FSIsEnabledForUser",
    "/aiserver.v1.FileSyncService/FSConfig",
    "/aiserver.v1.FileSyncService/FSUploadFile",
];

pub fn is_tab_path(path: &str) -> bool {
    TAB_PATHS.contains(&path)
}

pub fn router() -> Router<CursorSessionRegistry> {
    TAB_PATHS.into_iter().fold(Router::new(), |router, path| {
        router.route(path, post(forward))
    })
}

async fn forward(
    State(registry): State<CursorSessionRegistry>,
    Extension(upstream): Extension<proxy::CursorProxy>,
    request: Request<Body>,
) -> Result<Response<Body>> {
    let settings = registry.store().tab_settings().await?;
    match settings.service_url() {
        Some(service_url) => proxy::forward_to_service(&upstream, request, service_url).await,
        None => proxy::forward(Extension(upstream), request).await,
    }
}

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

    #[test]
    fn matches_only_legacy_tab_routes() {
        assert_eq!(TAB_PATHS.len(), 17);
        assert!(is_tab_path("/aiserver.v1.AiService/StreamCpp"));
        assert!(is_tab_path("/aiserver.v1.FileSyncService/FSUploadFile"));
        assert!(!is_tab_path("/aiserver.v1.AiService/AvailableModels"));
    }
}