From 4b1b34290da88d621a6fe93eddf0e29525dd9d90 Mon Sep 17 00:00:00 2001
From: rafliesa <rflisaa@gmail.com>
Date: Fri, 28 Mar 2025 09:49:12 +0700
Subject: [PATCH] Implement unsubscribe function in Notification controller.

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

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index ef4a9ce..346ff5d 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -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![notification::subscribe])
+            .mount("notification", routes![notification::subscribe, notification::unsubscribe])
     });
 }
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index e644857..bfb6959 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -1,5 +1,5 @@
 use rocket::response::status::Created;
-use rocket::serde::json::Json;
+use rocket::serde::json::{self, Json};
 
 use bambangshop::Result;
 use crate::model::subscriber::{self, Subscriber};
@@ -12,3 +12,11 @@ pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Cre
         Err(e) => Err(e)
     }
 }
+
+#[post("/unsubscribe/<product_type>?<url>")]
+pub fn unsubscribe(product_type: &str, url: &str) -> Result<Json<Subscriber>> {
+    return match NotificationService::unsubscribe(product_type, url) {
+        Ok(f) => Ok(Json::from(f)),
+        Err(e) => Err(e)
+    }
+}
-- 
GitLab