Add OpenAPI docs

This commit is contained in:
Andras Schmelczer 2024-12-08 15:15:10 +00:00
parent 53f1a3dcc6
commit d5a5367e1e
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
5 changed files with 51 additions and 20 deletions

View file

@ -1,13 +1,18 @@
use crate::app_state::AppState;
use aide::{
axum::{
routing::{delete, get, put},
ApiRouter,
},
openapi::{Info, OpenApi},
scalar::Scalar,
};
use anyhow::Context;
use anyhow::Result;
use axum::extract::DefaultBodyLimit;
use axum::extract::WebSocketUpgrade;
use axum::response::Response;
use axum::routing::delete;
use axum::{routing::get, routing::put, Router};
use axum::response::{IntoResponse, Response};
use axum::{extract::DefaultBodyLimit, Extension};
use axum::{extract::WebSocketUpgrade, Json};
use log::info;
mod delete_document;
mod fetch_latest_document_version;
mod fetch_latest_documents;
@ -20,35 +25,48 @@ pub async fn create_server(app_state: AppState) -> Result<()> {
&app_state.config.server.host, &app_state.config.server.port
);
let app = Router::new()
.route(
let mut api = OpenApi {
info: Info {
description: Some("an example API".to_string()),
..Info::default()
},
..OpenApi::default()
};
let app = ApiRouter::new()
.api_route(
"/vaults/:vault_id/documents/latest",
get(fetch_latest_documents::fetch_latest_documents),
)
.route(
.api_route(
"/vaults/:vault_id/documents/:document_id/versions/:parent_version_id",
put(update_document::update_document),
)
.route(
.api_route(
"/vaults/:vault_id/documents/:document_id",
delete(delete_document::delete_document),
)
.route(
.api_route(
"/vaults/:vault_id/documents/:document_id/versions/latest",
get(fetch_latest_document_version::fetch_latest_document_version),
)
.route("/ws", get(handler))
.api_route("/ws", get(handler))
.route("/", Scalar::new("/api.json").axum_route())
.route("/api.json", axum::routing::get(serve_api))
.layer(DefaultBodyLimit::max(
app_state.config.server.max_body_size_mb * 1024 * 1024,
))
.with_state(app_state);
.with_state(app_state)
.finish_api(&mut api)
.layer(Extension(api))
.into_make_service();
let listener = tokio::net::TcpListener::bind(address.clone())
.await
.with_context(|| format!("Failed to bind to address: {}", address))?;
info!(
"Listening on {}",
"Listening on http://{}",
listener
.local_addr()
.context("Failed to get local address")?
@ -65,3 +83,7 @@ async fn handler(ws: WebSocketUpgrade) -> Response {
// ...
})
}
async fn serve_api(Extension(api): Extension<OpenApi>) -> impl IntoResponse {
Json(api)
}