60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os/exec"
|
||
|
|
"text/template"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Alert struct {
|
||
|
|
Name string
|
||
|
|
Command []string
|
||
|
|
CommandShell string `yaml:"command_shell"`
|
||
|
|
commandTemplate []template.Template
|
||
|
|
commandShellTemplate template.Template
|
||
|
|
}
|
||
|
|
|
||
|
|
func (alert Alert) IsValid() bool {
|
||
|
|
atLeastOneCommand := (alert.CommandShell != "" || alert.Command != nil)
|
||
|
|
atMostOneCommand := (alert.CommandShell == "" || alert.Command == nil)
|
||
|
|
return atLeastOneCommand && atMostOneCommand
|
||
|
|
}
|
||
|
|
|
||
|
|
func (alert *Alert) BuildTemplates() {
|
||
|
|
if alert.commandTemplate == nil && alert.Command != nil {
|
||
|
|
// build template
|
||
|
|
fmt.Println("Building template for command...")
|
||
|
|
} else if alert.commandShellTemplate == nil && alert.CommandShell != "" {
|
||
|
|
alert.commandShellTemplate = template.Must(
|
||
|
|
template.New(alert.Name).Parse(alert.CommandShell),
|
||
|
|
)
|
||
|
|
} else {
|
||
|
|
panic("No template?")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (alert Alert) Send(notice AlertNotice) {
|
||
|
|
var cmd *exec.Cmd
|
||
|
|
|
||
|
|
if alert.commandTemplate != nil {
|
||
|
|
// build template
|
||
|
|
fmt.Println("Send command thing...")
|
||
|
|
} else if alert.commandShellTemplate != nil {
|
||
|
|
var commandBuffer bytes.Buffer
|
||
|
|
err := alert.commandShellTemplate.Execute(&commandBuffer, notice)
|
||
|
|
// TODO handle error
|
||
|
|
cmd = exec.Command(commandBuffer.String())
|
||
|
|
} else {
|
||
|
|
panic("No template?")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type AlertNotice struct {
|
||
|
|
MonitorName string
|
||
|
|
AlertCount int64
|
||
|
|
FailureCount int64
|
||
|
|
LastCheckOutput string
|
||
|
|
LastSuccess time.Time
|
||
|
|
}
|