forked from evergreen-ci/evergreen
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_api.go
103 lines (87 loc) · 2.99 KB
/
config_api.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
package evergreen
import (
"context"
"fmt"
"strings"
"github.com/evergreen-ci/pail"
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
type ClientBinary struct {
Arch string `yaml:"arch" json:"arch"`
OS string `yaml:"os" json:"os"`
URL string `yaml:"url" json:"url"`
DisplayName string `yaml:"display_name" json:"display_name"`
}
type ClientConfig struct {
ClientBinaries []ClientBinary
S3ClientBinaries []ClientBinary
LatestRevision string
S3URLPrefix string
}
func (c *ClientConfig) populateClientBinaries(ctx context.Context, bucket pail.Bucket, prefix string) error {
if !strings.HasSuffix(prefix, "/") {
prefix += "/"
}
iter, err := bucket.List(ctx, prefix)
if err != nil {
return errors.Wrap(err, "listing client bucket")
}
for iter.Next(ctx) {
item := strings.TrimPrefix(iter.Item().Name(), prefix)
// The item is expected to be of the form <os>_<arch>/evergreen{.exe}
name := strings.Split(item, "/")
// Skip files that either don't conform to the expected form or aren't the evergreen executable (e.g. .signed).
if len(name) != 2 || !strings.Contains(name[1], "evergreen") {
continue
}
// Skip files for invalid platforms.
displayName, ok := ValidArchDisplayNames[name[0]]
if !ok {
continue
}
osArchParts := strings.Split(name[0], "_")
c.ClientBinaries = append(c.ClientBinaries, ClientBinary{
URL: fmt.Sprintf("%s/%s", c.S3URLPrefix, item),
OS: osArchParts[0],
Arch: osArchParts[1],
DisplayName: displayName,
})
}
if err = iter.Err(); err != nil {
return errors.Wrap(err, "iterating client bucket contents")
}
return nil
}
// APIConfig holds relevant log and listener settings for the API server.
type APIConfig struct {
HttpListenAddr string `bson:"http_listen_addr" json:"http_listen_addr" yaml:"httplistenaddr"`
GithubWebhookSecret string `bson:"github_webhook_secret" json:"github_webhook_secret" yaml:"github_webhook_secret"`
}
func (c *APIConfig) SectionId() string { return "api" }
func (c *APIConfig) Get(ctx context.Context) error {
res := GetEnvironment().DB().Collection(ConfigCollection).FindOne(ctx, byId(c.SectionId()))
if err := res.Err(); err != nil {
if err == mongo.ErrNoDocuments {
*c = APIConfig{}
return nil
}
return errors.Wrapf(err, "getting config section '%s'", c.SectionId())
}
if err := res.Decode(&c); err != nil {
return errors.Wrapf(err, "decoding config section '%s'", c.SectionId())
}
return nil
}
func (c *APIConfig) Set(ctx context.Context) error {
_, err := GetEnvironment().DB().Collection(ConfigCollection).UpdateOne(ctx, byId(c.SectionId()), bson.M{
"$set": bson.M{
"http_listen_addr": c.HttpListenAddr,
"github_webhook_secret": c.GithubWebhookSecret,
},
}, options.Update().SetUpsert(true))
return errors.Wrapf(err, "updating config section '%s'", c.SectionId())
}
func (c *APIConfig) ValidateAndDefault() error { return nil }