mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
7cff71c35f
* Refactor transaction hooks. Add preCommit * Add BlobStore * Use blobStore for tag images * Use blobStore for studio images * Use blobStore for performer images * Use blobStore for scene covers * Don't generate screenshots in legacy directory * Run post-hooks outside original transaction * Use blobStore for movie images * Remove unnecessary DestroyImage methods * Add missing filter for scene cover * Add covers to generate options * Add generate cover option to UI * Add screenshot migration * Delete thumb files as part of screenshot migration
78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
package scene
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/stashapp/stash/pkg/fsutil"
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
"github.com/stashapp/stash/pkg/models/paths"
|
|
)
|
|
|
|
func MigrateHash(p *paths.Paths, oldHash string, newHash string) {
|
|
oldPath := filepath.Join(p.Generated.Markers, oldHash)
|
|
newPath := filepath.Join(p.Generated.Markers, newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
|
|
scenePaths := p.Scene
|
|
oldPath = scenePaths.GetVideoPreviewPath(oldHash)
|
|
newPath = scenePaths.GetVideoPreviewPath(newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
|
|
oldPath = scenePaths.GetWebpPreviewPath(oldHash)
|
|
newPath = scenePaths.GetWebpPreviewPath(newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
|
|
oldPath = scenePaths.GetTranscodePath(oldHash)
|
|
newPath = scenePaths.GetTranscodePath(newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
|
|
oldVttPath := scenePaths.GetSpriteVttFilePath(oldHash)
|
|
newVttPath := scenePaths.GetSpriteVttFilePath(newHash)
|
|
migrateSceneFiles(oldVttPath, newVttPath)
|
|
|
|
oldPath = scenePaths.GetSpriteImageFilePath(oldHash)
|
|
newPath = scenePaths.GetSpriteImageFilePath(newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
migrateVttFile(newVttPath, oldPath, newPath)
|
|
|
|
oldPath = scenePaths.GetInteractiveHeatmapPath(oldHash)
|
|
newPath = scenePaths.GetInteractiveHeatmapPath(newHash)
|
|
migrateSceneFiles(oldPath, newPath)
|
|
}
|
|
|
|
func migrateSceneFiles(oldName, newName string) {
|
|
oldExists, err := fsutil.FileExists(oldName)
|
|
if err != nil && !os.IsNotExist(err) {
|
|
logger.Errorf("Error checking existence of %s: %s", oldName, err.Error())
|
|
return
|
|
}
|
|
|
|
if oldExists {
|
|
logger.Infof("renaming %s to %s", oldName, newName)
|
|
if err := os.Rename(oldName, newName); err != nil {
|
|
logger.Errorf("error renaming %s to %s: %s", oldName, newName, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
// #2481: migrate vtt file contents in addition to renaming
|
|
func migrateVttFile(vttPath, oldSpritePath, newSpritePath string) {
|
|
contents, err := os.ReadFile(vttPath)
|
|
if err != nil {
|
|
logger.Errorf("Error reading %s for vtt migration: %v", vttPath, err)
|
|
return
|
|
}
|
|
|
|
oldSpriteBasename := filepath.Base(oldSpritePath)
|
|
newSpriteBasename := filepath.Base(newSpritePath)
|
|
|
|
contents = bytes.ReplaceAll(contents, []byte(oldSpriteBasename), []byte(newSpriteBasename))
|
|
|
|
if err := os.WriteFile(vttPath, contents, 0644); err != nil {
|
|
logger.Errorf("Error writing %s for vtt migration: %v", vttPath, err)
|
|
return
|
|
}
|
|
}
|