add the fetch site API

src/bin/server.rs

@@ -92,7 +92,13 @@ async fn main() -> Result<(), rocket::Error> {
         .mount("/api/sites", routes![site::all])
         .mount(
             "/api/site",
-            routes![site::add, site::update, site::delete, site::analytics],
+            routes![
+                site::get,
+                site::add,
+                site::update,
+                site::delete,
+                site::analytics
+            ],
         )
         .mount("/api/upload", routes![upload])
         .mount(upload_url, FileServer::from(upload_dir))

src/handlers/site.rs

@@ -5,6 +5,7 @@ use crate::errors::ServiceError;
 use crate::models::category::Category;
 use crate::models::site::Site;
 use crate::request::site::{CreateSite, UpdateSite};
+use crate::response::site::Site as SiteResponse;
 use crate::response::site::SiteWithCategory;
 use crate::response::WithTotal;
 use crate::MySQLDb;
@@ -259,3 +260,18 @@ pub async fn analytics(id: &str, db: &mut Connection<MySQLDb>) -> Result<(), Ser
 
     Ok(())
 }
+
+pub async fn get_site(id: i64, db: &mut Connection<MySQLDb>) -> Result<SiteResponse, ServiceError> {
+    let record = query_as::<_, Site>(
+        r#"SELECT id, name, url, description, icon, sort_order, visit_count, created_at, updated_at FROM site WHERE id = ?"#,
+    )
+    .bind(id)
+    .fetch_one(&mut ***db)
+    .await
+    .map_err(|e| match e {
+        sqlx::Error::RowNotFound => ServiceError::BadRequest(String::from("Site not found")),
+        _ => ServiceError::DatabaseError(e),
+    })?;
+
+    Ok(record.into())
+}

src/response/site.rs

@@ -1,6 +1,8 @@
 use serde::{Deserialize, Serialize};
 use sqlx::FromRow;
 
+use crate::models::site::Site as SiteModel;
+
 #[derive(Debug, Serialize, Deserialize, FromRow)]
 pub struct Site {
     pub id: i64,
@@ -21,3 +23,16 @@ pub struct SiteWithCategory {
     pub category: String,
     pub visit_count: i64,
 }
+
+impl From<SiteModel> for Site {
+    fn from(site: SiteModel) -> Self {
+        Self {
+            id: site.id,
+            name: site.name,
+            url: site.url,
+            description: site.description,
+            icon: site.icon,
+            visit_count: site.visit_count,
+        }
+    }
+}

src/routes/site.rs

@@ -9,11 +9,36 @@ use crate::guards::jwt::Middleware;
 use crate::handlers::site;
 use crate::handlers::site::get_sites;
 use crate::request::site::{CreateSite, UpdateSite};
-use crate::response::site::SiteWithCategory;
+use crate::response::site::{Site, SiteWithCategory};
 use crate::response::WithTotal;
 use crate::utils::standardize_url;
 use crate::MySQLDb;
 
+#[get("/<id>")]
+pub async fn get(
+    id: i64,
+    config: &State<Config>,
+    mut db: Connection<MySQLDb>,
+) -> Result<Json<Site>, Status> {
+    let mut site = site::get_site(id, &mut db).await.map_err(|e| {
+        error!("{}", e);
+
+        e.status()
+    })?;
+
+    let icon = site.icon.clone();
+
+    let icon = if icon.starts_with("http") || icon.starts_with("https") {
+        icon
+    } else {
+        format!("{}/{}", config.upload_url, icon)
+    };
+
+    site.icon = icon;
+
+    Ok(Json(site))
+}
+
 #[get("/?<page>&<size>&<search>")]
 pub async fn all(
     page: Option<i64>,