mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 12:54:38 +03:00
File storage rewrite (#2676)
* Restructure data layer part 2 (#2599) * Refactor and separate image model * Refactor image query builder * Handle relationships in image query builder * Remove relationship management methods * Refactor gallery model/query builder * Add scenes to gallery model * Convert scene model * Refactor scene models * Remove unused methods * Add unit tests for gallery * Add image tests * Add scene tests * Convert unnecessary scene value pointers to values * Convert unnecessary pointer values to values * Refactor scene partial * Add scene partial tests * Refactor ImagePartial * Add image partial tests * Refactor gallery partial update * Add partial gallery update tests * Use zero/null package for null values * Add files and scan system * Add sqlite implementation for files/folders * Add unit tests for files/folders * Image refactors * Update image data layer * Refactor gallery model and creation * Refactor scene model * Refactor scenes * Don't set title from filename * Allow galleries to freely add/remove images * Add multiple scene file support to graphql and UI * Add multiple file support for images in graphql/UI * Add multiple file for galleries in graphql/UI * Remove use of some deprecated fields * Remove scene path usage * Remove gallery path usage * Remove path from image * Move funscript to video file * Refactor caption detection * Migrate existing data * Add post commit/rollback hook system * Lint. Comment out import/export tests * Add WithDatabase read only wrapper * Prepend tasks to list * Add 32 pre-migration * Add warnings in release and migration notes
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
package file
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/fs"
|
||||
"os"
|
||||
|
||||
"github.com/stashapp/stash/pkg/logger"
|
||||
"github.com/stashapp/stash/pkg/txn"
|
||||
)
|
||||
|
||||
const deleteFileSuffix = ".delete"
|
||||
@@ -66,6 +68,19 @@ func NewDeleter() *Deleter {
|
||||
}
|
||||
}
|
||||
|
||||
// RegisterHooks registers post-commit and post-rollback hooks.
|
||||
func (d *Deleter) RegisterHooks(ctx context.Context, mgr txn.Manager) {
|
||||
mgr.AddPostCommitHook(ctx, func(ctx context.Context) error {
|
||||
d.Commit()
|
||||
return nil
|
||||
})
|
||||
|
||||
mgr.AddPostRollbackHook(ctx, func(ctx context.Context) error {
|
||||
d.Rollback()
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
// Files designates files to be deleted. Each file marked will be renamed to add
|
||||
// a `.delete` suffix. An error is returned if a file could not be renamed.
|
||||
// Note that if an error is returned, then some files may be left renamed.
|
||||
@@ -159,3 +174,17 @@ func (d *Deleter) renameForDelete(path string) error {
|
||||
func (d *Deleter) renameForRestore(path string) error {
|
||||
return d.RenamerRemover.Rename(path+deleteFileSuffix, path)
|
||||
}
|
||||
|
||||
func Destroy(ctx context.Context, destroyer Destroyer, f File, fileDeleter *Deleter, deleteFile bool) error {
|
||||
if err := destroyer.Destroy(ctx, f.Base().ID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if deleteFile {
|
||||
if err := fileDeleter.Files([]string{f.Base().Path}); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user