#!/usr/bin/env python3
"""
Classify emails using AI
Usage: python3 classify.py <email_json>
"""

import json
import sys

def classify_email(email_data):
    """Use AI to classify email"""
    
    prompt = f"""Classify this email into one category:

From: {email_data['from']}
Subject: {email_data['subject']}
Body: {email_data['body'][:500]}

Categories:
1. URGENT - Needs immediate attention (client, boss, deadline, problem)
2. NEWSLETTER - Marketing, updates, subscriptions
3. FOLLOW_UP - Waiting for your reply or follow-up needed
4. FYI - Information only, no action needed
5. SPAM - Unwanted, promotional

Respond in JSON format:
{{
  "category": "urgent|newsletter|follow_up|fyi|spam",
  "confidence": 0-100,
  "reason": "why this category",
  "action": "star|archive|reply|read|delete",
  "priority": "high|medium|low",
  "suggested_response": "draft reply if applicable"
}}"""

    # Call AI model
    result = call_ai_model(prompt)
    return json.loads(result)

def main():
    # Read email from stdin or file
    if len(sys.argv) > 1:
        with open(sys.argv[1]) as f:
            email = json.load(f)
    else:
        email = json.load(sys.stdin)
    
    classification = classify_email(email)
    print(json.dumps(classification, indent=2))

if __name__ == "__main__":
    main()

#!/usr/bin/env python3
"""
Generate daily email digest
Usage: python3 daily-digest.py
"""

import subprocess
import json
from datetime import datetime

def fetch_important_emails():
    """Fetch starred/important emails"""
    result = subprocess.run(
        ["gog", "gmail", "search", "is:starred OR is:important", "--max=20", "--json"],
        capture_output=True,
        text=True
    )
    return json.loads(result.stdout)

def fetch_todays_emails():
    """Fetch today's emails"""
    result = subprocess.run(
        ["gog", "gmail", "search", "newer_than:1d", "--max=50", "--json"],
        capture_output=True,
        text=True
    )
    return json.loads(result.stdout)

def generate_digest():
    """Generate digest message"""
    
    important = fetch_important_emails()
    today = fetch_todays_emails()
    
    message = f"📧 *Email Digest - {datetime.now().strftime('%A, %d %B')}*\n\n"
    
    # Important emails
    if important:
        message += f"⭐ *Important ({len(important)}):*\n"
        for email in important[:5]:
            sender = email['from'].split('<')[0].strip()
            subject = email['subject'][:40]
            message += f"• {subject} - _{sender}_\n"
        message += "\n"
    
    # Today's summary
    message += f"📊 *Today's Summary:*\n"
    message += f"• Total received: {len(today)}\n"
    message += f"• Important: {len(important)}\n"
    message += f"• Need reply: {len([e for e in today if 'Re:' not in e['subject']])}\n"
    
    return message

def send_to_telegram(message):
    """Send digest to Telegram"""
    import os
    import requests
    
    token = os.getenv("TELEGRAM_BOT_TOKEN")
    chat_id = os.getenv("TELEGRAM_CHAT_ID")
    
    url = f"https://api.telegram.org/bot{token}/sendMessage"
    requests.post(url, data={
        "chat_id": chat_id,
        "text": message,
        "parse_mode": "Markdown"
    })

if __name__ == "__main__":
    digest = generate_digest()
    print(digest)
    send_to_telegram(digest)

#!/usr/bin/env python3
"""
Draft replies for common email types
"""

import json
import subprocess

def draft_reply(email, classification):
    """Generate draft reply using AI"""
    
    prompt = f"""Draft a professional email reply:

Original email:
From: {email['from']}
Subject: {email['subject']}
Body: {email['body'][:300]}

Category: {classification['category']}

Draft a concise, professional reply. Keep it brief but helpful."""

    reply = call_ai_model(prompt)
    return reply

def create_draft(email_id, reply_body):
    """Create Gmail draft"""
    subprocess.run([
        "gog", "gmail", "draft",
        "--reply-to", email_id,
        "--body", reply_body
    ])

# Usage: Process emails marked for reply

VIP_SENDERS = ['boss@company.com', 'client@vip.com']

def check_vip(email):
    if any(vip in email['from'] for vip in VIP_SENDERS):
        return {'category': 'urgent', 'action': 'star'}

def is_follow_up(email):
    # Check if email is part of existing thread
    if 'Re:' in email['subject'] or email.get('threadId'):
        return True