package mtypes

import (
	"math"
	"strconv"
	"sync/atomic"

	"github.com/KusakabeSi/EtherGuard-VPN/conn"
)

// Nonnegative integer ID of vertex
type Vertex uint16

const (
	NodeID_Broadcast Vertex = math.MaxUint16 - iota // Normal boardcast, boardcast with route table
	NodeID_Spread    Vertex = math.MaxUint16 - iota // p2p mode: boardcast to every know peer and prevent dup. super mode: send to supernode
	NodeID_SuperNode Vertex = math.MaxUint16 - iota
	NodeID_Invalid   Vertex = math.MaxUint16 - iota
	NodeID_Special   Vertex = NodeID_Invalid
)

type EdgeConfig struct {
	Interface             InterfaceConf    `yaml:"Interface"`
	NodeID                Vertex           `yaml:"NodeID"`
	NodeName              string           `yaml:"NodeName"`
	PostScript            string           `yaml:"PostScript"`
	DefaultTTL            uint8            `yaml:"DefaultTTL"`
	L2FIBTimeout          float64          `yaml:"L2FIBTimeout"`
	PrivKey               string           `yaml:"PrivKey"`
	ListenPort            int              `yaml:"ListenPort"`
	DisableAf             conn.EnabledAf   `yaml:"DisabledAf"`
	AfPrefer              int              `yaml:"AfPrefer"`
	LogLevel              LoggerInfo       `yaml:"LogLevel"`
	DynamicRoute          DynamicRouteInfo `yaml:"DynamicRoute"`
	NextHopTable          NextHopTable     `yaml:"NextHopTable"`
	ResetEndPointInterval float64          `yaml:"ResetEndPointInterval"`
	Peers                 []PeerInfo       `yaml:"Peers"`
}

type SuperConfig struct {
	NodeName                string                  `yaml:"NodeName"`
	PostScript              string                  `yaml:"PostScript"`
	PrivKeyV4               string                  `yaml:"PrivKeyV4"`
	PrivKeyV6               string                  `yaml:"PrivKeyV6"`
	ListenPort              int                     `yaml:"ListenPort"`
	ListenPort_EdgeAPI      string                  `yaml:"ListenPort_EdgeAPI"`
	ListenPort_ManageAPI    string                  `yaml:"ListenPort_ManageAPI"`
	API_Prefix              string                  `yaml:"API_Prefix"`
	RePushConfigInterval    float64                 `yaml:"RePushConfigInterval"`
	HttpPostInterval        float64                 `yaml:"HttpPostInterval"`
	PeerAliveTimeout        float64                 `yaml:"PeerAliveTimeout"`
	SendPingInterval        float64                 `yaml:"SendPingInterval"`
	DampingFilterRadius     uint64                  `yaml:"DampingFilterRadius"`
	LogLevel                LoggerInfo              `yaml:"LogLevel"`
	Passwords               Passwords               `yaml:"Passwords"`
	GraphRecalculateSetting GraphRecalculateSetting `yaml:"GraphRecalculateSetting"`
	NextHopTable            NextHopTable            `yaml:"NextHopTable"`
	EdgeTemplate            string                  `yaml:"EdgeTemplate"`
	UsePSKForInterEdge      bool                    `yaml:"UsePSKForInterEdge"`
	ResetEndPointInterval   float64                 `yaml:"ResetEndPointInterval"`
	Peers                   []SuperPeerInfo         `yaml:"Peers"`
}

type Passwords struct {
	ShowState   string `yaml:"ShowState"`
	AddPeer     string `yaml:"AddPeer"`
	DelPeer     string `yaml:"DelPeer"`
	UpdatePeer  string `yaml:"UpdatePeer"`
	UpdateSuper string `yaml:"UpdateSuper"`
}

type InterfaceConf struct {
	IType         string `yaml:"IType"`
	Name          string `yaml:"Name"`
	VPPIFaceID    uint32 `yaml:"VPPIFaceID"`
	VPPBridgeID   uint32 `yaml:"VPPBridgeID"`
	MacAddrPrefix string `yaml:"MacAddrPrefix"`
	IPv4CIDR      string `yaml:"IPv4CIDR"`
	IPv6CIDR      string `yaml:"IPv6CIDR"`
	IPv6LLPrefix  string `yaml:"IPv6LLPrefix"`
	MTU           uint16 `yaml:"MTU"`
	RecvAddr      string `yaml:"RecvAddr"`
	SendAddr      string `yaml:"SendAddr"`
	L2HeaderMode  string `yaml:"L2HeaderMode"`
}

type PeerInfo struct {
	NodeID              Vertex `yaml:"NodeID"`
	PubKey              string `yaml:"PubKey"`
	PSKey               string `yaml:"PSKey"`
	EndPoint            string `yaml:"EndPoint"`
	PersistentKeepalive uint32 `yaml:"PersistentKeepalive"`
	Static              bool   `yaml:"Static"`
}

type SuperPeerInfo struct {
	NodeID         Vertex  `yaml:"NodeID"`
	Name           string  `yaml:"Name"`
	PubKey         string  `yaml:"PubKey"`
	PSKey          string  `yaml:"PSKey"`
	AdditionalCost float64 `yaml:"AdditionalCost"`
	SkipLocalIP    bool    `yaml:"SkipLocalIP"`
	EndPoint       string  `yaml:"EndPoint"`
	ExternalIP     string  `yaml:"ExternalIP"`
}

