102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/utils"
|
||
|
|
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/services"
|
||
|
|
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/services/app"
|
||
|
|
"github.com/dexidp/dex/storage"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
appID string
|
||
|
|
appClientID string
|
||
|
|
appClientSecret string
|
||
|
|
appName string
|
||
|
|
appRedirectURIs []string
|
||
|
|
appInteractive bool
|
||
|
|
)
|
||
|
|
|
||
|
|
var appAddCmd = &cobra.Command{
|
||
|
|
Use: "add",
|
||
|
|
Short: "Add a new app to the storage",
|
||
|
|
Long: `Add a new app to the storage.
|
||
|
|
|
||
|
|
Parameters to provide:
|
||
|
|
- id: Unique ID to represent the app in the storage
|
||
|
|
- name: Human readable name to represent the app.
|
||
|
|
- redirect-uri: list of allowed redirection URIs for this app
|
||
|
|
|
||
|
|
Optional parameters:
|
||
|
|
- client-id: Client ID used by the OpenIDConnect protocol, automatically generated if not provided
|
||
|
|
- client-secret: Client secret used by the OpenIDConnect protocol, automatically generated if not provided
|
||
|
|
- interactive: Pass this parameter to use a prompt to pass unset parameters (client id and secret)`,
|
||
|
|
Run: func(cmd *cobra.Command, args []string) {
|
||
|
|
addNewApp()
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
func generateSecret(interactive bool, currentValue, valueName string) (string, error) {
|
||
|
|
if currentValue != "" {
|
||
|
|
return currentValue, nil
|
||
|
|
}
|
||
|
|
if !interactive {
|
||
|
|
val, err := services.GenerateRandomHex(services.IDSecretSize)
|
||
|
|
if err != nil {
|
||
|
|
return "", fmt.Errorf("failed to generate %s: %w", valueName, err)
|
||
|
|
}
|
||
|
|
return val, nil
|
||
|
|
}
|
||
|
|
fmt.Printf("Enter value for %s, use an empty value to automatically generate it.\n", valueName)
|
||
|
|
var enteredVal string
|
||
|
|
fmt.Scanln(&enteredVal)
|
||
|
|
if enteredVal == "" {
|
||
|
|
return generateSecret(false, currentValue, valueName)
|
||
|
|
}
|
||
|
|
return enteredVal, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func addNewApp() {
|
||
|
|
c := utils.InitConfig("")
|
||
|
|
s := utils.InitStorage(c)
|
||
|
|
|
||
|
|
clientID, err := generateSecret(appInteractive, appClientID, "client ID")
|
||
|
|
if err != nil {
|
||
|
|
utils.Fail(err.Error())
|
||
|
|
}
|
||
|
|
clientSecret, err := generateSecret(appInteractive, appClientSecret, "client secret")
|
||
|
|
if err != nil {
|
||
|
|
utils.Fail(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
appConf := storage.Client{
|
||
|
|
ID: clientID,
|
||
|
|
Secret: clientSecret,
|
||
|
|
Name: appName,
|
||
|
|
RedirectURIs: appRedirectURIs,
|
||
|
|
}
|
||
|
|
if err := app.New(s).AddApp(appConf); err != nil {
|
||
|
|
utils.Failf("Failed to add new app to storage: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("New app %s added.\n", appName)
|
||
|
|
printProperty("Client ID", clientID, 1)
|
||
|
|
printProperty("Client secret", clientSecret, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
appCmd.AddCommand(appAddCmd)
|
||
|
|
|
||
|
|
appAddCmd.Flags().StringVarP(&appName, "name", "n", "", "Name to represent the app")
|
||
|
|
appAddCmd.Flags().StringVarP(&appClientID, "id", "i", "", "ID to identify the app in the storage")
|
||
|
|
appAddCmd.Flags().StringVarP(&appClientSecret, "secret", "s", "", "OpenIDConnect client secret")
|
||
|
|
appAddCmd.Flags().StringSliceVarP(&appRedirectURIs, "redirect-uri", "r", []string{}, "Allowed redirect URI")
|
||
|
|
|
||
|
|
appAddCmd.Flags().BoolVar(&appInteractive, "interactive", false, "Set the client ID and secret in an interactive way")
|
||
|
|
|
||
|
|
appAddCmd.MarkFlagRequired("name")
|
||
|
|
appAddCmd.MarkFlagRequired("redirect-uri")
|
||
|
|
}
|