24 create convert functionality for videos

.env.example

@@ -14,6 +14,7 @@ SECRET=some-other-super-secret
 
 WEBSOCKET_HEARTBEAT_INTERVAL=15000
 
+CACHE=/cache
 ASSETS=/assets
 WEB=/web
 

apps/server/internal/dto/helper_types.go

@@ -0,0 +1,41 @@
+package dto
+
+import "github.com/slugger7/exorcist/apps/server/internal/ffmpeg"
+
+type Dimension struct {
+	Height *int `json:"height"`
+	Width  *int `json:"width"`
+}
+
+func (d *Dimension) ToFfmpegDto() *ffmpeg.Dimension {
+	v := ffmpeg.Dimension{}
+	if d.Height != nil {
+		v.Height = new(int)
+		*v.Height = *d.Height
+	}
+
+	if d.Width != nil {
+		v.Width = new(int)
+		*v.Width = *d.Width
+	}
+
+	return &v
+}
+
+func (d *Dimension) FromFfmpegDto(m *ffmpeg.Dimension) *Dimension {
+	if m.Height != nil {
+		if d.Height == nil {
+			d.Height = new(int)
+		}
+		*d.Height = *m.Height
+	}
+
+	if m.Width != nil {
+		if d.Width == nil {
+			d.Width = new(int)
+		}
+		*d.Width = *m.Width
+	}
+
+	return d
+}

apps/server/internal/dto/job.go

