dex/cmd/dexctl/command_config.go

package main

import (
	"fmt"
	"io"
	"os"

	"github.com/coreos/dex/connector"
	"github.com/spf13/cobra"
)

var (
	cmdGetConnectorConfigs = &cobra.Command{
		Use:     "get-connector-configs",
		Short:   "Enumerate current IdP connector configs.",
		Long:    "Enumerate current IdP connector configs.",
		Example: `  dexctl get-connector-configs --db-url=${DB_URL}`,
		Run:     wrapRun(runGetConnectorConfigs),
	}

	cmdSetConnectorConfigs = &cobra.Command{
		Use:     "set-connector-configs",
		Short:   "Overwrite the current IdP connector configs with those from a local file. Provide the argument '-' to read from stdin.",
		Long:    "Overwrite the current IdP connector configs with those from a local file. Provide the argument '-' to read from stdin.",
		Example: `  dexctl set-connector-configs --db-url=${DB_URL} ./static/conn_conf.json`,
		Run:     wrapRun(runSetConnectorConfigs),
	}
)

func init() {
	rootCmd.AddCommand(cmdGetConnectorConfigs)
	rootCmd.AddCommand(cmdSetConnectorConfigs)
}

func runSetConnectorConfigs(cmd *cobra.Command, args []string) int {
	if len(args) != 1 {
		stderr("Provide a single argument.")
		return 2
	}

	var r io.Reader
	if from := args[0]; from == "-" {
		r = os.Stdin
	} else {
		f, err := os.Open(from)
		if err != nil {
			stderr("Unable to open specified file: %v", err)
			return 1
		}
		defer f.Close()
		r = f
	}

	cfgs, err := connector.ReadConfigs(r)
	if err != nil {
		stderr("Unable to decode connector configs: %v", err)
		return 1
	}

	if err := getDriver().SetConnectorConfigs(cfgs); err != nil {
		stderr(err.Error())
		return 1
	}

	fmt.Printf("Saved %d connector config(s)\n", len(cfgs))
	return 0
}

func runGetConnectorConfigs(cmd *cobra.Command, args []string) int {
	if len(args) != 0 {
		stderr("Provide zero arguments.")
		return 2
	}

	cfgs, err := getDriver().ConnectorConfigs()
	if err != nil {
		stderr("Unable to retrieve connector configs: %v", err)
		return 1
	}

	fmt.Printf("Found %d connector config(s)\n", len(cfgs))

	for _, cfg := range cfgs {
		fmt.Println()
		fmt.Printf("ID:   %v\n", cfg.ConnectorID())
		fmt.Printf("Type: %v\n", cfg.ConnectorType())
	}

	return 0
}