-
Notifications
You must be signed in to change notification settings - Fork 272
/
tinkerbellmachineconfig.go
133 lines (108 loc) Β· 3.57 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
package v1alpha1
import (
"fmt"
"net/url"
"strings"
"golang.org/x/crypto/ssh"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
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: Ubuntu,
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 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 config.Spec.OSImageURL != "" {
if _, err := url.ParseRequestURI(config.Spec.OSImageURL); err != nil {
return fmt.Errorf("parsing osImageOverride: %v", err)
}
}
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
}
}
func normalizeSSHKeys(machineConfig *TinkerbellMachineConfig) {
_ = stripCommentsFromSSHKeys(machineConfig)
}
func stripCommentsFromSSHKeys(machine *TinkerbellMachineConfig) error {
public, _, _, _, err := ssh.ParseAuthorizedKey([]byte(machine.Spec.Users[0].SshAuthorizedKeys[0]))
if err != nil {
return err
}
machine.Spec.Users[0].SshAuthorizedKeys[0] = strings.TrimSpace(string(ssh.MarshalAuthorizedKey(public)))
return nil
}