From 159207dd5f371a02c8b415739ddbafeef4aaab51 Mon Sep 17 00:00:00 2001
From: Seednode <git@seedno.de>
Date: Sat, 24 Sep 2022 20:28:34 -0500
Subject: [PATCH] Completely reworked file list generation to fix Successive
 weighting

---
 cmd/files.go   | 98 ++++++++++++++++++++++++++++++++++++--------------
 cmd/version.go |  2 +-
 cmd/web.go     |  2 --
 3 files changed, 73 insertions(+), 29 deletions(-)

diff --git a/cmd/files.go b/cmd/files.go
index 7af9085..dc3fcb6 100644
--- a/cmd/files.go
+++ b/cmd/files.go
@@ -22,20 +22,22 @@ var (
 	ErrNoImagesFound = fmt.Errorf("no supported image formats found")
 )
 
-func appendPaths(paths []string, path, filter string) ([]string, error) {
+func appendPaths(m map[string][]string, path, filter string) (map[string][]string, error) {
 	absolutePath, err := filepath.Abs(path)
 	if err != nil {
-		return paths, err
+		return nil, err
 	}
 
+	directory, _ := filepath.Split(absolutePath)
+
 	switch {
 	case filter != "" && strings.Contains(path, filter):
-		paths = append(paths, absolutePath)
+		m[directory] = append(m[directory], path)
 	case filter == "":
-		paths = append(paths, absolutePath)
+		m[directory] = append(m[directory], path)
 	}
 
-	return paths, nil
+	return m, nil
 }
 
 func getFirstFile(path string) (string, error) {
@@ -98,11 +100,13 @@ func getNextFile(path string) (string, error) {
 }
 
 func checkNextFile(path string) (bool, error) {
-	if _, err := os.Stat(path); err == nil {
+	_, err := os.Stat(path)
+	switch {
+	case err == nil:
 		return true, nil
-	} else if errors.Is(err, os.ErrNotExist) {
+	case errors.Is(err, os.ErrNotExist):
 		return false, nil
-	} else {
+	default:
 		return false, err
 	}
 }
@@ -124,9 +128,7 @@ func checkIfImage(path string) (bool, error) {
 	return false, nil
 }
 
-func getFiles(path, filter string) ([]string, error) {
-	var paths []string
-
+func getFiles(m map[string][]string, path, filter string) (map[string][]string, error) {
 	err := filepath.WalkDir(path, func(p string, info os.DirEntry, err error) error {
 		if err != nil {
 			return err
@@ -136,17 +138,17 @@ func getFiles(path, filter string) ([]string, error) {
 		case !Recursive && info.IsDir() && p != path:
 			return filepath.SkipDir
 		case Filter != "" && !info.IsDir():
-			paths, err = appendPaths(paths, p, Filter)
+			m, err = appendPaths(m, p, Filter)
 			if err != nil {
 				return err
 			}
 		case filter != "" && !info.IsDir():
-			paths, err = appendPaths(paths, p, filter)
+			m, err = appendPaths(m, p, filter)
 			if err != nil {
 				return err
 			}
 		default:
-			paths, err = appendPaths(paths, p, "")
+			m, err = appendPaths(m, p, "")
 			if err != nil {
 				return err
 			}
@@ -158,34 +160,78 @@ func getFiles(path, filter string) ([]string, error) {
 		return nil, err
 	}
 
-	return paths, nil
+	return m, nil
 }
 
-func getFileList(paths []string, filter string) ([]string, error) {
-	fileList := []string{}
+func getFileList(paths []string, filter string) (map[string][]string, error) {
+	fileMap := map[string][]string{}
+	var err error
 
 	for i := 0; i < len(paths); i++ {
-
-		f, err := getFiles(paths[i], filter)
+		fileMap, err = getFiles(fileMap, paths[i], filter)
 		if err != nil {
 			return nil, err
 		}
-
-		fileList = append(fileList, f...)
 	}
 
-	return fileList, nil
+	return fileMap, nil
+}
+
+func cleanFilename(filename string) string {
+	filename = filename[:len(filename)-(len(filepath.Ext(filename))+3)]
+
+	return filename
+}
+
+func prepareDirectory(directory []string) []string {
+	_, first := filepath.Split(directory[0])
+	first = cleanFilename(first)
+
+	_, last := filepath.Split(directory[len(directory)-1])
+	last = cleanFilename(last)
+
+	if first == last {
+		d := append([]string{}, directory[0])
+		return d
+	} else {
+		return directory
+	}
+}
+
+func prepareDirectories(m map[string][]string) []string {
+	rand.Seed(time.Now().UnixNano())
+
+	directories := []string{}
+
+	keys := make([]string, len(m))
+
+	i := 0
+	for k := range m {
+		keys[i] = k
+		i++
+	}
+
+	switch {
+	case Successive:
+		for i := 0; i < len(keys); i++ {
+			directories = append(directories, prepareDirectory(m[keys[i]])...)
+		}
+	default:
+		for i := 0; i < len(keys); i++ {
+			directories = append(directories, m[keys[i]]...)
+		}
+	}
+
+	return directories
 }
 
 func pickFile(args []string, filter string) (string, error) {
-	fileList, err := getFileList(args, filter)
+	fileMap, err := getFileList(args, filter)
 	if err != nil {
 		return "", err
 	}
 
-	rand.Seed(time.Now().UnixNano())
-
-	rand.Shuffle(len(fileList), func(i, j int) { fileList[i], fileList[j] = fileList[j], fileList[i] })
+	fileList := prepareDirectories(fileMap)
 
 	for i := 0; i < len(fileList); i++ {
 		filePath := fileList[i]
diff --git a/cmd/version.go b/cmd/version.go
index 71339df..ec45f48 100644
--- a/cmd/version.go
+++ b/cmd/version.go
@@ -10,7 +10,7 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var Version = "0.10.0"
+var Version = "0.11.0"
 
 func init() {
 	rootCmd.AddCommand(versionCmd)
diff --git a/cmd/web.go b/cmd/web.go
index 7231df5..ae0f7b7 100644
--- a/cmd/web.go
+++ b/cmd/web.go
@@ -74,8 +74,6 @@ func serveStaticFile(w http.ResponseWriter, r http.Request, paths []string) erro
 		return err
 	}
 
-	fmt.Println("Prefixed file path is " + prefixedFilePath)
-
 	filePath := strings.TrimPrefix(prefixedFilePath, PREFIX)
 
 	var matchesPrefix = false