init

generator/docker.go

@@ -0,0 +1,65 @@
+package generator
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/compose-spec/compose-go/v2/cli"
+	"github.com/compose-spec/compose-go/v2/types"
+	"github.com/kevinmidboe/traefik-etcd-advertiser/client/etcd"
+)
+
+func dockerLabelToEtcdKey(key string) string {
+	return strings.ReplaceAll(key, ".", "/")
+}
+
+func ParseDockerCompose(composeFilePath string) (*types.Project, error) {
+	ctx := context.Background()
+
+	options, err := cli.NewProjectOptions(
+		[]string{composeFilePath},
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	project, err := options.LoadProject(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	return project, nil
+}
+
+func createPacket2(config *types.Project) []etcd.EtcdPacket {
+	blocks := []etcd.EtcdPacket{}
+
+	fmt.Println("DockerToEtcd")
+	if config.Services == nil || len(config.Services) < 1 {
+		fmt.Println("services not found - skipping")
+		return blocks
+	}
+
+	for serviceName, _ := range config.Services {
+		if config.Services[serviceName].Labels == nil {
+			fmt.Println("sevice, but no labels found - continuing")
+			continue
+		}
+
+		for key, val := range config.Services[serviceName].Labels {
+			blocks = append(blocks, etcd.EtcdPacket{
+				Key:   dockerLabelToEtcdKey(key),
+				Value: val,
+			})
+		}
+	}
+
+	return blocks
+}
+
+func DockerToEtcd(config *types.Project, packetList *[]etcd.EtcdPacket) {
+	items := createPacket2(config)
+
+	*packetList = append(*packetList, items...)
+}

generator/traefik.go

@@ -0,0 +1,83 @@
+package generator
+
+import (
+	"fmt"
+	"log/slog"
+	"math"
+	"strconv"
+
+	"github.com/kevinmidboe/traefik-etcd-advertiser/client/etcd"
+	"github.com/kevinmidboe/traefik-etcd-advertiser/converter"
+	"github.com/traefik/traefik/v3/pkg/config/dynamic"
+)
+
+const traefikPrefix = "traefik"
+
+func isKnownGenericType(value interface{}) bool {
+	switch v := value.(type) {
+	case string:
+		return true
+	case float64:
+		return true
+	case bool:
+		return true
+	default:
+		slog.Debug(fmt.Sprintf("found unknown generic %s\n", v))
+	}
+
+	return false
+}
+
+func convertToGeneric(value interface{}) string {
+	switch v := value.(type) {
+	case string:
+		return v
+	case float64:
+		return fmt.Sprintf("%d", int(math.Floor(v)))
+	case bool:
+		return strconv.FormatBool(v)
+	}
+
+	return "unknown type"
+}
+
+// recursively walks the JSON object and creates internal
+// `etcdPackets` per leaf node, returns list of packets.
+func createPacket(item interface{}, parentKey string) []etcd.EtcdPacket {
+	blocks := []etcd.EtcdPacket{}
+
+	switch itemD := item.(type) {
+	// input is JSON object
+	case map[string]interface{}:
+		for key, value := range itemD {
+			// check for generic value type vs nested object,
+			// either create block or recursively call obj again
+
+			if isKnownGenericType(value) {
+				blocks = append(blocks, etcd.EtcdPacket{
+					Key:   parentKey + "/" + key,
+					Value: convertToGeneric(value),
+				})
+			} else {
+				blocks = append(blocks, createPacket(itemD[key], fmt.Sprintf("%s/%s", parentKey, key))...)
+			}
+		}
+	// input is JSON list
+	case []interface{}:
+		for i, item := range itemD {
+			blocks = append(blocks, createPacket(item, parentKey+"/"+strconv.Itoa(i))...)
+		}
+	}
+
+	return blocks
+}
+
+func TraefikToEtcd(config *dynamic.Configuration, packetList *[]etcd.EtcdPacket) {
+	// always convert to json before converting to etcd
+	data := converter.TraefikToJSON(config)
+
+	// generate list of etcd commands
+	items := createPacket(data, traefikPrefix)
+	*packetList = append(*packetList, items...)
+}
+