Kashif Khan
92e9157267
* initial code changes * move args before clone command * add comments * added flag for github source to support backward compatibility * added flag for gitlab source to support backward compatibility * inverse the logic for enterprise * remove print statement * remove flag defaults * updated comments and removed CloneRepoUsingTokenInHeader function * false->true
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package github
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
gogit "github.com/go-git/go-git/v5"
|
|
"github.com/google/go-github/v67/github"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/common"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/context"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/pb/credentialspb"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/sources/git"
|
|
)
|
|
|
|
type basicAuthConnector struct {
|
|
apiClient *github.Client
|
|
username string
|
|
password string
|
|
}
|
|
|
|
var _ Connector = (*basicAuthConnector)(nil)
|
|
|
|
func NewBasicAuthConnector(apiEndpoint string, cred *credentialspb.BasicAuth) (Connector, error) {
|
|
const httpTimeoutSeconds = 60
|
|
httpClient := common.RetryableHTTPClientTimeout(int64(httpTimeoutSeconds))
|
|
httpClient.Transport = &github.BasicAuthTransport{
|
|
Username: cred.Username,
|
|
Password: cred.Password,
|
|
}
|
|
|
|
apiClient, err := createGitHubClient(httpClient, apiEndpoint)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not create API client: %w", err)
|
|
}
|
|
|
|
return &basicAuthConnector{
|
|
apiClient: apiClient,
|
|
username: cred.Username,
|
|
password: cred.Password,
|
|
}, nil
|
|
}
|
|
|
|
func (c *basicAuthConnector) APIClient() *github.Client {
|
|
return c.apiClient
|
|
}
|
|
|
|
func (c *basicAuthConnector) Clone(ctx context.Context, repoURL string, args ...string) (string, *gogit.Repository, error) {
|
|
return git.CloneRepoUsingToken(ctx, c.password, repoURL, c.username, true, args...)
|
|
}
|