package handler

import (
	"github.com/spf13/pflag"

	"github.com/wencaiwulue/kubevpn/v2/pkg/daemon/rpc"
)

type ExtraRouteInfo struct {
	ExtraCIDR   []string
	ExtraDomain []string
	ExtraNodeIP bool
}

func ParseExtraRouteFromRPC(route *rpc.ExtraRoute) *ExtraRouteInfo {
	if route == nil {
		return &ExtraRouteInfo{}
	}
	return &ExtraRouteInfo{
		ExtraCIDR:   route.ExtraCIDR,
		ExtraDomain: route.ExtraDomain,
		ExtraNodeIP: route.ExtraNodeIP,
	}
}

func (e ExtraRouteInfo) ToRPC() *rpc.ExtraRoute {
	return &rpc.ExtraRoute{
		ExtraCIDR:   e.ExtraCIDR,
		ExtraDomain: e.ExtraDomain,
		ExtraNodeIP: e.ExtraNodeIP,
	}
}

func AddExtraRoute(flags *pflag.FlagSet, route *ExtraRouteInfo) {
	flags.StringArrayVar(&route.ExtraCIDR, "extra-cidr", []string{}, "Extra network CIDR string, add those cidr network to route table, eg: --extra-cidr 192.168.0.159/24 --extra-cidr 192.168.1.160/32")
	flags.StringArrayVar(&route.ExtraDomain, "extra-domain", []string{}, "Extra domain string, the resolved IP will add to route table, eg: --extra-domain test.abc.com --extra-domain foo.test.com")
	flags.BoolVar(&route.ExtraNodeIP, "extra-node-ip", false, "Extra node IP, add cluster node IP to route table.")
}
