diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 782fc656ec155f1cfd6685a86b27e0a83025f5e7..e42fb6548f17ec8366076af02d8641a44cf561aa 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -4,6 +4,6 @@ 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 1bf721ad89fc16a54c2e0a525e1ff2449ceff92a..93ca2c63e1657bb8370f6814cecd37997799e427 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(f) => Ok(Json::from(f)), Err(e) => Err(e) } +} + +#[post("/receive", format = "application/json", 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