From b3171c42b6d3bd8e7909d4dd837f264f49a8ac59 Mon Sep 17 00:00:00 2001
From: lin <ichenzhl@qq.com>
Date: Thu, 27 Jun 2024 18:06:10 +0800
Subject: [PATCH] add MarshalToString

---
 viper.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/viper.go b/viper.go
index d2f85d8..20b6a07 100644
--- a/viper.go
+++ b/viper.go
@@ -1601,6 +1601,11 @@ func (v *Viper) SafeWriteConfigAs(filename string) error {
 	return v.writeConfig(filename, false)
 }
 
+// MarshalToString writes current configuration to a string
+func MarshalToString(configTypeOps ...string) (string, error) {
+	return v.MarshalToString(configTypeOps...)
+}
+
 func (v *Viper) writeConfig(filename string, force bool) error {
 	v.logger.Info("attempting to write configuration to file")
 
@@ -1639,6 +1644,30 @@ func (v *Viper) writeConfig(filename string, force bool) error {
 	return f.Sync()
 }
 
+func (v *Viper) MarshalToString(configTypeOps ...string) (string, error) {
+	v.logger.Info("MarshalToString: attempting to marshal configuration to string")
+
+	//default by origin configType
+	var configType = v.configType
+	//set from param if configTypeOps len > 0
+	if len(configTypeOps) > 0 {
+		configType = configTypeOps[0]
+	}
+	if configType == "" {
+		return "", fmt.Errorf("MarshalToString: config type could not be determined")
+	}
+
+	if !stringInSlice(configType, SupportedExts) {
+		return "", UnsupportedConfigError(configType)
+	}
+
+	str, err := v.marshalToString(configType)
+	if err != nil {
+		return "", err
+	}
+	return str, nil
+}
+
 func (v *Viper) unmarshalReader(in io.Reader, c map[string]any) error {
 	buf := new(bytes.Buffer)
 	buf.ReadFrom(in)
@@ -1683,6 +1712,25 @@ func (v *Viper) marshalWriter(f afero.File, configType string) error {
 	return nil
 }
 
+// Marshal a map into string.
+func (v *Viper) marshalToString(configType string) (string, error) {
+	c := v.AllSettings()
+	switch configType {
+	case "yaml", "yml", "json", "toml", "hcl", "tfvars", "ini", "prop", "props", "properties", "dotenv", "env":
+		encoder, err := v.encoderRegistry.Encoder(configType)
+		if err != nil {
+			return "", ConfigMarshalError{err}
+		}
+
+		b, err := encoder.Encode(c)
+		if err != nil {
+			return "", ConfigMarshalError{err}
+		}
+		return string(b), nil
+	}
+	return "", ConfigMarshalError{fmt.Errorf("config type could not be determined, type:%s", configType)}
+}
+
 func keyExists(k string, m map[string]any) string {
 	lk := strings.ToLower(k)
 	for mk := range m {