/
main.go
156 lines (127 loc) · 3.95 KB
/
main.go
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
package main
import (
"fmt"
"io/ioutil"
"os"
"strings"
"github.com/pkg/errors"
"sigs.k8s.io/kustomize/kyaml/fn/framework"
"sigs.k8s.io/kustomize/kyaml/fn/framework/command"
"sigs.k8s.io/kustomize/kyaml/kio"
kyaml "sigs.k8s.io/kustomize/kyaml/yaml"
"istio.io/istio/operator/pkg/helm"
"istio.io/istio/operator/pkg/manifest"
"istio.io/istio/operator/pkg/name"
"istio.io/istio/operator/pkg/object"
"istio.io/pkg/log"
)
const (
istioOperatorKind = "IstioOperator"
istioOperatorApiVersion = "install.istio.io/v1alpha1"
)
type IstioOperator struct {
Value string `yaml:"value" json:"value"`
}
func init() {
logOptions := defaultLogOptions()
log.Configure(logOptions)
}
func main() {
fn := func(items []*kyaml.RNode) ([]*kyaml.RNode, error) {
result := []*kyaml.RNode{}
for _, item := range items {
if item.GetApiVersion() == istioOperatorApiVersion && item.GetKind() == istioOperatorKind {
manifests, err := generateIstioManifests(item)
if err != nil {
return nil, err
}
result = append(result, manifests...)
} else {
// Just append the non IstioOperator manifests
result = append(result, item)
}
}
return result, nil
}
p := framework.SimpleProcessor{Filter: kio.FilterFunc(fn)}
cmd := command.Build(p, command.StandaloneDisabled, false)
command.AddGenerateDockerfile(cmd)
if err := cmd.Execute(); err != nil {
fmt.Printf("Error while generating manifests: %v", err)
os.Exit(1)
}
}
// Copied from istioctl code
// https://github.com/istio/istio/blob/764e6688921e326bdaa530d5ef3a9ed5d87372f7/operator/cmd/mesh/manifest-generate.go#L159
// orderedManifests generates a list of manifests from the given map sorted by the default object order
// This allows
func orderedManifests(mm name.ManifestMap) ([]string, error) {
var rawOutput []string
var output []string
for _, mfs := range mm {
rawOutput = append(rawOutput, mfs...)
}
objects, err := object.ParseK8sObjectsFromYAMLManifest(strings.Join(rawOutput, helm.YAMLSeparator))
if err != nil {
return nil, err
}
// For a given group of objects, sort in order to avoid missing dependencies, such as creating CRDs first
objects.Sort(object.DefaultObjectOrder())
for _, obj := range objects {
yml, err := obj.YAML()
if err != nil {
return nil, err
}
output = append(output, string(yml))
}
return output, nil
}
func generateIstioManifests(item *kyaml.RNode) ([]*kyaml.RNode, error) {
result := []*kyaml.RNode{}
file, err := ioutil.TempFile("/tmp", "istio-operator.*.yaml")
if err != nil {
return nil, errors.Wrap(err, "failed to create temporary file")
}
defer os.Remove(file.Name())
itemYAML, err := item.String()
if err != nil {
return nil, errors.Wrap(err, "failed to parse item")
}
_, err = file.Write([]byte(itemYAML))
if err != nil {
return nil, errors.Wrap(err, "failed to write file")
}
if err = file.Close(); err != nil {
return nil, errors.Wrap(err, "failed to close file")
}
manifests, _, err := manifest.GenManifests([]string{file.Name()}, nil, false, nil, nil)
if err != nil {
return nil, errors.Wrap(err, "failed to render the manifests")
}
ordered, err := orderedManifests(manifests)
if err != nil {
return nil, errors.Wrap(err, "failed to order manifests")
}
for _, m := range ordered {
n, err := kyaml.Parse(m)
if err != nil {
return nil, errors.Wrap(err, "failed to parse manifest")
}
result = append(result, n)
}
return result, nil
}
func defaultLogOptions() *log.Options {
o := log.DefaultOptions()
// These scopes are, at the default "INFO" level, too chatty for command line use
o.SetOutputLevel("validation", log.FatalLevel)
o.SetOutputLevel("processing", log.FatalLevel)
o.SetOutputLevel("analysis", log.FatalLevel)
o.SetOutputLevel("installer", log.FatalLevel)
o.SetOutputLevel("translator", log.FatalLevel)
o.SetOutputLevel("adsc", log.FatalLevel)
o.SetOutputLevel("default", log.FatalLevel)
o.SetOutputLevel("klog", log.FatalLevel)
o.SetOutputLevel("kube", log.FatalLevel)
return o
}