mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 04:14:39 +03:00
5495d72849
* 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
94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stashapp/stash/pkg/file"
|
|
"github.com/stashapp/stash/pkg/gallery"
|
|
"github.com/stashapp/stash/pkg/image"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/scene"
|
|
"github.com/stashapp/stash/pkg/sqlite"
|
|
"github.com/stashapp/stash/pkg/txn"
|
|
)
|
|
|
|
type ImageReaderWriter interface {
|
|
models.ImageReaderWriter
|
|
image.FinderCreatorUpdater
|
|
}
|
|
|
|
type GalleryReaderWriter interface {
|
|
models.GalleryReaderWriter
|
|
gallery.FinderCreatorUpdater
|
|
}
|
|
|
|
type SceneReaderWriter interface {
|
|
models.SceneReaderWriter
|
|
scene.CreatorUpdater
|
|
}
|
|
|
|
type FileReaderWriter interface {
|
|
file.Store
|
|
file.Finder
|
|
Query(ctx context.Context, options models.FileQueryOptions) (*models.FileQueryResult, error)
|
|
GetCaptions(ctx context.Context, fileID file.ID) ([]*models.VideoCaption, error)
|
|
}
|
|
|
|
type FolderReaderWriter interface {
|
|
file.FolderStore
|
|
Find(ctx context.Context, id file.FolderID) (*file.Folder, error)
|
|
}
|
|
|
|
type Repository struct {
|
|
models.TxnManager
|
|
|
|
File FileReaderWriter
|
|
Folder FolderReaderWriter
|
|
Gallery GalleryReaderWriter
|
|
Image ImageReaderWriter
|
|
Movie models.MovieReaderWriter
|
|
Performer models.PerformerReaderWriter
|
|
Scene SceneReaderWriter
|
|
SceneMarker models.SceneMarkerReaderWriter
|
|
ScrapedItem models.ScrapedItemReaderWriter
|
|
Studio models.StudioReaderWriter
|
|
Tag models.TagReaderWriter
|
|
SavedFilter models.SavedFilterReaderWriter
|
|
}
|
|
|
|
func (r *Repository) WithTxn(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithTxn(ctx, r, fn)
|
|
}
|
|
|
|
func sqliteRepository(d *sqlite.Database) Repository {
|
|
txnRepo := d.TxnRepository()
|
|
|
|
return Repository{
|
|
TxnManager: txnRepo,
|
|
File: d.File,
|
|
Folder: d.Folder,
|
|
Gallery: d.Gallery,
|
|
Image: d.Image,
|
|
Movie: txnRepo.Movie,
|
|
Performer: txnRepo.Performer,
|
|
Scene: d.Scene,
|
|
SceneMarker: txnRepo.SceneMarker,
|
|
ScrapedItem: txnRepo.ScrapedItem,
|
|
Studio: txnRepo.Studio,
|
|
Tag: txnRepo.Tag,
|
|
SavedFilter: txnRepo.SavedFilter,
|
|
}
|
|
}
|
|
|
|
type SceneService interface {
|
|
Destroy(ctx context.Context, scene *models.Scene, fileDeleter *scene.FileDeleter, deleteGenerated, deleteFile bool) error
|
|
}
|
|
|
|
type ImageService interface {
|
|
Destroy(ctx context.Context, image *models.Image, fileDeleter *image.FileDeleter, deleteGenerated, deleteFile bool) error
|
|
}
|
|
|
|
type GalleryService interface {
|
|
Destroy(ctx context.Context, i *models.Gallery, fileDeleter *image.FileDeleter, deleteGenerated, deleteFile bool) ([]*models.Image, error)
|
|
}
|