""" Service d'envoi d'emails pour les notifications de licence """ import os import smtplib from email.mime.text import MIMEText from email.mime.multipart import MIMEMultipart from typing import Optional import logging logger = logging.getLogger(__name__) class EmailService: """Service d'envoi d'emails via SMTP""" def __init__(self): """Initialise le service email avec les variables d'environnement""" self.smtp_host = os.getenv('SMTP_HOST', 'smtp.gmail.com') self.smtp_port = int(os.getenv('SMTP_PORT', '587')) self.smtp_use_ssl = os.getenv('SMTP_USE_SSL', 'False').lower() == 'true' self.smtp_user = os.getenv('SMTP_USER', '') self.smtp_password = os.getenv('SMTP_PASSWORD', '') self.from_email = os.getenv('FROM_EMAIL', self.smtp_user) self.from_name = os.getenv('FROM_NAME', 'Web Sentinel') # Mode debug : si pas de config SMTP, on logue seulement self.debug_mode = not (self.smtp_user and self.smtp_password) if self.debug_mode: logger.warning("⚠️ Mode email DEBUG activé (pas de configuration SMTP)") def send_license_email( self, to_email: str, api_key: str, tier: str, customer_name: Optional[str] = None ) -> bool: """ Envoie l'email de bienvenue avec la clé API Args: to_email: Email du destinataire api_key: Clé API générée tier: Niveau d'abonnement customer_name: Nom du client (optionnel) Returns: True si envoi réussi, False sinon """ subject = f"🎉 Bienvenue sur Web Sentinel {tier.upper()} - Votre clé API" # Corps de l'email en HTML html_body = self._generate_license_email_html( api_key=api_key, tier=tier, customer_name=customer_name or to_email.split('@')[0] ) # Corps de l'email en texte brut (fallback) text_body = self._generate_license_email_text( api_key=api_key, tier=tier, customer_name=customer_name or to_email.split('@')[0] ) return self._send_email( to_email=to_email, subject=subject, html_body=html_body, text_body=text_body ) def _generate_license_email_html( self, api_key: str, tier: str, customer_name: str ) -> str: """Génère le corps HTML de l'email de licence""" tier_features = { 'STARTER': [ '✅ 10 domaines maximum', '✅ Scans de sécurité complets', '✅ Rapports HTML/JSON', '✅ Support email' ], 'PRO': [ '✅ 50 domaines maximum', '✅ Scans de sécurité avancés', '✅ Tests invasifs autorisés', '✅ Rapports détaillés avec historique', '✅ API REST complète', '✅ Support prioritaire' ], 'ENTERPRISE': [ '✅ Domaines illimités', '✅ Scans de sécurité avancés', '✅ Tests invasifs autorisés', '✅ Multi-utilisateurs (10 utilisateurs)', '✅ API REST complète', '✅ Intégration CI/CD', '✅ Support 24/7 dédié', '✅ Fonctionnalités personnalisées' ] } features = tier_features.get(tier.upper(), tier_features['STARTER']) features_html = ''.join([f'
  • {f}
    • ' for f in features]) return f"""

    🎉 Bienvenue sur Web Sentinel !

    Votre abonnement {tier.upper()} est maintenant actif

    Bonjour {customer_name},

    Merci d'avoir souscrit à Web Sentinel {tier.upper()} ! Votre compte est maintenant actif et prêt à l'emploi.

    🔑 Votre clé API

    {api_key}

    ⚠️ Gardez cette clé secrète et ne la partagez jamais

    🚀 Démarrage rapide

    Installation :

    pip install web-sentinel

    Utilisation :

    web-sentinel scan example.com --api-key {api_key[:20]}...

    Ou configurez la clé globalement :

    export WEB_SENTINEL_API_KEY={api_key[:20]}...

    ✨ Vos fonctionnalités {tier.upper()}

    💬 Besoin d'aide ?

    Notre équipe support est disponible :
    📧 Email : support@web-sentinel.com
    📚 Documentation : docs.web-sentinel.com

    Merci de votre confiance,
    L'équipe Web Sentinel

    © 2025 Web Sentinel - Tous droits réservés
    Cet email contient des informations confidentielles

    """ def _generate_license_email_text( self, api_key: str, tier: str, customer_name: str ) -> str: """Génère le corps texte brut de l'email de licence""" return f""" 🎉 Bienvenue sur Web Sentinel {tier.upper()} ! Bonjour {customer_name}, Merci d'avoir souscrit à Web Sentinel {tier.upper()} ! Votre compte est maintenant actif. 🔑 VOTRE CLÉ API : {api_key} ⚠️ Gardez cette clé secrète et ne la partagez jamais. 🚀 DÉMARRAGE RAPIDE : Installation : pip install web-sentinel Utilisation : web-sentinel scan example.com --api-key {api_key} Ou configurez la clé globalement : export WEB_SENTINEL_API_KEY={api_key} 💬 BESOIN D'AIDE ? - Email : support@web-sentinel.com - Documentation : https://docs.web-sentinel.com Merci de votre confiance, L'équipe Web Sentinel --- © 2025 Web Sentinel - Tous droits réservés Cet email contient des informations confidentielles """ def _send_email( self, to_email: str, subject: str, html_body: str, text_body: str ) -> bool: """ Envoie un email via SMTP Args: to_email: Email du destinataire subject: Sujet de l'email html_body: Corps HTML text_body: Corps texte brut Returns: True si envoi réussi, False sinon """ # Mode debug : on logue seulement if self.debug_mode: logger.info(f""" ╔══════════════════════════════════════════════════════════════════ ║ 📧 EMAIL DEBUG MODE ╠══════════════════════════════════════════════════════════════════ ║ To: {to_email} ║ Subject: {subject} ╠══════════════════════════════════════════════════════════════════ {text_body} ╚══════════════════════════════════════════════════════════════════ """) return True # Mode production : envoi réel try: msg = MIMEMultipart('alternative') msg['Subject'] = subject msg['From'] = f"{self.from_name} <{self.from_email}>" msg['To'] = to_email # Ajouter les deux versions part1 = MIMEText(text_body, 'plain', 'utf-8') part2 = MIMEText(html_body, 'html', 'utf-8') msg.attach(part1) msg.attach(part2) # Connexion SMTP (SSL ou TLS selon configuration) if self.smtp_use_ssl: # Port 465 : SSL direct with smtplib.SMTP_SSL(self.smtp_host, self.smtp_port, timeout=30) as server: server.login(self.smtp_user, self.smtp_password) server.send_message(msg) else: # Port 587 : TLS (STARTTLS) with smtplib.SMTP(self.smtp_host, self.smtp_port, timeout=30) as server: server.starttls() server.login(self.smtp_user, self.smtp_password) server.send_message(msg) logger.info(f"✅ Email envoyé à {to_email}") return True except Exception as e: logger.error(f"❌ Erreur envoi email à {to_email}: {e}") return False def send_subscription_cancelled_email( self, to_email: str, tier: str, customer_name: Optional[str] = None ) -> bool: """ Envoie un email de confirmation d'annulation Args: to_email: Email du destinataire tier: Niveau d'abonnement customer_name: Nom du client (optionnel) Returns: True si envoi réussi, False sinon """ subject = f"Confirmation d'annulation - Web Sentinel {tier.upper()}" html_body = f"""

    Abonnement annulé

    Bonjour {customer_name or to_email.split('@')[0]},

    Votre abonnement Web Sentinel {tier.upper()} a été annulé avec succès.

    Vous conservez l'accès à vos fonctionnalités jusqu'à la fin de votre période de facturation en cours.

    Nous espérons vous revoir bientôt !

    L'équipe Web Sentinel

    """ text_body = f""" Abonnement annulé Bonjour {customer_name or to_email.split('@')[0]}, Votre abonnement Web Sentinel {tier.upper()} a été annulé avec succès. Vous conservez l'accès à vos fonctionnalités jusqu'à la fin de votre période de facturation en cours. Nous espérons vous revoir bientôt ! L'équipe Web Sentinel """ return self._send_email(to_email, subject, html_body, text_body)