mirror of
https://github.com/christianselig/apollo-backend
synced 2024-11-21 19:37:41 +00:00
add notification test endpoints
This commit is contained in:
Andre Medeiros
parent
f9b9c595cf
commit
a42ddbeeb3
2 changed files with 196 additions and 0 deletions
|
|
@ -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")
|
||||
|
|
|
|||
190
internal/api/notifications.go
Normal file
190
internal/api/notifications.go
Normal file
|
|
@ -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")
|
||||
}
|
||||
Loading…
Reference in a new issue