silvermobi/services/push.go

package services

import (
	"context"
	"google.golang.org/protobuf/types/known/structpb"

	"github.com/appleboy/gorush/rpc/proto"
	"github.com/rs/zerolog/log"
	"google.golang.org/grpc"
)

const (
	// PlatFormIos constant is 1 for iOS
	PushToIos = iota + 1
	// PlatFormAndroid constant is 2 for Android
	PushToAndroid
	// PlatFormHuawei constant is 3 for Huawei
	PushToHuawei
	// PlatformUnifiedPush constant is 4 for UnifiedPush
	PushToUnifiedPush
	// PlatformSMSFactor constant is 5 for SMSFactor
	PushToSMSFactor
)

type Notification struct {
	Platform   int32
	Recipients []string
	Message    string
	Title      string
	Data       *structpb.Struct
	ID         string
}

type PushService struct {
	Client proto.GorushClient
}

func NewPushService(address string) (*PushService, error) {
	conn, err := grpc.Dial(address, grpc.WithInsecure())

	if err != nil {
		return nil, err
	}

	return &PushService{
		Client: proto.NewGorushClient(conn),
	}, nil
}

func (s *PushService) Send(notification Notification) error {
	log.Debug().
		Int32("Platform", notification.Platform).
		Strs("recipients", notification.Recipients).
		Str("notification_message", notification.Message).
		Str("notification_title", notification.Title).
		Msg("Send notification")

	resp, err := s.Client.Send(context.Background(), &proto.NotificationRequest{
		Data:     notification.Data,
		ID:       notification.ID,
		Platform: notification.Platform,
		Tokens:   notification.Recipients,
		Message:  notification.Message,
		Title:    notification.Title,
		Priority: proto.NotificationRequest_HIGH,
		Alert: &proto.Alert{
			Title: notification.Title,
			Body:  notification.Message,
		},
	})

	if err != nil {
		return err
	}

	log.Debug().Str("response", resp.String()).Msg("notification sent")

	return nil
}