fix: only write yml when origin format is yaml

boypt · Aug 16, 2021 · 06cf167 · 06cf167
1 parent 246f012
commit 06cf167
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/engine/config.go b/engine/config.go
@@ -6,6 +6,7 @@ import (
 	"os"
 	"path/filepath"
 	"reflect"
+	"strings"
 	"time"
 
 	"github.com/spf13/viper"
@@ -222,7 +223,10 @@ func (c *Config) SyncViper(nc Config) {
 
 func (c *Config) WriteDefault() error {
 	cf := viper.ConfigFileUsed()
-	return c.WriteYaml(cf)
+	if strings.HasSuffix(cf, ".yml") || strings.HasSuffix(cf, ".yaml") {
+		return c.WriteYaml(cf)
+	}
+	return viper.WriteConfig()
 }
 
 func (c *Config) WriteYaml(cf string) error {