iniutils for Golang

liwya

// Copyright 2017 ecofast(无尽愿). All rights reserved.　　
// Use of this source code is governed by a BSD-style license.
　　

　　
// Package iniutils was translated from TMemIniFile in Delphi(2007) RTL,
　　
// which loads an entire INI file into memory
　　
// and allows all operations to be performed on the memory image.
　　
// The image can then be written out to the disk file.
　　
package iniutils
　　

　　
import (
　　"bufio"
　　"bytes"
　　"fmt"
　　"os"
　　"strconv"
　　"strings"
　　

　　"github.com/ecofast/sysutils"
　　
)
　　

　　
type IniFile struct {
　　fileName      string
　　caseSensitive bool
　　sections      map[string][]string
　　
}
　　

　　
func NewIniFile(filename string, casesensitive bool) *IniFile {
　　ini := &IniFile{
　　fileName:      filename,
　　caseSensitive: casesensitive,
　　sections:      make(map[string][]string),
　　}
　　ini.loadValues()
　　return ini
　　
}
　　

　　
func (ini *IniFile) FileName() string {
　　return ini.fileName
　　
}
　　

　　
func (ini *IniFile) CaseSensitive() bool {
　　return ini.caseSensitive
　　
}
　　

　　
func (ini *IniFile) String() string {
　　var buf bytes.Buffer
　　for sec, lst := range ini.sections {
　　buf.WriteString(fmt.Sprintf("[%s]\n", sec))
　　for _, s := range lst {
　　buf.WriteString(fmt.Sprintf("%s\n", s))
　　}
　　buf.WriteString("\n")
　　}
　　return buf.String()
　　
}
　　

　　
func (ini *IniFile) getRealValue(s string) string {
　　if !ini.caseSensitive {
　　return strings.ToLower(s)
　　}
　　return s
　　
}
　　

　　
func (ini *IniFile) loadValues() {
　　if !sysutils.FileExists(ini.fileName) {
　　return
　　}
　　

　　file, err := os.Open(ini.fileName)
　　if err != nil {
　　return
　　}
　　defer file.Close()
　　

　　section := ""
　　scanner := bufio.NewScanner(file)
　　for scanner.Scan() {
　　s := scanner.Text()
　　s = strings.TrimSpace(s)
　　s = ini.getRealValue(s)
　　if s != "" && s[0] != ';' {
　　if s[0] == '[' && s[len(s)-1] == ']' {
　　s = s[1 : len(s)-1]
　　section = s
　　} else {
　　if section != "" {
　　if pos := strings.Index(s, "="); pos > 0 {
　　if sl, ok := ini.sections[section]; ok {
　　ini.sections[section] = append(sl, s)
　　} else {
　　ini.sections[section] = []string{s}
　　}
　　} else {

　　// ingore invalid>　　//
　　}
　　}
　　}
　　}
　　}
　　
}
　　

　　
func (ini *IniFile) flush() {
　　file, err := os.Create(ini.fileName)
　　sysutils.CheckError(err)
　　defer file.Close()
　　

　　fw := bufio.NewWriter(file)
　　for sec, lst := range ini.sections {
　　_, err = fw.WriteString(fmt.Sprintf("[%s]\n", sec))
　　sysutils.CheckError(err)
　　

　　for _, s := range lst {
　　_, err = fw.WriteString(fmt.Sprintf("%s\n", s))
　　sysutils.CheckError(err)
　　}
　　

　　_, err = fw.WriteString("\n")
　　sysutils.CheckError(err)
　　}
　　fw.Flush()
　　
}
　　

　　
func (ini *IniFile) SectionExists(section string) bool {
　　sec := ini.getRealValue(section)
　　if _, ok := ini.sections[sec]; ok {
　　return true
　　}
　　return false
　　
}
　　

　　
func (ini *IniFile) ReadSections() []string {
　　var ss []string
　　for sec, _ := range ini.sections {
　　ss = append(ss, sec)
　　}
　　return ss
　　
}
　　

　　
func (ini *IniFile) EraseSection(section string) {
　　sec := ini.getRealValue(section)
　　delete(ini.sections, sec)
　　
}
　　

　　
func (ini *IniFile) ReadSectionIdents(section string) []string {
　　var ss []string
　　sec := ini.getRealValue(section)
　　if sl, ok := ini.sections[sec]; ok {
　　for _, s := range sl {
　　if pos := strings.Index(s, "="); pos > 0 {
　　ss = append(ss, s[0:pos])
　　}
　　}
　　}
　　return ss
　　
}
　　

　　
func (ini *IniFile) ReadSectionValues(section string) []string {
　　var ss []string
　　sec := ini.getRealValue(section)
　　if sl, ok := ini.sections[sec]; ok {
　　for _, s := range sl {
　　ss = append(ss, s)
　　}
　　}
　　return ss
　　
}
　　


　　
func (ini *IniFile) DeleteIdent(section,>　　sec := ini.getRealValue(section)
　　id := ini.getRealValue(ident)
　　if sl, ok := ini.sections[sec]; ok {
　　for i := 0; i < len(sl); i++ {
　　s := sl
　　if pos := strings.Index(s, "="); pos > 0 {

　　if s[0:pos] ==>　　var ss []string
　　for j := 0; j < i; j++ {
　　ss = append(ss, sl[j])
　　}
　　for j := i + 1; j < len(sl); j++ {
　　ss = append(ss, sl[j])
　　}
　　ini.sections[sec] = ss
　　return
　　}
　　}
　　}
　　}
　　
}
　　


　　
func (ini *IniFile)>　　sec := ini.getRealValue(section)
　　id := ini.getRealValue(ident)
　　if sl, ok := ini.sections[sec]; ok {
　　for _, s := range sl {
　　if pos := strings.Index(s, "="); pos > 0 {

　　if s[0:pos] ==>　　return true
　　}
　　}
　　}
　　}
　　return false
　　
}
　　


　　
func (ini *IniFile) ReadString(section,>　　sec := ini.getRealValue(section)
　　id := ini.getRealValue(ident)
　　if sl, ok := ini.sections[sec]; ok {
　　for _, s := range sl {
　　if pos := strings.Index(s, "="); pos > 0 {

　　if s[0:pos] ==>　　return s[pos+1:]
　　}
　　}
　　}
　　}
　　return defaultValue
　　
}
　　


　　
func (ini *IniFile) WriteString(section,>　　sec := ini.getRealValue(section)
　　id := ini.getRealValue(ident)
　　if sl, ok := ini.sections[sec]; ok {
　　for i := 0; i < len(sl); i++ {
　　s := sl
　　if pos := strings.Index(s, "="); pos > 0 {

　　if s[0:pos] ==>　　var ss []string
　　for j := 0; j < i; j++ {
　　ss = append(ss, sl[j])
　　}

　　ss = append(ss,>　　for j := i + 1; j < len(sl); j++ {
　　ss = append(ss, sl[j])
　　}
　　ini.sections[sec] = ss
　　return
　　}
　　}
　　}

　　ini.sections[sec] = append(sl,>　　} else {
　　ini.sections[sec] = []string{ident + "=" + value}
　　}
　　
}
　　


　　
func (ini *IniFile) ReadInt(section,>
　　s := ini.ReadString(section,>　　if ret, err := strconv.Atoi(s); err == nil {
　　return ret
　　} else {
　　return defaultValue
　　}
　　
}
　　


　　
func (ini *IniFile) WriteInt(section,>
　　ini.WriteString(section,>　　
}
　　


　　
func (ini *IniFile) ReadBool(section,>
　　s := ini.ReadString(section,>　　return sysutils.StrToBool(s)
　　
}
　　


　　
func (ini *IniFile) WriteBool(section,>
　　ini.WriteString(section,>　　
}
　　


　　
func (ini *IniFile) ReadFloat(section,>
　　s := ini.ReadString(section,>　　if s != "" {
　　if ret, err := strconv.ParseFloat(s, 64); err == nil {
　　return ret
　　}
　　}
　　return defaultValue
　　
}
　　


　　
func (ini *IniFile) WriteFloat(section,>
　　ini.WriteString(section,>　　
}
　　

　　
func (ini *IniFile) Close() {
　　ini.flush()
　　
}
　　

　　
func (ini *IniFile) Clear() {
　　ini.sections = make(map[string][]string)
　　
}
　　


　　
func IniReadString(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　return inifile.ReadString(section,>　　
}
　　


　　
func IniWriteString(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　inifile.WriteString(section,>　　
}
　　


　　
func IniReadInt(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　return inifile.ReadInt(section,>　　
}
　　


　　
func IniWriteInt(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　inifile.WriteInt(section,>　　
}
　　

　　
func IniSectionExists(fileName, section string) bool {
　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()
　　return inifile.SectionExists(section)
　　
}
　　

　　
func IniReadSectionIdents(fileName, section string) []string {
　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()
　　return inifile.ReadSectionIdents(section)
　　
}
　　

　　
func IniReadSections(fileName string) []string {
　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()
　　return inifile.ReadSections()
　　
}
　　

　　
func IniReadSectionValues(fileName, section string) []string {
　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()
　　return inifile.ReadSectionValues(section)
　　
}
　　

　　
func IniEraseSection(fileName, section string) {
　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()
　　inifile.EraseSection(section)
　　
}
　　


　　
func IniIdentExists(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　return inifile.IdentExists(section,>　　
}
　　


　　
func IniDeleteIdent(fileName, section,>　　inifile := NewIniFile(fileName, false)
　　defer inifile.Close()

　　inifile.DeleteIdent(section,>　　
}