Add hideToken func.

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

gorush/log.go

@@ -7,7 +7,7 @@ import (
 	"github.com/Sirupsen/logrus"
 	"github.com/gin-gonic/gin"
 	"os"
-	// "time"
+	"strings"
 )
 
 var (
@@ -181,6 +181,20 @@ func typeForPlatForm(platform int) string {
 	}
 }
 
+func hideToken(token string, starLen int) string {
+	if len(token) == 0 {
+		return ""
+	}
+
+	start := token[len(token)-starLen:]
+	end := token[0:starLen]
+
+	result := strings.Replace(token, start, strings.Repeat("*", starLen), -1)
+	result = strings.Replace(result, end, strings.Repeat("*", starLen), -1)
+
+	return result
+}
+
 // LogPush record user push request and server response.
 func LogPush(status, token string, req PushNotification, errPush error) {
 	var plat, platColor, output string

gorush/log_test.go

@@ -88,3 +88,8 @@ func TestPlatFormColor(t *testing.T) {
 	assert.Equal(t, yellow, colorForPlatForm(PlatFormAndroid))
 	assert.Equal(t, reset, colorForPlatForm(1000000))
 }
+
+func TestHideToken(t *testing.T) {
+	assert.Equal(t, "", hideToken("", 2))
+	assert.Equal(t, "**345678**", hideToken("1234567890", 2))
+}