package netlink

import (
	"net/netip"

	"github.com/vishvananda/netlink"
)

type Route struct {
	LinkIndex int
	Dst       netip.Prefix
	Src       netip.Addr
	Gw        netip.Addr
	Priority  int
	Family    int
	Table     int
	Type      int
}

func (n *NetLink) RouteList(link *Link, family int) (
	routes []Route, err error) {
	netlinkLink := linkToNetlinkLink(link)
	netlinkRoutes, err := netlink.RouteList(netlinkLink, family)
	if err != nil {
		return nil, err
	}

	routes = make([]Route, len(netlinkRoutes))
	for i := range netlinkRoutes {
		routes[i] = netlinkRouteToRoute(netlinkRoutes[i])
	}
	return routes, nil
}

func (n *NetLink) RouteAdd(route Route) error {
	netlinkRoute := routeToNetlinkRoute(route)
	return netlink.RouteAdd(&netlinkRoute)
}

func (n *NetLink) RouteDel(route Route) error {
	netlinkRoute := routeToNetlinkRoute(route)
	return netlink.RouteDel(&netlinkRoute)
}

func (n *NetLink) RouteReplace(route Route) error {
	netlinkRoute := routeToNetlinkRoute(route)
	return netlink.RouteReplace(&netlinkRoute)
}

func netlinkRouteToRoute(netlinkRoute netlink.Route) (route Route) {
	return Route{
		LinkIndex: netlinkRoute.LinkIndex,
		Dst:       netIPNetToNetipPrefix(netlinkRoute.Dst),
		Src:       netIPToNetipAddress(netlinkRoute.Src),
		Gw:        netIPToNetipAddress(netlinkRoute.Gw),
		Priority:  netlinkRoute.Priority,
		Family:    netlinkRoute.Family,
		Table:     netlinkRoute.Table,
		Type:      netlinkRoute.Type,
	}
}

func routeToNetlinkRoute(route Route) (netlinkRoute netlink.Route) {
	return netlink.Route{
		LinkIndex: route.LinkIndex,
		Dst:       netipPrefixToIPNet(route.Dst),
		Src:       netipAddrToNetIP(route.Src),
		Gw:        netipAddrToNetIP(route.Gw),
		Priority:  route.Priority,
		Family:    route.Family,
		Table:     route.Table,
		Type:      route.Type,
	}
}