/
conf.go
60 lines (48 loc) · 1.62 KB
/
conf.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
// SPDX-License-Identifier: Apache-2.0
// Copyright Authors of Tetragon
package main
import (
"strings"
"github.com/cilium/tetragon/pkg/option"
"github.com/spf13/viper"
)
var (
adminTgConfDir = "/etc/tetragon/"
adminTgConfDropIn = "/etc/tetragon/tetragon.conf.d/"
packageTgConfDropIns = []string{
"/usr/lib/tetragon/tetragon.conf.d/",
"/usr/local/lib/tetragon/tetragon.conf.d/",
}
)
func readConfigSettings(defaultConfDir string, defaultConfDropIn string, dropInsDir []string) {
viper.SetEnvPrefix("tetragon")
replacer := strings.NewReplacer("-", "_")
viper.SetEnvKeyReplacer(replacer)
viper.AutomaticEnv()
// First set default conf file and format
viper.SetConfigName("tetragon")
viper.SetConfigType("yaml")
// Read default drop-ins directories
for _, dir := range dropInsDir {
option.ReadConfigDir(dir)
}
// Look into cwd first, this is needed for quick development only
option.ReadConfigFile(".", "tetragon.yaml")
// Look for /etc/tetragon/tetragon.yaml
option.ReadConfigFile(defaultConfDir, "tetragon.yaml")
// Look into default /etc/tetragon/tetragon.conf.d/ now
option.ReadConfigDir(defaultConfDropIn)
// Read now the passed key --config-dir
if viper.IsSet(option.KeyConfigDir) {
configDir := viper.GetString(option.KeyConfigDir)
// viper.IsSet could return true on an empty string reset
if configDir != "" {
err := option.ReadConfigDir(configDir)
if err != nil {
log.WithField(option.KeyConfigDir, configDir).WithError(err).Fatal("Failed to read config from directory")
} else {
log.WithField(option.KeyConfigDir, configDir).Info("Loaded config from directory")
}
}
}
}