Merge pull request #251 from cjvaughter/config-file

Add config file support

config.go

@@ -0,0 +1,101 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"io/ioutil"
+	"encoding/json"
+)
+
+type SymbolMap map[string]SymbolTemplate
+type ShellMap map[string]ShellInfo
+type ThemeMap map[string]Theme
+type AliasMap map[string]string
+
+type Config struct {
+	CwdMode                string    `json:"cwd-mode"`
+	CwdMaxDepth            int       `json:"cwd-max-depth"`
+	CwdMaxDirSize          int       `json:"cwd-max-dir-size"`
+	ColorizeHostname       bool      `json:"colorize-hostname"`
+	HostnameOnlyIfSSH      bool      `json:"hostname-only-if-ssh"`
+	SshAlternateIcon       bool      `json:"alternate-ssh-icon"`
+	EastAsianWidth         bool      `json:"east-asian-width"`
+	PromptOnNewLine        bool      `json:"newline"`
+	StaticPromptIndicator  bool      `json:"static-prompt-indicator"`
+	VenvNameSizeLimit      int       `json:"venv-name-size-limit"`
+	Jobs                   int       `json:"-"`
+	GitAssumeUnchangedSize int64     `json:"git-assume-unchanged-size"`
+	GitDisableStats        []string  `json:"git-disable-stats"`
+	GitMode                string    `json:"git-mode"`
+	Mode                   string    `json:"mode"`
+	Theme                  string    `json:"theme"`
+	Shell                  string    `json:"shell"`
+	Modules                []string  `json:"modules"`
+	ModulesRight           []string  `json:"modules-right"`
+	Priority               []string  `json:"priority"`
+	MaxWidthPercentage     int       `json:"max-width-percentage"`
+	TruncateSegmentWidth   int       `json:"truncate-segment-width"`
+	PrevError              int       `json:"-"`
+	NumericExitCodes       bool      `json:"numeric-exit-codes"`
+	IgnoreRepos            []string  `json:"ignore-repos"`
+	ShortenGKENames        bool      `json:"shorten-gke-names"`
+	ShortenEKSNames        bool      `json:"shorten-eks-names"`
+	ShellVar               string    `json:"shell-var"`
+	ShellVarNoWarnEmpty    bool      `json:"shell-var-no-warn-empty"`
+	TrimADDomain           bool      `json:"trim-ad-domain"`
+	PathAliases            AliasMap  `json:"path-aliases"`
+	Duration               string    `json:"-"`
+	DurationMin            string    `json:"duration-min"`
+	Eval                   bool      `json:"eval"`
+	Condensed              bool      `json:"condensed"`
+	Modes                  SymbolMap `json:"modes"`
+	Shells                 ShellMap  `json:"shells"`
+	Themes                 ThemeMap  `json:"themes"`
+}
+
+func (mode *SymbolTemplate) UnmarshalJSON(data []byte) error {
+	type Alias SymbolTemplate
+	tmp := defaults.Modes[defaults.Mode]
+	err := json.Unmarshal(data, (*Alias)(&tmp))
+	if err == nil {
+		*mode = tmp
+	}
+	return err
+}
+
+func (theme *Theme) UnmarshalJSON(data []byte) error {
+	type Alias Theme
+	tmp := defaults.Themes[defaults.Theme]
+	err := json.Unmarshal(data, (*Alias)(&tmp))
+	if err == nil {
+		*theme = tmp
+	}
+	return err
+}
+
+func configPath() string {
+	home, _ := os.UserHomeDir()
+	return filepath.Join(home, ".config", "powerline-go", "config.json")
+}
+
+func (cfg *Config) Load() error {
+	path := configPath()
+	file, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil // fail silently
+	}
+	return json.Unmarshal(file, cfg)
+}
+
+func (cfg *Config) Save() error {
+	path := configPath()
+	tmp := cfg
+	tmp.Themes = map[string]Theme{}
+	tmp.Modes = map[string]SymbolTemplate{}
+	tmp.Shells = map[string]ShellInfo{}
+	data, err := json.MarshalIndent(tmp, "", "    ")
+	if err != nil {
+		return err
+	}
+	return ioutil.WriteFile(path, data, 0644)
+}