Merge remote-tracking branch 'upstream/master' into feat-downtime

Author: lutfailham96

Makefile

@@ -25,4 +25,4 @@ format:
 	find . -name "*.go" -not -path "*/vendor/*" -not -path "*/.git/*" -not -path "*/volumes/*" | xargs gofmt -s -d -w
 
 clean:
-	rm tob tob.exe *.txt
+	rm -f tob tob.exe *.txt

cmd/tob/main.go

@@ -5,12 +5,15 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
+	"runtime"
 	"syscall"
 
 	"github.com/telkomdev/tob"
 	"github.com/telkomdev/tob/config"
 	"github.com/telkomdev/tob/notificators/discord"
 	"github.com/telkomdev/tob/notificators/email"
+	"github.com/telkomdev/tob/notificators/slack"
+	"github.com/telkomdev/tob/notificators/telegram"
 )
 
 func main() {
@@ -22,7 +25,7 @@ func main() {
 	}
 
 	if args.ShowVersion {
-		fmt.Printf("%s version %s\n", os.Args[0], tob.Version)
+		fmt.Printf("%s version %s (runtime: %s)\n", os.Args[0], tob.Version, runtime.Version())
 		os.Exit(0)
 	}
 
@@ -61,7 +64,26 @@ func main() {
 		os.Exit(1)
 	}
 
-	notificators := []tob.Notificator{emailNotificator, discordNotificator}
+	// slack notificator
+	slackNotificator, err := slack.NewSlack(configs)
+	if err != nil {
+		fmt.Println("error: ", err)
+		os.Exit(1)
+	}
+
+	// telegram notificator
+	telegramNotificator, err := telegram.NewTelegram(configs)
+	if err != nil {
+		fmt.Println("error: ", err)
+		os.Exit(1)
+	}
+
+	notificators := []tob.Notificator{
+		emailNotificator,
+		discordNotificator,
+		slackNotificator,
+		telegramNotificator,
+	}
 
 	ctx, cancel := context.WithCancel(context.Background())
 	// ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
@@ -96,6 +118,6 @@ func waitNotify(kill chan os.Signal, runner *tob.Runner) {
 	select {
 	case <-kill:
 		runner.Stop() <- true
-		fmt.Println("kill process")
+		fmt.Println("kill tob")
 	}
 }

config_example.json

@@ -42,6 +42,13 @@
             "enable": true
         },
 
+        "elasticsearch_cluster_one": {
+            "kind": "web",
+            "url": "http://elastic:devpass@localhost:9200/_cluster/health?pretty",
+            "checkInterval": 10,
+            "enable": true
+        },
+
         "dummy_one": {
             "kind": "dummy",
             "url": "https://minio-storage.com/minio/health/ready",
@@ -55,9 +62,23 @@
             "name": "BoyBot",
             "url": "https://discord.com/api/webhooks/YOUR_API_ID/YOUR_API_ID",
             "avatarUrl": "https://i.imgur.com/KHzN7QR.png",
+            "mentions": [
+                "@8XXXXXXXXXXXXXXXXX",
+                "@here"
+            ],
             "enable": true
         },
 
+        "slack": {
+            "name": "BoyBot",
+            "webhookUrl": "https://hooks.slack.com/services/XXXXXXXXX/XXXXXXXXXXX/XXXXXXXXXXXXXXXXXXXXXX",
+            "mentions": [
+                "@UXXXXXXXX",
+                "!here"
+            ],
+            "enable": false
+        },
+
         "email": {
             "authEmail": "[email protected]",
             "authPassword": "xxxx",
@@ -66,6 +87,12 @@
             "from": "[email protected]",
             "to": ["[email protected]", "[email protected]"],
             "subject": "Tob Notification",
+            "enable": false
+        },
+
+        "telegram": {
+            "botToken": "XXXXXXXXXX:XXXXXXXXXXX_XXXXXXXXXXXXXXXXXXXXXXX",
+            "groupId": "-111111111",
             "enable": true
         }
     },

notificators/discord/discord.go

@@ -4,8 +4,10 @@ import (
 	"bytes"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"github.com/telkomdev/tob/config"
 	"github.com/telkomdev/tob/httpx"
+	"strings"
 )
 
 // DiscordMessage represent discord request
