🎉 Initial Commit

pkg/internal/provider/apns.go

@@ -0,0 +1,48 @@
+package provider
+
+import (
+	"git.solsynth.dev/hypernet/pusher/pkg/pushkit"
+	"github.com/sideshow/apns2"
+	payload2 "github.com/sideshow/apns2/payload"
+	"github.com/spf13/viper"
+)
+
+type AppleNotifyProvider struct {
+	topic string
+	conn  *apns2.Client
+}
+
+func (v *AppleNotifyProvider) Push(in pushkit.Notification, tk string) error {
+	data := payload2.
+		NewPayload().
+		AlertTitle(in.Title).
+		AlertBody(in.Body).
+		Category(in.Topic).
+		Custom("metadata", in.Metadata).
+		Sound("default").
+		MutableContent()
+	if len(in.Subtitle) > 0 {
+		data = data.AlertSubtitle(in.Subtitle)
+	}
+	if avatar, ok := in.Metadata["avatar"]; ok {
+		data = data.Custom("avatar", avatar)
+	}
+	if picture, ok := in.Metadata["picture"]; ok {
+		data = data.Custom("picture", picture)
+	}
+	rawData, err := data.MarshalJSON()
+	if err != nil {
+		return err
+	}
+	payload := &apns2.Notification{
+		DeviceToken: tk,
+		Topic:       viper.GetString(v.topic),
+		Payload:     rawData,
+	}
+	_, err := v.conn.Push(payload)
+	return err
+}
+
+func (v *AppleNotifyProvider) GetName() string {
+	return "apns"
+}

pkg/internal/provider/caller.go

@@ -0,0 +1,67 @@
+package provider
+
+import (
+	"fmt"
+	"git.solsynth.dev/hypernet/pusher/pkg/pushkit"
+	"github.com/rs/zerolog/log"
+	"sync"
+	"time"
+)
+
+var notifyProviders = make(map[string]NotificationProvider)
+
+func AddProvider(in NotificationProvider) {
+	notifyProviders[in.GetName()] = in
+}
+
+func PushNotification(in pushkit.NotificationPushRequest) error {
+	prov, ok := notifyProviders[in.Provider]
+	if !ok {
+		return fmt.Errorf("provider not found")
+	}
+	start := time.Now()
+	err := prov.Push(in.Notification, in.Token)
+	if err != nil {
+		log.Warn().Err(err).
+			Str("tk", in.Token).
+			Str("provider", prov.GetName()).
+			Dur("elapsed", time.Since(start)).
+			Msg("Push notification failed once")
+	} else {
+		log.Debug().
+			Str("tk", in.Token).
+			Str("provider", prov.GetName()).
+			Dur("elapsed", time.Since(start)).
+			Msg("Pushed one notification")
+	}
+	return err
+}
+
+func PushNotificationBatch(in pushkit.NotificationPushBatchRequest) {
+	var wg sync.WaitGroup
+	for idx, key := range in.Providers {
+		prov, ok := notifyProviders[key]
+		if !ok {
+			continue
+		}
+		go func() {
+			wg.Add(1)
+			defer wg.Done()
+			start := time.Now()
+			err := prov.Push(in.Notification, in.Tokens[idx])
+			if err != nil {
+				log.Warn().Err(err).
+					Str("tk", in.Tokens[idx]).
+					Str("provider", prov.GetName()).
+					Dur("elapsed", time.Since(start)).
+					Msg("Push notification failed once")
+			} else {
+				log.Debug().
+					Str("tk", in.Tokens[idx]).
+					Str("provider", prov.GetName()).
+					Dur("elapsed", time.Since(start)).
+					Msg("Pushed one notification")
+			}
+		}()
+	}
+}

pkg/internal/provider/firebase.go

@@ -0,0 +1,40 @@
+package provider
+
+import (
+	"context"
+	firebase "firebase.google.com/go"
+	"firebase.google.com/go/messaging"
+	"fmt"
+	"git.solsynth.dev/hypernet/pusher/pkg/pushkit"
+)
+
+type FirebaseNotifyProvider struct {
+	conn *firebase.App
+}
+
+func (v *FirebaseNotifyProvider) Push(in pushkit.Notification, tk string) error {
+	ctx := context.Background()
+	client, err := v.conn.Messaging(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to create firebase client")
+	}
+
+	var subtitle string
+	if len(in.Subtitle) > 0 {
+		subtitle = "\n" + in.Subtitle
+	}
+	message := &messaging.Message{
+		Notification: &messaging.Notification{
+			Title: in.Title,
+			Body:  subtitle + in.Body,
+		},
+		Token: tk,
+	}
+
+	_, err = client.Send(ctx, message)
+	return err
+}
+
+func (v *FirebaseNotifyProvider) GetName() string {
+	return "fcm"
+}

pkg/internal/provider/initializer.go

@@ -0,0 +1,36 @@
+package provider
+
+import (
+	"context"
+	firebase "firebase.google.com/go"
+	"github.com/sideshow/apns2"
+	"github.com/sideshow/apns2/token"
+	"google.golang.org/api/option"
+)
+
+func InitFCM(in string) error {
+	opt := option.WithCredentialsFile(in)
+	app, err := firebase.NewApp(context.Background(), nil, opt)
+	if err != nil {
+		return err
+	} else {
+		AddProvider(&FirebaseNotifyProvider{app})
+	}
+
+	return nil
+}
+
+func InitAPN(in, keyId, teamId, topic string) error {
+	authKey, err := token.AuthKeyFromFile(in)
+	if err != nil {
+		return err
+	} else {
+		AddProvider(&AppleNotifyProvider{topic, apns2.NewTokenClient(&token.Token{
+			AuthKey: authKey,
+			KeyID:   keyId,
+			TeamID:  teamId,
+		}).Production()})
+	}
+
+	return nil
+}

pkg/internal/provider/interface.go

@@ -0,0 +1,11 @@
+package provider
+
+import (
+	"git.solsynth.dev/hypernet/pusher/pkg/pushkit"
+)
+
+type NotificationProvider interface {
+	Push(in pushkit.Notification, tk string) error
+
+	GetName() string
+}