main.py

import asyncio
import signal
from adapter import Config, MSGraphClient, SMTPServer
from adapter.msgraph import MockMSGraphClient


async def main():
    config = Config()
    
    if config.use_mock_msgraph:
        print("Using Mock MS Graph Client for testing")
        msgraph_client = MockMSGraphClient(simulate_failures=config.mock_simulate_failures)
    else:
        print("Using Real MS Graph Client")
        msgraph_client = MSGraphClient(
            tenant_id=config.msgraph_tenant_id,
            client_id=config.msgraph_client_id,
            client_secret=config.msgraph_client_secret
        )
    
    smtp_server = SMTPServer(
        host='0.0.0.0',
        port=config.smtp_port,
        tls_cert_path=config.smtp_tls_cert_path,
        tls_key_path=config.smtp_tls_key_path,
        msgraph_client=msgraph_client,
        user_mappings=config.smtp_users
    )
    
    try:
        await smtp_server.start()
        
        while True:
            await asyncio.sleep(1)
    
    except KeyboardInterrupt:
        print("\nShutting down...")
        if isinstance(msgraph_client, MockMSGraphClient):
            msgraph_client.print_summary()
        await smtp_server.stop()
    
    except Exception as e:
        print(f"Error: {str(e)}")
        await smtp_server.stop()
        raise


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