webpush.go

package notella

import (
	"encoding/json"
	"strings"
	"sync"

	"git.inpt.fr/churros/notella/db"
	"github.com/SherClockHolmes/webpush-go"
	ll "github.com/gwennlbh/label-logger-go"
)

type WebPushNotification struct {
	Title              string                  `json:"title"`
	Actions            []webpushAction         `json:"actions"`
	Badge              string                  `json:"badge"`
	Icon               string                  `json:"icon"`
	Image              string                  `json:"image"`
	Body               string                  `json:"body"`
	Renotify           bool                    `json:"renotify"`
	RequireInteraction bool                    `json:"requireInteraction"`
	Silent             bool                    `json:"silent"`
	Tag                string                  `json:"tag"`
	Timestamp          int64                   `json:"timestamp"`
	Vibrate            []int                   `json:"vibrate"`
	Data               webpushNotificationData `json:"data"`
}

type webpushAction struct {
	Action string `json:"action"`
	Label  string `json:"label"`
	Icon   string `json:"icon"`
}

type webpushNotificationData struct {
	Group            string                 `json:"group"`
	Channel          db.NotificationChannel `json:"channel"`
	SubscriptionName string                 `json:"subscriptionName"`
	Goto             string                 `json:"goto"`
}

func (msg Message) WebPush(groupId string) WebPushNotification {
	actions := make([]webpushAction, len(msg.Actions))
	for i, action := range msg.Actions {
		actions[i] = webpushAction{
			Action: action.Action,
			Label:  action.Label,
			Icon:   "",
		}
	}

	return WebPushNotification{
		Title:   msg.Title,
		Actions: actions,
		Badge:   "",
		Icon:    "",
		Image:   msg.Image,
		Body:    msg.Body,
		Data: webpushNotificationData{
			Group:            groupId,
			Channel:          msg.Channel(),
			SubscriptionName: "",
			Goto:             msg.Action,
		},
	}
}

func (msg Message) SendWebPush(groupId string, subs []Subscription) error {

	jsoned, err := json.Marshal(msg.WebPush(groupId))
	if err != nil {
		ll.ErrorDisplay("could not marshal notification to json", err)
	}

	var wg sync.WaitGroup
	wg.Add(len(subs))
	for _, sub := range subs {
		go func(wg *sync.WaitGroup, sub Subscription) {
			if config.DryRunMode {
				exempt := false
				for _, username := range config.DryRunExceptions {
					if username == sub.Owner.Uid {
						exempt = true
					}
				}
				if !exempt {
					ll.Warn("dry run mode enabled, not sending webpush notification to %s", sub.Owner.Uid)
					wg.Done()
					return
				}
			}
			resp, err := webpush.SendNotification(jsoned, &sub.Webpush, &webpush.Options{
				TTL:             30,
				Subscriber:      config.ContactEmail,
				VAPIDPublicKey:  config.VapidPublicKey,
				VAPIDPrivateKey: config.VapidPrivateKey,
			})
			wg.Done()

			if err != nil {
				ll.ErrorDisplay("could not send notification to %s", err, sub.Owner.Uid)
			} else if resp.StatusCode == 410 {
				ll.Log("Deleting", "yellow", "invalid webpush subscription %s", sub.Webpush.Endpoint)
				sub.Destroy()
			} else if resp.StatusCode >= 400 {
				ll.Error("could not send notification to %s: HTTP %d", sub.Owner.Uid, resp.StatusCode)
			}

		}(&wg, sub)
	}
	wg.Wait()

	return nil
}

func (sub Subscription) IsWebpush() bool {
	// Native subscriptions don't use the https: protocol
	return strings.HasPrefix(sub.Webpush.Endpoint, "https://")
}