Merge pull request #22 from 403unlocker/feat/readFromConfigFile

feat(): Add reading config file from home dir

Author: ArmanTaheriGhaleTaki

Makefile

@@ -2,6 +2,9 @@
 OUTPUT = 403unlocker
 MAIN = cmd/403unlockercli/main.go
 BIN_DIR = ~/.local/bin
+CONFIG_DIR= ~/.config/403unlocker
+DNS_CONFIG_FILE_URL=https://raw.githubusercontent.com/403unlocker/403Unlocker-cli/refs/heads/main/config/dns.conf
+DOCKER_CONFIG_FILE_URL=https://raw.githubusercontent.com/403unlocker/403Unlocker-cli/refs/heads/main/config/dockerRegistry.conf
 
 .DEFAULT_GOAL := help
 
@@ -34,8 +37,14 @@ clean:
 install: build 
 	@echo "Installing $(OUTPUT) to $(BIN_DIR)..."
 	@install -m 755 $(OUTPUT) $(BIN_DIR)
+	@echo "Downloading config files dns.conf to $(CONFIG_DIR)..."
+	@wget $(DNS_CONFIG_FILE_URL) -q -P $(CONFIG_DIR)
+	@echo "Downloading dockerRegistry.conf $(CONFIG_DIR)..."
+	@wget $(DOCKER_CONFIG_FILE_URL) -q -P $(CONFIG_DIR)
 
 
 uninstall:
 	@echo "Removing $(OUTPUT) from $(BIN_DIR)..."
 	@rm -f $(BIN_DIR)/$(OUTPUT)
+	@echo "Removing $(OUTPUT) from $(CONFIG_DIR)..."
+	@rm -rf $(CONFIG_DIR)

internal/check/function.go

@@ -41,16 +41,28 @@ func CheckWithDNS(c *cli.Context) error {
 	url := c.Args().First()
 	url = ensureHTTPS(url)
 
+	fmt.Println("URL: ", url)
+
 	// Print header
 	fmt.Println("\n+--------------------+------------+")
 	fmt.Printf("| %-18s | %-10s |\n", "DNS Server", "Status")
 	fmt.Println("+--------------------+------------+")
 
-	dnsList, err := ReadDNSFromFile("config/dns.conf")
+	dnsList, err := ReadDNSFromFile(common.DNS_CONFIG_FILE)
 	if err != nil {
-		fmt.Println(err)
-		return err
+		err = common.DownloadConfigFile(common.DNS_CONFIG_URL, common.DNS_CONFIG_FILE)
+		if err != nil {
+			return err
+		}
+
+		dnsList, err = ReadDNSFromFile(common.DNS_CONFIG_FILE)
+
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
 	}
+
 	var wg sync.WaitGroup
 	for _, dns := range dnsList {
 		wg.Add(1)
@@ -86,6 +98,12 @@ func CheckWithDNS(c *cli.Context) error {
 }
 
 func ReadDNSFromFile(filename string) ([]string, error) {
+	homeDir := os.Getenv("HOME")
+	if homeDir == "" {
+		fmt.Println("HOME environment variable not set")
+		os.Exit(1)
+	}
+	filename = homeDir + "/" + filename
 	data, err := os.ReadFile(filename)
 	if err != nil {
 		return nil, err

internal/common/function.go

@@ -1,16 +1,31 @@
 package common
 
-import "fmt"
-
-var Reset = "\033[0m"
-var Red = "\033[31m"
-var Green = "\033[32m"
-var Yellow = "\033[33m"
-var Blue = "\033[34m"
-var Magenta = "\033[35m"
-var Cyan = "\033[36m"
-var Gray = "\033[37m"
-var White = "\033[97m"
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+)
+
+const (
+	// Color
+	Reset   = "\033[0m"
+	Red     = "\033[31m"
+	Green   = "\033[32m"
+	Yellow  = "\033[33m"
+	Blue    = "\033[34m"
+	Magenta = "\033[35m"
+	Cyan    = "\033[36m"
+	Gray    = "\033[37m"
+	White   = "\033[97m"
+
+	// DNS config
+	DNS_CONFIG_FILE    = ".config/403unlocker/dns.conf"
+	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"
+)
 
 // FormatDataSize converts the size in bytes to a human-readable string in KB, MB, or GB.
 func FormatDataSize(bytes int64) string {
@@ -31,3 +46,51 @@ func FormatDataSize(bytes int64) string {
 		return fmt.Sprintf("%d Bytes", bytes)
 	}
 }
+
+func DownloadConfigFile(url, path string) error {
+
+	homeDir := os.Getenv("HOME")
+	if homeDir == "" {
+		fmt.Println("HOME environment variable not set")
+		os.Exit(1)
+	}
+	filePath := homeDir + "/" + path
+
+	dir := filepath.Dir(filePath)
+	if err := os.MkdirAll(dir, os.ModePerm); err != nil {
+		fmt.Printf("Error creating directory: %v\n", err)
+		return err
+	}
+
+	out, err := os.Create(filePath)
+
+	if err != nil {
+		fmt.Println(err)
+
+		return err
+	}
+	defer out.Close()
+
+	if err != nil {
+		fmt.Println("Could not download config file.")
+		return err
+	}
+
+	resp, err := http.Get(url)
+
+	if err != nil {
+		fmt.Println("Could not get the response: ", err)
+		return err
+	}
+
+	defer resp.Body.Close()
+
+	_, err = io.Copy(out, resp.Body)
+
+	if err != nil {
+		fmt.Println("Could not copy content file")
+		return err
+	}
+
+	return nil
+}

internal/dns/function.go

@@ -32,10 +32,19 @@ func URLValidator(URL string) bool {
 func CheckWithURL(c *cli.Context) error {
 	fileToDownload := c.Args().First()
 	timeout := c.Int("timeout")
-	dnsList, err := check.ReadDNSFromFile("config/dns.conf")
+
+	dnsList, err := check.ReadDNSFromFile(common.DNS_CONFIG_FILE)
 	if err != nil {
-		fmt.Println("Error reading DNS list:", err)
-		return err
+		err = common.DownloadConfigFile(common.DNS_CONFIG_URL, common.DNS_CONFIG_FILE)
+		if err != nil {
+			return err
+		}
+		dnsList, err = check.ReadDNSFromFile(common.DNS_CONFIG_FILE)
+
+		if err != nil {
+			fmt.Println("Error reading DNS list:", err)
+			return err
+		}
 	}
 
 	dnsSizeMap := make(map[string]int64)

internal/docker/function.go

@@ -112,10 +112,19 @@ func CheckWithDockerImage(c *cli.Context) error {
 		return fmt.Errorf("image name cannot be empty")
 	}
 
-	registryList, err := check.ReadDNSFromFile("config/dockerRegistry.conf")
+	registryList, err := check.ReadDNSFromFile(common.DOCKER_CONFIG_FILE)
 	if err != nil {
-		log.Printf("Error reading registry list: %v", err)
-		return err
+		err = common.DownloadConfigFile(common.DOCKER_CONFIG_URL, common.DOCKER_CONFIG_FILE)
+		if err != nil {
+			return err
+		}
+
+		registryList, err = check.ReadDNSFromFile(common.DOCKER_CONFIG_FILE)
+		if err != nil {
+			log.Printf("Error reading registry list: %v", err)
+			return err
+		}
+
 	}
 
 	// Find the longest registry name first