Add dict function

Author: md5

template.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"bytes"
+	"errors"
 	"io"
 	"io/ioutil"
 	"log"
@@ -56,6 +57,21 @@ func contains(item map[string]string, key string) bool {
 	return false
 }
 
+func dict(values ...interface{}) (map[string]interface{}, error) {
+	if len(values)%2 != 0 {
+		return nil, errors.New("invalid dict call")
+	}
+	dict := make(map[string]interface{}, len(values)/2)
+	for i := 0; i < len(values); i+=2 {
+		key, ok := values[i].(string)
+		if !ok {
+			return nil, errors.New("dict keys must be strings")
+		}
+		dict[key] = values[i+1]
+	}
+	return dict, nil
+}
+
 func generateFile(config Config, containers Context) bool {
 	templatePath := config.Template
 	tmpl, err := template.New(filepath.Base(templatePath)).Funcs(template.FuncMap{
@@ -65,6 +81,7 @@ func generateFile(config Config, containers Context) bool {
 		"groupByMulti": groupByMulti,
 		"split":        strings.Split,
 		"replace":      strings.Replace,
+		"dict":         dict,
 	}).ParseFiles(templatePath)
 	if err != nil {
 		log.Fatalf("unable to parse template: %s", err)

template_test.go

@@ -99,3 +99,36 @@ func TestGroupByMulti(t *testing.T) {
 		t.Fatalf("expected 2 got %s", groups["demo3.localhost"][0].ID)
 	}
 }
+
+func TestDict(t *testing.T) {
+	containers := []*RuntimeContainer{
+		&RuntimeContainer{
+			Env: map[string]string{
+				"VIRTUAL_HOST": "demo1.localhost",
+			},
+			ID: "1",
+		},
+		&RuntimeContainer{
+			Env: map[string]string{
+				"VIRTUAL_HOST": "demo1.localhost,demo3.localhost",
+			},
+			ID: "2",
+		},
+		&RuntimeContainer{
+			Env: map[string]string{
+				"VIRTUAL_HOST": "demo2.localhost",
+			},
+			ID: "3",
+		},
+	}
+	d, err := dict("/", containers)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if d["/"] == nil {
+		t.Fatalf("did not find containers in dict: %s", d)
+	}
+	if d["MISSING"] != nil {
+		t.Fail()
+	}
+}