mirror of
https://github.com/stashapp/stash.git
synced 2025-12-07 08:54:10 +01: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
106 lines
1.9 KiB
Go
106 lines
1.9 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/stashapp/stash/pkg/file"
|
|
)
|
|
|
|
// Image stores the metadata for a single image.
|
|
type Image struct {
|
|
ID int `json:"id"`
|
|
|
|
Title string `json:"title"`
|
|
Rating *int `json:"rating"`
|
|
Organized bool `json:"organized"`
|
|
OCounter int `json:"o_counter"`
|
|
StudioID *int `json:"studio_id"`
|
|
|
|
// transient - not persisted
|
|
Files []*file.ImageFile
|
|
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
|
|
GalleryIDs []int `json:"gallery_ids"`
|
|
TagIDs []int `json:"tag_ids"`
|
|
PerformerIDs []int `json:"performer_ids"`
|
|
}
|
|
|
|
func (i Image) PrimaryFile() *file.ImageFile {
|
|
if len(i.Files) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return i.Files[0]
|
|
}
|
|
|
|
func (i Image) Path() string {
|
|
if p := i.PrimaryFile(); p != nil {
|
|
return p.Path
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func (i Image) Checksum() string {
|
|
if p := i.PrimaryFile(); p != nil {
|
|
v := p.Fingerprints.Get(file.FingerprintTypeMD5)
|
|
if v == nil {
|
|
return ""
|
|
}
|
|
|
|
return v.(string)
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// GetTitle returns the title of the image. If the Title field is empty,
|
|
// then the base filename is returned.
|
|
func (i Image) GetTitle() string {
|
|
if i.Title != "" {
|
|
return i.Title
|
|
}
|
|
|
|
if p := i.PrimaryFile(); p != nil {
|
|
return p.Basename
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
type ImageCreateInput struct {
|
|
*Image
|
|
FileIDs []file.ID
|
|
}
|
|
|
|
type ImagePartial struct {
|
|
Title OptionalString
|
|
Rating OptionalInt
|
|
Organized OptionalBool
|
|
OCounter OptionalInt
|
|
StudioID OptionalInt
|
|
CreatedAt OptionalTime
|
|
UpdatedAt OptionalTime
|
|
|
|
GalleryIDs *UpdateIDs
|
|
TagIDs *UpdateIDs
|
|
PerformerIDs *UpdateIDs
|
|
}
|
|
|
|
func NewImagePartial() ImagePartial {
|
|
updatedTime := time.Now()
|
|
return ImagePartial{
|
|
UpdatedAt: NewOptionalTime(updatedTime),
|
|
}
|
|
}
|
|
|
|
type Images []*Image
|
|
|
|
func (i *Images) Append(o interface{}) {
|
|
*i = append(*i, o.(*Image))
|
|
}
|
|
|
|
func (i *Images) New() interface{} {
|
|
return &Image{}
|
|
}
|