@@ -11,7 +11,7 @@ import (
 
 type CreateJobDTO struct {
 	Type     model.JobTypeEnum      `json:"type" binding:"required" tstype:"model.JobTypeEnum"`
-	Data     map[string]interface{} `json:"data" tstype:"ScanPathData | GenerateThumbnailData"`
+	Data     map[string]interface{} `json:"data" tstype:"ScanPathData | GenerateThumbnailData | GenerateChaptersData | ConvertData | RefreshMetadata | RefreshLibraryMetadata"`
 	Priority *JobPriority           `json:"priority"`
 }
 

apps/server/internal/dto/job_data.go

@@ -3,6 +3,7 @@ package dto
 import (
 	"github.com/google/uuid"
 	"github.com/slugger7/exorcist/apps/server/internal/db/exorcist/public/model"
+	"github.com/slugger7/exorcist/apps/server/internal/ffmpeg"
 )
 
 type ScanPathData struct {
@@ -11,13 +12,11 @@ type ScanPathData struct {
 
 type GenerateThumbnailData struct {
 	MediaId uuid.UUID `json:"mediaId"`
-	Path    string    `json:"path"`
+	Path    string    `json:"path" tstype:"-"`
 	// Optional: If set to 0, timestamp at 25% of video playback will be used. Value in seconds
-	Timestamp float64 `json:"timestamp"`
-	// Optional: If set to 0, video height will be used
-	Height int `json:"height"`
-	// Optional: If set to 0, video widtch will be used
-	Width        int                          `json:"width"`
+	Timestamp    float64                      `json:"timestamp"`
+	Height       *int                         `json:"height"`
+	Width        *int                         `json:"width"`
 	RelationType *model.MediaRelationTypeEnum `json:"relationType"`
 	Metadata     *ThumbnailMetadataDTO        `json:"metadata"`
 }
@@ -41,8 +40,36 @@ type RefreshLibraryMetadata struct {
 type GenerateChaptersData struct {
 	MediaId      uuid.UUID `json:"mediaId"`
 	Interval     float64   `json:"interval"`
-	Height       int       `json:"height"`
-	Width        int       `json:"width"`
+	Height       *int      `json:"height"`
+	Width        *int      `json:"width"`
 	MaxDimension int       `json:"maxDimension"`
 	Overwrite    bool      `json:"overwrite"`
 }
+
+type ConvertData struct {
+	MediaId            uuid.UUID `json:"mediaId" binding:"required"`
+	Dimension          Dimension `json:"dimension"`
+	Filename           string    `json:"filename" binding:"required"`
+	CopyTags           *bool     `json:"copyTags"`
+	CopyPeople         *bool     `json:"copyPeople"`
+	Path               string    `json:"path" tstype:"-"` // omitted for clients
+	ConstantRateFactor *int      `json:"constantRateFactor"`
+	VariableBitrate    *int      `json:"variableBitrate"`
+	ForcePixelFormat   *string   `json:"forcePixelFormat"`
+}
+
+func (d *ConvertData) ToFfmpegDto() *ffmpeg.ConvertDto {
+	v := &ffmpeg.ConvertDto{
+		Dimension: ffmpeg.Dimension{
+			Height: new(int),
+			Width:  new(int),
+		},
+		ConstantRateFactor: d.ConstantRateFactor,
+		VariableBitrate:    d.VariableBitrate,
+		ForcePixelFormat:   d.ForcePixelFormat,
+	}
+	*v.Dimension.Height = *d.Dimension.Height
+	*v.Dimension.Width = *d.Dimension.Width
+
+	return v
+}

apps/server/internal/dto/media_relation.go

@@ -20,7 +20,7 @@ type MediaRelationDto struct {
 	// to the correct type if needed.
 	// This is only used to give the client a full json object without them needing
 	// to parse the json string
-	Metadata any `json:"metadata"`
+	Metadata any `json:"metadata" tstype:"ThumbnailMetadataDTO | ChapterMetadadataDTO | null"`
 }
 
 func (d *MediaRelationDto) FromModel(m models.MediaRelation) MediaRelationDto {

apps/server/internal/environment/environment.go

@@ -32,6 +32,7 @@ type EnvironmentVariables struct {
 	Port                       int
 	Secret                     string
 	LogLevel                   string
+	Cache                      string
 	Assets                     string
 	Web                        *string
 	JobRunner                  bool
@@ -58,6 +59,7 @@ const (
 	PORT                         OsEnv = "PORT"
 	SECRET                       OsEnv = "SECRET"
 	LOG_LEVEL                    OsEnv = "LOG_LEVEL"
+	CACHE                        OsEnv = "CACHE"
 	ASSETS                       OsEnv = "ASSETS"
 	WEB                          OsEnv = "WEB"
 	JOB_RUNNER                   OsEnv = "JOB_RUNNER"
@@ -94,6 +96,7 @@ func RefreshEnvironmentVariables() {
 		Port:                       getIntValue(PORT),
 		Secret:                     os.Getenv(SECRET),
 		LogLevel:                   getValueOrDefault(LOG_LEVEL, "debug"),
+		Cache:                      os.Getenv(CACHE),
 		Assets:                     os.Getenv(ASSETS),
 		Web:                        getValueOrNil(WEB),
 		JobRunner:                  getBoolValue(JOB_RUNNER, true),

apps/server/internal/ffmpeg/convert.go

@@ -0,0 +1,49 @@
+package ffmpeg
+
+import (
+	"fmt"
+	"os"
+
+	errs "github.com/slugger7/exorcist/apps/server/internal/errors"
+	ffmpeg_go "github.com/u2takey/ffmpeg-go"
+)
+
+type ConvertDto struct {
+	InputFilePath      string
+	OutputFilePath     string
+	Dimension          Dimension
+	ConstantRateFactor *int
+	VariableBitrate    *int
+	ForcePixelFormat   *string
+}
+
+func Convert(c ConvertDto) error {
+	if *c.Dimension.Height <= 0 {
+		return fmt.Errorf(ErrNegativeHeight, *c.Dimension.Height)
+	}
+	if *c.Dimension.Width <= 0 {
+		return fmt.Errorf(ErrNegativeWidth, *c.Dimension.Width)
+	}
+
+	ouptutArgs := ffmpeg_go.KwArgs{"vf": fmt.Sprintf("scale=%v:%v", *c.Dimension.Width, *c.Dimension.Height)}
+
+	if c.ConstantRateFactor != nil {
+		ouptutArgs["crf"] = *c.ConstantRateFactor
+	}
+
+	if c.ForcePixelFormat != nil {
+		ouptutArgs["pix_fmt"] = *c.ForcePixelFormat
+	}
+
+	err := ffmpeg_go.Input(c.InputFilePath).Output(c.OutputFilePath,
+		ouptutArgs).
+		Run()
+	if err != nil {
+		str := err.Error()
+		_ = str
+		_ = os.Remove(c.OutputFilePath)
+		return errs.BuildError(err, "error converting %v to %v", c.InputFilePath, c.OutputFilePath)
+	}
+
+	return nil
+}

apps/server/internal/ffmpeg/image.go

@@ -21,6 +21,56 @@ func ScaleHeightByWidth(currentHeight, currentWidth, wantedWidth int) int {
 	return int(float32(currentHeight) / float32(currentWidth) * float32(wantedWidth))
 }
 
+type Dimension struct {
+	Height *int `json:"height"`
+	Width  *int `json:"width"`
+}
+
+func DetermineDimensions(wanted, current Dimension) Dimension {
+	if wanted.Height != nil && wanted.Width != nil {
+		return wanted
+	}
+
+	if wanted.Height == nil && wanted.Width == nil {
+		return current
+	}
+
+	if wanted.Height != nil {
+		scaledWidth := ScaleWidthByHeight(*current.Height, *current.Width, *wanted.Height)
+		return Dimension{
+			Height: wanted.Height,
+			Width:  &scaledWidth,
+		}
+	} else {
+		scaledHeight := ScaleHeightByWidth(*current.Height, *current.Width, *wanted.Width)
+		return Dimension{
+			Height: &scaledHeight,
+			Width:  wanted.Width,
+		}
+	}
+}
+
+func ScaleByMaxDimension(maxDimension int, currentDimension Dimension) *Dimension {
+	d := Dimension{
+		Height: new(int),
+		Width:  new(int),
+	}
+	*d.Height = *currentDimension.Height
+	*d.Width = *currentDimension.Width
+
+	if *d.Width > maxDimension {
+		*d.Height = ScaleHeightByWidth(*d.Height, *d.Width, maxDimension)
+		*d.Width = maxDimension
+	}
+
+	if *d.Height > maxDimension {
+		*d.Width = ScaleWidthByHeight(*d.Height, *d.Width, maxDimension)
+		*d.Height = maxDimension
+	}
+
+	return &d
+}
+
 func ImageAt(vid string, time float64, img string, width, height int) error {
 	if width <= 0 {
 		return fmt.Errorf(ErrNegativeWidth, width)