refactor continues

agent/tactical/shared/shared_windows.go

@@ -0,0 +1,70 @@
+package shared
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"time"
+
+	"github.com/amidaware/rmmagent/agent/system"
+	"github.com/amidaware/rmmagent/agent/tactical/config"
+	"github.com/amidaware/rmmagent/agent/utils"
+	"github.com/go-resty/resty/v2"
+)
+
+func GetPython(force bool) {
+	if utils.FileExists(system.GetPythonBin()) && !force {
+		return
+	}
+
+	var archZip string
+	var folder string
+	switch runtime.GOARCH {
+	case "amd64":
+		archZip = "py38-x64.zip"
+		folder = "py38-x64"
+	case "386":
+		archZip = "py38-x32.zip"
+		folder = "py38-x32"
+	}
+	pyFolder := filepath.Join(system.GetProgramDirectory(), folder)
+	pyZip := filepath.Join(system.GetProgramDirectory(), archZip)
+	defer os.Remove(pyZip)
+
+	if force {
+		os.RemoveAll(pyFolder)
+	}
+
+	config := config.NewAgentConfig()
+	rClient := resty.New()
+	rClient.SetTimeout(20 * time.Minute)
+	rClient.SetRetryCount(10)
+	rClient.SetRetryWaitTime(1 * time.Minute)
+	rClient.SetRetryMaxWaitTime(15 * time.Minute)
+	if len(config.Proxy) > 0 {
+		rClient.SetProxy(config.Proxy)
+	}
+
+	url := fmt.Sprintf("https://github.com/amidaware/rmmagent/releases/download/v2.0.0/%s", archZip)
+	r, err := rClient.R().SetOutput(pyZip).Get(url)
+	if err != nil {
+		return
+	}
+	if r.IsError() {
+		return
+	}
+
+	err = utils.Unzip(pyZip, system.GetProgramDirectory())
+	if err != nil {
+	}
+}
+
+func RunMigrations() {
+	for _, i := range []string{"nssm.exe", "nssm-x86.exe"} {
+		nssm := filepath.Join(system.GetProgramDirectory(), i)
+		if utils.FileExists(nssm) {
+			os.Remove(nssm)
+		}
+	}
+}