From 2e3eaaebc018eea51150a332ece1cf40a54a3d7d Mon Sep 17 00:00:00 2001
From: raja <rafaelraja.sitorus@gmail.com>
Date: Fri, 28 Mar 2025 21:27:03 +0800
Subject: [PATCH] Implement publish function in Program service and Program c
 ontroller.

---
 src/controller/mod.rs     |  4 ++--
 src/controller/product.rs |  8 ++++++++
 src/service/product.rs    | 14 ++++++++++++++
 3 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index df6df1a..4233ca8 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -4,9 +4,9 @@ pub mod notification;
 use rocket::fairing::AdHoc;
 
 pub fn route_stage() -> AdHoc {
-    return AdHoc::on_ignite("Initializing controller routes...", |rocket| async {
+    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..ab9159f 100644
--- a/src/controller/product.rs
+++ b/src/controller/product.rs
@@ -37,3 +37,11 @@ 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)
+    };
+}
diff --git a/src/service/product.rs b/src/service/product.rs
index 775bf09..64261da 100644
--- a/src/service/product.rs
+++ b/src/service/product.rs
@@ -42,4 +42,18 @@ impl ProductService {
 
         return Ok(Json::from(product));
     }
+
+    pub fn publish(id: usize) -> Result<Product> {
+        let product_opt: Option<Product> = 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 = product_opt.unwrap();
+        
+        NotificationService::notify(&product.product_type, "PROMOTION", product.clone());
+        return Ok(product);
+    }
 }
-- 
GitLab