-
Notifications
You must be signed in to change notification settings - Fork 234
/
machine_defaults.go
79 lines (68 loc) · 1.88 KB
/
machine_defaults.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
package scale
import (
"strconv"
"github.com/superfly/flyctl/api"
"github.com/superfly/flyctl/internal/appconfig"
)
type defaultValues struct {
image string
guest *api.MachineGuest
guestPerGroup map[string]*api.MachineGuest
releaseId string
releaseVersion string
appConfig *appconfig.Config
}
func newDefaults(appConfig *appconfig.Config, latest api.Release, machines []*api.Machine) *defaultValues {
var (
defaultGroupName = appConfig.DefaultProcessName()
guest *api.MachineGuest
releaseId = latest.ID
releaseVersion = strconv.Itoa(latest.Version)
image = latest.ImageRef
guestPerGroup = make(map[string]*api.MachineGuest)
)
for _, m := range machines {
groupName := m.ProcessGroup()
if _, ok := guestPerGroup[groupName]; ok {
continue
} else if m.Config.Guest != nil {
guestPerGroup[groupName] = m.Config.Guest
if groupName == defaultGroupName {
guest = m.Config.Guest
}
}
}
// In case we haven't found a guest for the default,
// scan all the existing groups and pick the first
if guest == nil {
for _, name := range appConfig.ProcessNames() {
if v, ok := guestPerGroup[name]; ok {
guest = v
break
}
}
}
return &defaultValues{
image: image,
guest: guest,
guestPerGroup: guestPerGroup,
releaseId: releaseId,
releaseVersion: releaseVersion,
appConfig: appConfig,
}
}
func (d *defaultValues) ToMachineConfig(groupName string) (*api.MachineConfig, error) {
mc, err := d.appConfig.ToMachineConfig(groupName, nil)
if err != nil {
return nil, err
}
if guest, ok := d.guestPerGroup[groupName]; ok {
mc.Guest = guest
} else {
mc.Guest = d.guest
}
mc.Image = d.image
mc.Metadata[api.MachineConfigMetadataKeyFlyReleaseId] = d.releaseId
mc.Metadata[api.MachineConfigMetadataKeyFlyReleaseVersion] = d.releaseVersion
return mc, nil
}