import logging
from typing import Generator, Optional
from src.config import Config
from src.email_client import fetch_unseen_emails, mark_as_seen, Email
from src.ai_client import summarize_email
from src.tg_bot import send_summary, format_summary

logger = logging.getLogger(__name__)


def poll_accounts(cfg: Config) -> Generator[tuple[int, Email], None, None]:
    for idx, acct in enumerate(cfg.email_accounts):
        try:
            logger.info(f"检查邮箱: {acct.username}")
            emails = fetch_unseen_emails(acct)
            if emails:
                logger.info(f"  发现 {len(emails)} 封新邮件")
            for mail in emails:
                yield idx, mail
        except Exception as e:
            logger.error(f"轮询 {acct.username} 失败: {e}", exc_info=True)


def ai_process(cfg: Config, acct_idx: int, mail: Email) -> Optional[dict]:
    logger.info(f"  正在摘要: {mail.subject}")
    summary = summarize_email(cfg.ai, mail.recipient, mail.subject, mail.sender, mail.body, mail.account_email)
    summary["recipient"] = mail.recipient
    if "@" not in summary.get("sender", ""):
        summary["sender"] = mail.sender
    text = format_summary(summary, mail.account_email)
    return {
        "text": text,
        "data": summary,
        "original_body": mail.body,
        "original_sender": mail.sender,
        "original_recipient": mail.recipient,
        "original_reply_to": mail.reply_to,
        "acct_idx": acct_idx,
        "account_email": mail.account_email,
        "uid": mail.uid,
    }


def tg_send_and_mark(cfg: Config, info: dict):
    acct = cfg.email_accounts[info["acct_idx"]]
    send_summary(cfg.telegram, cfg.telegram.chat_id,
                 info["text"], info["data"],
                 info["original_body"], info["acct_idx"],
                 original_sender=info.get("original_sender", ""),
                 original_recipient=info.get("original_recipient", ""),
                 original_reply_to=info.get("original_reply_to", ""),
                 account_email=info.get("account_email", ""))
    mark_as_seen(acct, [info["uid"]])
    logger.info(f"  已发送并标记已读")