#!/usr/bin/env python3
"""Check unread emails, grouped by sender."""
import os
import sys
from collections import defaultdict

# Set credentials path
os.environ.setdefault('GMAIL_CREDENTIALS_PATH', os.path.expanduser('~/.gmail-mcp/credentials.json'))

from gmail_mcp.utils.GCP.gmail_auth import get_gmail_service

def main():
    days = int(sys.argv[1]) if len(sys.argv) > 1 else 7
    max_results = int(sys.argv[2]) if len(sys.argv) > 2 else 25

    service = get_gmail_service()
    results = service.users().messages().list(
        userId='me',
        q=f'is:unread newer_than:{days}d',
        maxResults=max_results
    ).execute()

    by_sender = defaultdict(list)
    for msg in results.get('messages', []):
        detail = service.users().messages().get(
            userId='me',
            id=msg['id'],
            format='metadata',
            metadataHeaders=['From', 'Subject']
        ).execute()
        headers = {h['name']: h['value'] for h in detail['payload']['headers']}
        sender = headers.get('From', 'Unknown').split('<')[0].strip().strip('"')
        by_sender[sender].append(headers.get('Subject', '(no subject)')[:50])

    if not by_sender:
        print("No unread emails in the last", days, "days")
        return

    print(f"Unread emails (last {days} days):\n")
    for sender, subjects in sorted(by_sender.items(), key=lambda x: -len(x[1])):
        print(f"* {sender} ({len(subjects)})")
        for s in subjects[:2]:
            print(f"  - {s}")
        if len(subjects) > 2:
            print(f"  - ...+{len(subjects)-2} more")

if __name__ == '__main__':
    main()