Restructure go project (#2356)

* Move main to cmd
* Move api to internal
* Move logger and manager to internal
* Move shell hiding code to separate package
* Decouple job from desktop and utils
* Decouple session from config
* Move static into internal
* Decouple config from dlna
* Move desktop to internal
* Move dlna to internal
* Decouple remaining packages from config
* Move config into internal
* Move jsonschema and paths to models
* Make ffmpeg functions private
* Move file utility methods into fsutil package
* Move symwalk into fsutil
* Move single-use util functions into client package
* Move slice functions to separate packages
* Add env var to suppress windowsgui arg
* Move hash functions into separate package
* Move identify to internal
* Move autotag to internal
* Touch UI when generating backend

internal/api/images.go

@@ -0,0 +1,113 @@
+package api
+
+import (
+	"io"
+	"io/fs"
+	"os"
+	"strings"
+
+	"github.com/stashapp/stash/internal/manager/config"
+	"github.com/stashapp/stash/internal/static"
+	"github.com/stashapp/stash/pkg/hash"
+	"github.com/stashapp/stash/pkg/logger"
+)
+
+type imageBox struct {
+	box   fs.FS
+	files []string
+}
+
+var imageExtensions = []string{
+	".jpg",
+	".jpeg",
+	".png",
+	".gif",
+	".svg",
+	".webp",
+}
+
+func newImageBox(box fs.FS) (*imageBox, error) {
+	ret := &imageBox{
+		box: box,
+	}
+
+	err := fs.WalkDir(box, ".", func(path string, d fs.DirEntry, err error) error {
+		if d.IsDir() {
+			return nil
+		}
+
+		baseName := strings.ToLower(d.Name())
+		for _, ext := range imageExtensions {
+			if strings.HasSuffix(baseName, ext) {
+				ret.files = append(ret.files, path)
+				break
+			}
+		}
+
+		return nil
+	})
+
+	return ret, err
+}
+
+var performerBox *imageBox
+var performerBoxMale *imageBox
+var performerBoxCustom *imageBox
+
+func initialiseImages() {
+	var err error
+	performerBox, err = newImageBox(&static.Performer)
+	if err != nil {
+		logger.Warnf("error loading performer images: %v", err)
+	}
+	performerBoxMale, err = newImageBox(&static.PerformerMale)
+	if err != nil {
+		logger.Warnf("error loading male performer images: %v", err)
+	}
+	initialiseCustomImages()
+}
+
+func initialiseCustomImages() {
+	customPath := config.GetInstance().GetCustomPerformerImageLocation()
+	if customPath != "" {
+		logger.Debugf("Loading custom performer images from %s", customPath)
+		// We need to set performerBoxCustom at runtime, as this is a custom path, and store it in a pointer.
+		var err error
+		performerBoxCustom, err = newImageBox(os.DirFS(customPath))
+		if err != nil {
+			logger.Warnf("error loading custom performer from %s: %v", customPath, err)
+		}
+	} else {
+		performerBoxCustom = nil
+	}
+}
+
+func getRandomPerformerImageUsingName(name, gender, customPath string) ([]byte, error) {
+	var box *imageBox
+
+	// If we have a custom path, we should return a new box in the given path.
+	if performerBoxCustom != nil && len(performerBoxCustom.files) > 0 {
+		box = performerBoxCustom
+	}
+
+	if box == nil {
+		switch strings.ToUpper(gender) {
+		case "FEMALE":
+			box = performerBox
+		case "MALE":
+			box = performerBoxMale
+		default:
+			box = performerBox
+		}
+	}
+
+	imageFiles := box.files
+	index := hash.IntFromString(name) % uint64(len(imageFiles))
+	img, err := box.box.Open(imageFiles[index])
+	if err != nil {
+		return nil, err
+	}
+	defer img.Close()
+
+	return io.ReadAll(img)
+}