move downloader into own package

config.go

@@ -20,8 +20,10 @@ type Config struct {
 }
 
 const (
-	configDirName = "epochcli"
+	configDirName    = "epochcli"
-	configName    = "config.toml"
+	configName       = "config.toml"
+	defaultWowPath   = "/path/to/wow"
+	defaultLaunchCmd = "not configured"
 )
 
 var (

main.go

@@ -1,27 +1,15 @@
 package main
 
 import (
-	"crypto/md5"
-	"encoding/hex"
 	"flag"
 	"fmt"
-	"io"
+	"git.burkey.co/eburk/epochcli/pkg/epoch"
-	"io/fs"
 	"log"
-	"net/http"
 	"os"
 	"os/exec"
-	"path/filepath"
-	"regexp"
 	"strings"
 )
 
-const (
-	manifestUrl      = "https://updater.project-epoch.net/api/v2/manifest"
-	defaultWowPath   = "/path/to/wow"
-	defaultLaunchCmd = "not configured"
-)
-
 func main() {
 	outOfDate, err := needUpdate()
 	if err != nil {
@@ -61,14 +49,14 @@ func main() {
 		log.Fatalf("WowDir in %s is still the default setting, exiting", cfgPath)
 	}
 
-	stats, err := downloadUpdate(config, forceFlag)
+	stats, err := epoch.DownloadUpdate(config.WowDir, forceFlag, config.RemoveUnknownPatches)
 	if err != nil {
 		log.Fatal(err)
 	}
 
-	fmt.Printf("%d files updated\n\n", stats.updated)
+	fmt.Printf("%d files updated\n\n", stats.Updated)
-	if stats.current > 0 {
+	if stats.Current > 0 {
-		fmt.Printf("%d files are already up to date\n\n", stats.current)
+		fmt.Printf("%d files are already up to date\n\n", stats.Current)
 	}
 
 	if configRun {
@@ -102,115 +90,3 @@ func main() {
 		}
 	}
 }
-
-type DownloadStats struct {
-	updated int
-	current int
-}
-
-func downloadUpdate(config *Config, force bool) (DownloadStats, error) {
-	var stats DownloadStats
-
-	manifest, err := getManifest()
-	if err != nil {
-		log.Fatalf("Failed to get manifest: %v\n", err)
-	}
-
-	for _, file := range manifest.Files {
-		path := strings.ReplaceAll(file.Path, `\`, `/`)
-		path = strings.TrimLeft(path, `\`)
-
-		localPath := filepath.Join(config.WowDir, path)
-		localDir := filepath.Dir(localPath)
-		if _, err = os.Stat(localDir); os.IsNotExist(err) {
-			err = os.MkdirAll(localDir, 0755)
-			if err != nil {
-				return stats, fmt.Errorf("failed to create directory %s: %v", localDir, err)
-			}
-		}
-
-		if !force {
-			if _, err = os.Stat(localPath); err == nil {
-				data, err := os.ReadFile(localPath)
-				if err != nil {
-					return stats, err
-				}
-				hashBytes := md5.Sum(data)
-				hash := hex.EncodeToString(hashBytes[:])
-				if hash == file.Hash {
-					fmt.Printf("File %s is up to date\n", localPath)
-					stats.current += 1
-					continue
-				}
-			}
-		}
-
-		fmt.Printf("Updating %s...\n", localPath)
-
-		outFile, err := os.Create(localPath)
-		if err != nil {
-			return stats, err
-		}
-
-		for _, url := range []string{file.Urls.Cloudflare, file.Urls.Digitalocean, file.Urls.None} {
-			resp, err := http.Get(url)
-			if err != nil {
-				outFile.Close()
-				return stats, err
-			}
-			defer resp.Body.Close()
-			if resp.StatusCode != http.StatusOK {
-				outFile.Close()
-				return stats, fmt.Errorf("failed to download update from %s, status code: %d", url, resp.StatusCode)
-			}
-
-			_, err = io.Copy(outFile, resp.Body)
-			if err != nil {
-				outFile.Close()
-				return stats, err
-			}
-
-			break
-		}
-
-		outFile.Close()
-		stats.updated += 1
-	}
-
-	if config.RemoveUnknownPatches {
-		patches := make([]string, 0)
-		patchreg := regexp.MustCompile(`patch-[A-Za-z].MPQ`)
-
-		for _, file := range manifest.Files {
-			if patchreg.MatchString(file.Path) {
-				patches = append(patches, strings.Split(file.Path, "Data\\")[1])
-			}
-		}
-
-		err = filepath.WalkDir(filepath.Join(config.WowDir, "Data"), func(path string, d fs.DirEntry, err error) error {
-			if !d.IsDir() && patchreg.MatchString(d.Name()) {
-				del := true
-				for _, patch := range patches {
-					if patch == d.Name() {
-						del = false
-						break
-					}
-				}
-				if del {
-					err = os.Remove(path)
-					if err != nil {
-						return err
-					}
-					fmt.Println("Removed unknown patch", d.Name())
-				}
-			}
-			return nil
-		})
-
-		if err != nil {
-			log.Fatalf("failed to delete unknown patches: %s", err)
-		}
-	}
-
-	return stats, nil
-}

pkg/epoch/manifest.go

@@ -1,4 +1,4 @@
-package main
+package epoch
 
 import (
 	"encoding/json"
@@ -6,6 +6,10 @@ import (
 	"net/http"
 )
 
+const (
+	manifestUrl = "https://updater.project-epoch.net/api/v2/manifest"
+)
+
 type File struct {
 	Path   string `json:"Path"`
 	Hash   string `json:"Hash"`
@@ -25,7 +29,7 @@ type Manifest struct {
 	CheckedAt string `json:"checked_at"`
 }
 
-func getManifest() (*Manifest, error) {
+func GetManifest() (*Manifest, error) {
 	resp, err := http.Get(manifestUrl)
 	if err != nil {
 		return nil, err

pkg/epoch/update.go

@@ -0,0 +1,127 @@
+package epoch
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"fmt"
+	"io"
+	"io/fs"
+	"log"
+	"net/http"
+	"os"
+	"path/filepath"
+	"regexp"
+	"strings"
+)
+
+type DownloadStats struct {
+	Updated int
+	Current int
+}
+
+func DownloadUpdate(wowdir string, force bool, removeUnknown bool) (DownloadStats, error) {
+	var stats DownloadStats
+
+	manifest, err := GetManifest()
+	if err != nil {
+		log.Fatalf("Failed to get manifest: %v\n", err)
+	}
+
+	for _, file := range manifest.Files {
+		path := strings.ReplaceAll(file.Path, `\`, `/`)
+		path = strings.TrimLeft(path, `\`)
+
+		localPath := filepath.Join(wowdir, path)
+		localDir := filepath.Dir(localPath)
+		if _, err = os.Stat(localDir); os.IsNotExist(err) {
+			err = os.MkdirAll(localDir, 0755)
+			if err != nil {
+				return stats, fmt.Errorf("failed to create directory %s: %v", localDir, err)
+			}
+		}
+
+		if !force {
+			if _, err = os.Stat(localPath); err == nil {
+				data, err := os.ReadFile(localPath)
+				if err != nil {
+					return stats, err
+				}
+				hashBytes := md5.Sum(data)
+				hash := hex.EncodeToString(hashBytes[:])
+				if hash == file.Hash {
+					fmt.Printf("File %s is up to date\n", localPath)
+					stats.Current += 1
+					continue
+				}
+			}
+		}
+
+		fmt.Printf("Updating %s...\n", localPath)
+
+		outFile, err := os.Create(localPath)
+		if err != nil {
+			return stats, err
+		}
+
+		for _, url := range []string{file.Urls.Cloudflare, file.Urls.Digitalocean, file.Urls.None} {
+			resp, err := http.Get(url)
+			if err != nil {
+				outFile.Close()
+				return stats, err
+			}
+			defer resp.Body.Close()
+			if resp.StatusCode != http.StatusOK {
+				outFile.Close()
+				return stats, fmt.Errorf("failed to download update from %s, status code: %d", url, resp.StatusCode)
+			}
+
+			_, err = io.Copy(outFile, resp.Body)
+			if err != nil {
+				outFile.Close()
+				return stats, err
+			}
+
+			break
+		}
+
+		outFile.Close()
+		stats.Updated += 1
+	}
+
+	if removeUnknown {
+		patches := make([]string, 0)
+		patchreg := regexp.MustCompile(`patch-[A-Za-z].MPQ`)
+
+		for _, file := range manifest.Files {
+			if patchreg.MatchString(file.Path) {
+				patches = append(patches, strings.Split(file.Path, "Data\\")[1])
+			}
+		}
+
+		err = filepath.WalkDir(filepath.Join(wowdir, "Data"), func(path string, d fs.DirEntry, err error) error {
+			if !d.IsDir() && patchreg.MatchString(d.Name()) {
+				del := true
+				for _, patch := range patches {
+					if patch == d.Name() {
+						del = false
+						break
+					}
+				}
+				if del {
+					err = os.Remove(path)
+					if err != nil {
+						return err
+					}
+					fmt.Println("Removed unknown patch", d.Name())
+				}
+			}
+			return nil
+		})
+
+		if err != nil {
+			log.Fatalf("failed to delete unknown patches: %s", err)
+		}
+	}
+
+	return stats, nil
+}