From 1adbbc5063cdb992e845144a4c95c9c9726d1b96 Mon Sep 17 00:00:00 2001
From: lantry-glitch <allan.kwek.18@gmail.com>
Date: Fri, 28 Mar 2025 15:53:44 +0700
Subject: [PATCH] Implement unsubscribe function in Notification controller.

---
 src/controller/mod.rs          | 2 +-
 src/controller/notification.rs | 9 +++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index 038c473..2e4d6d9 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -4,7 +4,7 @@ use rocket::fairing::AdHoc;
 pub fn route_stage() -> AdHoc {
     return AdHoc::on_ignite("Initializing controller routes...", |rocket| async {
         rocket
-            .mount("/", routes![notification::subscribe])
+            .mount("/", routes![notification::subscribe, notification::unsubscribe])
 
     });
 }
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index 24a3dc6..37c67cc 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -13,3 +13,12 @@ pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> {
         Err(e) => Err(e)
     };
 }
+
+#[get("/unsubscribe/<product_type>")]
+pub fn unsubscribe(product_type: &str) -> Result<Json<SubscriberRequest>> {
+    return match NotificationService::unsubscribe(product_type) {
+        Ok(f) => Ok(Json::from(f)),
+        Err(e) => Err(e)
+    };
+}
+
-- 
GitLab