Dinko Zdravac
e4ba6c815e
* Server file structure * Create HTTP server - graceful shutdown - routes - logging, CORS * gateways WIP * gateways API + swagger docs complete * Mixnodes API + swagger docs complete * Services API + swagger docs complete * Commit summary insert * Make troubleshooting DB easier * Summary API + swagger docs * Client log changes * QOL improvements - remove implicit panics via `as` - safer DTO conversions - add logging - new config
86 lines
2.5 KiB
Rust
86 lines
2.5 KiB
Rust
use anyhow::anyhow;
|
|
use axum::{response::Redirect, Router};
|
|
use tokio::net::ToSocketAddrs;
|
|
use tower_http::{cors::CorsLayer, trace::TraceLayer};
|
|
use utoipa::OpenApi;
|
|
use utoipa_swagger_ui::SwaggerUi;
|
|
|
|
use crate::http::{server::HttpServer, state::AppState};
|
|
|
|
pub(crate) mod gateways;
|
|
pub(crate) mod mixnodes;
|
|
pub(crate) mod services;
|
|
pub(crate) mod summary;
|
|
pub(crate) mod testruns;
|
|
|
|
pub(crate) struct RouterBuilder {
|
|
unfinished_router: Router<AppState>,
|
|
}
|
|
|
|
impl RouterBuilder {
|
|
pub(crate) fn with_default_routes() -> Self {
|
|
let router = Router::new()
|
|
.merge(
|
|
SwaggerUi::new("/swagger")
|
|
.url("/api-docs/openapi.json", super::api_docs::ApiDoc::openapi()),
|
|
)
|
|
.route(
|
|
"/",
|
|
axum::routing::get(|| async { Redirect::permanent("/swagger") }),
|
|
)
|
|
.nest(
|
|
"/v2",
|
|
Router::new()
|
|
.nest("/gateways", gateways::routes())
|
|
.nest("/mixnodes", mixnodes::routes())
|
|
.nest("/services", services::routes())
|
|
.nest("/summary", summary::routes()),
|
|
// .nest("/testruns", testruns::_routes()),
|
|
);
|
|
|
|
Self {
|
|
unfinished_router: router,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn with_state(self, state: AppState) -> RouterWithState {
|
|
RouterWithState {
|
|
router: self.finalize_routes().with_state(state),
|
|
}
|
|
}
|
|
|
|
fn finalize_routes(self) -> Router<AppState> {
|
|
// layers added later wrap earlier layers
|
|
self.unfinished_router
|
|
// CORS layer needs to wrap other API layers
|
|
.layer(setup_cors())
|
|
// logger should be outermost layer
|
|
.layer(TraceLayer::new_for_http())
|
|
}
|
|
}
|
|
|
|
pub(crate) struct RouterWithState {
|
|
router: Router,
|
|
}
|
|
|
|
impl RouterWithState {
|
|
pub(crate) async fn build_server<A: ToSocketAddrs>(
|
|
self,
|
|
bind_address: A,
|
|
) -> anyhow::Result<HttpServer> {
|
|
tokio::net::TcpListener::bind(bind_address)
|
|
.await
|
|
.map(|listener| HttpServer::new(self.router, listener))
|
|
.map_err(|err| anyhow!("Couldn't bind to address due to {}", err))
|
|
}
|
|
}
|
|
|
|
fn setup_cors() -> CorsLayer {
|
|
use axum::http::Method;
|
|
CorsLayer::new()
|
|
.allow_origin(tower_http::cors::Any)
|
|
.allow_methods([Method::POST, Method::GET, Method::PATCH, Method::OPTIONS])
|
|
.allow_headers(tower_http::cors::Any)
|
|
.allow_credentials(false)
|
|
}
|