{title}

mod auth; mod chat; mod kb; pub mod obj; mod projects; mod settings; mod timers; mod workflows; use std::sync::Arc; use axum::{ body::Body, extract::{Path, State, Request}, http::StatusCode, response::{IntoResponse, Response}, routing::{get, any}, Json, Router, }; use crate::AppState; pub(crate) type ApiResult = Result, Response>; pub(crate) fn db_err(e: sqlx::Error) -> Response { tracing::error!("Database error: {}", e); (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()).into_response() } pub fn router(state: Arc) -> Router { Router::new() .merge(projects::router(state.clone())) .merge(workflows::router(state.clone())) .merge(timers::router(state.clone())) .merge(kb::router(state.clone())) .merge(settings::router(state.clone())) .merge(chat::router(state.clone())) .merge(auth::router(state.clone())) .route("/projects/{id}/files/{*path}", get(serve_project_file)) .route("/projects/{id}/app/{*path}", any(proxy_to_service).with_state(state.clone())) .route("/projects/{id}/app/", any(proxy_to_service_root).with_state(state)) } async fn proxy_to_service_root( State(state): State>, Path(project_id): Path, req: Request, ) -> Response { proxy_impl(&state, &project_id, "/", req).await } async fn proxy_to_service( State(state): State>, Path((project_id, path)): Path<(String, String)>, req: Request, ) -> Response { proxy_impl(&state, &project_id, &format!("/{}", path), req).await } async fn proxy_impl( state: &AppState, project_id: &str, path: &str, req: Request, ) -> Response { let port = match state.agent_mgr.get_service_port(project_id).await { Some(p) => p, None => return (StatusCode::SERVICE_UNAVAILABLE, "服务未启动").into_response(), }; let query = req.uri().query().map(|q| format!("?{}", q)).unwrap_or_default(); let url = format!("http://127.0.0.1:{}{}{}", port, path, query); let client = reqwest::Client::new(); let method = req.method().clone(); let headers = req.headers().clone(); let body_bytes = match axum::body::to_bytes(req.into_body(), 10 * 1024 * 1024).await { Ok(b) => b, Err(_) => return (StatusCode::BAD_REQUEST, "请求体过大").into_response(), }; let mut upstream_req = client.request(method, &url); for (key, val) in headers.iter() { if key != "host" { upstream_req = upstream_req.header(key, val); } } upstream_req = upstream_req.body(body_bytes); match upstream_req.send().await { Ok(resp) => { let status = StatusCode::from_u16(resp.status().as_u16()).unwrap_or(StatusCode::BAD_GATEWAY); let resp_headers = resp.headers().clone(); let body = resp.bytes().await.unwrap_or_default(); let mut response = (status, body).into_response(); for (key, val) in resp_headers.iter() { if let Ok(name) = axum::http::header::HeaderName::from_bytes(key.as_ref()) { response.headers_mut().insert(name, val.clone()); } } response } Err(_) => (StatusCode::BAD_GATEWAY, "无法连接到后端服务").into_response(), } } async fn serve_project_file( Path((project_id, file_path)): Path<(String, String)>, ) -> Response { let full_path = std::path::PathBuf::from("/app/data/workspaces") .join(&project_id) .join(&file_path); // Prevent path traversal if file_path.contains("..") { return (StatusCode::BAD_REQUEST, "Invalid path").into_response(); } match tokio::fs::read(&full_path).await { Ok(bytes) => { // Render markdown files as HTML if full_path.extension().is_some_and(|e| e == "md") { let md = String::from_utf8_lossy(&bytes); let html = render_markdown_page(&md, &file_path); return ( [(axum::http::header::CONTENT_TYPE, "text/html; charset=utf-8".to_string())], html, ).into_response(); } let mime = mime_guess::from_path(&full_path) .first_or_octet_stream() .to_string(); ( [(axum::http::header::CONTENT_TYPE, mime)], bytes, ).into_response() } Err(_) => (StatusCode::NOT_FOUND, "File not found").into_response(), } } fn render_markdown_page(markdown: &str, title: &str) -> String { use pulldown_cmark::{Parser, Options, html}; let mut opts = Options::empty(); opts.insert(Options::ENABLE_TABLES); opts.insert(Options::ENABLE_STRIKETHROUGH); opts.insert(Options::ENABLE_TASKLISTS); let parser = Parser::new_ext(markdown, opts); let mut body = String::new(); html::push_html(&mut body, parser); format!(r#" {title}

{body}

"#, title = title, body = body) }