all

Downloads/Voltaserve/conversion/pipeline/video_pipeline.go

@@ -0,0 +1,49 @@
+package pipeline
+
+import (
+	"os"
+	"path/filepath"
+	"voltaserve/client"
+	"voltaserve/core"
+	"voltaserve/helper"
+	"voltaserve/identifier"
+	"voltaserve/infra"
+	"voltaserve/processor"
+)
+
+type videoPipeline struct {
+	pipelineIdentifier *identifier.PipelineIdentifier
+	videoProc          *processor.VideoProcessor
+	s3                 *infra.S3Manager
+	apiClient          *client.APIClient
+}
+
+func NewVideoPipeline() core.Pipeline {
+	return &videoPipeline{
+		pipelineIdentifier: identifier.NewPipelineIdentifier(),
+		videoProc:          processor.NewVideoProcessor(),
+		s3:                 infra.NewS3Manager(),
+		apiClient:          client.NewAPIClient(),
+	}
+}
+
+func (p *videoPipeline) Run(opts core.PipelineRunOptions) error {
+	inputPath := filepath.FromSlash(os.TempDir() + "/" + helper.NewID() + filepath.Ext(opts.Key))
+	if err := p.s3.GetFile(opts.Key, inputPath, opts.Bucket); err != nil {
+		return err
+	}
+	thumbnail, err := p.videoProc.Base64Thumbnail(inputPath)
+	if err != nil {
+		return err
+	}
+	if err := p.apiClient.UpdateSnapshot(core.SnapshotUpdateOptions{
+		Options:   opts,
+		Thumbnail: &thumbnail,
+	}); err != nil {
+		return err
+	}
+	if err := os.Remove(inputPath); err != nil {
+		return err
+	}
+	return nil
+}