From 4e21828caa75c42fcf84d9601ab18c713c99b3fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=9E=A8=E7=85=8A?= Date: Thu, 1 Oct 2020 10:23:35 +0800 Subject: [PATCH] init program --- .gitignore | 5 +++ common.go | 38 ++++++++++++++++ config.go | 37 +++++++++++++++ config.simple.xml | 112 ++++++++++++++++++++++++++++++++++++++++++++++ go.mod | 8 ++++ go.sum | 53 ++++++++++++++++++++++ hook.go | 54 ++++++++++++++++++++++ index.go | 13 ++++++ main.go | 21 +++++++++ 9 files changed, 341 insertions(+) create mode 100644 .gitignore create mode 100644 common.go create mode 100644 config.go create mode 100644 config.simple.xml create mode 100644 go.mod create mode 100644 go.sum create mode 100644 hook.go create mode 100644 index.go create mode 100644 main.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..4b2108a --- /dev/null +++ b/.gitignore @@ -0,0 +1,5 @@ +.* +nbproject +ftpsync.settings +tags +config.xml \ No newline at end of file diff --git a/common.go b/common.go new file mode 100644 index 0000000..7128cc5 --- /dev/null +++ b/common.go @@ -0,0 +1,38 @@ +package main + +import ( + "errors" + "github.com/sirupsen/logrus" + "os" + "os/exec" + "path/filepath" + "strings" +) + +func fechCurrentPath() string { + // fetch workdir from env + workdir := os.Getenv("WORKDIR") + if workdir != "" { + return workdir + "/" + } + + file, err := exec.LookPath(os.Args[0]) + if err != nil { + logrus.Panicln(err) + return "" + } + path, err := filepath.Abs(file) + if err != nil { + logrus.Panicln(err) + return "" + } + i := strings.LastIndex(path, "/") + if i < 0 { + i = strings.LastIndex(path, "\\") + } + if i < 0 { + logrus.Panicln(errors.New(`Can't find "/" or "\"`)) + return "" + } + return path[0 : i+1] +} diff --git a/config.go b/config.go new file mode 100644 index 0000000..6b60adb --- /dev/null +++ b/config.go @@ -0,0 +1,37 @@ +package main + +import ( + "encoding/xml" + "github.com/sirupsen/logrus" + "io/ioutil" +) + +type configXml struct { + Webhook struct { + Username string + Password string + } + Shell struct { + Cmd struct { + Name string + Args []string + } + } +} + +var Config = new(configXml) + +func loadConfig() { + // load xml file + configFile := fechCurrentPath() + "config.xml" + content, err := ioutil.ReadFile(configFile) + if err != nil { + logrus.Panic(err) + } + + // parse xml + err = xml.Unmarshal(content, Config) + if err != nil { + logrus.Panicln(err) + } +} diff --git a/config.simple.xml b/config.simple.xml new file mode 100644 index 0000000..a036795 --- /dev/null +++ b/config.simple.xml @@ -0,0 +1,112 @@ + + + + YourUserName + YourPassword + + + + + beep + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + -n + -f + 3000 + -l + 1000 + -n + -f + 2000 + -l + 1000 + + + \ No newline at end of file diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..e10b0ec --- /dev/null +++ b/go.mod @@ -0,0 +1,8 @@ +module github.com/flyhope/grafana-shell + +go 1.15 + +require ( + github.com/gin-gonic/gin v1.6.3 + github.com/sirupsen/logrus v1.7.0 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..3e88dac --- /dev/null +++ b/go.sum @@ -0,0 +1,53 @@ +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.6.3 h1:ahKqKTFpO5KTPHxWZjEdPScmYaGtLo8Y4DMHoEsnp14= +github.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwvtwp4M= +github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A= +github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= +github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q= +github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8= +github.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no= +github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA= +github.com/go-playground/validator/v10 v10.2.0 h1:KgJ0snyC2R9VXYN2rneOtQcw5aHQB1Vv0sFl1UcHBOY= +github.com/go-playground/validator/v10 v10.2.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GOhaH6EGOAJShg8Id5JGkI= +github.com/golang/protobuf v1.3.3 h1:gyjaxf+svBWX08ZjK86iN9geUJF0H6gp2IRKX6Nf6/I= +github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/json-iterator/go v1.1.9 h1:9yzud/Ht36ygwatGx56VwCZtlI/2AD15T1X2sjSuGns= +github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4= +github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y= +github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII= +github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY= +github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLDQ0W1YjYsBW+p8U2u7vzgW2SQVmlNazg= +github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/sirupsen/logrus v1.7.0 h1:ShrD1U9pZB12TX0cVy0DtePoCH97K8EtX+mg7ZARUtM= +github.com/sirupsen/logrus v1.7.0/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk= +github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= +github.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo= +github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw= +github.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs= +github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY= +golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 h1:YyJpGZS1sBuBCzLAR1VEpK193GlqGZbnPFnPV/5Rsb4= +golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200116001909-b77594299b42 h1:vEOn+mP2zCOVzKckCZy6YsCtDblrpj/w7B9nxGNELpg= +golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= diff --git a/hook.go b/hook.go new file mode 100644 index 0000000..a6a1767 --- /dev/null +++ b/hook.go @@ -0,0 +1,54 @@ +package main + +import ( + "github.com/gin-gonic/gin" + "github.com/sirupsen/logrus" + "os/exec" +) + +// Grafana web hook struct +type hookAlert struct { + EvalMatches []struct { + Metric string `json:"metric"` + Tags interface{} `json:"tags"` + Value float64 `json:"value"` + } `json:"evalMatches"` + ImageUrl string `json:"imageUrl"` + Message string `json:"message"` + RuleId int `json:"ruleId"` + RuleName string `json:"ruleName"` + RuleUrl string `json:"ruleUrl"` + State string `json:"state"` + Title string `json:"title"` +} + +// basicAuth +func basicAuth() gin.HandlerFunc { + return gin.BasicAuth(gin.Accounts{ + Config.Webhook.Username: Config.Webhook.Password, + }) +} + +// grafana webhook handler +func handlerWebhook(c *gin.Context) { + // parse json + data := new(hookAlert) + err := c.BindJSON(data) + if err != nil { + logrus.Warnln(err) + return + } + + // only except exec + if data.State != "ok" { + cmd := exec.Command(Config.Shell.Cmd.Name, Config.Shell.Cmd.Args...) + if err = cmd.Run(); err != nil { + logrus.Warnln(err) + return + } + c.JSON(200, false) + return + } + + c.JSON(200, false) +} diff --git a/index.go b/index.go new file mode 100644 index 0000000..dc53089 --- /dev/null +++ b/index.go @@ -0,0 +1,13 @@ +package main + +import ( + "github.com/gin-gonic/gin" +) + +// home index handler +func indexHandler(c *gin.Context) { + response := map[string]interface{}{ + "useage": "go to grafana add /webhook/ for webhook", + } + c.JSON(200, response) +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..6aa7163 --- /dev/null +++ b/main.go @@ -0,0 +1,21 @@ +package main + +import ( + "github.com/gin-gonic/gin" + "github.com/sirupsen/logrus" +) + +func main() { + loadConfig() + + router := gin.Default() + router.NoRoute(indexHandler) + + webhook := router.Group("/webhook", basicAuth()) + webhook.POST("/", handlerWebhook) + webhook.PUT("/", handlerWebhook) + err := router.Run(":8850") + if err != nil { + logrus.Warnln(err) + } +}