sap-jenkins-library/pkg/github/github.go

package github

import (
	"context"
	"net/url"
	"strings"
	"time"

	piperhttp "github.com/SAP/jenkins-library/pkg/http"
	"github.com/google/go-github/v45/github"
	"github.com/pkg/errors"
	"golang.org/x/oauth2"
)

type githubCreateIssueService interface {
	Create(ctx context.Context, owner string, repo string, issue *github.IssueRequest) (*github.Issue, *github.Response, error)
}

type githubSearchIssuesService interface {
	Issues(ctx context.Context, query string, opts *github.SearchOptions) (*github.IssuesSearchResult, *github.Response, error)
}

type githubCreateCommentService interface {
	CreateComment(ctx context.Context, owner string, repo string, number int, comment *github.IssueComment) (*github.IssueComment, *github.Response, error)
}

type ClientBuilder struct {
	token        string // GitHub token, required
	baseURL      string // GitHub API URL, required
	uploadURL    string // Base URL for uploading files, optional
	timeout      time.Duration
	maxRetries   int
	trustedCerts []string // Trusted TLS certificates, optional
}

func NewClientBuilder(token, baseURL string) *ClientBuilder {
	if !strings.HasSuffix(baseURL, "/") {
		baseURL += "/"
	}

	return &ClientBuilder{
		token:        token,
		baseURL:      baseURL,
		uploadURL:    "",
		timeout:      0,
		maxRetries:   0,
		trustedCerts: nil,
	}
}

func (b *ClientBuilder) WithTrustedCerts(trustedCerts []string) *ClientBuilder {
	b.trustedCerts = trustedCerts
	return b
}

func (b *ClientBuilder) WithUploadURL(uploadURL string) *ClientBuilder {
	if !strings.HasSuffix(uploadURL, "/") {
		uploadURL += "/"
	}

	b.uploadURL = uploadURL
	return b
}

func (b *ClientBuilder) WithTimeout(timeout time.Duration) *ClientBuilder {
	b.timeout = timeout
	return b
}

func (b *ClientBuilder) WithMaxRetries(maxRetries int) *ClientBuilder {
	b.maxRetries = maxRetries
	return b
}

func (b *ClientBuilder) Build() (context.Context, *github.Client, error) {
	baseURL, err := url.Parse(b.baseURL)
	if err != nil {
		return nil, nil, errors.Wrap(err, "failed to parse baseURL")
	}

	uploadURL, err := url.Parse(b.uploadURL)
	if err != nil {
		return nil, nil, errors.Wrap(err, "failed to parse uploadURL")
	}

	if b.timeout == 0 {
		b.timeout = 30 * time.Second
	}

	if b.maxRetries == 0 {
		b.maxRetries = 5
	}

	piperHttp := piperhttp.Client{}
	piperHttp.SetOptions(piperhttp.ClientOptions{
		TrustedCerts:             b.trustedCerts,
		DoLogRequestBodyOnDebug:  true,
		DoLogResponseBodyOnDebug: true,
		TransportTimeout:         b.timeout,
		MaxRetries:               b.maxRetries,
	})
	ctx := context.WithValue(context.Background(), oauth2.HTTPClient, piperHttp.StandardClient())
	tokenSource := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: b.token, TokenType: "Bearer"})

	client := github.NewClient(oauth2.NewClient(ctx, tokenSource))
	client.BaseURL = baseURL
	client.UploadURL = uploadURL
	return ctx, client, nil
}