Initial commit

app/plugins/torrent_clients/base.py

@@ -0,0 +1,219 @@
+"""
+Classe de base pour les plugins de clients torrent.
+Tous les plugins doivent hériter de cette classe.
+"""
+
+from abc import ABC, abstractmethod
+from typing import Optional, Dict, Any, List
+from dataclasses import dataclass
+
+
+@dataclass
+class TorrentClientConfig:
+    """Configuration d'un client torrent"""
+    host: str
+    port: int = 0  # 0 = pas de port explicite (utiliser le port par défaut du protocole)
+    username: str = ""
+    password: str = ""
+    use_ssl: bool = False
+    path: str = ""  # Chemin optionnel (ex: /qbittorrent)
+    
+    @property
+    def base_url(self) -> str:
+        protocol = "https" if self.use_ssl else "http"
+        
+        # Si le port est 0 ou vide, ne pas l'inclure dans l'URL
+        if self.port and self.port > 0:
+            url = f"{protocol}://{self.host}:{self.port}"
+        else:
+            url = f"{protocol}://{self.host}"
+        
+        # Ajouter le chemin s'il existe
+        if self.path:
+            # S'assurer que le chemin commence par / et ne finit pas par /
+            path = self.path.strip('/')
+            if path:
+                url = f"{url}/{path}"
+        
+        return url
+
+
+@dataclass 
+class TorrentInfo:
+    """Informations sur un torrent"""
+    hash: str
+    name: str
+    size: int
+    progress: float  # 0.0 à 1.0
+    status: str  # downloading, seeding, paused, error, etc.
+    download_speed: int  # bytes/s
+    upload_speed: int  # bytes/s
+    seeds: int
+    peers: int
+    save_path: str
+
+
+class TorrentClientPlugin(ABC):
+    """
+    Classe abstraite pour les plugins de clients torrent.
+    
+    Pour créer un nouveau plugin :
+    1. Créer un fichier dans plugins/torrent_clients/
+    2. Hériter de TorrentClientPlugin
+    3. Implémenter toutes les méthodes abstraites
+    4. Définir PLUGIN_NAME, PLUGIN_DESCRIPTION, PLUGIN_VERSION
+    """
+    
+    # Métadonnées du plugin (à surcharger)
+    PLUGIN_NAME: str = "Base Plugin"
+    PLUGIN_DESCRIPTION: str = "Plugin de base"
+    PLUGIN_VERSION: str = "1.0.0"
+    PLUGIN_AUTHOR: str = "Unknown"
+    
+    # Capacités du plugin
+    SUPPORTS_TORRENT_FILES: bool = True  # Supporte les URLs .torrent en plus des magnets
+    
+    def __init__(self, config: TorrentClientConfig):
+        self.config = config
+        self._connected = False
+    
+    @abstractmethod
+    def connect(self) -> bool:
+        """
+        Établit la connexion avec le client torrent.
+        Retourne True si la connexion réussit.
+        """
+        pass
+    
+    @abstractmethod
+    def disconnect(self) -> None:
+        """Ferme la connexion avec le client torrent."""
+        pass
+    
+    @abstractmethod
+    def is_connected(self) -> bool:
+        """Vérifie si la connexion est active."""
+        pass
+    
+    @abstractmethod
+    def add_torrent_url(self, url: str, save_path: Optional[str] = None, 
+                        category: Optional[str] = None, paused: bool = False) -> bool:
+        """
+        Ajoute un torrent via son URL (magnet ou .torrent).
+        
+        Args:
+            url: URL du torrent (magnet:// ou http://.torrent)
+            save_path: Chemin de sauvegarde (optionnel)
+            category: Catégorie/Label (optionnel)
+            paused: Démarrer en pause (défaut: False)
+            
+        Returns:
+            True si l'ajout réussit
+        """
+        pass
+    
+    @abstractmethod
+    def add_torrent_file(self, file_content: bytes, filename: str,
+                         save_path: Optional[str] = None,
+                         category: Optional[str] = None, paused: bool = False) -> bool:
+        """
+        Ajoute un torrent via son contenu fichier.
+        
+        Args:
+            file_content: Contenu binaire du fichier .torrent
+            filename: Nom du fichier
+            save_path: Chemin de sauvegarde (optionnel)
+            category: Catégorie/Label (optionnel)
+            paused: Démarrer en pause (défaut: False)
+            
+        Returns:
+            True si l'ajout réussit
+        """
+        pass
+    
+    @abstractmethod
+    def get_torrents(self) -> List[TorrentInfo]:
+        """
+        Récupère la liste de tous les torrents.
+        
+        Returns:
+            Liste des torrents
+        """
+        pass
+    
+    @abstractmethod
+    def get_torrent(self, torrent_hash: str) -> Optional[TorrentInfo]:
+        """
+        Récupère les informations d'un torrent spécifique.
+        
+        Args:
+            torrent_hash: Hash du torrent
+            
+        Returns:
+            Informations du torrent ou None si non trouvé
+        """
+        pass
+    
+    @abstractmethod
+    def pause_torrent(self, torrent_hash: str) -> bool:
+        """Met un torrent en pause."""
+        pass
+    
+    @abstractmethod
+    def resume_torrent(self, torrent_hash: str) -> bool:
+        """Reprend un torrent en pause."""
+        pass
+    
+    @abstractmethod
+    def delete_torrent(self, torrent_hash: str, delete_files: bool = False) -> bool:
+        """
+        Supprime un torrent.
+        
+        Args:
+            torrent_hash: Hash du torrent
+            delete_files: Supprimer aussi les fichiers téléchargés
+            
+        Returns:
+            True si la suppression réussit
+        """
+        pass
+    
+    @abstractmethod
+    def get_categories(self) -> List[str]:
+        """Récupère la liste des catégories/labels disponibles."""
+        pass
+    
+    def get_info(self) -> Dict[str, Any]:
+        """Retourne les informations du plugin."""
+        return {
+            "name": self.PLUGIN_NAME,
+            "description": self.PLUGIN_DESCRIPTION,
+            "version": self.PLUGIN_VERSION,
+            "author": self.PLUGIN_AUTHOR,
+            "connected": self.is_connected()
+        }
+    
+    def test_connection(self) -> Dict[str, Any]:
+        """
+        Teste la connexion au client.
+        
+        Returns:
+            {"success": bool, "message": str, "version": str (si connecté)}
+        """
+        try:
+            if self.connect():
+                return {
+                    "success": True,
+                    "message": "Connexion réussie",
+                    "client": self.PLUGIN_NAME
+                }
+            else:
+                return {
+                    "success": False,
+                    "message": "Échec de la connexion"
+                }
+        except Exception as e:
+            return {
+                "success": False,
+                "message": str(e)
+            }