-
Notifications
You must be signed in to change notification settings - Fork 2
/
generate_presets.go
113 lines (93 loc) · 2.3 KB
/
generate_presets.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
package main
import (
"fmt"
"go/format"
"io/ioutil"
"log"
"os"
"path/filepath"
"regexp"
"sort"
"strings"
)
const chartsDir = "charts"
const outputFile = "pkg/generated/presets.go"
func main() {
filesMap := make(map[string]string)
var chartNames []string
err := filepath.Walk(chartsDir, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.Mode()&os.ModeSymlink != 0 {
return filepath.SkipDir
}
if info.IsDir() && path != chartsDir && filepath.Dir(path) == chartsDir {
fmt.Printf("read: %s\n", path)
if info.Name() != "talm" {
chartNames = append(chartNames, info.Name())
}
return nil
}
if info.IsDir() {
return nil
}
relPath, err := filepath.Rel(chartsDir, path)
if err != nil {
return err
}
content, err := ioutil.ReadFile(path)
if err != nil {
return err
}
if filepath.Base(path) == "Chart.yaml" {
regex := regexp.MustCompile(`(name|version): \S+`)
content = regex.ReplaceAll(content, []byte("$1: %s"))
}
filesMap[relPath] = string(content)
return nil
})
if err != nil {
log.Fatal(err)
}
generateCode(filesMap, chartNames)
}
func generateCode(filesMap map[string]string, chartNames []string) {
var builder strings.Builder
// Sort keys
keys := make([]string, 0, len(filesMap))
for k := range filesMap {
keys = append(keys, k)
}
sort.Strings(keys)
builder.WriteString("package generated\n\n")
builder.WriteString("var PresetFiles = map[string]string{\n")
for _, key := range keys {
content := filesMap[key]
escapedContent := strings.ReplaceAll(content, "`", "` + \"`\" + `")
builder.WriteString(` "` + key + `": ` + "`" + escapedContent + "`,\n")
}
builder.WriteString("}\n\n")
builder.WriteString("var AvailablePresets = []string{\n")
// Generic chart should be first
for _, name := range chartNames {
if name == "generic" {
builder.WriteString(` "` + name + `",` + "\n")
}
}
for _, name := range chartNames {
if name != "generic" {
builder.WriteString(` "` + name + `",` + "\n")
}
}
builder.WriteString("}\n")
formattedSource, err := format.Source([]byte(builder.String()))
if err != nil {
log.Fatal(err)
}
err = ioutil.WriteFile(outputFile, []byte(formattedSource), 0644)
if err != nil {
log.Fatal(err)
}
fmt.Printf("written: %s\n", outputFile)
}