goini/goini.go

package goini

import (
	"bytes"
	"errors"
	"io/ioutil"
	"reflect"
	"regexp"
	"strconv"
	"strings"
)

// GoINI GoINI数据结构
type GoINI struct {
	tag         string
	data        []byte
	dataMap     map[string]map[string]string
	nameList    []string
	dataKeyList map[string][]string
	commonField string
}

// NewGoINI 获取GoINI对象
func NewGoINI() *GoINI {
	return &GoINI{
		tag:         "goini",
		commonField: "BuiltCommon",
	}
}

// parse ini配置文件解析器
func (ini *GoINI) parse() {
	currentName := ini.commonField
	iniData := make(map[string]map[string]string)
	iniData[currentName] = make(map[string]string)
	ini.nameList = append(ini.nameList, currentName)
	ini.dataKeyList = make(map[string][]string)
	for _, v := range bytes.Split(ini.data, []byte("\n")) {
		line := bytes.TrimSpace(v)
		lineLen := len(line)
		if line == nil || lineLen < 3 ||
			line[0] == 35 || line[0] == 59 {
			continue
		}
		if line[0] == 91 && line[lineLen-1] == 93 {
			name := bytes.TrimSpace(line[1 : lineLen-1])
			if name != nil {
				currentName = string(name)
				if _, ok := iniData[currentName]; !ok {
					iniData[currentName] = make(map[string]string)
					ini.nameList = append(ini.nameList, currentName)
				}
			}
			continue
		}
		split := bytes.IndexByte(line, 61)
		if split == -1 {
			split = bytes.IndexByte(line, 32)
			if split == -1 {
				continue
			}
		}
		key := bytes.TrimSpace(line[0:split])
		value := bytes.TrimSpace(line[split+1:])
		valueLen := len(value)
		keyStr, valueStr := string(key), string(value)
		if keyStr == "" {
			continue
		}
		if _, ok := iniData[currentName][keyStr]; ok {
			continue
		}
		ini.dataKeyList[currentName] = append(ini.dataKeyList[currentName], keyStr)
		if value == nil {
			iniData[currentName][keyStr] = ""
			continue
		}
		if valueLen >= 2 &&
			((value[0] == 34 && value[valueLen-1] == 34) ||
				(value[0] == 39 && value[valueLen-1] == 39) ||
				(value[0] == 96 && value[valueLen-1] == 96)) {
			iniData[currentName][keyStr] = string(value[1 : valueLen-1])
			continue
		}
		iniData[currentName][keyStr] = valueStr
	}
	ini.dataMap = iniData
}

// String 获取GoINI对象字符串形式
func (ini *GoINI) String() string {
	var iniLines []string
	for _, name := range ini.GetNames("") {
		if name != ini.commonField {
			if len(iniLines) < 1 {
				iniLines = append(iniLines, "["+name+"]")
			} else {
				iniLines = append(iniLines, "\n["+name+"]")
			}
		}
		for _, key := range ini.GetNameKeys(name, "") {
			nameValue := ini.GetString(name, key, "")
			if ok, _ := regexp.MatchString("^\\s|\\s$", nameValue); ok {
				iniLines = append(iniLines, key+" = \""+nameValue+"\"")
				continue
			}
			nameValueLen := len(nameValue)
			if nameValueLen >= 2 &&
				((nameValue[0] == 34 && nameValue[nameValueLen-1] == 34) ||
					(nameValue[0] == 39 && nameValue[nameValueLen-1] == 39) ||
					(nameValue[0] == 96 && nameValue[nameValueLen-1] == 96)) {
				tag := `"`
				if nameValue[0] == 34 {
					tag = "`"
				}
				iniLines = append(iniLines, key+" = "+tag+nameValue+tag)
			} else {
				iniLines = append(iniLines, key+" = "+nameValue)
			}
		}
	}
	return strings.Join(iniLines, "\n")
}

// SetTag 设置结构体的tag键名称
func (ini *GoINI) SetTag(tag string) {
	ini.tag = tag
}

// SetData 从代码读取配置并解析
func (ini *GoINI) SetData(fileData []byte) {
	ini.data = bytes.TrimSpace(fileData)
	ini.parse()
}

// LoadFile 从文件读取配置并解析
func (ini *GoINI) LoadFile(fileName string) error {
	b, err := ioutil.ReadFile(fileName)
	if err != nil {
		return err
	}
	ini.data = bytes.TrimSpace(b)
	ini.parse()
	return nil
}

