diff --git a/src/controller/mod.rs b/src/controller/mod.rs index e5f99979a404c7f08724e6a56a061d8f65354f6f..dc66b8adf8b3bcc30cd783598062f4e293b22e9b 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -5,6 +5,6 @@ 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 c469e491da26cc341814da293fd7620908a238d8..ce8ae619148902dd0736c2cdb31eed5e37c05e65 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -14,4 +14,12 @@ pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { Ok(f) => Ok(Json::from(f)), 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) + }; } \ No newline at end of file