Nash
17f6c98119
* GitHub source logger clean up * applied pr comments * applied pr comments * applied pr comments * applied PR review comments
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package github
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
|
|
gogit "github.com/go-git/go-git/v5"
|
|
"github.com/google/go-github/v63/github"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/common"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/context"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/sources/git"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
type tokenConnector struct {
|
|
apiClient *github.Client
|
|
token string
|
|
isGitHubEnterprise bool
|
|
handleRateLimit func(context.Context, error) bool
|
|
user string
|
|
userMu sync.Mutex
|
|
}
|
|
|
|
var _ connector = (*tokenConnector)(nil)
|
|
|
|
func newTokenConnector(apiEndpoint string, token string, handleRateLimit func(context.Context, error) bool) (*tokenConnector, error) {
|
|
const httpTimeoutSeconds = 60
|
|
httpClient := common.RetryableHTTPClientTimeout(int64(httpTimeoutSeconds))
|
|
tokenSource := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: token})
|
|
httpClient.Transport = &oauth2.Transport{
|
|
Base: httpClient.Transport,
|
|
Source: oauth2.ReuseTokenSource(nil, tokenSource),
|
|
}
|
|
|
|
apiClient, err := createGitHubClient(httpClient, apiEndpoint)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not create API client: %w", err)
|
|
}
|
|
|
|
return &tokenConnector{
|
|
apiClient: apiClient,
|
|
token: token,
|
|
isGitHubEnterprise: !strings.EqualFold(apiEndpoint, cloudEndpoint),
|
|
handleRateLimit: handleRateLimit,
|
|
}, nil
|
|
}
|
|
|
|
func (c *tokenConnector) APIClient() *github.Client {
|
|
return c.apiClient
|
|
}
|
|
|
|
func (c *tokenConnector) Clone(ctx context.Context, repoURL string) (string, *gogit.Repository, error) {
|
|
if err := c.setUserIfUnset(ctx); err != nil {
|
|
return "", nil, err
|
|
}
|
|
return git.CloneRepoUsingToken(ctx, c.token, repoURL, c.user)
|
|
}
|
|
|
|
func (c *tokenConnector) IsGithubEnterprise() bool {
|
|
return c.isGitHubEnterprise
|
|
}
|
|
|
|
func (c *tokenConnector) getUser(ctx context.Context) (string, error) {
|
|
var (
|
|
user *github.User
|
|
err error
|
|
)
|
|
for {
|
|
user, _, err = c.apiClient.Users.Get(ctx, "")
|
|
if c.handleRateLimit(ctx, err) {
|
|
continue
|
|
}
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not get GitHub user: %w", err)
|
|
}
|
|
break
|
|
}
|
|
return user.GetLogin(), nil
|
|
}
|
|
|
|
func (c *tokenConnector) setUserIfUnset(ctx context.Context) error {
|
|
c.userMu.Lock()
|
|
defer c.userMu.Unlock()
|
|
|
|
if c.user != "" {
|
|
return nil
|
|
}
|
|
|
|
user, err := c.getUser(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.user = user
|
|
return nil
|
|
}
|