package config

import (
	"fmt"
	"time"

	"github.com/spf13/pflag"
	"github.com/spf13/viper"
	"github.com/taosdata/taosadapter/v3/version"
)

type Monitor struct {
	Disable                   bool
	CollectDuration           time.Duration
	DisableClientIP           bool
	InCGroup                  bool
	PauseQueryMemoryThreshold float64
	PauseAllMemoryThreshold   float64
	Identity                  string
	WriteToTD                 bool
	User                      string
	Password                  string
	WriteInterval             time.Duration
}

func initMonitor() {
	viper.SetDefault("monitor.disable", false)
	_ = viper.BindEnv("monitor.disable", "TAOS_ADAPTER_MONITOR_DISABLE")
	pflag.Bool("monitor.disable", false, `Whether to disable monitoring. Env "TAOS_ADAPTER_MONITOR_DISABLE"`)

	viper.SetDefault("monitor.collectDuration", time.Second*3)
	_ = viper.BindEnv("monitor.collectDuration", "TAOS_ADAPTER_MONITOR_COLLECT_DURATION")
	pflag.Duration("monitor.collectDuration", time.Second*3, `Set monitor duration. Env "TAOS_ADAPTER_MONITOR_COLLECT_DURATION"`)

	viper.SetDefault("monitor.disableCollectClientIP", false)
	_ = viper.BindEnv("monitor.disableCollectClientIP", "TAOS_ADAPTER_MONITOR_DISABLE_COLLECT_CLIENT_IP")
	pflag.Bool("monitor.disableCollectClientIP", false, `Whether to disable collecting clientIP. Env "TAOS_ADAPTER_MONITOR_DISABLE_COLLECT_CLIENT_IP"`)

	viper.SetDefault("monitor.incgroup", false)
	_ = viper.BindEnv("monitor.incgroup", "TAOS_ADAPTER_MONITOR_INCGROUP")
	pflag.Bool("monitor.incgroup", false, `Whether running in cgroup. Env "TAOS_ADAPTER_MONITOR_INCGROUP"`)

	viper.SetDefault("monitor.pauseQueryMemoryThreshold", 70)
	_ = viper.BindEnv("monitor.pauseQueryMemoryThreshold", "TAOS_ADAPTER_MONITOR_PAUSE_QUERY_MEMORY_THRESHOLD")
	pflag.Float64("monitor.pauseQueryMemoryThreshold", 70, `Memory percentage threshold for pause query. Env "TAOS_ADAPTER_MONITOR_PAUSE_QUERY_MEMORY_THRESHOLD"`)

	viper.SetDefault("monitor.pauseAllMemoryThreshold", 80)
	_ = viper.BindEnv("monitor.pauseAllMemoryThreshold", "TAOS_ADAPTER_MONITOR_PAUSE_ALL_MEMORY_THRESHOLD")
	pflag.Float64("monitor.pauseAllMemoryThreshold", 80, `Memory percentage threshold for pause all. Env "TAOS_ADAPTER_MONITOR_PAUSE_ALL_MEMORY_THRESHOLD"`)

	viper.SetDefault("monitor.identity", "")
	_ = viper.BindEnv("monitor.identity", "TAOS_ADAPTER_MONITOR_IDENTITY")
	pflag.String("monitor.identity", "", `The identity of the current instance, or 'hostname:port' if it is empty. Env "TAOS_ADAPTER_MONITOR_IDENTITY"`)

	viper.SetDefault("monitor.writeToTD", false)
	_ = viper.BindEnv("monitor.writeToTD", "TAOS_ADAPTER_MONITOR_WRITE_TO_TD")
	pflag.Bool("monitor.writeToTD", false, fmt.Sprintf(`Whether write metrics to %s. Env "TAOS_ADAPTER_MONITOR_WRITE_TO_TD"`, version.CUS_NAME))

	viper.SetDefault("monitor.user", "root")
	_ = viper.BindEnv("monitor.user", "TAOS_ADAPTER_MONITOR_USER")
	pflag.String("monitor.user", "root", fmt.Sprintf(`%s user. Env "TAOS_ADAPTER_MONITOR_USER"`, version.CUS_NAME))

	viper.SetDefault("monitor.password", "taosdata")
	_ = viper.BindEnv("monitor.password", "TAOS_ADAPTER_MONITOR_PASSWORD")
	pflag.String("monitor.password", "taosdata", fmt.Sprintf(`%s password. Env "TAOS_ADAPTER_MONITOR_PASSWORD"`, version.CUS_NAME))

	viper.SetDefault("monitor.writeInterval", time.Second*30)
	_ = viper.BindEnv("monitor.writeInterval", "TAOS_ADAPTER_MONITOR_WRITE_INTERVAL")
	pflag.Duration("monitor.writeInterval", time.Second*30, fmt.Sprintf(`Set write to %s interval. Env "TAOS_ADAPTER_MONITOR_WRITE_INTERVAL"`, version.CUS_NAME))
}

func (p *Monitor) setValue() {
	p.Disable = viper.GetBool("monitor.disable")
	p.CollectDuration = viper.GetDuration("monitor.collectDuration")
	p.InCGroup = viper.GetBool("monitor.incgroup")
	p.DisableClientIP = viper.GetBool("monitor.disableCollectClientIP")
	p.PauseQueryMemoryThreshold = viper.GetFloat64("monitor.pauseQueryMemoryThreshold")
	p.PauseAllMemoryThreshold = viper.GetFloat64("monitor.pauseAllMemoryThreshold")
	p.Identity = viper.GetString("monitor.identity")
	p.WriteToTD = viper.GetBool("monitor.writeToTD")
	p.User = viper.GetString("monitor.user")
	p.Password = viper.GetString("monitor.password")
	p.WriteInterval = viper.GetDuration("monitor.writeInterval")
}