Move zip files while moving folders (stashapp#4374)

dogwithakeyboard · Feb 18, 2024 · fc4fcde · fc4fcde
1 parent 647d611
commit fc4fcde
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 41 deletions.
diff --git a/pkg/file/folder.go b/pkg/file/folder.go
@@ -7,6 +7,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/stashapp/stash/pkg/logger"
 	"github.com/stashapp/stash/pkg/models"
 )
 
@@ -46,9 +47,21 @@ func GetOrCreateFolderHierarchy(ctx context.Context, fc models.FolderFinderCreat
 	return folder, nil
 }
 
-// TransferZipFolderHierarchy creates the folder hierarchy for zipFileID under newPath, and removes
+func transferZipHierarchy(ctx context.Context, folderStore models.FolderReaderWriter, files models.FileFinderUpdater, zipFileID models.FileID, oldPath string, newPath string) error {
+	if err := transferZipFolderHierarchy(ctx, folderStore, zipFileID, oldPath, newPath); err != nil {
+		return fmt.Errorf("moving folder hierarchy for file %s: %w", oldPath, err)
+	}
+
+	if err := transferZipFileEntries(ctx, folderStore, files, zipFileID, oldPath, newPath); err != nil {
+		return fmt.Errorf("moving zip file contents for file %s: %w", oldPath, err)
+	}
+
+	return nil
+}
+
+// transferZipFolderHierarchy creates the folder hierarchy for zipFileID under newPath, and removes
 // ZipFileID from folders under oldPath.
