Update app.py

app.py

@@ -1,27 +1,39 @@
 import logging
 import ssl
+import smtplib
+from email.mime.text import MIMEText
+from flask import Flask, request, jsonify
 from aiosmtpd.controller import Controller
 from aiosmtpd.handlers import Sink
 from aiosmtpd.smtp import SMTP, AuthResult, LoginPassword
-from email.policy import default
-from typing import Optional
+from dotenv import load_dotenv
+import os
+
+# Load environment variables
+load_dotenv()
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("smtp_server")
+logger = logging.getLogger("app")
 
 # SMTP Server Configuration
-SMTP_HOST = "0.0.0.0"  # Bind to all interfaces
-SMTP_PORT = 587         # Standard SMTP submission port
-TLS_CERTFILE = "cert.pem"  # Path to your SSL certificate
-TLS_KEYFILE = "key.pem"    # Path to your SSL private key
+SMTP_HOST = "0.0.0.0"
+SMTP_PORT = 587
+SMTP_USER = os.getenv("SMTP_USER", "noreply@yourdomain.com")  # Use a valid domain
+SMTP_PASSWORD = os.getenv("SMTP_PASSWORD", "password")
+
+# WhatsApp webhook verification token
+WHATSAPP_WEBHOOK_TOKEN = os.getenv("WHATSAPP_WEBHOOK_TOKEN", "your_webhook_token")
 
-# Authentication Credentials (for demo purposes; use a secure vault in production)
+# Authentication Credentials
 VALID_USERS = {
-    "user1": "password1",
-    "user2": "password2",
+    SMTP_USER: SMTP_PASSWORD,
 }
 
+# Flask app for WhatsApp bot
+app = Flask(__name__)
+
+# SMTP Server Handler
 class AuthenticatedSMTPHandler(Sink):
     async def handle_AUTH(self, server: SMTP, args: str, mechanism: str) -> Optional[AuthResult]:
         if mechanism.upper() != "LOGIN":
@@ -40,39 +52,99 @@ class AuthenticatedSMTPHandler(Sink):
             return AuthResult(success=False, message="Authentication failed")
 
     async def handle_DATA(self, server: SMTP, session, envelope):
-        # Log the received email
         logger.info(f"Received email from: {envelope.mail_from}")
         logger.info(f"Recipients: {envelope.rcpt_tos}")
         logger.info(f"Message data:\n{envelope.content.decode('utf-8')}")
         return "250 Message accepted for delivery"
 
+# Start SMTP Server
 def start_smtp_server():
-    # Create SSL context for TLS
+    # Enable SSL/TLS
     ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
-    ssl_context.load_cert_chain(TLS_CERTFILE, TLS_KEYFILE)
+    ssl_context.load_cert_chain("cert.pem", "key.pem")  # Load your certificate and key
 
-    # Create the SMTP handler
     handler = AuthenticatedSMTPHandler()
-
-    # Start the SMTP server with TLS and authentication
     controller = Controller(
         handler,
         hostname=SMTP_HOST,
         port=SMTP_PORT,
         enable_SMTPUTF8=True,
-        ssl_context=ssl_context,
+        ssl_context=ssl_context,  # Enable SSL/TLS
         auth_required=True,
     )
 
-    logger.info(f"Starting SMTP server on {SMTP_HOST}:{SMTP_PORT} with TLS and authentication...")
+    logger.info(f"Starting SMTP server on {SMTP_HOST}:{SMTP_PORT} with SSL/TLS...")
     controller.start()
+    return controller
+
+# WhatsApp Webhook
+@app.route("/webhook", methods=["GET", "POST"])
+def webhook():
+    if request.method == "GET":
+        # Verify webhook
+        mode = request.args.get("hub.mode")
+        token = request.args.get("hub.verify_token")
+        challenge = request.args.get("hub.challenge")
+
+        if mode == "subscribe" and token == WHATSAPP_WEBHOOK_TOKEN:
+            logger.info("Webhook verified successfully")
+            return challenge, 200
+        else:
+            logger.error("Webhook verification failed")
+            return "Verification failed", 403
+
+    elif request.method == "POST":
+        # Handle incoming messages
+        data = request.json
+        logger.info(f"Incoming webhook data: {data}")
+
+        try:
+            # Extract message details
+            message = data["entry"][0]["changes"][0]["value"]["messages"][0]
+            sender_number = message["from"]
+            message_body = message["text"]["body"]
+
+            # Parse the message (format: "send email to <recipient> subject <subject> body <body>")
+            parts = message_body.split("subject")
+            recipient_part = parts[0].replace("send email to", "").strip()
+            subject_part = parts[1].split("body")[0].strip()
+            body_part = parts[1].split("body")[1].strip()
+
+            recipient = recipient_part
+            subject = subject_part
+            body = body_part
+
+            # Send email
+            send_email(recipient, subject, body)
+            response_msg = f"Email sent to {recipient} with subject '{subject}'."
+        except Exception as e:
+            logger.error(f"Error processing message: {e}")
+            response_msg = "Invalid format. Use: 'send email to <recipient> subject <subject> body <body>'."
+
+        # Send response back to WhatsApp
+        return jsonify({"status": "success", "message": response_msg}), 200
+
+# Send Email Function
+def send_email(recipient, subject, body):
+    msg = MIMEText(body)
+    msg['Subject'] = subject
+    msg['From'] = SMTP_USER  # Use a valid "From" address
+    msg['To'] = recipient
+
+    with smtplib.SMTP(SMTP_HOST, SMTP_PORT) as server:
+        server.starttls()  # Enable TLS
+        server.login(SMTP_USER, SMTP_PASSWORD)
+        server.sendmail(SMTP_USER, [recipient], msg.as_string())
+    logger.info(f"Email sent to {recipient}")
+
+# Main Function
+if __name__ == "__main__":
+    # Start SMTP server
+    smtp_controller = start_smtp_server()
+
+    # Start Flask app for WhatsApp bot
     try:
-        # Keep the server running
-        while True:
-            pass
+        app.run(host="0.0.0.0", port=5000)
     except KeyboardInterrupt:
         logger.info("Shutting down SMTP server...")
-        controller.stop()
-
-if __name__ == "__main__":
-    start_smtp_server()
+        smtp_controller.stop()