# -*- coding: utf-8 -*-
"""
召回完成检测器（Recall Completion Detector）

ETL 数据更新后，扫描所有 MAIN 关系对的结算记录，
记录广义召回事件（recall_events），匹配活跃任务标记 completed，
对所有到店的 MAIN 关联客户生成回访任务（follow_up_visit）。

由 trigger_jobs 中的 recall_completion_check 配置驱动（event: etl_data_updated）。

CHANGE 2026-04-08 | Fix-13 改造：
- 扫描范围从"有 active 任务的客户"扩大为"所有 os_label='MAIN' 的关联客户"
- 新增 recall_events 事件表记录广义召回（按天去重）
- 无 active 任务的客户到店也生成 follow_up_visit
"""

import json
import logging
from datetime import timedelta

from app.trace.decorators import trace_service

logger = logging.getLogger(__name__)


def _get_connection():
    """延迟导入 get_connection，避免模块级导入失败。"""
    from app.database import get_connection

    return get_connection()


def _insert_history(
    cur,
    task_id: int,
    action: str,
    old_status: str | None = None,
    new_status: str | None = None,
    old_task_type: str | None = None,
    new_task_type: str | None = None,
    detail: dict | None = None,
) -> None:
    """在 coach_task_history 中记录变更。"""
    cur.execute(
        """
        INSERT INTO biz.coach_task_history
            (task_id, action, old_status, new_status,
             old_task_type, new_task_type, detail)
        VALUES (%s, %s, %s, %s, %s, %s, %s)
        """,
        (
            task_id,
            action,
            old_status,
            new_status,
            old_task_type,
            new_task_type,
            json.dumps(detail) if detail else None,
        ),
    )


@trace_service(description_zh="执行维客检测", description_en="Run recall detection")
def run(payload: dict | None = None, job_id: int | None = None) -> dict:
    """
    召回完成检测主流程。

    CHANGE 2026-04-08 | Fix-13 改造：扫描所有 MAIN 关系对。
    1. 从 biz.sites 获取所有活跃门店
    2. 对每个 site_id，通过 _fdw_context 扫描 MAIN 关系对的结算记录
    3. 有结算 → 写 recall_events + 完成任务（如有）+ 生成回访
    """
    completed_count = 0
    event_count = 0

    conn = _get_connection()
    try:
        # ── 1. 从业务库获取所有活跃门店 ──
        with conn.cursor() as cur:
            cur.execute(
                "SELECT site_id FROM biz.sites WHERE is_active = true"
            )
            site_ids = [r[0] for r in cur.fetchall()]
        conn.commit()

        # ── 2. 逐 site_id 处理 ──
        for site_id in site_ids:
            try:
                result = _process_site(conn, site_id)
                completed_count += result["completed"]
                event_count += result["events"]
            except Exception:
                logger.exception(
                    "处理门店召回检测失败: site_id=%s", site_id
                )
                conn.rollback()

        # ── 更新 last_run_at（兼容 trigger_scheduler 调度记录） ──
        if job_id is not None:
            from app.services.trigger_scheduler import update_job_last_run_at

            with conn.cursor() as cur:
                cur.execute("BEGIN")
                update_job_last_run_at(cur, job_id)
            conn.commit()

    finally:
        conn.close()

    logger.info(
        "召回完成检测完成: completed_count=%d, event_count=%d",
        completed_count, event_count,
    )
    return {"completed_count": completed_count, "event_count": event_count}


