add the handler for fetching all sites

src/bin/server.rs

@@ -1,10 +1,9 @@
 use dotenvy::dotenv;
 use rocket::{self, routes};
 use rocket_db_pools::Database;
-use sqlx::MySqlPool;
 
 use startpage::routes::upload::upload;
-use startpage::routes::{auth, category, user};
+use startpage::routes::{auth, category, site, user};
 use startpage::state::AppState;
 use startpage::utils::calculate_expires;
 use startpage::Db;
@@ -75,6 +74,7 @@ async fn main() -> Result<(), rocket::Error> {
                 category::delete_site,
             ],
         )
+        .mount("/api/sites", routes![site::all])
         .mount("/api/upload", routes![upload])
         .launch()
         .await?;

src/handlers.rs

@@ -1,3 +1,4 @@
 pub mod auth;
 pub mod category;
+pub mod site;
 pub mod user;

src/handlers/category.rs

@@ -10,7 +10,7 @@ use crate::response;
 use crate::response::WithTotal;
 use crate::Db;
 
-pub async fn get_all_categories(
+pub async fn get_categories(
     page: i64,
     size: i64,
     db: &mut Connection<Db>,

src/handlers/site.rs

@@ -0,0 +1,27 @@
+use rocket_db_pools::Connection;
+use sqlx::{query, query_as, Row};
+
+use crate::errors::ServiceError;
+use crate::response::site::SiteWithCategory;
+use crate::response::WithTotal;
+use crate::Db;
+
+pub async fn get_sites(
+    page: i64,
+    size: i64,
+    db: &mut Connection<Db>,
+) -> Result<WithTotal<SiteWithCategory>, ServiceError> {
+    let count = query(r#"SELECT COUNT(id) AS count FROM site"#)
+        .fetch_one(&mut ***db)
+        .await?
+        .get::<i64, &str>("count");
+
+    let sites = query_as::<_, SiteWithCategory>(
+        r#"SELECT site.id AS id, site.name AS name, site.url AS url, site.icon AS icon, site.description AS description, category.name AS category FROM site INNER JOIN category INNER JOIN category_site ON site.id = category_site.site_id AND category.id = category_site.category_id LIMIT ? OFFSET ?"#,
+    ).bind(size).bind(page * size).fetch_all(&mut ***db).await?;
+
+    Ok(WithTotal {
+        total: count,
+        data: sites,
+    })
+}

src/response/site.rs

@@ -9,3 +9,13 @@ pub struct Site {
     pub description: String,
     pub icon: String,
 }
+
+#[derive(Debug, Serialize, Deserialize, FromRow)]
+pub struct SiteWithCategory {
+    pub id: i64,
+    pub name: String,
+    pub url: String,
+    pub description: String,
+    pub icon: String,
+    pub category: String,
+}

src/routes.rs

@@ -1,4 +1,5 @@
 pub mod auth;
 pub mod category;
+pub mod site;
 pub mod upload;
 pub mod user;

src/routes/category.rs

@@ -7,15 +7,14 @@ use rocket::{delete, get, post, put};
 use rocket_db_pools::Connection;
 
 use crate::handlers::category::{
-    self, add_category, delete_category, get_all_categories, modify_site, update_category,
+    self, add_category, delete_category, get_categories, modify_site, update_category,
 };
 use crate::middlewares::JwtMiddleware;
 use crate::request::category::{CreateCategory, UpdateCategory};
 use crate::request::site::{CreateSite, UpdateSite};
 use crate::response::category::Category;
 use crate::response::site::Site;
 use crate::response::WithTotal;
-use crate::state::AppState;
 use crate::Db;
 
 #[get("/?<page>&<size>")]
@@ -28,13 +27,13 @@ pub async fn all(
 
     let size = size.unwrap_or(10);
 
-    let categories = get_all_categories(page, size, &mut db).await.map_err(|e| {
+    let result = get_categories(page, size, &mut db).await.map_err(|e| {
         error!("{}", e);
 
         e.into()
     })?;
 
-    Ok(Json(categories))
+    Ok(Json(result))
 }
 
 #[put("/<id>", format = "json", data = "<category>")]

src/routes/site.rs

@@ -0,0 +1,29 @@
+use log::error;
+use rocket::get;
+use rocket::http::Status;
+use rocket::serde::json::Json;
+use rocket_db_pools::Connection;
+
+use crate::handlers::site::get_sites;
+use crate::response::site::SiteWithCategory;
+use crate::response::WithTotal;
+use crate::Db;
+
+#[get("/?<page>&<size>")]
+pub async fn all(
+    page: Option<i64>,
+    size: Option<i64>,
+    mut db: Connection<Db>,
+) -> Result<Json<WithTotal<SiteWithCategory>>, Status> {
+    let page = page.unwrap_or(0);
+
+    let size = size.unwrap_or(10);
+
+    let result = get_sites(page, size, &mut db).await.map_err(|e| {
+        error!("{}", e);
+
+        e.into()
+    })?;
+
+    Ok(Json(result))
+}