From a42ddbeeb3d628829962f7a4d066fe1fb12e960a Mon Sep 17 00:00:00 2001 From: Andre Medeiros Date: Sat, 7 May 2022 12:53:42 -0400 Subject: [PATCH] add notification test endpoints --- internal/api/api.go | 6 ++ internal/api/notifications.go | 190 ++++++++++++++++++++++++++++++++++ 2 files changed, 196 insertions(+) create mode 100644 internal/api/notifications.go diff --git a/internal/api/api.go b/internal/api/api.go index 44b1e26..d073211 100644 --- a/internal/api/api.go +++ b/internal/api/api.go @@ -96,6 +96,12 @@ func (a *api) Routes() *mux.Router { r.HandleFunc("/v1/device", a.upsertDeviceHandler).Methods("POST") r.HandleFunc("/v1/device/{apns}", a.deleteDeviceHandler).Methods("DELETE") r.HandleFunc("/v1/device/{apns}/test", a.testDeviceHandler).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/comment_reply", generateNotificationTester(a, commentReply)).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/post_reply", generateNotificationTester(a, postReply)).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/private_message", generateNotificationTester(a, privateMessage)).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/subreddit_watcher", generateNotificationTester(a, subredditWatcher)).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/trending_post", generateNotificationTester(a, trendingPost)).Methods("POST") + r.HandleFunc("/v1/device/{apns}/test/username_mention", generateNotificationTester(a, usernameMention)).Methods("POST") r.HandleFunc("/v1/device/{apns}/account", a.upsertAccountHandler).Methods("POST") r.HandleFunc("/v1/device/{apns}/accounts", a.upsertAccountsHandler).Methods("POST") diff --git a/internal/api/notifications.go b/internal/api/notifications.go new file mode 100644 index 0000000..7ce243d --- /dev/null +++ b/internal/api/notifications.go @@ -0,0 +1,190 @@ +package api + +import ( + "context" + "fmt" + "net/http" + + "github.com/gorilla/mux" + "github.com/sideshow/apns2" + "github.com/sideshow/apns2/payload" + "github.com/sirupsen/logrus" +) + +const ( + commentReplyNotificationTitleFormat = "%s in %s" + postReplyNotificationTitleFormat = "%s to %s" + privateMessageNotificationTitleFormat = "Message from %s" + subredditNotificationBodyFormat = "r/%s: \u201c%s\u201d" + subredditNotificationTitleFormat = "📣 \u201c%s\u201d Watcher" + trendingNotificationTitleFormat = "🔥 r/%s Trending" + usernameMentionNotificationTitleFormat = "Mention in \u201c%s\u201d" +) + +type notificationGenerator func(*payload.Payload) + +func generateNotificationTester(a *api, fun notificationGenerator) func(w http.ResponseWriter, r *http.Request) { + return func(w http.ResponseWriter, r *http.Request) { + vars := mux.Vars(r) + + ctx := context.Background() + tok := vars["apns"] + + d, err := a.deviceRepo.GetByAPNSToken(ctx, tok) + if err != nil { + a.logger.WithFields(logrus.Fields{ + "err": err, + }).Info("failed fetching device from database") + a.errorResponse(w, r, 500, err.Error()) + return + } + + p := payload.NewPayload() + fun(p) + + notification := &apns2.Notification{} + notification.Topic = "com.christianselig.Apollo" + notification.DeviceToken = d.APNSToken + notification.Payload = p + + client := apns2.NewTokenClient(a.apns) + if !d.Sandbox { + client = client.Production() + } + + if _, err := client.Push(notification); err != nil { + a.logger.WithFields(logrus.Fields{ + "err": err, + }).Info("failed to send test notification") + a.errorResponse(w, r, 500, err.Error()) + return + } + w.WriteHeader(http.StatusOK) + } +} + +func privateMessage(pl *payload.Payload) { + title := fmt.Sprintf(privateMessageNotificationTitleFormat, "welcomebot") + + pl. + AlertBody("**Welcome to r/GriefSupport!**\n\nWe're glad you found us, but sad you needed to. We're here to support you during whatever difficulties you're going through."). + AlertSummaryArg("welcomebot"). + Custom("account_id", "1ia22"). + Custom("author", "welcomebot"). + Custom("destination_author", "changelog"). + Custom("parent_id", ""). + Custom("post_title", ""). + Custom("subreddit", ""). + MutableContent(). + Sound("traloop.wav"). + AlertTitle(title). + AlertSubtitle("Welcome to r/GriefSupport!"). + Category("inbox-private-message"). + Custom("comment_id", "1d2oouy"). + Custom("type", "private-message") +} + +func commentReply(pl *payload.Payload) { + title := fmt.Sprintf(commentReplyNotificationTitleFormat, "Equinox_Shift", "Protests set to disrupt Ottawa's downtown for 3rd straight weekend") + + pl. + AlertBody("They don't even go here."). + Custom("account_id", "1ia22"). + Custom("author", "Equinox_Shift"). + Custom("destination_author", "changelog"). + Custom("parent_id", "t1_hwonb97"). + Custom("post_title", "Protests set to disrupt Ottawa's downtown for 3rd straight weekend"). + Custom("subreddit", "ottawa"). + AlertTitle(title). + MutableContent(). + Sound("traloop.wav"). + Category("inbox-comment-reply"). + Custom("comment_id", "hwp66zg"). + Custom("post_id", "sqqk29"). + Custom("subject", "comment"). + Custom("subreddit", "ottawa"). + Custom("type", "comment"). + ThreadID("comment") +} +func postReply(pl *payload.Payload) { + title := fmt.Sprintf(postReplyNotificationTitleFormat, "Ryfter", "Quest 2 use during chemo") + + pl. + AlertBody("As others have said, [Real Fishing VR](https://www.oculus.com/experiences/quest/2582932495064035). Especially if he likes to fish. My dad and mom were blown away by it."). + Custom("account_id", "1ia22"). + Custom("author", "Ryfter"). + Custom("destination_author", "changelog"). + Custom("parent_id", "t3_t0qn4z"). + Custom("post_title", "Quest 2 use during chemo"). + Custom("subreddit", "OculusQuest2"). + AlertTitle(title). + MutableContent(). + Sound("traloop.wav"). + Category("inbox-comment-reply"). + Custom("comment_id", "hyg01ip"). + Custom("post_id", "t0qn4z"). + Custom("subject", "comment"). + Custom("subreddit", "OculusQuest2"). + Custom("type", "post"). + ThreadID("comment") +} +func usernameMention(pl *payload.Payload) { + title := fmt.Sprintf(usernameMentionNotificationTitleFormat, "testimg") + + pl. + AlertBody("yo u/changelog what's good"). + Custom("account_id", "1ia22"). + Custom("author", "iamthatis"). + Custom("destination_author", "changelog"). + Custom("parent_id", "t3_u02338"). + Custom("post_title", "testimg"). + Custom("subreddit", "calicosummer"). + MutableContent(). + Sound("traloop.wav"). + AlertTitle(title). + Category("inbox-username-mention-no-context"). + Custom("comment_id", "i6xobpa"). + Custom("post_id", "u02338"). + Custom("subject", "comment"). + Custom("subreddit", "calicosummer"). + Custom("type", "username") + +} +func subredditWatcher(pl *payload.Payload) { + title := fmt.Sprintf(subredditNotificationTitleFormat, "bug pics") + body := fmt.Sprintf(subredditNotificationBodyFormat, "pics", "A Goliath Stick Insect. Aware of my presence she let me get close enough for a photo. (OC)") + + pl. + AlertTitle(title). + AlertBody(body). + AlertSummaryArg("pics"). + Category("subreddit-watcher"). + Custom("post_title", "A Goliath Stick Insect. Aware of my presence she let me get close enough for a photo. (OC)"). + Custom("post_id", "ufzaml"). + Custom("subreddit", "pics"). + Custom("author", "befarked247"). + Custom("post_age", 1651409659.0). + ThreadID("subreddit-watcher"). + MutableContent(). + Sound("traloop.wav"). + Custom("thumbnail", "https://a.thumbs.redditmedia.com/Lr4b-YHLTNu1LFuyUY1Zic8kHy3ojX06gLcZOuqxrr0.jpg") +} + +func trendingPost(pl *payload.Payload) { + title := fmt.Sprintf(trendingNotificationTitleFormat, "pics") + + pl. + AlertTitle(title). + AlertBody("A Goliath Stick Insect. Aware of my presence she let me get close enough for a photo. (OC)"). + AlertSummaryArg("pics"). + Category("trending-post"). + Custom("post_title", "A Goliath Stick Insect. Aware of my presence she let me get close enough for a photo. (OC)"). + Custom("post_id", "ufzaml"). + Custom("subreddit", "pics"). + Custom("author", "befarked247"). + Custom("post_age", 1651409659.0). + ThreadID("trending-post"). + MutableContent(). + Sound("traloop.wav"). + Custom("thumbnail", "https://a.thumbs.redditmedia.com/Lr4b-YHLTNu1LFuyUY1Zic8kHy3ojX06gLcZOuqxrr0.jpg") +}