-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.go
126 lines (115 loc) · 2.64 KB
/
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
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
125
126
package config
import (
"context"
"errors"
"fmt"
"github.com/fsnotify/fsnotify"
"io/ioutil"
"os"
"github.com/AnatolyRugalev/kube-commander/pb"
"google.golang.org/protobuf/encoding/protojson"
"path/filepath"
"sigs.k8s.io/yaml"
)
const DefaultName = ".kubecom.yaml"
func DefaultPath() (string, error) {
home, err := os.UserHomeDir()
if err != nil {
return "", fmt.Errorf("error obtaining user home dir: %w", err)
}
return filepath.Join(home, DefaultName), nil
}
type Event struct {
Config *pb.Config
Err error
}
func Exists(path string) bool {
_, err := os.Stat(path)
return !errors.Is(err, os.ErrNotExist)
}
func Init(path string) error {
return ioutil.WriteFile(path, nil, 0755)
}
func Watch(ctx context.Context, path string, ch chan<- Event) error {
watcher, err := fsnotify.NewWatcher()
if err != nil {
return fmt.Errorf("could not start watch notifier: %w", err)
}
err = watcher.Add(path)
if err != nil {
return err
}
go func() {
defer watcher.Close()
config, err := Load(path)
if err != nil {
ch <- Event{Err: err}
} else {
ch <- Event{Config: config}
}
for {
select {
case <-ctx.Done():
ch <- Event{Err: ctx.Err()}
return
case err, ok := <-watcher.Errors:
if !ok {
return
}
ch <- Event{Err: err}
return
case event, ok := <-watcher.Events:
if !ok {
return
}
if event.Op&fsnotify.Write == fsnotify.Write {
config, err := Load(path)
if err != nil {
ch <- Event{Err: err}
} else {
ch <- Event{Config: config}
}
}
}
}
}()
return nil
}
func Save(path string, config *pb.Config) error {
err := os.MkdirAll(filepath.Dir(path), 0755)
if err != nil {
return fmt.Errorf("could not create configuration file directory")
}
jsonB, err := protojson.Marshal(config)
if err != nil {
return fmt.Errorf("error marshalling config: %w", err)
}
yamlB, err := yaml.JSONToYAML(jsonB)
if err != nil {
return fmt.Errorf("error converting configuration to YAML: %w", err)
}
err = ioutil.WriteFile(path, yamlB, 0755)
if err != nil {
return fmt.Errorf("error writing config file: %w", err)
}
return nil
}
func Load(path string) (*pb.Config, error) {
yamlB, err := ioutil.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("error reading configuration file: %w", err)
}
config := pb.Config{}
if len(yamlB) == 0 {
return &config, nil
}
jsonB, err := yaml.YAMLToJSON(yamlB)
if err != nil {
return nil, fmt.Errorf("error converting configuration to JSON: %w", err)
}
err = protojson.Unmarshal(jsonB, &config)
if err != nil {
return nil, fmt.Errorf("error unmarshalling configuration: %w", err)
}
return &config, nil
}