package local

import (
	"fmt"
	"net"
	"regexp"
	"strings"
	"time"

	"github.com/gravitl/netmaker/netclient/ncutils"
)

// GetDefaultRoute - Gets the default route (ip and interface) on a windows machine
func GetDefaultRoute() (string, string, error) {
	var ipaddr string
	var iface string
	var err error
	var outLine string
	output, err := ncutils.RunCmd("netsh interface ipv4 show route", false)
	if err != nil {
		return ipaddr, iface, err
	}
	for _, line := range strings.Split(strings.TrimSuffix(output, "\n"), "\n") {
		if strings.Contains(line, "0.0.0.0/0") {
			outLine = line
			break
		}
	}
	if outLine == "" {
		return ipaddr, iface, fmt.Errorf("could not find default gateway")
	}
	space := regexp.MustCompile(`\s+`)
	outputSlice := strings.Split(strings.TrimSpace(space.ReplaceAllString(outLine, " ")), " ")
	ipaddr = outputSlice[len(outputSlice)-1]
	if err = ncutils.CheckIPAddress(ipaddr); err != nil {
		return ipaddr, iface, fmt.Errorf("invalid output for ip address check: " + err.Error())
	}
	iface = "irrelevant"
	return ipaddr, iface, err
}

func setRoute(iface string, addr *net.IPNet, address string) error {
	var err error
	_, err = ncutils.RunCmd("route ADD "+addr.String()+" "+address, false)
	time.Sleep(time.Second >> 2)
	ncutils.RunCmd("route CHANGE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
	return err
}

func deleteRoute(iface string, addr *net.IPNet, address string) error {
	var err error
	_, err = ncutils.RunCmd("route DELETE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
	return err
}

func setCidr(iface, address string, addr *net.IPNet) {
	ncutils.RunCmd("route ADD "+addr.String()+" "+address, false)
	time.Sleep(time.Second >> 2)
	ncutils.RunCmd("route CHANGE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
}

func removeCidr(iface string, addr *net.IPNet, address string) {
	ncutils.RunCmd("route DELETE "+addr.String(), false)
}