package vpn

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/c3os-io/c3os/internal/machine"
	"github.com/c3os-io/c3os/internal/machine/systemd"
	"github.com/c3os-io/c3os/internal/utils"
	"github.com/c3os-io/c3os/pkg/config"
	yip "github.com/mudler/yip/pkg/schema"
)

func Setup(instance, apiAddress, rootDir string, start bool, c *config.Config) error {

	if c.C3OS == nil || c.C3OS.NetworkToken == "" {
		return fmt.Errorf("no network token defined")
	}

	svc, err := machine.EdgeVPN(instance, rootDir)
	if err != nil {
		return err
	}

	apiAddress = strings.ReplaceAll(apiAddress, "https://", "")
	apiAddress = strings.ReplaceAll(apiAddress, "http://", "")

	vpnOpts := map[string]string{
		"EDGEVPNTOKEN":         c.C3OS.NetworkToken,
		"API":                  "true",
		"APILISTEN":            apiAddress,
		"EDGEVPNLOWPROFILEVPN": "true",
		"DHCP":                 "true",
		"DHCPLEASEDIR":         "/usr/local/.c3os/lease",
	}
	// Override opts with user-supplied
	for k, v := range c.VPN {
		vpnOpts[k] = v
	}

	if c.C3OS.DNS {
		vpnOpts["DNSADDRESS"] = "127.0.0.1:53"
		vpnOpts["DNSFORWARD"] = "true"
		if !utils.IsOpenRCBased() {
			if _, err := os.Stat("/etc/sysconfig/network/config"); err == nil {
				utils.WriteEnv("/etc/sysconfig/network/config", map[string]string{
					"NETCONFIG_DNS_STATIC_SERVERS": "127.0.0.1",
				})
				if utils.Flavor() == "opensuse" {
					// TODO: This is dependant on wickedd, move this out in its own network detection block
					svc, err := systemd.NewService(systemd.WithName("wickedd"))
					if err == nil {
						svc.Restart()
					}
				}
			}
		}
		if err := config.SaveCloudConfig("dns", yip.YipConfig{
			Name: "DNS Configuration",
			Stages: map[string][]yip.Stage{
				config.NetworkStage.String(): {{Dns: yip.DNS{Nameservers: []string{"127.0.0.1"}}}}},
		}); err != nil {
			fmt.Println("Failed installing DNS")
		}
	}

	os.MkdirAll("/etc/systemd/system.conf.d/", 0600)
	// Setup edgevpn instance
	err = utils.WriteEnv(filepath.Join(rootDir, "/etc/systemd/system.conf.d/edgevpn-c3os.env"), vpnOpts)
	if err != nil {
		return err
	}

	err = svc.WriteUnit()
	if err != nil {
		return err
	}

	if start {
		err = svc.Start()
		if err != nil {
			return err
		}

		return svc.Enable()
	}
	return nil
}