import asyncio
from aiosmtpd.controller import Controller
import logging

# Configure logging
logging.basicConfig(level=logging.INFO, filename="./shared/emails.log", filemode="a",
                    format="%(asctime)s - %(message)s")


def log_email(email_content):
    logging.info("Received email:\n%s", email_content)


class EmailHandler:
    async def handle_DATA(self, server, session, envelope):
        try:
            # Decode email content and log it
            email_content = envelope.content.decode('utf-8')
            log_email(email_content)
            print(f"Email received:\n{email_content}")
        except Exception as e:
            print(f"Error processing email: {e}")
        return '250 Message accepted for delivery'


async def main():
    handler = EmailHandler()
    controller = Controller(handler, hostname="0.0.0.0", port=25)

    print("Starting SMTP server on 0.0.0.0:25")
    controller.start()

    try:
        await asyncio.Event().wait()  # Keep the server running
    except KeyboardInterrupt:
        print("Stopping SMTP server...")
    finally:
        controller.stop()


if __name__ == "__main__":
    asyncio.run(main())

from flask import Flask, Response
import os

# Define the log file path
LOG_FILE_PATH = "./shared/emails.log"

# Initialize the Flask app
app = Flask(__name__)

@app.route('/')
def serve_log_file():
    if os.path.exists(LOG_FILE_PATH):
        # Open the log file and read its content
        with open(LOG_FILE_PATH, 'rb') as log_file:
            log_data = log_file.read()

        # Return the log file as a plain text response
        return Response(log_data, mimetype='text/plain')
    else:
        # If the log file doesn't exist, return a 404
        return "Log file not found.", 404

if __name__ == "__main__":
    # Start the Flask server on port 8000
    app.run(host='0.0.0.0', port=8000)