Firebase.go

package wfb

import (
	"context"
	"fmt"
	"os"
	"path"
	"sync"

	firebase "firebase.google.com/go"
	// "firebase.google.com/go/auth"
	"firebase.google.com/go/messaging"
	"google.golang.org/api/option"

	"github.com/wasabee-project/Wasabee-Server/config"
	"github.com/wasabee-project/Wasabee-Server/log"
	wm "github.com/wasabee-project/Wasabee-Server/messaging"
)

var msg *messaging.Client
var fbctx context.Context
var multicastFantoutMutex sync.Mutex

// Start is the main startup function for the Firebase subsystem
func Start(ctx context.Context) {
	c := config.Get()
	keypath := path.Join(c.Certs, c.FirebaseKey)

	if _, err := os.Stat(keypath); err != nil {
		log.Debugw("firebase key does not exist, not starting", "key", keypath)
		return
	}

	log.Infow("startup", "subsystem", "Firebase", "version", firebase.Version, "message", "Firebase starting", "key", keypath)

	app, err := firebase.NewApp(ctx, nil, option.WithCredentialsFile(keypath))
	if err != nil {
		err := fmt.Errorf("error initializing firebase messaging: %v", err)
		log.Error(err)
		return
	}

	msg, err = app.Messaging(ctx)
	if err != nil {
		log.Error(err)
		return
	}

	wm.RegisterMessageBus("firebase", wm.Bus{
		SendMessage:      sendMessage,
		SendTarget:       sendTarget,
		SendAnnounce:     sendAnnounce,
		AddToRemote:      addToRemote,
		RemoveFromRemote: removeFromRemote,
		// SendAssignment: sendAssignment,
		AgentDeleteOperation: agentDeleteOperation,
		DeleteOperation:      deleteOperation,
	})

	fbctx = ctx
	ratelimitinit()
	config.SetFirebaseRunning(true)

	<-ctx.Done()

	log.Infow("Shutdown", "message", "Firebase Shutting down")
	config.SetFirebaseRunning(false)
}