49 lines
1012 B
Go
49 lines
1012 B
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("Merge", func() Task { return &mergeTask{} })
|
|
}
|
|
|
|
type mergeTask struct {
|
|
}
|
|
|
|
func (t mergeTask) Run(ctx context.Context, pctx *plancontext.Context, s solver.Solver, v *compiler.Value) (*compiler.Value, error) {
|
|
inputs, err := v.Lookup("inputs").List()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
inputStates := make([]llb.State, len(inputs))
|
|
for i, input := range inputs {
|
|
inputFS, err := pctx.FS.FromValue(input)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
inputState, err := inputFS.State()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
inputStates[i] = inputState
|
|
}
|
|
|
|
st := llb.Merge(inputStates)
|
|
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(),
|
|
})
|
|
}
|