-
Notifications
You must be signed in to change notification settings - Fork 2.9k
/
machinedriver_data.go
73 lines (67 loc) · 2.35 KB
/
machinedriver_data.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
package app
import (
"github.com/rancher/types/apis/management.cattle.io/v3"
"github.com/rancher/types/config"
"github.com/sirupsen/logrus"
"k8s.io/apimachinery/pkg/apis/meta/v1"
)
func addMachineDrivers(management *config.ManagementContext) error {
if err := addMachineDriver("amazonec2", "local://", "", true, true, management); err != nil {
return err
}
if err := addMachineDriver("digitalocean", "local://", "", true, true, management); err != nil {
return err
}
if err := addMachineDriver("exoscale", "local://", "", false, true, management); err != nil {
return err
}
if err := addMachineDriver("openstack", "local://", "", false, true, management); err != nil {
return err
}
if err := addMachineDriver("otc", "https://obs.otc.t-systems.com/dockermachinedriver/docker-machine-driver-otc",
"e98f246f625ca46f5e037dc29bdf00fe", false, false, management); err != nil {
return err
}
if err := addMachineDriver("packet", "https://github.com/packethost/docker-machine-driver-packet/releases/download/v0.1.5/docker-machine-driver-packet_linux-amd64.zip",
"8231136e899311d54eca0b560b1feb96", false, false, management); err != nil {
return err
}
if err := addMachineDriver("rackspace", "local://", "", false, true, management); err != nil {
return err
}
if err := addMachineDriver("softlayer", "local://", "", false, true, management); err != nil {
return err
}
return addMachineDriver("vmwarevsphere", "local://", "", true, true, management)
}
func addMachineDriver(name, url, checksum string, active, builtin bool, management *config.ManagementContext) error {
lister := management.Management.NodeDrivers("").Controller().Lister()
cli := management.Management.NodeDrivers("")
m, _ := lister.Get("", name)
if m != nil {
if m.Spec.Builtin != builtin || m.Spec.URL != url || m.Spec.Checksum != checksum || m.Spec.DisplayName != name {
logrus.Infof("Updating node driver %v", name)
m.Spec.Builtin = builtin
m.Spec.URL = url
m.Spec.Checksum = checksum
m.Spec.DisplayName = name
_, err := cli.Update(m)
return err
}
return nil
}
logrus.Infof("Creating node driver %v", name)
_, err := cli.Create(&v3.NodeDriver{
ObjectMeta: v1.ObjectMeta{
Name: name,
},
Spec: v3.NodeDriverSpec{
Active: active,
Builtin: builtin,
URL: url,
DisplayName: name,
Checksum: checksum,
},
})
return err
}