discord: multiple discord feature

Author: wuriyanto48

cmd/tob/main.go

@@ -45,7 +45,7 @@ func main() {
 	}
 
 	// init Notificator
-	notificators, err := tob.InitNotificatorFactory(configs)
+	notificators, err := tob.InitNotificatorFactory(configs, args.Verbose)
 	if err != nil {
 		fmt.Println("error: ", err)
 		os.Exit(1)

notificator.go

@@ -23,9 +23,9 @@ type Notificator interface {
 }
 
 // InitNotificatorFactory will init all notificator
-func InitNotificatorFactory(configs config.Config) ([]Notificator, error) {
+func InitNotificatorFactory(configs config.Config, verbose bool) ([]Notificator, error) {
 	// discord notificator
-	discordNotificator, err := discord.NewDiscord(configs)
+	discordNotificator, err := discord.NewDiscord(configs, verbose, Logger)
 	if err != nil {
 		return nil, err
 	}

notificators/discord/discord.go

@@ -7,6 +7,8 @@ import (
 	"fmt"
 	"github.com/telkomdev/tob/config"
 	"github.com/telkomdev/tob/httpx"
+	"io"
+	"log"
 	"strings"
 )
 
@@ -23,8 +25,8 @@ type DiscordResponse struct {
 	Message string `json:"message"`
 }
 
-// Discord represent discord notificator
-type Discord struct {
+// DiscordConfig represent discord config
+type DiscordConfig struct {
 	threadURL string
 	name      string
 	avatarURL string
@@ -33,67 +35,89 @@ type Discord struct {
 	enabled   bool
 }
 
+// Discord represent discord notificator
+type Discord struct {
+	configs []DiscordConfig
+	logger  *log.Logger
+	verbose bool
+}
+
 // NewDiscord Discord's constructor
-func NewDiscord(configs config.Config) (*Discord, error) {
+func NewDiscord(configs config.Config, verbose bool, logger *log.Logger) (*Discord, error) {
 	notificatorConfigInterface, ok := configs["notificator"]
 	if !ok {
 		return nil, errors.New("error: cannot find notificator field in the config file")
 	}
 
 	notificatorConfig := notificatorConfigInterface.(map[string]interface{})
 
-	discordConfigInterface, ok := notificatorConfig["discord"]
+	discordConfigInterfaces, ok := notificatorConfig["discord"]
 	if !ok {
 		return nil, errors.New("error: cannot find discord field in the config file")
 	}
 
-	discordConfig := discordConfigInterface.(map[string]interface{})
+	var discordConfigs []DiscordConfig
+	discordConfigList := discordConfigInterfaces.([]interface{})
+	for _, discordConfigInterface := range discordConfigList {
+		discordConfig, ok := discordConfigInterface.(map[string]interface{})
+		if !ok {
+			return nil, errors.New("error: discord config is not valid")
+		}
 
-	name, ok := discordConfig["name"].(string)
-	if !ok {
-		return nil, errors.New("error: cannot find discord name field in the config file")
-	}
+		name, ok := discordConfig["name"].(string)
+		if !ok {
+			return nil, errors.New("error: cannot find discord name field in the config file")
+		}
 
-	threadURL, ok := discordConfig["url"].(string)
-	if !ok {
-		return nil, errors.New("error: cannot find discord url field in the config file")
-	}
+		threadURL, ok := discordConfig["url"].(string)
+		if !ok {
+			return nil, errors.New("error: cannot find discord url field in the config file")
+		}
 
-	avatarURL, ok := discordConfig["avatarUrl"].(string)
-	if !ok {
-		return nil, errors.New("error: cannot find discord avatarUrl field in the config file")
-	}
+		avatarURL, ok := discordConfig["avatarUrl"].(string)
+		if !ok {
+			return nil, errors.New("error: cannot find discord avatarUrl field in the config file")
+		}
 
-	mentionsInterface, ok := discordConfig["mentions"].([]interface{})
-	if !ok {
-		return nil, errors.New("error: cannot find discord mentions field in the config file")
-	}
+		mentionsInterface, ok := discordConfig["mentions"].([]interface{})
+		if !ok {
+			return nil, errors.New("error: cannot find discord mentions field in the config file")
+		}
+
+		var mentions []string
+		for _, mentionInterface := range mentionsInterface {
+			mention, ok := mentionInterface.(string)
+			if !ok {
+				return nil, errors.New("error: mention field is not valid string")
+			}
 
-	var mentions []string
-	for _, mentionInterface := range mentionsInterface {
-		mention, ok := mentionInterface.(string)
+			mentions = append(mentions, mention)
+		}
+
+		enabled, ok := discordConfig["enable"].(bool)
 		if !ok {
-			return nil, errors.New("error: mention field is not valid string")
+			return nil, errors.New("error: cannot find discord enable field in the config file")
 		}
 
-		mentions = append(mentions, mention)
-	}
+		headers := make(map[string]string)
+		headers["Content-Type"] = "application/json"
 
-	enabled, ok := discordConfig["enable"].(bool)
-	if !ok {
-		return nil, errors.New("error: cannot find discord enable field in the config file")
-	}
+		conf := DiscordConfig{
+			name:      name,
+			threadURL: threadURL,
+			avatarURL: avatarURL,
+			enabled:   enabled,
+			headers:   headers,
+			mentions:  mentions,
+		}
 
-	headers := make(map[string]string)
-	headers["Content-Type"] = "application/json"
+		discordConfigs = append(discordConfigs, conf)
+	}
 
 	return &Discord{
-		name:      name,
-		threadURL: threadURL,
-		avatarURL: avatarURL,
-		enabled:   enabled,
-		headers:   headers,
-		mentions:  mentions,
+		configs: discordConfigs,
+		logger:  logger,
+		verbose: verbose,
 	}, nil
 }
 
@@ -104,44 +128,59 @@ func (d *Discord) Provider() string {
 
 // Send will send notification
 func (d *Discord) Send(msg string) error {
-	var messageBuilder strings.Builder
-
-	messageBuilder.WriteString("Hey ")
-	for _, mention := range d.mentions {
-		if strings.Contains(mention, "here") {
-			messageBuilder.WriteString(fmt.Sprintf("%s", mention))
-		} else {
-			messageBuilder.WriteString(fmt.Sprintf("<%s>", mention))
+	for _, conf := range d.configs {
+		if conf.enabled {
+			var messageBuilder strings.Builder
+
+			messageBuilder.WriteString("Hey ")
+			for _, mention := range conf.mentions {
+				if strings.Contains(mention, "here") {
+					messageBuilder.WriteString(fmt.Sprintf("%s", mention))
+				} else {
+					messageBuilder.WriteString(fmt.Sprintf("<%s>", mention))
+				}
+
+				messageBuilder.WriteString(", ")
+			}
+
+			messageBuilder.WriteString(" ")
+			messageBuilder.WriteString(msg)
+
+			content := messageBuilder.String()
+
+			discordMessage := DiscordMessage{
+				Username:  conf.name,
+				AvatarURL: conf.avatarURL,
+				Content:   content,
+			}
+
+			messageJSON, err := json.Marshal(discordMessage)
+			if err != nil {
+				return err
+			}
+
+			go func(threadURL string, body io.Reader, headers map[string]string, timeout int) {
+				_, err = httpx.HTTPPost(threadURL, body, headers, timeout)
+				if err != nil {
+					if d.verbose {
+						d.logger.Println(err)
+					}
+				}
+			}(conf.threadURL, bytes.NewBuffer(messageJSON), conf.headers, 5)
 		}
-
-		messageBuilder.WriteString(", ")
-	}
-
-	messageBuilder.WriteString(" ")
-	messageBuilder.WriteString(msg)
-
-	msg = messageBuilder.String()
-
-	discordMessage := DiscordMessage{
-		Username:  d.name,
-		AvatarURL: d.avatarURL,
-		Content:   msg,
-	}
-
-	messageJSON, err := json.Marshal(discordMessage)
-	if err != nil {
-		return err
-	}
-
-	_, err = httpx.HTTPPost(d.threadURL, bytes.NewBuffer(messageJSON), d.headers, 5)
-	if err != nil {
-		return err
 	}
 
 	return nil
 }
 
 // IsEnabled will return enable status
 func (d *Discord) IsEnabled() bool {
-	return d.enabled
+	if len(d.configs) > 0 {
+		temp := d.configs[0].enabled
+		for _, conf := range d.configs {
+			temp = temp || conf.enabled
+		}
+		return temp
+	}
+	return false
 }

runner.go

@@ -142,6 +142,9 @@ func (r *Runner) InitServices() error {
 			service.Enable(serviceEnabled)
 			service.SetConfig(conf)
 
+			// by default service is recovered
+			service.SetRecover(true)
+
 			err = service.Connect()
 			if err != nil {
 				return err