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",
@@ -175,19 +180,20 @@ func run(c *cli.Context) error {
 			Finished: c.Float64("job.finished"),
 		},
 		Config: Config{
-			Token:    c.String("token"),
-			Debug:    c.Bool("debug"),
-			To:       c.StringSlice("to"),
-			Message:  c.StringSlice("message"),
-			Photo:    c.StringSlice("photo"),
-			Document: c.StringSlice("document"),
-			Sticker:  c.StringSlice("sticker"),
-			Audio:    c.StringSlice("audio"),
-			Voice:    c.StringSlice("voice"),
-			Location: c.StringSlice("location"),
-			Video:    c.StringSlice("video"),
-			Venue:    c.StringSlice("venue"),
-			Format:   c.String("format"),
+			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"),
+			Document:   c.StringSlice("document"),
+			Sticker:    c.StringSlice("sticker"),
+			Audio:      c.StringSlice("audio"),
+			Voice:      c.StringSlice("voice"),
+			Location:   c.StringSlice("location"),
+			Video:      c.StringSlice("video"),
+			Venue:      c.StringSlice("venue"),
+			Format:     c.String("format"),
 		},
 	}
 

plugin.go

@@ -36,19 +36,20 @@ type (
 
 	// Config for the plugin.
 	Config struct {
-		Token    string
-		Debug    bool
-		To       []string
-		Message  []string
-		Photo    []string
-		Document []string
-		Sticker  []string
-		Audio    []string
-		Voice    []string
-		Location []string
-		Video    []string
-		Venue    []string
-		Format   string
+		Token      string
+		Debug      bool
+		MatchEmail bool
+		To         []string
+		Message    []string
+		Photo      []string
+		Document   []string
+		Sticker    []string
+		Audio      []string
+		Voice      []string
+		Location   []string
+		Video      []string
+		Venue      []string
+		Format     string
 	}
 
 	// Plugin values.
@@ -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, ":"))
+
+		// check id
+		id, err := strconv.ParseInt(idArray[0], 10, 64)
+		if err != nil {
+			continue
 		}
+
+		// 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)
 	}
 
-	id, err := strconv.ParseInt(ids[0], 10, 64)
-	if err != nil {
-		return int64(0), false
+	if matchEmail == true && attachEmail == false {
+		return emails
 	}
 
-	return id, true
+	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) {