-func TransferZipFolderHierarchy(ctx context.Context, folderStore models.FolderReaderWriter, zipFileID models.FileID, oldPath string, newPath string) error {
+func transferZipFolderHierarchy(ctx context.Context, folderStore models.FolderReaderWriter, zipFileID models.FileID, oldPath string, newPath string) error {
 	zipFolders, err := folderStore.FindByZipFileID(ctx, zipFileID)
 	if err != nil {
 		return err
@@ -74,12 +87,14 @@ func TransferZipFolderHierarchy(ctx context.Context, folderStore models.FolderRe
 		}
 
 		// add ZipFileID to new folder
+		logger.Debugf("adding zip file %s to folder %s", zipFileID, newFolder.Path)
 		newFolder.ZipFileID = &zipFileID
 		if err = folderStore.Update(ctx, newFolder); err != nil {
 			return err
 		}
 
 		// remove ZipFileID from old folder
+		logger.Debugf("removing zip file %s from folder %s", zipFileID, oldFolder.Path)
 		oldFolder.ZipFileID = nil
 		if err = folderStore.Update(ctx, oldFolder); err != nil {
 			return err
@@ -88,3 +103,42 @@ func TransferZipFolderHierarchy(ctx context.Context, folderStore models.FolderRe
 
 	return nil
 }
+
+func transferZipFileEntries(ctx context.Context, folders models.FolderFinderCreator, files models.FileFinderUpdater, zipFileID models.FileID, oldPath, newPath string) error {
+	// move contained files if file is a zip file
+	zipFiles, err := files.FindByZipFileID(ctx, zipFileID)
+	if err != nil {
+		return fmt.Errorf("finding contained files in file %s: %w", oldPath, err)
+	}
+	for _, zf := range zipFiles {
+		zfBase := zf.Base()
+		oldZfPath := zfBase.Path
+		oldZfDir := filepath.Dir(oldZfPath)
+
+		// sanity check - ignore files which aren't under oldPath
+		if !strings.HasPrefix(oldZfPath, oldPath) {
+			continue
+		}
+
+		relZfDir, err := filepath.Rel(oldPath, oldZfDir)
+		if err != nil {
+			return fmt.Errorf("moving contained file %s: %w", zfBase.ID, err)
+		}
+		newZfDir := filepath.Join(newPath, relZfDir)
+
+		// folder should have been created by transferZipFolderHierarchy
+		newZfFolder, err := GetOrCreateFolderHierarchy(ctx, folders, newZfDir)
+		if err != nil {
+			return fmt.Errorf("getting or creating folder hierarchy: %w", err)
+		}
+
+		// update file parent folder
+		zfBase.ParentFolderID = newZfFolder.ID
+		logger.Debugf("moving %s to folder %s", zfBase.Path, newZfFolder.Path)
+		if err := files.Update(ctx, zf); err != nil {
+			return fmt.Errorf("updating file %s: %w", oldZfPath, err)
+		}
+	}
+
+	return nil
+}
diff --git a/pkg/file/move.go b/pkg/file/move.go
@@ -7,7 +7,6 @@ import (
 	"io/fs"
 	"os"
 	"path/filepath"
-	"strings"
 	"time"
 
 	"github.com/stashapp/stash/pkg/logger"
@@ -88,44 +87,10 @@ func (m *Mover) Move(ctx context.Context, f models.File, folder *models.Folder,
 		return fmt.Errorf("file %s already exists", newPath)
 	}
 
-	if err := TransferZipFolderHierarchy(ctx, m.Folders, fBase.ID, oldPath, newPath); err != nil {
+	if err := transferZipHierarchy(ctx, m.Folders, m.Files, fBase.ID, oldPath, newPath); err != nil {
 		return fmt.Errorf("moving folder hierarchy for file %s: %w", fBase.Path, err)
 	}
 
-	// move contained files if file is a zip file
-	zipFiles, err := m.Files.FindByZipFileID(ctx, fBase.ID)
-	if err != nil {
-		return fmt.Errorf("finding contained files in file %s: %w", fBase.Path, err)
-	}
-	for _, zf := range zipFiles {
-		zfBase := zf.Base()
-		oldZfPath := zfBase.Path
-		oldZfDir := filepath.Dir(oldZfPath)
-
-		// sanity check - ignore files which aren't under oldPath
-		if !strings.HasPrefix(oldZfPath, oldPath) {
-			continue
-		}
-
-		relZfDir, err := filepath.Rel(oldPath, oldZfDir)
-		if err != nil {
-			return fmt.Errorf("moving contained file %s: %w", zfBase.ID, err)
-		}
-		newZfDir := filepath.Join(newPath, relZfDir)
-
-		// folder should have been created by moveZipFolderHierarchy
-		newZfFolder, err := GetOrCreateFolderHierarchy(ctx, m.Folders, newZfDir)
-		if err != nil {
-			return fmt.Errorf("getting or creating folder hierarchy: %w", err)
-		}
-
-		// update file parent folder
-		zfBase.ParentFolderID = newZfFolder.ID
-		if err := m.Files.Update(ctx, zf); err != nil {
-			return fmt.Errorf("updating file %s: %w", oldZfPath, err)
-		}
-	}
-
 	fBase.ParentFolderID = folder.ID
 	fBase.Basename = basename
 	fBase.UpdatedAt = time.Now()

diff --git a/pkg/file/scan.go b/pkg/file/scan.go
@@ -653,6 +653,7 @@ func (s *scanJob) handleFile(ctx context.Context, f scanFile) error {
 		}
 
 		if ff == nil {
+			// returns a file only if it is actually new
 			ff, err = s.onNewFile(ctx, f)
 			return err
 		}
@@ -740,7 +741,10 @@ func (s *scanJob) onNewFile(ctx context.Context, f scanFile) (models.File, error
 	}
 
 	if renamed != nil {
-		return renamed, nil
+		// handle rename should have already handled the contents of the zip file
+		// so shouldn't need to scan it again
+		// return nil so it doesn't
+		return nil, nil
 	}
 
 	// if not renamed, queue file for creation
@@ -901,8 +905,8 @@ func (s *scanJob) handleRename(ctx context.Context, f models.File, fp []models.F
 		}
 
 		if s.isZipFile(fBase.Basename) {
-			if err := TransferZipFolderHierarchy(ctx, s.Repository.Folder, fBase.ID, otherBase.Path, fBase.Path); err != nil {
-				return fmt.Errorf("moving folder hierarchy for renamed zip file %q: %w", fBase.Path, err)
+			if err := transferZipHierarchy(ctx, s.Repository.Folder, s.Repository.File, fBase.ID, otherBase.Path, fBase.Path); err != nil {
+				return fmt.Errorf("moving zip hierarchy for renamed zip file %q: %w", fBase.Path, err)
 			}
 		}