mirror of
https://github.com/hanzoai/cd.git
synced 2026-10-30 14:56:45 +00:00
zeekayandHanzo Dev
cbf668a980
go fix applied the modernizers: promoted-field keys in struct literals,
strings.Cut, SplitSeq, new(expr), slices and maps helpers, wg.Go, reflect.TypeFor
and errors.AsType. omitzero is left out, since omitting a zero struct such as
time.Time{} is a per-field wire decision and not a mechanical one.
Two changes the modernizers could not make for themselves. AsType needs its type
argument to be an error, so the git exit-status reader embeds error in the
interface it asks for. And an error built by concatenating a tag list into its
own format string now says so with a verb, which is what vet was pointing at.
Co-authored-by: Hanzo Dev <[email protected]>
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package pull_request
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/hashicorp/go-retryablehttp"
|
|
gitlab "gitlab.com/gitlab-org/api/client-go"
|
|
|
|
"github.com/hanzoai/cd/applicationset/utils"
|
|
"github.com/hanzoai/cd/util/proxy"
|
|
)
|
|
|
|
type GitLabService struct {
|
|
client *gitlab.Client
|
|
project string
|
|
labels []string
|
|
pullRequestState string
|
|
}
|
|
|
|
var _ PullRequestService = (*GitLabService)(nil)
|
|
|
|
func NewGitLabService(token, url, project string, labels []string, pullRequestState string, scmRootCAPath string, insecure bool, caCerts []byte, proxyURL, noProxy string) (PullRequestService, error) {
|
|
var clientOptionFns []gitlab.ClientOptionFunc
|
|
|
|
// Set a custom Gitlab base URL if one is provided
|
|
if url != "" {
|
|
clientOptionFns = append(clientOptionFns, gitlab.WithBaseURL(url))
|
|
}
|
|
|
|
if token == "" {
|
|
token = os.Getenv("GITLAB_TOKEN")
|
|
}
|
|
|
|
tr := http.DefaultTransport.(*http.Transport).Clone()
|
|
tr.TLSClientConfig = utils.GetTlsConfig(scmRootCAPath, insecure, caCerts)
|
|
tr.Proxy = proxy.GetCallback(proxyURL, noProxy)
|
|
|
|
retryClient := retryablehttp.NewClient()
|
|
retryClient.HTTPClient.Transport = tr
|
|
|
|
clientOptionFns = append(clientOptionFns, gitlab.WithHTTPClient(retryClient.HTTPClient))
|
|
|
|
client, err := gitlab.NewClient(token, clientOptionFns...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating Gitlab client: %w", err)
|
|
}
|
|
|
|
return &GitLabService{
|
|
client: client,
|
|
project: project,
|
|
labels: labels,
|
|
pullRequestState: pullRequestState,
|
|
}, nil
|
|
}
|
|
|
|
func (g *GitLabService) List(ctx context.Context) ([]*PullRequest, error) {
|
|
// Filter the merge requests on labels, if they are specified.
|
|
var labels *gitlab.LabelOptions
|
|
if len(g.labels) > 0 {
|
|
var labelsList gitlab.LabelOptions = g.labels
|
|
labels = &labelsList
|
|
}
|
|
|
|
snippetsListOptions := gitlab.ExploreSnippetsOptions{
|
|
PerPage: 100,
|
|
}
|
|
opts := &gitlab.ListProjectMergeRequestsOptions{
|
|
ListOptions: snippetsListOptions.ListOptions,
|
|
Labels: labels,
|
|
}
|
|
|
|
if g.pullRequestState != "" {
|
|
opts.State = &g.pullRequestState
|
|
}
|
|
|
|
pullRequests := []*PullRequest{}
|
|
for {
|
|
mrs, resp, err := g.client.MergeRequests.ListProjectMergeRequests(g.project, opts, gitlab.WithContext(ctx))
|
|
if err != nil {
|
|
if resp != nil && resp.StatusCode == http.StatusNotFound {
|
|
// return a custom error indicating that the repository is not found,
|
|
// but also returning the empty result since the decision to continue or not in this case is made by the caller
|
|
return pullRequests, NewRepositoryNotFoundError(err)
|
|
}
|
|
return nil, fmt.Errorf("error listing merge requests for project '%s': %w", g.project, err)
|
|
}
|
|
for _, mr := range mrs {
|
|
pullRequests = append(pullRequests, &PullRequest{
|
|
Number: mr.IID,
|
|
Title: mr.Title,
|
|
Branch: mr.SourceBranch,
|
|
TargetBranch: mr.TargetBranch,
|
|
HeadSHA: mr.SHA,
|
|
Labels: mr.Labels,
|
|
Author: mr.Author.Username,
|
|
})
|
|
}
|
|
if resp.NextPage == 0 {
|
|
break
|
|
}
|
|
opts.Page = resp.NextPage
|
|
}
|
|
return pullRequests, nil
|
|
}
|