diff --git a/src/controller/mod.rs b/src/controller/mod.rs index dfcb94393746b92e525671cbd8ed2372f73211ae..25cc704e24f34dc37b782b17302a90e0e69c4872 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -3,7 +3,8 @@ pub mod notification; pub fn route_stage() -> AdHoc { return AdHoc::on_ignite("Initializing controller routes...", |rocket| async { rocket - .mount("/", routes![notification::subscribe, notification::unsubscribe]) + .mount("/", routes![notification::subscribe, notification::unsubscribe, + notification::receive]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index af4edbb38d066443b34323057780bc24d6623a10..5a5cf588b97d315797c4ccffdf1db95f1f3358fe 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -19,4 +19,12 @@ pub fn unsubscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { Ok(r) => Ok(Json::from(r)), Err(e) => Err(e) }; +} + +#[post("/receive", data = "<notification>")] +pub fn receive(notification: Json<Notification>) -> Result<Json<Notification>> { + return match NotificationService::receive_notification(notification.into_inner()) { + Ok(f) => Ok(Json::from(f)), + Err(e) => Err(e) + }; } \ No newline at end of file