stash/internal/manager/repository.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
	GetManyFileIDs(ctx context.Context, ids []int) ([][]file.ID, error)
}

type GalleryReaderWriter interface {
	models.GalleryReaderWriter
	gallery.FinderCreatorUpdater
	gallery.Finder
	models.FileLoader
	GetManyFileIDs(ctx context.Context, ids []int) ([][]file.ID, error)
}

type SceneReaderWriter interface {
	models.SceneReaderWriter
	scene.CreatorUpdater
	GetManyFileIDs(ctx context.Context, ids []int) ([][]file.ID, error)
}

type FileReaderWriter interface {
	file.Store
	Query(ctx context.Context, options models.FileQueryOptions) (*models.FileQueryResult, error)
	GetCaptions(ctx context.Context, fileID file.ID) ([]*models.VideoCaption, error)
	IsPrimary(ctx context.Context, fileID file.ID) (bool, error)
}

type FolderReaderWriter interface {
	file.FolderStore
}

type Repository struct {
	models.TxnManager

	File           FileReaderWriter
	Folder         FolderReaderWriter
	Gallery        GalleryReaderWriter
	GalleryChapter models.GalleryChapterReaderWriter
	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 (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,
		ScrapedItem:    txnRepo.ScrapedItem,
		Studio:         txnRepo.Studio,
		Tag:            txnRepo.Tag,
		SavedFilter:    txnRepo.SavedFilter,
	}
}

type SceneService interface {
	Create(ctx context.Context, input *models.Scene, fileIDs []file.ID, coverImage []byte) (*models.Scene, error)
	AssignFile(ctx context.Context, sceneID int, fileID file.ID) 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 file.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
}