1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
|
package dasel
import (
"fmt"
)
var FilterFunc = BasicFunction{
name: "filter",
runFn: func(c *Context, s *Step, args []string) (Values, error) {
if err := requireXOrMoreArgs("filter", args, 1); err != nil {
return nil, err
}
input := s.inputs()
runComparison := func(value Value, selector string) (bool, error) {
gotValues, err := c.subSelect(value, selector)
if err != nil {
return false, err
}
if len(gotValues) > 1 {
return false, fmt.Errorf("filter expects selector to return a single value")
}
if len(gotValues) == 0 {
return false, nil
}
return IsTruthy(gotValues[0]), nil
}
res := make(Values, 0)
for _, val := range input {
valPassed := true
for _, cmp := range args {
pass, err := runComparison(val, cmp)
if err != nil {
return nil, err
}
if !pass {
valPassed = false
break
}
}
if valPassed {
res = append(res, val)
}
}
return res, nil
},
}
|