added notification support for discord

_examples/example.config.toml

@@ -28,3 +28,8 @@ ssh_key = "/home/vsts/.beast/secret.key"
 url = "[email protected]:sdslabs/nonexistent.git"
 name = "nonexistent"
 branch = "nonexistent"
+
+[[webhooks]]
+url = "https://discordapp.com/api/webhooks/615970326093758464/Dui7gVZVIaGvdys87I9O2Gn9Bx3ssNkdgkxSf3etEXN0ClxlNYSeTflbJd0obO81a_m1"
+service_name= "discord"
+status=true

core/config/config.go

@@ -86,8 +86,7 @@ type BeastConfig struct {
 	AvailableSidecars  []string  `toml:"available_sidecars"`
 	GitRemote          GitRemote `toml:"remote"`
 	JWTSecret          string    `toml:"jwt_secret"`
-	SlackWebHookURL    string    `toml:"slack_webhook"`
-	DiscordWebHookURL  string    `toml:"discord_webhook"`
+	Webhooks           []Webhook `toml:webhook`
 	TickerFrequency    int       `toml:"ticker_frequency"`
 
 	RemoteSyncPeriod time.Duration `toml:"-"`
@@ -216,6 +215,12 @@ func (config *GitRemote) ValidateGitConfig() error {
 	return nil
 }
 
+type Webhook struct {
+	URL         string `toml:"url"`
+	ServiceName string `toml:"service_name"`
+	Status      bool   `toml:"status"`
+}
+
 // From the path of the config file provided as an arguement this function
 // loads the parse the config file and load it into the BeastConfig
 // structure. After parsing it validates the data in the config file and returns

core/manager/challenge.go

@@ -530,7 +530,7 @@ func StartUndeployChallenge(challengeName string, purge bool) error {
 		notify.SendNotification(notify.Success, msg)
 	}
 
-	log.Infof("Notification for the event sent to slack.")
+	log.Infof("Notification for the event sent.")
 	return err
 }
 

core/manager/pipeline.go

@@ -412,5 +412,5 @@ func StartDeployPipeline(challengeDir string, skipStage bool, skipCommit bool) {
 		notify.SendNotification(notify.Success, msg)
 	}
 
-	log.Debugf("%s: Notification sent to slack", challengeName)
+	log.Debugf("%s: Notification sent", challengeName)
 }

pkg/notify/main.go

@@ -75,42 +75,36 @@ func NewNotifier(URL string, ProviderType ProviderTypeEnum) Notifier {
 
 func (req *Request) FillReqParams() error {
 	req.PostPayload = PostPayload{
-		Username: "Beast",
-		IconUrl:  "https://i.ibb.co/sjC5dRY/beast-eye-39371.png",
-		Channel:  "#beast",
+		Username: USERNAME,
+		IconUrl:  ICON_URL,
+		Channel:  CHANNEL_NAME,
 	}
 	return nil
 }
 
 func SendNotification(nType NotificationType, message string) error {
-	if config.Cfg.SlackWebHookURL != "" {
-		slackNotifier := NewNotifier(config.Cfg.SlackWebHookURL, SlackProvider)
-
-		err := slackNotifier.SendNotification(nType, message)
-		if err != nil {
-			log.Errorf("Error while sending notification to slack : %s", err)
-			fmt.Errorf("NOTIFICATION_SEND_ERROR: %s", err)
-		}
-
-		log.Infof("Notfication sent to slack.")
-	} else {
-		log.Warnf("No slack webhook url provided in beast config, cannot send notification.")
-		fmt.Errorf("No webhook URL in beast config.")
-	}
-
-	if config.Cfg.DiscordWebHookURL != "" {
-		discordNotifier := NewNotifier(config.Cfg.DiscordWebHookURL, DiscordProvider)
-
-		err := discordNotifier.SendNotification(nType, message)
-		if err != nil {
-			log.Errorf("Error while sending notification to discord : %s", err)
-			return fmt.Errorf("NOTIFICATION_SEND_ERROR: %s", err)
+	for _, webhook := range config.Cfg.Webhooks {
+		if webhook.ServiceName != "" || webhook.Status == true {
+			var Provider ProviderTypeEnum
+			if webhook.ServiceName == "slack" {
+				Provider = SlackProvider
+			}
+			if webhook.ServiceName == "discord" {
+				Provider = DiscordProvider
+			}
+			Notifier := NewNotifier(webhook.URL, Provider)
+
+			err := Notifier.SendNotification(nType, message)
+			if err != nil {
+				log.Errorf("Error while sending notification to %s : %s", webhook.ServiceName, err)
+				fmt.Errorf("NOTIFICATION_SEND_ERROR: %s", err)
+			}
+
+			log.Infof("Notfication sent to %s.", webhook.ServiceName)
+		} else {
+			log.Warnf("No %s webhook url provided in beast config, cannot send notification.", webhook.ServiceName)
+			fmt.Errorf("No webhook URL in beast config.")
 		}
-
-		log.Infof("Notfication sent to discord.")
-		return nil
-	} else {
-		log.Warnf("No discord webhook url provided in beast config, cannot send notification.")
-		return fmt.Errorf("No webhook URL in beast config.")
 	}
+	return nil
 }

pkg/notify/notification.go → pkg/notify/notification_constants.go

@@ -15,3 +15,9 @@ const (
 	WarningColor NotificationColor = "#FF4500"
 	SuccessColor NotificationColor = "#32CD32"
 )
+
+const (
+	USERNAME     string = "Beast"
+	ICON_URL     string = "https://i.ibb.co/sjC5dRY/beast-eye-39371.png"
+	CHANNEL_NAME string = "#beast"
+)