iniutils for Golang
// 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
}
func NewIniFile(filename string, casesensitive bool) *IniFile {
ini := &IniFile{
fileName: filename,
caseSensitive: casesensitive,
sections: make(map[]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 != ';' {
if s == '[' && s == ']' {
s = s
section = s
} else {
if section != "" {
if pos := strings.Index(s, "="); pos > 0 {
if sl, ok := ini.sections; ok {
ini.sections = append(sl, s)
} else {
ini.sections = []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; 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; ok {
for _, s := range sl {
if pos := strings.Index(s, "="); pos > 0 {
ss = append(ss, s)
}
}
}
return ss
}
func (ini *IniFile) ReadSectionValues(section string) []string {
var ss []string
sec := ini.getRealValue(section)
if sl, ok := ini.sections; 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; ok {
for i := 0; i < len(sl); i++ {
s := sl
if pos := strings.Index(s, "="); pos > 0 {
if s ==> var ss []string
for j := 0; j < i; j++ {
ss = append(ss, sl)
}
for j := i + 1; j < len(sl); j++ {
ss = append(ss, sl)
}
ini.sections = ss
return
}
}
}
}
}
func (ini *IniFile)> sec := ini.getRealValue(section)
id := ini.getRealValue(ident)
if sl, ok := ini.sections; ok {
for _, s := range sl {
if pos := strings.Index(s, "="); pos > 0 {
if s ==> return true
}
}
}
}
return false
}
func (ini *IniFile) ReadString(section,> sec := ini.getRealValue(section)
id := ini.getRealValue(ident)
if sl, ok := ini.sections; ok {
for _, s := range sl {
if pos := strings.Index(s, "="); pos > 0 {
if s ==> return s
}
}
}
}
return defaultValue
}
func (ini *IniFile) WriteString(section,> sec := ini.getRealValue(section)
id := ini.getRealValue(ident)
if sl, ok := ini.sections; ok {
for i := 0; i < len(sl); i++ {
s := sl
if pos := strings.Index(s, "="); pos > 0 {
if s ==> var ss []string
for j := 0; j < i; j++ {
ss = append(ss, sl)
}
ss = append(ss,> for j := i + 1; j < len(sl); j++ {
ss = append(ss, sl)
}
ini.sections = ss
return
}
}
}
ini.sections = append(sl,> } else {
ini.sections = []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)
}
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,>
}
页:
[1]