From 3aece38e9dfab762efc52afe066ce418c36e673a Mon Sep 17 00:00:00 2001
From: Jonas Platte <jplatte+git@posteo.de>
Date: Sun, 13 Feb 2022 13:59:27 +0100
Subject: [PATCH] Add a not-found route

---
 src/main.rs | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index a96bef50..6aa08704 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,6 +11,7 @@ use std::{future::Future, io, net::SocketAddr, sync::Arc, time::Duration};
 
 use axum::{
     extract::{FromRequest, MatchedPath},
+    handler::Handler,
     response::IntoResponse,
     routing::{get, on, MethodFilter},
     Router,
@@ -22,10 +23,13 @@ use figment::{
 };
 use http::{
     header::{self, HeaderName},
-    Method,
+    Method, Uri,
 };
 use opentelemetry::trace::{FutureExt, Tracer};
-use ruma::{api::IncomingRequest, Outgoing};
+use ruma::{
+    api::{client::error::ErrorKind, IncomingRequest},
+    Outgoing,
+};
 use tokio::{signal, sync::RwLock};
 use tower::ServiceBuilder;
 use tower_http::{
@@ -321,6 +325,7 @@ fn routes() -> Router {
         .ruma_route(server_server::get_profile_information_route)
         .ruma_route(server_server::get_keys_route)
         .ruma_route(server_server::claim_keys_route)
+        .fallback(not_found.into_service())
 }
 
 async fn shutdown_signal(handle: ServerHandle) {
@@ -349,6 +354,10 @@ async fn shutdown_signal(handle: ServerHandle) {
     handle.graceful_shutdown(Some(Duration::from_secs(30)));
 }
 
+async fn not_found(_uri: Uri) -> impl IntoResponse {
+    Error::BadRequest(ErrorKind::NotFound, "Unknown or unimplemented route")
+}
+
 trait RouterExt {
     fn ruma_route<H, T>(self, handler: H) -> Self
     where