diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 038c47365e27cad53780c042d1e27d7c95b0191c..2e4d6d98e158f78ae585e061ce929692a14058e7 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 24a3dc6bf3f79cc34d9bce23b852edb69a8c2d0e..37c67cc8d1968b6bab44462387b0c60cb2c3619a 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) + }; +} +