Send notifications by Expo API

4 files changed, 70 insertions, 2 deletions

diff --git a/src/config.rs b/src/config.rs
index 7eefff8..a703010 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -16,6 +16,9 @@ pub struct Configuration {
 
     /// Host URL
     pub allowed_host: String,
+
+    /// Token used by Expo API to send a notification
+    pub expo_access_token: String,
 }
 
 impl Configuration {
diff --git a/src/expo.rs b/src/expo.rs
new file mode 100644
index 0000000..1a6a4e5
--- /dev/null
+++ b/src/expo.rs
@@ -0,0 +1,31 @@
+use expo_push_notification_client::{Expo, ExpoClientOptions, ExpoPushMessage, ValidationError};
+
+/// Connection to an Expo client
+static mut EXPO_CONNECTION: Option<Expo> = None;
+
+/// Setup a new Expo API
+pub fn setup(access_token: String) {
+    unsafe {
+        EXPO_CONNECTION = Some(Expo::new(ExpoClientOptions {
+            access_token: Some(access_token),
+        }))
+    }
+}
+
+/// Send notifications using Expo
+pub async fn send(tokens: Vec<String>, body: String, title: String) -> Result<(), ValidationError> {
+    let expo = unsafe {
+        EXPO_CONNECTION
+            .clone()
+            .expect("You need to call `setup()` first")
+    };
+
+    let expo_push_message = ExpoPushMessage::builder(tokens)
+        .body(body)
+        .title(title)
+        .build()?;
+
+    let _ = expo.send_push_notifications(expo_push_message).await;
+
+    Ok(())
+}
diff --git a/src/graphql/mutation.rs b/src/graphql/mutation.rs
index 4387b77..dfffad8 100644
--- a/src/graphql/mutation.rs
+++ b/src/graphql/mutation.rs
@@ -1,4 +1,5 @@
 use crate::{
+    expo,
     graphql::types::{
         alert,
         jwt::{self, Authentication},
@@ -282,8 +283,8 @@ impl Mutation {
                     .collect();
 
                 let mut notification_ids = vec![];
-                for id in position_ids {
-                    let notification = notification::Notification::new(client, alert.id, id)
+                for id in &position_ids {
+                    let notification = notification::Notification::new(client, alert.id, *id)
                         .await
                         .unwrap();
                     notification_ids.push(notification);
@@ -298,6 +299,37 @@ impl Mutation {
                     .await
                     .unwrap();
 
+                let placeholders: Vec<String> = (1..=position_ids.len())
+                    .map(|i| format!("${}", i))
+                    .collect();
+                let query = format!(
+                    "SELECT u.notification_token FROM positions p JOIN users u ON u.id = p.user_id
+                    WHERE p.id IN ({}) AND notification_token IS NOT NULL",
+                    placeholders.join(", ")
+                );
+
+                let tokens: Vec<String> = client
+                    .query(
+                        &query,
+                        &position_ids
+                            .iter()
+                            .map(|id| id as &(dyn tokio_postgres::types::ToSql + Sync))
+                            .collect::<Vec<&(dyn tokio_postgres::types::ToSql + Sync)>>(),
+                    )
+                    .await
+                    .unwrap()
+                    .iter()
+                    .map(|row| format!("ExponentPushToken[{}]", row.get::<usize, String>(0)))
+                    .collect();
+
+                expo::send(
+                    tokens,
+                    "New Alert!".to_string(),
+                    "Keep an eye open".to_string(),
+                )
+                .await
+                .unwrap();
+
                 Ok(alert)
             }
         }
diff --git a/src/main.rs b/src/main.rs
index c288001..6988711 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,7 @@
 mod config;
 mod db;
 mod errors;
+mod expo;
 mod graphql;
 mod logger;
 mod routes;
@@ -27,6 +28,7 @@ use tracing::Span;
 /// Create the app: setup everything and returns a `Router`
 async fn create_app() -> Router {
     logger::setup();
+    expo::setup(CONFIG.expo_access_token.clone());
     let dbclient = db::setup().await.unwrap();
 
     let state = state::AppState {