use crate::ApiDoc;
use crate::app_state::AppState;
use crate::config::get_global_file_size_config;
use crate::handlers::{
    document_handler, health_handler, markdown_handler, private_oss_handler, task_handler,
    toc_handler,
};
use axum::{
    Router,
    extract::DefaultBodyLimit,
    routing::{delete, get, post},
};
use tower::ServiceBuilder;
use tower_http::{cors::CorsLayer, trace::TraceLayer};
use utoipa::OpenApi;
use utoipa_swagger_ui::SwaggerUi;

/// 创建应用路由
pub fn create_routes(state: AppState) -> Router {
    Router::new()
        // 健康检查路由
        .route("/health", get(health_handler::health_check))
        .route("/ready", get(health_handler::ready_check))
        // OpenAPI 文档路由 - 使用 utoipa-swagger-ui 内置支持
        .merge(SwaggerUi::new("/api/docs").url("/api/docs/openapi.json", ApiDoc::openapi()))
        // 文档处理路由
        .nest("/api/v1/documents", document_routes())
        // 任务管理路由
        .nest("/api/v1/tasks", task_routes())
        // OSS 服务路由
        .nest("/api/v1/oss", oss_routes())
        // 添加中间件
        .layer(
            ServiceBuilder::new()
                .layer(TraceLayer::new_for_http())
                .layer(CorsLayer::permissive())
                .layer(DefaultBodyLimit::max(
                    get_global_file_size_config().max_file_size.bytes() as usize,
                )),
        )
        .with_state(state)
}

/// 文档处理相关路由
fn document_routes() -> Router<AppState> {
    Router::new()
        // 文档上传和解析
        .route("/upload", post(document_handler::upload_document))
        .route(
            "/uploadFromUrl",
            post(document_handler::download_document_from_url),
        )
        // 结构化文档生成
        .route(
            "/structured",
            post(document_handler::generate_structured_document),
        )
        // Markdown处理接口
        .route(
            "/markdown/parse",
            post(markdown_handler::parse_markdown_sections),
        )
        .route(
            "/markdown/sections",
            post(markdown_handler::parse_markdown_sections),
        )
        // 解析器管理和状态
        .route("/formats", get(document_handler::get_supported_formats))
        .route("/parser/stats", get(document_handler::get_parser_stats))
        .route("/parser/health", get(document_handler::check_parser_health))
}

/// 任务管理相关路由
fn task_routes() -> Router<AppState> {
    Router::new()
        // 任务CRUD操作
        .route("/", post(task_handler::create_task))
        .route("/", get(task_handler::list_tasks))
        .route("/{task_id}", get(task_handler::get_task))
        .route("/{task_id}", delete(task_handler::delete_task))
        .route("/{task_id}/result", get(task_handler::get_task_result))
        // 任务操作
        .route("/{task_id}/cancel", post(task_handler::cancel_task))
        .route("/{task_id}/retry", post(task_handler::retry_task))
        .route("/{task_id}/progress", get(task_handler::get_task_progress))
        // 批量操作
        .route("/batch", post(task_handler::batch_operation_tasks))
        // 任务统计和管理
        .route("/stats", get(task_handler::get_task_stats))
        .route("/cleanup", post(task_handler::cleanup_expired_tasks))
        // Markdown 结果
        .route(
            "/{task_id}/markdown/download",
            get(markdown_handler::download_markdown),
        )
        .route(
            "/{task_id}/markdown/url",
            get(markdown_handler::get_markdown_url),
        )
        // 目录和章节
        .route("/{task_id}/toc", get(toc_handler::get_document_toc))
        .route(
            "/{task_id}/section/{section_id}",
            get(toc_handler::get_section_content),
        )
        .route("/{task_id}/sections", get(toc_handler::get_all_sections))
}

/// 私有桶OSS服务相关路由
fn oss_routes() -> Router<AppState> {
    Router::new()
        // 文件上传到OSS
        .route("/upload", post(private_oss_handler::upload_file_to_oss))
        // 获取上传签名URL（4小时有效）
        .route(
            "/upload-sign-url",
            get(private_oss_handler::get_upload_sign_url),
        )
        // 获取下载签名URL（4小时有效）
        .route(
            "/download-sign-url",
            get(private_oss_handler::get_download_sign_url),
        )
        // 删除OSS文件
        .route("/delete", get(private_oss_handler::delete_file_from_oss))
}