package peer

import (
	"log"
	"net/netip"
	"time"

	"vppn/peer/control"
	"vppn/peer/wginterface"
)

func (a *App) onTick() {
	wgPeers := a.devPeers()

	now := time.Now().UnixNano()

	for _, wgPeer := range wgPeers {
		p, ok := a.peersByKey[wgPeer.PublicKey]
		if !ok {
			log.Printf("Wireguard peer not known. Removing: %v", wgPeer.PublicKey)
			a.devRemove(&Peer{wgPeer: wgPeer})
			continue
		}
		p.wgPeer = wgPeer
		if a.isPublic && !p.IsPublic {
			if ep := p.WGEndpoint(); ep != p.EndpointV4 {
				log.Printf("Client %s %s endpoint: %s -> %s", p.Name, p.VPNIP, p.EndpointV4, ep)
				p.EndpointV4 = ep
			}
		}

		// Send pings to peers where we're the client.
		if p.Role == control.Client {
			a.sendPing(p, now)
		}

		switch p.State {
		case StateRelayed:
			// If we have an ep to probe, add it.
			if ep := p.PreferredEndpoint(); ep.IsValid() {
				a.devAddProbe(p, ep)
			}

		case StateProbing:
			// Promote probing peers to direct once alive (direct path confirmed
			// working).
			if time.Since(p.LastHandshakeTime()) < 2*wginterface.ProbeKeepalive {
				a.devPromote(p)
			}

		case StateDirect:
			if p.IsPublic || a.isPublic || p.Up() {
				break
			}

			// Stale non-public direct peer: demote to relayed and wait for new IP
			// information.
			p.EndpointV4 = netip.AddrPort{}
			p.EndpointV6 = netip.AddrPort{}
			p.EndpointLAN = netip.AddrPort{}
			a.devAddPeer(p)
		}
	}

	// Ensure we have a live relay (if we're not public).
	if !a.isPublic && (a.relay == nil || !a.relay.Up()) {
		a.switchActiveRelay()
	}
}