added support for username:password

Signed-off-by: Richard Jones <richard@dagger.io>
This commit is contained in:
Richard Jones
2021-12-22 11:02:52 -07:00
parent a49d794409
commit 0295dc6340
3 changed files with 62 additions and 37 deletions

View File

@@ -3,8 +3,10 @@ package task
import (
"context"
"net/url"
"strings"
"github.com/moby/buildkit/client/llb"
"github.com/rs/zerolog/log"
"go.dagger.io/dagger/compiler"
"go.dagger.io/dagger/plancontext"
"go.dagger.io/dagger/solver"
@@ -18,53 +20,65 @@ type gitPullTask struct {
}
func (c gitPullTask) Run(ctx context.Context, pctx *plancontext.Context, s solver.Solver, v *compiler.Value) (*compiler.Value, error) {
remote, err := v.Lookup("remote").String()
if err != nil {
return nil, err
}
ref, err := v.Lookup("ref").String()
if err != nil {
return nil, err
var gitPull struct {
Remote string
Ref string
KeepGitDir bool
Username string
}
remoteRedacted := remote
if u, err := url.Parse(remote); err == nil {
remoteRedacted = u.Redacted()
if err := v.Decode(&gitPull); err != nil {
return nil, err
}
gitOpts := []llb.GitOption{}
var opts struct {
KeepGitDir bool
}
if err := v.Decode(&opts); err != nil {
return nil, err
}
lg := log.Ctx(ctx)
if opts.KeepGitDir {
if gitPull.KeepGitDir {
lg.Debug().Str("keepGitDir", "true").Msg("adding git option")
gitOpts = append(gitOpts, llb.KeepGitDir())
}
// Secret
if authToken := v.Lookup("authToken"); authToken.Exists() {
if gitPull.Username != "" {
pwd := v.Lookup("password")
pwdSecret, err := pctx.Secrets.FromValue(pwd)
if err != nil {
return nil, err
}
remote, err := url.Parse(gitPull.Remote)
if err != nil {
return nil, err
}
remote.User = url.UserPassword(gitPull.Username, strings.TrimSpace(pwdSecret.PlainText()))
gitPull.Remote = remote.String()
} else if authToken := v.Lookup("authToken"); plancontext.IsSecretValue(authToken) {
authTokenSecret, err := pctx.Secrets.FromValue(authToken)
if err != nil {
return nil, err
}
lg.Debug().Str("authToken", "***").Msg("adding git option")
gitOpts = append(gitOpts, llb.AuthTokenSecret(authTokenSecret.ID()))
}
if authHeader := v.Lookup("authHeader"); authHeader.Exists() {
} else if authHeader := v.Lookup("authHeader"); plancontext.IsSecretValue(authHeader) {
authHeaderSecret, err := pctx.Secrets.FromValue(authHeader)
if err != nil {
return nil, err
}
lg.Debug().Str("authHeader", "***").Msg("adding git option")
gitOpts = append(gitOpts, llb.AuthHeaderSecret(authHeaderSecret.ID()))
}
gitOpts = append(gitOpts, withCustomName(v, "FetchGit %s@%s", remoteRedacted, ref))
remoteRedacted := gitPull.Remote
if u, err := url.Parse(gitPull.Remote); err == nil {
remoteRedacted = u.Redacted()
}
st := llb.Git(remote, ref, gitOpts...)
gitOpts = append(gitOpts, withCustomName(v, "GitPull %s@%s", remoteRedacted, gitPull.Ref))
st := llb.Git(gitPull.Remote, gitPull.Ref, gitOpts...)
result, err := s.Solve(ctx, st, pctx.Platform.Get())
if err != nil {