package cli

import (
	"fmt"

	"github.com/DataDrake/cli-ng/v2/cmd"
	"github.com/hyprspace/hyprspace/rpc"
)

var Route = cmd.Sub{
	Name:  "route",
	Alias: "r",
	Short: "Control routing",
	Args:  &RouteArgs{},
	Run:   RouteRun,
}

type RouteArgs struct {
	Action        string
	InterfaceName string
	Args          []string `zero:"true"`
}

func RouteRun(r *cmd.Root, c *cmd.Sub) {
	// Parse Command Args
	args := c.Args.(*RouteArgs)

	action := rpc.RouteAction(args.Action)
	rArgs := rpc.RouteArgs{
		Action: action,
		Args:   args.Args,
	}
	reply := rpc.Route(args.InterfaceName, rArgs)
	for _, r := range reply.Routes {
		var target string
		connectStatus := ""
		if r.IsRelay {
			target = fmt.Sprintf("%s relay target %s", r.RelayAddr, r.TargetAddr)
		} else {
			target = fmt.Sprintf("%s direct", r.TargetAddr)
		}
		if r.IsConnected {
			connectStatus = " connected"
		}
		fmt.Printf("%s via %s%s\n", &r.Network, target, connectStatus)
	}
}