From 931bef0aeb057c295b1e23b8b1b754c1f3f605a7 Mon Sep 17 00:00:00 2001
From: Andre Medeiros <me@andremedeiros.info>
Date: Mon, 5 Jul 2021 21:39:09 -0400
Subject: [PATCH] iterate over devices

---
 cmd/apollo-worker/main.go | 37 ++++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 9 deletions(-)

diff --git a/cmd/apollo-worker/main.go b/cmd/apollo-worker/main.go
index d6b5490..02286c1 100644
--- a/cmd/apollo-worker/main.go
+++ b/cmd/apollo-worker/main.go
@@ -103,16 +103,35 @@ func accountWorker(id int, rc *reddit.Client, db *sql.DB, logger *log.Logger, qu
 				log.Fatal(err)
 			}
 
+			devices := []string{}
+			query = `
+				SELECT apns_token FROM devices
+				LEFT JOIN devices_accounts ON devices.id = devices_accounts.device_id
+				WHERE devices_accounts.account_id = $1`
+
+			rows, err := tx.Query(query, account.ID)
+			if err != nil {
+				logger.Fatal(err)
+			}
+			for rows.Next() {
+				var device string
+				rows.Scan(&device)
+				devices = append(devices, device)
+			}
+			rows.Close()
+
 			for _, msg := range msgs.MessageListing.Messages {
-				notification := &apns2.Notification{}
-				notification.DeviceToken = "9e1eb4c68d24a8f43eb92ed0a65f46aadbfbdbfe0a15ef4b5c34a9a4deb9ca49"
-				notification.Topic = "com.christianselig.Apollo"
-				notification.Payload = payload.NewPayload().AlertTitle(msg.Subject).AlertBody(msg.Body)
-				res, err := client.Push(notification)
-				if err != nil {
-					logger.Printf("Error sending push: %s", err)
-				} else {
-					logger.Printf("Push response: %v %v %v\n", res.StatusCode, res.ApnsID, res.Reason)
+				for _, device := range devices {
+					notification := &apns2.Notification{}
+					notification.DeviceToken = device
+					notification.Topic = "com.christianselig.Apollo"
+					notification.Payload = payload.NewPayload().AlertTitle(msg.Subject).AlertBody(msg.Body)
+					res, err := client.Push(notification)
+					if err != nil {
+						logger.Printf("Error sending push: %s", err)
+					} else {
+						logger.Printf("Push response: %v %v %v\n", res.StatusCode, res.ApnsID, res.Reason)
+					}
 				}
 			}