Add Somewhat working logic

python_rpc/http_downloader.py

@@ -1,48 +1,162 @@
 import aria2p
+from typing import Union, List
+import logging
+import os
+from pathlib import Path
+from aria2p import API, Client, Download
+import requests
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class HttpDownloader:
     def __init__(self):
-        self.download = None
-        self.aria2 = aria2p.API(
-            aria2p.Client(
-                host="http://localhost",
-                port=6800,
-                secret=""
-            )
-        )
+        self.downloads = []  # vom păstra toate download-urile active
+        self.aria2 = API(Client(host="http://localhost", port=6800))
+        self.download = None  # pentru compatibilitate cu codul vechi
 
-    def start_download(self, url: str, save_path: str, header: str, out: str = None):
-        if self.download:
-            self.aria2.resume([self.download])
-        else:
-            downloads = self.aria2.add(url, options={"header": header, "dir": save_path, "out": out})
+    def unlock_alldebrid_link(self, link: str) -> str:
+        """Deblochează un link AllDebrid și returnează link-ul real de descărcare."""
+        api_key = os.getenv('ALLDEBRID_API_KEY')
+        if not api_key:
+            logger.error("AllDebrid API key nu a fost găsită în variabilele de mediu")
+            return link
+
+        try:
+            response = requests.post(
+                "https://api.alldebrid.com/v4/link/unlock",
+                params={
+                    "agent": "hydra",
+                    "apikey": api_key,
+                    "link": link
+                }
+            )
+            data = response.json()
             
-            self.download = downloads[0]
+            if data.get("status") == "success":
+                return data["data"]["link"]
+            else:
+                logger.error(f"Eroare la deblocarea link-ului AllDebrid: {data.get('error', {}).get('message', 'Unknown error')}")
+                return link
+        except Exception as e:
+            logger.error(f"Eroare la apelul API AllDebrid: {str(e)}")
+            return link
+
+    def start_download(self, url: Union[str, List[str]], save_path: str, header: str = None, out: str = None):
+        logger.info(f"Starting download with URL: {url}, save_path: {save_path}, header: {header}, out: {out}")
+        
+        # Pentru AllDebrid care returnează un link per fișier
+        if isinstance(url, list):
+            logger.info(f"Multiple URLs detected: {len(url)} files to download")
+            self.downloads = []
+            
+            # Deblocăm toate link-urile AllDebrid
+            unlocked_urls = []
+            for single_url in url:
+                logger.info(f"Unlocking AllDebrid URL: {single_url}")
+                unlocked_url = self.unlock_alldebrid_link(single_url)
+                if unlocked_url:
+                    unlocked_urls.append(unlocked_url)
+                    logger.info(f"URL deblocat cu succes: {unlocked_url}")
+            
+            # Descărcăm folosind link-urile deblocate
+            for unlocked_url in unlocked_urls:
+                logger.info(f"Adding download for unlocked URL: {unlocked_url}")
+                options = {
+                    "dir": save_path
+                }
+                if header:
+                    if isinstance(header, list):
+                        options["header"] = header
+                    else:
+                        options["header"] = [header]
+                
+                try:
+                    download = self.aria2.add_uris([unlocked_url], options=options)
+                    logger.info(f"Download added successfully: {download.gid}")
+                    self.downloads.append(download)
+                except Exception as e:
+                    logger.error(f"Error adding download for URL {unlocked_url}: {str(e)}")
+            
+            if self.downloads:
+                self.download = self.downloads[0]  # păstrăm primul pentru referință
+            else:
+                logger.error("No downloads were successfully added!")
+                
+        # Pentru RealDebrid/alte servicii care returnează un singur link pentru tot
+        else:
+            logger.info(f"Single URL download: {url}")
+            options = {
+                "dir": save_path
+            }
+            if header:
+                if isinstance(header, list):
+                    options["header"] = header
+                else:
+                    options["header"] = [header]
+            if out:
+                options["out"] = out
+                
+            try:
+                download = self.aria2.add_uris([url], options=options)
+                self.download = download
+                self.downloads = [self.download]
+                logger.info(f"Single download added successfully: {self.download.gid}")
+            except Exception as e:
+                logger.error(f"Error adding single download: {str(e)}")
     
     def pause_download(self):
-        if self.download:
-            self.aria2.pause([self.download])
+        try:
+            for download in self.downloads:
+                download.pause()
+        except Exception as e:
+            logger.error(f"Error pausing downloads: {str(e)}")
     
     def cancel_download(self):
-        if self.download:
-            self.aria2.remove([self.download])
-            self.download = None
+        try:
+            for download in self.downloads:
+                download.remove()
+        except Exception as e:
+            logger.error(f"Error canceling downloads: {str(e)}")
 
     def get_download_status(self):
-        if self.download == None:
+        try:
+            if not self.downloads:
+                return None
+
+            total_size = 0
+            downloaded = 0
+            download_speed = 0
+            active_downloads = []
+
+            for download in self.downloads:
+                try:
+                    download.update()
+                    if download.is_active:
+                        active_downloads.append(download)
+                        total_size += download.total_length
+                        downloaded += download.completed_length
+                        download_speed += download.download_speed
+                except Exception as e:
+                    logger.error(f"Error updating download status for {download.gid}: {str(e)}")
+
+            if not active_downloads:
+                return None
+
+            # Folosim primul download pentru numele folderului
+            folder_path = os.path.dirname(active_downloads[0].files[0].path)
+            folder_name = os.path.basename(folder_path)
+
+            return {
+                "progress": downloaded / total_size if total_size > 0 else 0,
+                "numPeers": 0,  # nu este relevant pentru HTTP
+                "numSeeds": 0,  # nu este relevant pentru HTTP
+                "downloadSpeed": download_speed,
+                "bytesDownloaded": downloaded,
+                "fileSize": total_size,
+                "folderName": folder_name,
+                "status": "downloading"
+            }
+        except Exception as e:
+            logger.error(f"Error getting download status: {str(e)}")
             return None
-
-        download = self.aria2.get_download(self.download.gid)
-
-        response = {
-            'folderName': download.name,
-            'fileSize': download.total_length,
-            'progress': download.completed_length / download.total_length if download.total_length else 0,
-            'downloadSpeed': download.download_speed,
-            'numPeers': 0,
-            'numSeeds': 0,
-            'status': download.status,
-            'bytesDownloaded': download.completed_length,
-        }
-
-        return response