From da151015cd42af048e8e5401a64452a50a7c56c7 Mon Sep 17 00:00:00 2001
From: Piotr Grabowski <fau999@gmail.com>
Date: Sat, 3 Sep 2022 21:10:52 +0200
Subject: [PATCH] improve(archive): for dir loader, look for supported images
 recursively in subdirectories instead of just in root

---
 archive/dir.go | 71 ++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 63 insertions(+), 8 deletions(-)

diff --git a/archive/dir.go b/archive/dir.go
index 43c49b7..e01587f 100644
--- a/archive/dir.go
+++ b/archive/dir.go
@@ -20,6 +20,8 @@ package archive
 
 import (
 	"errors"
+	"fmt"
+	"log"
 	"os"
 	"path/filepath"
 	"sort"
@@ -36,19 +38,26 @@ type Dir struct {
 
 /* Reads filenames from a directory, and sorts them */
 func NewDir(path string) (*Dir, error) {
-	var err error
+	subPath, err := findFirstDirContainingSupportedImage(path)
+	if err != nil {
+		return nil, fmt.Errorf("searching for supported images in provided path: %v", err)
+	}
+	if subPath == nil {
+		return nil, errors.New("Could not find a directory containing supported images")
+	}
 
-	ar := new(Dir)
-
-	ar.name = filepath.Base(path)
-	ar.path = path
-
-	dir, err := os.Open(path)
+	dir, err := os.Open(*subPath)
 	if err != nil {
 		return nil, err
 	}
 	defer dir.Close()
 
+	ar := Dir{
+		filenames: nil,
+		name:      filepath.Base(*subPath),
+		path:      *subPath,
+	}
+
 	filenames, err := dir.Readdirnames(-1)
 	if err != nil {
 		return nil, err
@@ -68,7 +77,7 @@ func NewDir(path string) (*Dir, error) {
 
 	sort.Sort(ar.filenames)
 
-	return ar, nil
+	return &ar, nil
 }
 
 func (ar *Dir) checkbounds(i int) error {
@@ -117,3 +126,49 @@ func (ar *Dir) Len() *int {
 func (ar *Dir) Close() error {
 	return nil
 }
+
+func findFirstDirContainingSupportedImage(path string) (*string, error) {
+	dir, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer dir.Close()
+
+	entries, err := dir.ReadDir(-1)
+	if err != nil {
+		return nil, err
+	}
+
+	var subdirs []string
+	for _, entry := range entries {
+		if entry.IsDir() {
+			if subdirs == nil {
+				subdirs = make([]string, 0, len(entries))
+			}
+			subdirs = append(subdirs, entry.Name())
+		} else if extensionMatches(entry.Name(), imageExtensions) {
+			// Found supported image directly in path
+			return &path, nil
+		}
+	}
+
+	if len(subdirs) == 0 {
+		// Did not find supported image directly in path and have no subdirectories to check
+		return nil, nil
+	} else {
+		// Check subdirectories
+		for _, subDir := range subdirs {
+			var subDirPath = filepath.Join(path, subDir)
+			res, err := findFirstDirContainingSupportedImage(subDirPath)
+			if err != nil {
+				log.Printf("Error while searching for supported images in subdirectory: %v", err)
+			} else if res != nil {
+				// Found supported image in subdirectory
+				return res, nil
+			}
+		}
+	}
+
+	// Did not find supported image directly in path or in subdirectories
+	return nil, nil
+}