-
Notifications
You must be signed in to change notification settings - Fork 272
/
tinkerbellmachineconfig.go
137 lines (116 loc) Β· 3.87 KB
/
tinkerbellmachineconfig.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
package v1alpha1
import (
"fmt"
"os"
"strings"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/yaml"
)
const TinkerbellMachineConfigKind = "TinkerbellMachineConfig"
// +kubebuilder:object:generate=false
type TinkerbellMachineConfigGenerateOpt func(config *TinkerbellMachineConfigGenerate)
// Used for generating yaml for generate clusterconfig command.
func NewTinkerbellMachineConfigGenerate(name string, opts ...TinkerbellMachineConfigGenerateOpt) *TinkerbellMachineConfigGenerate {
machineConfig := &TinkerbellMachineConfigGenerate{
TypeMeta: metav1.TypeMeta{
Kind: TinkerbellMachineConfigKind,
APIVersion: SchemeBuilder.GroupVersion.String(),
},
ObjectMeta: ObjectMeta{
Name: name,
},
Spec: TinkerbellMachineConfigSpec{
HardwareSelector: HardwareSelector{},
OSFamily: Bottlerocket,
Users: []UserConfiguration{
{
Name: "ec2-user",
SshAuthorizedKeys: []string{"ssh-rsa AAAA..."},
},
},
},
}
for _, opt := range opts {
opt(machineConfig)
}
return machineConfig
}
func (c *TinkerbellMachineConfigGenerate) APIVersion() string {
return c.TypeMeta.APIVersion
}
func (c *TinkerbellMachineConfigGenerate) Kind() string {
return c.TypeMeta.Kind
}
func (c *TinkerbellMachineConfigGenerate) Name() string {
return c.ObjectMeta.Name
}
func GetTinkerbellMachineConfigs(fileName string) (map[string]*TinkerbellMachineConfig, error) {
configs := make(map[string]*TinkerbellMachineConfig)
content, err := os.ReadFile(fileName)
if err != nil {
return nil, fmt.Errorf("unable to read file due to: %v", err)
}
for _, c := range strings.Split(string(content), YamlSeparator) {
var config TinkerbellMachineConfig
if err = yaml.UnmarshalStrict([]byte(c), &config); err == nil {
if config.Kind == TinkerbellMachineConfigKind {
configs[config.Name] = &config
continue
}
}
_ = yaml.Unmarshal([]byte(c), &config) // this is to check if there is a bad spec in the file
if config.Kind == TinkerbellMachineConfigKind {
return nil, fmt.Errorf("unable to unmarshall content from file due to: %v", err)
}
}
if len(configs) == 0 {
return nil, fmt.Errorf("unable to find kind %v in file", TinkerbellMachineConfigKind)
}
return configs, nil
}
func WithTemplateRef(ref ProviderRefAccessor) TinkerbellMachineConfigGenerateOpt {
return func(c *TinkerbellMachineConfigGenerate) {
c.Spec.TemplateRef = Ref{
Kind: ref.Kind(),
Name: ref.Name(),
}
}
}
func validateTinkerbellMachineConfig(config *TinkerbellMachineConfig) error {
if err := validateObjectMeta(config.ObjectMeta); err != nil {
return fmt.Errorf("TinkerbellMachineConfig: %v", err)
}
if len(config.Spec.HardwareSelector) == 0 {
return fmt.Errorf("TinkerbellMachineConfig: missing spec.hardwareSelector: %s", config.Name)
}
if len(config.Spec.HardwareSelector) != 1 {
return fmt.Errorf(
"TinkerbellMachineConfig: spec.hardwareSelector must contain only 1 key-value pair: %s",
config.Name,
)
}
if config.Spec.OSFamily == "" {
return fmt.Errorf("TinkerbellMachineConfig: missing spec.osFamily: %s", config.Name)
}
if config.Spec.OSFamily != Ubuntu && config.Spec.OSFamily != Bottlerocket && config.Spec.OSFamily != RedHat {
return fmt.Errorf(
"TinkerbellMachineConfig: unsupported spec.osFamily (%v); Please use one of the following: %s, %s, %s",
config.Spec.OSFamily,
Ubuntu,
RedHat,
Bottlerocket,
)
}
if len(config.Spec.Users) == 0 {
return fmt.Errorf("TinkerbellMachineConfig: missing spec.Users: %s", config.Name)
}
if err := validateHostOSConfig(config.Spec.HostOSConfiguration, config.Spec.OSFamily); err != nil {
return fmt.Errorf("HostOSConfiguration is invalid for TinkerbellMachineConfig %s: %v", config.Name, err)
}
return nil
}
func setTinkerbellMachineConfigDefaults(machineConfig *TinkerbellMachineConfig) {
if machineConfig.Spec.OSFamily == "" {
machineConfig.Spec.OSFamily = Bottlerocket
}
}