File: inpututil.go

package info (click to toggle)
golang-k8s-sigs-kustomize-kyaml 0.19.0%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 7,148 kB
  • sloc: makefile: 220; sh: 68
file content (65 lines) | stat: -rw-r--r-- 1,826 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
61
62
63
64
65
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0

package inpututil

import (
	"sigs.k8s.io/kustomize/kyaml/errors"
	"sigs.k8s.io/kustomize/kyaml/kio/kioutil"
	"sigs.k8s.io/kustomize/kyaml/yaml"
)

type MapInputsEFn func(*yaml.RNode, yaml.ResourceMeta) error

// MapInputsE runs the function against each input Resource, providing the parsed metadata
func MapInputsE(inputs []*yaml.RNode, fn MapInputsEFn) error {
	for i := range inputs {
		meta, err := inputs[i].GetMeta()
		if err != nil {
			return errors.Wrap(err)
		}
		if err := fn(inputs[i], meta); err != nil {
			return WrapErrorWithFile(err, meta)
		}
	}
	return nil
}

type MapInputsFn func(*yaml.RNode, yaml.ResourceMeta) ([]*yaml.RNode, error)

// MapInputs runs the function against each input Resource, providing the parsed metadata
// and returning the aggregated result
func MapInputs(inputs []*yaml.RNode, fn MapInputsFn) ([]*yaml.RNode, error) {
	var outputs []*yaml.RNode
	for i := range inputs {
		meta, err := inputs[i].GetMeta()
		if err != nil {
			return nil, errors.Wrap(err)
		}
		o, err := fn(inputs[i], meta)
		if err != nil {
			return nil, WrapErrorWithFile(err, meta)
		}
		outputs = append(outputs, o...)
	}
	return outputs, nil
}

// WrapErrorWithFile returns the original error wrapped with information about the file
// that the Resource was parsed from.
func WrapErrorWithFile(err error, meta yaml.ResourceMeta) error {
	if err == nil {
		return err
	}
	path := meta.Annotations[kioutil.PathAnnotation]
	index := meta.Annotations[kioutil.IndexAnnotation]
	if path == "" {
		path = meta.Annotations[kioutil.LegacyPathAnnotation]
	}
	if index == "" {
		index = meta.Annotations[kioutil.LegacyPathAnnotation]
	}
	return errors.WrapPrefixf(err, "%s [%s]",
		meta.Annotations[path],
		meta.Annotations[index])
}