Amtool implementation

.gitignore

@@ -1,5 +1,6 @@
 /data/
 /alertmanager
+/amtool
 *.yml
 *.yaml
 /.build

.promu.yml

@@ -4,6 +4,8 @@ build:
     binaries:
         - name: alertmanager
           path: ./cmd/alertmanager
+        - name: amtool
+          path: ./cmd/amtool
     flags: -a -tags netgo
     ldflags: |
         -X {{repoPath}}/vendor/github.com/prometheus/common/version.Version={{.Version}}

artifacts/README.md

@@ -0,0 +1,18 @@
+# Generating amtool artifacts
+
+Amtool comes with the option to create a number of eaze-of-use artifacts that can be created.
+
+    go run generate_amtool_artifacts.go
+
+## Bash completion
+
+The bash completion file can be added to `/etc/bash_completion.d/`.
+
+## Man pages
+
+Man pages can be added to the man directory of your choice
+
+    cp artifacts/*.1 /usr/local/share/man/man1/
+    sudo mandb
+
+Then you should be able to view the man pages as expected.

artifacts/generate_amtool_artifacts.go

@@ -0,0 +1,17 @@
+package main
+
+import (
+	"github.com/spf13/cobra/doc"
+
+	"github.com/prometheus/alertmanager/cli"
+)
+
+func main() {
+	cli.RootCmd.GenBashCompletionFile("amtool_completion.sh")
+	header := &doc.GenManHeader{
+		Title:   "amtool",
+		Section: "1",
+	}
+
+	doc.GenManTree(cli.RootCmd, header, ".")
+}

cli/alert.go

@@ -0,0 +1,180 @@
+package cli
+
+import (
+	"encoding/json"
+	"errors"
+	"net/http"
+	"path"
+	"time"
+
+	"github.com/prometheus/alertmanager/cli/format"
+	"github.com/prometheus/alertmanager/types"
+	"github.com/prometheus/common/model"
+	"github.com/spf13/cobra"
+	flag "github.com/spf13/pflag"
+	"github.com/spf13/viper"
+)
+
+type alertmanagerAlertResponse struct {
+	Status    string       `json:"status"`
+	Data      model.Alerts `json:"data,omitempty"`
+	ErrorType string       `json:"errorType,omitempty"`
+	Error     string       `json:"error,omitempty"`
+}
+
+var alertFlags *flag.FlagSet
+
+// alertCmd represents the alert command
+var alertCmd = &cobra.Command{
+	Use:   "alert",
+	Short: "View and search through current alerts",
+	Long: `View and search through current alerts.
+
+  Amtool has a simplified prometheus query syntax, but contains robust support for
+  bash variable expansions. The non-option section of arguments constructs a list
+  of "Matcher Groups" that will be used to filter your query. The following
+  examples will attempt to show this behaviour in action:
+
+  amtool alert query alertname=foo node=bar
+
+  	This query will match all alerts with the alertname=foo and node=bar label
+  	value pairs set.
+
+  amtool alert query foo node=bar
+
+  	If alertname is ommited and the first argument does not contain a '=' or a
+  	'=~' then it will be assumed to be the value of the alertname pair.
+
+  amtool alert query 'alertname=~foo.*'
+
+  	As well as direct equality, regex matching is also supported. The '=~' syntax
+  	(similar to prometheus) is used to represent a regex match. Regex matching
+  	can be used in combination with a direct match.
+
+  amtool alert query alertname=foo node={bar,baz}
+
+  	This query will match all alerts with the alertname=foo label value pair
+  	and EITHER node=bar or node=baz.
+
+  amtool alert query alertname=foo{a,b} node={bar,baz}
+
+	Similar to the previous example this query will match all alerts with any
+	combination of alertname=fooa or alertname=foob AND node=bar or node=baz.
+
+	`,
+	RunE: queryAlerts,
+}
+
+func init() {
+	RootCmd.AddCommand(alertCmd)
+	alertCmd.Flags().Bool("expired", false, "Show expired alerts as well as active")
+	alertCmd.Flags().BoolP("silenced", "s", false, "Show silenced alerts")
+	alertFlags = alertCmd.Flags()
+}
+
+func fetchAlerts() (model.Alerts, error) {
+	alertResponse := alertmanagerAlertResponse{}
+
+	u, err := GetAlertmanagerURL()
+	if err != nil {
+		return model.Alerts{}, err
+	}
+
+	u.Path = path.Join(u.Path, "/api/v1/alerts")
+	res, err := http.Get(u.String())
+	if err != nil {
+		return model.Alerts{}, err
+	}
+
+	defer res.Body.Close()
+	decoder := json.NewDecoder(res.Body)
+
+	err = decoder.Decode(&alertResponse)
+	if err != nil {
+		return model.Alerts{}, errors.New("Unable to decode json response")
+	}
+	return alertResponse.Data, nil
+}
+
+func queryAlerts(cmd *cobra.Command, args []string) error {
+	alerts, err := fetchAlerts()
+	if err != nil {
+		return err
+	}
+
+	silences, err := fetchSilences()
+	if err != nil {
+		return err
+	}
+
+	var groups []types.Matchers
+	if len(args) > 0 {
+		matchers, err := parseMatchers(args, RESOLVE_FUZZY)
+		if err != nil {
+			return err
+		}
+		groups = parseMatcherGroups(matchers)
+		if err != nil {
+			return err
+		}
+	}
+
+	expired, err := alertFlags.GetBool("expired")
+	if err != nil {
+		return err
+	}
+
+	showSilenced, err := alertFlags.GetBool("silenced")
+	if err != nil {
+		return err
+	}
+
+	displayAlerts := model.Alerts{}
+	for _, alert := range alerts {
+		// If we are only returning current alerts and this one has already expired skip it
+		if !expired {
+			if !alert.EndsAt.IsZero() && alert.EndsAt.Before(time.Now()) {
+				continue
+			}
+		}
+
+		if !showSilenced {
+			// If any silence mutes this alert don't show it
+			silenced := false
+			for _, silence := range silences {
+				// Need to call Init before Mutes
+				err = silence.Init()
+				if err != nil {
+					return err
+				}
+
+				if silence.Mutes(alert.Labels) {
+					silenced = true
+					break
+				}
+			}
+			if silenced {
+				continue
+			}
+		}
+
+		// If the user hasn't specified and match groups then let it through
+		if len(groups) < 1 {
+			displayAlerts = append(displayAlerts, alert)
+			continue
+		}
+
+		for _, matchers := range groups {
+			if matchers.Match(alert.Labels) {
+				displayAlerts = append(displayAlerts, alert)
+				break
+			}
+		}
+	}
+
+	formatter, found := format.Formatters[viper.GetString("output")]
+	if !found {
+		return errors.New("Unknown output formatter")
+	}
+	return formatter.FormatAlerts(displayAlerts)
+}

cli/config.go

@@ -0,0 +1,86 @@
+package cli
+
+import (
+	"encoding/json"
+	"errors"
+	"net/http"
+	"path"
+	"time"
+
+	"github.com/prometheus/alertmanager/cli/format"
+	"github.com/prometheus/alertmanager/config"
+	"github.com/spf13/cobra"
+	//flag "github.com/spf13/pflag"
+	"github.com/spf13/viper"
+)
+
+// Config is the response type of alertmanager config endpoint
+// Duped in cli/format needs to be moved to common/model
+type Config struct {
+	Config      string            `json:"config"`
+	ConfigJSON  config.Config     `json:configJSON`
+	VersionINFO map[string]string `json:"versionInfo"`
+	Uptime      time.Time         `json:"uptime"`
+}
+
+type alertmanagerStatusResponse struct {
+	Status    string `json:"status"`
+	Data      Config `json:"data,omitempty"`
+	ErrorType string `json:"errorType,omitempty"`
+	Error     string `json:"error,omitempty"`
+}
+
+// alertCmd represents the alert command
+var configCmd = &cobra.Command{
+	Use:   "config",
+	Short: "View the running config",
+	Long: `View current config
+
+The amount of output is controlled by the output selection flag:
+	- Simple: Print just the running config
+	- Extended: Print the running config as well as uptime and all version info
+	- Json: Print entire config object as json`,
+	RunE: queryConfig,
+}
+
+func init() {
+	RootCmd.AddCommand(configCmd)
+}
+
+func fetchConfig() (Config, error) {
+	configResponse := alertmanagerStatusResponse{}
+	u, err := GetAlertmanagerURL()
+	if err != nil {
+		return Config{}, err
+	}
+
+	u.Path = path.Join(u.Path, "/api/v1/status")
+	res, err := http.Get(u.String())
+	if err != nil {
+		return Config{}, err
+	}
+
+	defer res.Body.Close()
+	decoder := json.NewDecoder(res.Body)
+
+	err = decoder.Decode(&configResponse)
+	if err != nil {
+		return Config{}, err
+	}
+
+	return configResponse.Data, nil
+}
+
+func queryConfig(cmd *cobra.Command, args []string) error {
+	config, err := fetchConfig()
+	if err != nil {
+		return err
+	}
+
+	formatter, found := format.Formatters[viper.GetString("output")]
+	if !found {
+		return errors.New("Unknown output formatter")
+	}
+
+	return formatter.FormatConfig(format.Config(config))
+}

cli/format/format.go

@@ -0,0 +1,38 @@
+package format
+
+import (
+	"io"
+	"time"
+
+	"github.com/prometheus/alertmanager/config"
+	"github.com/prometheus/alertmanager/types"
+	"github.com/prometheus/common/model"
+	"github.com/spf13/viper"
+)
+
+const DefaultDateFormat = "2006-01-02 15:04:05 MST"
+
+// Config representation
+// Need to get this moved to the prometheus/common/model repo having is duplicated here is smelly
+type Config struct {
+	Config      string            `json:"config"`
+	ConfigJSON  config.Config     `json:configJSON`
+	VersionINFO map[string]string `json:"versionInfo"`
+	Uptime      time.Time         `json:"uptime"`
+}
+
+// Formatter needs to be implemented for each new output formatter
+type Formatter interface {
+	SetOutput(io.Writer)
+	FormatSilences([]types.Silence) error
+	FormatAlerts(model.Alerts) error
+	FormatConfig(Config) error
+}
+
+// Formatters is a map of cli argument name to formatter inferface object
+var Formatters map[string]Formatter = map[string]Formatter{}
+
+func FormatDate(input time.Time) string {
+	dateformat := viper.GetString("date.format")
+	return input.Format(dateformat)
+}