//go:build plus

package nodeconfigs

import (
	"fmt"
	"github.com/iwind/TeaGo/rands"
	"net"
	"runtime"
)

// NewTOAConfig 默认的TOA配置
func NewTOAConfig() *TOAConfig {
	return &TOAConfig{
		IsOn:         false,
		Debug:        false,
		OptionTypeV4: 0xfe,
		OptionTypeV6: 0xfe,
		MinQueueId:   100,
		AutoSetup:    true,
	}
}

// TOAConfig TOA相关配置
type TOAConfig struct {
	IsOn          bool     `yaml:"isOn" json:"isOn"` // 是否启用
	Debug         bool     `yaml:"debug" json:"debug"`
	OldOptionType uint8    `yaml:"optionType" json:"optionType"`     // Deprecated 已过期，使用 OptionTypeV4 和 OptionTypeV6 代替
	OptionTypeV4  uint8    `yaml:"optionTypeV4" json:"optionTypeV4"` // IPv4的类型
	OptionTypeV6  uint8    `yaml:"OptionTypeV6" json:"optionTypeV6"`
	MinQueueId    uint8    `yaml:"minQueueId" json:"minQueueId"`
	AutoSetup     bool     `yaml:"autoSetup" json:"autoSetup"`
	MinLocalPort  uint16   `yaml:"minLocalPort" json:"minLocalPort"` // 本地可使用的最小端口 TODO
	MaxLocalPort  uint16   `yaml:"maxLocalPort" json:"maxLocalPort"` // 本地可使用的最大端口 TODO
	SockPath      string   `yaml:"sockPath" json:"sockPath"`         // Sock文件路径 TODO
	ByPassPorts   []uint16 `yaml:"byPassPorts" json:"byPassPorts"`   // 忽略的端口 TODO

	minLocalPort int
	maxLocalPort int
}

func (this *TOAConfig) Init() error {
	// LocalPort
	minPort := this.MinLocalPort
	maxPort := this.MaxLocalPort
	if minPort == 0 {
		minPort = 1025
	}
	if maxPort == 0 {
		maxPort = 65534
	}
	if minPort > maxPort {
		minPort, maxPort = maxPort, minPort
	}
	this.minLocalPort = int(minPort)
	this.maxLocalPort = int(maxPort)

	// 兼容
	if this.OptionTypeV4 == 0 && this.OldOptionType > 0 {
		this.OptionTypeV4 = this.OldOptionType
	}
	if this.OptionTypeV6 == 0 && this.OldOptionType > 0 {
		this.OptionTypeV6 = this.OldOptionType
	}

	return nil
}

// SockFile Sock路径
func (this *TOAConfig) SockFile() string {
	if len(this.SockPath) == 0 {
		return "/tmp/edge-toa.sock"
	}
	return this.SockPath
}

// RandLocalPort 获取随机端口
func (this *TOAConfig) RandLocalPort() uint16 {
	listener, err := net.Listen("tcp", ":0")
	if err != nil {
		return uint16(rands.Int(this.minLocalPort, this.maxLocalPort))
	}
	_ = listener.Close()
	return uint16(listener.Addr().(*net.TCPAddr).Port)
}

// AsArgs 转换为参数的形式
func (this *TOAConfig) AsArgs() (args []string) {
	args = append(args, "run")
	args = append(args, "-option-type-v4="+fmt.Sprintf("%d", this.OptionTypeV4))
	args = append(args, "-option-type-v6="+fmt.Sprintf("%d", this.OptionTypeV6))
	args = append(args, "-min-queue-id="+fmt.Sprintf("%d", this.MinQueueId))
	args = append(args, "-max-queue-id="+fmt.Sprintf("%d", this.MaxQueueId()))
	if this.AutoSetup {
		args = append(args, "-auto-setup")
	}
	if this.Debug {
		args = append(args, "-debug")
	}
	return
}

// MaxQueueId 获取队列ID最大值
func (this *TOAConfig) MaxQueueId() uint8 {
	var maxQueueId = int(this.MinQueueId) + runtime.NumCPU() - 1
	if maxQueueId > 255 {
		maxQueueId = 255
	}
	return uint8(maxQueueId)
}