112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
package fcm
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
firebase "firebase.google.com/go"
|
|
"firebase.google.com/go/messaging"
|
|
"git.selly.red/Selly-Modules/mongodb"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"google.golang.org/api/option"
|
|
)
|
|
|
|
// MongoDBConfig ...
|
|
type MongoDBConfig struct {
|
|
Host, User, Password, DBName, Mechanism, Source string
|
|
}
|
|
|
|
// Config ...
|
|
type Config struct {
|
|
// Project id
|
|
ProjectID string
|
|
// Original is JSON format, but encoded with base64, need to call base64 decode to get byte data
|
|
Credential string
|
|
// MongoDB config, for save logs
|
|
MongoDB MongoDBConfig
|
|
}
|
|
|
|
// Result of each send
|
|
type Result struct {
|
|
BatchID string
|
|
Success int
|
|
Failure int
|
|
ErrorTokens []string
|
|
}
|
|
|
|
// List topics
|
|
const (
|
|
TopicAll = "all"
|
|
TopicIOS = "iOS"
|
|
TopicAndroid = "android"
|
|
)
|
|
|
|
var allowedTopics = []string{TopicAll, TopicIOS, TopicAndroid}
|
|
|
|
// Service ...
|
|
type Service struct {
|
|
Config
|
|
Client *messaging.Client
|
|
DB *mongo.Database
|
|
}
|
|
|
|
var s *Service
|
|
|
|
// NewInstance for push notification
|
|
func NewInstance(config Config) error {
|
|
if config.ProjectID == "" || config.Credential == "" || config.MongoDB.Host == "" {
|
|
return errors.New("please provide all information that needed: projectId, credential, postgresql")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
// Connect MongoDB
|
|
_, err := mongodb.Connect(mongodb.Config{
|
|
Host: config.MongoDB.Host,
|
|
DBName: config.MongoDB.DBName,
|
|
Monitor: nil,
|
|
TLS: nil,
|
|
Standalone: &mongodb.ConnectStandaloneOpts{
|
|
AuthMechanism: config.MongoDB.Mechanism,
|
|
AuthSource: config.MongoDB.Source,
|
|
Username: config.MongoDB.User,
|
|
Password: config.MongoDB.Password,
|
|
},
|
|
})
|
|
if err != nil {
|
|
fmt.Println("FCM - Connect MongoDB error", err)
|
|
return err
|
|
}
|
|
|
|
// Setup
|
|
credential := base64Decode(config.Credential)
|
|
opts := option.WithCredentialsJSON(credential)
|
|
app, err := firebase.NewApp(context.Background(), &firebase.Config{
|
|
ProjectID: config.ProjectID,
|
|
}, opts)
|
|
if err != nil {
|
|
fmt.Println("FCM - New app error", err)
|
|
return errors.New("error when init Firebase app")
|
|
}
|
|
|
|
// Init messaging client
|
|
client, err := app.Messaging(ctx)
|
|
if err != nil {
|
|
fmt.Println("FCM - init messaging client error", err)
|
|
return errors.New("error when init Firebase messaging client")
|
|
}
|
|
|
|
s = &Service{
|
|
Config: config,
|
|
Client: client,
|
|
DB: mongodb.GetInstance(),
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetInstance ...
|
|
func GetInstance() *Service {
|
|
return s
|
|
}
|