feat(): make command line windows compatible (#29)

Author: ArmanTaheriGhaleTaki

internal/common/function.go

@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strings"
 )
 
@@ -29,6 +30,9 @@ const (
 	DOCKER_CONFIG_FILE      = ".config/403unlocker/dockerRegistry.conf"
 	DNS_CONFIG_URL          = "https://raw.githubusercontent.com/403unlocker/403Unlocker-cli/refs/heads/main/config/dns.conf"
 	DOCKER_CONFIG_URL       = "https://raw.githubusercontent.com/403unlocker/403Unlocker-cli/refs/heads/main/config/dockerRegistry.conf"
+
+	// OS names
+	WINDOWS_OS_NAME = "windows"
 )
 
 // FormatDataSize converts the size in bytes to a human-readable string in KB, MB, or GB.
@@ -52,14 +56,12 @@ func FormatDataSize(bytes int64) string {
 }
 
 func DownloadConfigFile(url, path string) error {
-
-	homeDir := os.Getenv("HOME")
+	homeDir := GetHomeDir()
 	if homeDir == "" {
 		fmt.Println("HOME environment variable not set")
 		os.Exit(1)
 	}
-	filePath := homeDir + "/" + path
-
+	filePath := AddPathToDir(homeDir, path)
 	dir := filepath.Dir(filePath)
 	if err := os.MkdirAll(dir, os.ModePerm); err != nil {
 		fmt.Printf("Error creating directory: %v\n", err)
@@ -100,14 +102,12 @@ func DownloadConfigFile(url, path string) error {
 }
 
 func WriteDNSToFile(filename string, dnsList []string) error {
-	homeDir := os.Getenv("HOME")
+	homeDir := GetHomeDir()
 	if homeDir == "" {
 		fmt.Println("HOME environment variable not set")
 		os.Exit(1)
 	}
-
-	filename = homeDir + "/" + filename
-
+	filename = AddPathToDir(homeDir, filename)
 	_, err := os.Stat(filename)
 	if os.IsNotExist(err) {
 		file, err := os.Create(filename)
@@ -130,12 +130,12 @@ func WriteDNSToFile(filename string, dnsList []string) error {
 }
 
 func ReadDNSFromFile(filename string) ([]string, error) {
-	homeDir := os.Getenv("HOME")
+	homeDir := GetHomeDir()
 	if homeDir == "" {
 		fmt.Println("HOME environment variable not set")
 		os.Exit(1)
 	}
-	filename = homeDir + "/" + filename
+	filename = AddPathToDir(homeDir, filename)
 	data, err := os.ReadFile(filename)
 	if err != nil {
 		return nil, err
@@ -164,3 +164,24 @@ func ChangeDNS(dns string) *http.Client {
 	}
 	return client
 }
+func GetHomeDir() string {
+	if runtime.GOOS == WINDOWS_OS_NAME {
+		return os.Getenv("USERPROFILE")
+	} else {
+		return os.Getenv("HOME")
+	}
+}
+func GetTempDir() string {
+	if runtime.GOOS == WINDOWS_OS_NAME {
+		return os.Getenv("TEMP")
+	} else {
+		return "/tmp"
+	}
+}
+func AddPathToDir(baseDir, extraDir string) string {
+	if runtime.GOOS == WINDOWS_OS_NAME {
+		return baseDir + "\\" + extraDir
+	} else {
+		return baseDir + "/" + extraDir
+	}
+}

internal/dns/function.go

@@ -161,7 +161,9 @@ func CheckWithURL(c *cli.Context) error {
 	fmt.Printf("| %-18s | %-14s |\n", "DNS Server", "Download Speed")
 	fmt.Println("+--------------------+----------------+")
 
-	tempDir := time.Now().UnixMilli()
+	rand := time.Now().UnixMilli()
+	tempDir := common.AddPathToDir(common.GetTempDir(), strconv.Itoa(int(rand)))
+
 	var wg sync.WaitGroup
 	for _, dns := range dnsList {
 		ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout)*time.Second)
@@ -170,8 +172,8 @@ func CheckWithURL(c *cli.Context) error {
 		clientWithCustomDNS := common.ChangeDNS(dns)
 		client := grab.NewClient()
 		client.HTTPClient = clientWithCustomDNS
+		req, err := grab.NewRequest(fmt.Sprintf("%v", tempDir), fileToDownload)
 
-		req, err := grab.NewRequest(fmt.Sprintf("/tmp/%v", tempDir), fileToDownload)
 		if err != nil {
 			fmt.Fprintf(os.Stderr, "Error creating request for DNS %s: %v\n", dns, err)
 		}
@@ -213,6 +215,6 @@ func CheckWithURL(c *cli.Context) error {
 		fmt.Println("No DNS server was able to download any data.")
 	}
 
-	os.RemoveAll(fmt.Sprintf("/tmp/%v", tempDir))
+	os.RemoveAll(fmt.Sprintf("%v", tempDir))
 	return nil
 }

internal/docker/function.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"path/filepath"
 	"regexp"
+	"strconv"
 	"strings"
 	"sync/atomic"
 	"time"
@@ -102,8 +103,8 @@ func CheckWithDockerImage(c *cli.Context) error {
 	registrySizeMap := make(map[string]int64)
 	timeout := c.Int("timeout")
 	imageName := c.Args().First()
-	tempDir := time.Now().UnixMilli()
-
+	rand := time.Now().UnixMilli()
+	tempDir := common.AddPathToDir(common.GetTempDir(), strconv.Itoa(int(rand)))
 	fmt.Printf("\nTimeout: %d seconds\n", timeout)
 	fmt.Printf("Docker Image: %s\n\n", imageName)
 
@@ -147,7 +148,7 @@ func CheckWithDockerImage(c *cli.Context) error {
 		ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout)*time.Second)
 		defer cancel()
 
-		size, err := DownloadDockerImage(ctx, imageName, registry, fmt.Sprintf("/tmp/%v", tempDir))
+		size, err := DownloadDockerImage(ctx, imageName, registry, fmt.Sprintf("%v", tempDir))
 		if err != nil {
 			fmt.Printf("| %-*s | %s%-16s%s |\n",
 				maxLength, registry,
@@ -181,6 +182,6 @@ func CheckWithDockerImage(c *cli.Context) error {
 		fmt.Println("No registry was able to download any data.")
 	}
 
-	os.RemoveAll(fmt.Sprintf("/tmp/%v", tempDir))
+	os.RemoveAll(fmt.Sprintf("%v", tempDir))
 	return nil
 }