user-config/home_config.go

// Package homeConfig eases the use of config files in a user's home directory
package homeConfig

import (
	"bufio"
	"errors"
	"fmt"
	"os"
	"strings"

	"launchpad.net/go-xdg"
)

// Config is a stuct for managing the config
type Config struct {
	name string
	// ConfigFiles are files that have key/value pairs
	ConfigFiles []ConfigFile
	// RawFiles are other files (dbs, etc.)
	RawFiles map[string]string
}

// ConfigFile is a file that has key/value pairs in the config
type ConfigFile struct {
	Name   string
	values map[string]string
}

// NewConfig generates a Config struct
func NewConfig(name string) (*Config, error) {
	c := &Config{name: name}
	if err := c.Load(); err != nil {
		return nil, err
	}
	return c, nil
}

// Set at the config level sets a value in the <c.name>.conf file
func (c *Config) Set(k, v string) error {
	for i := range c.ConfigFiles {
		if c.ConfigFiles[i].Name == c.name+".conf" {
			c.ConfigFiles[i].Set(k, v)
			return nil
		}
	}
	return errors.New("Couldn't find " + c.name + ".config file")
}

// Get at the config level retrieves a value from the <c.name>.conf file
func (c *Config) Get(k string) (string, error) {
	for i := range c.ConfigFiles {
		if c.ConfigFiles[i].Name == c.name+".conf" {
			return c.ConfigFiles[i].Get(k), nil
		}
	}
	return "", errors.New("Couldn't find " + c.name + ".config file")
}

// Set sets a key/value pair in cf
func (cf *ConfigFile) Set(k, v string) {
	cf.values[k] = v
}

// Get gets a key/value pair from cf
func (cf *ConfigFile) Get(k string) string {
	return cf.values[k]
}

// Load loads config files into the config
func (c *Config) Load() error {
	// Clear the data, we're reloading
	c.ConfigFiles = c.ConfigFiles[:0]
	c.RawFiles = make(map[string]string)

	if strings.TrimSpace(c.name) == "" {
		return errors.New("Invalid Config Name: " + c.name)
	}

	var cfgPath string
	cfgPath = xdg.Config.Dirs()[0]
	//cfgPath = os.Getenv("HOME")
	if cfgPath != "" {
		//cfgPath = cfgPath + "/.config"
		//if err := verifyOrCreateDirectory(cfgPath); err != nil {
		//	return err
		//}
		cfgPath = cfgPath + "/" + c.name
		if err := c.VerifyOrCreateDirectory(cfgPath); err != nil {
			return err
		}
		// We always have a <name>.conf file
		cfgPath = cfgPath + "/" + c.name + ".conf"
	}
	fmt.Println(cfgPath)
	return nil

	if cfgPath != "" {
		file, err := os.Open(cfgPath)
		if err != nil {
			// Couldn't load config even though one was specified
			return errors.New("Unable to load config")
		}
		defer file.Close()
		scanner := bufio.NewScanner(file)
		for scanner.Scan() {
			tstString := scanner.Text()
			prts := strings.Split(tstString, "=")
			if len(prts) > 1 {
			}
			if len(prts) > 1 {
			}
		}
	}
	return nil
}

// Save writes the config to file(s)
func (c *Config) Save() error {
	var cfgPath string
	var configLines []string
	//configLines = append(configLines, "server="+client.ServerAddr)
	//configLines = append(configLines, "key="+client.ServerKey)
	cfgPath = os.Getenv("HOME")
	if cfgPath != "" {
		cfgPath = cfgPath + "/.config"
		if err := c.VerifyOrCreateDirectory(cfgPath); err != nil {
			return err
		}
		cfgPath = cfgPath + "/" + c.name
	}
	if cfgPath != "" {
		file, err := os.Create(cfgPath)
		if err != nil {
			// Couldn't load config even though one was specified
			return err
		}
		defer file.Close()

		w := bufio.NewWriter(file)
		for _, line := range configLines {
			fmt.Fprintln(w, line)
		}
		if err = w.Flush(); err != nil {
			return err
		}
	}
	return nil
}

// VerifyOrCreateDirectory is a helper function for building an
// individual directory
func (c *Config) VerifyOrCreateDirectory(path string) error {
	var tstDir *os.File
	var tstDirInfo os.FileInfo
	var err error
	if tstDir, err = os.Open(path); err != nil {
		if err = os.Mkdir(path, 0755); err != nil {
			return err
		}
		if tstDir, err = os.Open(path); err != nil {
			return err
		}
	}
	if tstDirInfo, err = tstDir.Stat(); err != nil {
		return err
	}
	if !tstDirInfo.IsDir() {
		return errors.New(path + " exists and is not a directory")
	}
	// We were able to open the path and it was a directory
	return nil
}