Pull out hostnames more nicely

dotege.go

@@ -5,6 +5,7 @@ import (
 	"github.com/docker/docker/client"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 	"time"
 )
@@ -15,6 +16,16 @@ type Container struct {
 	Labels map[string]string
 }
 
+type LabelConfig struct {
+	Hostnames string
+}
+
+type Hostname struct {
+	Name         string
+	Alternatives map[string]bool
+	Containers   []Container
+}
+
 func monitorSignals() <-chan bool {
 	signals := make(chan os.Signal, 1)
 	done := make(chan bool, 1)
@@ -34,6 +45,9 @@ func main() {
 	done := monitorSignals()
 	containerChan := make(chan Container, 1)
 	expiryChan := make(chan string, 1)
+	labelConfig := LabelConfig{
+		Hostnames: "com.chameth.vhost",
+	}
 
 	cli, err := client.NewEnvClient()
 	if err != nil {
@@ -63,7 +77,10 @@ func main() {
 				delete(containers, name)
 				timer.Reset(100 * time.Millisecond)
 			case <-timer.C:
-				templateGenerator.Generate(Context{Containers: containers})
+				templateGenerator.Generate(Context{
+					Containers: containers,
+					Hostnames:  getHostnames(containers, labelConfig),
+				})
 			}
 		}
 	}()
@@ -75,3 +92,29 @@ func main() {
 		panic(err)
 	}
 }
+
+func getHostnames(containers map[string]Container, config LabelConfig) (hostnames map[string]Hostname) {
+	hostnames = make(map[string]Hostname)
+	for _, container := range containers {
+		if label, ok := container.Labels[config.Hostnames]; ok {
+			names := strings.Split(strings.Replace(label, ",", " ", -1), " ")
+			if hostname, ok := hostnames[names[0]]; ok {
+				hostname.Containers = append(hostname.Containers, container)
+			} else {
+				hostnames[names[0]] = Hostname{
+					Name:         names[0],
+					Alternatives: make(map[string]bool),
+					Containers:   []Container{container},
+				}
+			}
+			addAlternatives(hostnames[names[0]], names[1:])
+		}
+	}
+	return
+}
+
+func addAlternatives(hostname Hostname, alternatives []string) {
+	for _, alternative := range alternatives {
+		hostname.Alternatives[alternative] = true
+	}
+}

template_generator.go

@@ -11,6 +11,7 @@ import (
 
 type Context struct {
 	Containers map[string]Container
+	Hostnames map[string]Hostname
 }
 
 type TemplateConfig struct {

templates/domains.txt.tpl

@@ -1,5 +1,3 @@
-{{ range .Containers }}
-    {{- with index .Labels "com.chameth.vhost" -}}
-        {{ . | replace "," " " }}{{ "\n" }}
-    {{- end -}}
+{{ range .Hostnames }}
+    {{- .Name }}{{ range $san, $true := .Alternatives }} {{ $san }}{{ end }}
 {{ end }}