Initial Commit

home_config.go

@@ -0,0 +1,172 @@
+// 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
+}