Merge branch 'add-bitbucket-hooks' into v1

Author: joeybloggs

bitbucket/bitbucket.go

@@ -0,0 +1,179 @@
+package bitbucket
+
+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
+	uuid       string
+	eventFuncs map[Event]webhooks.ProcessPayloadFunc
+}
+
+// Config defines the configuration to create a new Bitbucket Webhook instance
+type Config struct {
+	UUID string
+}
+
+// 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"
+	PullRequestApprovalRemovedEvent 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.Bitbucket,
+		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 PullRequestApprovalRemovedEvent:
+		var pl PullRequestApprovalRemovedPayload
+		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)
+}