Auto fix MU1 file duration

Author: KillerX

activities/transcode.go

@@ -468,3 +468,38 @@ func (va VideoActivities) TranscodeToHAPActivity(ctx context.Context, input HAPI
 		OutputPath: paths.MustParse(transcodeResult.OutputPath),
 	}, nil
 }
+
+type FixDurationInput struct {
+	InputPath  paths.Path
+	OutputPath paths.Path
+}
+
+type FixDurationResult struct {
+	OutputPath paths.Path
+}
+
+func (va VideoActivities) FixDurationActivity(ctx context.Context, input FixDurationInput) (*FixDurationResult, error) {
+	log := activity.GetLogger(ctx)
+	activity.RecordHeartbeat(ctx, "FixDuration")
+	log.Info("Starting FixDurationActivity")
+
+	stop, progressCallback := registerProgressCallback(ctx)
+	defer close(stop)
+
+	args := []string{
+		"-i", input.InputPath.Local(),
+		"-c:v", "copy",
+		"-c:a", "copy",
+		"-map", "0",
+		input.OutputPath.Local(),
+	}
+
+	_, err := ffmpeg.Do(args, ffmpeg.StreamInfo{}, progressCallback)
+	if err != nil {
+		return nil, err
+	}
+
+	return &FixDurationResult{
+		OutputPath: input.OutputPath,
+	}, nil
+}

workflows/ingest/incremental_ingest.go

@@ -282,6 +282,19 @@ func doIncremental(ctx workflow.Context, params IncrementalParams) error {
 		NotificationChannel: &telegram.ChatOther,
 	})
 
+	// Fix duration metadata issues
+	fixDurationFuture := workflow.ExecuteChildWorkflow(ctx, miscworkflows.FixDurationVX, miscworkflows.FixDurationVXInput{
+		VXID: videoVXID,
+	})
+
+	// Handle errors in background
+	workflow.Go(ctx, func(ctx workflow.Context) {
+		err := fixDurationFuture.Get(ctx, nil)
+		if err != nil {
+			wfutils.SendTelegramText(ctx, telegram.ChatOther, fmt.Sprintf("🟥 Duration fix failed for %s: %v", videoVXID, err))
+		}
+	})
+
 	_ = wfutils.Execute(ctx, activities.Vidispine.CreateThumbnailsActivity, vsactivity.CreateThumbnailsParams{
 		AssetID: videoVXID,
 	}).Get(ctx, nil)

workflows/misc/fix_duration.go

@@ -0,0 +1,63 @@
+package miscworkflows
+
+import (
+	"fmt"
+
+	"github.com/bcc-code/bcc-media-flows/activities"
+	vsactivity "github.com/bcc-code/bcc-media-flows/activities/vidispine"
+	"github.com/bcc-code/bcc-media-flows/services/rclone"
+	"github.com/bcc-code/bcc-media-flows/services/telegram"
+	wfutils "github.com/bcc-code/bcc-media-flows/utils/workflows"
+	"go.temporal.io/sdk/workflow"
+)
+
+type FixDurationVXInput struct {
+	VXID string
+}
+
+func FixDurationVX(
+	ctx workflow.Context,
+	params FixDurationVXInput,
+) error {
+	logger := workflow.GetLogger(ctx)
+	logger.Info("Starting FixDurationVX")
+
+	ctx = workflow.WithActivityOptions(ctx, wfutils.GetDefaultActivityOptions())
+
+	// Get the original file from Vidispine
+	originalFile, err := wfutils.Execute(ctx, activities.Vidispine.GetFileFromVXActivity, vsactivity.GetFileFromVXParams{
+		VXID: params.VXID,
+		Tags: []string{"original"},
+	}).Result(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to get original file: %w", err)
+	}
+
+	// Get temp folder for processing
+	tempFolder, err := wfutils.GetWorkflowTempFolder(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to get temp folder: %w", err)
+	}
+
+	// Create output path in temp folder
+	outputPath := tempFolder.Append(originalFile.FilePath.Base()).SetExt(originalFile.FilePath.Ext())
+
+	// Run FFmpeg to fix duration
+	_, err = wfutils.Execute(ctx, activities.Video.FixDurationActivity, activities.FixDurationInput{
+		InputPath:  originalFile.FilePath,
+		OutputPath: outputPath,
+	}).Result(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to fix duration: %w", err)
+	}
+
+	// Replace the original file on disk with the fixed version
+	err = wfutils.MoveFile(ctx, outputPath, originalFile.FilePath, rclone.PriorityNormal)
+	if err != nil {
+		return fmt.Errorf("failed to replace original file: %w", err)
+	}
+
+	wfutils.SendTelegramText(ctx, telegram.ChatOther, fmt.Sprintf("✅ Duration fix completed for %s", params.VXID))
+
+	return nil
+}

workflows/workflows.go

@@ -24,6 +24,7 @@ var TriggerableWorkflows = []any{
 	miscworkflows.MergeAndImportSubtitlesFromCSV,
 	miscworkflows.UpdateAssetRelations,
 	miscworkflows.NormalizeAudioLevelWorkflow,
+	miscworkflows.FixDurationVX,
 	miscworkflows.MoveMBFile,
 	miscworkflows.MoveFilesWorkerFlow,
 	scheduled.CleanupTemp,
@@ -68,6 +69,7 @@ var WorkerWorkflows = []any{
 	ingestworkflows.Multitrack,
 	ingestworkflows.ImportSubtitles,
 	miscworkflows.NormalizeAudioLevelWorkflow,
+	miscworkflows.FixDurationVX,
 	vb_export.VBExport,
 	vb_export.VBExportToAbekas,
 	vb_export.VBExportToBStage,