Initial version

helm/deployment.go

@@ -0,0 +1,80 @@
+package helm
+
+import "strings"
+
+type Deployment struct {
+	*K8sBase `yaml:",inline"`
+	Spec     *DepSpec `yaml:"spec"`
+}
+
+func NewDeployment() *Deployment {
+	d := &Deployment{K8sBase: NewBase(), Spec: NewDepSpec()}
+	d.K8sBase.ApiVersion = "apps/v1"
+	d.K8sBase.Kind = "Deployment"
+	return d
+}
+
+type DepSpec struct {
+	Replicas int                    `yaml:"replicas"`
+	Selector map[string]interface{} `yaml:"selector"`
+	Template PodTemplate            `yaml:"template"`
+}
+
+func NewDepSpec() *DepSpec {
+	return &DepSpec{
+		Replicas: 1,
+	}
+}
+
+type Value struct {
+	Name  string      `yaml:"name"`
+	Value interface{} `yaml:"value"`
+}
+
+type ContainerPort struct {
+	Name          string
+	ContainerPort int `yaml:"containerPort"`
+}
+
+type Container struct {
+	Name    string           `yaml:"name,omitempty"`
+	Image   string           `yaml:"image"`
+	Ports   []*ContainerPort `yaml:"ports,omitempty"`
+	Env     []Value          `yaml:"env,omitempty"`
+	Command []string         `yaml:"command,omitempty"`
+}
+
+func NewContainer(name, image string, environment, labels map[string]string) *Container {
+	container := &Container{
+		Image: image,
+		Name:  name,
+		Env:   make([]Value, len(environment)),
+	}
+
+	toServices := make([]string, 0)
+	if bound, ok := labels[K+"/to-services"]; ok {
+		toServices = strings.Split(bound, ",")
+	}
+
+	idx := 0
+	for n, v := range environment {
+		for _, name := range toServices {
+			if name == n {
+				v = "{{ .Release.Name }}-" + v
+			}
+		}
+		container.Env[idx] = Value{Name: n, Value: v}
+		idx++
+	}
+	return container
+}
+
+type PodSpec struct {
+	InitContainers []*Container `yaml:"initContainers,omitempty"`
+	Containers     []*Container `yaml:"containers"`
+}
+
+type PodTemplate struct {
+	Metadata Metadata `yaml:"metadata"`
+	Spec     PodSpec  `yaml:"spec"`
+}