forked from mystiq/dex
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/coreos/dex/pkg/log"
|
|
"github.com/coreos/go-oidc/oidc"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
)
|
|
|
|
var (
|
|
rootCmd = &cobra.Command{
|
|
Use: "dexctl",
|
|
Short: "A command line tool for interacting with the dex system",
|
|
Long: "",
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
// initialize flags from environment
|
|
fs := cmd.Flags()
|
|
|
|
// don't override flags set by command line flags
|
|
alreadySet := make(map[string]bool)
|
|
fs.Visit(func(f *pflag.Flag) { alreadySet[f.Name] = true })
|
|
|
|
var err error
|
|
fs.VisitAll(func(f *pflag.Flag) {
|
|
if err != nil || alreadySet[f.Name] {
|
|
return
|
|
}
|
|
key := "DEXCTL_" + strings.ToUpper(strings.Replace(f.Name, "-", "_", -1))
|
|
if val := os.Getenv(key); val != "" {
|
|
err = fs.Set(f.Name, val)
|
|
}
|
|
})
|
|
return err
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cmd.Help()
|
|
os.Exit(2)
|
|
},
|
|
}
|
|
|
|
global struct {
|
|
endpoint string
|
|
creds oidc.ClientCredentials
|
|
dbURL string
|
|
help bool
|
|
logDebug bool
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
log.EnableTimestamps()
|
|
|
|
rootCmd.PersistentFlags().StringVar(&global.endpoint, "endpoint", "", "URL of dex API")
|
|
rootCmd.PersistentFlags().StringVar(&global.creds.ID, "client-id", "", "dex API user ID")
|
|
rootCmd.PersistentFlags().StringVar(&global.creds.Secret, "client-secret", "", "dex API user password")
|
|
rootCmd.PersistentFlags().StringVar(&global.dbURL, "db-url", "", "DSN-formatted database connection string")
|
|
rootCmd.PersistentFlags().BoolVar(&global.logDebug, "log-debug", false, "Log debug-level information")
|
|
}
|
|
|
|
func main() {
|
|
if err := rootCmd.Execute(); err != nil {
|
|
os.Exit(2)
|
|
}
|
|
}
|
|
|
|
func wrapRun(run func(cmd *cobra.Command, args []string) int) func(cmd *cobra.Command, args []string) {
|
|
return func(cmd *cobra.Command, args []string) {
|
|
os.Exit(run(cmd, args))
|
|
}
|
|
}
|
|
|
|
func getDriver() (drv driver) {
|
|
var err error
|
|
switch {
|
|
case len(global.dbURL) > 0:
|
|
drv, err = newDBDriver(global.dbURL)
|
|
case len(global.endpoint) > 0:
|
|
if len(global.creds.ID) == 0 || len(global.creds.Secret) == 0 {
|
|
err = errors.New("--client-id/--client-secret flags unset")
|
|
break
|
|
}
|
|
pcfg, err := oidc.FetchProviderConfig(http.DefaultClient, global.endpoint)
|
|
if err != nil {
|
|
stderr("Unable to fetch provider config: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
drv, err = newAPIDriver(pcfg, global.creds)
|
|
default:
|
|
err = errors.New("--endpoint/--db-url flags unset")
|
|
}
|
|
|
|
if err != nil {
|
|
stderr("Unable to configure dexctl driver: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return
|
|
}
|