Add installation hook

goodfirstissue/handler.go

@@ -1,5 +1,7 @@
 package function
 
+// package main
+
 import (
 	"fmt"
 	"io/ioutil"
@@ -21,6 +23,13 @@ func init() {
 	twitterClient, twitterClientInitErr = twitter.NewClient()
 }
 
+// func main() {
+// 	http.HandleFunc("/", Handle)
+// 	if err := http.ListenAndServe(":8080", nil); err != nil {
+// 		panic(err)
+// 	}
+// }
+
 //Handle handles the function call
 func Handle(w http.ResponseWriter, r *http.Request) {
 	if twitterClient == nil {
@@ -52,6 +61,17 @@ func Handle(w http.ResponseWriter, r *http.Request) {
 	}
 	logrus.Tracef("%s", string(body))
 
+	//per https://developer.github.com/webhooks/#events integration_installation and integration_installation_repositories events are
+	// deprecated and replaced by installation and installation_repositories. But github is still sending integration_installation and
+	// integration_installation_repositories and go-github don't understand these.
+
+	switch t {
+	case "integration_installation":
+		t = "installation"
+	case "integration_installation_repositories":
+		t = "installation_repositories"
+	}
+
 	e, err := github.ParseWebHook(t, body)
 	if err != nil {
 		logrus.Error("failed to parsepayload. error: ", err.Error())
@@ -60,31 +80,47 @@ func Handle(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	msg := ""
-	if o, ok := e.(*github.IssuesEvent); ok && goodFirstIssue(o.Issue.Labels) {
-		switch {
-		case *o.Action == "opened":
-			msg = "a new #goodfirstissue opened"
-		case *o.Action == "reopened":
-			msg = "a #goodfirstissue got reopened"
-		case *o.Action == "labeled":
-			msg = "an issue just got labeled #goodfirstissue"
-		case *o.Action == "unassigned":
-			msg = "an issue just got available for assignment #goodfirstissue"
-		}
+	var msgs []string
 
-		if msg != "" {
-			msg += fmt.Sprintf(" for %s.\n#%s\n%s", *o.Repo.FullName, *o.Repo.Language, *o.Issue.HTMLURL)
-			twitterClient.Tweet(msg)
-		}
+	switch o := e.(type) {
+	case *github.IssuesEvent:
+		msgs = handleIssuesEvent(o)
+	case *github.InstallationEvent:
+		msgs = handleInstallationEvent(o)
+	case *github.InstallationRepositoriesEvent:
+		msgs = handleInstallationRepositoriesEvent(o)
 	}
 
-	if msg == "" {
-		//comment 1
-		msg = "OK"
+	for _, msg := range msgs {
+		twitterClient.Tweet(msg)
 	}
+
 	w.WriteHeader(http.StatusOK)
-	w.Write([]byte(msg))
+	w.Write([]byte("OK"))
+}
+
+func handleIssuesEvent(o *github.IssuesEvent) []string {
+	if !goodFirstIssue(o.Issue.Labels) {
+		return nil
+	}
+
+	msg := ""
+	switch {
+	case *o.Action == "opened":
+		msg = "a new #goodfirstissue opened"
+	case *o.Action == "reopened":
+		msg = "a #goodfirstissue got reopened"
+	case *o.Action == "labeled":
+		msg = "an issue just got labeled #goodfirstissue"
+	case *o.Action == "unassigned":
+		msg = "an issue just got available for assignment #goodfirstissue"
+	}
+
+	if msg != "" {
+		msg += fmt.Sprintf(" for %s.\n#%s\n%s", *o.Repo.FullName, *o.Repo.Language, *o.Issue.HTMLURL)
+	}
+
+	return []string{msg}
 }
 
 func goodFirstIssue(labels []github.Label) bool {
@@ -102,3 +138,40 @@ func goodFirstIssue(labels []github.Label) bool {
 
 	return false
 }
+
+func handleInstallationEvent(o *github.InstallationEvent) []string {
+	if stringValue(o.Action) != "created" {
+		return nil
+	}
+
+	msgs := []string{}
+
+	for _, r := range o.Repositories {
+		htmlURL := fmt.Sprintf("%s/%s", stringValue(o.Installation.Account.HTMLURL), stringValue(r.Name))
+		msgs = append(msgs, fmt.Sprintf("yay!! Lets welcome %s to #goodfirstissue", htmlURL))
+	}
+
+	return msgs
+}
+
+func handleInstallationRepositoriesEvent(o *github.InstallationRepositoriesEvent) []string {
+	if stringValue(o.Action) != "added" {
+		return nil
+	}
+
+	msgs := []string{}
+	for _, r := range o.RepositoriesAdded {
+		htmlURL := fmt.Sprintf("%s/%s", stringValue(o.Installation.Account.HTMLURL), stringValue(r.Name))
+		msgs = append(msgs, fmt.Sprintf("yay!! Lets welcome %s to #goodfirstissue", htmlURL))
+	}
+
+	return msgs
+}
+
+func stringValue(s *string) string {
+	if s == nil {
+		return ""
+	}
+
+	return *s
+}