mirror of
https://github.com/stashapp/stash.git
synced 2025-12-09 09:53:40 +01:00
4a0c4c4847
Rather than passing a pointer to a waitgroup into task.Start(..) functions, handle the waitgroup.Done() at the callsite. This makes waitgroup handling local to its definition rather than it being spread out over multiple files. Tasks now simply execute, and the policy of waiting on them is handled by the caller.
89 lines
2.6 KiB
Go
89 lines
2.6 KiB
Go
package manager
|
|
|
|
import (
|
|
"github.com/stashapp/stash/pkg/ffmpeg"
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
"github.com/stashapp/stash/pkg/manager/config"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/utils"
|
|
)
|
|
|
|
type GeneratePreviewTask struct {
|
|
Scene models.Scene
|
|
ImagePreview bool
|
|
|
|
Options models.GeneratePreviewOptionsInput
|
|
|
|
Overwrite bool
|
|
fileNamingAlgorithm models.HashAlgorithm
|
|
}
|
|
|
|
func (t *GeneratePreviewTask) Start() {
|
|
videoFilename := t.videoFilename()
|
|
videoChecksum := t.Scene.GetHash(t.fileNamingAlgorithm)
|
|
imageFilename := t.imageFilename()
|
|
|
|
if !t.Overwrite && !t.required() {
|
|
return
|
|
}
|
|
|
|
videoFile, err := ffmpeg.NewVideoFile(instance.FFProbePath, t.Scene.Path, false)
|
|
if err != nil {
|
|
logger.Errorf("error reading video file: %s", err.Error())
|
|
return
|
|
}
|
|
|
|
const generateVideo = true
|
|
generator, err := NewPreviewGenerator(*videoFile, videoChecksum, videoFilename, imageFilename, instance.Paths.Generated.Screenshots, generateVideo, t.ImagePreview, t.Options.PreviewPreset.String())
|
|
|
|
if err != nil {
|
|
logger.Errorf("error creating preview generator: %s", err.Error())
|
|
return
|
|
}
|
|
generator.Overwrite = t.Overwrite
|
|
|
|
// set the preview generation configuration from the global config
|
|
generator.Info.ChunkCount = *t.Options.PreviewSegments
|
|
generator.Info.ChunkDuration = *t.Options.PreviewSegmentDuration
|
|
generator.Info.ExcludeStart = *t.Options.PreviewExcludeStart
|
|
generator.Info.ExcludeEnd = *t.Options.PreviewExcludeEnd
|
|
generator.Info.Audio = config.GetInstance().GetPreviewAudio()
|
|
|
|
if err := generator.Generate(); err != nil {
|
|
logger.Errorf("error generating preview: %s", err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
func (t GeneratePreviewTask) required() bool {
|
|
sceneHash := t.Scene.GetHash(t.fileNamingAlgorithm)
|
|
videoExists := t.doesVideoPreviewExist(sceneHash)
|
|
imageExists := !t.ImagePreview || t.doesImagePreviewExist(sceneHash)
|
|
return !imageExists || !videoExists
|
|
}
|
|
|
|
func (t *GeneratePreviewTask) doesVideoPreviewExist(sceneChecksum string) bool {
|
|
if sceneChecksum == "" {
|
|
return false
|
|
}
|
|
|
|
videoExists, _ := utils.FileExists(instance.Paths.Scene.GetStreamPreviewPath(sceneChecksum))
|
|
return videoExists
|
|
}
|
|
|
|
func (t *GeneratePreviewTask) doesImagePreviewExist(sceneChecksum string) bool {
|
|
if sceneChecksum == "" {
|
|
return false
|
|
}
|
|
|
|
imageExists, _ := utils.FileExists(instance.Paths.Scene.GetStreamPreviewImagePath(sceneChecksum))
|
|
return imageExists
|
|
}
|
|
|
|
func (t *GeneratePreviewTask) videoFilename() string {
|
|
return t.Scene.GetHash(t.fileNamingAlgorithm) + ".mp4"
|
|
}
|
|
|
|
func (t *GeneratePreviewTask) imageFilename() string {
|
|
return t.Scene.GetHash(t.fileNamingAlgorithm) + ".webp"
|
|
}
|