This commit is contained in:
Alessandro (Ale) Segala 2024-11-13 09:36:21 -08:00 committed by GitHub
commit bdfda6150f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 6 deletions

View File

@ -1644,17 +1644,27 @@ func (v *Viper) writeConfig(filename string, force bool) error {
if v.config == nil {
v.config = make(map[string]any)
}
flags := os.O_CREATE | os.O_TRUNC | os.O_WRONLY
if !force {
flags |= os.O_EXCL
}
f, err := v.fs.OpenFile(filename, flags, v.configPermissions)
exists, err := afero.Exists(v.fs, filename)
if err != nil {
return err
}
if exists && !force {
return fmt.Errorf("file already exists: %s", filename)
}
f, err := afero.TempFile(v.fs, filepath.Dir(filename), filepath.Base(filename)+"-")
if err != nil {
return err
}
defer f.Close()
if err := v.marshalWriter(f, configType); err != nil {
f.Close()
return err
}
f.Close()
err = v.fs.Rename(f.Name(), filename)
if err != nil {
return err
}