-
Notifications
You must be signed in to change notification settings - Fork 3
/
variable_config.go
95 lines (80 loc) · 1.96 KB
/
variable_config.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
package unpuzzled
import (
"errors"
"flag"
"io/ioutil"
"github.com/Jeffail/gabs"
"github.com/pelletier/go-toml"
log "github.com/sirupsen/logrus"
)
type ConfigVariable struct {
*StringVariable
Type ParsingType
config configGetter
}
type configGetter interface {
GetByVariable(string) (interface{}, error)
}
var (
ErrFailedToLoadToml = errors.New("Failed to load toml.")
ErrFailedToLoadJson = errors.New("Failed to load json.")
ErrConfigValueNotSet = errors.New("Config variable is not set.")
)
func (c *ConfigVariable) apply(interface{}) {
log.Fatal("Config Variable does not apply values.")
}
func (c *ConfigVariable) ParseConfig(set *flag.FlagSet) error {
// ignore error, library should handle this.
stringPointer, _ := c.StringVariable.getFlagValue(set)
if stringPointer == nil {
return ErrConfigValueNotSet
}
value := stringPointer.(string)
data, err := ioutil.ReadFile(value)
if err != nil {
return err
}
switch c.Type {
case TomlConfig:
tree, err := toml.Load(string(data))
if err != nil {
log.WithFields(log.Fields{"err": err}).Fatal(ErrFailedToLoadToml)
return ErrFailedToLoadToml
}
config := &tomlConfig{
tree: tree,
}
c.config = config
case JsonConfig:
container, err := gabs.ParseJSON(data)
if err != nil {
log.WithFields(log.Fields{"err": err}).Fatal(ErrFailedToLoadJson)
return ErrFailedToLoadJson
}
config := &jsonConfig{
container: container,
}
c.config = config
default:
log.Fatal("Unimplemented config type")
}
return nil
}
func (c *ConfigVariable) getConfigValue(path string) (interface{}, error) {
if c.config != nil {
return c.config.GetByVariable(path)
}
return nil, nil
}
type tomlConfig struct {
tree *toml.Tree
}
func (t *tomlConfig) GetByVariable(path string) (interface{}, error) {
return t.tree.Get(path), nil
}
type jsonConfig struct {
container *gabs.Container
}
func (j *jsonConfig) GetByVariable(path string) (interface{}, error) {
return j.container.Path(path).Data(), nil
}