Issue #37: Added multi-cluster support for displaying cluster specifi…

…c information.

pkg/controller/controller.go

@@ -1,6 +1,7 @@
 package controller
 
 import (
+	"fmt"
 	"os"
 	"os/signal"
 	"strconv"
@@ -20,13 +21,13 @@ import (
 	"k8s.io/client-go/tools/cache"
 )
 
-var startTime time.Time
-
 const (
-	controllerStartMsg = "...and now my watch begins! :crossed_swords:"
-	controllerStopMsg  = "my watch has ended!"
+	controllerStartMsg = "...and now my watch begins for cluster '%s'! :crossed_swords:"
+	controllerStopMsg  = "my watch has ended for cluster '%s'!"
 )
 
+var startTime time.Time
+
 func findNamespace(ns string) string {
 	if ns == "all" {
 		return apiV1.NamespaceAll
@@ -39,7 +40,7 @@ func findNamespace(ns string) string {
 
 // RegisterInformers creates new informer controllers to watch k8s resources
 func RegisterInformers(c *config.Config) {
-	sendMessage(controllerStartMsg)
+	sendMessage(fmt.Sprintf(controllerStartMsg, c.Settings.ClusterName))
 	startTime = time.Now().Local()
 
 	// Get resync period
@@ -127,7 +128,7 @@ func RegisterInformers(c *config.Config) {
 	signal.Notify(sigterm, syscall.SIGTERM)
 	signal.Notify(sigterm, syscall.SIGINT)
 	<-sigterm
-	sendMessage(controllerStopMsg)
+	sendMessage(fmt.Sprintf(controllerStopMsg, c.Settings.ClusterName))
 }
 
 func registerEventHandlers(resourceType string, events []string) (handlerFns cache.ResourceEventHandlerFuncs) {

pkg/execute/executor.go

@@ -1,6 +1,7 @@
 package execute
 
 import (
+	"fmt"
 	"io/ioutil"
 	"os"
 	"os/exec"
@@ -34,10 +35,10 @@ var validNotifierCommands = map[string]bool{
 var kubectlBinary = "/usr/local/bin/kubectl"
 
 const (
-	notifierStartMsg   = "Brace yourselves, notifications are coming."
-	notifierStopMsg    = "Sure! I won't send you notifications anymore."
+	notifierStartMsg   = "Brace yourselves, notifications are coming from cluster '%s'."
+	notifierStopMsg    = "Sure! I won't send you notifications from cluster '%s' anymore."
 	unsupportedCmdMsg  = "Command not supported. Please run '@BotKube help' to see supported commands."
-	kubectlDisabledMsg = "Sorry, the admin hasn't given me the permission to execute kubectl command."
+	kubectlDisabledMsg = "Sorry, the admin hasn't given me the permission to execute kubectl command on cluster '%s'."
 )
 
 // Executor is an interface for processes to execute commands
@@ -49,13 +50,15 @@ type Executor interface {
 type DefaultExecutor struct {
 	Message      string
 	AllowKubectl bool
+	ClusterName  string
 }
 
 // NewDefaultExecutor returns new Executor object
-func NewDefaultExecutor(msg string, allowkubectl bool) Executor {
+func NewDefaultExecutor(msg string, allowkubectl bool, clustername string) Executor {
 	return &DefaultExecutor{
 		Message:      msg,
 		AllowKubectl: allowkubectl,
+		ClusterName:  clustername,
 	}
 }
 
@@ -64,12 +67,12 @@ func (e *DefaultExecutor) Execute() string {
 	args := strings.Split(e.Message, " ")
 	if validKubectlCommands[args[0]] {
 		if !e.AllowKubectl {
-			return kubectlDisabledMsg
+			return fmt.Sprintf(kubectlDisabledMsg, e.ClusterName)
 		}
-		return runKubectlCommand(args)
+		return fmt.Sprintf("Cluster: %s\n%s", e.ClusterName, runKubectlCommand(args))
 	}
 	if validNotifierCommands[args[0]] {
-		return runNotifierCommand(args)
+		return runNotifierCommand(args, e.ClusterName)
 	}
 	return unsupportedCmdMsg
 }
@@ -129,14 +132,14 @@ func runKubectlCommand(args []string) string {
 }
 
 // TODO: Have a seperate cli which runs bot commands
-func runNotifierCommand(args []string) string {
+func runNotifierCommand(args []string, clustername string) string {
 	switch len(args) {
 	case 1:
 		if strings.ToLower(args[0]) == "help" {
 			return printHelp()
 		}
 		if strings.ToLower(args[0]) == "ping" {
-			return "pong"
+			return fmt.Sprintf("pong from cluster '%s'", clustername)
 		}
 	case 2:
 		if args[0] != "notifier" {
@@ -145,26 +148,26 @@ func runNotifierCommand(args []string) string {
 		if args[1] == "start" {
 			config.Notify = true
 			log.Logger.Info("Notifier enabled")
-			return notifierStartMsg
+			return fmt.Sprintf(notifierStartMsg, clustername)
 		}
 		if args[1] == "stop" {
 			config.Notify = false
 			log.Logger.Info("Notifier disabled")
-			return notifierStopMsg
+			return fmt.Sprintf(notifierStopMsg, clustername)
 		}
 		if args[1] == "status" {
 			if config.Notify == false {
-				return "stopped"
+				return fmt.Sprintf("Notifications are off for cluster '%s'", clustername)
 			}
-			return "running"
+			return fmt.Sprintf("Notifications are on for cluster '%s'", clustername)
 		}
 		if args[1] == "showconfig" {
 			out, err := showControllerConfig()
 			if err != nil {
 				log.Logger.Error("Error in executing showconfig command: ", err)
 				return "Error in getting configuration!"
 			}
-			return out
+			return fmt.Sprintf("Showing config for cluster '%s'\n\n%s", clustername, out)
 		}
 	}
 	return printDefaultMsg()

pkg/slack/slack.go

@@ -14,8 +14,8 @@ import (
 type Bot struct {
 	Token        string
 	AllowKubectl bool
+	ClusterName  string
 	ChannelName  string
-	CheckChannel bool
 }
 
 // slackMessage contains message details to execute command and send back the result
@@ -37,8 +37,8 @@ func NewSlackBot() *Bot {
 	return &Bot{
 		Token:        c.Communications.Slack.Token,
 		AllowKubectl: c.Settings.AllowKubectl,
+		ClusterName:  c.Settings.ClusterName,
 		ChannelName:  c.Communications.Slack.Channel,
-		CheckChannel: c.Settings.CheckChannel,
 	}
 }
 
@@ -95,7 +95,7 @@ func (b *Bot) Start() {
 				InMessage: inMessage,
 				RTM:       rtm,
 			}
-			sm.HandleMessage(b.AllowKubectl)
+			sm.HandleMessage(b.AllowKubectl, b.ClusterName)
 
 		case *slack.RTMError:
 			logging.Logger.Errorf("Slack RMT error: %+v", ev.Error())
@@ -108,8 +108,8 @@ func (b *Bot) Start() {
 	}
 }
 
-func (sm *slackMessage) HandleMessage(allowkubectl bool) {
-	e := execute.NewDefaultExecutor(sm.InMessage, allowkubectl)
+func (sm *slackMessage) HandleMessage(allowkubectl bool, clustername string) {
+	e := execute.NewDefaultExecutor(sm.InMessage, allowkubectl, clustername)
 	sm.OutMessage = e.Execute()
 	sm.OutMsgLength = len(sm.OutMessage)
 	sm.Send()