92 lines
1.7 KiB
Go
92 lines
1.7 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/moby/buildkit/client/llb"
|
|
"go.dagger.io/dagger/compiler"
|
|
"go.dagger.io/dagger/plancontext"
|
|
"go.dagger.io/dagger/solver"
|
|
)
|
|
|
|
func init() {
|
|
Register("Copy", func() Task { return ©Task{} })
|
|
}
|
|
|
|
type copyTask struct {
|
|
}
|
|
|
|
func (t *copyTask) Run(ctx context.Context, pctx *plancontext.Context, s *solver.Solver, v *compiler.Value) (*compiler.Value, error) {
|
|
var err error
|
|
|
|
input, err := pctx.FS.FromValue(v.Lookup("input"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
inputState, err := input.State()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
contents, err := pctx.FS.FromValue(v.Lookup("contents"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
contentsState, err := contents.State()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sourcePath, err := v.Lookup("source").String()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
destPath, err := v.Lookup("dest").String()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var filters struct {
|
|
Include []string
|
|
Exclude []string
|
|
}
|
|
|
|
if err := v.Decode(&filters); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// FIXME: allow more configurable llb options
|
|
// For now we define the following convenience presets.
|
|
opts := &llb.CopyInfo{
|
|
CopyDirContentsOnly: true,
|
|
CreateDestPath: true,
|
|
AllowWildcard: true,
|
|
IncludePatterns: filters.Include,
|
|
ExcludePatterns: filters.Exclude,
|
|
}
|
|
|
|
outputState := inputState.File(
|
|
llb.Copy(
|
|
contentsState,
|
|
sourcePath,
|
|
destPath,
|
|
opts,
|
|
),
|
|
withCustomName(v, "Copy %s %s", sourcePath, destPath),
|
|
)
|
|
|
|
result, err := s.Solve(ctx, outputState, pctx.Platform.Get())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fs := pctx.FS.New(result)
|
|
|
|
return compiler.NewValue().FillFields(map[string]interface{}{
|
|
"output": fs.MarshalCUE(),
|
|
})
|
|
}
|