89 lines
1.4 KiB
Go
89 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
const touchedFile = ".touched"
|
|
|
|
var touched = loadTouched()
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
var imageExts = map[string]bool{
|
|
".jpeg": true,
|
|
".jpg": true,
|
|
".png": true,
|
|
".gif": true,
|
|
".bmp": true,
|
|
".tiff": true,
|
|
".tif": true,
|
|
".webp": true,
|
|
}
|
|
|
|
func main() {
|
|
err := filepath.WalkDir(".", func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if d.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
ext := strings.ToLower(filepath.Ext(path))
|
|
switch {
|
|
case ext == ".mov":
|
|
wg.Go(func() {
|
|
ConvertVideo(path)
|
|
})
|
|
case imageExts[ext]:
|
|
if touched[path] {
|
|
fmt.Printf("Skipping %s (already touched)\n", path)
|
|
return nil
|
|
}
|
|
wg.Go(func() { OptimizeImage(path) })
|
|
}
|
|
return nil
|
|
})
|
|
|
|
wg.Wait()
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func loadTouched() map[string]bool {
|
|
set := map[string]bool{}
|
|
data, err := os.ReadFile(touchedFile)
|
|
if err != nil {
|
|
return set
|
|
}
|
|
for _, line := range strings.Split(string(data), "\n") {
|
|
line = strings.TrimSpace(line)
|
|
if line != "" && !set[line] {
|
|
set[line] = true
|
|
}
|
|
}
|
|
return set
|
|
}
|
|
|
|
func markTouched(path string) {
|
|
f, err := os.OpenFile(touchedFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
fmt.Printf("Error updating %s: %v\n", touchedFile, err)
|
|
return
|
|
}
|
|
defer f.Close()
|
|
|
|
if _, err := fmt.Fprintln(f, path); err == nil {
|
|
touched[path] = true
|
|
}
|
|
}
|