diff --git a/src/github.com/matrix-org/go-neb/services/github/github_webhook.go b/src/github.com/matrix-org/go-neb/services/github/github_webhook.go
index ef7e196..9d4417b 100644
--- a/src/github.com/matrix-org/go-neb/services/github/github_webhook.go
+++ b/src/github.com/matrix-org/go-neb/services/github/github_webhook.go
@@ -36,7 +36,7 @@ const WebhookServiceType = "github-webhook"
// "!qmElAGdFYCHoCJuaNt:localhost": {
// Repos: {
// "matrix-org/go-neb": {
-// Events: ["push", "issues", "pull_request"]
+// Events: ["push", "issues", "pull_request", "labels"]
// }
// }
// }
@@ -57,10 +57,13 @@ type WebhookService struct {
// The webhook events to listen for. Currently supported:
// push : When users push to this repository.
// pull_request : When a pull request is made to this repository.
- // issues : When an issue is opened/closed.
+ // issues : When an issue is opened/edited/closed/reopened.
// issue_comment : When an issue or pull request is commented on.
// pull_request_review_comment : When a line comment is made on a pull request.
- // Full list: https://developer.github.com/webhooks/#events
+ // labels : When any issue or pull request is labeled/unlabeled. Unique to Go-NEB.
+ // milestones : When any issue or pull request is milestoned/demilestoned. Unique to Go-NEB.
+ // assign : When any issue or pull request is assigned/unassigned. Unique to Go-NEB.
+ // Most of these events are directly from: https://developer.github.com/webhooks/#events
Events []string
}
}
diff --git a/src/github.com/matrix-org/go-neb/services/github/webhook/webhook.go b/src/github.com/matrix-org/go-neb/services/github/webhook/webhook.go
index 8d1b625..d610edb 100644
--- a/src/github.com/matrix-org/go-neb/services/github/webhook/webhook.go
+++ b/src/github.com/matrix-org/go-neb/services/github/webhook/webhook.go
@@ -61,14 +61,15 @@ func OnReceiveRequest(r *http.Request, secretToken string) (string, *github.Repo
return "", nil, nil, &errors.HTTPError{nil, "pong", 200}
}
- htmlStr, repo, err := parseGithubEvent(eventType, content)
+ htmlStr, repo, refinedType, err := parseGithubEvent(eventType, content)
if err != nil {
log.WithError(err).Print("Failed to parse github event")
return "", nil, nil, &errors.HTTPError{nil, "Failed to parse github event", 500}
}
msg := matrix.GetHTMLMessage("m.notice", htmlStr)
- return eventType, repo, &msg, nil
+
+ return refinedType, repo, &msg, nil
}
// checkMAC reports whether messageMAC is a valid HMAC tag for message.
@@ -80,24 +81,26 @@ func checkMAC(message, messageMAC, key []byte) bool {
}
// parseGithubEvent parses a github event type and JSON data and returns an explanatory
-// HTML string and the github repository this event affects, or an error.
-func parseGithubEvent(eventType string, data []byte) (string, *github.Repository, error) {
+// HTML string, the github repository and the refined event type, or an error.
+func parseGithubEvent(eventType string, data []byte) (string, *github.Repository, string, error) {
if eventType == "pull_request" {
var ev github.PullRequestEvent
if err := json.Unmarshal(data, &ev); err != nil {
- return "", nil, err
+ return "", nil, eventType, err
}
- return pullRequestHTMLMessage(ev), ev.Repo, nil
+ refinedEventType := refineEventType(eventType, ev.Action)
+ return pullRequestHTMLMessage(ev), ev.Repo, refinedEventType, nil
} else if eventType == "issues" {
var ev github.IssuesEvent
if err := json.Unmarshal(data, &ev); err != nil {
- return "", nil, err
+ return "", nil, eventType, err
}
- return issueHTMLMessage(ev), ev.Repo, nil
+ refinedEventType := refineEventType(eventType, ev.Action)
+ return issueHTMLMessage(ev), ev.Repo, refinedEventType, nil
} else if eventType == "push" {
var ev github.PushEvent
if err := json.Unmarshal(data, &ev); err != nil {
- return "", nil, err
+ return "", nil, eventType, err
}
// The 'push' event repository format is subtly different from normal, so munge the bits we need.
@@ -109,21 +112,36 @@ func parseGithubEvent(eventType string, data []byte) (string, *github.Repository
Name: ev.Repo.Name,
FullName: &fullName,
}
- return pushHTMLMessage(ev), &repo, nil
+ return pushHTMLMessage(ev), &repo, eventType, nil
} else if eventType == "issue_comment" {
var ev github.IssueCommentEvent
if err := json.Unmarshal(data, &ev); err != nil {
- return "", nil, err
+ return "", nil, eventType, err
}
- return issueCommentHTMLMessage(ev), ev.Repo, nil
+ return issueCommentHTMLMessage(ev), ev.Repo, eventType, nil
} else if eventType == "pull_request_review_comment" {
var ev github.PullRequestReviewCommentEvent
if err := json.Unmarshal(data, &ev); err != nil {
- return "", nil, err
+ return "", nil, eventType, err
}
- return prReviewCommentHTMLMessage(ev), ev.Repo, nil
+ return prReviewCommentHTMLMessage(ev), ev.Repo, eventType, nil
+ }
+ return "", nil, eventType, fmt.Errorf("Unrecognized event type")
+}
+
+func refineEventType(eventType string, action *string) string {
+ if action == nil {
+ return eventType
+ }
+ a := *action
+ if a == "assigned" || a == "unassigned" {
+ return "assign"
+ } else if a == "milestoned" || a == "demilestoned" {
+ return "milestones"
+ } else if a == "labeled" || a == "unlabeled" {
+ return "labels"
}
- return "", nil, fmt.Errorf("Unrecognized event type")
+ return eventType
}
func pullRequestHTMLMessage(p github.PullRequestEvent) string {
diff --git a/src/github.com/matrix-org/go-neb/services/github/webhook/webhook_test.go b/src/github.com/matrix-org/go-neb/services/github/webhook/webhook_test.go
index 3a4b810..4994de6 100644
--- a/src/github.com/matrix-org/go-neb/services/github/webhook/webhook_test.go
+++ b/src/github.com/matrix-org/go-neb/services/github/webhook/webhook_test.go
@@ -10,6 +10,7 @@ var ghtests = []struct {
jsonBody string
outHTML string
outFullRepo string
+ outType string
}{
{"issues",
`{
@@ -165,7 +166,7 @@ var ghtests = []struct {
}
}`,
`[DummyAccount/reponame] DummyAccount closed issue #15: aaaaaa [closed] - https://github.com/DummyAccount/reponame/issues/15`,
- "DummyAccount/reponame"},
+ "DummyAccount/reponame", "issues"},
// ==================================================================
{
"issue_comment",
@@ -350,7 +351,7 @@ var ghtests = []struct {
}
}`,
"[DummyAccount/arepo] DummyAccount commented on DummyAccount's issue #15: aaaaaa - https://github.com/DummyAccount/arepo/issues/15",
- "DummyAccount/arepo",
+ "DummyAccount/arepo", "issue_comment",
},
// ==================================================================
{
@@ -561,7 +562,7 @@ var ghtests = []struct {
}
}`,
"[matrix-org/sytest] NegativeMjark pushed 2 commits to develop: https://github.com/matrix-org/sytest/commit/4a05c601f6b806110e63160cf7cf41b37787461f
NegativeMjark: Fix arguments to postgres connector to work with go
NegativeMjark: Add necessary info to the second postgres db",
- "matrix-org/sytest",
+ "matrix-org/sytest", "push",
},
// ==================================================================
{
@@ -1032,7 +1033,7 @@ var ghtests = []struct {
}
}`,
"[matrix-org/matrix-react-sdk] richvdh assigned pull request #303: Factor out common parts of room creation [open] to dbkr - https://github.com/matrix-org/matrix-react-sdk/pull/303",
- "matrix-org/matrix-react-sdk",
+ "matrix-org/matrix-react-sdk", "assign",
},
// ==================================================================
{
@@ -1500,25 +1501,28 @@ var ghtests = []struct {
}
}`,
"[matrix-org/synapse] erikjohnston made a line comment on negzi's pull request #860 (assignee: None): Fix a bug caused by a change in auth_handler function - https://github.com/matrix-org/synapse/pull/860#discussion_r66413356",
- "matrix-org/synapse",
+ "matrix-org/synapse", "pull_request_review_comment",
},
}
func TestParseGithubEvent(t *testing.T) {
for _, gh := range ghtests {
- outHTML, outRepo, outErr := parseGithubEvent(gh.eventType, []byte(gh.jsonBody))
+ outHTML, outRepo, outType, outErr := parseGithubEvent(gh.eventType, []byte(gh.jsonBody))
if outErr != nil {
t.Fatal(outErr)
}
if strings.TrimSpace(outHTML) != strings.TrimSpace(gh.outHTML) {
- t.Fatalf("ParseGithubEvent(%s) => HTML output does not match. Got:\n%s\n\nExpected:\n%s", gh.eventType,
+ t.Errorf("ParseGithubEvent(%s) => HTML output does not match. Got:\n%s\n\nExpected:\n%s", gh.eventType,
strings.TrimSpace(outHTML), strings.TrimSpace(gh.outHTML))
}
if outRepo == nil {
- t.Fatalf("ParseGithubEvent(%s) => Repo is nil", gh.eventType)
+ t.Errorf("ParseGithubEvent(%s) => Repo is nil", gh.eventType)
}
if *outRepo.FullName != gh.outFullRepo {
- t.Fatalf("ParseGithubEvent(%s) => Repo: Want %s got %s", gh.eventType, gh.outFullRepo, *outRepo.FullName)
+ t.Errorf("ParseGithubEvent(%s) => Repo: Want %s got %s", gh.eventType, gh.outFullRepo, *outRepo.FullName)
+ }
+ if outType != gh.outType {
+ t.Errorf("ParseGithubEvent(%s) => Event type: Want %s got %s", gh.eventType, gh.outType, outType)
}
}
}