🎉 Initial Commit

pkg/internal/services/connections.go

@@ -0,0 +1,95 @@
+package services
+
+import (
+	"math/rand"
+	"sync"
+
+	"git.solsynth.dev/hypernet/nexus/pkg/internal/models"
+	"github.com/gofiber/contrib/websocket"
+)
+
+var (
+	wsMutex sync.Mutex
+	wsConn  = make(map[uint]map[uint64]*websocket.Conn)
+)
+
+func ClientRegister(user models.Account, conn *websocket.Conn) uint64 {
+	wsMutex.Lock()
+	if wsConn[user.ID] == nil {
+		wsConn[user.ID] = make(map[uint64]*websocket.Conn)
+	}
+	clientId := rand.Uint64()
+	wsConn[user.ID][clientId] = conn
+	wsMutex.Unlock()
+
+	return clientId
+}
+
+func ClientUnregister(user models.Account, id uint64) {
+	wsMutex.Lock()
+	if wsConn[user.ID] == nil {
+		wsConn[user.ID] = make(map[uint64]*websocket.Conn)
+	}
+	delete(wsConn[user.ID], id)
+	wsMutex.Unlock()
+}
+
+func ClientCount(uid uint) int {
+	return len(wsConn[uid])
+}
+
+func WebsocketPush(uid uint, body []byte) (count int, success int, errs []error) {
+	for _, conn := range wsConn[uid] {
+		if err := conn.WriteMessage(1, body); err != nil {
+			errs = append(errs, err)
+		} else {
+			success++
+		}
+		count++
+	}
+	return
+}
+
+func WebsocketPushDirect(clientId uint64, body []byte) (count int, success int, errs []error) {
+	for _, m := range wsConn {
+		if conn, ok := m[clientId]; ok {
+			if err := conn.WriteMessage(1, body); err != nil {
+				errs = append(errs, err)
+			} else {
+				success++
+			}
+			count++
+		}
+	}
+	return
+}
+
+func WebsocketPushBatch(uidList []uint, body []byte) (count int, success int, errs []error) {
+	for _, uid := range uidList {
+		for _, conn := range wsConn[uid] {
+			if err := conn.WriteMessage(1, body); err != nil {
+				errs = append(errs, err)
+			} else {
+				success++
+			}
+			count++
+		}
+	}
+	return
+}
+
+func WebsocketPushBatchDirect(clientIdList []uint64, body []byte) (count int, success int, errs []error) {
+	for _, clientId := range clientIdList {
+		for _, m := range wsConn {
+			if conn, ok := m[clientId]; ok {
+				if err := conn.WriteMessage(1, body); err != nil {
+					errs = append(errs, err)
+				} else {
+					success++
+				}
+				count++
+			}
+		}
+	}
+	return
+}

pkg/internal/services/pusher_conn.go

@@ -0,0 +1,43 @@
+package services
+
+import (
+	"context"
+	firebase "firebase.google.com/go"
+	"github.com/sideshow/apns2"
+	"github.com/sideshow/apns2/token"
+	"github.com/spf13/viper"
+	"google.golang.org/api/option"
+)
+
+// ExtFire is a Firebase App client
+var ExtFire *firebase.App
+
+func SetupFirebase() error {
+	opt := option.WithCredentialsFile(viper.GetString("firebase_credentials"))
+	app, err := firebase.NewApp(context.Background(), nil, opt)
+	if err != nil {
+		return err
+	} else {
+		ExtFire = app
+	}
+
+	return nil
+}
+
+// ExtAPNS is an Apple Push Notification Services client
+var ExtAPNS *apns2.Client
+
+func SetupAPNS() error {
+	authKey, err := token.AuthKeyFromFile(viper.GetString("apns_credentials"))
+	if err != nil {
+		return err
+	}
+
+	ExtAPNS = apns2.NewTokenClient(&token.Token{
+		AuthKey: authKey,
+		KeyID:   viper.GetString("apns_credentials_key"),
+		TeamID:  viper.GetString("apns_credentials_team"),
+	}).Production()
+
+	return nil
+}