Merge branch 'main' of https://github.com/KeksPirates/SoftwareManager

Add bzzhr implementation and improve file preallocation on Windows

src/data/sources/steamrip.py

@@ -113,7 +113,7 @@ class SteamripScraper:
 
         try:
             consoleLog(f"Found download link: {best}")
-            if "buzzheavier" in best:
+            if "buzzheavier" or "bzzhr" in best:
                 link = scrape_buzzheavier(best)
                 return link
             elif "gofile" in best:

src/interface/dialogs/contextmenu.py

@@ -12,7 +12,6 @@ import threading
 import webbrowser
 import subprocess
 import platform
-import time
 import os
 
 

src/network/direct_download/__init__.py

@@ -20,12 +20,14 @@ def add_direct_download(url: str, title: str, dl_path: Optional[str] = None, hea
             consoleLog(f"Download already active: {title}")
             return
 
-    filename = (
+    raw_filename = (
         detect_filename_from_headers(url, DirectDownloadHandle.USER_AGENT)
         or extract_filename_from_url(url)
         or sanitize_filename(title) + ".zip"
     )
 
+    filename = sanitize_filename(raw_filename)
+
     handle = DirectDownloadHandle(url, filename, dl_path, headers, single_threaded)
     with state.downloads_lock:
         state.active_downloads[url] = handle

src/network/direct_download/handle.py

@@ -10,6 +10,7 @@ import requests
 import hashlib
 import json
 import time
+import sys
 import os
 
 
@@ -231,8 +232,24 @@ class DirectDownloadHandle:
     def _preallocate_file(self, total_size: int):
         if os.path.exists(self._file_path) and os.path.getsize(self._file_path) == total_size:
             return
+
         with open(self._file_path, "wb") as f:
-            f.truncate(total_size)
+            if sys.platform == "win32":
+                import msvcrt
+                import ctypes
+                from ctypes.wintypes import DWORD
+
+                handle = msvcrt.get_osfhandle(f.fileno())
+                kernel32 = ctypes.windll.kernel32
+
+                bytes_returned = DWORD()
+                kernel32.DeviceIoControl(
+                    handle, 590020, None, 0, None, 0, ctypes.byref(bytes_returned), None
+                )
+                kernel32.SetFilePointerEx(handle, ctypes.c_int64(total_size), None, 0)
+                kernel32.SetEndOfFile(handle)
+            else:
+                f.truncate(total_size)
 
     def _compute_chunks(self, total_size: int) -> list[ChunkSpec]:
         num_chunks = min(self.NUM_THREADS, max(1, total_size // self.MIN_CHUNK_SIZE))

src/network/direct_download/utils.py

@@ -34,7 +34,7 @@ def detect_filename_from_headers(url: str, user_agent: str) -> Optional[str]:
         if "filename=" in cd:
             parts = cd.split("filename=")
             if len(parts) > 1:
-                fname = parts[-1].strip().strip('"').strip("'")
+                fname = parts[1].split(";")[0].strip().strip('"').strip("'")
                 if fname:
                     return fname
     except Exception as e:

src/network/libtorrent_misc.py

@@ -4,7 +4,6 @@ from PySide6.QtCore import QObject, QTimer
 from utils.data.state import state
 from plyer import notification
 import libtorrent as lt
-import time
 import os