diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 80195c40910ace8157102b9c97495f2e5d822632..b1d8df3d48fedcaf5843728fe18632a21cb8d6db 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -5,6 +5,7 @@ use rocket::fairing::AdHoc; 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 f65f0f97f53a4a28e0038e40f754f53a5db3d2ba..bc64944ba1e84649c90c8fc0c43c45a0fa1a1642 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -21,3 +21,11 @@ pub fn unsubscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { }; } +#[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) + }; +} +