-
Notifications
You must be signed in to change notification settings - Fork 3
/
flag.go
76 lines (59 loc) · 1.39 KB
/
flag.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
package cmd
import (
"encoding/base64"
"errors"
"os"
"path/filepath"
"strings"
"github.com/alexfalkowski/go-service/file"
"github.com/alexfalkowski/go-service/marshaller"
)
var (
// ErrInvalidKind for cmd.
ErrInvalidKind = errors.New("invalid kind")
// ConfigFlag for cmd.
ConfigFlag string
)
// Config for cmd.
type Config struct {
kind string
Data []byte
factory *marshaller.Factory
}
// NewConfig for cmd.
func NewConfig(factory *marshaller.Factory) (*Config, error) {
k, l := splitFlag()
switch k {
case "env":
k := file.Extension(os.Getenv(l))
d, err := os.ReadFile(filepath.Clean(os.Getenv(l)))
return &Config{kind: k, Data: d, factory: factory}, err
case "file":
k := file.Extension(l)
d, err := os.ReadFile(filepath.Clean(l))
return &Config{kind: k, Data: d, factory: factory}, err
case "mem":
k, l = splitMemory(l)
d, err := base64.StdEncoding.DecodeString(l)
return &Config{kind: k, Data: d, factory: factory}, err
}
return nil, ErrInvalidKind
}
// Marshaller for cmd.
func (c *Config) Marshaller() (marshaller.Marshaller, error) {
return c.factory.Create(c.kind)
}
func splitFlag() (string, string) {
c := strings.Split(ConfigFlag, ":")
if len(c) != 2 {
return "env", "CONFIG_FILE"
}
return c[0], c[1]
}
func splitMemory(l string) (string, string) {
c := strings.Split(l, "=>")
if len(c) != 2 {
return "yaml", l
}
return c[0], c[1]
}