forked from gopasspw/gopass
/
io.go
124 lines (114 loc) · 2.51 KB
/
io.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package config
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"github.com/justwatchcom/gopass/utils/fsutil"
"github.com/pkg/errors"
"gopkg.in/yaml.v2"
)
// Load will try to load the config from one of the default locations
func Load() *Config {
for _, l := range configLocations() {
cfg, err := load(l)
if err == ErrConfigNotFound {
continue
}
if err != nil {
panic(err)
}
if debug {
fmt.Printf("[DEBUG] Loaded config from %s: %+v\n", l, cfg)
}
return cfg
}
cfg := New()
cfg.Root.Path = PwStoreDir("")
return cfg
}
func load(cf string) (*Config, error) {
// deliberately using os.Stat here, a symlinked
// config is OK
if _, err := os.Stat(cf); err != nil {
return nil, ErrConfigNotFound
}
buf, err := ioutil.ReadFile(cf)
if err != nil {
fmt.Printf("Error reading config from %s: %s\n", cf, err)
return nil, ErrConfigNotFound
}
cfg, err := decode(buf)
if err != nil {
fmt.Printf("Error reading config from %s: %s\n", cf, err)
return nil, ErrConfigNotParsed
}
if cfg.Mounts == nil {
cfg.Mounts = make(map[string]*StoreConfig)
}
return cfg, nil
}
func checkOverflow(m map[string]interface{}, section string) error {
if len(m) < 1 {
return nil
}
var keys []string
for k := range m {
keys = append(keys, k)
}
return errors.Errorf("unknown fields in %s: %+v", section, keys)
}
type configer interface {
Config() *Config
CheckOverflow() error
}
func decode(buf []byte) (*Config, error) {
cfgs := []configer{
&Config{
Root: &StoreConfig{
AutoSync: true,
},
},
&Pre140{
AutoSync: true,
},
&Pre130{},
}
for _, cfg := range cfgs {
if err := yaml.Unmarshal(buf, cfg); err != nil {
continue
}
if err := cfg.CheckOverflow(); err != nil {
if debug {
fmt.Printf("[DEBUG] Extra elements in config: %s\n", err)
}
continue
}
if debug {
fmt.Printf("[DEBUG] Loaded config: %+v\n", cfg)
}
return cfg.Config(), nil
}
return nil, ErrConfigNotParsed
}
// Save saves the config
func (c *Config) Save() error {
buf, err := yaml.Marshal(c)
if err != nil {
return errors.Wrapf(err, "failed to marshal YAML")
}
cfgLoc := configLocation()
cfgDir := filepath.Dir(cfgLoc)
if !fsutil.IsDir(cfgDir) {
if err := os.MkdirAll(cfgDir, 0700); err != nil {
return errors.Wrapf(err, "failed to create dir '%s'", cfgDir)
}
}
if err := ioutil.WriteFile(cfgLoc, buf, 0600); err != nil {
return errors.Wrapf(err, "failed to write config file to '%s'", cfgLoc)
}
if debug {
fmt.Printf("[DEBUG] Saved config to %s: %+v\n", cfgLoc, c)
}
return nil
}