From bcf0e91d0b86a7855e256a06bfcaa65c4de03526 Mon Sep 17 00:00:00 2001
From: Christopher Ramey <noreply@binarythought.com>
Date: Mon, 18 Jan 2021 12:47:03 -0900
Subject: [PATCH] added manual alarm triggering for testing

---
 alarm/alarm_email.go | 63 ++++++++++++++++++++++----------------------
 main.go              | 39 ++++++++++++++++++++++-----
 2 files changed, 64 insertions(+), 38 deletions(-)

diff --git a/alarm/alarm_email.go b/alarm/alarm_email.go
index 65046e9..db2b20a 100644
--- a/alarm/alarm_email.go
+++ b/alarm/alarm_email.go
@@ -7,12 +7,11 @@ import (
 
 const (
 	TK_NONE = iota
-  TK_TO
-  TK_SMTP
-  TK_FROM
+	TK_TO
+	TK_SMTP
+	TK_FROM
 )
 
-
 type AlarmEmail struct {
 	Type  string
 	Name  string
@@ -29,44 +28,44 @@ func NewAlarmEmail(name string) *AlarmEmail {
 }
 
 func (a *AlarmEmail) Alarm() error {
-	fmt.Printf("email alarm")
+	fmt.Printf("email alarm\n")
 	return nil
 }
 
 func (a *AlarmEmail) Parse(tk string) (bool, error) {
 	switch a.state {
-		case TK_NONE:
-			switch strings.ToLower(tk){
-				case "to":
-					a.state = TK_TO
-				case "from":
-					a.state = TK_FROM
-				case "smtp":
-					a.state = TK_SMTP
-				default:
-					if len(a.To) < 1 {
-						return false, fmt.Errorf("email alarm requires to address")
-					}
-					return false, nil
+	case TK_NONE:
+		switch strings.ToLower(tk) {
+		case "to":
+			a.state = TK_TO
+		case "from":
+			a.state = TK_FROM
+		case "smtp":
+			a.state = TK_SMTP
+		default:
+			if len(a.To) < 1 {
+				return false, fmt.Errorf("email alarm requires to address")
 			}
+			return false, nil
+		}
 
-		case TK_FROM:
-			a.From = tk
-			a.state = TK_NONE
+	case TK_FROM:
+		a.From = tk
+		a.state = TK_NONE
 
-		case TK_SMTP:
-			a.SMTP = tk
-			a.state = TK_NONE
+	case TK_SMTP:
+		a.SMTP = tk
+		a.state = TK_NONE
 
-		case TK_TO:
-			if strings.TrimSpace(tk) == "" {
-				return false, fmt.Errorf("to address cannot be empty")
-			}
-			a.To = append(a.To, tk)
-			a.state = TK_NONE
+	case TK_TO:
+		if strings.TrimSpace(tk) == "" {
+			return false, fmt.Errorf("to address cannot be empty")
+		}
+		a.To = append(a.To, tk)
+		a.state = TK_NONE
 
-		default:
-			return false, fmt.Errorf("invalid state in alarm_email")
+	default:
+		return false, fmt.Errorf("invalid state in alarm_email")
 	}
 	return true, nil
 }
diff --git a/main.go b/main.go
index ae4f051..0adfde2 100644
--- a/main.go
+++ b/main.go
@@ -49,10 +49,36 @@ func main() {
 
 		fmt.Printf("config is OK\n")
 
+	case "alarm":
+		an := flag.Arg(1)
+		if an == "" {
+			fmt.Fprintf(os.Stderr, "alarm name required\n")
+			os.Exit(1)
+		}
+
+		cfg, err := config.ReadConfig(*cfgPath, 0)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "%s\n", err.Error())
+			os.Exit(1)
+		}
+
+		al, exists := cfg.Alarms[an]
+		if !exists {
+			fmt.Fprintf(os.Stderr, "group or host is not defined\n")
+			os.Exit(1)
+		}
+
+		err = al.Alarm()
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "alarm failed: %s\n", err.Error())
+			os.Exit(1)
+		}
+		fmt.Printf("alarm sounded successfully\n")
+
 	case "check":
-		tn := flag.Arg(1)
-		if tn == "" {
-			fmt.Fprintf(os.Stderr, "check requires a host or group\n")
+		cn := flag.Arg(1)
+		if cn == "" {
+			fmt.Fprintf(os.Stderr, "check host or group name required\n")
 			os.Exit(1)
 		}
 
@@ -62,20 +88,20 @@ func main() {
 			os.Exit(1)
 		}
 
-		group, exists := cfg.Groups[tn]
+		gr, exists := cfg.Groups[cn]
 		if !exists {
 			fmt.Fprintf(os.Stderr, "group or host is not defined\n")
 			os.Exit(1)
 		}
 
-		err = group.Check(*debuglvl)
+		err = gr.Check(*debuglvl)
 		if err != nil {
 			fmt.Fprintf(os.Stderr, "check failed: %s\n", err.Error())
 			os.Exit(1)
 		}
 		fmt.Printf("check successful\n")
 
-	case "":
+	case "help", "":
 		printUsage()
 
 	default:
@@ -92,4 +118,5 @@ func printUsage() {
 	fmt.Printf("Actions:\n")
 	fmt.Printf("  verify configuration:     %s [args] config\n", os.Args[0])
 	fmt.Printf("  run a check manually:     %s [args] check <host/group>\n", os.Args[0])
+	fmt.Printf("  test an alarm:            %s [args] alarm <name>\n", os.Args[0])
 }