Moved config set/unset to internal/config/modify.go as Store and Purge.
This commit is contained in:
1 parent
539e88e73c
commit
09644d0691
4 files changed
+102
-67
No files matched your search
+10
-64
@@ -4,8 +4,6 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"reflect"
|
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/cli"
|
"git.avsdev.uk/Go/job-search/internal/cli"
|
||||||
"git.avsdev.uk/Go/job-search/internal/config"
|
"git.avsdev.uk/Go/job-search/internal/config"
|
||||||
@@ -13,65 +11,6 @@ import (
|
|||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
)
|
)
|
||||||
|
|
||||||
func setConfig(conf *config.JobSearchConfig, key string, value string) error {
|
|
||||||
var confValue any = nil
|
|
||||||
var confType reflect.Kind
|
|
||||||
var err error
|
|
||||||
|
|
||||||
confType, err = conf.TypeOf(key)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
switch confType {
|
|
||||||
case reflect.Bool:
|
|
||||||
confValue, err = strconv.ParseBool(value)
|
|
||||||
case reflect.String:
|
|
||||||
confValue = value
|
|
||||||
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
||||||
confValue, err = strconv.ParseInt(value, 10, 64)
|
|
||||||
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
||||||
confValue, err = strconv.ParseUint(value, 10, 64)
|
|
||||||
case reflect.Float32, reflect.Float64:
|
|
||||||
confValue, err = strconv.ParseFloat(value, 64)
|
|
||||||
default:
|
|
||||||
err = errors.ErrUnsupported
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("config value type not accepted, expected <%s>", confType)
|
|
||||||
}
|
|
||||||
|
|
||||||
viper.Set(key, confValue)
|
|
||||||
if err := viper.WriteConfig(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
fmt.Print(key, " = ", confValue, " <", confType, ">\n")
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
func unsetConfig(conf *config.JobSearchConfig, key string) error {
|
|
||||||
if _, err := conf.TypeOf(key); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
v := viper.New()
|
|
||||||
v.SetConfigFile(viper.ConfigFileUsed())
|
|
||||||
|
|
||||||
keys := viper.AllKeys()
|
|
||||||
for i := range keys {
|
|
||||||
k := keys[i]
|
|
||||||
if k != key {
|
|
||||||
v.Set(k, viper.Get(k))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := v.WriteConfig(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func commandSet(opts *cli.Options) *cobra.Command {
|
func commandSet(opts *cli.Options) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "set [config_key] [config_value]",
|
Use: "set [config_key] [config_value]",
|
||||||
@@ -111,10 +50,17 @@ func commandSet(opts *cli.Options) *cobra.Command {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if unset {
|
if unset {
|
||||||
return unsetConfig(conf, args[0])
|
return config.PurgeValue(conf, args[0])
|
||||||
} else {
|
|
||||||
return setConfig(conf, args[0], args[1])
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
confValue, confType, err := config.StoreValue(conf, args[0], args[1])
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Print(args[0], " = ", confValue, " <", confType, ">\n")
|
||||||
|
|
||||||
|
return nil
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
cmd.Flags().BoolP("unset", "x", false, "clear configuration setting")
|
cmd.Flags().BoolP("unset", "x", false, "clear configuration setting")
|
||||||
|
|||||||
@@ -54,16 +54,20 @@ func LoadFromViper(conf *JobSearchConfig) {
|
|||||||
if keyType == reflect.Invalid {
|
if keyType == reflect.Invalid {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
var err error
|
||||||
switch keyType {
|
switch keyType {
|
||||||
case reflect.String:
|
case reflect.String:
|
||||||
conf.SetValue(key, viper.GetString(key))
|
err = conf.SetValue(key, viper.GetString(key))
|
||||||
case reflect.Int:
|
case reflect.Int:
|
||||||
conf.SetValue(key, viper.GetInt(key))
|
err = conf.SetValue(key, viper.GetInt(key))
|
||||||
case reflect.Bool:
|
case reflect.Bool:
|
||||||
conf.SetValue(key, viper.GetBool(key))
|
err = conf.SetValue(key, viper.GetBool(key))
|
||||||
default:
|
default:
|
||||||
slog.Error(fmt.Sprintf("unknown key type: %s", keyType))
|
slog.Error(fmt.Sprintf("unknown key type: %s", keyType))
|
||||||
}
|
}
|
||||||
|
if err != nil {
|
||||||
|
slog.Warn("ignoring invalid config value", "key", key, "err", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package config
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/joho/godotenv"
|
"github.com/joho/godotenv"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
@@ -33,6 +34,7 @@ func loadConfig(cfgFile string) error {
|
|||||||
LoadDefaultConfig()
|
LoadDefaultConfig()
|
||||||
|
|
||||||
viper.SetEnvPrefix(EnvPrefix)
|
viper.SetEnvPrefix(EnvPrefix)
|
||||||
|
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
||||||
viper.AutomaticEnv() // read in environment variables that match
|
viper.AutomaticEnv() // read in environment variables that match
|
||||||
|
|
||||||
err := viper.ReadInConfig()
|
err := viper.ReadInConfig()
|
||||||
|
|||||||
@@ -0,0 +1,83 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
)
|
||||||
|
|
||||||
|
// StoreValue parses value as the type of key, checks conf accepts it, sets it
|
||||||
|
// in conf and Viper and writes the config file Viper is using. It returns the
|
||||||
|
// parsed value and its kind.
|
||||||
|
func StoreValue(
|
||||||
|
conf *JobSearchConfig, key string, value string,
|
||||||
|
) (any, reflect.Kind, error) {
|
||||||
|
var confValue any = nil
|
||||||
|
var confType reflect.Kind
|
||||||
|
var err error
|
||||||
|
|
||||||
|
confType, err = conf.TypeOf(key)
|
||||||
|
if err != nil {
|
||||||
|
return nil, confType, err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch confType {
|
||||||
|
case reflect.Bool:
|
||||||
|
confValue, err = strconv.ParseBool(value)
|
||||||
|
case reflect.String:
|
||||||
|
confValue = value
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
confValue, err = strconv.ParseInt(value, 10, 64)
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||||
|
confValue, err = strconv.ParseUint(value, 10, 64)
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
confValue, err = strconv.ParseFloat(value, 64)
|
||||||
|
default:
|
||||||
|
err = errors.ErrUnsupported
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, confType, fmt.Errorf(
|
||||||
|
"config value type not accepted, expected <%s>", confType,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reject values the configuration itself would not accept, e.g. an
|
||||||
|
// unknown log level or an unparseable duration.
|
||||||
|
if err := conf.SetValue(key, confValue); err != nil {
|
||||||
|
return nil, confType, err
|
||||||
|
}
|
||||||
|
|
||||||
|
viper.Set(key, confValue)
|
||||||
|
if err := viper.WriteConfig(); err != nil {
|
||||||
|
return nil, confType, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return confValue, confType, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// PurgeValue rewrites the config file Viper is using without key.
|
||||||
|
func PurgeValue(conf *JobSearchConfig, key string) error {
|
||||||
|
if _, err := conf.TypeOf(key); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
v := viper.New()
|
||||||
|
v.SetConfigFile(viper.ConfigFileUsed())
|
||||||
|
|
||||||
|
keys := viper.AllKeys()
|
||||||
|
for i := range keys {
|
||||||
|
k := keys[i]
|
||||||
|
if k != key {
|
||||||
|
v.Set(k, viper.Get(k))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := v.WriteConfig(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Reference in new issue
Block a user