Merge pull request #24 from appleboy/match

Support MATCH_EMAIL flag

README.md

@@ -66,6 +66,7 @@ docker run --rm \
   -e PLUGIN_VENUE=24.9163213,121.1424972,title,address \
   -e PLUGIN_VIDEO=tests/video.mp4 \
   -e PLUGIN_DEBUG=true \
+  -e PLUGIN_ONLY_MATCH_EMAIL=false \
   -e PLUGIN_FORMAT=markdown \
   -e DRONE_REPO_OWNER=appleboy \
   -e DRONE_REPO_NAME=go-hello \

main.go

@@ -77,6 +77,11 @@ func main() {
 			Usage:  "enable debug message",
 			EnvVar: "PLUGIN_DEBUG",
 		},
+		cli.BoolFlag{
+			Name:   "match.email",
+			Usage:  "send message when only match email",
+			EnvVar: "PLUGIN_ONLY_MATCH_EMAIL",
+		},
 		cli.StringFlag{
 			Name:   "format",
 			Value:  "markdown",
@@ -177,6 +182,7 @@ func run(c *cli.Context) error {
 		Config: Config{
 			Token:      c.String("token"),
 			Debug:      c.Bool("debug"),
+			MatchEmail: c.Bool("match.email"),
 			To:         c.StringSlice("to"),
 			Message:    c.StringSlice("message"),
 			Photo:      c.StringSlice("photo"),

plugin.go

@@ -38,6 +38,7 @@ type (
 	Config struct {
 		Token      string
 		Debug      bool
+		MatchEmail bool
 		To         []string
 		Message    []string
 		Photo      []string
@@ -135,22 +136,43 @@ func convertLocation(value string) (Location, bool) {
 	}, false
 }
 
-func parseTo(value, authorEmail string) (int64, bool) {
-	ids := trimElement(strings.Split(value, ":"))
+func parseTo(to []string, authorEmail string, matchEmail bool) []int64 {
+	var emails []int64
+	var ids []int64
+	attachEmail := true
 
-	if len(ids) > 1 {
-		if email := ids[1]; email != authorEmail {
-			log.Println("email not match")
-			return int64(0), false
-		}
-	}
+	for _, value := range to {
+		idArray := trimElement(strings.Split(value, ":"))
 
-	id, err := strconv.ParseInt(ids[0], 10, 64)
+		// check id
+		id, err := strconv.ParseInt(idArray[0], 10, 64)
 		if err != nil {
-		return int64(0), false
+			continue
 		}
 
-	return id, true
+		// check match author email
+		if len(idArray) > 1 {
+			if email := idArray[1]; email != authorEmail {
+				continue
+			}
+
+			emails = append(emails, id)
+			attachEmail = false
+			continue
+		}
+
+		ids = append(ids, id)
+	}
+
+	if matchEmail == true && attachEmail == false {
+		return emails
+	}
+
+	for _, value := range emails {
+		ids = append(ids, value)
+	}
+
+	return ids
 }
 
 // Exec executes the plugin.
@@ -179,6 +201,7 @@ func (p Plugin) Exec() error {
 
 	bot.Debug = p.Config.Debug
 
+	ids := parseTo(p.Config.To, p.Build.Email, p.Config.MatchEmail)
 	photos := fileExist(trimElement(p.Config.Photo))
 	documents := fileExist(trimElement(p.Config.Document))
 	stickers := fileExist(trimElement(p.Config.Sticker))
@@ -189,12 +212,7 @@ func (p Plugin) Exec() error {
 	venues := trimElement(p.Config.Venue)
 
 	// send message.
-	for _, to := range p.Config.To {
-		user, enable := parseTo(to, p.Build.Email)
-		if !enable {
-			continue
-		}
-
+	for _, user := range ids {
 		for _, value := range trimElement(message) {
 			txt, err := template.RenderTrim(value, p)
 			if err != nil {

plugin_test.go

@@ -133,30 +133,19 @@ func TestTrimElement(t *testing.T) {
 }
 
 func TestParseTo(t *testing.T) {
-	id, enable := parseTo("1234567890", "test@gmail.com")
+	input := []string{"0", "1:1@gmail.com", "2:2@gmail.com", "3:3@gmail.com", "4", "5"}
 
-	assert.Equal(t, true, enable)
-	assert.Equal(t, int64(1234567890), id)
+	ids := parseTo(input, "1@gmail.com", false)
+	assert.Equal(t, []int64{0, 4, 5, 1}, ids)
 
-	id, enable = parseTo("1234567890:test2@gmail.com", "test@gmail.com")
+	ids = parseTo(input, "1@gmail.com", true)
+	assert.Equal(t, []int64{1}, ids)
 
-	assert.Equal(t, false, enable)
-	assert.Equal(t, int64(0), id)
+	ids = parseTo(input, "a@gmail.com", false)
+	assert.Equal(t, []int64{0, 4, 5}, ids)
 
-	id, enable = parseTo("1234567890:test@gmail.com", "test@gmail.com")
-
-	assert.Equal(t, true, enable)
-	assert.Equal(t, int64(1234567890), id)
-
-	id, enable = parseTo("測試:test@gmail.com", "test@gmail.com")
-
-	assert.Equal(t, false, enable)
-	assert.Equal(t, int64(0), id)
-
-	id, enable = parseTo("測試", "test@gmail.com")
-
-	assert.Equal(t, false, enable)
-	assert.Equal(t, int64(0), id)
+	ids = parseTo(input, "a@gmail.com", true)
+	assert.Equal(t, []int64{0, 4, 5}, ids)
 }
 
 func TestCheckFileExist(t *testing.T) {