mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 04:44:37 +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
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package image
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/models/json"
|
|
"github.com/stashapp/stash/pkg/models/jsonschema"
|
|
"github.com/stashapp/stash/pkg/studio"
|
|
)
|
|
|
|
// ToBasicJSON converts a image object into its JSON object equivalent. It
|
|
// does not convert the relationships to other objects, with the exception
|
|
// of cover image.
|
|
func ToBasicJSON(image *models.Image) *jsonschema.Image {
|
|
newImageJSON := jsonschema.Image{
|
|
Checksum: image.Checksum(),
|
|
Title: image.Title,
|
|
CreatedAt: json.JSONTime{Time: image.CreatedAt},
|
|
UpdatedAt: json.JSONTime{Time: image.UpdatedAt},
|
|
}
|
|
|
|
if image.Rating != nil {
|
|
newImageJSON.Rating = *image.Rating
|
|
}
|
|
|
|
newImageJSON.Organized = image.Organized
|
|
newImageJSON.OCounter = image.OCounter
|
|
|
|
newImageJSON.File = getImageFileJSON(image)
|
|
|
|
return &newImageJSON
|
|
}
|
|
|
|
func getImageFileJSON(image *models.Image) *jsonschema.ImageFile {
|
|
ret := &jsonschema.ImageFile{}
|
|
|
|
f := image.PrimaryFile()
|
|
|
|
ret.ModTime = json.JSONTime{Time: f.ModTime}
|
|
ret.Size = f.Size
|
|
ret.Width = f.Width
|
|
ret.Height = f.Height
|
|
|
|
return ret
|
|
}
|
|
|
|
// GetStudioName returns the name of the provided image's studio. It returns an
|
|
// empty string if there is no studio assigned to the image.
|
|
func GetStudioName(ctx context.Context, reader studio.Finder, image *models.Image) (string, error) {
|
|
if image.StudioID != nil {
|
|
studio, err := reader.Find(ctx, *image.StudioID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if studio != nil {
|
|
return studio.Name.String, nil
|
|
}
|
|
}
|
|
|
|
return "", nil
|
|
}
|
|
|
|
// GetGalleryChecksum returns the checksum of the provided image. It returns an
|
|
// empty string if there is no gallery assigned to the image.
|
|
// func GetGalleryChecksum(reader models.GalleryReader, image *models.Image) (string, error) {
|
|
// gallery, err := reader.FindByImageID(image.ID)
|
|
// if err != nil {
|
|
// return "", fmt.Errorf("error getting image gallery: %v", err)
|
|
// }
|
|
|
|
// if gallery != nil {
|
|
// return gallery.Checksum, nil
|
|
// }
|
|
|
|
// return "", nil
|
|
// }
|