forked from mystiq/dex
91 lines
2.1 KiB
Go
91 lines
2.1 KiB
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
|
|
}
|