Release v16.0.2: Windows Support, Retry Logic, and CLI

setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="speedster",
-    version="16.0.1", # UPDATED VERSION
+    version="16.0.2", # Major Update: CLI & Windows Support
     packages=find_packages(),
     install_requires=[
         "aiohttp>=3.9.0",
@@ -10,8 +10,13 @@ setup(
         "nest_asyncio",
         "requests"
     ],
+    entry_points={
+        'console_scripts': [
+            'speedster=speedster:cli_main', # Mendaftarkan command 'speedster'
+        ],
+    },
     author="Bl4ckSpaces",
-    description="Ultra-fast, asynchronous, memory-efficient universal downloader (Nitro Edition).",
-    keywords="downloader, aiohttp, async, civitai, huggingface, speedster",
+    description="Universal cross-platform downloader (Windows/Linux/Colab) with connection retry logic.",
+    keywords="downloader, aiohttp, async, civitai, huggingface, speedster, cli",
     python_requires=">=3.8",
 )

speedster/__init__.py

@@ -1,7 +1,9 @@
 import os
+import sys
 import asyncio
 import aiohttp
 import urllib.parse
+import argparse
 from tqdm.asyncio import tqdm
 import nest_asyncio
 
@@ -9,99 +11,134 @@ import nest_asyncio
 nest_asyncio.apply()
 
 class Speedster:
-    def __init__(self, num_threads=16, chunk_size_mb=2):
+    def __init__(self, num_threads=16, chunk_size_mb=2, max_retries=5):
         self.num_threads = num_threads
-        self.chunk_size = chunk_size_mb * 1024 * 1024
-
+        self.chunk_size = chunk_size_mb * 1024 * 1024 
+        self.max_retries = max_retries
+        self.is_windows = os.name == 'nt' # Deteksi Windows
+        
         self.default_headers = {
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
             "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
             "Accept-Language": "en-US,en;q=0.5",
         }
 
+    # --- CROSS-PLATFORM WRITER ---
+    def _write_data(self, fd, data, offset):
+        if self.is_windows:
+            # Windows Fallback: Lseek + Write (Blocking but safe)
+            os.lseek(fd, offset, 0)
+            os.write(fd, data)
+        else:
+            # Linux/Mac/Android: Pwrite (Atomic & Fast)
+            os.pwrite(fd, data, offset)
+
     async def _fetch_chunk(self, session, url, start, end, fd, headers, pbar):
         chunk_headers = headers.copy()
         chunk_headers['Range'] = f'bytes={start}-{end}'
-
-        try:
-            async with session.get(url, headers=chunk_headers, timeout=60) as response:
-                response.raise_for_status()
-                offset = start
-                async for chunk in response.content.iter_chunked(self.chunk_size):
-                    if chunk:
-                        os.pwrite(fd, chunk, offset)
-                        offset += len(chunk)
-                        pbar.update(len(chunk))
-        except Exception as e:
-            # print(f"Retry chunk {start}...")
-            pass
+        
+        # --- RETRY LOGIC (NYAWA CADANGAN) ---
+        for attempt in range(self.max_retries):
+            try:
+                async with session.get(url, headers=chunk_headers, timeout=60) as response:
+                    response.raise_for_status()
+                    offset = start
+                    async for chunk in response.content.iter_chunked(self.chunk_size):
+                        if chunk:
+                            self._write_data(fd, chunk, offset)
+                            offset += len(chunk)
+                            pbar.update(len(chunk))
+                return # Sukses, keluar dari fungsi
+            except Exception as e:
+                if attempt < self.max_retries - 1:
+                    # Tunggu sebentar sebelum coba lagi (Exponential Backoff)
+                    await asyncio.sleep(1 * (attempt + 1))
+                else:
+                    # Nyawa habis
+                    tqdm.write(f"❌ Gagal pada chunk {start}-{end} setelah {self.max_retries} kali coba. Error: {e}")
+                    raise e
 
     async def _download_logic(self, url, dest_path, headers):
         async with aiohttp.ClientSession() as session:
             # 1. Get Metadata
