Merge pull request #1 from 0xma12k/dev

mindmarky · web-flow · commit 816e93d384e9 · 2021-07-22T15:22:47.000+07:00
Add graylog line notify gateway feature
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -0,0 +1,33 @@
+name: goreleaser
+
+on:
+  push:
+    tags:
+      - '*'
+
+permissions:
+  contents: write
+
+jobs:
+  goreleaser:
+    runs-on: ubuntu-latest
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+      -
+        name: Set up Go
+        uses: actions/setup-go@v2
+        with:
+          go-version: 1.16
+      -
+        name: Run GoReleaser
+        uses: goreleaser/goreleaser-action@v2
+        with:
+          distribution: goreleaser
+          version: latest
+          args: release --rm-dist
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,3 @@
+bin/*
+.vscode/*
+.debug/*
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,25 @@
+FROM golang:1.16 as builder
+
+WORKDIR /app
+
+COPY go.mod .
+
+COPY go.sum .
+
+RUN go mod download
+
+COPY . .
+
+RUN CGO_ENABLED=0 GOOS=linux go build -o app 
+
+FROM alpine:latest
+
+ENV TZ=Asia/Bangkok
+
+WORKDIR /app/
+
+COPY --from=builder /app/app .
+
+EXPOSE 3000
+
+ENTRYPOINT ["./app"]
diff --git a/config/config.yml b/config/config.yml
@@ -0,0 +1,10 @@
+---
+
+server_port: 3000
+log_level: debug
+
+default_template: default
+
+templates:
+  - name: default
+    path: templates/default.templ
diff --git a/controllers/webhookhandler.go b/controllers/webhookhandler.go
@@ -0,0 +1,43 @@
+package controllers
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/0xma12k/graylog-line-notify-gateway/internal/entity"
+	"github.com/0xma12k/graylog-line-notify-gateway/internal/templ"
+	"github.com/0xma12k/graylog-line-notify-gateway/notify/line"
+	log "github.com/sirupsen/logrus"
+)
+
+// Webhook ...
+func Webhook(w http.ResponseWriter, r *http.Request) {
+	params := r.URL.Query()
+	templName := params.Get("template")
+	lineToken := params.Get("token")
+
+	// Decode reqeust body to Graylog struct object
+	var graylog entity.GraylogJSON
+	if err := json.NewDecoder(r.Body).Decode(&graylog); err != nil {
+		log.Error(err)
+	}
+
+	// Execute template
+	msg, err := templ.ExecuteTemplate(templName, &graylog)
+	if err != nil {
+		log.Error(err)
+
+		defaultMsg, err := templ.DefaultExecute(&graylog)
+		if err != nil {
+			log.Fatal(err)
+		}
+		msg = defaultMsg
+	}
+
+	statusCode, bodyBytes := line.Notify(msg, lineToken)
+
+	w.WriteHeader(statusCode)
+	w.Write(bodyBytes)
+
+	defer r.Body.Close()
+}
diff --git a/go.mod b/go.mod
@@ -0,0 +1,9 @@
+module github.com/0xma12k/graylog-line-notify-gateway
+
+go 1.16
+
+require (
+	github.com/gorilla/mux v1.8.0
+	github.com/sirupsen/logrus v1.8.1
+	gopkg.in/yaml.v2 v2.4.0
+)
diff --git a/go.sum b/go.sum
@@ -0,0 +1,16 @@
+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/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=
+github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=
+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.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
+github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
+github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+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=
+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.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -0,0 +1,83 @@
+package config
+
+import (
+	"flag"
+	"os"
+	"path"
+	"runtime"
+	"strings"
+
+	"github.com/sirupsen/logrus"
+	"gopkg.in/yaml.v2"
+)
+
+// Config ...
+type Config struct {
+	Port            string `yaml:"server_port"`
+	LogLevel        string `yaml:"log_level"`
+	DefaultTemplate string `yaml:"default_template"`
+	Templates       []struct {
+		Name string `yaml:"name"`
+		Path string `yaml:"path"`
+	} `yaml:"templates"`
+}
+
+var config Config
+
+// Init ...
+func Init() {
+	logrus.SetFormatter(&logrus.TextFormatter{
+		DisableColors: true,
+		FullTimestamp: true,
+		CallerPrettyfier: func(f *runtime.Frame) (string, string) {
+			s := strings.Split(f.Function, ".")
+			funcname := s[len(s)-1]
+			_, filename := path.Split(f.File)
+			return funcname, filename
+		},
+	})
+	logrus.SetReportCaller(true)
+
+	logrus.Info("loading configuration from yaml file")
+	path := ""
+	flag.StringVar(&path, "config", "", "`file` to configuration line notify gateway")
+	flag.Parse()
+
+	if path != "" {
+		logrus.Info("open configuration file at " + path)
+		f, err := os.Open(path)
+		if err != nil {
+			logrus.Fatal(err)
+		}
+		defer f.Close()
+
+		decoder := yaml.NewDecoder(f)
+		err = decoder.Decode(&config)
+		if err != nil {
+			logrus.Fatal(err)
+		}
+	} else {
+		logrus.Info("not provide configuration file, server run on default port")
+		config.Port = "3000"
+	}
+	logLevel()
+}
+
+func logLevel() {
+	switch config.LogLevel {
+	case "info":
+		logrus.SetLevel(logrus.InfoLevel)
+	case "debug":
+		logrus.SetLevel(logrus.DebugLevel)
+	case "warnning":
+		logrus.SetLevel(logrus.WarnLevel)
+	default:
+		logrus.SetLevel(logrus.InfoLevel)
+	}
+	logrus.SetOutput(os.Stdout)
+}
+
+// Get ..
+func Get() Config {
+	return config
+}
diff --git a/internal/entity/graylog.go b/internal/entity/graylog.go
@@ -0,0 +1,40 @@
+package entity
+
+// GraylogJSON represent json object receive from Graylog notification, see reference at https://docs.graylog.org/en/3.3/pages/alerts.html
+type GraylogJSON struct {
+	EventDefinitionID          string `json:"event_definition_id"`
+	EventDefinitionType        string `json:"event_definition_type"`
+	EventDefinitionTitle       string `json:"event_definition_title"`
+	EventDefinitionDescription string `json:"event_definition_description"`
+	JobDefinitionID            string `json:"job_definition_id"`
+	JobTriggerID               string `json:"job_trigger_id"`
+	Event                      struct {
+		ID                  string                 `json:"id"`
+		EventDefinitionID   string                 `json:"event_definition_id"`
+		EventDefinitionType string                 `json:"event_definition_type"`
+		OriginContext       string                 `json:"origin_context"`
+		Timestamp           string                 `json:"timestamp"`
+		TimestampProcessing string                 `json:"timestamp_processing"`
+		TimerangeStart      string                 `json:"timerange_start"`
+		TimerangeEnd        string                 `json:"timerange_end"`
+		Streams             []string               `json:"streams"`
+		SourceStreams       []string               `json:"source_streams"`
+		Alert               bool                   `json:"alert"`
+		Message             string                 `json:"message"`
+		Source              string                 `json:"source"`
+		KeyTuple            []string               `json:"key_tuple"`
+		Key                 string                 `json:"key"`
+		Priority            int64                  `json:"priority"`
+		Fields              map[string]interface{} `json:"fields"`
+	} `json:"event"`
+	Backlog []struct {
+		ID        string                 `json:"id"`
+		Index     string                 `json:"index"`
+		Source    string                 `json:"source"`
+		Message   string                 `json:"message"`
+		Timestamp string                 `json:"timestamp"`
+		SteamIDs  []string               `json:"stream_ids"`
+		Fields    map[string]interface{} `json:"fields"`
+	} `json:"backlog"`
+	// Backlog []interface{} `json:"backlog"`
+}
diff --git a/internal/templ/templ.go b/internal/templ/templ.go
@@ -0,0 +1,57 @@
+package templ
+
+import (
+	"bytes"
+	"fmt"
+	"text/template"
+
+	"github.com/0xma12k/graylog-line-notify-gateway/internal/config"
+	"github.com/0xma12k/graylog-line-notify-gateway/internal/entity"
+	"github.com/sirupsen/logrus"
+)
+
+var defaultTempl string = "{{.EventDefinitionTitle}}"
+
+func DefaultExecute(graylog *entity.GraylogJSON) (string, error) {
+	var buf bytes.Buffer
+	logrus.Debug("using build-in template")
+	t := template.Must(template.New("default").Parse(defaultTempl))
+	if err := t.Execute(&buf, graylog); err != nil {
+		return "", err
+	}
+	return buf.String(), nil
+}
+
+func ExecuteTemplate(templName string, graylog *entity.GraylogJSON) (string, error) {
+
+	var buf bytes.Buffer
+
+	if templName == "" {
+		templName = config.Get().DefaultTemplate
+	}
+
+	path, err := findTemplatePath(templName)
+	if err != nil {
+		return "", err
+	}
+
+	tmpl, err := template.ParseFiles(path)
+	if err != nil {
+		return "", err
+	}
+	t := template.Must(tmpl, err)
+
+	if err := t.Execute(&buf, graylog); err != nil {
+		return "", err
+	}
+	return buf.String(), nil
+}
+
+func findTemplatePath(name string) (string, error) {
+	for _, templ := range config.Get().Templates {
+		if name == templ.Name {
+			return templ.Path, nil
+		}
+	}
+	return "", fmt.Errorf("template not found")
+}
diff --git a/main.go b/main.go
@@ -0,0 +1,14 @@
+package main
+
+import (
+	"net/http"
+
+	"github.com/0xma12k/graylog-line-notify-gateway/internal/config"
+	"github.com/sirupsen/logrus"
+)
+
+func main() {
+	config.Init()
+	logrus.Info("server is running on port " + config.Get().Port)
+	http.ListenAndServe(":"+config.Get().Port, Router())
+}
diff --git a/notify/line/line.go b/notify/line/line.go
@@ -0,0 +1,48 @@
+package line
+
+import (
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+
+	"github.com/sirupsen/logrus"
+)
+
+// Notify ...
+func Notify(message, token string) (int, []byte) {
+	endpoint := "https://notify-api.line.me/api/notify"
+	tokenBearer := "Bearer " + token
+	data := url.Values{}
+	data.Set("message", message)
+
+	client := &http.Client{}
+	r, err := http.NewRequest("POST", endpoint, strings.NewReader(data.Encode())) // URL-encoded payload
+	if err != nil {
+		logrus.Error(err)
+	}
+	r.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+	r.Header.Add("Authorization", tokenBearer)
+
+	resp, err := client.Do(r)
+	if err != nil {
+		logrus.Error(err)
+	}
+
+	bodyBytes, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		logrus.Error(err)
+	}
+
+	resMsg := string(bodyBytes)
+
+	logrus.WithFields(
+		logrus.Fields{
+			"status": resp.StatusCode,
+			"resp":   resMsg,
+		}).Debug("called service line notify service")
+
+	defer resp.Body.Close()
+
+	return resp.StatusCode, bodyBytes
+}
diff --git a/route.go b/route.go
diff --git a/templates/default.templ b/templates/default.templ
