"""Mail plugin — IMAP/SMTP, threading, templates, rules, PGP, delegates.""" from __future__ import annotations import asyncio import logging from datetime import UTC from typing import Any from app.plugins.base import BasePlugin from app.plugins.manifest import ( FrontendDetailTab, FrontendMenuItem, FrontendPageRoute, FrontendSettingsPage, PluginManifest, PluginRouteDef, ) logger = logging.getLogger(__name__) async def _mail_restore_handler( db, entity, action: str, snapshot: dict, context: dict, ) -> dict: """Special restore handler for Mail entities (moved from core, P0-7 fix). Mail restore has IMAP semantics: - delete: move back from trash to original folder (if folder still exists) - update: revert metadata fields - create: soft-delete (undo send only works for drafts) """ import uuid from datetime import datetime from sqlalchemy import select user_id = context.get("user_id") tenant_id = context.get("tenant_id") if action == "delete": if entity is None: raise ValueError("Mail entity not found for restore") entity.deleted_at = None if user_id: entity.updated_by = user_id if hasattr(entity, "updated_by") else None original_folder_id = snapshot.get("folder_id") if original_folder_id and hasattr(entity, "folder_id"): try: folder_uuid = uuid.UUID(str(original_folder_id)) from app.plugins.builtins.mail.models import MailFolder folder_q = select(MailFolder).where( MailFolder.id == folder_uuid, MailFolder.tenant_id == tenant_id, MailFolder.deleted_at.is_(None), ) folder_result = await db.execute(folder_q) folder = folder_result.scalar_one_or_none() if folder: entity.folder_id = folder_uuid else: logger.warning( "Original mail folder %s no longer exists, " "restoring mail without folder assignment", original_folder_id, ) except (ValueError, Exception) as e: logger.warning("Failed to restore mail folder: %s", e) await db.flush() return {"id": str(entity.id), "restored": True, "entity_type": "mail"} elif action == "update": if entity is None: raise ValueError("Mail entity not found for restore") from app.core.restore_registry import _DEFAULT_EXCLUDED excluded = _DEFAULT_EXCLUDED | { "message_id", "rfc822_size", "raw_path", "account_id", "folder_id", } for key, value in snapshot.items(): if hasattr(entity, key) and key not in excluded: setattr(entity, key, value) await db.flush() return {"id": str(entity.id), "restored": True, "entity_type": "mail"} elif action == "create": if entity is None: raise ValueError("Mail entity not found for restore") entity.deleted_at = datetime.now(UTC) await db.flush() return {"id": str(entity.id), "restored": True, "entity_type": "mail", "note": "soft-deleted (undo create)"} raise ValueError(f"Unsupported action for mail restore: {action}") async def _auto_sync_loop() -> None: """Background loop: process pending sync queue, then sync all active mail accounts every 5 minutes.""" from app.core.db import get_session_factory from app.plugins.builtins.mail.services import auto_sync_all_accounts, process_sync_queue while True: try: factory = get_session_factory() async with factory() as db: await process_sync_queue(db) await db.commit() except Exception as exc: logger.warning("process_sync_queue error: %s", exc) try: await auto_sync_all_accounts() except Exception as exc: logger.warning("auto_sync error: %s", exc) await asyncio.sleep(60) class MailPlugin(BasePlugin): """Mail plugin for email management: IMAP sync, SMTP send, threading, rules, PGP.""" _auto_sync_task: asyncio.Task | None = None manifest = PluginManifest( name="mail", version="1.3.0", display_name="Mail", description=( "Email management: IMAP sync, SMTP send, threading, " "templates, rules, vacation, PGP, delegates, labels." ), dependencies=[], routes=[ PluginRouteDef( path="/api/v1/mail", module="app.plugins.builtins.mail.routes", router_attr="router", ), ], events=[], migrations=["0001_initial.sql", "0006_flag_type.sql", "0007_sync_queue.sql", "0008_sync_queue_deleted_at.sql", "0009_remove_mail_soft_delete.sql", "0010_add_deleted_at.sql"], permissions=["mail:read", "mail:send", "mail:config", "mail:share", "mail:write", "mail:delete"], menu_items=[ FrontendMenuItem(label_key='nav.mail', label='E-Mail', path='/mail', icon='Mail', order=30), ], page_routes=[ FrontendPageRoute(path='/mail', component='@/pages/Mail', protected=True), FrontendPageRoute(path='/mail/settings', component='@/pages/MailSettings', protected=True), ], settings_pages=[ FrontendSettingsPage(path='mail', label_key='settings.mail', label='Mail', component='@/pages/MailSettings', icon='Mail', order=50), ], detail_tabs=[ FrontendDetailTab(entity_type='contact', label_key='tabs.email', label='E-Mails', component='@/components/contact/ContactMailTab', icon='Mail', order=20, permission='mail:read'), ], author="LeoCRM Team", min_app_version="1.0.0", hooks=["mail.before_send", "mail.after_send"], contract_version="1.0.0", ) async def on_activate( self, db, service_container, event_bus ) -> None: """Activate plugin: register events, restore, history + start auto-sync.""" await super().on_activate(db, service_container, event_bus) # Register restore config for Mail entities (P0-7 fix) from app.core.restore_registry import RestoreConfig, get_restore_registry from app.plugins.builtins.mail.models import Mail get_restore_registry().register(RestoreConfig( entity_type="mail", model_class=Mail, restore_permission="mail:write", excluded_fields=frozenset({"message_id", "rfc822_size", "raw_path", "account_id", "folder_id"}), special_handler=_mail_restore_handler, )) # Register history hooks for Mail entities (P0-8 fix) from app.core.history_hooks import register_history_hooks from app.core.hooks import get_hook_registry register_history_hooks( get_hook_registry(), "mail", "mail.after_create", "mail.after_update", "mail.after_delete", owner_tag="mail", ) if self._auto_sync_task is None or self._auto_sync_task.done(): self._auto_sync_task = asyncio.create_task(_auto_sync_loop()) logger.info("Mail plugin: auto-sync background task started") def get_notification_types(self) -> list[dict[str, Any]]: """Return the notification types this mail plugin registers.""" return [ {"type_key": "mail_new", "category": "mail", "label": "Neue E-Mail empfangen", "description": "Benachrichtigung bei neuen E-Mails", "is_enabled_by_default": True}, {"type_key": "mail_error", "category": "mail", "label": "IMAP-Verbindungsfehler", "description": "Fehler bei der Verbindung zum Mailserver", "is_enabled_by_default": True}, {"type_key": "mail_auth", "category": "mail", "label": "IMAP-Login-Fehler", "description": "Anmeldung am Mailserver fehlgeschlagen", "is_enabled_by_default": True}, {"type_key": "mail_quota", "category": "mail", "label": "Postfach fast voll", "description": "Warnung bei hohem Postfach-Füllstand", "is_enabled_by_default": True}, {"type_key": "mail_sync_error", "category": "mail", "label": "Sync-Fehler", "description": "Synchronisierung fehlgeschlagen", "is_enabled_by_default": True}, {"type_key": "mail_sent", "category": "mail", "label": "E-Mail gesendet", "description": "Bestätigung beim Senden einer E-Mail", "is_enabled_by_default": False}, {"type_key": "mail_send_error", "category": "mail", "label": "SMTP-Sendefehler", "description": "E-Mail konnte nicht gesendet werden", "is_enabled_by_default": True}, {"type_key": "mail_draft", "category": "mail", "label": "Entwurf gespeichert", "description": "Bestätigung beim Speichern eines Entwurfs", "is_enabled_by_default": False}, {"type_key": "mail_account", "category": "mail", "label": "Account deaktiviert", "description": "Warnung bei deaktiviertem Mail-Account", "is_enabled_by_default": True}, {"type_key": "mail_folder", "category": "mail", "label": "Ordner erstellt/gelöscht", "description": "Bestätigung bei Ordner-Operationen", "is_enabled_by_default": False}, ] async def on_deactivate( self, db, service_container, event_bus ) -> None: """Deactivate plugin: stop auto-sync task + unregister events.""" # Contract abmelden from app.plugins.builtins.contracts import get_contract_registry get_contract_registry().unregister(self.manifest.name) if self._auto_sync_task is not None and not self._auto_sync_task.done(): self._auto_sync_task.cancel() try: await self._auto_sync_task except asyncio.CancelledError: pass self._auto_sync_task = None logger.info("Mail plugin: auto-sync background task stopped") # Unregister history hooks (free functions, not bound methods) from app.core.hooks import get_hook_registry get_hook_registry().unregister_actions_by_owner("mail.after_create", "mail") get_hook_registry().unregister_actions_by_owner("mail.after_update", "mail") get_hook_registry().unregister_actions_by_owner("mail.after_delete", "mail") # Unregister restore config for Mail entities from app.core.restore_registry import get_restore_registry get_restore_registry().unregister("mail") await super().on_deactivate(db, service_container, event_bus)