added manual alarm triggering for testing

alarm/alarm_email.go

@@ -7,12 +7,11 @@ import (
 
 const (
 	TK_NONE = iota
-  TK_TO
+	TK_TO
-  TK_SMTP
+	TK_SMTP
-  TK_FROM
+	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:
+	case TK_NONE:
-			switch strings.ToLower(tk){
+		switch strings.ToLower(tk) {
-				case "to":
+		case "to":
-					a.state = TK_TO
+			a.state = TK_TO
-				case "from":
+		case "from":
-					a.state = TK_FROM
+			a.state = TK_FROM
-				case "smtp":
+		case "smtp":
-					a.state = TK_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_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
-
 		default:
-			return false, fmt.Errorf("invalid state in alarm_email")
+			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_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
+
+	default:
+		return false, fmt.Errorf("invalid state in alarm_email")
 	}
 	return true, nil
 }

main.go

@@ -49,10 +49,10 @@ func main() {
 
 		fmt.Printf("config is OK\n")
 
-	case "check":
+	case "alarm":
-		tn := flag.Arg(1)
+		an := flag.Arg(1)
-		if tn == "" {
+		if an == "" {
-			fmt.Fprintf(os.Stderr, "check requires a host or group\n")
+			fmt.Fprintf(os.Stderr, "alarm name required\n")
 			os.Exit(1)
 		}
 
@@ -62,20 +62,46 @@ func main() {
 			os.Exit(1)
 		}
 
-		group, exists := cfg.Groups[tn]
+		al, exists := cfg.Alarms[an]
 		if !exists {
 			fmt.Fprintf(os.Stderr, "group or host is not defined\n")
 			os.Exit(1)
 		}
 
-		err = group.Check(*debuglvl)
+		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":
+		cn := flag.Arg(1)
+		if cn == "" {
+			fmt.Fprintf(os.Stderr, "check host or group 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)
+		}
+
+		gr, exists := cfg.Groups[cn]
+		if !exists {
+			fmt.Fprintf(os.Stderr, "group or host is not defined\n")
+			os.Exit(1)
+		}
+
+		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])
 }