-
Notifications
You must be signed in to change notification settings - Fork 102
/
config.go
94 lines (80 loc) · 2.63 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
// Copyright 2018 Palantir Technologies, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package server
import (
"os"
"strconv"
"time"
"github.com/c2h5oh/datasize"
"github.com/palantir/go-baseapp/baseapp"
"github.com/palantir/go-baseapp/baseapp/datadog"
"github.com/palantir/go-githubapp/githubapp"
"github.com/palantir/policy-bot/server/handler"
"github.com/pkg/errors"
"gopkg.in/yaml.v2"
)
const (
DefaultEnvPrefix = "POLICYBOT_"
)
type Config struct {
Server baseapp.HTTPConfig `yaml:"server"`
Logging LoggingConfig `yaml:"logging"`
Cache CachingConfig `yaml:"cache"`
Github githubapp.Config `yaml:"github"`
Sessions SessionsConfig `yaml:"sessions"`
Options handler.PullEvaluationOptions `yaml:"options"`
Files handler.FilesConfig `yaml:"files"`
Datadog datadog.Config `yaml:"datadog"`
Workers WorkerConfig `yaml:"workers"`
}
type LoggingConfig struct {
Level string `yaml:"level" json:"level"`
Text bool `yaml:"text" json:"text"`
}
func (c *LoggingConfig) SetValuesFromEnv(prefix string) {
if v, ok := os.LookupEnv(prefix + "LOG_LEVEL"); ok {
c.Level = v
}
if v, ok := os.LookupEnv(prefix + "LOG_TEXT"); ok {
if b, err := strconv.ParseBool(v); err == nil {
c.Text = b
}
}
}
type CachingConfig struct {
MaxSize datasize.ByteSize `yaml:"max_size"`
}
type WorkerConfig struct {
Workers int `yaml:"workers"`
QueueSize int `yaml:"queue_size"`
GithubTimeout time.Duration `yaml:"github_timeout"`
}
type SessionsConfig struct {
Key string `yaml:"key"`
Lifetime string `yaml:"lifetime"`
}
func ParseConfig(bytes []byte) (*Config, error) {
var c Config
if err := yaml.UnmarshalStrict(bytes, &c); err != nil {
return nil, errors.Wrapf(err, "failed unmarshalling yaml")
}
c.Options.SetValuesFromEnv(DefaultEnvPrefix + "OPTIONS_")
c.Server.SetValuesFromEnv(DefaultEnvPrefix)
c.Logging.SetValuesFromEnv(DefaultEnvPrefix)
c.Github.SetValuesFromEnv("")
if v, ok := os.LookupEnv(DefaultEnvPrefix + "SESSIONS_KEY"); ok {
c.Sessions.Key = v
}
return &c, nil
}