028baf2
pub mod metrics;
pub mod openapi;
pub mod spa;
use axum::Router;
use export::AppState;
use export::middleware::AppTraceLayer;
use session::{SessionLayer, store};
use tower_http::limit::RequestBodyLimitLayer;
use tower_http::trace::{DefaultMakeSpan, DefaultOnResponse, TraceLayer};
use tracing::Level;
const MAX_REQUEST_BODY_BYTES: usize = 2 * 1024 * 1024;
fn build_cors_layer(web: &config::web::WebConfig) -> tower_http::cors::CorsLayer {
use axum::http::{HeaderName, header};
use tower_http::cors::{AllowOrigin, CorsLayer};
if web.cors_allow_origins.is_empty() {
return CorsLayer::new();
}
let origins: Vec<axum::http::HeaderValue> = web
.cors_allow_origins
.iter()
.filter_map(|o| o.parse().ok())
.collect();
CorsLayer::new()
.allow_origin(AllowOrigin::list(origins))
.allow_credentials(web.cors_allow_credentials)
.allow_methods([
axum::http::Method::GET,
axum::http::Method::POST,
axum::http::Method::PUT,
axum::http::Method::PATCH,
axum::http::Method::DELETE,
axum::http::Method::OPTIONS,
])
.allow_headers([
header::CONTENT_TYPE,
header::AUTHORIZATION,
HeaderName::from_static("x-request-id"),
])
}
pub async fn router(state: AppState) -> Result<Router, axum_session::SessionError> {
let store = store::connect(state.redis.clone(), &state.config.session).await?;
let cors = build_cors_layer(&state.config.web);
let auth = auth::services::AuthService::new(state.clone());
let chat = chat::services::service::ChatService::new(state.clone());
let issues = issues::services::service::IssueService::new(state.clone());
let space = space::services::service::SpaceService::new(state.clone());
let user = user::services::service::UserService::new(state.clone());
let git = git::service::GitService::new(state.clone());
let pulls = pulls::services::service::PullService::new(state.clone());
let health_state = state;
let public: Router = Router::new()
.merge(auth::routes::router(auth.clone()).with_state(auth.clone()))
.merge(chat::routes::router(chat.clone(), space.clone()).with_state(chat))
.merge(issues::routes::router(issues.clone()).with_state(issues))
.merge(space::routes::router(space.clone()).with_state(space))
.merge(user::routes::router(user.clone()).with_state(user))
.merge(git::routes::router(git.clone()).with_state(git.clone()))
.merge(pulls::routes::router().with_state(pulls));
let internal: Router = Router::new()
.merge(git::routes::internal_router(git.clone()).with_state(git))
.route_layer(axum::middleware::from_fn_with_state(
auth.clone(),
auth::share::guard::admin_guard,
));
let router: Router = Router::new()
.nest("/api/v1", public)
.nest("/api/v1/internal", internal)
.nest("/pages", auth::routes::pages_router(auth.clone()))
.merge(export::health::router(health_state))
.fallback(spa::fallback)
.layer(cors)
.layer(SessionLayer::new(store))
.layer(RequestBodyLimitLayer::new(MAX_REQUEST_BODY_BYTES))
.layer(
TraceLayer::new_for_http()
.make_span_with(
DefaultMakeSpan::new()
.level(Level::INFO)
.include_headers(false),
)
.on_response(DefaultOnResponse::new().level(Level::INFO)),
)
.layer(metrics::metrics_layer())
.layer(AppTraceLayer::new());
Ok(router)
}