Skip to content

Commit

Permalink
init program
Browse files Browse the repository at this point in the history
  • Loading branch information
@flyhope
flyhope committed Oct 1, 2020
1 parent dcaaed6 commit 4e21828
Show file tree
Hide file tree
Showing 9 changed files with 341 additions and 0 deletions.
5 changes: 5 additions & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
.*
nbproject
ftpsync.settings
tags
config.xml
38 changes: 38 additions & 0 deletions common.go
View file
Original file line number Diff line number Diff line change
@@ -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]
}
37 changes: 37 additions & 0 deletions config.go
View file
Original file line number Diff line number Diff line change
@@ -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)
}
}
112 changes: 112 additions & 0 deletions config.simple.xml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,112 @@
<?xml version="1.0" encoding="UTF-8"?>
<Config>
<Webhook>
<Username>YourUserName</Username>
<Password>YourPassword</Password>
</Webhook>

<Shell>
<Cmd>
<Name>beep</Name>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>3000</Args>
<Args>-l</Args>
<Args>1000</Args>
<Args>-n</Args>
<Args>-f</Args>
<Args>2000</Args>
<Args>-l</Args>
<Args>1000</Args>
</Cmd>
</Shell>
</Config>
8 changes: 8 additions & 0 deletions go.mod
View file
Original file line number Diff line number Diff line change
@@ -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
)
53 changes: 53 additions & 0 deletions go.sum
View file
Original file line number Diff line number Diff line change
@@ -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=
54 changes: 54 additions & 0 deletions hook.go
View file
Original file line number Diff line number Diff line change
@@ -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)
}
13 changes: 13 additions & 0 deletions index.go
View file
Original file line number Diff line number Diff line change
@@ -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)
}
21 changes: 21 additions & 0 deletions main.go
View file
Original file line number Diff line number Diff line change
@@ -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)
}
}

0 comments on commit 4e21828

Please sign in to comment.