package wireguard

import (
	"context"
	"net/netip"
	"time"

	tun "github.com/sagernet/sing-tun"
	"github.com/sagernet/sing/common/json/badoption"
	"github.com/sagernet/sing/common/logger"
	M "github.com/sagernet/sing/common/metadata"
	N "github.com/sagernet/sing/common/network"
)

type EndpointOptions struct {
	Context                    context.Context
	Logger                     logger.ContextLogger
	System                     bool
	Handler                    tun.Handler
	UDPTimeout                 time.Duration
	ICMPTimeout                time.Duration
	Dialer                     N.Dialer
	CreateDialer               func(interfaceName string) N.Dialer
	Name                       string
	MTU                        uint32
	Address                    []netip.Prefix
	PrivateKey                 string
	ListenPort                 uint16
	ResolvePeer                func(domain string) (netip.Addr, error)
	Peers                      []PeerOptions
	Workers                    int
	PreallocatedBuffersPerPool uint32
	DisablePauses              bool
	Amnezia                    *AmneziaOptions
}

type PeerOptions struct {
	Endpoint                    M.Socksaddr
	PublicKey                   string
	PreSharedKey                string
	AllowedIPs                  []netip.Prefix
	PersistentKeepaliveInterval uint16
}

type AmneziaOptions struct {
	JC                     int
	JMin                   int
	JMax                   int
	S1                     int
	S2                     int
	S3                     int
	S4                     int
	H1                     *badoption.Range[uint32]
	H2                     *badoption.Range[uint32]
	H3                     *badoption.Range[uint32]
	H4                     *badoption.Range[uint32]
	I1                     string
	I2                     string
	I3                     string
	I4                     string
	I5                     string
	HeaderProtectionKey    string
	ContentPaddingAddition *badoption.Range[uint32]
	RekeyAfterTime         *badoption.Range[uint32]
	RekeyTimeout           *badoption.Range[uint32]
	RejectAfterTime        *badoption.Range[uint32]
	KeepaliveTimeout       *badoption.Range[uint32]
	MaxHandshakeAttempts   *badoption.Range[uint32]
}