mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 08:26:00 +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
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/stashapp/stash/internal/manager/config"
|
|
"github.com/stashapp/stash/pkg/fsutil"
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/txn"
|
|
"github.com/stashapp/stash/pkg/utils"
|
|
)
|
|
|
|
type StreamRequestContext struct {
|
|
context.Context
|
|
ResponseWriter http.ResponseWriter
|
|
}
|
|
|
|
func NewStreamRequestContext(w http.ResponseWriter, r *http.Request) *StreamRequestContext {
|
|
return &StreamRequestContext{
|
|
Context: r.Context(),
|
|
ResponseWriter: w,
|
|
}
|
|
}
|
|
|
|
func (c *StreamRequestContext) Cancel() {
|
|
hj, ok := (c.ResponseWriter).(http.Hijacker)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
// hijack and close the connection
|
|
conn, _, _ := hj.Hijack()
|
|
if conn != nil {
|
|
conn.Close()
|
|
}
|
|
}
|
|
|
|
func KillRunningStreams(scene *models.Scene, fileNamingAlgo models.HashAlgorithm) {
|
|
instance.ReadLockManager.Cancel(scene.Path())
|
|
|
|
sceneHash := scene.GetHash(fileNamingAlgo)
|
|
|
|
if sceneHash == "" {
|
|
return
|
|
}
|
|
|
|
transcodePath := GetInstance().Paths.Scene.GetTranscodePath(sceneHash)
|
|
instance.ReadLockManager.Cancel(transcodePath)
|
|
}
|
|
|
|
type SceneCoverGetter interface {
|
|
GetCover(ctx context.Context, sceneID int) ([]byte, error)
|
|
}
|
|
|
|
type SceneServer struct {
|
|
TxnManager txn.Manager
|
|
SceneCoverGetter SceneCoverGetter
|
|
}
|
|
|
|
func (s *SceneServer) StreamSceneDirect(scene *models.Scene, w http.ResponseWriter, r *http.Request) {
|
|
fileNamingAlgo := config.GetInstance().GetVideoFileNamingAlgorithm()
|
|
|
|
filepath := GetInstance().Paths.Scene.GetStreamPath(scene.Path(), scene.GetHash(fileNamingAlgo))
|
|
streamRequestCtx := NewStreamRequestContext(w, r)
|
|
|
|
// #2579 - hijacking and closing the connection here causes video playback to fail in Safari
|
|
// We trust that the request context will be closed, so we don't need to call Cancel on the
|
|
// returned context here.
|
|
_ = GetInstance().ReadLockManager.ReadLock(streamRequestCtx, filepath)
|
|
http.ServeFile(w, r, filepath)
|
|
}
|
|
|
|
func (s *SceneServer) ServeScreenshot(scene *models.Scene, w http.ResponseWriter, r *http.Request) {
|
|
filepath := GetInstance().Paths.Scene.GetScreenshotPath(scene.GetHash(config.GetInstance().GetVideoFileNamingAlgorithm()))
|
|
|
|
// fall back to the scene image blob if the file isn't present
|
|
screenshotExists, _ := fsutil.FileExists(filepath)
|
|
if screenshotExists {
|
|
http.ServeFile(w, r, filepath)
|
|
} else {
|
|
var cover []byte
|
|
err := txn.WithTxn(r.Context(), s.TxnManager, func(ctx context.Context) error {
|
|
cover, _ = s.SceneCoverGetter.GetCover(ctx, scene.ID)
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
logger.Warnf("read transaction failed while serving screenshot: %v", err)
|
|
}
|
|
|
|
if err = utils.ServeImage(cover, w, r); err != nil {
|
|
logger.Warnf("unable to serve screenshot image: %v", err)
|
|
}
|
|
}
|
|
}
|