def _process_site(conn, site_id: int) -> dict:
    """
    处理单个门店的广义召回检测。

    CHANGE 2026-04-08 | Fix-13 改造：
    1. 从 ETL 查询所有 os_label='MAIN' 的 (assistant_id, member_id) 对
    2. 批量查询这些客户的最新结算记录
    3. 对每个有新结算的关系对：写 recall_events + 完成任务 + 生成回访
    """
    completed = 0
    events = 0

    from app.services.fdw_queries import _fdw_context

    # ── 1. 获取本门店所有 MAIN 关系对 ──
    with _fdw_context(conn, site_id) as cur:
        cur.execute(
            """
            SELECT assistant_id, member_id
            FROM app.v_dws_member_assistant_relation_index
            WHERE os_label = 'MAIN'
            """
        )
        main_pairs = [(r[0], r[1]) for r in cur.fetchall()]

    if not main_pairs:
        return {"completed": 0, "events": 0}

    # ── 2. 批量查询这些客户的最新结算时间 ──
    member_ids = list({mid for _, mid in main_pairs})
    settlement_map: dict[tuple[int, int], object] = {}  # (assistant_id, member_id) → latest_pay_time

    with _fdw_context(conn, site_id) as cur:
        cur.execute(
            """
            SELECT sl.site_assistant_id AS assistant_id,
                   sh.member_id,
                   MAX(sh.pay_time) AS latest_pay_time
            FROM app.v_dwd_settlement_head sh
            JOIN app.v_dwd_assistant_service_log sl
              ON sl.order_settle_id = sh.order_settle_id
             AND sl.is_delete = 0
            WHERE sh.member_id = ANY(%s)
              AND sh.settle_type IN (1, 3)
            GROUP BY sl.site_assistant_id, sh.member_id
            """,
            (member_ids,),
        )
        for row in cur.fetchall():
            settlement_map[(row[0], row[1])] = row[2]

    # ── 3. 获取本门店所有 active 的召回/回访任务（用于匹配） ──
    active_tasks_map: dict[tuple[int, int], list] = {}  # (assistant_id, member_id) → [(id, task_type, created_at)]
    with conn.cursor() as cur:
        cur.execute(
            """
            SELECT id, assistant_id, member_id, task_type, created_at
            FROM biz.coach_tasks
            WHERE site_id = %s
              AND status = 'active'
              AND task_type IN ('high_priority_recall', 'priority_recall', 'follow_up_visit')
            """,
            (site_id,),
        )
        for row in cur.fetchall():
            key = (row[1], row[2])
            active_tasks_map.setdefault(key, []).append(
                {"id": row[0], "task_type": row[3], "created_at": row[4]}
            )
    conn.commit()

    # ── 4. 逐关系对处理 ──
    for assistant_id, member_id in main_pairs:
        latest_pay = settlement_map.get((assistant_id, member_id))
        if latest_pay is None:
            continue

        active_tasks = active_tasks_map.get((assistant_id, member_id), [])

        try:
            result = _process_pair(
                conn, site_id, assistant_id, member_id,
                latest_pay, active_tasks,
            )
            completed += result["completed"]
            events += result["events"]
        except Exception:
            logger.exception(
                "处理关系对失败: site_id=%s, assistant_id=%s, member_id=%s",
                site_id, assistant_id, member_id,
            )
            conn.rollback()

    return {"completed": completed, "events": events}


