feat(格式配置): 支持格式化配置

goini.go

@@ -21,7 +21,6 @@ var (
 
 // GoINI 表示INI配置解析器
 type GoINI struct {
-	data                  []byte
 	filename              string
 	fileLines             []iniLine
 	structTag             string
@@ -43,8 +42,7 @@ func NewGoINI() *GoINI {
 // 入参: fileData 配置文件内容
 func (ini *GoINI) SetData(fileData []byte) {
 	ini.filename = ""
-	ini.data = append(ini.data[:0], fileData...)
-	ini.parse()
+	ini.parse(string(fileData))
 }
 
 // LoadFile 从文件读取配置并解析
@@ -56,28 +54,76 @@ func (ini *GoINI) LoadFile(filename string) error {
 		return err
 	}
 	ini.filename = filename
-	ini.data = append(ini.data[:0], b...)
-	ini.parse()
+	ini.parse(string(b))
 	return nil
 }
 
 // Save 将配置写回LoadFile读取的文件
 // 返回: error 错误信息
 func (ini *GoINI) Save() error {
-	if ini.filename == "" {
-		return errFilenameRequired
+	return ini.save(ini.String())
+}
+
+// SaveFormat 将配置格式化并写回LoadFile读取的文件
+// 返回: error 错误信息
+func (ini *GoINI) SaveFormat() error {
+	formatted := ini.formatString()
+	if err := ini.save(formatted); err != nil {
+		return err
 	}
-	return ini.SaveFile(ini.filename)
+	ini.parse(formatted)
+	return nil
 }
 
 // SaveFile 将配置写入指定文件
-// 入参: filename 配置文件路径
+// 入参: filename 配置文件路径, perm 文件权限
 // 返回: error 错误信息
-func (ini *GoINI) SaveFile(filename string) error {
+func (ini *GoINI) SaveFile(filename string, perm os.FileMode) error {
+	return ini.saveFile(filename, ini.String(), perm)
+}
+
+// SaveFormatFile 将配置格式化并写入指定文件
+// 入参: filename 配置文件路径, perm 文件权限
+// 返回: error 错误信息
+func (ini *GoINI) SaveFormatFile(filename string, perm os.FileMode) error {
+	formatted := ini.formatString()
+	if err := ini.saveFile(filename, formatted, perm); err != nil {
+		return err
+	}
+	ini.parse(formatted)
+	return nil
+}
+
+func (ini *GoINI) save(data string) error {
+	if ini.filename == "" {
+		return errFilenameRequired
+	}
+	info, err := os.Stat(ini.filename)
+	if err != nil {
+		return err
+	}
+	return ini.saveFile(ini.filename, data, info.Mode().Perm())
+}
+
+func (ini *GoINI) saveFile(filename string, data string, perm os.FileMode) error {
 	if filename == "" {
 		return errFilenameRequired
 	}
-	if err := os.WriteFile(filename, []byte(ini.String()), 0o666); err != nil {
+	perm = perm.Perm()
+	writePerm := perm | 0o200
+	if info, err := os.Stat(filename); err == nil {
+		if info.Mode().IsRegular() {
+			if err := os.Chmod(filename, writePerm); err != nil {
+				return err
+			}
+		}
+	} else if !errors.Is(err, os.ErrNotExist) {
+		return err
+	}
+	if err := os.WriteFile(filename, []byte(data), writePerm); err != nil {
+		return err
+	}
+	if err := os.Chmod(filename, perm); err != nil {
 		return err
 	}
 	ini.filename = filename