Reorg

pkg/manager/paths/paths.go

@@ -0,0 +1,27 @@
+package paths
+
+import (
+	"github.com/stashapp/stash/pkg/manager/jsonschema"
+)
+
+type Paths struct {
+	Config    *jsonschema.Config
+	Generated *generatedPaths
+	JSON      *jsonPaths
+
+	Gallery      *galleryPaths
+	Scene        *scenePaths
+	SceneMarkers *sceneMarkerPaths
+}
+
+func NewPaths(config *jsonschema.Config) *Paths {
+	p := Paths{}
+	p.Config = config
+	p.Generated = newGeneratedPaths(p)
+	p.JSON = newJSONPaths(p)
+
+	p.Gallery = newGalleryPaths(p.Config)
+	p.Scene = newScenePaths(p)
+	p.SceneMarkers = newSceneMarkerPaths(p)
+	return &p
+}

pkg/manager/paths/paths_gallery.go

@@ -0,0 +1,24 @@
+package paths
+
+import (
+	"github.com/stashapp/stash/pkg/manager/jsonschema"
+	"path/filepath"
+)
+
+type galleryPaths struct {
+	config *jsonschema.Config
+}
+
+func newGalleryPaths(c *jsonschema.Config) *galleryPaths {
+	gp := galleryPaths{}
+	gp.config = c
+	return &gp
+}
+
+func (gp *galleryPaths) GetExtractedPath(checksum string) string {
+	return filepath.Join(gp.config.Cache, checksum)
+}
+
+func (gp *galleryPaths) GetExtractedFilePath(checksum string, fileName string) string {
+	return filepath.Join(gp.config.Cache, checksum, fileName)
+}

pkg/manager/paths/paths_generated.go

@@ -0,0 +1,40 @@
+package paths
+
+import (
+	"github.com/stashapp/stash/pkg/utils"
+	"path/filepath"
+)
+
+type generatedPaths struct {
+	Screenshots string
+	Vtt         string
+	Markers     string
+	Transcodes  string
+	Tmp         string
+}
+
+func newGeneratedPaths(p Paths) *generatedPaths {
+	gp := generatedPaths{}
+	gp.Screenshots = filepath.Join(p.Config.Metadata, "screenshots")
+	gp.Vtt = filepath.Join(p.Config.Metadata, "vtt")
+	gp.Markers = filepath.Join(p.Config.Metadata, "markers")
+	gp.Transcodes = filepath.Join(p.Config.Metadata, "transcodes")
+	gp.Tmp = filepath.Join(p.Config.Metadata, "tmp")
+	return &gp
+}
+
+func (gp *generatedPaths) GetTmpPath(fileName string) string {
+	return filepath.Join(gp.Tmp, fileName)
+}
+
+func (gp *generatedPaths) EnsureTmpDir() {
+	_ = utils.EnsureDir(gp.Tmp)
+}
+
+func (gp *generatedPaths) EmptyTmpDir() {
+	_ = utils.EmptyDir(gp.Tmp)
+}
+
+func (gp *generatedPaths) RemoveTmpDir() {
+	_ = utils.RemoveDir(gp.Tmp)
+}

pkg/manager/paths/paths_json.go

@@ -0,0 +1,38 @@
+package paths
+
+import (
+	"path/filepath"
+)
+
+type jsonPaths struct {
+	MappingsFile string
+	ScrapedFile  string
+
+	Performers string
+	Scenes     string
+	Galleries  string
+	Studios    string
+}
+
+func newJSONPaths(p Paths) *jsonPaths {
+	jp := jsonPaths{}
+	jp.MappingsFile = filepath.Join(p.Config.Metadata, "mappings.json")
+	jp.ScrapedFile = filepath.Join(p.Config.Metadata, "scraped.json")
+	jp.Performers = filepath.Join(p.Config.Metadata, "performers")
+	jp.Scenes = filepath.Join(p.Config.Metadata, "scenes")
+	jp.Galleries = filepath.Join(p.Config.Metadata, "galleries")
+	jp.Studios = filepath.Join(p.Config.Metadata, "studios")
+	return &jp
+}
+
+func (jp *jsonPaths) PerformerJSONPath(checksum string) string {
+	return filepath.Join(jp.Performers, checksum+".json")
+}
+
+func (jp *jsonPaths) SceneJSONPath(checksum string) string {
+	return filepath.Join(jp.Scenes, checksum+".json")
+}
+
+func (jp *jsonPaths) StudioJSONPath(checksum string) string {
+	return filepath.Join(jp.Studios, checksum+".json")
+}

