package main import ( "bytes" "encoding/json" log "github.com/sirupsen/logrus" "io/ioutil" "net/http" "strconv" "strings" "text/template" // "github.com/davecgh/go-spew/spew" // "os" ) func setupListener() { httpURI := cfg.Section("http").Key("http_uri").String() httpPort := cfg.Section("http").Key("http_port").String() mux := http.NewServeMux() mux.HandleFunc(httpURI, PostHandler) log.Debugf("listening on port %s", httpPort) log.Fatal(http.ListenAndServe(":"+httpPort, mux)) } // PostHandler converts post request body to string func PostHandler(w http.ResponseWriter, r *http.Request) { if r.Method == "POST" { body, err := ioutil.ReadAll(r.Body) if err != nil { http.Error(w, "Error reading request body", http.StatusInternalServerError) } var postData GiteaPostData json.Unmarshal(body, &postData) message := generateMessage(postData, r.Header.Get("X-Gitea-Event")) args := strings.Split(r.URL.String(), "/") room := args[len(args)-1] log.Debugf("Posting to room: %s", room) if mygiteabot.checkToken(room, postData.Secret) { mygiteabot.SendToRoom(room, message) } else { log.Warningf("Wrong token %s for room: %s", postData.Secret, room) } } else { http.Error(w, "Invalid request method", http.StatusMethodNotAllowed) } } //generateMessage generates the message string for a given event func generateMessage(data GiteaPostData, eventHeader string) string { templ := template.New("notification") var tpl bytes.Buffer switch eventHeader { case "push": templ.Parse("[{{.Repository.Fullname}}] {{.Pusher.FullName}} pushed " + strconv.Itoa(len(data.Commits)) + " commit(s) to {{.Repository.Name}}") case "issues": switch data.Action { case "assigned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} assigned issue #{{.Issue.Number}} {{.Issue.Title}} to {{}} ") case "closed": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} closed issue #{{.Issue.Number}} {{.Issue.Title}}") case "demilestoned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} removed milestone TODO from issue #{{.Issue.Number}} {{.Issue.Title}}") case "edited": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} edited issue #{{.Issue.Number}} {{.Issue.Title}}") case "label_cleared": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} cleared labels from issue #{{.Issue.Number}} {{.Issue.Title}}") case "label_updated": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} updated labels of issue #{{.Issue.Number}} {{.Issue.Title}}") case "milestoned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} added issue #{{.Issue.Number}} {{.Issue.Title}} to milestone TODO") case "opened": templ.Parse("{{.Repository.FullName}}: {{.Issue.User.FullName}} opened issue #{{.Issue.Number}} {{.Issue.Title}}") case "reopened": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} re-opened issue #{{.Issue.Number}} {{.Issue.Title}}") case "synchronized": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} synchronized issue #{{.Issue.Number}} {{.Issue.Title}}") case "unassigned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} removed assignee from issue #{{.Issue.Number}} {{.Issue.Title}}") } case "fork": templ.Parse("[{{.Repository.FullName}}] {{.Sender.FullName}} forked repository {{.Repository.Parent.FullName}} to {{.Repository.FullName}}") case "pull_request": switch data.Action { case "assigned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} assigned pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\" to {{.PullRequest.Assignee.FullName}}") case "closed": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} closed pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "demilestoned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} removed milestone TODO from pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "edited": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} edited pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "label_cleared": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} removed labels from pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "label_updated": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} updated labels from pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "milestoned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} added pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\" to milestone TODO") case "opened": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} opened pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "reopened": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} re-opened pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "synchronized": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} synchronized pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "unassigned": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} removed assinee from pull-request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") } case "issue_comment": switch data.Action { case "created": templ.Parse("{{.Sender.FullName}} commented on #{{.Issue.Number}} {{.Issue.Title}}: {{.Comment.Body}}") case "deleted": templ.Parse("{{.Sender.FullName}} deleted commented on #{{.Issue.Number}} {{.Issue.Title}}") } case "repository": switch data.Action { case "created": templ.Parse("{{.Sender.FullName}} created repository {{}}") case "deleted": templ.Parse("{{.Sender.FullName}} deleted repository {{}}") } case "release": switch data.Action { case "published": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} published release {{}}") case "updated": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} updated release {{}}") case "deleted": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} deleted release {{}}") } case "pull_request_approved": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} approved pull request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "pull_request_rejected": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} rejected pull request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\"") case "pull_request_comment": templ.Parse("{{.Repository.FullName}}: {{.Sender.FullName}} commented on pull request #{{.PullRequest.Number}} \"{{.PullRequest.Title}}\": {{.Comment.Body}}") default: log.Warningf("Unknown action: %s for eventHeader %s", data.Action, eventHeader) templ.Parse("Gitea did something unexpected, seriously wtf was that?! Event: " + eventHeader + " Action: " + data.Action) } if err := templ.Execute(&tpl, data); err != nil { log.Fatal(err) } return tpl.String() }