def _process_pair(
    conn,
    site_id: int,
    assistant_id: int,
    member_id: int,
    latest_pay_time,
    active_tasks: list[dict],
) -> dict:
    """
    处理单个 MAIN 关系对的召回检测。

    CHANGE 2026-04-08 | Fix-13 改造：
    - 写 recall_events（ON CONFLICT DO NOTHING 按天去重）
    - 有 active 召回任务且 pay_time > created_at → 完成任务
    - 关闭旧回访 → 新建回访（48h）
    - 无 active 任务也生成回访

    返回: {"completed": int, "events": int}
    """
    completed = 0
    events = 0

    with conn.cursor() as cur:
        cur.execute("BEGIN")

        # ── 1. 写 recall_events（按天去重） ──
        # 先查是否有匹配的召回任务（用于填充 task_id/task_type）
        recall_tasks = [
            t for t in active_tasks
            if t["task_type"] in ("high_priority_recall", "priority_recall")
            and latest_pay_time > t["created_at"]
        ]
        event_task_id = recall_tasks[0]["id"] if recall_tasks else None
        event_task_type = recall_tasks[0]["task_type"] if recall_tasks else None

        cur.execute(
            """
            INSERT INTO biz.recall_events
                (site_id, assistant_id, member_id, pay_time, task_id, task_type)
            VALUES (%s, %s, %s, %s, %s, %s)
            ON CONFLICT (site_id, assistant_id, member_id, (date_trunc('day', pay_time AT TIME ZONE 'Asia/Shanghai')))
            DO NOTHING
            RETURNING id
            """,
            (site_id, assistant_id, member_id, latest_pay_time,
             event_task_id, event_task_type),
        )
        inserted = cur.fetchone()
        if inserted is None:
            # 今天已记录过，跳过后续处理（避免重复生成回访）
            conn.commit()
            return {"completed": 0, "events": 0}

        events = 1

        # ── 2. 完成匹配的召回任务 ──
        has_active_recall = len(recall_tasks) > 0

        for task in recall_tasks:
            cur.execute(
                """
                UPDATE biz.coach_tasks
                SET status = 'completed',
                    completed_at = %s,
                    completed_task_type = %s,
                    completion_type = 'auto',
                    updated_at = NOW()
                WHERE id = %s AND status = 'active'
                """,
                (latest_pay_time, task["task_type"], task["id"]),
            )
            _insert_history(
                cur,
                task["id"],
                action="completed",
                old_status="active",
                new_status="completed",
                old_task_type=task["task_type"],
                new_task_type=task["task_type"],
                detail={
                    "service_time": str(latest_pay_time),
                    "completed_task_type": task["task_type"],
                },
            )
            completed += 1

        # ── 3. 关闭已有的 active 回访任务 ──
        cur.execute(
            """
            SELECT id FROM biz.coach_tasks
            WHERE site_id = %s AND assistant_id = %s AND member_id = %s
              AND task_type = 'follow_up_visit' AND status = 'active'
            """,
            (site_id, assistant_id, member_id),
        )
        old_follow_ups = cur.fetchall()
        for (old_id,) in old_follow_ups:
            cur.execute(
                """
                UPDATE biz.coach_tasks
                SET status = 'inactive', updated_at = NOW()
                WHERE id = %s
                """,
                (old_id,),
            )
            _insert_history(
                cur, old_id,
                action="superseded_by_new_visit",
                old_status="active", new_status="inactive",
                old_task_type="follow_up_visit", new_task_type="follow_up_visit",
                detail={"reason": "new_service_record", "service_time": str(latest_pay_time)},
            )

        # ── 4. 创建新的回访任务（48h 过期） ──
        expires_at = (
            latest_pay_time + timedelta(hours=48)
            if hasattr(latest_pay_time, '__add__') else None
        )
        cur.execute(
            """
            INSERT INTO biz.coach_tasks
                (site_id, assistant_id, member_id, task_type, status,
                 expires_at, created_at, updated_at)
            VALUES (%s, %s, %s, 'follow_up_visit', 'active', %s, NOW(), NOW())
            RETURNING id
            """,
            (site_id, assistant_id, member_id, expires_at),
        )
        new_follow_up_id = cur.fetchone()[0]
        _insert_history(
            cur, new_follow_up_id,
            action="created",
            old_status=None, new_status="active",
            new_task_type="follow_up_visit",
            detail={
                "reason": "service_record_detected",
                "service_time": str(latest_pay_time),
                "had_recall": has_active_recall,
            },
        )

    conn.commit()

    # ── 5. 触发 recall_completed 事件（仅当有召回任务被完成时） ──
    if has_active_recall:
        try:
            from app.services.trigger_scheduler import fire_event
            fire_event(
                "recall_completed",
                {
                    "site_id": site_id,
                    "assistant_id": assistant_id,
                    "member_id": member_id,
                    "service_time": str(latest_pay_time),
                },
            )
        except Exception:
            logger.exception(
                "触发 recall_completed 事件失败: site_id=%s, assistant_id=%s, member_id=%s",
                site_id, assistant_id, member_id,
            )

    return {"completed": completed, "events": events}