v0.1.0

backend/app/core/imap.py

@@ -0,0 +1,42 @@
+import imaplib
+
+from app.core.logging import get_logger
+
+"""IMAP utility functions for connecting to mail servers and fetching folders."""
+
+logger = get_logger(__name__)
+
+
+def _test_imap_connection(server, username, password):
+    """Test the IMAP connection with the given credentials."""
+    logger.info(f"Testing IMAP connection to {server} for user {username}")
+    try:
+        mail = imaplib.IMAP4_SSL(server)
+        mail.login(username, password)
+        mail.logout()
+        logger.info("IMAP connection successful")
+        return True, "Connection successful"
+    except Exception as e:
+        logger.error(f"IMAP connection failed: {e}")
+        return False, str(e)
+
+
+def get_folders(server, username, password):
+    """Fetch a list of IMAP folders from the mail server."""
+    logger.info(f"Fetching IMAP folders from {server} for user {username}")
+    try:
+        mail = imaplib.IMAP4_SSL(server)
+        mail.login(username, password)
+        status, folders = mail.list()
+        mail.logout()
+        if status == "OK":
+            folder_list = [
+                folder.decode().split(' "/" ')[1].strip('"') for folder in folders
+            ]
+            logger.info(f"Found {len(folder_list)} folders")
+            return folder_list
+        logger.warning(f"Failed to list IMAP folders, status: {status}")
+        return []
+    except Exception as e:
+        logger.error(f"Error fetching IMAP folders: {e}")
+        return []