From df0b442d5d38d67ed380c8daaf893c48548c6c96 Mon Sep 17 00:00:00 2001 From: Brad Rydzewski Date: Mon, 5 Oct 2015 23:17:59 -0700 Subject: [PATCH] just some initial work to enable bitbucket pull requests --- remote/bitbucket/bitbucket.go | 44 +++++++++++++++++++++++++++++++---- remote/bitbucket/helper.go | 31 ++++++++++++++++++++++++ 2 files changed, 71 insertions(+), 4 deletions(-) diff --git a/remote/bitbucket/bitbucket.go b/remote/bitbucket/bitbucket.go index 26107f3b..51502883 100644 --- a/remote/bitbucket/bitbucket.go +++ b/remote/bitbucket/bitbucket.go @@ -360,12 +360,17 @@ func (bb *Bitbucket) Deactivate(u *model.User, r *model.Repo, link string) error // and returns the required data in a standard format. func (bb *Bitbucket) Hook(r *http.Request) (*model.Repo, *model.Build, error) { - // only a subset of hooks are processed by drone - if r.Header.Get("X-Event-Key") != "repo:push" { - return nil, nil, nil + switch r.Header.Get("X-Event-Key") { + case "repo:push": + return bb.pushHook(r) + case "pullrequest:created", "pullrequest:updated": + return bb.pullHook(r) } - // extract the hook payload + return nil, nil, nil +} + +func (bb *Bitbucket) pushHook(r *http.Request) (*model.Repo, *model.Build, error) { payload := []byte(r.FormValue("payload")) if len(payload) == 0 { defer r.Body.Close() @@ -405,3 +410,34 @@ func (bb *Bitbucket) Hook(r *http.Request) (*model.Repo, *model.Build, error) { return nil, nil, nil } + +func (bb *Bitbucket) pullHook(r *http.Request) (*model.Repo, *model.Build, error) { + payload := []byte(r.FormValue("payload")) + if len(payload) == 0 { + defer r.Body.Close() + payload, _ = ioutil.ReadAll(r.Body) + } + + hook := PullRequestHook{} + err := json.Unmarshal(payload, &hook) + if err != nil { + return nil, nil, err + } + if hook.PullRequest.State != "OPEN" { + return nil, nil, nil + } + + return convertRepo(&hook.Repo), &model.Build{ + Event: model.EventPull, + Commit: hook.PullRequest.Dest.Commit.Hash, + Ref: fmt.Sprintf("refs/heads/%s", hook.PullRequest.Dest.Branch.Name), + Refspec: fmt.Sprintf("https://bitbucket.org/%s.git", hook.PullRequest.Source.Repo.FullName), + Remote: cloneLink(hook.PullRequest.Dest.Repo), + Link: hook.PullRequest.Links.Html.Href, + Branch: hook.PullRequest.Dest.Branch.Name, + Message: hook.PullRequest.Desc, + Avatar: hook.Actor.Links.Avatar.Href, + Author: hook.Actor.Login, + Timestamp: hook.PullRequest.Updated.UTC().Unix(), + }, nil +} diff --git a/remote/bitbucket/helper.go b/remote/bitbucket/helper.go index 17b7c00b..6d44b8bb 100644 --- a/remote/bitbucket/helper.go +++ b/remote/bitbucket/helper.go @@ -53,6 +53,37 @@ func convertRepo(from *Repo) *model.Repo { return &repo } +// cloneLink is a helper function that tries to extract the +// clone url from the repository object. +func cloneLink(repo Repo) string { + var clone string + + // above we manually constructed the repository clone url. + // below we will iterate through the list of clone links and + // attempt to instead use the clone url provided by bitbucket. + for _, link := range repo.Links.Clone { + if link.Name == "https" { + clone = link.Href + } + } + + // if no repository name is provided, we use the Html link. + // this excludes the .git suffix, but will still clone the repo. + if len(clone) == 0 { + clone = repo.Links.Html.Href + } + + // if bitbucket tries to automatically populate the user + // in the url we must strip it out. + cloneurl, err := url.Parse(clone) + if err == nil { + cloneurl.User = nil + clone = cloneurl.String() + } + + return clone +} + // convertRepoLite is a helper function used to convert a Bitbucket // repository structure to the simplified Drone repository structure. func convertRepoLite(from *Repo) *model.RepoLite {