Notifications

firebase/functions/index.js

@@ -1,14 +1,107 @@
 const {onRequest} = require("firebase-functions/v2/https");
 const {getAuth} = require("firebase-admin/auth");
 const {getFirestore} = require("firebase-admin/firestore");
-const admin = require("firebase-admin");
 const logger = require("firebase-functions/logger");
+const functions = require('firebase-functions');
+const admin = require('firebase-admin');
+const {Expo} = require('expo-server-sdk');
 
-try {
-    admin.initializeApp();
-} catch (error) {
-    console.error(error)
-}
+admin.initializeApp();
+const db = admin.firestore();
+
+let expo = new Expo({accessToken: process.env.EXPO_ACCESS_TOKEN});
+
+// Firestore trigger that listens for new events in the 'Events' collection
+exports.sendNotificationOnEventCreation = functions.firestore
+    .document('Events/{eventId}')
+    .onCreate(async (snapshot, context) => {
+        const eventData = snapshot.data();
+
+        const pushTokens = await getPushTokensForEvent(eventData);
+
+        if (!pushTokens.length) {
+            console.log('No push tokens available for the event.');
+            return;
+        }
+
+        let messages = [];
+        for (let pushToken of pushTokens) {
+            if (!Expo.isExpoPushToken(pushToken)) {
+                console.error(`Push token ${pushToken} is not a valid Expo push token`);
+                continue;
+            }
+
+            messages.push({
+                to: pushToken,
+                sound: 'default',
+                title: 'New Event Added!',
+                body: `An event "${eventData.title}" has been added. Check it out!`,
+                data: {eventId: context.params.eventId},
+            });
+        }
+
+        let chunks = expo.chunkPushNotifications(messages);
+        let tickets = [];
+        for (let chunk of chunks) {
+            try {
+                let ticketChunk = await expo.sendPushNotificationsAsync(chunk);
+
+                tickets.push(...ticketChunk);
+
+                for (let ticket of ticketChunk) {
+                    if (ticket.status === 'ok') {
+                        console.log('Notification successfully sent:', ticket.id);
+                    } else if (ticket.status === 'error') {
+                        console.error(`Notification error: ${ticket.message}`);
+
+                        if (ticket.details && ticket.details.error) {
+                            console.error('Error details:', ticket.details.error);
+                            if (ticket.details.error === 'DeviceNotRegistered') {
+                                console.log(`Removing invalid push token: ${ticket.to}`);
+                                await removeInvalidPushToken(ticket.to);
+                            }
+                        }
+                    }
+                }
+
+            } catch (error) {
+                console.error('Error sending notification:', error);
+            }
+        }
+
+        let receiptIds = [];
+        for (let ticket of tickets) {
+            if (ticket.id) {
+                receiptIds.push(ticket.id);
+            }
+        }
+
+        let receiptIdChunks = expo.chunkPushNotificationReceiptIds(receiptIds);
+        for (let chunk of receiptIdChunks) {
+            try {
+                let receipts = await expo.getPushNotificationReceiptsAsync(chunk);
+                console.log('Receipts:', receipts);
+
+                for (let receiptId in receipts) {
+                    let { status, message, details } = receipts[receiptId];
+
+                    if (status === 'ok') {
+                        console.log(`Notification with receipt ID ${receiptId} was delivered successfully`);
+                    } else if (status === 'error') {
+                        console.error(`Notification error: ${message}`);
+                        if (details && details.error) {
+                            console.error(`Error details: ${details.error}`);
+                        }
+                    }
+                }
+            } catch (error) {
+                console.error('Error retrieving receipts:', error);
+            }
+        }
+
+
+        return null;
+    });
 
 exports.createSubUser = onRequest(async (request, response) => {
     const authHeader = request.get('Authorization');
@@ -97,3 +190,20 @@ exports.generateCustomToken = onRequest(async (request, response) => {
         response.status(500).json({error: "Failed to generate custom token"});
     }
 });
+
+async function getPushTokensForEvent() {
+    const usersRef = db.collection('Profiles');
+    const snapshot = await usersRef.get();
+    let pushTokens = [];
+
+    snapshot.forEach(doc => {
+        const data = doc.data();
+        if (data.pushToken) {
+            pushTokens.push(data.pushToken);
+        }
+    });
+
+    console.log('Push Tokens:', pushTokens);
+    return pushTokens;
+}
+