Add .Env to access local environment variables

Fixes #17

Author: jwilder

docker-gen.go

@@ -79,6 +79,19 @@ type ConfigFile struct {
 	Config []Config
 }
 
+type Context []*RuntimeContainer
+
+func (c *Context) Env() map[string]string {
+
+	env := make(map[string]string)
+	for _, i := range os.Environ() {
+		parts := strings.Split(i, "=")
+		env[parts[0]] = parts[1]
+	}
+	return env
+
+}
+
 func (c *ConfigFile) filterWatches() ConfigFile {
 	configWithWatches := []Config{}
 

template.go

@@ -55,7 +55,7 @@ func contains(item map[string]string, key string) bool {
 	return false
 }
 
-func generateFile(config Config, containers []*RuntimeContainer) bool {
+func generateFile(config Config, containers Context) bool {
 	templatePath := config.Template
 	tmpl, err := template.New(filepath.Base(templatePath)).Funcs(template.FuncMap{
 		"contains":     contains,
@@ -68,7 +68,7 @@ func generateFile(config Config, containers []*RuntimeContainer) bool {
 		log.Fatalf("unable to parse template: %s", err)
 	}
 
-	filteredContainers := []*RuntimeContainer{}
+	filteredContainers := Context{}
 	if config.OnlyPublished {
 		for _, container := range containers {
 			if len(container.PublishedAddresses()) > 0 {
@@ -99,7 +99,7 @@ func generateFile(config Config, containers []*RuntimeContainer) bool {
 
 	var buf bytes.Buffer
 	multiwriter := io.MultiWriter(dest, &buf)
-	err = tmpl.ExecuteTemplate(multiwriter, filepath.Base(templatePath), filteredContainers)
+	err = tmpl.ExecuteTemplate(multiwriter, filepath.Base(templatePath), &filteredContainers)
 	if err != nil {
 		log.Fatalf("template error: %s\n", err)
 	}

templates/etcd.tmpl

@@ -1,6 +1,8 @@
 
 #!/bin/bash
 
+# Genenerated by {{ .Env.USER }}
+
 {{range $key, $value := .}}
 {{ $addrLen := len $value.Addresses }}
 {{ if gt $addrLen 0 }}