Add Bitbucket logic.

2016-01-07 14:19:13 -05:00
parent b297674e2b
commit 4598745601
4 changed files with 220 additions and 58 deletions
@@ -1,18 +1,179 @@
 package bitbucket

-import "gopkg.in/go-playground/webhooks.v1"
+import (
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+
+	"gopkg.in/go-playground/webhooks.v1"
+)

 // Webhook instance contains all methods needed to process events
 type Webhook struct {
 	provider   webhooks.Provider
-	secret     string
+	uuid       string
 	eventFuncs map[Event]webhooks.ProcessPayloadFunc
 }

-// Config defines the configuration to create a new GitHubWebhook instance
+// Config defines the configuration to create a new Bitbucket Webhook instance
 type Config struct {
-	Secret string
+	UUID string
 }

-// Event defines a GitHub hook event type
+// Event defines a Bitbucket hook event type
 type Event string
+
+// Bitbucket hook types
+const (
+	RepoPushEvent                  Event = "repo:push"
+	RepoForkEvent                  Event = "repo:fork"
+	RepoCommitCommentCreatedEvent  Event = "repo:commit_comment_created"
+	RepoCommitStatusCreatedEvent   Event = "repo:commit_status_created"
+	RepoCommitStatusUpdatedEvent   Event = "repo:commit_status_updated"
+	IssueCreatedEvent              Event = "issue:created"
+	IssueUpdatedEvent              Event = "issue:updated"
+	IssueCommentCreatedEvent       Event = "issue:comment_created"
+	PullRequestCreatedEvent        Event = "pullrequest:created"
+	PullRequestUpdatedEvent        Event = "pullrequest:updated"
+	PullRequestApprovedEvent       Event = "pullrequest:approved"
+	PullRequestUnapprovedEvent     Event = "pullrequest:unapproved"
+	PullRequestMergedEvent         Event = "pullrequest:fulfilled"
+	PullRequestDeclinedEvent       Event = "pullrequest:rejected"
+	PullRequestCommentCreatedEvent Event = "pullrequest:comment_created"
+	PullRequestCommentUpdatedEvent Event = "pullrequest:comment_updated"
+	PullRequestCommentDeletedEvent Event = "pull_request:comment_deleted"
+)
+
+// New creates and returns a WebHook instance denoted by the Provider type
+func New(config *Config) *Webhook {
+	return &Webhook{
+		provider:   webhooks.GitHub,
+		uuid:       config.UUID,
+		eventFuncs: map[Event]webhooks.ProcessPayloadFunc{},
+	}
+}
+
+// Provider returns the current hooks provider ID
+func (hook Webhook) Provider() webhooks.Provider {
+	return hook.provider
+}
+
+// RegisterEvents registers the function to call when the specified event(s) are encountered
+func (hook Webhook) RegisterEvents(fn webhooks.ProcessPayloadFunc, events ...Event) {
+
+	for _, event := range events {
+		hook.eventFuncs[event] = fn
+	}
+}
+
+// ParsePayload parses and verifies the payload and fires off the mapped function, if it exists.
+func (hook Webhook) ParsePayload(w http.ResponseWriter, r *http.Request) {
+
+	uuid := r.Header.Get("X-Hook-UUID")
+	if uuid == "" {
+		http.Error(w, "400 Bad Request - Missing X-Hook-UUID Header", http.StatusBadRequest)
+		return
+	}
+
+	if uuid != hook.uuid {
+		http.Error(w, "403 Forbidden - Missing X-Hook-UUID does not match", http.StatusForbidden)
+		return
+	}
+
+	event := r.Header.Get("X-Event-Key")
+	if event == "" {
+		http.Error(w, "400 Bad Request - Missing X-Event-Key Header", http.StatusBadRequest)
+		return
+	}
+
+	bitbucketEvent := Event(event)
+
+	fn, ok := hook.eventFuncs[bitbucketEvent]
+	// if no event registered
+	if !ok {
+		return
+	}
+
+	payload, err := ioutil.ReadAll(r.Body)
+	if err != nil || len(payload) == 0 {
+		http.Error(w, "Error reading Body", http.StatusInternalServerError)
+		return
+	}
+
+	switch bitbucketEvent {
+	case RepoPushEvent:
+		var pl RepoPushPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case RepoForkEvent:
+		var pl RepoForkPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case RepoCommitCommentCreatedEvent:
+		var pl RepoCommitCommentCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case RepoCommitStatusCreatedEvent:
+		var pl RepoCommitStatusCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case RepoCommitStatusUpdatedEvent:
+		var pl RepoCommitStatusUpdatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case IssueCreatedEvent:
+		var pl IssueCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case IssueUpdatedEvent:
+		var pl IssueUpdatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case IssueCommentCreatedEvent:
+		var pl IssueCommentCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestCreatedEvent:
+		var pl PullRequestCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestUpdatedEvent:
+		var pl PullRequestUpdatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestApprovedEvent:
+		var pl PullRequestApprovedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestUnapprovedEvent:
+		var pl PullRequestUnapprovedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestMergedEvent:
+		var pl PullRequestMergedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestDeclinedEvent:
+		var pl PullRequestDeclinedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestCommentCreatedEvent:
+		var pl PullRequestCommentCreatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestCommentUpdatedEvent:
+		var pl PullRequestCommentUpdatedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	case PullRequestCommentDeletedEvent:
+		var pl PullRequestCommentDeletedPayload
+		json.Unmarshal([]byte(payload), &pl)
+		hook.runProcessPayloadFunc(fn, pl)
+	}
+}
+
+func (hook Webhook) runProcessPayloadFunc(fn webhooks.ProcessPayloadFunc, results interface{}) {
+	go func(fn webhooks.ProcessPayloadFunc, results interface{}) {
+		fn(results)
+	}(fn, results)
+}