more cleaning logics

bot/core/download.go

@@ -194,7 +194,15 @@ func DownloadMedias(
 	}
 
 	if firstError != nil {
-		return results, firstError
+		for _, result := range results {
+			if result.FilePath != "" {
+				os.Remove(result.FilePath)
+			}
+			if result.ThumbnailFilePath != "" {
+				os.Remove(result.ThumbnailFilePath)
+			}
+		}
+		return nil, firstError
 	}
 
 	if len(results) > 1 {

main.go

@@ -14,6 +14,7 @@ func main() {
 	if err != nil {
 		log.Fatal("error loading .env file")
 	}
+	util.ClearDownlods()
 	ok := util.CheckFFmpeg()
 	if !ok {
 		log.Fatal("ffmpeg executable not found. please install it or add it to your PATH")

util/download.go

@@ -481,9 +481,19 @@ func downloadSegments(
 
 	errChan := make(chan error, len(segmentURLs))
 	var errMutex sync.Mutex
+
 	var firstErr error
 
 	downloadedFiles := make([]string, len(segmentURLs))
+	defer func() {
+		if firstErr != nil {
+			for _, path := range downloadedFiles {
+				if path != "" {
+					os.Remove(path)
+				}
+			}
+		}
+	}()
 
 	downloadCtx, cancelDownload := context.WithCancel(ctx)
 	defer cancelDownload()

util/misc.go

@@ -110,3 +110,11 @@ func CheckFFmpeg() bool {
 	_, err := exec.LookPath("ffmpeg")
 	return err == nil
 }
+
+func ClearDownlods() {
+	downloadsDir := os.Getenv("DOWNLOADS_DIR")
+	if downloadsDir == "" {
+		downloadsDir = "downloads"
+	}
+	os.RemoveAll(downloadsDir)
+}