stash/pkg/models/tag.go

package models

import (
	"github.com/jmoiron/sqlx"
)

type TagReader interface {
	Find(id int) (*Tag, error)
	FindMany(ids []int) ([]*Tag, error)
	FindBySceneID(sceneID int) ([]*Tag, error)
	FindBySceneMarkerID(sceneMarkerID int) ([]*Tag, error)
	FindByName(name string, nocase bool) (*Tag, error)
	FindByNames(names []string, nocase bool) ([]*Tag, error)
	// Count() (int, error)
	All() ([]*Tag, error)
	// AllSlim() ([]*Tag, error)
	// Query(tagFilter *TagFilterType, findFilter *FindFilterType) ([]*Tag, int, error)
	GetTagImage(tagID int) ([]byte, error)
}

type TagWriter interface {
	Create(newTag Tag) (*Tag, error)
	Update(updatedTag Tag) (*Tag, error)
	// Destroy(id string) error
	UpdateTagImage(tagID int, image []byte) error
	// DestroyTagImage(tagID int) error
}

type TagReaderWriter interface {
	TagReader
	TagWriter
}

func NewTagReaderWriter(tx *sqlx.Tx) TagReaderWriter {
	return &tagReaderWriter{
		tx: tx,
		qb: NewTagQueryBuilder(),
	}
}

type tagReaderWriter struct {
	tx *sqlx.Tx
	qb TagQueryBuilder
}

func (t *tagReaderWriter) Find(id int) (*Tag, error) {
	return t.qb.Find(id, t.tx)
}

func (t *tagReaderWriter) FindMany(ids []int) ([]*Tag, error) {
	return t.qb.FindMany(ids)
}

func (t *tagReaderWriter) All() ([]*Tag, error) {
	return t.qb.All()
}

func (t *tagReaderWriter) FindBySceneMarkerID(sceneMarkerID int) ([]*Tag, error) {
	return t.qb.FindBySceneMarkerID(sceneMarkerID, t.tx)
}

func (t *tagReaderWriter) FindByName(name string, nocase bool) (*Tag, error) {
	return t.qb.FindByName(name, t.tx, nocase)
}

func (t *tagReaderWriter) FindByNames(names []string, nocase bool) ([]*Tag, error) {
	return t.qb.FindByNames(names, t.tx, nocase)
}

func (t *tagReaderWriter) GetTagImage(tagID int) ([]byte, error) {
	return t.qb.GetTagImage(tagID, t.tx)
}

func (t *tagReaderWriter) FindBySceneID(sceneID int) ([]*Tag, error) {
	return t.qb.FindBySceneID(sceneID, t.tx)
}

func (t *tagReaderWriter) Create(newTag Tag) (*Tag, error) {
	return t.qb.Create(newTag, t.tx)
}

func (t *tagReaderWriter) Update(updatedTag Tag) (*Tag, error) {
	return t.qb.Update(updatedTag, t.tx)
}

func (t *tagReaderWriter) UpdateTagImage(tagID int, image []byte) error {
	return t.qb.UpdateTagImage(tagID, image, t.tx)
}