From 6e79fff1182e927a0eec53a8751d2e0a5b2ebb6b Mon Sep 17 00:00:00 2001
From: psyhomb <htchak19@gmail.com>
Date: Tue, 25 Feb 2020 02:39:38 +0100
Subject: [PATCH] Add support for Go template to -tags flag

---
 VERSION          |   2 +-
 bridge/bridge.go | 135 +++++++++++++++++++++++++++++++++++++++++++++--
 registrator.go   |   2 +-
 3 files changed, 133 insertions(+), 6 deletions(-)

diff --git a/VERSION b/VERSION
index 02a819f21..aea985200 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-v7
+v7.3.1
diff --git a/bridge/bridge.go b/bridge/bridge.go
index f02ca99c9..ba938d8cf 100644
--- a/bridge/bridge.go
+++ b/bridge/bridge.go
@@ -1,6 +1,7 @@
 package bridge
 
 import (
+	"bytes"
 	"errors"
 	"log"
 	"net"
@@ -11,6 +12,7 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"text/template"
 
 	dockerapi "github.com/fsouza/go-dockerclient"
 )
@@ -202,7 +204,7 @@ func (b *Bridge) add(containerId string, quiet bool) {
 
 	// Extract configured host port mappings, relevant when using --net=host
 	for port, _ := range container.Config.ExposedPorts {
-		published := []dockerapi.PortBinding{ {"0.0.0.0", port.Port()}, }
+		published := []dockerapi.PortBinding{{"0.0.0.0", port.Port()}}
 		ports[string(port)] = servicePort(container, port, published)
 	}
 
@@ -309,7 +311,7 @@ func (b *Bridge) newService(port ServicePort, isgroup bool) *Service {
 				service.IP = containerIp
 			}
 			log.Println("using container IP " + service.IP + " from label '" +
-				b.config.UseIpFromLabel  + "'")
+				b.config.UseIpFromLabel + "'")
 		} else {
 			log.Println("Label '" + b.config.UseIpFromLabel +
 				"' not found in container configuration")
@@ -332,13 +334,138 @@ func (b *Bridge) newService(port ServicePort, isgroup bool) *Service {
 		}
 	}
 
+	// Use container inspect data to populate tags list
+	// https://github.com/fsouza/go-dockerclient/blob/master/container.go#L441-L483
+	ForceTags := b.config.ForceTags
+	if len(ForceTags) != 0 {
+		// Template functions
+		fm := template.FuncMap{
+			// Slice function
+			"slice": func(v string, i ...int) string {
+				if len(i) == 1 {
+					if len(v) >= i[0] {
+						return v[i[0]:]
+					}
+				}
+
+				if len(i) == 2 {
+					if len(v) >= i[0] && len(v) >= i[1] {
+						if i[0] == 0 {
+							return v[:i[1]]
+						}
+						if i[1] < i[0] {
+							return v[i[0]:]
+						}
+						return v[i[0]:i[1]]
+					}
+				}
+
+				return v
+			},
+			// sIndex function
+			"sIndex": func(s []string, i int) string {
+				if i < 0 {
+					i = i * -1
+					if i >= len(s) {
+						return s[0]
+					}
+					return s[len(s)-i]
+				}
+
+				if i >= len(s) {
+					return s[len(s)-1]
+				}
+
+				return s[i]
+			},
+			// mIndex function
+			"mIndex": func(m map[string]string, k string) string {
+				return m[k]
+			},
+			// toUpper function
+			"toUpper": func(v string) string {
+				return strings.ToUpper(v)
+			},
+			// toLower function
+			"toLower": func(v string) string {
+				return strings.ToLower(v)
+			},
+			// join function
+			"join": func(s []string, sep string) string {
+				return strings.Join(s, sep)
+			},
+			// split function
+			"split": func(v, sep string) []string {
+				return strings.Split(v, sep)
+			},
+			// splitIndex function
+			"splitIndex": func(v, sep string, i int) string {
+				l := strings.Split(v, sep)
+
+				if i < 0 {
+					i = i * -1
+					if i >= len(l) {
+						return l[0]
+					}
+					return l[len(l)-i]
+				}
+
+				if i >= len(l) {
+					return l[len(l)-1]
+				}
+
+				return l[i]
+			},
+			// matchFirstElement function
+			"matchFirstElement": func(s []string, r string) string {
+				var m string
+
+				re := regexp.MustCompile(r)
+				for _, e := range s {
+					if re.MatchString(e) {
+						m = e
+						break
+					}
+				}
+
+				return m
+			},
+			// matchAllElements function
+			"matchAllElements": func(s []string, r string) []string {
+				var m []string
+
+				re := regexp.MustCompile(r)
+				for _, e := range s {
+					if re.MatchString(e) {
+						m = append(m, e)
+					}
+				}
+
+				return m
+			},
+		}
+
+		tmpl, err := template.New("tags").Funcs(fm).Parse(ForceTags)
+		if err != nil {
+			log.Fatalf("%s template parsing failed with error: %s", ForceTags, err)
+		}
+
+		var b bytes.Buffer
+		err = tmpl.Execute(&b, container)
+		if err != nil {
+			log.Fatalf("%s template execution failed with error: %s", ForceTags, err)
+		}
+
+		ForceTags = b.String()
+	}
+
 	if port.PortType == "udp" {
 		service.Tags = combineTags(
-			mapDefault(metadata, "tags", ""), b.config.ForceTags, "udp")
+			mapDefault(metadata, "tags", ""), ForceTags, "udp")
 		service.ID = service.ID + ":udp"
 	} else {
 		service.Tags = combineTags(
-			mapDefault(metadata, "tags", ""), b.config.ForceTags)
+			mapDefault(metadata, "tags", ""), ForceTags)
 	}
 
 	id := mapDefault(metadata, "id", "")
diff --git a/registrator.go b/registrator.go
index e26d3397a..bcb483124 100644
--- a/registrator.go
+++ b/registrator.go
@@ -25,7 +25,7 @@ var explicit = flag.Bool("explicit", false, "Only register containers which have
 var useIpFromLabel = flag.String("useIpFromLabel", "", "Use IP which is stored in a label assigned to the container")
 var refreshInterval = flag.Int("ttl-refresh", 0, "Frequency with which service TTLs are refreshed")
 var refreshTtl = flag.Int("ttl", 0, "TTL for services (default is no expiry)")
-var forceTags = flag.String("tags", "", "Append tags for all registered services")
+var forceTags = flag.String("tags", "", "Append tags for all registered services (supports Go template)")
 var resyncInterval = flag.Int("resync", 0, "Frequency with which services are resynchronized")
 var deregister = flag.String("deregister", "always", "Deregister exited services \"always\" or \"on-success\"")
 var retryAttempts = flag.Int("retry-attempts", 0, "Max retry attempts to establish a connection with the backend. Use -1 for infinite retries")