traefik/cmd/storeconfig/storeconfig.go

package storeconfig

import (
	"encoding/json"
	"fmt"
	stdlog "log"

	"github.com/abronan/valkeyrie/store"
	"github.com/containous/flaeg"
	"github.com/containous/staert"
	"github.com/containous/traefik/cmd"
)

// NewCmd builds a new StoreConfig command
func NewCmd(traefikConfiguration *cmd.TraefikConfiguration, traefikPointersConfiguration *cmd.TraefikConfiguration) *flaeg.Command {
	return &flaeg.Command{
		Name:                  "storeconfig",
		Description:           `Stores the static traefik configuration into a Key-value stores. Traefik will not start.`,
		Config:                traefikConfiguration,
		DefaultPointersConfig: traefikPointersConfiguration,
		Metadata: map[string]string{
			"parseAllSources": "true",
		},
	}
}

// Run store config in KV
func Run(kv *staert.KvSource, traefikConfiguration *cmd.TraefikConfiguration) func() error {
	return func() error {
		if kv == nil {
			return fmt.Errorf("error using command storeconfig, no Key-value store defined")
		}

		fileConfig := traefikConfiguration.Providers.File
		if fileConfig != nil {
			traefikConfiguration.Providers.File = nil
			if len(fileConfig.Filename) == 0 && len(fileConfig.Directory) == 0 {
				fileConfig.Filename = traefikConfiguration.ConfigFile
			}
		}

		jsonConf, err := json.Marshal(traefikConfiguration.Configuration)
		if err != nil {
			return err
		}
		stdlog.Printf("Storing configuration: %s\n", jsonConf)

		err = kv.StoreConfig(traefikConfiguration.Configuration)
		if err != nil {
			return err
		}

		if fileConfig != nil {
			jsonConf, err = json.Marshal(fileConfig)
			if err != nil {
				return err
			}

			stdlog.Printf("Storing file configuration: %s\n", jsonConf)
			config, err := fileConfig.BuildConfiguration()
			if err != nil {
				return err
			}

			stdlog.Print("Writing config to KV")
			err = kv.StoreConfig(config)
			if err != nil {
				return err
			}
		}

		// if traefikConfiguration.Configuration.ACME != nil {
		// 	account := &acme.Account{}
		//
		// 	accountInitialized, err := keyExists(kv, traefikConfiguration.Configuration.ACME.Storage)
		// 	if err != nil && err != store.ErrKeyNotFound {
		// 		return err
		// 	}
		//
		// 	// Check to see if ACME account object is already in kv store
		// 	if traefikConfiguration.Configuration.ACME.OverrideCertificates || !accountInitialized {
		//
		// 		// Stores the ACME Account into the KV Store
		// 		// Certificates in KV Stores will be overridden
		// 		meta := cluster.NewMetadata(account)
		// 		err = meta.Marshall()
		// 		if err != nil {
		// 			return err
		// 		}
		//
		// 		source := staert.KvSource{
		// 			Store:  kv,
		// 			Prefix: traefikConfiguration.Configuration.ACME.Storage,
		// 		}
		//
		// 		err = source.StoreConfig(meta)
		// 		if err != nil {
		// 			return err
		// 		}
		// 	}
		// }
		return nil
	}
}

// func keyExists(source *staert.KvSource, key string) (bool, error) {
// 	list, err := source.List(key, nil)
// 	if err != nil {
// 		return false, err
// 	}
//
// 	return len(list) > 0, nil
// }

// CreateKvSource creates KvSource
// TLS support is enable for Consul and Etcd backends
func CreateKvSource(traefikConfiguration *cmd.TraefikConfiguration) (*staert.KvSource, error) {
	var kv *staert.KvSource
	var kvStore store.Store
	var err error

	switch {
	case traefikConfiguration.Providers.Consul != nil:
		kvStore, err = traefikConfiguration.Providers.Consul.CreateStore()
		kv = &staert.KvSource{
			Store:  kvStore,
			Prefix: traefikConfiguration.Providers.Consul.Prefix,
		}
	case traefikConfiguration.Providers.Etcd != nil:
		kvStore, err = traefikConfiguration.Providers.Etcd.CreateStore()
		kv = &staert.KvSource{
			Store:  kvStore,
			Prefix: traefikConfiguration.Providers.Etcd.Prefix,
		}
	case traefikConfiguration.Providers.Zookeeper != nil:
		kvStore, err = traefikConfiguration.Providers.Zookeeper.CreateStore()
		kv = &staert.KvSource{
			Store:  kvStore,
			Prefix: traefikConfiguration.Providers.Zookeeper.Prefix,
		}
	case traefikConfiguration.Providers.Boltdb != nil:
		kvStore, err = traefikConfiguration.Providers.Boltdb.CreateStore()
		kv = &staert.KvSource{
			Store:  kvStore,
			Prefix: traefikConfiguration.Providers.Boltdb.Prefix,
		}
	}
	return kv, err
}