Cody Rose
ae793d7030
GitHub Realtime needs to be able to authenticate. This PR adds relevant fields to the relevant protobuf message, and exports/tweaks some stuff from the github source package to permit its (eventual) use by the GitHub Realtime implementation: The Connector interface and its constructors are now exported Connector.Clone now accepts additional clone arguments The github-specific connection construction has been moved into github.go The injected rate limit handler no longer accepts error reporters that it never uses I also added an API endpoint field to the GitHub Realtime proto message, because we need that too and I was here.
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, args...)
|
|
}
|