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
	}
}