gitea-matrix-bot/listener.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"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.Printf("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) {

	fmt.Println(r.URL)

	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)

		// fmt.Printf("%+v\n", postData)
		// fmt.Println("=================================================")
		// spew.Dump(postData)
		// fmt.Println("=================================================")
		// fmt.Println(string(body))
		// fmt.Println("=================================================")
		message := generateMessage(postData, r.Header.Get("X-Gitea-Event"))

		args := strings.Split(r.URL.String(), "/")
		room := args[len(args)-1]
		fmt.Println("Posting to room: ")

		if mygiteabot.checkToken(room, postData.Secret) {
			mygiteabot.SendToRoom(room, message)
		} else {
			fmt.Println("Wrong token for room: " + room)
			fmt.Println("Secret: " + postData.Secret)
		}

		// fmt.Println("=================================================")
		// fmt.Fprint(w, "POST done")
	} 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

	// fmt.Println("======================")

	// fmt.Println(eventHeader)
	// fmt.Println(data.Action)

	// fmt.Println("======================")

	switch eventHeader {

	case "push":
		templ.Parse("{{.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("{{.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:

		fmt.Println("Unknown action: " + eventHeader + " " + data.Action)
		templ.Parse("Gitea did something unexpected, seriously wtf was that?! Event: " + eventHeader + " Action: " + data.Action)

	}

	if err := templ.Execute(&tpl, data); err != nil {
		panic(err)
	}

	return tpl.String()

	// tmplIssue, err := template.New("test").Parse("{{.User}} {{.Action}} Issue {{.IssueID}} in repository {{.Repo}}")

}