From 257339e80c6d5d5b06f6712c68b38d6024603b5b Mon Sep 17 00:00:00 2001
From: Andrew4Coding <andrewdevitoaryo@gmail.com>
Date: Fri, 28 Mar 2025 01:16:35 +0800
Subject: [PATCH] Implement subscribe function in Notification controller.

---
 src/controller/mod.rs          |  4 ++--
 src/controller/notification.rs | 10 +++++++++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index fe2103d..fcdeefc 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -1,5 +1,5 @@
 pub mod product;
-pub  mod notification;
+pub mod notification;
 
 use rocket::fairing::AdHoc;
 
@@ -7,6 +7,6 @@ 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("/notification", routes![])
+            .mount("/notification", routes![notification::subscribe])
     });
 }
\ No newline at end of file
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index 7305876..0f18dd2 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -3,4 +3,12 @@ use rocket::serde::json::Json;
 
 use bambangshop::Result;
 use crate::model::subscriber::Subscriber;
-use crate::service::notification::NotificationService;
\ No newline at end of file
+use crate::service::notification::NotificationService;
+
+# [post("/subscribe/<product_type>", data = "<subscriber>")]
+pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Created<Json<Subscriber>>> {
+    return match NotificationService::subscribe(product_type, subscriber.into_inner()) {
+        Ok(f) => Ok(Created::new("/").body(Json::from(f))),
+        Err(e) => Err(e)
+    };
+}
\ No newline at end of file
-- 
GitLab