type LoggerInfo struct {
	LogLevel    string `yaml:"LogLevel"`
	LogTransit  bool   `yaml:"LogTransit"`
	LogNormal   bool   `yaml:"LogNormal"`
	LogControl  bool   `yaml:"LogControl"`
	LogInternal bool   `yaml:"LogInternal"`
	LogNTP      bool   `yaml:"LogNTP"`
}

func (v *Vertex) ToString() string {
	switch *v {
	case NodeID_Broadcast:
		return "Boardcast"
	case NodeID_Spread:
		return "Spread"
	case NodeID_SuperNode:
		return "Super"
	case NodeID_Invalid:
		return "Invalid"
	default:
		return strconv.Itoa(int(*v))
	}
}

type DynamicRouteInfo struct {
	SendPingInterval     float64   `yaml:"SendPingInterval"`
	PeerAliveTimeout     float64   `yaml:"PeerAliveTimeout"`
	TimeoutCheckInterval float64   `yaml:"TimeoutCheckInterval"`
	ConnNextTry          float64   `yaml:"ConnNextTry"`
	DupCheckTimeout      float64   `yaml:"DupCheckTimeout"`
	AdditionalCost       float64   `yaml:"AdditionalCost"`
	DampingFilterRadius  uint64    `yaml:"DampingFilterRadius"`
	SaveNewPeers         bool      `yaml:"SaveNewPeers"`
	SuperNode            SuperInfo `yaml:"SuperNode"`
	P2P                  P2PInfo   `yaml:"P2P"`
	NTPConfig            NTPInfo   `yaml:"NTPConfig"`
}

type NTPInfo struct {
	UseNTP           bool     `yaml:"UseNTP"`
	MaxServerUse     int      `yaml:"MaxServerUse"`
	SyncTimeInterval float64  `yaml:"SyncTimeInterval"`
	NTPTimeout       float64  `yaml:"NTPTimeout"`
	Servers          []string `yaml:"Servers"`
}

type SuperInfo struct {
	UseSuperNode         bool     `yaml:"UseSuperNode"`
	PSKey                string   `yaml:"PSKey"`
	EndpointV4           string   `yaml:"EndpointV4"`
	PubKeyV4             string   `yaml:"PubKeyV4"`
	EndpointV6           string   `yaml:"EndpointV6"`
	PubKeyV6             string   `yaml:"PubKeyV6"`
	EndpointEdgeAPIUrl   string   `yaml:"EndpointEdgeAPIUrl"`
	SkipLocalIP          bool     `yaml:"SkipLocalIP"`
	AdditionalLocalIP    []string `yaml:"AdditionalLocalIP"`
	SuperNodeInfoTimeout float64  `yaml:"SuperNodeInfoTimeout"`
}

type P2PInfo struct {
	UseP2P                  bool                    `yaml:"UseP2P"`
	SendPeerInterval        float64                 `yaml:"SendPeerInterval"`
	GraphRecalculateSetting GraphRecalculateSetting `yaml:"GraphRecalculateSetting"`
}

type GraphRecalculateSetting struct {
	StaticMode                bool      `yaml:"StaticMode"`
	ManualLatency             DistTable `yaml:"ManualLatency"`
	JitterTolerance           float64   `yaml:"JitterTolerance"`
	JitterToleranceMultiplier float64   `yaml:"JitterToleranceMultiplier"`
	TimeoutCheckInterval      float64   `yaml:"TimeoutCheckInterval"`
	RecalculateCoolDown       float64   `yaml:"RecalculateCoolDown"`
}

type DistTable map[Vertex]map[Vertex]float64
type NextHopTable map[Vertex]map[Vertex]Vertex

type API_connurl struct {
	ExternalV4 map[string]float64
	ExternalV6 map[string]float64
	LocalV4    map[string]float64
	LocalV6    map[string]float64
}

func (Connurl *API_connurl) IsEmpty() bool {
	return len(Connurl.ExternalV4)+len(Connurl.ExternalV6)+len(Connurl.LocalV4)+len(Connurl.LocalV6) == 0
}

func (Connurl *API_connurl) GetList(UseLocal bool) (ret map[string]float64) {
	ret = make(map[string]float64)
	if UseLocal {
		if Connurl.LocalV4 != nil {
			for k, v := range Connurl.LocalV4 {
				ret[k] = v
			}
		}
		if Connurl.LocalV6 != nil {
			for k, v := range Connurl.LocalV6 {
				ret[k] = v
			}
		}
	}
	if Connurl.ExternalV4 != nil {
		for k, v := range Connurl.ExternalV4 {
			ret[k] = v
		}
	}
	if Connurl.ExternalV6 != nil {
		for k, v := range Connurl.ExternalV6 {
			ret[k] = v
		}
	}
	return
}

type API_Peerinfo struct {
	NodeID  Vertex
	PSKey   string
	Connurl *API_connurl
}

type API_SuperParams struct {
	SendPingInterval    float64
	HttpPostInterval    float64
	PeerAliveTimeout    float64
	DampingFilterRadius uint64
	AdditionalCost      float64
}

type StateHash struct {
	Peer       atomic.Value //[32]byte
	SuperParam atomic.Value //[32]byte
	NhTable    atomic.Value //[32]byte
}

type API_Peers map[string]API_Peerinfo // map[PubKey]API_Peerinfo

type JWTSecret [32]byte

const chars = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"