90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/noerw/osem_notify/core"
|
|
"github.com/noerw/osem_notify/utils"
|
|
)
|
|
|
|
var (
|
|
clearCache bool
|
|
)
|
|
|
|
func init() {
|
|
debugCmd.AddCommand(debugNotificationsCmd)
|
|
debugCacheCmd.PersistentFlags().BoolVarP(&clearCache, "clear", "", false, "reset the notifications cache")
|
|
debugCmd.AddCommand(debugCacheCmd)
|
|
rootCmd.AddCommand(debugCmd)
|
|
}
|
|
|
|
var debugCmd = &cobra.Command{
|
|
Use: "debug",
|
|
Short: "Run some debugging checks on osem_notify itself",
|
|
Long: "osem_notify debug <feature> tests the functionality of the given feature",
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
log.SetLevel(log.DebugLevel)
|
|
},
|
|
PersistentPostRun: func(cmd *cobra.Command, args []string) {
|
|
utils.PrintConfig()
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cmd.Help()
|
|
},
|
|
}
|
|
|
|
var debugCacheCmd = &cobra.Command{
|
|
Use: "cache",
|
|
Short: "Print or clear the notifications cache",
|
|
Long: "osem_notify debug cache prints the contents of the notifications cache",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if clearCache {
|
|
return core.ClearCache()
|
|
}
|
|
core.PrintCache()
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var debugNotificationsCmd = &cobra.Command{
|
|
Use: "notifications",
|
|
Short: "Verify that notifications are working",
|
|
Long: `osem_notify debug notifications sends a test notification according
|
|
to healthchecks.default.notifications.options as defined in the config file`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
defaultNotifyConf := &core.NotifyConfig{}
|
|
err := viper.UnmarshalKey("healthchecks.default", defaultNotifyConf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for transport, notifier := range core.Notifiers {
|
|
notLog := log.WithField("transport", transport)
|
|
opts := defaultNotifyConf.Notifications
|
|
notLog.Infof("testing notifer %s with options %v", transport, opts.Options)
|
|
n, err := notifier.New(opts)
|
|
if err != nil {
|
|
notLog.Warnf("could not initialize %s notifier: %s", transport, err)
|
|
continue
|
|
}
|
|
|
|
host, _ := os.Hostname()
|
|
err = n.Submit(core.Notification{
|
|
Subject: "Test notification from openSenseMap notifier",
|
|
Body: fmt.Sprintf("Your notification set up on %s is working fine!", host),
|
|
})
|
|
if err != nil {
|
|
notLog.Warnf("could not submit test notification for %s notifier: %s", transport, err)
|
|
continue
|
|
}
|
|
notLog.Info("Test notification (successfully?) submitted, check the specified inbox")
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|