Further re-organize the project structure

.github/workflows/tests.yml

@@ -32,4 +32,4 @@ jobs:
         run: make check-gofmt
 
       - name: Run tests
-        run: go test -v ./internal/dockergen
+        run: go test -v ./internal/...

Makefile

@@ -44,16 +44,16 @@ get-deps:
 	go mod download
 
 check-gofmt:
-	if [ -n "$(shell gofmt -l ./cmd/docker-gen)" ]; then \
+	if [ -n "$(shell go fmt ./cmd/...)" ]; then \
 		echo 1>&2 'The following files need to be formatted:'; \
 		gofmt -l ./cmd/docker-gen; \
 		exit 1; \
 	fi
-	if [ -n "$(shell gofmt -l ./internal/dockergen)" ]; then \
+	if [ -n "$(shell go fmt ./internal/...)" ]; then \
 		echo 1>&2 'The following files need to be formatted:'; \
 		gofmt -l ./internal/dockergen; \
 		exit 1; \
 	fi
 
 test:
-	go test ./internal/dockergen
+	go test ./internal/...

cmd/docker-gen/main.go

@@ -9,7 +9,8 @@ import (
 
 	"github.com/BurntSushi/toml"
 	docker "github.com/fsouza/go-dockerclient"
-	"github.com/nginx-proxy/docker-gen/internal/dockergen"
+	"github.com/nginx-proxy/docker-gen/internal/config"
+	"github.com/nginx-proxy/docker-gen/internal/generator"
 )
 
 type stringslice []string
