Add fallback handlers for 404 and 405

1 files changed, 16 insertions, 1 deletions

diff --git a/src/routers/mod.rs b/src/routers/mod.rs
index b7bc1e6..f9ec049 100644
--- a/src/routers/mod.rs
+++ b/src/routers/mod.rs
@@ -1,10 +1,25 @@
 mod account;
 mod queue;
 
+use axum::extract::OriginalUri;
 use utoipa::OpenApi;
 use utoipa_axum::router::OpenApiRouter;
 
-use crate::{AppOpenApi, AppState};
+use crate::{AppOpenApi, AppState, ClientError, ErrorResponse};
+
+pub async fn not_found(OriginalUri(path): OriginalUri) -> ErrorResponse {
+    ClientError::NotFound {
+        path: path.to_string(),
+    }
+    .into()
+}
+
+pub async fn method_not_allowed(OriginalUri(path): OriginalUri) -> ErrorResponse {
+    ClientError::MethodNotAllowed {
+        path: path.to_string(),
+    }
+    .into()
+}
 
 pub fn router() -> OpenApiRouter<AppState> {
     OpenApiRouter::with_openapi(AppOpenApi::openapi())