writeini.go

package goconfig

import (
	"fmt"
	"io/ioutil"
)

// 格式文件, sit 0, 1 ,
func writeFile(key, value, module string, notes ...string) {
	// 判断是不是组
	if module == "" {
		//先添加key
		kvconfig.newKeyValue(key, value, notes...)
	} else {
		// 组
		for i, g := range kvconfig.Groups {
			if string(g.name) == module {
				kvconfig.addGroupKeyValue(i, key, value, notes...)
				return
			}
		}
		// 不存在就新建
		kvconfig.newGroupKeyValue(module, key, value, notes...)

	}
}

func FlushWrite() {
	content := getWrite()
	if err := ioutil.WriteFile(kvconfig.Filepath, content, 0644); err != nil {
		panic(err)
	}
}

func getWrite() []byte {
	kvconfig.Write = nil
	for _, v := range kvconfig.Lines {
		// 打印注释
		for _, n := range v.note {
			line := fmt.Sprintf("%s %s\n", WELL, string(n))
			kvconfig.Write = append(kvconfig.Write, []byte(line)...)
		}
		// 打印kv
		kv := fmt.Sprintf("%s = %s\n", v.key, string(v.value))
		kvconfig.Write = append(kvconfig.Write, []byte(kv)...)
	}
	for _, v := range kvconfig.Groups {
		// 模块添加换行
		kvconfig.Write = append(kvconfig.Write, []byte("\n")...)
		// 打印组注释
		for _, n := range v.note {
			line := fmt.Sprintf("%s %s\n", WELL, string(n))
			kvconfig.Write = append(kvconfig.Write, []byte(line)...)
		}
		// 打印组
		g := fmt.Sprintf("%s%s%s\n", MODEL_START, string(v.name), MODEL_END)
		kvconfig.Write = append(kvconfig.Write, []byte(g)...)
		for _, gn := range v.group {
			// 组key 注释
			for _, nn := range gn.note {
				line := fmt.Sprintf("%s %s\n", WELL, string(nn))
				kvconfig.Write = append(kvconfig.Write, []byte(line)...)
			}
			// 打印kv
			kv := fmt.Sprintf("%s = %s\n", gn.key, string(gn.value))
			kvconfig.Write = append(kvconfig.Write, []byte(kv)...)
		}
	}
	return kvconfig.Write
}

func GetBytesAndClear() []byte {
	defer func() { kvconfig = nil }()
	return getWrite()
}