(wip) PWA Push Notifications

pages/api/notify.ts

@@ -1,6 +1,8 @@
 
 const webPush = require('web-push')
 
+import common from '../../src/helpers/common';
+
 //generate and store VAPID keys in .env.local if not already done
 if (!process.env.NEXT_PUBLIC_WEB_PUSH_PUBLIC_KEY || !process.env.WEB_PUSH_PRIVATE_KEY) {
     const { publicKey, privateKey } = webPush.generateVAPIDKeys()
@@ -28,28 +30,66 @@ const Notification = async (req, res) => {
         res.end(process.env.NEXT_PUBLIC_WEB_PUSH_PUBLIC_KEY)
         res.end()
     }
-    // on PUT store the subscription object in the database
+    if (req.method == 'PUT') {
+        // store the subscription object in the database
+        // publisher.pushSubscription = subscription
+        const prisma = common.getPrismaClient();
+        const { subscription, id } = req.body
+        const publisher = await prisma.publisher.update({
+            where: { id },
+            data: { pushSubscription: subscription }
+        })
+        console.log('Subscription for publisher', id, 'updated:', subscription)
+        res.statusCode = 200
+        res.end()
+    }
+    if (req.method == 'DELETE') {
+        // remove the subscription object from the database
+        // publisher.pushSubscription = null
+        const prisma = common.getPrismaClient();
+        const { id } = req.body
+        const publisher = await prisma.publisher.update({
+            where: { id },
+            data: { pushSubscription: null }
+        })
+        console.log('Subscription for publisher', id, 'deleted')
+        res.statusCode = 200
+        res.end()
+    }
+
 
     if (req.method == 'POST') {
-        const { subscription } = req.body
-
-        await webPush
-            .sendNotification(
-                subscription,
-                JSON.stringify({ title: 'Hello Web Push', message: 'Your web push notification is here!' })
-            )
-            .then(response => {
-                res.writeHead(response.statusCode, response.headers).end(response.body)
-            })
-            .catch(err => {
-                if ('statusCode' in err) {
-                    res.writeHead(err.statusCode, err.headers).end(err.body)
-                } else {
-                    console.error(err)
-                    res.statusCode = 500
-                    res.end()
-                }
-            })
+        const { subscription, id, broadcast, title = 'Hello Web Push', message = 'Your web push notification is here!' } = req.body
+        if (broadcast) {
+            await broadcastPush(title, message)
+            res.statusCode = 200
+            res.end()
+            return
+        }
+        else if (id) {
+            await sendPush(id, title, message)
+            res.statusCode = 200
+            res.end()
+            return
+        } else if (subscription) {
+            await webPush
+                .sendNotification(
+                    subscription,
+                    JSON.stringify({ title, message })
+                )
+                .then(response => {
+                    res.writeHead(response.statusCode, response.headers).end(response.body)
+                })
+                .catch(err => {
+                    if ('statusCode' in err) {
+                        res.writeHead(err.statusCode, err.headers).end(err.body)
+                    } else {
+                        console.error(err)
+                        res.statusCode = 500
+                        res.end()
+                    }
+                })
+        }
     } else {
         res.statusCode = 405
         res.end()
@@ -57,3 +97,48 @@ const Notification = async (req, res) => {
 }
 
 export default Notification
+
+//export pushNotification(userId or email) for use in other files
+export const sendPush = async (id, title, message) => {
+    const prisma = common.getPrismaClient();
+    const publisher = await prisma.publisher.findUnique({
+        where: { id }
+    })
+    if (!publisher.pushSubscription) {
+        console.log('No push subscription found for publisher', id)
+        return
+    }
+
+    await webPush
+        .sendNotification(
+            publisher.pushSubscription,
+            JSON.stringify({ title, message })
+        )
+        .then(response => {
+            console.log('Push notification sent to publisher', id)
+        })
+        .catch(err => {
+            console.error('Error sending push notification to publisher', id, ':', err)
+        })
+}
+//export breoadcastNotification for use in other files
+export const broadcastPush = async (title, message) => {
+    const prisma = common.getPrismaClient();
+    const publishers = await prisma.publisher.findMany({
+        where: { pushSubscription: { not: null } }
+    })
+
+    for (const publisher of publishers) {
+        await webPush
+            .sendNotification(
+                publisher.pushSubscription,
+                JSON.stringify({ title, message })
+            )
+            .then(response => {
+                console.log('Push notification sent to publisher', publisher.id)
+            })
+            .catch(err => {
+                console.error('Error sending push notification to publisher', publisher.id, ':', err)
+            })
+    }
+}