mirror of
https://gitlab.com/psuapp/psu.git
synced 2024-08-30 18:12:34 +00:00
99 lines
2.4 KiB
Go
99 lines
2.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/greenled/portainer-stack-utils/common"
|
|
"github.com/mitchellh/go-homedir"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
"os"
|
|
"sort"
|
|
)
|
|
|
|
// configCmd represents the config command
|
|
var configCmd = &cobra.Command{
|
|
Use: "config KEY [VALUE]",
|
|
Short: "Get and set configuration options",
|
|
Example: "psu config user admin",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if viper.GetBool("config.list") {
|
|
// List config keys
|
|
keys := viper.AllKeys()
|
|
sort.Slice(keys, func(i, j int) bool {
|
|
return keys[i] < keys[j]
|
|
})
|
|
for _, key := range keys {
|
|
fmt.Println(key)
|
|
}
|
|
} else {
|
|
if len(args) >= 1 {
|
|
// Check if it's a valid key
|
|
var keyExists bool
|
|
for _, key := range viper.AllKeys() {
|
|
if key == args[0] {
|
|
keyExists = true
|
|
break
|
|
}
|
|
}
|
|
if !keyExists {
|
|
log.Fatalf("Unkonwn configuration key \"%s\"", args[0])
|
|
}
|
|
|
|
// Create new viper
|
|
commandViper := viper.New()
|
|
|
|
// Set config file name
|
|
var configFile string
|
|
if viper.ConfigFileUsed() != "" {
|
|
// Use config file from viper
|
|
configFile = viper.ConfigFileUsed()
|
|
} else {
|
|
// Find home directory
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Use $HOME/.psu.yaml
|
|
configFile = fmt.Sprintf("%s%s.psu.yaml", home, string(os.PathSeparator))
|
|
}
|
|
commandViper.SetConfigFile(configFile)
|
|
|
|
// Read config from file
|
|
if configReadingErr := commandViper.ReadInConfig(); configReadingErr != nil {
|
|
common.PrintVerbose(fmt.Sprintf("Could not read configuration from \"%s\". Expect all configuration values to be unset.", configFile))
|
|
}
|
|
|
|
if len(args) == 1 {
|
|
// Get config
|
|
fmt.Println(commandViper.Get(args[0]))
|
|
} else {
|
|
// Set config
|
|
commandViper.Set(args[0], args[1])
|
|
|
|
// Make sure the config file exists
|
|
_, fileCreationErr := os.Create(configFile)
|
|
if fileCreationErr != nil {
|
|
common.CheckError(fileCreationErr)
|
|
}
|
|
|
|
// Write te config file
|
|
configWritingErr := commandViper.WriteConfig()
|
|
if configWritingErr != nil {
|
|
common.CheckError(configWritingErr)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(configCmd)
|
|
|
|
configCmd.Flags().BoolP("list", "l", false, "list all config keys")
|
|
viper.BindPFlag("config.list", configCmd.Flags().Lookup("list"))
|
|
}
|