Merge pull request #18 from masterful/multiple-domains

Allow for multiple domains by using a name for upstream

Author: codesuki

main.go

@@ -17,6 +17,7 @@ import (
 )
 
 type container struct {
+	Name    string
 	Host    string
 	Port    string
 	Address string
@@ -156,10 +157,10 @@ func writeConfig(params []*container) error {
 	containers = make(map[string][]*container)
 	// Remap the containers as a 2D array with the domain as the index
 	for _, v := range params {
-		if _, ok := containers[v.Host]; !ok {
-			containers[v.Host] = make([]*container, 0)
+		if _, ok := containers[v.Name]; !ok {
+			containers[v.Name] = make([]*container, 0)
 		}
-		containers[v.Host] = append(containers[v.Host], v)
+		containers[v.Name] = append(containers[v.Name], v)
 	}
 	tmpl, err := newTemplate(filepath.Base(*templateFile)).ParseFiles(*templateFile)
 	if err != nil {

scanner.go

@@ -123,6 +123,7 @@ func (s *scanner) extractContainer(t *ecs.Task, cd *ecs.ContainerDefinition) (*c
 		return nil, errors.New("[" + *cd.Name + "] no valid port configuration found. skipping")
 	}
 	return &container{
+		Name:    strings.Fields(virtualHost)[0],
 		Host:    virtualHost,
 		Port:    port,
 		Env:     envVariables,

templates/nginx.tmpl

@@ -9,18 +9,18 @@ server {
         return 503;
 }
 
-{{ range $domain, $container := . }}
-upstream {{ $domain }} {
+{{ range $name, $container := . }}
+upstream {{ $name }} {
     {{ range $_, $value := $container }}
          server {{ $value.Address }}:{{ $value.Port }};
     {{ end }}
 }
 server {
-        server_name {{ $domain }};
+        server_name {{ (index $container 0).Host }};
         listen 80;
         access_log /var/log/nginx/access.log vhost;
         location / {
-                proxy_pass http://{{ $domain }};
+                proxy_pass http://{{ $name }};
         }
 }
 {{ end }}