Init
This commit is contained in:
commit
23900f0350
|
|
@ -0,0 +1 @@
|
|||
config.conf
|
||||
|
|
@ -0,0 +1,59 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/genofire/golang-lib/log"
|
||||
"github.com/mattn/go-xmpp"
|
||||
"github.com/pierrre/githubhook"
|
||||
|
||||
configuration "github.com/genofire/hook2xmpp/config"
|
||||
"github.com/genofire/hook2xmpp/github"
|
||||
ownXMPP "github.com/genofire/hook2xmpp/xmpp"
|
||||
)
|
||||
|
||||
func main() {
|
||||
configFile := "config.conf"
|
||||
flag.StringVar(&configFile, "config", configFile, "path of configuration file")
|
||||
flag.Parse()
|
||||
|
||||
// load config
|
||||
config := configuration.ReadConfigFile(configFile)
|
||||
client, err := xmpp.NewClientNoTLS(config.XMPP.Host, config.XMPP.Username, config.XMPP.Password, config.XMPP.Debug)
|
||||
if err != nil {
|
||||
log.Log.Panic(err)
|
||||
}
|
||||
|
||||
log.Log.Infof("Started hock2xmpp with %s", client.JID())
|
||||
|
||||
client.SendHtml(xmpp.Chat{Remote: config.XMPP.StartupNotify, Type: "chat", Text: "Startup of hock2xmpp"})
|
||||
go ownXMPP.Start(client)
|
||||
|
||||
githubHandler := github.NewHandler(client, config.Hooks)
|
||||
handler := &githubhook.Handler{
|
||||
Delivery: githubHandler.Deliviery,
|
||||
}
|
||||
http.Handle("/github", handler)
|
||||
|
||||
srv := &http.Server{
|
||||
Addr: config.WebserverBind,
|
||||
}
|
||||
go func() {
|
||||
if err := srv.ListenAndServe(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Wait for system signal
|
||||
sigs := make(chan os.Signal, 1)
|
||||
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||
sig := <-sigs
|
||||
|
||||
srv.Close()
|
||||
|
||||
log.Log.Info("received", sig)
|
||||
}
|
||||
|
|
@ -0,0 +1,46 @@
|
|||
package config
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
|
||||
"github.com/BurntSushi/toml"
|
||||
|
||||
"github.com/genofire/golang-lib/log"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
WebserverBind string `toml:"webserver_bind"`
|
||||
|
||||
XMPP struct {
|
||||
Host string `toml:"host"`
|
||||
Username string `toml:"username"`
|
||||
Password string `toml:"password"`
|
||||
Debug bool `toml:"debug"`
|
||||
StartupNotify string `toml:"startup_notify"`
|
||||
} `toml:"xmpp"`
|
||||
|
||||
Hooks []Hook `toml:"hooks"`
|
||||
}
|
||||
|
||||
type Hook struct {
|
||||
NotifyUser []string `toml:"notify_user"`
|
||||
NotifyMuc []string `toml:"notify_muc"`
|
||||
|
||||
Type string `toml:"type"`
|
||||
Github struct {
|
||||
Project string `toml:"project"`
|
||||
} `toml:"github"`
|
||||
}
|
||||
|
||||
func ReadConfigFile(path string) *Config {
|
||||
config := &Config{}
|
||||
file, err := ioutil.ReadFile(path)
|
||||
if err != nil {
|
||||
log.Log.Panic(err)
|
||||
}
|
||||
if err := toml.Unmarshal(file, config); err != nil {
|
||||
log.Log.Panic(err)
|
||||
}
|
||||
|
||||
return config
|
||||
}
|
||||
|
|
@ -0,0 +1,11 @@
|
|||
webserver_bind = ":8080"
|
||||
|
||||
[xmpp]
|
||||
host = "fireorbit.de"
|
||||
username = "bot@fireorbit.de"
|
||||
password = "example"
|
||||
startup_notify = "geno@fireorbit.de"
|
||||
|
||||
[[github.project]]
|
||||
repository = "yanic"
|
||||
notify_user = geno@fireorbit.de
|
||||
|
|
@ -0,0 +1,126 @@
|
|||
package github
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/genofire/golang-lib/log"
|
||||
"github.com/genofire/hook2xmpp/config"
|
||||
xmpp "github.com/mattn/go-xmpp"
|
||||
|
||||
ownXMPP "github.com/genofire/hook2xmpp/xmpp"
|
||||
)
|
||||
|
||||
type Handler struct {
|
||||
client *xmpp.Client
|
||||
hooks map[string]config.Hook
|
||||
}
|
||||
|
||||
func NewHandler(client *xmpp.Client, newHooks []config.Hook) *Handler {
|
||||
hooks := make(map[string]config.Hook)
|
||||
|
||||
for _, hook := range newHooks {
|
||||
if hook.Type == "github" {
|
||||
hooks[hook.Github.Project] = hook
|
||||
}
|
||||
}
|
||||
return &Handler{
|
||||
client: client,
|
||||
hooks: hooks,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) Deliviery(event string, deliveryID string, payloadOrigin interface{}) {
|
||||
msg := PayloadToString(event, payloadOrigin)
|
||||
payload := payloadOrigin.(map[string]interface{})
|
||||
repository := payload["repository"].(map[string]interface{})
|
||||
repoName := repository["full_name"].(string)
|
||||
|
||||
hook, ok := h.hooks[repoName]
|
||||
if !ok {
|
||||
log.Log.Errorf("No hook found for: '%s'", repoName)
|
||||
return
|
||||
}
|
||||
|
||||
ownXMPP.Notify(h.client, hook, msg)
|
||||
}
|
||||
|
||||
var eventMsg = map[string]string{
|
||||
"commit_comment_created": "Commit comment",
|
||||
"status_error": "Commit status: error",
|
||||
"status_failure": "Commit status: failure",
|
||||
"status_pending": "Commit status: pending",
|
||||
"status_success": "Commit status: success",
|
||||
"create_branch": "Create branch",
|
||||
"create_tag": "Create tag",
|
||||
"delete_branch": "Delete branch",
|
||||
"delete_tag": "Delete tag",
|
||||
"issue_comment_created": "Issue comment",
|
||||
"issue_comment_deleted": "Issue comment: deleted",
|
||||
"issue_comment_edited": "Issue comment: edited",
|
||||
"issue_assigned": "Issue: assigned",
|
||||
"issue_closed": "Issue: closed",
|
||||
"issue_edited": "Issue: edited",
|
||||
"issue_labeled": "Issue: labeled",
|
||||
"issue_opened": "Issue: opened",
|
||||
"issue_reopened": "Issue: reopened",
|
||||
"issue_unassigned": "Issue: unassigned",
|
||||
"issue_unlabeled": "Issue: unlabeled",
|
||||
"pr_review_created": "Pull request review comment",
|
||||
"pr_review_deleted": "Pull request review comment: deleted",
|
||||
"pr_review_edited": "Pull request review comment: edited",
|
||||
"pr_assigned": "Pull request: assigned",
|
||||
"pr_closed": "Pull request: closed",
|
||||
"pr_edited": "Pull request: edited",
|
||||
"pr_labeled": "Pull request: labeled",
|
||||
"pr_opened": "Pull request: opened",
|
||||
"pr_reopened": "Pull request: reopened",
|
||||
"pr_synchronize": "Pull request: synchronize",
|
||||
"pr_unassigned": "Pull request: unassigned",
|
||||
"pr_unlabeled": "Pull request: unlabeled",
|
||||
"push": "Push",
|
||||
"release_published": "Release published",
|
||||
"member_added": "Repo: added collaborator",
|
||||
"team_add": "Repo: added to a team",
|
||||
"fork": "Repo: forked",
|
||||
"public": "Repo: made public",
|
||||
"watch_started": "Repo: starred",
|
||||
"gollum_created": "Wiki: created page",
|
||||
"gollum_edited": "Wiki: edited page",
|
||||
}
|
||||
|
||||
func PayloadToString(event string, payloadOrigin interface{}) string {
|
||||
payload := payloadOrigin.(map[string]interface{})
|
||||
|
||||
repository := payload["repository"].(map[string]interface{})
|
||||
repoName := repository["full_name"].(string)
|
||||
|
||||
msg := fmt.Sprintf("[%s]", repoName)
|
||||
|
||||
if event == "push" {
|
||||
pusher := payload["pusher"].(map[string]interface{})
|
||||
commits := payload["commits"].([]interface{})
|
||||
added := 0
|
||||
removed := 0
|
||||
modified := 0
|
||||
for _, commitOrigin := range commits {
|
||||
commit := commitOrigin.(map[string]interface{})
|
||||
added += len(commit["added"].([]interface{}))
|
||||
removed += len(commit["removed"].([]interface{}))
|
||||
modified += len(commit["modified"].([]interface{}))
|
||||
}
|
||||
msg = fmt.Sprintf("%s %s - pushed %d commit(s) to %s [+%d/-%d/\u00B1%d] \n %s", msg, pusher["name"], len(commits), strings.TrimLeft(payload["ref"].(string), "refs/heads/"), added, removed, modified, payload["compare"])
|
||||
} else if event == "issues" || event == "issue_comment" {
|
||||
sender := payload["sender"].(map[string]interface{})
|
||||
issue := payload["issue"].(map[string]interface{})
|
||||
msg = fmt.Sprintf("%s %s - %s action #%f: %s \n %s", msg, sender["login"], payload["action"], issue["number"], issue["title"], issue["html_url"])
|
||||
} else {
|
||||
sender := payload["sender"].(map[string]interface{})
|
||||
text := eventMsg[event]
|
||||
if text == "" {
|
||||
text = event
|
||||
}
|
||||
msg = fmt.Sprintf("%s %s - %s", msg, sender["login"], text)
|
||||
}
|
||||
return msg
|
||||
}
|
||||
|
|
@ -0,0 +1,36 @@
|
|||
package xmpp
|
||||
|
||||
import (
|
||||
"github.com/genofire/golang-lib/log"
|
||||
"github.com/genofire/hook2xmpp/config"
|
||||
|
||||
xmpp "github.com/mattn/go-xmpp"
|
||||
)
|
||||
|
||||
func Start(client *xmpp.Client) {
|
||||
for {
|
||||
m, err := client.Recv()
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
switch v := m.(type) {
|
||||
case xmpp.Chat:
|
||||
if v.Type == "chat" {
|
||||
log.Log.Infof("from %s: %s", v.Remote, v.Text)
|
||||
}
|
||||
if v.Type == "groupchat" {
|
||||
}
|
||||
case xmpp.Presence:
|
||||
// do nothing
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func Notify(client *xmpp.Client, hook config.Hook, msg string) {
|
||||
for _, muc := range hook.NotifyMuc {
|
||||
client.SendHtml(xmpp.Chat{Remote: muc, Type: "groupchat", Text: msg})
|
||||
}
|
||||
for _, user := range hook.NotifyUser {
|
||||
client.SendHtml(xmpp.Chat{Remote: user, Type: "chat", Text: msg})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue