refactor: file download process

core/utils.go

@@ -0,0 +1,62 @@
+package core
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/storage"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+func saveFileWithRetry(task *types.Task, destPath string) error {
+	for i := 0; i <= config.Cfg.Retry; i++ {
+		if err := storage.Save(task.Storage, task.Ctx, destPath, task.StoragePath); err != nil {
+			if i == config.Cfg.Retry {
+				return fmt.Errorf("Failed to save file: %w", err)
+			}
+			logger.L.Errorf("Failed to save file: %s, retrying...", err)
+			continue
+		}
+		return nil
+	}
+	return nil
+}
+
+func getProgressBar(progress float64, totalCount int) string {
+	bar := ""
+	barSize := 100 / totalCount
+	for i := 0; i < totalCount; i++ {
+		if int(progress)/barSize > i {
+			bar += "█"
+		} else {
+			bar += "░"
+		}
+	}
+	return bar
+}
+
+func cleanCacheFile(destPath string) {
+	if config.Cfg.Temp.CacheTTL > 0 {
+		common.RmFileAfter(destPath, time.Duration(config.Cfg.Temp.CacheTTL)*time.Second)
+	} else {
+		if err := os.Remove(destPath); err != nil {
+			logger.L.Errorf("Failed to purge file: %s", err)
+		}
+	}
+}
+
+func calculateBarTotalCount(fileSize int64) int {
+	barTotalCount := 5
+	if fileSize > 1024*1024*1000 {
+		barTotalCount = 50
+	} else if fileSize > 1024*1024*500 {
+		barTotalCount = 20
+	} else if fileSize > 1024*1024*200 {
+		barTotalCount = 10
+	}
+	return barTotalCount
+}