mirror of
https://github.com/adnanh/webhook.git
synced 2025-05-12 00:24:45 +00:00
90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os/exec"
|
|
"time"
|
|
|
|
"github.com/adnanh/webhook/hooks"
|
|
|
|
"github.com/go-martini/martini"
|
|
|
|
l4g "code.google.com/p/log4go"
|
|
)
|
|
|
|
const (
|
|
version string = "1.0.0"
|
|
)
|
|
|
|
var (
|
|
webhooks *hooks.Hooks
|
|
appStart time.Time
|
|
ip = flag.String("ip", "", "ip the webhook server should listen on")
|
|
port = flag.Int("port", 9000, "port the webhook server should listen on")
|
|
hooksFilename = flag.String("hooks", "hooks.json", "path to the json file containing defined hooks the webhook should serve")
|
|
logFilename = flag.String("log", "webhook.log", "path to the log file")
|
|
)
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
|
|
fileLogWriter := l4g.NewFileLogWriter(*logFilename, true)
|
|
fileLogWriter.SetRotateDaily(true)
|
|
|
|
martini.Env = "production"
|
|
|
|
l4g.AddFilter("file", l4g.FINE, fileLogWriter)
|
|
}
|
|
|
|
func main() {
|
|
appStart = time.Now()
|
|
var e error
|
|
|
|
webhooks, e = hooks.New(*hooksFilename)
|
|
|
|
if e != nil {
|
|
l4g.Warn("Error occurred while loading hooks from %s: %s", *hooksFilename, e)
|
|
}
|
|
|
|
web := martini.Classic()
|
|
|
|
web.Get("/", rootHandler)
|
|
web.Get("/hook/:id", hookHandler)
|
|
web.Post("/hook/:id", hookHandler)
|
|
|
|
l4g.Info("Starting webhook %s with %d hook(s) on %s:%d", version, webhooks.Count(), *ip, *port)
|
|
|
|
web.RunOnAddr(fmt.Sprintf("%s:%d", *ip, *port))
|
|
}
|
|
|
|
func rootHandler() string {
|
|
return fmt.Sprintf("webhook %s running for %s serving %d hook(s)\n", version, time.Since(appStart).String(), webhooks.Count())
|
|
}
|
|
|
|
func hookHandler(req *http.Request, params martini.Params) string {
|
|
p := make(map[string]interface{})
|
|
|
|
if req.Header.Get("Content-Type") == "application/json" {
|
|
decoder := json.NewDecoder(req.Body)
|
|
decoder.UseNumber()
|
|
|
|
err := decoder.Decode(&p)
|
|
|
|
if err != nil {
|
|
l4g.Warn("Error occurred while trying to parse the payload as JSON")
|
|
}
|
|
}
|
|
|
|
go func(id string, params interface{}) {
|
|
if hook := webhooks.Match(id, params); hook != nil {
|
|
cmd := exec.Command(hook.Command, "", "", hook.Cwd)
|
|
out, _ := cmd.Output()
|
|
l4g.Info("Hook %s triggered successfully! Command output:\n%s", hook.ID, out)
|
|
}
|
|
}(params["id"], p)
|
|
|
|
return "Got it, thanks. :-)"
|
|
}
|