Update main.go

main.go

@@ -160,33 +160,28 @@ func unrarSource(source, dest string) error {
 	cmd := exec.Command("unrar", "x", "-y", source, dest+string(os.PathSeparator))
 	return cmd.Run()
 }
-
-// --- Download & Zip Logic ---
 func downloadFile(url string, customName string, destFolder string) (string, error) {
-
-    resp, err := http.Get(url)
-    if err != nil {
-        return "", err
-    }
-
-    filename := customName
-    if filename == "" {
-        filename = "download_" + uuid.New().String()
-    }
-
-    filename = filepath.Base(filename)
-    finalPath := filepath.Join(destFolder, filename)
-
-    out, err := os.Create(finalPath)
-    if err != nil {
-        return "", err
-    }
-    defer out.Close()
-
-    _, err = io.Copy(out, resp.Body)
-    return finalPath, err
-}
-
+	resp, err := http.Get(url)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	filename := customName
+	if filename == "" {
+		if cd := resp.Header.Get("Content-Disposition"); cd != "" {
+			_, params, err := mime.ParseMediaType(cd)
+			if err == nil {
+				filename = params["filename"]
+			}
+		}
+		if filename == "" {
+			filename = filepath.Base(resp.Request.URL.Path)
+		}
+		if filename == "" || filename == "." || filename == "/" {
+			filename = "download_" + uuid.New().String()
+		}
+	}
 
 	filename = filepath.Base(filename)
 	finalPath := filepath.Join(destFolder, filename)
@@ -198,7 +193,6 @@ func downloadFile(url string, customName string, destFolder string) (string, err
 	_, err = io.Copy(out, resp.Body)
 	return finalPath, err
 }
-
 func zipFile(sourcePath string, destFolder string) (string, error) {
 	filename := filepath.Base(sourcePath)
 	zipName := fmt.Sprintf("%s_%s.zip", strings.TrimSuffix(filename, filepath.Ext(filename)), uuid.New().String()[:8])