84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package notifiers
|
|
|
|
import (
|
|
"github.com/statping/statping/types/failures"
|
|
"github.com/statping/statping/types/notifications"
|
|
"github.com/statping/statping/types/notifier"
|
|
"github.com/statping/statping/types/services"
|
|
"github.com/statping/statping/utils"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var _ notifier.Notifier = (*commandLine)(nil)
|
|
|
|
type commandLine struct {
|
|
*notifications.Notification
|
|
}
|
|
|
|
func (c *commandLine) Select() *notifications.Notification {
|
|
return c.Notification
|
|
}
|
|
|
|
var Command = &commandLine{¬ifications.Notification{
|
|
Method: "command",
|
|
Title: "Command",
|
|
Description: "Shell Command allows you to run a customized shell/bash Command on the local machine it's running on.",
|
|
Author: "Hunter Long",
|
|
AuthorUrl: "https://github.com/hunterlong",
|
|
Delay: time.Duration(1 * time.Second),
|
|
Icon: "fas fa-terminal",
|
|
Host: "/bin/bash",
|
|
Limits: 60,
|
|
Form: []notifications.NotificationForm{{
|
|
Type: "text",
|
|
Title: "Executable Path",
|
|
Placeholder: "/usr/bin/curl",
|
|
DbField: "host",
|
|
SmallText: "You can use '/bin/sh', '/bin/bash', '/usr/bin/curl' or an absolute path for an application.",
|
|
}, {
|
|
Type: "text",
|
|
Title: "Command to Run on OnSuccess",
|
|
Placeholder: "http://localhost:8080/health",
|
|
DbField: "var1",
|
|
SmallText: "<b>Accepts Variables</b> This Command will run when a service is receiving a Successful event.",
|
|
}, {
|
|
Type: "text",
|
|
Title: "Command to Run on OnFailure",
|
|
Placeholder: "http://localhost:8080/health",
|
|
DbField: "var2",
|
|
SmallText: "<b>Accepts Variables</b> This Command will run when a service is receiving a Failing event.",
|
|
}}},
|
|
}
|
|
|
|
func runCommand(app string, cmd ...string) (string, string, error) {
|
|
utils.Log.Infof("Command notifier sending: %s %s", app, strings.Join(cmd, " "))
|
|
outStr, errStr, err := utils.Command(app, cmd...)
|
|
return outStr, errStr, err
|
|
}
|
|
|
|
// OnSuccess for commandLine will trigger successful service
|
|
func (c *commandLine) OnSuccess(s *services.Service) error {
|
|
msg := c.GetValue("var1")
|
|
tmpl := ReplaceVars(msg, s, nil)
|
|
_, _, err := runCommand(c.Host, tmpl)
|
|
return err
|
|
}
|
|
|
|
// OnFailure for commandLine will trigger failing service
|
|
func (c *commandLine) OnFailure(s *services.Service, f *failures.Failure) error {
|
|
msg := c.GetValue("var2")
|
|
tmpl := ReplaceVars(msg, s, f)
|
|
_, _, err := runCommand(c.Host, tmpl)
|
|
return err
|
|
}
|
|
|
|
// OnTest for commandLine triggers when this notifier has been saved
|
|
func (c *commandLine) OnTest() (string, error) {
|
|
tmpl := ReplaceVars(c.Var1, exampleService, exampleFailure)
|
|
in, out, err := runCommand(c.Host, tmpl)
|
|
utils.Log.Infoln(in)
|
|
utils.Log.Infoln(out)
|
|
return out, err
|
|
}
|