// GetNames 获取配置文件分区列表
func (ini *GoINI) GetNames(match string) []string {
	if match == "" {
		return ini.nameList
	}
	var matchNameList []string
	for _, name := range ini.nameList {
		if b, e := regexp.MatchString(match, name); e == nil && b {
			matchNameList = append(matchNameList, name)
		}
	}
	return matchNameList
}

// GetNameKeys 获取分区下配置项列表
func (ini *GoINI) GetNameKeys(name string, match string) []string {
	var keyList []string
	if name == "" {
		name = ini.commonField
	}
	if keyList, ok := ini.dataKeyList[name]; ok {
		if match == "" {
			return keyList
		}
		var matchKeyList []string
		for _, key := range keyList {
			if b, e := regexp.MatchString(match, key); e == nil && b {
				matchKeyList = append(matchKeyList, key)
			}
		}
		return matchKeyList
	}
	return keyList
}

// GetBool 获取单个配置项的布尔值
func (ini *GoINI) GetBool(name string, key string, value bool) bool {
	boolStr := ini.GetString(name, key, "")
	switch strings.ToLower(boolStr) {
	case "y", "yes", "on":
		return true
	case "n", "no", "off":
		return false
	}
	if b, e := strconv.ParseBool(boolStr); e == nil {
		return b
	}
	return value
}

// GetInt64 获取单个配置项的数字值
func (ini *GoINI) GetInt64(name string, key string, value int64) int64 {
	int64Str := ini.GetString(name, key, "")
	if i, e := strconv.ParseInt(int64Str, 10, 64); e == nil {
		return i
	}
	return value
}

// GetString 获取单个配置项的字符值
func (ini *GoINI) GetString(name string, key string, value string) string {
	if 0 == len(name) {
		name = ini.commonField
	}
	if v, ok := ini.dataMap[name]; ok {
		if vv, ok := v[key]; ok {
			return vv
		}
	}
	return value
}

// GetFloat64 获取单个配置项的小数值
func (ini *GoINI) GetFloat64(name string, key string, value float64) float64 {
	float64Str := ini.GetString(name, key, "")
	if f, e := strconv.ParseFloat(float64Str, 64); e == nil {
		return f
	}
	return value
}

// MapToStruct 将配置映射到一个结构体
func (ini *GoINI) MapToStruct(ptr interface{}) (err error) {
	t := reflect.TypeOf(ptr)
	v := reflect.ValueOf(ptr)
	if t.Kind() != reflect.Ptr {
		err = errors.New("input struct ptr")
		return
	}
	t = t.Elem()
	v = v.Elem()
	for i := 0; i < t.NumField(); i++ {
		if !v.CanInterface() {
			continue
		}
		k := t.Field(i).Tag.Get(ini.tag)
		if k == "" {
			k = t.Field(i).Name
		}
		switch v.Field(i).Kind() {
		case reflect.Struct:
			tt := v.Field(i).Type()
			vv := v.Field(i)
			for ii := 0; ii < tt.NumField(); ii++ {
				if !v.CanInterface() {
					continue
				}
				kk := tt.Field(ii).Tag.Get(ini.tag)
				if kk == "" {
					kk = t.Field(ii).Name
				}
				switch vv.Field(ii).Kind() {
				case reflect.Bool:
					vv.Field(ii).SetBool(ini.GetBool(k, kk, false))
				case reflect.String:
					vv.Field(ii).SetString(ini.GetString(k, kk, ""))
				case reflect.Float32, reflect.Float64:
					vv.Field(ii).SetFloat(ini.GetFloat64(k, kk, 0))
				case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
					vv.Field(ii).SetInt(ini.GetInt64(k, kk, 0))
				case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
					vv.Field(ii).SetUint(uint64(ini.GetInt64(k, kk, 0)))
				}
			}
		case reflect.Bool:
			v.Field(i).SetBool(ini.GetBool(ini.commonField, k, false))
		case reflect.String:
			v.Field(i).SetString(ini.GetString(ini.commonField, k, ""))
		case reflect.Float32, reflect.Float64:
			v.Field(i).SetFloat(ini.GetFloat64(ini.commonField, k, 0))
		case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
			v.Field(i).SetInt(ini.GetInt64(ini.commonField, k, 0))
		case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
			v.Field(i).SetUint(uint64(ini.GetInt64(ini.commonField, k, 0)))
		}
	}
	return
}