forked from maistra/istio-operator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
116 lines (103 loc) · 3.36 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
package main
import (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"os"
"path"
// Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
"github.com/operator-framework/operator-sdk/pkg/log/zap"
"github.com/spf13/pflag"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/util/sets"
_ "k8s.io/client-go/plugin/pkg/client/auth"
"k8s.io/helm/pkg/releaseutil"
logf "sigs.k8s.io/controller-runtime/pkg/runtime/log"
"sigs.k8s.io/yaml"
)
var log = logf.Log.WithName("cmd")
func main() {
// Add the zap logger flag set to the CLI. The flag set must
// be added before calling pflag.Parse().
pflag.CommandLine.AddFlagSet(zap.FlagSet())
// Add flags registered by imported packages (e.g. glog and
// controller-runtime)
pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
pflag.Parse()
logf.SetLogger(zap.Logger())
files := pflag.Args()
written := sets.NewString()
for _, file := range files {
log.Info(fmt.Sprintf("processing file %s", file))
processFile(file, written)
}
log.Info("completed")
}
func processFile(fileName string, written sets.String) {
if fileInfo, err := os.Stat(fileName); err == nil {
if fileInfo.IsDir() {
if file, err := os.Open(fileName); err == nil {
if files, err := file.Readdirnames(-1); err == nil {
for _, dirFile := range files {
processFile(path.Join(fileName, dirFile), written)
}
} else {
log.Error(err, fmt.Sprintf("error reading files in dir %s", fileName))
panic(err)
}
} else {
panic(err)
}
} else {
dirName := path.Dir(fileName)
rawYaml := readFile(fileName)
objects := releaseutil.SplitManifests(rawYaml)
for _, object := range objects {
if rawJSON, err := yaml.YAMLToJSON([]byte(object)); err == nil {
parsed := make(map[string]interface{})
if err := json.Unmarshal(rawJSON, &parsed); err != nil {
log.Error(err, fmt.Sprintf("%s contains invalid YAML: \"%s\"", fileName, object))
panic(err)
}
if kind, ok, _ := unstructured.NestedString(parsed, "kind"); ok && kind == "CustomResourceDefinition" {
if name, ok, err := unstructured.NestedString(parsed, "metadata", "name"); ok && !written.Has(name) {
written.Insert(name)
outFileName := path.Join(dirName, fmt.Sprintf("%s.crd.yaml", name))
if _, err := os.Stat(outFileName); err == nil {
panic(fmt.Errorf("output file exists: %s", outFileName))
} else if os.IsNotExist(err) {
log.Info(fmt.Sprintf("writing CRD %s to %s", name, outFileName))
if err := ioutil.WriteFile(outFileName, []byte(object), 0664); err != nil {
log.Error(err, fmt.Sprintf("error writing CRD to %s", outFileName))
panic(err)
}
} else {
panic(err)
}
} else if !ok {
log.Error(err, "error retrieving name from CRD")
panic(fmt.Errorf("CRD is missing name field: \n%s", parsed))
} else {
log.Info(fmt.Sprintf("skipping CRD %s", name))
}
} else {
log.Info(fmt.Sprintf("skipping resource \"%s\" in %s", kind, fileName))
}
} else {
log.Error(err, fmt.Sprintf("error converting raw yaml to json: %s", object))
}
}
}
} else {
panic(err)
}
}
func readFile(name string) string {
data, err := ioutil.ReadFile(name)
if err != nil {
log.Error(err, fmt.Sprintf("error reading file: %s", name))
panic(err)
}
return string(data)
}