93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
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"
|
|
)
|
|
|
|
func init() {
|
|
Register("GitPull", func() Task { return &gitPullTask{} })
|
|
}
|
|
|
|
type gitPullTask struct {
|
|
}
|
|
|
|
func (c gitPullTask) Run(ctx context.Context, pctx *plancontext.Context, s solver.Solver, v *compiler.Value) (*compiler.Value, error) {
|
|
var gitPull struct {
|
|
Remote string
|
|
Ref string
|
|
KeepGitDir bool
|
|
Username string
|
|
}
|
|
|
|
if err := v.Decode(&gitPull); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
gitOpts := []llb.GitOption{}
|
|
|
|
lg := log.Ctx(ctx)
|
|
|
|
if gitPull.KeepGitDir {
|
|
lg.Debug().Str("keepGitDir", "true").Msg("adding git option")
|
|
gitOpts = append(gitOpts, llb.KeepGitDir())
|
|
}
|
|
|
|
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()))
|
|
} 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()))
|
|
}
|
|
|
|
remoteRedacted := gitPull.Remote
|
|
if u, err := url.Parse(gitPull.Remote); err == nil {
|
|
remoteRedacted = u.Redacted()
|
|
}
|
|
|
|
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 {
|
|
return nil, err
|
|
}
|
|
|
|
fs := pctx.FS.New(result)
|
|
return compiler.NewValue().FillFields(map[string]interface{}{
|
|
"output": fs.MarshalCUE(),
|
|
})
|
|
}
|