feat(serve): 404 for document not found

mdn · Dec 5, 2024 · 3435feb · 3435feb
1 parent 5f0f7e6
commit 3435feb
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/crates/rari-cli/serve.rs b/crates/rari-cli/serve.rs
@@ -157,16 +157,23 @@ fn get_search_index(locale: Locale) -> Result<Vec<SearchItem>, DocError> {
 }
 
 #[derive(Debug)]
-struct AppError(anyhow::Error);
+struct AppError(DocError);
 
 impl IntoResponse for AppError {
     fn into_response(self) -> Response<Body> {
-        (StatusCode::INTERNAL_SERVER_ERROR, error!("🤷: {}", self.0)).into_response()
+        match self.0 {
+            DocError::RariIoError(_) | DocError::IOError(_) | DocError::PageNotFound(_, _) => {
+                (StatusCode::NOT_FOUND, "").into_response()
+            }
+
+            _ => (StatusCode::INTERNAL_SERVER_ERROR, error!("🤷: {}", self.0)).into_response(),
+        }
     }
 }
+
 impl<E> From<E> for AppError
 where
-    E: Into<anyhow::Error>,
+    E: Into<DocError>,
 {
     fn from(err: E) -> Self {
         Self(err.into())