2017-03-05 09:56:08 +02:00
|
|
|
package yaml
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
libcompose "github.com/docker/libcompose/yaml"
|
2019-11-14 21:16:03 +02:00
|
|
|
"gopkg.in/yaml.v3"
|
2017-03-05 09:56:08 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type (
|
|
|
|
// AuthConfig defines registry authentication credentials.
|
|
|
|
AuthConfig struct {
|
|
|
|
Username string
|
|
|
|
Password string
|
|
|
|
Email string
|
|
|
|
}
|
|
|
|
|
|
|
|
// Containers denotes an ordered collection of containers.
|
|
|
|
Containers struct {
|
|
|
|
Containers []*Container
|
|
|
|
}
|
|
|
|
|
|
|
|
// Container defines a container.
|
|
|
|
Container struct {
|
|
|
|
AuthConfig AuthConfig `yaml:"auth_config,omitempty"`
|
|
|
|
CapAdd []string `yaml:"cap_add,omitempty"`
|
|
|
|
CapDrop []string `yaml:"cap_drop,omitempty"`
|
|
|
|
Command libcompose.Command `yaml:"command,omitempty"`
|
|
|
|
Commands libcompose.Stringorslice `yaml:"commands,omitempty"`
|
|
|
|
CPUQuota libcompose.StringorInt `yaml:"cpu_quota,omitempty"`
|
|
|
|
CPUSet string `yaml:"cpuset,omitempty"`
|
|
|
|
CPUShares libcompose.StringorInt `yaml:"cpu_shares,omitempty"`
|
|
|
|
Detached bool `yaml:"detach,omitempty"`
|
|
|
|
Devices []string `yaml:"devices,omitempty"`
|
2017-09-08 02:43:33 +02:00
|
|
|
Tmpfs []string `yaml:"tmpfs,omitempty"`
|
2017-03-05 09:56:08 +02:00
|
|
|
DNS libcompose.Stringorslice `yaml:"dns,omitempty"`
|
|
|
|
DNSSearch libcompose.Stringorslice `yaml:"dns_search,omitempty"`
|
|
|
|
Entrypoint libcompose.Command `yaml:"entrypoint,omitempty"`
|
|
|
|
Environment libcompose.SliceorMap `yaml:"environment,omitempty"`
|
|
|
|
ExtraHosts []string `yaml:"extra_hosts,omitempty"`
|
|
|
|
Group string `yaml:"group,omitempty"`
|
|
|
|
Image string `yaml:"image,omitempty"`
|
|
|
|
Isolation string `yaml:"isolation,omitempty"`
|
|
|
|
Labels libcompose.SliceorMap `yaml:"labels,omitempty"`
|
|
|
|
MemLimit libcompose.MemStringorInt `yaml:"mem_limit,omitempty"`
|
|
|
|
MemSwapLimit libcompose.MemStringorInt `yaml:"memswap_limit,omitempty"`
|
|
|
|
MemSwappiness libcompose.MemStringorInt `yaml:"mem_swappiness,omitempty"`
|
|
|
|
Name string `yaml:"name,omitempty"`
|
|
|
|
NetworkMode string `yaml:"network_mode,omitempty"`
|
2017-09-08 02:43:33 +02:00
|
|
|
IpcMode string `yaml:"ipc_mode,omitempty"`
|
2017-03-05 09:56:08 +02:00
|
|
|
Networks libcompose.Networks `yaml:"networks,omitempty"`
|
|
|
|
Privileged bool `yaml:"privileged,omitempty"`
|
|
|
|
Pull bool `yaml:"pull,omitempty"`
|
|
|
|
ShmSize libcompose.MemStringorInt `yaml:"shm_size,omitempty"`
|
|
|
|
Ulimits libcompose.Ulimits `yaml:"ulimits,omitempty"`
|
|
|
|
Volumes libcompose.Volumes `yaml:"volumes,omitempty"`
|
2017-04-10 12:39:50 +02:00
|
|
|
Secrets Secrets `yaml:"secrets,omitempty"`
|
2017-11-18 00:49:01 +02:00
|
|
|
Sysctls libcompose.SliceorMap `yaml:"sysctls,omitempty"`
|
2017-03-05 09:56:08 +02:00
|
|
|
Constraints Constraints `yaml:"when,omitempty"`
|
|
|
|
Vargs map[string]interface{} `yaml:",inline"`
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
// UnmarshalYAML implements the Unmarshaller interface.
|
2019-11-14 21:16:03 +02:00
|
|
|
func (c *Containers) UnmarshalYAML(value *yaml.Node) error {
|
|
|
|
containers := map[string]Container{}
|
|
|
|
err := value.Decode(&containers)
|
|
|
|
if err != nil {
|
2017-03-05 09:56:08 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-11-14 21:16:03 +02:00
|
|
|
for i, n := range value.Content {
|
|
|
|
if i%2 == 1 {
|
|
|
|
container := Container{}
|
|
|
|
err := n.Decode(&container)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2017-03-05 09:56:08 +02:00
|
|
|
|
2019-11-14 21:16:03 +02:00
|
|
|
if container.Name == "" {
|
|
|
|
container.Name = fmt.Sprintf("%v", value.Content[i-1].Value)
|
|
|
|
}
|
|
|
|
c.Containers = append(c.Containers, &container)
|
2017-03-05 09:56:08 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|