mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 08:26:00 +01:00
c364346a59
* Add mockery config file * Move basic file/folder structs to models * Fix hack due to import loop * Move file interfaces to models * Move folder interfaces to models * Move scene interfaces to models * Move scene marker interfaces to models * Move image interfaces to models * Move gallery interfaces to models * Move gallery chapter interfaces to models * Move studio interfaces to models * Move movie interfaces to models * Move performer interfaces to models * Move tag interfaces to models * Move autotag interfaces to models * Regenerate mocks
83 lines
2.8 KiB
Go
83 lines
2.8 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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 Repository struct {
|
|
models.TxnManager
|
|
|
|
File models.FileReaderWriter
|
|
Folder models.FolderReaderWriter
|
|
Gallery models.GalleryReaderWriter
|
|
GalleryChapter models.GalleryChapterReaderWriter
|
|
Image models.ImageReaderWriter
|
|
Movie models.MovieReaderWriter
|
|
Performer models.PerformerReaderWriter
|
|
Scene models.SceneReaderWriter
|
|
SceneMarker models.SceneMarkerReaderWriter
|
|
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 (r *Repository) WithReadTxn(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithReadTxn(ctx, r, fn)
|
|
}
|
|
|
|
func (r *Repository) WithDB(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithDatabase(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,
|
|
GalleryChapter: txnRepo.GalleryChapter,
|
|
Image: d.Image,
|
|
Movie: txnRepo.Movie,
|
|
Performer: txnRepo.Performer,
|
|
Scene: d.Scene,
|
|
SceneMarker: txnRepo.SceneMarker,
|
|
Studio: txnRepo.Studio,
|
|
Tag: txnRepo.Tag,
|
|
SavedFilter: txnRepo.SavedFilter,
|
|
}
|
|
}
|
|
|
|
type SceneService interface {
|
|
Create(ctx context.Context, input *models.Scene, fileIDs []models.FileID, coverImage []byte) (*models.Scene, error)
|
|
AssignFile(ctx context.Context, sceneID int, fileID models.FileID) error
|
|
Merge(ctx context.Context, sourceIDs []int, destinationID int, values models.ScenePartial) error
|
|
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
|
|
DestroyZipImages(ctx context.Context, zipFile models.File, fileDeleter *image.FileDeleter, deleteGenerated bool) ([]*models.Image, error)
|
|
}
|
|
|
|
type GalleryService interface {
|
|
AddImages(ctx context.Context, g *models.Gallery, toAdd ...int) error
|
|
RemoveImages(ctx context.Context, g *models.Gallery, toRemove ...int) error
|
|
|
|
Destroy(ctx context.Context, i *models.Gallery, fileDeleter *image.FileDeleter, deleteGenerated, deleteFile bool) ([]*models.Image, error)
|
|
|
|
ValidateImageGalleryChange(ctx context.Context, i *models.Image, updateIDs models.UpdateIDs) error
|
|
|
|
Updated(ctx context.Context, galleryID int) error
|
|
}
|