file.go

package ini

import (
	"io"
)

// This implements the full ini.StreamReadWriter interface
type file struct {
	sections                   map[string]*section
	reader                     io.Reader
	environmentOverrideEnabled bool
	environmentOverridePrefix  string
}

func (f *file) EnableEnvironmentVariableOverrides(prefix string) {
	f.environmentOverrideEnabled = true
	f.environmentOverridePrefix = prefix
}

func (f *file) DisableEnvironmentVariableOverrides() {
	f.environmentOverrideEnabled = false
}

// Returns a named Section. A Section will be created if one does not already exist for the given name.
func (f *file) section(name string) *section {
	theSection := f.sections[name]
	if theSection == nil {
		theSection = &section{
			file:         f,
			name:         name,
			stringValues: make(map[string]string),
			arrayValues:  make(map[string][]string),
		}
		f.sections[name] = theSection
	}
	return theSection
}

func (f *file) Sections() (value []string) {
	value = make([]string, 0, len(f.sections))
	for sect, _ := range f.sections {
		value = append(value, sect)
	}
	return
}

func (f *file) Values(section string) (value map[string]string) {
	value = make(map[string]string)
	sect := f.section(section)
	if sect != nil {
		for k, v := range sect.stringValues {
			value[k] = v
		}
	}
	return
}

// Looks up a value for a key in a section and returns that value, along with a boolean result similar to a map lookup.
func (f *file) Get(section, key string) (value string, ok bool) {
	return f.section(section).Get(key)
}

// Set the value for a key in a section, along with a boolean result similar to a map lookup.
func (f *file) Set(section, key string, value string) (ok bool) {
	return f.section(section).Set(key, value)
}

// Set a key in a section to an integer value
func (f *file) SetInt(section, key string, value int) (ok bool) {
	return f.section(section).SetInt(key, value)
}

// Set a key in a section to a boolean value
func (f *file) SetBool(section, key string, value bool) (ok bool) {
	return f.section(section).SetBool(key, value)
}

// Set a key in a section to an array
func (f *file) SetArr(section, key string, value []string) (ok bool) {
	return f.section(section).SetArr(key, value)
}

// Looks up a value for a key in a section and returns that value, along with a boolean result similar to a map lookup.
// The `ok` boolean will be false in the event that the value could not be parsed as an int
func (f *file) GetInt(section, key string) (value int, ok bool) {
	return f.section(section).GetInt(key)
}

// Looks up a value for a key in a section and returns that value, along with a boolean result similar to a map lookup.
// The `ok` boolean will be false in the event that the value could not be parsed as a bool
func (f *file) GetBool(section, key string) (value bool, ok bool) {
	return f.section(section).GetBool(key)
}

// Looks up a value for an array key in a section and returns that value, along with a boolean result similar to a map lookup.
func (f *file) GetArr(section, key string) (value []string, ok bool) {
	return f.section(section).GetArr(key)
}

func (f *file) Remove(section, key string) {
	f.section(section).Remove(key)
}

func (f *file) RemoveSection(section string) {
	_, found := f.sections[section]
	if found {
		delete(f.sections, section)
	}
}

func (f *file) Copy(w Setter) {
	for secName, sec := range f.sections {
		for keyName, val := range sec.stringValues {
			w.Set(secName, keyName, val)
		}
		for keyName, arVal := range sec.arrayValues {
			w.SetArr(secName, keyName, arVal)
		}
	}
}