fix: fix email pool auto

core/email_generator.py

@@ -17,6 +17,7 @@ from dataclasses import dataclass, field
 from pathlib import Path
 
 from dotenv import load_dotenv
+import requests
 
 
 # Name pools for generating realistic emails
@@ -194,11 +195,11 @@ class EmailGenerator:
         - email@domain.com (uses main IMAP password)
         - email@domain.com:password (uses specific password for this email)
         """
-        if not self.config.email_pool:
-            raise ValueError("Email pool is empty")
-        
         if self._pool_index >= len(self.config.email_pool):
-            raise ValueError("Email pool exhausted - no more emails available")
+            if not self._maybe_add_mailapi_address():
+                if not self.config.email_pool:
+                    raise ValueError("Email pool is empty")
+                raise ValueError("Email pool exhausted - no more emails available")
         
         entry = self.config.email_pool[self._pool_index]
         self._pool_index += 1
@@ -225,6 +226,58 @@ class EmailGenerator:
             display_name=name,
             imap_password=imap_password
         )
+
+    def _maybe_add_mailapi_address(self) -> bool:
+        backend = os.environ.get('EMAIL_BACKEND', 'imap').strip().lower()
+        auto_create = os.environ.get('MAIL_API_AUTO_CREATE', '1').strip().lower()
+        if backend != 'mailapi' or auto_create in ('0', 'false', 'no'):
+            return False
+        address = self._create_mailapi_address()
+        if not address:
+            return False
+        self.config.email_pool.append(address)
+        return True
+
+    def _create_mailapi_address(self) -> Optional[str]:
+        base_url = os.environ.get('MAIL_API_BASE_URL', '').strip()
+        admin_pwd = os.environ.get('MAIL_API_ADMIN_PWD', '').strip()
+        domain = os.environ.get('MAIL_API_DOMAIN', '').strip()
+        if not domain:
+            domain = os.environ.get('EMAIL_DOMAIN', '').strip()
+        timeout = int(os.environ.get('MAIL_API_TIMEOUT', '15'))
+
+        if not base_url or not admin_pwd or not domain:
+            return None
+
+        name = self._generate_mailapi_name()
+        url = base_url.rstrip('/') + '/admin/new_address'
+        headers = {
+            'x-admin-auth': admin_pwd,
+            'Content-Type': 'application/json'
+        }
+        payload = {
+            'enablePrefix': True,
+            'name': name,
+            'domain': domain
+        }
+
+        try:
+            resp = requests.post(url, json=payload, headers=headers, timeout=timeout)
+            if resp.status_code != 200:
+                return None
+            data = resp.json() if resp.content else {}
+            address = data.get('address')
+            if not address:
+                address = f"{name}@{domain}"
+            return address
+        except Exception:
+            return None
+
+    def _generate_mailapi_name(self) -> str:
+        letters1 = ''.join(random.choices(string.ascii_lowercase, k=5))
+        numbers = ''.join(random.choices(string.digits, k=random.randint(1, 3)))
+        letters2 = ''.join(random.choices(string.ascii_lowercase, k=random.randint(1, 3)))
+        return letters1 + numbers + letters2
     
     def _generate_random_name(self) -> str:
         """Generate a random realistic name"""