Move health and docs routes to meta module

Signed-off-by: Lilly Rose Berner <[email protected]>

Author: LostLuma

echo/src/main.rs

@@ -6,7 +6,7 @@ use database::{background_task::BackgroundTask, PgDatabase};
 use figment::providers::{Format, Json};
 use rocket::{catchers, fairing::AdHoc, figment, routes};
 use rocket_db_pools::Database;
-use routes::{docs, health, pastes, security};
+use routes::{meta, pastes, security};
 use scanners::InitScanners;
 
 pub mod config;
@@ -21,11 +21,11 @@ pub mod utils;
 #[rocket::launch]
 fn rocket() -> _ {
     let routes = routes![
-        routes::get_root,
         cors::snatcher,
-        docs::get_docs,
-        docs::get_spec,
-        health::health,
+        meta::get_root,
+        meta::get_docs,
+        meta::get_spec,
+        meta::get_health,
         pastes::get_paste,
         pastes::create_paste_simple,
         pastes::create_paste_structured,

echo/src/routes/docs.rs

@@ -0,0 +1,49 @@
+use rocket::{
+    get,
+    response::{
+        content::{RawHtml, RawJson},
+        Redirect,
+    },
+    serde::json::Json,
+    uri,
+};
+use rocket_db_pools::{
+    sqlx::{self, Row},
+    Connection,
+};
+use scalar_doc::{Documentation, Theme};
+
+use crate::{
+    database::PgDatabase,
+    result::{HTTPError, Result},
+};
+
+#[get("/")]
+pub fn get_root() -> Redirect {
+    Redirect::temporary(uri!("/docs"))
+}
+
+#[get("/docs")]
+pub fn get_docs() -> RawHtml<String> {
+    let docs = Documentation::new("MystBin API Documentation", "/docs/spec.json")
+        .theme(Theme::Alternate)
+        .build()
+        .unwrap();
+
+    RawHtml(docs)
+}
+
+#[get("/docs/spec.json")]
+pub fn get_spec() -> RawJson<&'static str> {
+    RawJson(include_str!("../../openapi.json"))
+}
+
+#[get("/health")]
+pub async fn get_health(mut db: Connection<PgDatabase>) -> Result<String> {
+    let result = sqlx::query("SELECT 'ok'").fetch_one(&mut **db).await;
+
+    match result {
+        Ok(row) => Ok(row.get(0)),
+        Err(_) => Err(HTTPError::new(503, "Database connection unhealthy.")),
+    }
+}

echo/src/routes/health.rs

@@ -1,11 +1,3 @@
-use rocket::{get, response::Redirect, uri};
-
-pub mod docs;
-pub mod health;
+pub mod meta;
 pub mod pastes;
 pub mod security;
-
-#[get("/")]
-pub fn get_root() -> Redirect {
-    Redirect::temporary(uri!("/docs"))
-}