improved file read/copy logic

out.go

@@ -1,74 +1,183 @@
 package main
 
 import (
+	_ "embed"
 	"fmt"
+	"image"
+	"image/color"
+	"image/png"
+	"log"
 	"os"
 	"path"
 	"path/filepath"
+	"slices"
 	"strings"
 )
 
+//go:embed silence.wav
+var SILENCE_WAV []byte
+
 const FILEPERM = 0644
-const INIPREFIX = `[SETTINGS]
-Sensitivity=6
-Splash1len=2
-Splash2len=2
-`
 
-func CopyOut(roomFolders []RoomFolder, outpath string) error {
-	inistr := INIPREFIX
-	
-	for _, roomFolder := range roomFolders {
-		err := copyFolder(roomFolder, outpath)
-		if err != nil {
-			return fmt.Errorf("error copying folder %s: %+v", roomFolder.path, err)
-		}
-		inistr += roomFolder.cfg.ToIni()
-	}
-
-	outini := filepath.Join(outpath, "CF.ini")
-	fmt.Printf("Writing %s\n", outini)
-	return os.WriteFile(outini, []byte(inistr), FILEPERM)
-}
-
-func copyFolder(roomFolder RoomFolder, outpath string) error {
-	nightdir := fmt.Sprintf("NIGHT_%d", roomFolder.cfg.night)
-	roomdir := fmt.Sprintf("ROOM_%d", roomFolder.cfg.room)
-
-	roomoutpath := filepath.Join(outpath, "DATA", nightdir, roomdir)
-
-	os.MkdirAll(roomoutpath, FILEPERM)
-		fmt.Printf(`Copying:
-    %s
- to %s
-`, roomFolder.path, roomoutpath)
-
-	entries, err := os.ReadDir(roomFolder.path)
+// only deletes DATA/NIGHT_* folders and the CF.ini file
+// will not touch TITLE or any game files
+// so this can be ran safely in your game's directory
+func EmptyOut(outpath string) error {
+	err := emptyOutData(outpath)
 	if err != nil {
 		return err
 	}
 
+	cfpath := path.Join(outpath, "CF.ini")
+	exists, err := Exists(cfpath)
+	if err != nil {
+		return err
+	}
+	if !exists {
+		return nil
+	}
+
+	log.Printf("INFO: removing %s", cfpath)
+	return os.Remove(cfpath)
+}
+
+func emptyOutData(outpath string) error {
+	datapath := path.Join(outpath, "DATA")
+	exists, err := Exists(datapath)
+	if err != nil {
+		return err
+	}
+	if !exists {
+		return nil
+	}
+
+	entries, err := os.ReadDir(datapath)
+	if err != nil {
+		return err
+	}
+
+	nightprefix := "NIGHT_"
 	for _, entry := range entries {
-		// TODO whitelist files?
-		entrypath := path.Join(roomFolder.path, entry.Name())
-
-		if entry.IsDir() {
-			fmt.Printf("WARN: unexpected folder found in room folder: %s\n", entrypath)
+		if entry.Name()[0] == '.' || !entry.IsDir() {
 			continue
 		}
 
-		if strings.ToLower(entry.Name()) == "room.ini" {
+		name := strings.ToUpper(entry.Name())
+
+		if len(name) >= len(nightprefix) && name[:len(nightprefix)] == nightprefix {
+			nightpath := path.Join(datapath, entry.Name())
+			log.Printf("INFO: removing %s", nightpath)
+			err = os.RemoveAll(nightpath)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func CopyOut(roomFolders []RoomFolder, outpath string, splash1len, splash2len int) error {
+	inistr := fmt.Sprintf(`[SETTINGS]
+Sensitivity=6
+Splash1len=%d
+Splash2len=%d
+`, splash1len, splash2len)
+
+	for _, roomFolder := range roomFolders {
+		err := copyFolder(roomFolder, outpath)
+		if err != nil {
+			return fmt.Errorf("error copying folder %s: %+v", roomFolder.Path, err)
+		}
+		inistr += roomFolder.Cfg.ToIni()
+	}
+
+	outini := filepath.Join(outpath, "CF.ini")
+	log.Println("INFO:")
+	log.Printf("INFO: Writing %s", outini)
+	return os.WriteFile(outini, []byte(inistr), FILEPERM)
+}
+
+func copyFolder(roomFolder RoomFolder, outpath string) error {
+	nightdir := fmt.Sprintf("NIGHT_%d", roomFolder.Cfg.night)
+	roomdir := fmt.Sprintf("ROOM_%d", roomFolder.Cfg.room)
+
+	roomoutpath := filepath.Join(outpath, "DATA", nightdir, roomdir)
+
+	os.MkdirAll(roomoutpath, FILEPERM)
+
+	log.Println("INFO:")
+	log.Printf("INFO: Copying %s", roomFolder.Path)
+	log.Printf("INFO:      to %s", roomoutpath)
+	if roomFolder.IsDefaultCfg {
+		log.Println("INFO: room.ini was not found, using default settings")
+	}
+
+	entries, err := os.ReadDir(roomFolder.Path)
+	if err != nil {
+		return err
+	}
+
+	extraRequiredImages := gameExtraRequiredImages()
+	extraRequiredAudio := gameExtraRequiredAudios()
+
+	for _, entry := range entries {
+		if entry.Name()[0] == '.' || entry.IsDir() {
 			continue
 		}
 
+		name, ext := SplitExt(entry.Name())
+
+		if !isAcceptedFile(name, ext) {
+			continue
+		}
+
+		if index := slices.Index(extraRequiredImages, name); index > -1 && isAcceptedImageExt(ext) {
+			extraRequiredImages = append(extraRequiredImages[:index], extraRequiredImages[index+1:]...)
+		}
+
+		if index := slices.Index(extraRequiredAudio, name); index > -1 && isAcceptedAudioExt(ext) {
+			extraRequiredAudio = append(extraRequiredAudio[:index], extraRequiredAudio[index+1:]...)
+		}
+
+		frompath := path.Join(roomFolder.Path, entry.Name())
 		topath := path.Join(roomoutpath, entry.Name())
-
-		err := copyFile(entrypath, topath)
+		err := copyFile(frompath, topath)
 		if err != nil {
 			return err
 		}
 	}
 
+	if len(extraRequiredImages) > 0 {
+		log.Printf("INFO: necessary optional images will use a transparent file: %v", extraRequiredImages)
+
+		transparent := image.NewRGBA(image.Rectangle{image.Point{0, 0}, image.Point{1, 1}})
+		transparent.Set(0, 0, color.RGBA{0xff, 0xff, 0xff, 0x01})
+
+		for _, imagename := range extraRequiredImages {
+			imagepath := path.Join(roomoutpath, imagename+".PNG")
+
+			f, err := os.Create(imagepath)
+			if err != nil {
+				return err
+			}
+			png.Encode(f, transparent)
+		}
+	}
+
+	if len(extraRequiredAudio) > 0 {
+		log.Printf("INFO: necessary optional audios will use a silent file: %v", extraRequiredAudio)
+
+		for _, audioname := range extraRequiredAudio {
+			audiopath := path.Join(roomoutpath, audioname+".WAV")
+
+			err := os.WriteFile(audiopath, SILENCE_WAV, FILEPERM)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
 	return nil
 }