package lldp

import (
	"time"

	"github.com/google/gopacket"
	"github.com/google/gopacket/afpacket"
	"github.com/jellydator/ttlcache/v3"
	"github.com/jetkvm/kvm/internal/logging"
	"github.com/rs/zerolog"
)

var defaultLogger = logging.GetSubsystemLogger("lldp")

type LLDP struct {
	l         *zerolog.Logger
	tPacket   *afpacket.TPacket
	pktSource *gopacket.PacketSource

	enableRx bool
	enableTx bool

	packets       chan gopacket.Packet
	interfaceName string
	stop          chan struct{}

	neighbors *ttlcache.Cache[string, Neighbor]
}

type LLDPOptions struct {
	InterfaceName string
	EnableRx      bool
	EnableTx      bool
	Logger        *zerolog.Logger
}

func NewLLDP(opts *LLDPOptions) *LLDP {
	if opts.Logger == nil {
		opts.Logger = defaultLogger
	}

	if opts.InterfaceName == "" {
		opts.Logger.Fatal().Msg("InterfaceName is required")
	}

	return &LLDP{
		interfaceName: opts.InterfaceName,
		enableRx:      opts.EnableRx,
		enableTx:      opts.EnableTx,
		l:             opts.Logger,
		neighbors:     ttlcache.New(ttlcache.WithTTL[string, Neighbor](1 * time.Hour)),
	}
}

func (l *LLDP) Start() error {
	if l.enableRx {
		l.l.Info().Msg("setting up AF_PACKET")
		if err := l.setUpCapture(); err != nil {
			l.l.Error().Err(err).Msg("unable to set up AF_PACKET")
			return err
		}
		l.startCapture()
	}

	go l.neighbors.Start()

	return nil
}