From bc995fc11890aec107442914692be209d6e65eda Mon Sep 17 00:00:00 2001
From: vissutagunawan <vglim3653@gmail.com>
Date: Thu, 27 Mar 2025 09:20:51 +0700
Subject: [PATCH] Implement subscribe function in Notification controller.

---
 src/controller/mod.rs          |  2 +-
 src/controller/notification.rs | 13 ++++++++++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index 38cfed4..1ab4717 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -15,6 +15,6 @@ pub fn route_stage() -> AdHoc {
                 product::delete
             ],
         )
-        .mount("/notification", routes![])
+        .mount("/notification", routes![notification::subscribe])
     });
 }
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index b411b3c..a59531d 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -3,4 +3,15 @@ use rocket::serde::json::Json;
 
 use crate::model::subscriber::Subscriber;
 use crate::service::notification::NotificationService;
-use bambangshop::Result;
\ No newline at end of file
+use bambangshop::Result;
+
+#[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("/".to_string()).body(Json::from(f))),
+         Err(e) => Err(e),
+     };
+ }
\ No newline at end of file
-- 
GitLab