0a890a2021
* make edit and service argument subcommands * generate uuid if non-exist, use default report_delay or ip_report_period value if not specified
68 lines
3.0 KiB
Go
68 lines
3.0 KiB
Go
package model
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/spf13/viper"
|
|
"sigs.k8s.io/yaml"
|
|
)
|
|
|
|
type AgentConfig struct {
|
|
Debug bool `mapstructure:"debug" json:"debug"`
|
|
|
|
Server string `mapstructure:"server" json:"server"` // 服务器地址
|
|
ClientSecret string `mapstructure:"client_secret" json:"client_secret"` // 客户端密钥
|
|
UUID string `mapstructure:"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获取更新
|
|
IPReportPeriod uint32 `mapstructure:"ip_report_period" json:"ip_report_period"` // IP上报周期
|
|
|
|
v *viper.Viper
|
|
}
|
|
|
|
// Read 从给定的文件目录加载配置文件
|
|
func (c *AgentConfig) Read(path string) error {
|
|
c.v = viper.New()
|
|
c.v.SetConfigFile(path)
|
|
err := c.v.ReadInConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = c.v.Unmarshal(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if c.ReportDelay == 0 {
|
|
c.ReportDelay = 1
|
|
}
|
|
|
|
if c.IPReportPeriod == 0 {
|
|
c.IPReportPeriod = 1800
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *AgentConfig) Save() error {
|
|
data, err := yaml.Marshal(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(c.v.ConfigFileUsed(), data, os.ModePerm)
|
|
}
|