pkg/manager/paths/paths_scene_markers.go

@@ -0,0 +1,24 @@
+package paths
+
+import (
+	"path/filepath"
+	"strconv"
+)
+
+type sceneMarkerPaths struct {
+	generated generatedPaths
+}
+
+func newSceneMarkerPaths(p Paths) *sceneMarkerPaths {
+	sp := sceneMarkerPaths{}
+	sp.generated = *p.Generated
+	return &sp
+}
+
+func (sp *sceneMarkerPaths) GetStreamPath(checksum string, seconds int) string {
+	return filepath.Join(sp.generated.Markers, checksum, strconv.Itoa(seconds)+".mp4")
+}
+
+func (sp *sceneMarkerPaths) GetStreamPreviewImagePath(checksum string, seconds int) string {
+	return filepath.Join(sp.generated.Markers, checksum, strconv.Itoa(seconds)+".webp")
+}

pkg/manager/paths/paths_scenes.go

@@ -0,0 +1,53 @@
+package paths
+
+import (
+	"github.com/stashapp/stash/pkg/utils"
+	"path/filepath"
+)
+
+type scenePaths struct {
+	generated generatedPaths
+}
+
+func newScenePaths(p Paths) *scenePaths {
+	sp := scenePaths{}
+	sp.generated = *p.Generated
+	return &sp
+}
+
+func (sp *scenePaths) GetScreenshotPath(checksum string) string {
+	return filepath.Join(sp.generated.Screenshots, checksum+".jpg")
+}
+
+func (sp *scenePaths) GetThumbnailScreenshotPath(checksum string) string {
+	return filepath.Join(sp.generated.Screenshots, checksum+".thumb.jpg")
+}
+
+func (sp *scenePaths) GetTranscodePath(checksum string) string {
+	return filepath.Join(sp.generated.Transcodes, checksum+".mp4")
+}
+
+func (sp *scenePaths) GetStreamPath(scenePath string, checksum string) string {
+	transcodePath := sp.GetTranscodePath(checksum)
+	transcodeExists, _ := utils.FileExists(transcodePath)
+	if transcodeExists {
+		return transcodePath
+	}
+	return scenePath
+}
+
+func (sp *scenePaths) GetStreamPreviewPath(checksum string) string {
+	return filepath.Join(sp.generated.Screenshots, checksum+".mp4")
+}
+
+func (sp *scenePaths) GetStreamPreviewImagePath(checksum string) string {
+	return filepath.Join(sp.generated.Screenshots, checksum+".webp")
+}
+
+func (sp *scenePaths) GetSpriteImageFilePath(checksum string) string {
+	return filepath.Join(sp.generated.Vtt, checksum+"_sprite.jpg")
+}
+
+func (sp *scenePaths) GetSpriteVttFilePath(checksum string) string {
+	return filepath.Join(sp.generated.Vtt, checksum+"_thumbs.vtt")
+}

pkg/manager/paths/paths_static.go

@@ -0,0 +1,44 @@
+package paths
+
+import (
+	"os"
+	"os/user"
+	"path/filepath"
+)
+
+type StaticPathsType struct {
+	ExecutionDirectory string
+	ConfigDirectory    string
+	ConfigFile         string
+	DatabaseFile       string
+
+	FFMPEG  string
+	FFProbe string
+}
+
+var StaticPaths = StaticPathsType{
+	ExecutionDirectory: getExecutionDirectory(),
+	ConfigDirectory:    getConfigDirectory(),
+	ConfigFile:         filepath.Join(getConfigDirectory(), "config.json"),
+	DatabaseFile:       filepath.Join(getConfigDirectory(), "stash-go.sqlite"),
+}
+
+func getExecutionDirectory() string {
+	ex, err := os.Executable()
+	if err != nil {
+		panic(err)
+	}
+	return filepath.Dir(ex)
+}
+
+func getHomeDirectory() string {
+	currentUser, err := user.Current()
+	if err != nil {
+		panic(err)
+	}
+	return currentUser.HomeDir
+}
+
+func getConfigDirectory() string {
+	return filepath.Join(getHomeDirectory(), ".stash")
+}