@@ -27,6 +29,7 @@ type Discord struct {
 	name      string
 	avatarURL string
 	headers   map[string]string
+	mentions  []string
 	enabled   bool
 }
 
@@ -61,6 +64,21 @@ func NewDiscord(configs config.Config) (*Discord, error) {
 		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")
+	}
+
+	var mentions []string
+	for _, mentionInterface := range mentionsInterface {
+		mention, ok := mentionInterface.(string)
+		if !ok {
+			return nil, errors.New("error: mention field is not valid string")
+		}
+
+		mentions = append(mentions, mention)
+	}
+
 	enabled, ok := discordConfig["enable"].(bool)
 	if !ok {
 		return nil, errors.New("error: cannot find discord enable field in the config file")
@@ -75,6 +93,7 @@ func NewDiscord(configs config.Config) (*Discord, error) {
 		avatarURL: avatarURL,
 		enabled:   enabled,
 		headers:   headers,
+		mentions:  mentions,
 	}, nil
 }
 
@@ -85,6 +104,24 @@ 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))
+		}
+
+		messageBuilder.WriteString(", ")
+	}
+
+	messageBuilder.WriteString(" ")
+	messageBuilder.WriteString(msg)
+
+	msg = messageBuilder.String()
+
 	discordMessage := DiscordMessage{
 		Username:  d.name,
 		AvatarURL: d.avatarURL,

notificators/slack/slack.go

@@ -0,0 +1,126 @@
+package slack
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/telkomdev/tob/config"
+	"github.com/telkomdev/tob/httpx"
+	"strings"
+)
+
+// https://api.slack.com/messaging/sending
+
+// SlackMessage represent Slack request
+type SlackMessage struct {
+	Text string `json:"text"`
+}
+
+// SlackResponse represent Slack response
+type SlackResponse struct {
+	Code    uint   `json:"code"`
+	Message string `json:"message"`
+}
+
+// Slack represent Slack notificator
+type Slack struct {
+	webhookURL string
+	headers    map[string]string
+	mentions   []string
+	enabled    bool
+}
+
+// NewSlack Slack's constructor
+func NewSlack(configs config.Config) (*Slack, 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{})
+
+	slackConfigInterface, ok := notificatorConfig["slack"]
+	if !ok {
+		return nil, errors.New("error: cannot find Slack field in the config file")
+	}
+
+	slackConfig := slackConfigInterface.(map[string]interface{})
+
+	webhookURL, ok := slackConfig["webhookUrl"].(string)
+	if !ok {
+		return nil, errors.New("error: cannot find Slack webhookUrl field in the config file")
+	}
+
+	mentionsInterface, ok := slackConfig["mentions"].([]interface{})
+	if !ok {
+		return nil, errors.New("error: cannot find Slack 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")
+		}
+
+		mentions = append(mentions, mention)
+	}
+
+	enabled, ok := slackConfig["enable"].(bool)
+	if !ok {
+		return nil, errors.New("error: cannot find Slack enable field in the config file")
+	}
+
+	headers := make(map[string]string)
+	headers["Content-Type"] = "application/json"
+
+	return &Slack{
+		webhookURL: webhookURL,
+		enabled:    enabled,
+		headers:    headers,
+		mentions:   mentions,
+	}, nil
+}
+
+// Provider will return Notificator provider
+func (d *Slack) Provider() string {
+	return "slack"
+}
+
+// Send will send notification
+func (d *Slack) Send(msg string) error {
+	var messageBuilder strings.Builder
+
+	messageBuilder.WriteString("Hey ")
+	for _, mention := range d.mentions {
+		messageBuilder.WriteString(fmt.Sprintf("<%s>", mention))
+		messageBuilder.WriteString(", ")
+	}
+
+	messageBuilder.WriteString(" ")
+	messageBuilder.WriteString(msg)
+
+	msg = messageBuilder.String()
+
+	slackMessage := SlackMessage{
+		Text: msg,
+	}
+
+	messageJSON, err := json.Marshal(slackMessage)
+	if err != nil {
+		return err
+	}
+
+	_, err = httpx.HTTPPost(d.webhookURL, bytes.NewBuffer(messageJSON), d.headers, 5)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// IsEnabled will return enable status
+func (d *Slack) IsEnabled() bool {
+	return d.enabled
+}