"""
任务二企业微信消息推送模块
用于发送同步失败通知
"""

import sys
from pathlib import Path
from typing import List, Dict
from datetime import datetime

# 将项目根目录添加到 Python 路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from src.wework_notifier import WeWorkNotifier
from src2.logger import get_task2_logger


def send_sync_failure_notification(access_token: str, agentid: str,
                                   receivers: str, failed_records: List[Dict]) -> bool:
    """
    发送同步失败通知

    Args:
        access_token: 企业微信access_token
        agentid: 应用ID
        receivers: 接收人列表
        failed_records: 失败记录列表，每条记录包含：
            - sheet_title: 子表标题
            - record_id: 记录ID
            - tapd_link: TAPD链接
            - error_message: 失败原因

    Returns:
        bool: 是否发送成功
    """
    if not failed_records:
        return True

    # 构造消息内容
    content = _build_sync_failure_message(failed_records)

    # 使用任务一的推送器发送消息
    notifier = WeWorkNotifier(access_token, agentid, receivers, logger=get_task2_logger())
    return notifier._send_text_message(content)


def _build_sync_failure_message(failed_records: List[Dict]) -> str:
    """
    构造同步失败消息内容（支持多表格、多子表分组）

    Args:
        failed_records: 失败记录列表，每条记录可包含：
            - doc_index: 表格序号（可选）
            - docid_short: 表格ID简写（可选）
            - sheet_title: 子表标题
            - record_id: 记录ID
            - tapd_link: TAPD链接
            - error_message: 失败原因

    Returns:
        str: 格式化的消息内容
    """
    # 按表格和子表分组
    records_by_doc = {}
    for record in failed_records:
        doc_index = record.get('doc_index', 1)
        docid_short = record.get('docid_short', '')
        doc_key = (doc_index, docid_short)

        if doc_key not in records_by_doc:
            records_by_doc[doc_key] = {}

        sheet_title = record.get('sheet_title', '未知子表')
        if sheet_title not in records_by_doc[doc_key]:
            records_by_doc[doc_key][sheet_title] = []
        records_by_doc[doc_key][sheet_title].append(record)

    # 消息头部
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    total_count = len(failed_records)
    doc_count = len(records_by_doc)

    lines = [
        "【autoTAPD 同步失败通知】",
        f"时间: {timestamp}",
        f"失败数量: {total_count} 条",
    ]

    # 如果有多个表格，显示表格数量
    if doc_count > 1:
        lines.append(f"涉及表格: {doc_count} 个")

    lines.extend([
        "",
        "以下记录同步失败，请检查：",
        "=" * 40
    ])

    # 按表格和子表分组显示失败记录
    global_idx = 1
    for (doc_index, docid_short), sheets in sorted(records_by_doc.items()):
        # 如果有多个表格，显示表格标识
        if doc_count > 1:
            doc_label = f"表格{doc_index}"
            if docid_short:
                doc_label += f" ({docid_short})"
            lines.append(f"\n{'#'*20}")
            lines.append(f"# {doc_label}")
            lines.append(f"{'#'*20}")

        for sheet_title, sheet_records in sheets.items():
            lines.append(f"\n【子表：{sheet_title}】")
            lines.append("")

            for record in sheet_records:
                record_id = record.get('record_id', '未知')
                tapd_link = record.get('tapd_link', '(无链接)')
                error_message = record.get('error_message', '未知错误')

                lines.append(f"[{global_idx}] 记录ID: {record_id}")
                lines.append(f"TAPD链接: {tapd_link}")
                lines.append(f"失败原因: {error_message}")
                lines.append("")

                global_idx += 1

    lines.append("=" * 40)
    lines.append("系统将在下次同步时自动重试失败记录。")

    return "\n".join(lines)


if __name__ == "__main__":
    print("=== 任务二推送模块 ===")
    print("此模块提供同步失败通知功能")
    print("请通过 sync_service 或 main.py 调用")