File: func_join.go

package info (click to toggle)
dasel 2.8.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 5,844 kB
  • sloc: sh: 53; python: 21; makefile: 21; xml: 20
file content (60 lines) | stat: -rw-r--r-- 1,330 bytes parent folder | download
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
55
56
57
58
59
60
package dasel

import (
	"github.com/tomwright/dasel/v2/util"
	"strings"
)

var JoinFunc = BasicFunction{
	name: "join",
	runFn: func(c *Context, s *Step, args []string) (Values, error) {
		if err := requireXOrMoreArgs("join", args, 1); err != nil {
			return nil, err
		}

		input := s.inputs()

		getValues := func(value Value, selector string) ([]string, error) {
			gotValues, err := c.subSelect(value, selector)
			if err != nil {
				return []string{}, err
			}

			res := make([]string, len(gotValues))
			for k, v := range gotValues {
				res[k] = util.ToString(v.Interface())
			}
			return res, nil
		}

		res := make(Values, 0)

		separator := args[0]
		args = args[1:]

		// No args - join all input values
		if len(args) == 0 {
			values := make([]string, len(input))
			for k, v := range input {
				values[k] = util.ToString(v.Interface())
			}
			res = append(res, ValueOf(strings.Join(values, separator)))
			return res, nil
		}

		// There are args - use each as a selector and join any resulting values.
		values := make([]string, 0)
		for _, val := range input {
			for _, cmp := range args {
				vals, err := getValues(val, cmp)
				if err != nil {
					return nil, err
				}
				values = append(values, vals...)
			}
		}
		res = append(res, ValueOf(strings.Join(values, separator)))

		return res, nil
	},
}