add the handler to delete a site for a specific category

src/bin/server.rs

@@ -72,6 +72,7 @@ async fn main() -> Result<(), rocket::Error> {
                 category::add_site,
                 category::get_sites,
                 category::update_site,
+                category::delete_site,
             ],
         )
         .launch()

src/handlers/category.rs

@@ -229,3 +229,34 @@ pub async fn modify_site(
 
     Ok(())
 }
+
+pub async fn delete_site(
+    category_id: &str,
+    site_id: &str,
+    state: &State<AppState>,
+) -> Result<(), ServiceError> {
+    query(
+        r#"SELECT category_id, site_id FROM category_site WHERE category_id = ? AND site_id = ?"#,
+    )
+    .bind(category_id)
+    .bind(site_id)
+    .fetch_one(&state.pool)
+    .await
+    .map_err(|e| match e {
+        sqlx::Error::RowNotFound => ServiceError::BadRequest(String::from("Site not found")),
+        _ => ServiceError::DatabaseError(e),
+    })?;
+
+    query(r#"DELETE FROM category_site WHERE category_id = ? AND site_id = ?"#)
+        .bind(category_id)
+        .bind(site_id)
+        .execute(&state.pool)
+        .await?;
+
+    query(r#"DELETE FROM site WHERE id = ?"#)
+        .bind(site_id)
+        .execute(&state.pool)
+        .await?;
+
+    Ok(())
+}

src/routes/category.rs

@@ -112,6 +112,24 @@ pub async fn update_site<'r>(
     Ok(())
 }
 
+#[delete("/<id>/site/<site_id>")]
+pub async fn delete_site(
+    id: &str,
+    site_id: &str,
+    state: &State<AppState>,
+    _jwt: JwtMiddleware,
+) -> Result<(), Status> {
+    category::delete_site(id, site_id, state)
+        .await
+        .map_err(|e| {
+            error!("{}", e);
+
+            e.into()
+        })?;
+
+    Ok(())
+}
+
 #[get("/<id>/sites")]
 pub async fn get_sites(id: &str, state: &State<AppState>) -> Result<Json<Vec<Site>>, Status> {
     let sites = category::get_sites(id, state).await.map_err(|e| {