提交qiming-mcp-proxy

qiming-mcp-proxy/document-parser/src/routes.rs

@@ -0,0 +1,126 @@
+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))
+}