162 lines
2.8 KiB
Go
162 lines
2.8 KiB
Go
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
|
|
}
|