netmaker/netclient/daemon/windows.go

package daemon

import (
	"fmt"
	"log"
	"os"
	"strings"

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

// SetupWindowsDaemon - sets up the Windows daemon service
func SetupWindowsDaemon() error {

	if !ncutils.FileExists(ncutils.GetNetclientPathSpecific() + "winsw.xml") {
		if err := writeServiceConfig(); err != nil {
			return err
		}
	}

	if !ncutils.FileExists(ncutils.GetNetclientPathSpecific() + "winsw.exe") {
		ncutils.Log("performing first time daemon setup")
		err := ncutils.GetEmbedded()
		if err != nil {
			return err
		}
		ncutils.Log("finished daemon setup")
	}
	// install daemon, will not overwrite
	ncutils.RunCmd(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1)+`winsw.exe install`, false)
	// start daemon, will not restart or start another
	ncutils.RunCmd(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1)+`winsw.exe start`, false)
	ncutils.Log(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1) + `winsw.exe start`)
	return nil
}

// RestartWindowsDaemon - restarts windows service
func RestartWindowsDaemon() {
	StopWindowsDaemon()
	// start daemon, will not restart or start another
	ncutils.RunCmd(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1)+`winsw.exe start`, false)
}

// CleanupWindows - cleans up windows files
func CleanupWindows() {
	if !ncutils.FileExists(ncutils.GetNetclientPathSpecific() + "winsw.xml") {
		writeServiceConfig()
	}
	StopWindowsDaemon()
	RemoveWindowsDaemon()
	os.RemoveAll(ncutils.GetNetclientPath())
	log.Println("Netclient on Windows, uninstalled")
}

func writeServiceConfig() error {
	serviceConfigPath := ncutils.GetNetclientPathSpecific() + "winsw.xml"
	scriptString := fmt.Sprintf(`<service>
<id>netclient</id>
<name>Netclient</name>
<description>Connects Windows nodes to one or more Netmaker networks.</description>
<executable>%v</executable>
<arguments>daemon</arguments>
<log mode="roll"></log>
</service>
`, strings.Replace(ncutils.GetNetclientPathSpecific()+"netclient.exe", `\\`, `\`, -1))
	if !ncutils.FileExists(serviceConfigPath) {
		err := os.WriteFile(serviceConfigPath, []byte(scriptString), 0600)
		if err != nil {
			return err
		}
		ncutils.Log("wrote the daemon config file to the Netclient directory")
	}
	return nil
}

// == Daemon ==

// StopWindowsDaemon - stops the Windows daemon
func StopWindowsDaemon() {
	ncutils.Log("stopping Windows, Netclient daemon")
	// stop daemon, will not overwrite
	ncutils.RunCmd(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1)+`winsw.exe stop`, true)
}

// RemoveWindowsDaemon - removes the Windows daemon
func RemoveWindowsDaemon() {
	// uninstall daemon, will not restart or start another
	ncutils.RunCmd(strings.Replace(ncutils.GetNetclientPathSpecific(), `\\`, `\`, -1)+`winsw.exe uninstall`, true)
	ncutils.Log("uninstalled Windows, Netclient daemon")
}

// func copyWinswOver() error {

// 	input, err := ioutil.ReadFile(".\\winsw.exe")
// 	if err != nil {
// 		ncutils.Log("failed to find winsw.exe")
// 		return err
// 	}
// 	if err = ioutil.WriteFile(ncutils.GetNetclientPathSpecific()+"winsw.exe", input, 0644); err != nil {
// 		ncutils.Log("failed to copy winsw.exe to " + ncutils.GetNetclientPath())
// 		return err
// 	}
// 	if err = os.Remove(".\\winsw.exe"); err != nil {
// 		ncutils.Log("failed to cleanup local winsw.exe, feel free to delete it")
// 		return err
// 	}
// 	ncutils.Log("finished copying winsw.exe")
// 	return nil
// }

// func downloadWinsw() error {
// 	fullURLFile := "https://github.com/winsw/winsw/releases/download/v2.11.0/WinSW-x64.exe"
// 	fileName := "winsw.exe"

// 	// Create the file
// 	file, err := os.Create(fileName)
// 	if err != nil {
// 		ncutils.Log("could not create file on OS for Winsw")
// 		return err
// 	}
// 	defer file.Close()

// 	client := http.Client{
// 		CheckRedirect: func(r *http.Request, via []*http.Request) error {
// 			r.URL.Opaque = r.URL.Path
// 			return nil
// 		},
// 	}
// 	// Put content on file
// 	ncutils.Log("downloading service tool...")
// 	resp, err := client.Get(fullURLFile)
// 	if err != nil {
// 		ncutils.Log("could not GET Winsw")
// 		return err
// 	}
// 	defer resp.Body.Close()

// 	_, err = io.Copy(file, resp.Body)
// 	if err != nil {
// 		ncutils.Log("could not mount winsw.exe")
// 		return err
// 	}
// 	ncutils.Log("finished downloading Winsw")
// 	return nil
// }