feat: add initial sshNotifier

common/alarm.go

@@ -51,7 +51,7 @@ var AlarmSendCmd = &cobra.Command{
     Run: func(cmd *cobra.Command, args []string) {
         Init()
         message, _ := cmd.Flags().GetString("message")
-        Alarm(message)
+        Alarm(message, "", "", false)
     },
 }
 
@@ -94,7 +94,7 @@ func AlarmCheckUp(service string, message string, noInterval bool) {
         return
     } else {
         os.Remove(file_path)
-        Alarm(messageFinal)
+        Alarm(messageFinal, "", "", false)
     }
 }
 
@@ -166,7 +166,7 @@ func AlarmCheckDown(service string, message string, noInterval bool) {
 
                 err = os.WriteFile(filePath, jsonData, 0644)
 
-                Alarm(messageFinal)
+                Alarm(messageFinal, "", "", false)
             }
             return
         }
@@ -185,7 +185,7 @@ func AlarmCheckDown(service string, message string, noInterval bool) {
                 LogError("Error writing to file: \n" + err.Error())
             }
 
-            Alarm(messageFinal)
+            Alarm(messageFinal, "", "", false)
         } else {
             if j.Locked == false {
                 // currentDate - oldDate in minutes
@@ -203,7 +203,7 @@ func AlarmCheckDown(service string, message string, noInterval bool) {
                         LogError("Error writing to file: \n" + err.Error())
                     }
 
-                    Alarm(messageFinal)
+                    Alarm(messageFinal, "", "", false)
                 }
             }
         }
@@ -232,7 +232,7 @@ func AlarmCheckDown(service string, message string, noInterval bool) {
 
 
         if Config.Alarm.Interval == 0 {
-            Alarm(messageFinal)
+            Alarm(messageFinal, "", "", false)
         }
     }        
 }
@@ -243,7 +243,7 @@ type ResponseData struct {
     Code string `json:"code"`
 }
 
-func Alarm(m string) {
+func Alarm(m string, customStream string, customTopic string, onlyFirstWebhook bool) {
     if Config.Alarm.Enabled == false {
         return
     }
@@ -253,6 +253,13 @@ func Alarm(m string) {
     body:= []byte(`{"text":"` + message + `"}`)
 
     for _, webhook_url := range Config.Alarm.Webhook_urls {
+
+		if customStream != "" && customTopic != "" {
+			// Remove everything after &
+			webhook_url = strings.Split(webhook_url, "&")[0]
+			webhook_url = webhook_url + "&stream=" + customStream + "&topic=" + customTopic
+		}
+
         r, err := http.NewRequest("POST", webhook_url, bytes.NewBuffer(body))
         r.Header.Set("Content-Type", "application/json")
 
@@ -286,5 +293,9 @@ func Alarm(m string) {
         }
 
         defer res.Body.Close()
+
+		if onlyFirstWebhook == true {
+			break
+		}
     }
 }

main.go

@@ -9,6 +9,7 @@ import (
 	"github.com/monobilisim/monokit/osHealth"
 	"github.com/monobilisim/monokit/shutdownNotifier"
 	"github.com/monobilisim/monokit/pritunlHealth"
+	"github.com/monobilisim/monokit/sshNotifier"
 	"github.com/spf13/cobra"
 	"os"
 )
@@ -49,6 +50,12 @@ func main() {
 		Run:   pritunlHealth.Main,
 	}
 
+	var sshNotifierCmd = &cobra.Command{
+		Use:   "sshNotifier",
+		Short: "SSH Notifier",
+		Run:   sshNotifier.Main,
+	}
+
 	//// Common
 	RootCmd.AddCommand(redmineCmd)
 	RootCmd.AddCommand(common.AlarmCmd)
@@ -215,6 +222,12 @@ func main() {
 	/// Kubernetes Health
 	RootCmd.AddCommand(k8sHealthCmd)
 
+	/// SSH Notifier
+	RootCmd.AddCommand(sshNotifierCmd)
+
+	sshNotifierCmd.Flags().BoolP("login", "1", false, "Login")
+	sshNotifierCmd.Flags().BoolP("logout", "0", false, "Logout")
+
 	kubeconfig := os.Getenv("KUBECONFIG")
 
 	if kubeconfig == "" {

mysqlHealth/mysql.go

@@ -344,7 +344,7 @@ func CheckDB() {
 		for _, table := range tables {
 			message += table + "\n"
 		}
-		common.Alarm(message)
+		common.Alarm(message, "", "", false)
 	}
 }
 

shutdownNotifier/main.go

@@ -15,9 +15,9 @@ func Main(cmd *cobra.Command, args []string) {
     poweroff, _ := cmd.Flags().GetBool("poweroff")
 
     if poweron {
-        common.Alarm("[ " + common.Config.Identifier + " ] [:info: Info] Server is up...")
+        common.Alarm("[ " + common.Config.Identifier + " ] [:info: Info] Server is up...", "", "", false)
     } else if poweroff {
-        common.Alarm("[ " + common.Config.Identifier + " ] [:warning: Warning] Server is shutting down...")   } else {
+        common.Alarm("[ " + common.Config.Identifier + " ] [:warning: Warning] Server is shutting down...", "", "", false)   } else {
         fmt.Println("No action specified")
     }