package prox5

import (
	"strings"
	"time"

	rl "github.com/yunginnanet/Rate5"

	"git.tcp.direct/kayos/prox5/internal/pools"
)

// https://pkg.go.dev/github.com/yunginnanet/Rate5#Policy
var defaultUseProxyRatelimiter = rl.Policy{
	Window: 55,
	Burst:  55,
}

var defaultBadProxyRateLimiter = rl.Policy{
	Window: 55,
	Burst:  25,
}

const (
	stateUnlocked uint32 = iota
	stateLocked
)

// Proxy represents an individual proxy
type Proxy struct {
	// Endpoint is the address:port of the proxy that we connect to
	Endpoint string
	// ProxiedIP is the address that we end up having when making proxied requests through this proxy
	ProxiedIP string
	// protocol is the version/Protocol (currently SOCKS* only) of the proxy
	protocol proto
	// lastValidated is the time this proxy was last verified working
	lastValidated time.Time
	// timesValidated is the amount of times the proxy has been validated.
	timesValidated int64
	// timesBad is the amount of times the proxy has been marked as bad.
	timesBad int64

	parent *Swamp
	lock   uint32
}

// UniqueKey is an implementation of the Identity interface from Rate5.
// See: https://pkg.go.dev/github.com/yunginnanet/Rate5#Identity
func (sock *Proxy) UniqueKey() string {
	return sock.Endpoint
}

// GetProto retrieves the known protocol value of the Proxy.
func (sock *Proxy) GetProto() ProxyProtocol {
	return sock.protocol.Get()
}

// GetProto safely retrieves the protocol value of the Proxy.
func (sock *Proxy) String() string {
	tout := ""
	if sock.parent.GetServerTimeoutStr() != "-1" {
		tbuf := pools.CopABuffer.Get().(*strings.Builder)
		tbuf.WriteString("?timeout=")
		tbuf.WriteString(sock.parent.GetServerTimeoutStr())
		tbuf.WriteString("s")
		tout = tbuf.String()
		pools.DiscardBuffer(tbuf)
	}
	buf := pools.CopABuffer.Get().(*strings.Builder)
	buf.WriteString(sock.GetProto().String())
	buf.WriteString("://")
	buf.WriteString(sock.Endpoint)
	if tout != "" {
		buf.WriteString(tout)
	}
	out := buf.String()
	pools.DiscardBuffer(buf)
	return out
}