package bitbucket import ( "fmt" "regexp" "net/url" "strings" "github.com/drone/drone/model" "github.com/drone/drone/remote/bitbucket/internal" "golang.org/x/oauth2" ) const ( statusPending = "INPROGRESS" statusSuccess = "SUCCESSFUL" statusFailure = "FAILED" ) const ( descPending = "this build is pending" descSuccess = "the build was successful" descFailure = "the build failed" descError = "oops, something went wrong" ) // convertStatus is a helper function used to convert a Drone status to a // Bitbucket commit status. func convertStatus(status string) string { switch status { case model.StatusPending, model.StatusRunning: return statusPending case model.StatusSuccess: return statusSuccess default: return statusFailure } } // convertDesc is a helper function used to convert a Drone status to a // Bitbucket status description. func convertDesc(status string) string { switch status { case model.StatusPending, model.StatusRunning: return descPending case model.StatusSuccess: return descSuccess case model.StatusFailure: return descFailure default: return descError } } // convertRepo is a helper function used to convert a Bitbucket repository // structure to the common Drone repository structure. func convertRepo(from *internal.Repo) *model.Repo { repo := model.Repo{ Clone: cloneLink(from), Owner: strings.Split(from.FullName, "/")[0], Name: strings.Split(from.FullName, "/")[1], FullName: from.FullName, Link: from.Links.Html.Href, IsPrivate: from.IsPrivate, Avatar: from.Owner.Links.Avatar.Href, Kind: from.Scm, Branch: "master", } if repo.Kind == model.RepoHg { repo.Branch = "default" } return &repo } // cloneLink is a helper function that tries to extract the clone url from the // repository object. func cloneLink(repo *internal.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 *internal.Repo) *model.RepoLite { return &model.RepoLite{ Owner: strings.Split(from.FullName, "/")[0], Name: strings.Split(from.FullName, "/")[1], FullName: from.FullName, Avatar: from.Owner.Links.Avatar.Href, } } // convertUser is a helper function used to convert a Bitbucket user account // structure to the Drone User structure. func convertUser(from *internal.Account, token *oauth2.Token) *model.User { return &model.User{ Login: from.Login, Token: token.AccessToken, Secret: token.RefreshToken, Expiry: token.Expiry.UTC().Unix(), Avatar: from.Links.Avatar.Href, } } // convertTeamList is a helper function used to convert a Bitbucket team list // structure to the Drone Team structure. func convertTeamList(from []*internal.Account) []*model.Team { var teams []*model.Team for _, team := range from { teams = append(teams, convertTeam(team)) } return teams } // convertTeam is a helper function used to convert a Bitbucket team account // structure to the Drone Team structure. func convertTeam(from *internal.Account) *model.Team { return &model.Team{ Login: from.Login, Avatar: from.Links.Avatar.Href, } } // convertPullHook is a helper function used to convert a Bitbucket pull request // hook to the Drone build struct holding commit information. func convertPullHook(from *internal.PullRequestHook) *model.Build { return &model.Build{ Event: model.EventPull, Commit: from.PullRequest.Dest.Commit.Hash, Ref: fmt.Sprintf("refs/heads/%s", from.PullRequest.Dest.Branch.Name), Refspec: fmt.Sprintf("%s:%s", from.PullRequest.Source.Branch.Name, from.PullRequest.Dest.Branch.Name, ), Remote: fmt.Sprintf("https://bitbucket.org/%s", from.PullRequest.Source.Repo.FullName), Link: from.PullRequest.Links.Html.Href, Branch: from.PullRequest.Dest.Branch.Name, Message: from.PullRequest.Desc, Avatar: from.Actor.Links.Avatar.Href, Author: from.Actor.Login, Timestamp: from.PullRequest.Updated.UTC().Unix(), } } // convertPushHook is a helper function used to convert a Bitbucket push // hook to the Drone build struct holding commit information. func convertPushHook(hook *internal.PushHook, change *internal.Change) *model.Build { build := &model.Build{ Commit: change.New.Target.Hash, Link: change.New.Target.Links.Html.Href, Branch: change.New.Name, Message: change.New.Target.Message, Avatar: hook.Actor.Links.Avatar.Href, Author: hook.Actor.Login, Timestamp: change.New.Target.Date.UTC().Unix(), } switch change.New.Type { case "tag", "annotated_tag", "bookmark": build.Event = model.EventTag build.Ref = fmt.Sprintf("refs/tags/%s", change.New.Name) default: build.Event = model.EventPush build.Ref = fmt.Sprintf("refs/heads/%s", change.New.Name) } if len(change.New.Target.Author.Raw) != 0 { build.Email = extractEmail(change.New.Target.Author.Raw) } return build } // regex for git author fields ("name ") var reGitMail = regexp.MustCompile("<(.*)>") // extracts the email from a git commit author string func extractEmail(gitauthor string) (author string) { matches := reGitMail.FindAllStringSubmatch(gitauthor,-1) if len(matches) == 1 { author = matches[0][1] } return }