From 680867b4bb5e2708f02fc3f8c21c3563975c44c7 Mon Sep 17 00:00:00 2001
From: WriterV2 <primary-v2@protonmail.com>
Date: Thu, 7 Nov 2024 18:35:27 +0100
Subject: Feat: Add admin API route to delete story by product id

---
 src/db/mod.rs     |  4 ++++
 src/db/product.rs |  7 ++++++-
 src/db/story.rs   | 13 +++++++++++++
 3 files changed, 23 insertions(+), 1 deletion(-)

(limited to 'src/db')

diff --git a/src/db/mod.rs b/src/db/mod.rs
index e44c9ad..51665a3 100644
--- a/src/db/mod.rs
+++ b/src/db/mod.rs
@@ -25,4 +25,8 @@ pub trait ProductDatabaseHandler {
     ) -> Result<Self, AppError>
     where
         Self: Sized;
+
+    async fn delete(pool: &SqlitePool, id: i64) -> Result<(), AppError>
+    where
+        Self: Sized;
 }
diff --git a/src/db/product.rs b/src/db/product.rs
index 06217c0..c21734e 100644
--- a/src/db/product.rs
+++ b/src/db/product.rs
@@ -5,7 +5,6 @@ use sqlx::{QueryBuilder, Sqlite, SqlitePool, Transaction};
 
 use crate::error::AppError;
 
-use super::producttag::ProductTag;
 use super::tag::Tag;
 use super::ProductDatabaseHandler;
 
@@ -50,6 +49,12 @@ impl Product {
         }
         Ok(product)
     }
+
+    pub async fn delete(tx: &mut Transaction<'static, Sqlite>, pid: i64) -> Result<(), AppError> {
+        sqlx::query!("DELETE FROM producttag WHERE pid = $1", pid).execute(&mut **tx).await?;
+        sqlx::query!("DELETE FROM product WHERE id = $1", pid).execute(&mut **tx).await?;
+        Ok(())
+    }
 }
 
 pub trait ProductMarker {
diff --git a/src/db/story.rs b/src/db/story.rs
index 137c964..9e5fdb9 100644
--- a/src/db/story.rs
+++ b/src/db/story.rs
@@ -99,6 +99,19 @@ impl ProductDatabaseHandler for Story {
 
         Ok(story)
     }
+
+    async fn delete(pool: &SqlitePool, pid: i64) -> Result<(), AppError>
+    where
+        Self: Sized,
+    {
+        let mut tx = pool.begin().await?;
+        sqlx::query!("DELETE FROM story WHERE pid = $1", pid)
+            .execute(&mut *tx)
+            .await?;
+        Product::delete(&mut tx, pid).await?;
+        tx.commit().await?;
+        Ok(())
+    }
 }
 
 impl PageBuilder for Story {
-- 
cgit v1.2.3