Create providers/mega_provider.py

providers/mega_provider.py

@@ -0,0 +1,78 @@
+import os
+import subprocess
+import tempfile
+
+class MegaProvider:
+    def __init__(self, email=None, password=None):
+        self.email = email
+        self.password = password
+    
+    def validate_credentials(self, email, password):
+        """Validate Mega credentials"""
+        megarc_path = tempfile.mktemp()
+        try:
+            with open(megarc_path, 'w') as f:
+                f.write(f"[Login]\nUsername = {email}\nPassword = {password}\n")
+            
+            result = subprocess.run(
+                ['megals', '--config', megarc_path, '/Root'],
+                capture_output=True,
+                text=True,
+                timeout=10
+            )
+            
+            return result.returncode == 0
+        finally:
+            if os.path.exists(megarc_path):
+                os.remove(megarc_path)
+    
+    def upload(self, file_path, progress_callback=None):
+        """Upload file to Mega"""
+        megarc_path = tempfile.mktemp()
+        try:
+            with open(megarc_path, 'w') as f:
+                f.write(f"[Login]\nUsername = {self.email}\nPassword = {self.password}\n")
+            
+            cmd = [
+                'megaput',
+                '--config', megarc_path,
+                '--path', '/Root',
+                file_path
+            ]
+            
+            process = subprocess.Popen(
+                cmd,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+                bufsize=1
+            )
+            
+            last_progress = 0
+            while True:
+                line = process.stdout.readline()
+                if not line and process.poll() is not None:
+                    break
+                
+                if line and progress_callback:
+                    # Parse progress
+                    import re
+                    match = re.search(r'(\d+)%', line)
+                    if match:
+                        progress = int(match.group(1))
+                        if progress != last_progress:
+                            last_progress = progress
+                            progress_callback(progress)
+            
+            if process.returncode == 0:
+                return {
+                    'provider': 'mega',
+                    'filename': os.path.basename(file_path),
+                    'size': os.path.getsize(file_path)
+                }
+            
+            return None
+            
+        finally:
+            if os.path.exists(megarc_path):
+                os.remove(megarc_path)