-            async with session.get(url, headers=headers, allow_redirects=True) as response:
-                final_url = response.url
-                total_size = int(response.headers.get('Content-Length', 0))
-
-                # Smart Filename Detection
-                filename = "downloaded_file.bin"
-                if os.path.isdir(dest_path):
-                    if 'Content-Disposition' in response.headers:
-                        try:
-                            cd = response.headers['Content-Disposition']
-                            filename = cd.split('filename=')[1].strip('"')
-                        except:
-                            filename = os.path.basename(urllib.parse.urlparse(str(final_url)).path)
-                    else:
-                        path_filename = os.path.basename(urllib.parse.urlparse(str(final_url)).path)
-                        if path_filename: filename = path_filename
-
-                    dest_path = os.path.join(dest_path, filename)
-
-            if total_size == 0:
-                print(f"❌ Server rejected size request.")
-                return
-
-            print(f"🚀 [Speedster Nitro] Target: {dest_path} | Size: {total_size / (1024**3):.2f} GB")
-
-            # 2. Prepare Disk
-            fd = os.open(dest_path, os.O_RDWR | os.O_CREAT)
-            os.ftruncate(fd, total_size)
-
-            # 3. Create Tasks
-            chunk_size = total_size // self.num_threads
-            chunks = []
-            for i in range(self.num_threads):
-                start = i * chunk_size
-                end = total_size - 1 if i == self.num_threads - 1 else (start + chunk_size - 1)
-                chunks.append((start, end))
-
-            pbar = tqdm(total=total_size, unit='iB', unit_scale=True, unit_divisor=1024, desc="⚡ SPEEDSTER")
-            tasks = [asyncio.create_task(self._fetch_chunk(session, final_url, s, e, fd, headers, pbar)) for s, e in chunks]
-
-            # 4. AWAIT ALL TASKS (Blocking)
-            await asyncio.gather(*tasks)
-
-            pbar.close()
-            os.close(fd)
-            print("✅ Download Selesai!")
-
-    def download(self, url, dest_path="/content", token=None):
+            try:
+                async with session.get(url, headers=headers, allow_redirects=True) as response:
+                    final_url = response.url
+                    total_size = int(response.headers.get('Content-Length', 0))
+                    
+                    # Smart Filename
+                    filename = "downloaded_file.bin"
+                    if os.path.isdir(dest_path):
+                        if 'Content-Disposition' in response.headers:
+                            try:
+                                cd = response.headers['Content-Disposition']
+                                filename = cd.split('filename=')[1].strip('"')
+                            except:
+                                filename = os.path.basename(urllib.parse.urlparse(str(final_url)).path)
+                        else:
+                            path_filename = os.path.basename(urllib.parse.urlparse(str(final_url)).path)
+                            if path_filename: filename = path_filename
+                        
+                        # Clean filename from weird URL params
+                        filename = filename.split('?')[0]
+                        dest_path = os.path.join(dest_path, filename)
+
+                if total_size == 0:
+                    print(f"❌ Server tidak memberikan ukuran file. Tidak bisa multi-thread.")
+                    return
+
+                print(f"🚀 [Speedster Universal] Target: {dest_path}")
+                print(f"📦 Size: {total_size / (1024**3):.2f} GB | Threads: {self.num_threads} | OS: {'Windows' if self.is_windows else 'Linux/Unix'}")
+
+                # 2. Prepare Disk
+                # Mode wb+ untuk binary write update
+                fd = os.open(dest_path, os.O_RDWR | os.O_CREAT | os.O_BINARY if self.is_windows else os.O_RDWR | os.O_CREAT)
+                os.ftruncate(fd, total_size)
+
+                # 3. Create Tasks
+                chunk_size = total_size // self.num_threads
+                chunks = []
+                for i in range(self.num_threads):
+                    start = i * chunk_size
+                    end = total_size - 1 if i == self.num_threads - 1 else (start + chunk_size - 1)
+                    chunks.append((start, end))
+
+                pbar = tqdm(total=total_size, unit='iB', unit_scale=True, unit_divisor=1024, desc="⚡ DOWNLOADING")
+                tasks = [asyncio.create_task(self._fetch_chunk(session, final_url, s, e, fd, headers, pbar)) for s, e in chunks]
+                
+                await asyncio.gather(*tasks)
+                
+                pbar.close()
+                os.close(fd)
+                print("✅ Download Selesai!")
+
+            except Exception as e:
+                print(f"❌ Critical Error: {e}")
+
+    def download(self, url, dest_path=".", token=None):
         headers = self.default_headers.copy()
         if token:
             headers["Authorization"] = f"Bearer {token}"
-
-        # --- FIX UTAMA DI SINI ---
+        
         loop = asyncio.get_event_loop()
         if loop.is_running():
-            # Jika ada loop berjalan (Colab), kita patch dan paksa tunggu
             nest_asyncio.apply()
             loop.run_until_complete(self._download_logic(url, dest_path, headers))
         else:
-            # Jika script python biasa
             asyncio.run(self._download_logic(url, dest_path, headers))
 
-# Global Wrapper
-def download(url, dest_path="/content", token=None, threads=16):
-    engine = Speedster(num_threads=threads)
-    engine.download(url, dest_path, token)
+# --- CLI ENTRY POINT ---
+def cli_main():
+    parser = argparse.ArgumentParser(description="Speedster v16.0.2 - Universal Downloader")
+    parser.add_argument("url", help="URL file/model yang ingin didownload")
+    parser.add_argument("--out", "-o", default=".", help="Folder tujuan (Default: folder saat ini)")
+    parser.add_argument("--token", "-t", help="Bearer token (Khusus Civitai/HuggingFace Private)")
+    parser.add_argument("--threads", type=int, default=16, help="Jumlah koneksi simultan (Default: 16)")
+    
+    args = parser.parse_args()
+    
+    engine = Speedster(num_threads=args.threads)
+    engine.download(args.url, args.out, args.token)
+
+if __name__ == "__main__":
+    cli_main()