feat: read config from env (#104)
* feat: read config from env * ci: skip some tests
This commit is contained in:
UUBulb
+98
-30
@@ -1,51 +1,75 @@
|
||||
package model
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
"github.com/hashicorp/go-uuid"
|
||||
"github.com/knadh/koanf/providers/env"
|
||||
"github.com/knadh/koanf/providers/file"
|
||||
"github.com/knadh/koanf/v2"
|
||||
"sigs.k8s.io/yaml"
|
||||
|
||||
"github.com/nezhahq/agent/pkg/util"
|
||||
)
|
||||
|
||||
type AgentConfig struct {
|
||||
Debug bool `mapstructure:"debug" json:"debug"`
|
||||
Debug bool `koanf:"debug" json:"debug"`
|
||||
|
||||
Server string `mapstructure:"server" json:"server"` // 服务器地址
|
||||
ClientSecret string `mapstructure:"client_secret" json:"client_secret"` // 客户端密钥
|
||||
UUID string `mapstructure:"uuid" json:"uuid"`
|
||||
Server string `koanf:"server" json:"server"` // 服务器地址
|
||||
ClientSecret string `koanf:"client_secret" json:"client_secret"` // 客户端密钥
|
||||
UUID string `koanf:"uuid" json:"uuid"`
|
||||
|
||||
HardDrivePartitionAllowlist []string `mapstructure:"hard_drive_partition_allowlist" json:"hard_drive_partition_allowlist"`
|
||||
NICAllowlist map[string]bool `mapstructure:"nic_allowlist" json:"nic_allowlist"`
|
||||
DNS []string `mapstructure:"dns" json:"dns"`
|
||||
GPU bool `mapstructure:"gpu" json:"gpu"` // 是否检查GPU
|
||||
Temperature bool `mapstructure:"temperature" json:"temperature"` // 是否检查温度
|
||||
SkipConnectionCount bool `mapstructure:"skip_connection_count" json:"skip_connection_count"` // 跳过连接数检查
|
||||
SkipProcsCount bool `mapstructure:"skip_procs_count" json:"skip_procs_count"` // 跳过进程数量检查
|
||||
DisableAutoUpdate bool `mapstructure:"disable_auto_update" json:"disable_auto_update"` // 关闭自动更新
|
||||
DisableForceUpdate bool `mapstructure:"disable_force_update" json:"disable_force_update"` // 关闭强制更新
|
||||
DisableCommandExecute bool `mapstructure:"disable_command_execute" json:"disable_command_execute"` // 关闭命令执行
|
||||
ReportDelay int `mapstructure:"report_delay" json:"report_delay"` // 报告间隔
|
||||
TLS bool `mapstructure:"tls" json:"tls"` // 是否使用TLS加密传输至服务端
|
||||
InsecureTLS bool `mapstructure:"insecure_tls" json:"insecure_tls"` // 是否禁用证书检查
|
||||
UseIPv6CountryCode bool `mapstructure:"use_ipv6_country_code" json:"use_ipv6_country_code"` // 默认优先展示IPv6旗帜
|
||||
UseGiteeToUpgrade bool `mapstructure:"use_gitee_to_upgrade" json:"use_gitee_to_upgrade"` // 强制从Gitee获取更新
|
||||
DisableNat bool `mapstructure:"disable_nat" json:"disable_nat"` // 关闭内网穿透
|
||||
DisableSendQuery bool `mapstructure:"disable_send_query" json:"disable_send_query"` // 关闭发送TCP/ICMP/HTTP请求
|
||||
IPReportPeriod uint32 `mapstructure:"ip_report_period" json:"ip_report_period"` // IP上报周期
|
||||
CustomIPApi []string `mapstructure:"custom_ip_api" json:"custom_ip_api"` // 自定义 IP API
|
||||
HardDrivePartitionAllowlist []string `koanf:"hard_drive_partition_allowlist" json:"hard_drive_partition_allowlist,omitempty"`
|
||||
NICAllowlist map[string]bool `koanf:"nic_allowlist" json:"nic_allowlist,omitempty"`
|
||||
DNS []string `koanf:"dns" json:"dns,omitempty"`
|
||||
GPU bool `koanf:"gpu" json:"gpu"` // 是否检查GPU
|
||||
Temperature bool `koanf:"temperature" json:"temperature"` // 是否检查温度
|
||||
SkipConnectionCount bool `koanf:"skip_connection_count" json:"skip_connection_count"` // 跳过连接数检查
|
||||
SkipProcsCount bool `koanf:"skip_procs_count" json:"skip_procs_count"` // 跳过进程数量检查
|
||||
DisableAutoUpdate bool `koanf:"disable_auto_update" json:"disable_auto_update"` // 关闭自动更新
|
||||
DisableForceUpdate bool `koanf:"disable_force_update" json:"disable_force_update"` // 关闭强制更新
|
||||
DisableCommandExecute bool `koanf:"disable_command_execute" json:"disable_command_execute"` // 关闭命令执行
|
||||
ReportDelay int `koanf:"report_delay" json:"report_delay"` // 报告间隔
|
||||
TLS bool `koanf:"tls" json:"tls"` // 是否使用TLS加密传输至服务端
|
||||
InsecureTLS bool `koanf:"insecure_tls" json:"insecure_tls"` // 是否禁用证书检查
|
||||
UseIPv6CountryCode bool `koanf:"use_ipv6_country_code" json:"use_ipv6_country_code"` // 默认优先展示IPv6旗帜
|
||||
UseGiteeToUpgrade bool `koanf:"use_gitee_to_upgrade" json:"use_gitee_to_upgrade"` // 强制从Gitee获取更新
|
||||
DisableNat bool `koanf:"disable_nat" json:"disable_nat"` // 关闭内网穿透
|
||||
DisableSendQuery bool `koanf:"disable_send_query" json:"disable_send_query"` // 关闭发送TCP/ICMP/HTTP请求
|
||||
IPReportPeriod uint32 `koanf:"ip_report_period" json:"ip_report_period"` // IP上报周期
|
||||
CustomIPApi []string `koanf:"custom_ip_api" json:"custom_ip_api,omitempty"` // 自定义 IP API
|
||||
|
||||
v *viper.Viper
|
||||
k *koanf.Koanf `json:"-"`
|
||||
filePath string `json:"-"`
|
||||
}
|
||||
|
||||
// Read 从给定的文件目录加载配置文件
|
||||
func (c *AgentConfig) Read(path string) error {
|
||||
c.v = viper.New()
|
||||
c.v.SetConfigFile(path)
|
||||
err := c.v.ReadInConfig()
|
||||
c.k = koanf.New("")
|
||||
c.filePath = path
|
||||
saveOnce := util.OnceValue(c.Save)
|
||||
|
||||
if _, err := os.Stat(path); err == nil {
|
||||
err = c.k.Load(file.Provider(path), new(kubeyaml))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
} else {
|
||||
defer saveOnce()
|
||||
}
|
||||
|
||||
err := c.k.Load(env.Provider("NZ_", "", func(s string) string {
|
||||
return strings.ToLower(strings.TrimPrefix(s, "NZ_"))
|
||||
}), nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = c.v.Unmarshal(c)
|
||||
|
||||
err = c.k.Unmarshal("", c)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -60,6 +84,27 @@ func (c *AgentConfig) Read(path string) error {
|
||||
c.IPReportPeriod = 30
|
||||
}
|
||||
|
||||
if c.Server == "" {
|
||||
return errors.New("server address should not be empty")
|
||||
}
|
||||
|
||||
if c.ClientSecret == "" {
|
||||
return errors.New("client_secret must be specified")
|
||||
}
|
||||
|
||||
if c.ReportDelay < 1 || c.ReportDelay > 4 {
|
||||
return errors.New("report-delay ranges from 1-4")
|
||||
}
|
||||
|
||||
if c.UUID == "" {
|
||||
if uuid, err := uuid.GenerateUUID(); err == nil {
|
||||
c.UUID = uuid
|
||||
return saveOnce()
|
||||
} else {
|
||||
return fmt.Errorf("generate UUID failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -68,5 +113,28 @@ func (c *AgentConfig) Save() error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return os.WriteFile(c.v.ConfigFileUsed(), data, os.ModePerm)
|
||||
|
||||
dir := filepath.Dir(c.filePath)
|
||||
if err := os.MkdirAll(dir, 0750); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return os.WriteFile(c.filePath, data, 0600)
|
||||
}
|
||||
|
||||
type kubeyaml struct{}
|
||||
|
||||
// Unmarshal parses the given YAML bytes.
|
||||
func (k *kubeyaml) Unmarshal(b []byte) (map[string]interface{}, error) {
|
||||
var out map[string]interface{}
|
||||
if err := yaml.Unmarshal(b, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// Marshal marshals the given config map to YAML bytes.
|
||||
func (k *kubeyaml) Marshal(o map[string]interface{}) ([]byte, error) {
|
||||
return yaml.Marshal(o)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user