Moved everything out of internal

internal/manager/task_scan.go

@@ -1,164 +0,0 @@
-package manager
-
-import (
-	"context"
-	"database/sql"
-	"github.com/stashapp/stash/internal/database"
-	"github.com/stashapp/stash/internal/ffmpeg"
-	"github.com/stashapp/stash/internal/logger"
-	"github.com/stashapp/stash/internal/models"
-	"github.com/stashapp/stash/internal/utils"
-	"path/filepath"
-	"strconv"
-	"sync"
-	"time"
-)
-
-type ScanTask struct {
-	FilePath string
-}
-
-func (t *ScanTask) Start(wg *sync.WaitGroup) {
-	if filepath.Ext(t.FilePath) == ".zip" {
-		t.scanGallery()
-	} else {
-		t.scanScene()
-	}
-
-	wg.Done()
-}
-
-func (t *ScanTask) scanGallery() {
-	qb := models.NewGalleryQueryBuilder()
-	gallery, _ := qb.FindByPath(t.FilePath)
-	if gallery != nil {
-		// We already have this item in the database, keep going
-		return
-	}
-
-	checksum, err := t.calculateChecksum()
-	if err != nil {
-		logger.Error(err.Error())
-		return
-	}
-
-	ctx := context.TODO()
-	tx := database.DB.MustBeginTx(ctx, nil)
-	gallery, _ = qb.FindByChecksum(checksum, tx)
-	if gallery != nil {
-		logger.Infof("%s already exists.  Updating path...", t.FilePath)
-		gallery.Path = t.FilePath
-		_, err = qb.Update(*gallery, tx)
-	} else {
-		logger.Infof("%s doesn't exist.  Creating new item...", t.FilePath)
-		currentTime := time.Now()
-		newGallery := models.Gallery{
-			Checksum: checksum,
-			Path: t.FilePath,
-			CreatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
-			UpdatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
-		}
-		_, err = qb.Create(newGallery, tx)
-	}
-
-	if err != nil {
-		logger.Error(err.Error())
-		_ = tx.Rollback()
-	} else if err := tx.Commit(); err != nil {
-		logger.Error(err.Error())
-	}
-}
-
-func (t *ScanTask) scanScene() {
-	ffprobe := ffmpeg.NewFFProbe(instance.Paths.FixedPaths.FFProbe)
-	ffprobeResult, err := ffprobe.ProbeVideo(t.FilePath)
-	if err != nil {
-		logger.Error(err.Error())
-		return
-	}
-
-	qb := models.NewSceneQueryBuilder()
-	scene, _ := qb.FindByPath(t.FilePath)
-	if scene != nil {
-		// We already have this item in the database, keep going
-		return
-	}
-
-	checksum, err := t.calculateChecksum()
-	if err != nil {
-		logger.Error(err.Error())
-		return
-	}
-
-	t.makeScreenshots(*ffprobeResult, checksum)
-
-	scene, _ = qb.FindByChecksum(checksum)
-	ctx := context.TODO()
-	tx := database.DB.MustBeginTx(ctx, nil)
-	if scene != nil {
-		logger.Infof("%s already exists.  Updating path...", t.FilePath)
-		scene.Path = t.FilePath
-		_, err = qb.Update(*scene, tx)
-	} else {
-		logger.Infof("%s doesn't exist.  Creating new item...", t.FilePath)
-		currentTime := time.Now()
-		newScene := models.Scene{
-			Checksum: checksum,
-			Path: t.FilePath,
-			Duration: sql.NullFloat64{Float64: ffprobeResult.Duration, Valid: true },
-			VideoCodec: sql.NullString{ String: ffprobeResult.VideoCodec, Valid: true},
-			AudioCodec: sql.NullString{ String: ffprobeResult.AudioCodec, Valid: true},
-			Width: sql.NullInt64{ Int64: int64(ffprobeResult.Width), Valid: true },
-			Height: sql.NullInt64{ Int64: int64(ffprobeResult.Height), Valid: true },
-			Framerate: sql.NullFloat64{ Float64: ffprobeResult.FrameRate, Valid: true },
-			Bitrate: sql.NullInt64{ Int64: ffprobeResult.Bitrate, Valid: true },
-			Size: sql.NullString{ String: strconv.Itoa(int(ffprobeResult.Size)), Valid: true },
-			CreatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
-			UpdatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
-		}
-		_, err = qb.Create(newScene, tx)
-	}
-
-	if err != nil {
-		logger.Error(err.Error())
-		_ = tx.Rollback()
-	} else if err := tx.Commit(); err != nil {
-		logger.Error(err.Error())
-	}
-}
-
-func (t *ScanTask) makeScreenshots(probeResult ffmpeg.FFProbeResult, checksum string) {
-	thumbPath := instance.Paths.Scene.GetThumbnailScreenshotPath(checksum)
-	normalPath := instance.Paths.Scene.GetScreenshotPath(checksum)
-
-	thumbExists, _ := utils.FileExists(thumbPath)
-	normalExists, _ := utils.FileExists(normalPath)
-	if thumbExists && normalExists {
-		logger.Debug("Screenshots already exist for this path... skipping")
-		return
-	}
-
-	t.makeScreenshot(probeResult, thumbPath, 5, 320)
-	t.makeScreenshot(probeResult, normalPath, 2, probeResult.Width)
-}
-
-func (t *ScanTask) makeScreenshot(probeResult ffmpeg.FFProbeResult, outputPath string, quality int, width int) {
-	encoder := ffmpeg.NewEncoder(instance.Paths.FixedPaths.FFMPEG)
-	options := ffmpeg.ScreenshotOptions{
-		OutputPath: outputPath,
-		Quality: quality,
-		Time: float64(probeResult.Duration) * 0.2,
-		Width: width,
-	}
-	encoder.Screenshot(probeResult, options)
-}
-
-func (t *ScanTask) calculateChecksum() (string, error) {
-	logger.Infof("%s not found.  Calculating checksum...", t.FilePath)
-	checksum, err := utils.MD5FromFilePath(t.FilePath)
-	if err != nil {
-		return "", err
-	}
-	logger.Debugf("Checksum calculated: %s", checksum)
-	return checksum, nil
-}