feat: add a config remover

mainawycliffe · Dec 10, 2019 · 9e38de6 · 9e38de6
1 parent 516d60a
commit 9e38de6
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/configs/viper.go b/configs/viper.go
@@ -0,0 +1,30 @@
+package configs
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/spf13/viper"
+)
+
+func UnsetViperConfig(keys ...string) error {
+	configMap := viper.AllSettings()
+	for _, key := range keys {
+		delete(configMap, strings.ToLower(key))
+	}
+	encodedConfig, err := json.MarshalIndent(configMap, "", " ")
+	if err != nil {
+		return err
+	}
+	err = viper.ReadConfig(bytes.NewReader(encodedConfig))
+	if err != nil {
+		return err
+	}
+	err = viper.WriteConfig()
+	if err != nil {
+		return fmt.Errorf("Error removing configs: %w", err)
+	}
+	return nil
+}