feat: implement gRPC support sound field for FCM (#794)

- Add `IsGRPC` field to `PushNotification` struct - Add logic to handle sound configuration for gRPC notifications in `GetAndroidNotification` - Set `IsGRPC` to `true` in the `Send` function of the `Server` struct Signed-off-by: Bo-Yi Wu <[email protected]>
appleboy · Jul 16, 2024 · cc8b9e4 · cc8b9e4
1 parent 1f48853
commit cc8b9e4
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/notify/notification.go b/notify/notification.go
@@ -115,6 +115,7 @@ type PushNotification struct {
 
 	// ref: https://github.com/sideshow/apns2/blob/54928d6193dfe300b6b88dad72b7e2ae138d4f0a/payload/builder.go#L7-L24
 	InterruptionLevel string `json:"interruption_level,omitempty"`
+	IsGRPC            bool   `json:"is_grpc,omitempty"`
 }
 
 // Bytes for queue message

diff --git a/notify/notification_fcm.go b/notify/notification_fcm.go
@@ -73,6 +73,22 @@ func GetAndroidNotification(req *PushNotification) []*messaging.Message {
 		}
 	}
 
+	// Check if the notification has a sound
+	if req.Sound != "" && req.IsGRPC {
+		req.APNS = &messaging.APNSConfig{
+			Payload: &messaging.APNSPayload{
+				Aps: &messaging.Aps{
+					Sound: req.Sound.(string),
+				},
+			},
+		}
+		req.Android = &messaging.AndroidConfig{
+			Notification: &messaging.AndroidNotification{
+				Sound: req.Sound.(string),
+			},
+		}
+	}
+
 	// Check if the notification is a topic
 	if req.IsTopic() {
 		message := &messaging.Message{

diff --git a/rpc/server.go b/rpc/server.go
@@ -79,6 +79,7 @@ func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*prot
 		Priority:         strings.ToLower(in.GetPriority().String()),
 		PushType:         in.PushType,
 		Development:      in.Development,
+		IsGRPC:           true,
 	}
 
 	if badge > 0 {