-
Notifications
You must be signed in to change notification settings - Fork 17
/
config.go
156 lines (129 loc) · 2.98 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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
package config
import (
"errors"
"os"
"path/filepath"
"time"
"github.com/pelletier/go-toml"
)
var (
configDirFn = os.UserHomeDir
readFileFn = os.ReadFile
)
const (
ConfigDirName = "/.deepsource/"
ConfigFileName = "/config.toml"
DefaultHostName = "deepsource.io"
)
type CLIConfig struct {
Host string `toml:"host"`
User string `toml:"user"`
Token string `toml:"token"`
TokenExpiresIn time.Time `toml:"token_expires_in,omitempty"`
}
var Cfg CLIConfig
// Sets the token expiry in the desired format
// Sets the token expiry in the desired format
func (cfg *CLIConfig) SetTokenExpiry(str string) {
t, _ := time.Parse(time.RFC3339, str)
cfg.TokenExpiresIn = t.UTC()
}
// Checks if the token has expired or not
func (cfg CLIConfig) IsExpired() bool {
if cfg.TokenExpiresIn.IsZero() {
return true
}
return time.Now().After(cfg.TokenExpiresIn)
}
// configDir returns the directory to store the config file.
func (CLIConfig) configDir() (string, error) {
home, err := configDirFn()
if err != nil {
return "", err
}
return filepath.Join(home, ConfigDirName), nil
}
// configPath returns the file path to the config file.
func (cfg CLIConfig) configPath() (string, error) {
home, err := cfg.configDir()
if err != nil {
return "", err
}
return filepath.Join(home, ConfigFileName), nil
}
// ReadFile reads the CLI config file.
func (cfg *CLIConfig) ReadConfigFile() error {
path, err := cfg.configPath()
if err != nil {
return err
}
// check if config exists
_, err = os.Stat(path)
if err != nil {
return nil
}
data, err := readFileFn(path)
if err != nil {
return err
}
err = toml.Unmarshal(data, cfg)
if err != nil {
return err
}
return nil
}
func GetConfig() (*CLIConfig, error) {
if Cfg.Token != "" {
return &Cfg, nil
}
err := Cfg.ReadConfigFile()
if err != nil {
return &Cfg, err
}
return &Cfg, nil
}
// WriteFile writes the CLI config to file.
func (cfg *CLIConfig) WriteFile() error {
data, err := toml.Marshal(cfg)
if err != nil {
return err
}
configDir, err := cfg.configDir()
if err != nil {
return err
}
if err := os.MkdirAll(configDir, os.ModePerm); err != nil {
return err
}
path, err := cfg.configPath()
if err != nil {
return err
}
// Create file
file, err := os.Create(path)
if err != nil {
return err
}
defer file.Close()
_, err = file.Write(data)
return err
}
// Deletes the config during logging out user
func (cfg *CLIConfig) Delete() error {
path, err := cfg.configPath()
if err != nil {
return err
}
return os.Remove(path)
}
func (cfg *CLIConfig) VerifyAuthentication() error {
// Checking if the user has authenticated / logged in or not
if cfg.Token == "" {
return errors.New("You are not logged into DeepSource. Run \"deepsource auth login\" to authenticate.")
}
// // Check if the token has already expired
// if cfg.IsExpired() {
// return errors.New("The authentication has expired. Run \"deepsource auth refresh\" to refresh the credentials.")
// }
return nil
}