Dinko Zdravac
cc983963d4
* Compile & copy wg probe * Node status agent WIP * Enable debug logging * Agent submits results - add clap to agent - agent runs network probe - /submit endpoint on NS API * Build clients with timeouts * Update logging and dev scripts * Replace /blaclisted endpoint * Testruns fully functional - task that queues testruns periodically - testruns read/write in DB * Probe scores fully working - testruns are assigned on API - submit updates testruns correctly on NS API side - agent registers with API - agent submits results correctly * Clippy fixes * PR feedback * Clippy again * PR feedback * Run clippy earlier in CI * Make refresh delay configurable in server & agent
89 lines
2.6 KiB
Rust
89 lines
2.6 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(
|
|
"/internal",
|
|
Router::new().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)
|
|
}
|