-
Notifications
You must be signed in to change notification settings - Fork 0
/
userconfig.go
66 lines (56 loc) · 1.21 KB
/
userconfig.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
// SPDX-FileCopyrightText: 2023 RH America LLC <info@rhamerica.com>
// SPDX-License-Identifier: GPL-3.0-only
package model
import (
"strings"
"github.com/bazurto/bz/lib/utils"
)
/*
server "github.com" {
token: "abcde..."
}
------------
{
server: {
"github.com": {
token: "abcde..."
}
}
}
*/
type UserConfig struct {
Servers []UserConfigServer `ion:"server" hcl:"server,block"`
}
type UserConfigServer struct {
Name string `ion:"name" hcl:",label"`
Token string `ion:"token" hcl:"token,optional"`
}
type UserConfigIon struct {
Servers map[string]UserConfigServer `ion:"server"`
}
func NewUserConfigFromFile(f string) (*UserConfig, error) {
var err error
cfg := UserConfig{}
if utils.IsIonFile(f) {
uci := UserConfigIon{}
err = utils.IonLoad(f, &uci)
if err == nil {
for serverName, attr := range uci.Servers {
attr.Name = serverName
cfg.Servers = append(cfg.Servers, attr)
}
}
} else {
err = utils.HclLoad(f, &cfg)
}
return &cfg, err
}
func (o *UserConfig) GetServerToken(serverName string) string {
var token string
for _, server := range o.Servers {
if strings.ToLower(server.Name) == strings.ToLower(serverName) {
token = server.Token
}
}
return token
}