mirror of
https://github.com/stashapp/stash.git
synced 2025-12-15 12:52:38 +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
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package scene
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"image"
|
|
"image/jpeg"
|
|
"os"
|
|
|
|
"github.com/stashapp/stash/pkg/file"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/models/paths"
|
|
|
|
"github.com/disintegration/imaging"
|
|
|
|
// needed to decode other image formats
|
|
_ "image/gif"
|
|
_ "image/png"
|
|
)
|
|
|
|
type CoverGenerator interface {
|
|
GenerateCover(ctx context.Context, scene *models.Scene, f *file.VideoFile) error
|
|
}
|
|
|
|
type ScreenshotSetter interface {
|
|
SetScreenshot(scene *models.Scene, imageData []byte) error
|
|
}
|
|
|
|
type PathsCoverSetter struct {
|
|
Paths *paths.Paths
|
|
FileNamingAlgorithm models.HashAlgorithm
|
|
}
|
|
|
|
func (ss *PathsCoverSetter) SetScreenshot(scene *models.Scene, imageData []byte) error {
|
|
checksum := scene.GetHash(ss.FileNamingAlgorithm)
|
|
return SetScreenshot(ss.Paths, checksum, imageData)
|
|
}
|
|
|
|
func writeImage(path string, imageData []byte) error {
|
|
f, err := os.Create(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
_, err = f.Write(imageData)
|
|
return err
|
|
}
|
|
|
|
func writeThumbnail(path string, thumbnail image.Image) error {
|
|
f, err := os.Create(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
return jpeg.Encode(f, thumbnail, nil)
|
|
}
|
|
|
|
func SetScreenshot(paths *paths.Paths, checksum string, imageData []byte) error {
|
|
thumbPath := paths.Scene.GetThumbnailScreenshotPath(checksum)
|
|
normalPath := paths.Scene.GetScreenshotPath(checksum)
|
|
|
|
img, _, err := image.Decode(bytes.NewReader(imageData))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// resize to 320 width maintaining aspect ratio, for the thumbnail
|
|
const width = 320
|
|
origWidth := img.Bounds().Max.X
|
|
origHeight := img.Bounds().Max.Y
|
|
height := width / origWidth * origHeight
|
|
|
|
thumbnail := imaging.Resize(img, width, height, imaging.Lanczos)
|
|
err = writeThumbnail(thumbPath, thumbnail)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = writeImage(normalPath, imageData)
|
|
|
|
return err
|
|
}
|