From f46a5bf4924bb7cc702d677512d4aa9831254cb9 Mon Sep 17 00:00:00 2001
From: Andrew4Coding <andrewdevitoaryo@gmail.com>
Date: Fri, 28 Mar 2025 11:35:43 +0800
Subject: [PATCH] Implement publish function in Program service and Program
 controller.

---
 src/controller/mod.rs     |  2 +-
 src/controller/product.rs |  9 +++++++++
 src/service/product.rs    | 18 ++++++++++++++++++
 3 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index df6df1a..9107e72 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -6,7 +6,7 @@ use rocket::fairing::AdHoc;
 pub fn route_stage() -> AdHoc {
     return AdHoc::on_ignite("Initializing controller routes...", |rocket| async {
         rocket
-            .mount("/product", routes![product::create, product::list, product::read, product::delete])
+            .mount("/product", routes![product::create, product::list, product::read, product::delete, product::publish])
             .mount("/notification", routes![notification::subscribe, notification::unsubscribe])
     });
 }
\ No newline at end of file
diff --git a/src/controller/product.rs b/src/controller/product.rs
index 1cc5384..9e976c8 100644
--- a/src/controller/product.rs
+++ b/src/controller/product.rs
@@ -37,3 +37,12 @@ pub fn delete(id: usize) -> Result<Json<Product>> {
         Err(e) => Err(e)
     };
 }
+
+
+#[post("/<id>/publish")]
+pub fn publish(id: usize) -> Result<Json<Product>> {
+    return match ProductService::publish(id) {
+        Ok(f) => Ok(Json::from(f)),
+        Err(e) => Err(e)
+    };
+}
\ No newline at end of file
diff --git a/src/service/product.rs b/src/service/product.rs
index 775bf09..44424d0 100644
--- a/src/service/product.rs
+++ b/src/service/product.rs
@@ -5,6 +5,8 @@ use bambangshop::{Result, compose_error_response};
 use crate::model::product::Product;
 use crate::repository::product::ProductRepository;
 
+use super::notification::NotificationService;
+
 pub struct ProductService;
 
 impl ProductService {
@@ -42,4 +44,20 @@ impl ProductService {
 
         return Ok(Json::from(product));
     }
+
+    pub fn publish(id: usize) -> Result<Product> {
+        let product_opt = ProductRepository::get_by_id(id);
+        if product_opt.is_none() {
+            return Err(compose_error_response(
+                Status::NotFound,
+                String::from("Product not found.")
+            ));
+        };
+
+        let product = product_opt.unwrap();
+
+        NotificationService.notify(&product.product_type, "PROMOTION", product.clone());
+
+        return Ok(product);
+    }
 }
-- 
GitLab