diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 1567487f77b9d46c16d18eef8ebf1b8d571c60d8..fe4a544b65cec3c65bedd9d1fa666dc986aff6f2 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 e1420a2bc52a3a38047f9665ac750691e259cf6f..851bd7ffc90b7f1ca7c5c6d13fc8fc0cb12f3e9c 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -1,3 +1,4 @@ +use reqwest::Url; use rocket::response::status::Created; use rocket::serde::json::Json; @@ -14,4 +15,12 @@ pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Cre Ok(f) => Ok(Created::new("/").body(Json::from(f))), 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) + }; } \ No newline at end of file