bfb28e3
pub mod metrics;
pub mod openapi;
use axum::Router;
use export::AppState;
use export::middleware::AppTraceLayer;
use session::{SessionLayer, store};
use tower_http::limit::RequestBodyLimitLayer;
use tower_http::trace::{DefaultMakeSpan, DefaultOnRequest, DefaultOnResponse, TraceLayer};
use tracing::Level;
const MAX_REQUEST_BODY_BYTES: usize = 2 * 1024 * 1024;
pub async fn router(state: AppState) -> Result<Router, axum_session::SessionError> {
let store = store::connect(state.redis.clone(), &state.config.session).await?;
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);
let public: Router = Router::new()
.merge(auth::routes::router(auth.clone()).with_state(auth.clone()))
.merge(chat::routes::router(chat.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()));
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()))
.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_request(DefaultOnRequest::new().level(Level::INFO))
.on_response(DefaultOnResponse::new().level(Level::INFO)),
)
.layer(metrics::metrics_layer())
.layer(AppTraceLayer::new());
Ok(router)
}