support Device Group HTTP POST Request (#299)

* support Device Group HTTP POST Request

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

* fix vendor

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

* check error exist

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

gorush/notification.go

@@ -117,7 +117,7 @@ func CheckMessage(req PushNotification) error {
 	var msg string
 
 	// ignore send topic mesaage from FCM
-	if !req.IsTopic() && len(req.Tokens) == 0 {
+	if !req.IsTopic() && len(req.Tokens) == 0 && len(req.To) == 0 {
 		msg = "the message must specify at least one registration ID"
 		LogAccess.Debug(msg)
 		return errors.New(msg)

gorush/notification_fcm.go

@@ -42,7 +42,9 @@ func GetAndroidNotification(req PushNotification) *fcm.Message {
 		DryRun:                req.DryRun,
 	}
 
-	notification.RegistrationIDs = req.Tokens
+	if len(req.Tokens) > 0 {
+		notification.RegistrationIDs = req.Tokens
+	}
 
 	if len(req.Priority) > 0 && req.Priority == "high" {
 		notification.Priority = "high"
@@ -133,17 +135,24 @@ Retry:
 	var newTokens []string
 	// result from Send messages to specific devices
 	for k, result := range res.Results {
+		to := ""
+		if k < len(req.Tokens) {
+			to = req.Tokens[k]
+		} else {
+			to = req.To
+		}
+
 		if result.Error != nil {
 			isError = true
-			newTokens = append(newTokens, req.Tokens[k])
-			LogPush(FailedPush, req.Tokens[k], req, result.Error)
+			newTokens = append(newTokens, to)
+			LogPush(FailedPush, to, req, result.Error)
 			if PushConf.Core.Sync {
-				req.AddLog(getLogPushEntry(FailedPush, req.Tokens[k], req, result.Error))
+				req.AddLog(getLogPushEntry(FailedPush, to, req, result.Error))
 			}
 			continue
 		}
 
-		LogPush(SucceededPush, req.Tokens[k], req, nil)
+		LogPush(SucceededPush, to, req, nil)
 	}
 
 	// result from Send messages to topics
@@ -168,6 +177,19 @@ Retry:
 		}
 	}
 
+	// Device Group HTTP Response
+	if len(res.FailedRegistrationIDs) > 0 {
+		isError = true
+		for _, id := range res.FailedRegistrationIDs {
+			newTokens = append(newTokens, id)
+		}
+
+		LogPush(FailedPush, notification.To, req, errors.New("device group: partial success or all fails"))
+		if PushConf.Core.Sync {
+			req.AddLog(getLogPushEntry(FailedPush, notification.To, req, errors.New("device group: partial success or all fails")))
+		}
+	}
+
 	if isError && retryCount < maxRetry {
 		retryCount++
 

gorush/notification_test.go

@@ -174,6 +174,32 @@ func TestSyncModeForTopicNotification(t *testing.T) {
 	assert.Equal(t, 1, len(logs))
 }
 
+func TestSyncModeForDeviceGroupNotification(t *testing.T) {
+	PushConf, _ = config.LoadConf("")
+
+	PushConf.Android.Enabled = true
+	PushConf.Android.APIKey = os.Getenv("ANDROID_API_KEY")
+	PushConf.Log.HideToken = false
+
+	// enable sync mode
+	PushConf.Core.Sync = true
+
+	req := RequestPush{
+		Notifications: []PushNotification{
+			// android
+			{
+				To:       "aUniqueKey",
+				Platform: PlatFormAndroid,
+				Message:  "This is a Firebase Cloud Messaging Device Group Message!",
+			},
+		},
+	}
+
+	count, logs := queueNotification(req)
+	assert.Equal(t, 1, count)
+	assert.Equal(t, 1, len(logs))
+}
+
 func TestSetProxyURL(t *testing.T) {
 
 	err := SetProxy("87.236.233.92:8080")