#!/usr/bin/env python3
import sys
import os
import re

# Add the src directory to the Python path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

from email_client.config import EMAIL_CONFIG
import imaplib
import logging

# Set up logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
log = logging.getLogger(__name__)

def delete_custom_labels():
    """Delete all custom Gmail labels except system labels."""
    # Connect to Gmail
    mail = imaplib.IMAP4_SSL(EMAIL_CONFIG["imap_server"], 993)  # Default IMAP SSL port
    log.info(f"Connecting to {EMAIL_CONFIG['imap_server']} with {EMAIL_CONFIG['email']}")
    mail.login(EMAIL_CONFIG["email"], EMAIL_CONFIG["password"])
    log.info("Login successful")
    
    try:
        # List all labels
        log.info("Listing labels...")
        typ, data = mail.list()
        if typ != 'OK':
            log.error(f"Failed to list labels: {typ}")
            return
        
        # Extract label names
        labels = []
        for item in data:
            if isinstance(item, bytes):
                decoded = item.decode('utf-8')
                log.debug(f"Raw label data: {decoded}")
                # Extract the label name from the response
                # Format is typically: '(\\HasNoChildren) "/" "LabelName"'
                match = re.search(r'"([^"]+)"$', decoded)
                if match:
                    label_name = match.group(1)
                    # Skip system labels
                    if '[Gmail]' in label_name or label_name == 'INBOX':
                        continue
                    labels.append(label_name)
        
        log.info(f"Found {len(labels)} custom labels to delete: {labels}")
        
        # Delete each custom label
        for label in labels:
            try:
                log.info(f"Deleting label: {label}")
                # Try with different encodings and formats
                try:
                    # Try direct deletion
                    status, response = mail.delete(label)
                    log.info(f"Delete response: {status}, {response}")
                    if status == 'OK':
                        log.info(f"Successfully deleted label: {label}")
                        continue
                except Exception as e:
                    log.warning(f"First attempt failed: {str(e)}")
                
                try:
                    # Try with quotes
                    status, response = mail.delete(f'"{label}"')
                    log.info(f"Delete response (quoted): {status}, {response}")
                    if status == 'OK':
                        log.info(f"Successfully deleted label with quotes: {label}")
                        continue
                except Exception as e:
                    log.warning(f"Second attempt failed: {str(e)}")
                
                try:
                    # Try with UTF-7 encoding
                    encoded_label = label.encode('utf-7')
                    status, response = mail.delete(encoded_label)
                    log.info(f"Delete response (utf-7): {status}, {response}")
                    if status == 'OK':
                        log.info(f"Successfully deleted label with utf-7: {label}")
                        continue
                except Exception as e:
                    log.warning(f"Third attempt failed: {str(e)}")
                
                log.error(f"All attempts to delete label '{label}' failed")
            except Exception as e:
                log.error(f"Error deleting label '{label}': {str(e)}")
    
    finally:
        # Logout
        mail.logout()
        log.info("Logged out")

if __name__ == "__main__":
    delete_custom_labels()