main.py

"""Main module for demonstrating the mail client."""

# ta-assignment/main.py

import contextlib
import logging

import gmail_client_impl  # noqa: F401
import mail_client_api

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def main() -> None:
    """Initialize the client and demonstrate all mail client methods."""
    # Now, get_client() returns a GmailClient instance...
    client = mail_client_api.get_client(interactive=False)

    # Test 1: Get messages (existing functionality)
    messages = list(client.get_messages(max_results=3))

    if not messages:
        return

    for _i, _msg in enumerate(messages, 1):
        pass

    # Test 2: Get a specific message by ID
    if messages:
        test_message_id = messages[0].id
        with contextlib.suppress(Exception):
            pass

    # Test 3: Mark a message as read
    if messages:
        test_message_id = messages[0].id
        with contextlib.suppress(Exception):
            success = client.mark_as_read(test_message_id)
            if success:
                pass
            else:
                pass

    # Test 4: Delete a message (WARNING: This is destructive!)
    # Only test if we have more than one message to avoid deleting all messages
    if len(messages) > 1:
        # Ask for confirmation before deleting
        delete_message_id = messages[-1].id  # Delete the last message
        try:
            confirmation = input("Type 'DELETE' to confirm deletion: ")
            if confirmation == "DELETE":
                success = client.delete_message(delete_message_id)
                if success:
                    logger.info("Message with ID %s deleted.", delete_message_id)
                else:
                    logger.info("Failed to delete message with ID %s.", delete_message_id)
        except EOFError:
            # This means that CircleCI or another non-interactive environment is not going to actually delete anything
            pass
    else:
        pass

    print("Demo complete.")  # noqa: T201


if __name__ == "__main__":
    main()