7b4421b9a0
Signed-off-by: Sam Alba <sam.alba@gmail.com>
89 lines
1.6 KiB
Go
89 lines
1.6 KiB
Go
package environment
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cuelang.org/go/cue"
|
|
"github.com/rs/zerolog/log"
|
|
"go.dagger.io/dagger/compiler"
|
|
)
|
|
|
|
func isReference(val cue.Value) bool {
|
|
isRef := func(v cue.Value) bool {
|
|
_, ref := v.ReferencePath()
|
|
|
|
if ref.String() == "" || v.Path().String() == ref.String() {
|
|
// not a reference
|
|
return false
|
|
}
|
|
|
|
for _, s := range ref.Selectors() {
|
|
if s.IsDefinition() {
|
|
// if we reference to a definition, we skip the check
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
op, vals := val.Expr()
|
|
if op == cue.NoOp {
|
|
return isRef(val)
|
|
}
|
|
|
|
for _, v := range vals {
|
|
// if the expr has an op (& or |, etc...), check the expr values, recursively
|
|
if isReference(v) {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return isRef(val)
|
|
}
|
|
|
|
func ScanInputs(ctx context.Context, value *compiler.Value) []*compiler.Value {
|
|
lg := log.Ctx(ctx)
|
|
inputs := []*compiler.Value{}
|
|
|
|
value.Walk(
|
|
func(val *compiler.Value) bool {
|
|
if isReference(val.Cue()) {
|
|
lg.Debug().Str("value.Path", val.Path().String()).Msg("found reference, stop walk")
|
|
return false
|
|
}
|
|
|
|
if !val.HasAttr("input") {
|
|
return true
|
|
}
|
|
|
|
lg.Debug().Str("value.Path", val.Path().String()).Msg("found input")
|
|
inputs = append(inputs, val)
|
|
|
|
return true
|
|
}, nil,
|
|
)
|
|
|
|
return inputs
|
|
}
|
|
|
|
func ScanOutputs(ctx context.Context, value *compiler.Value) []*compiler.Value {
|
|
lg := log.Ctx(ctx)
|
|
inputs := []*compiler.Value{}
|
|
|
|
value.Walk(
|
|
func(val *compiler.Value) bool {
|
|
if !val.HasAttr("output") {
|
|
return true
|
|
}
|
|
|
|
lg.Debug().Str("value.Path", val.Path().String()).Msg("found output")
|
|
inputs = append(inputs, val)
|
|
|
|
return true
|
|
}, nil,
|
|
)
|
|
|
|
return inputs
|
|
}
|