addrport.go

package core

import (
	"net"
	"net/netip"
)

// AddrPort attempts to extract a netip.AddrPort from an object
func AddrPort(v any) (netip.AddrPort, bool) {
	// known types first
	if addr, ok := typeSpecificAddrPort(v); ok {
		return addr, ok
	}

	// via interfaces
	if p, ok := v.(interface {
		AddrPort() netip.AddrPort
	}); ok {
		return p.AddrPort(), true
	}

	if p, ok := v.(interface {
		Addr() net.Addr
	}); ok {
		return AddrPort(p.Addr())
	}

	if p, ok := v.(interface {
		RemoteAddr() net.Addr
	}); ok {
		return AddrPort(p.RemoteAddr())
	}

	// sorry
	return netip.AddrPort{}, false
}

func typeSpecificAddrPort(v any) (netip.AddrPort, bool) {
	switch addr := v.(type) {
	case netip.AddrPort:
		return addr, true
	case *netip.AddrPort:
		return *addr, true
	case *net.TCPAddr:
		if ip, ok := netip.AddrFromSlice(addr.IP); ok {
			return netip.AddrPortFrom(ip, uint16(addr.Port)), true
		}
	case *net.UDPAddr:
		if ip, ok := netip.AddrFromSlice(addr.IP); ok {
			return netip.AddrPortFrom(ip, uint16(addr.Port)), true
		}
	}

	return netip.AddrPort{}, false
}