init

hdiff-any-game/utils.go

@@ -0,0 +1,161 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"sync"
+
+	"github.com/schollz/progressbar/v3"
+)
+
+func collectFiles(root string) (map[string]string, error) {
+	files := sync.Map{}
+	var wg sync.WaitGroup
+	dirs := make(chan string, 100)
+
+	workers := runtime.NumCPU() * 2
+	if workers < 1 {
+		workers = 1
+	}
+
+	for i := 0; i < workers; i++ {
+		go func() {
+			for dir := range dirs {
+				entries, err := os.ReadDir(dir)
+				if err != nil {
+					wg.Done()
+					continue
+				}
+				for _, e := range entries {
+					path := filepath.Join(dir, e.Name())
+					if e.IsDir() {
+						wg.Add(1)
+						dirs <- path
+					} else {
+						rel, _ := filepath.Rel(root, path)
+						files.Store(filepath.ToSlash(rel), path)
+					}
+				}
+				wg.Done()
+			}
+		}()
+	}
+	wg.Add(1)
+	dirs <- root
+
+	go func() {
+		wg.Wait()
+		close(dirs)
+	}()
+
+	wg.Wait()
+	out := make(map[string]string)
+	files.Range(func(k, v any) bool {
+		out[k.(string)] = v.(string)
+		return true
+	})
+
+	return out, nil
+}
+
+func copyFile(src, dst string) error {
+	in, err := os.Open(src)
+	if err != nil {
+		return err
+	}
+	defer in.Close()
+
+	out, err := os.Create(dst)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	_, err = io.Copy(out, in)
+	return err
+}
+
+func ZipWith7za(src, dest string) error {
+	if _, err := os.Stat(src); os.IsNotExist(err) {
+		return fmt.Errorf("source folder does not exist: %s", src)
+	}
+
+	if err := os.MkdirAll(filepath.Dir(dest), 0755); err != nil {
+		return err
+	}
+
+	files, err := os.ReadDir(src)
+	if err != nil {
+		return err
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("source folder is empty: %s", src)
+	}
+
+	sevenZipPath, err := filepath.Abs(filepath.Join("bin", "7za.exe"))
+	if err != nil {
+		return err
+	}
+
+	destAbs, err := filepath.Abs(filepath.Join(".", dest))
+	if err != nil {
+		return err
+	}
+
+	args := []string{"a", "-tzip", "-mx=1", "-mmt=on", destAbs}
+	for _, f := range files {
+		args = append(args, f.Name())
+	}
+
+	cmd := exec.Command(sevenZipPath, args...)
+	cmd.Dir = src
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return cmd.Run()
+}
+
+func RemoveFolderWithProgress(folder string) error {
+	var total int
+	filepath.Walk(folder, func(path string, info os.FileInfo, err error) error {
+		if err == nil {
+			total++
+		}
+		return nil
+	})
+
+	bar := progressbar.NewOptions(total,
+		progressbar.OptionSetDescription("Removing temp files"),
+		progressbar.OptionShowCount(),
+		progressbar.OptionSetWidth(30),
+		progressbar.OptionSetPredictTime(true),
+	)
+
+	err := filepath.Walk(folder, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			if err := os.Remove(path); err != nil {
+				return err
+			}
+			bar.Add(1)
+		}
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+	if err := os.RemoveAll(folder); err != nil {
+		return err
+	}
+
+	bar.Finish()
+	fmt.Println("\nTemp folder removed")
+	return nil
+}