Add support for labels and windows

README.md

@@ -10,6 +10,8 @@ The first argument specifies the target file, the second argument specifies the
 Each docker container will be added as `<container-name>.docker`, `<container-short-id>.docker`.  
 Which network will be used for containers with multiple networks is undefined behaviour.
 
+You can also set additional names (seprated by `;`) via the label `docker-hosts.hosts`.
+
 **Tip:** if you want to use this on a dedicated DNS server instead of using the hosts file directly, I suggest using [CoreDNS](https://coredns.io/) with the [hosts plugin](https://coredns.io/plugins/hosts/).
 
 ## Build and installation

main.go

@@ -28,7 +28,13 @@ func ContainerToHosts(container types.Container, tld string) string {
 		if details.IPAddress == "" {
 			continue
 		}
-		hosts = details.IPAddress
+		// <docker-hosts.host>
+		if container.Labels["docker-hosts.hosts"] != "" {
+			names := strings.Split(container.Labels["docker-hosts.hosts"], ";")
+			for _, name := range names {
+				hosts += "\t" + strings.TrimPrefix(name, "/") + "." + tld
+			}
+		}
 		for _, name := range container.Names {
 			// <name>.docker
 			hosts += "\t" + strings.TrimPrefix(name, "/") + "." + tld
@@ -128,11 +134,13 @@ func main() {
 		panic(err)
 	}
 
-	file := "/etc/hosts"
+	file := hostFile
 	if len(os.Args) > 1 {
 		file = os.Args[1]
 	}
 
+	fmt.Printf("Hostfile was selected as: %s\n", file)
+
 	tld := "docker"
 	if len(os.Args) > 2 {
 		tld = os.Args[2]

unix.go

@@ -0,0 +1,5 @@
+// +build !windows
+
+package main
+
+const hostFile = "/etc/hosts"

windows.go

@@ -0,0 +1,5 @@
+// +build windows
+
+package main
+
+const hostFile = "C:\\Windows\\System32\\drivers\\etc\\hosts"