/
file.go
203 lines (182 loc) · 5.62 KB
/
file.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
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
/*
Copyright 2020 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package common
import (
"fmt"
"os"
gofilepath "path/filepath"
"strconv"
"gopkg.in/yaml.v3"
)
// Config is the internal representation of the yaml that defines
// server configuration
type Config struct {
RuntimeEndpoint string
ImageEndpoint string
Timeout int
Debug bool
PullImageOnCreate bool
DisablePullOnRun bool
yamlData *yaml.Node //YAML representation of config
}
// ReadConfig reads from a file with the given name and returns a config or
// an error if the file was unable to be parsed.
func ReadConfig(filepath string) (*Config, error) {
data, err := os.ReadFile(filepath)
if err != nil {
return nil, err
}
yamlConfig := yaml.Node{}
err = yaml.Unmarshal(data, &yamlConfig)
if err != nil {
return nil, err
}
config, err := getConfigOptions(yamlConfig)
if err != nil {
return nil, err
}
return config, err
}
// WriteConfig writes config to file
// an error if the file was unable to be written to.
func WriteConfig(c *Config, filepath string) error {
if c == nil {
c = new(Config)
c.yamlData = new(yaml.Node)
}
setConfigOptions(c)
data, err := yaml.Marshal(c.yamlData)
if err != nil {
return err
}
if err := os.MkdirAll(gofilepath.Dir(filepath), 0o755); err != nil {
return err
}
return os.WriteFile(filepath, data, 0o644)
}
// Extracts config options from the yaml data which is loaded from file
func getConfigOptions(yamlData yaml.Node) (*Config, error) {
config := Config{}
config.yamlData = &yamlData
if yamlData.Content == nil || len(yamlData.Content) == 0 ||
yamlData.Content[0].Content == nil {
return &config, nil
}
contentLen := len(yamlData.Content[0].Content)
// YAML representation contains 2 yaml ScalarNodes per config option.
// One is config option name and other is the value of the option
// These ScalarNodes help preserve comments associated with
// the YAML entry
for indx := 0; indx < contentLen-1; {
configOption := yamlData.Content[0].Content[indx]
name := configOption.Value
value := yamlData.Content[0].Content[indx+1].Value
var err error
switch name {
case "runtime-endpoint":
config.RuntimeEndpoint = value
case "image-endpoint":
config.ImageEndpoint = value
case "timeout":
config.Timeout, err = strconv.Atoi(value)
if err != nil {
return nil, fmt.Errorf("parsing config option '%s': %w", name, err)
}
case "debug":
config.Debug, err = strconv.ParseBool(value)
if err != nil {
return nil, fmt.Errorf("parsing config option '%s': %w", name, err)
}
case "pull-image-on-create":
config.PullImageOnCreate, err = strconv.ParseBool(value)
if err != nil {
return nil, fmt.Errorf("parsing config option '%s': %w", name, err)
}
case "disable-pull-on-run":
config.DisablePullOnRun, err = strconv.ParseBool(value)
if err != nil {
return nil, fmt.Errorf("parsing config option '%s': %w", name, err)
}
default:
return nil, fmt.Errorf("Config option '%s' is not valid", name)
}
indx += 2
}
return &config, nil
}
// Set config options on yaml data for persistece to file
func setConfigOptions(config *Config) {
setConfigOption("runtime-endpoint", config.RuntimeEndpoint, config.yamlData)
setConfigOption("image-endpoint", config.ImageEndpoint, config.yamlData)
setConfigOption("timeout", strconv.Itoa(config.Timeout), config.yamlData)
setConfigOption("debug", strconv.FormatBool(config.Debug), config.yamlData)
setConfigOption("pull-image-on-create", strconv.FormatBool(config.PullImageOnCreate), config.yamlData)
setConfigOption("disable-pull-on-run", strconv.FormatBool(config.DisablePullOnRun), config.yamlData)
}
// Set config option on yaml
func setConfigOption(configName, configValue string, yamlData *yaml.Node) {
if yamlData.Content == nil || len(yamlData.Content) == 0 {
yamlData.Kind = yaml.DocumentNode
yamlData.Content = make([]*yaml.Node, 1)
yamlData.Content[0] = &yaml.Node{
Kind: yaml.MappingNode,
Tag: "!!map",
}
}
var contentLen = 0
var foundOption = false
if yamlData.Content[0].Content != nil {
contentLen = len(yamlData.Content[0].Content)
}
// Set value on existing config option
for indx := 0; indx < contentLen-1; {
name := yamlData.Content[0].Content[indx].Value
if name == configName {
yamlData.Content[0].Content[indx+1].Value = configValue
foundOption = true
break
}
indx += 2
}
// New config option to set
// YAML representation contains 2 yaml ScalarNodes per config option.
// One is config option name and other is the value of the option
// These ScalarNodes help preserve comments associated with
// the YAML entry
if !foundOption {
name := &yaml.Node{
Kind: yaml.ScalarNode,
Value: configName,
Tag: "!!str",
}
var tagType string
switch configName {
case "timeout":
tagType = "!!int"
case "debug":
tagType = "!!bool"
case "pull-image-on-create":
tagType = "!!bool"
case "disable-pull-on-run":
tagType = "!!bool"
default:
tagType = "!!str"
}
value := &yaml.Node{
Kind: yaml.ScalarNode,
Value: configValue,
Tag: tagType,
}
yamlData.Content[0].Content = append(yamlData.Content[0].Content, name, value)
}
}