@@ -27,7 +28,7 @@ var (
 	onlyPublished         bool
 	includeStopped        bool
 	configFiles           stringslice
-	configs               dockergen.ConfigFile
+	configs               config.ConfigFile
 	interval              int
 	keepBlankLines        bool
 	endpoint              string
@@ -133,11 +134,11 @@ func main() {
 			}
 		}
 	} else {
-		w, err := dockergen.ParseWait(wait)
+		w, err := config.ParseWait(wait)
 		if err != nil {
 			log.Fatalf("Error parsing wait interval: %s\n", err)
 		}
-		config := dockergen.Config{
+		cfg := config.Config{
 			Template:         flag.Arg(0),
 			Dest:             flag.Arg(1),
 			Watch:            watch,
@@ -152,10 +153,10 @@ func main() {
 			KeepBlankLines:   keepBlankLines,
 		}
 		if notifyContainerID != "" {
-			config.NotifyContainers[notifyContainerID] = notifyContainerSignal
+			cfg.NotifyContainers[notifyContainerID] = notifyContainerSignal
 		}
-		configs = dockergen.ConfigFile{
-			Config: []dockergen.Config{config}}
+		configs = config.ConfigFile{
+			Config: []config.Config{cfg}}
 	}
 
 	all := true
@@ -165,7 +166,7 @@ func main() {
 		}
 	}
 
-	generator, err := dockergen.NewGenerator(dockergen.GeneratorConfig{
+	generator, err := generator.NewGenerator(generator.GeneratorConfig{
 		Endpoint:   endpoint,
 		TLSKey:     tlsKey,
 		TLSCert:    tlsCert,

internal/dockergen/config.go → internal/config/config.go

@@ -1,4 +1,4 @@
-package dockergen
+package config
 
 import (
 	"errors"

internal/dockergen/config_test.go → internal/config/config_test.go

@@ -1,11 +1,29 @@
-package dockergen
+package config
 
 import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
+func TestFilterWatches(t *testing.T) {
+	testConfigFile := &ConfigFile{
+		Config: []Config{
+			{Template: "foo", Watch: true},
+			{Template: "bar"},
+			{Template: "baz", Watch: true},
+		},
+	}
+
+	expected := []Config{
+		{Template: "foo", Watch: true},
+		{Template: "baz", Watch: true},
+	}
+
+	configFile := testConfigFile.FilterWatches()
+	assert.Equal(t, expected, configFile.Config)
+}
+
 func TestParseWait(t *testing.T) {
 	incorrectIntervals := []string{
 		"500x",    // Incorrect min interval

internal/dockergen/context.go → internal/context/context.go

@@ -1,4 +1,4 @@
-package dockergen
+package context
 
 import (
 	"bufio"
@@ -8,6 +8,7 @@ import (
 	"sync"
 
 	docker "github.com/fsouza/go-dockerclient"
+	"github.com/nginx-proxy/docker-gen/internal/utils"
 )
 
 var (
@@ -19,7 +20,7 @@ var (
 type Context []*RuntimeContainer
 
 func (c *Context) Env() map[string]string {
-	return splitKeyValueSlice(os.Environ())
+	return utils.SplitKeyValueSlice(os.Environ())
 }
 
 func (c *Context) Docker() Docker {

internal/dockergen/context_test.go → internal/context/context_test.go

@@ -1,4 +1,4 @@
-package dockergen
+package context
 
 import (
 	"fmt"
@@ -163,3 +163,39 @@ func TestPublishedAddresses(t *testing.T) {
 
 	assert.ElementsMatch(t, expected, container.PublishedAddresses())
 }
+
+func TestRuntimeContainerEquals(t *testing.T) {
+	rc1 := &RuntimeContainer{
+		ID: "baz",
+		Image: DockerImage{
+			Registry: "foo/bar",
+		},
+	}
+	rc2 := &RuntimeContainer{
+		ID:   "baz",
+		Name: "qux",
+		Image: DockerImage{
+			Registry: "foo/bar",
+		},
+	}
+	assert.True(t, rc1.Equals(*rc2))
+	assert.True(t, rc2.Equals(*rc1))
+
+	rc2.Image.Tag = "quux"
+	assert.False(t, rc1.Equals(*rc2))
+	assert.False(t, rc2.Equals(*rc1))
+}
+
+func TestDockerImageString(t *testing.T) {
+	image := &DockerImage{Repository: "foo/bar"}
+	assert.Equal(t, "foo/bar", image.String())
+
+	image.Registry = "baz.io"
+	assert.Equal(t, "baz.io/foo/bar", image.String())
+
+	image.Tag = "qux"
+	assert.Equal(t, "baz.io/foo/bar:qux", image.String())
+
+	image.Registry = ""
+	assert.Equal(t, "foo/bar:qux", image.String())
+}

internal/dockergen/docker_client.go → internal/dockerclient/docker_cli.go

@@ -1,20 +1,40 @@
-package dockergen
+package dockerclient
 
 import (
 	"errors"
 	"fmt"
+	"os"
 	"strconv"
 	"strings"
 
 	docker "github.com/fsouza/go-dockerclient"
+	"github.com/nginx-proxy/docker-gen/internal/utils"
 )
 
+func GetEndpoint(endpoint string) (string, error) {
+	defaultEndpoint := "unix:///var/run/docker.sock"
+	if os.Getenv("DOCKER_HOST") != "" {
+		defaultEndpoint = os.Getenv("DOCKER_HOST")
+	}
+
+	if endpoint != "" {
+		defaultEndpoint = endpoint
+	}
+
+	_, _, err := parseHost(defaultEndpoint)
+	if err != nil {
+		return "", err
+	}
+
+	return defaultEndpoint, nil
+}
+
 func NewDockerClient(endpoint string, tlsVerify bool, tlsCert, tlsCaCert, tlsKey string) (*docker.Client, error) {
 	if strings.HasPrefix(endpoint, "unix:") {
 		return docker.NewClient(endpoint)
 	} else if tlsVerify || tlsEnabled(tlsCert, tlsCaCert, tlsKey) {
 		if tlsVerify {
-			if e, err := pathExists(tlsCaCert); !e || err != nil {
+			if e, err := utils.PathExists(tlsCaCert); !e || err != nil {
 				return nil, errors.New("TLS verification was requested, but CA cert does not exist")
 			}
 		}
@@ -26,7 +46,7 @@ func NewDockerClient(endpoint string, tlsVerify bool, tlsCert, tlsCaCert, tlsKey
 
 func tlsEnabled(tlsCert, tlsCaCert, tlsKey string) bool {
 	for _, v := range []string{tlsCert, tlsCaCert, tlsKey} {
-		if e, err := pathExists(v); e && err == nil {
+		if e, err := utils.PathExists(v); e && err == nil {
 			return true
 		}
 	}
@@ -98,7 +118,7 @@ func parseHost(addr string) (string, string, error) {
 	return proto, fmt.Sprintf("%s:%d", host, port), nil
 }
 
-func splitDockerImage(img string) (string, string, string) {
+func SplitDockerImage(img string) (string, string, string) {
 	index := 0
 	repository := img
 	var registry, tag string