feat: update

backend/app/jobs/remove_downloaded_video.go

@@ -0,0 +1,54 @@
+package jobs
+
+import (
+	"context"
+	"os"
+	"time"
+
+	. "github.com/riverqueue/river"
+	log "github.com/sirupsen/logrus"
+	_ "go.ipao.vip/atom"
+	"go.ipao.vip/atom/contracts"
+)
+
+var _ contracts.JobArgs = (*RemoveDownloadedVideo)(nil)
+
+type RemoveDownloadedVideo struct {
+	FilePath string `json:"file_path"`
+}
+
+func (s RemoveDownloadedVideo) InsertOpts() InsertOpts {
+	return InsertOpts{
+		Queue:       QueueDefault,
+		Priority:    PriorityDefault,
+		ScheduledAt: time.Now().Add(time.Minute * 10),
+	}
+}
+
+func (s RemoveDownloadedVideo) Kind() string     { return "remove_downloaded_video" }
+func (a RemoveDownloadedVideo) UniqueID() string { return a.Kind() }
+
+var _ Worker[RemoveDownloadedVideo] = (*RemoveDownloadedVideoWorker)(nil)
+
+// @provider(job)
+type RemoveDownloadedVideoWorker struct {
+	WorkerDefaults[RemoveDownloadedVideo]
+}
+
+func (w *RemoveDownloadedVideoWorker) NextRetry(job *Job[RemoveDownloadedVideo]) time.Time {
+	return time.Now().Add(30 * time.Second)
+}
+
+func (w *RemoveDownloadedVideoWorker) Work(ctx context.Context, job *Job[RemoveDownloadedVideo]) error {
+	log := log.WithField("job", job.Args.Kind())
+
+	log.Infof("[Start] Working on job with strings: %+v", job.Args)
+	defer log.Infof("[End] Finished %s", job.Args.Kind())
+
+	if err := os.Remove(job.Args.FilePath); err != nil {
+		log.Errorf("Error removing file: %v", err)
+		return err
+	}
+
+	return nil
+}

backend/app/jobs/video_cut.go

@@ -87,6 +87,13 @@ func (w *VideoCutWorker) Work(ctx context.Context, job *Job[VideoCut]) error {
 		return JobCancel(err)
 	}
 
+	// remove original file
+	if w.job.Add(&RemoveDownloadedVideo{
+		FilePath: input,
+	}); err != nil {
+		log.Errorf("Error removing original file: %v", err)
+	}
+
 	// save to database
 	return w.job.Add(&VideoStoreShort{
 		MediaHash: media.Hash,