Neo-ZQYY/apps/backend/app/services/note_service.py

"""
备注服务

负责备注 CRUD、星星评分存储与读取。
备注类型根据关联任务自动确定：follow_up_visit 任务 → follow_up，否则 normal。
星星评分不参与回访完成判定，不参与 AI 分析，仅存储。
"""

import json
import logging

from fastapi import HTTPException
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 _record_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 中记录变更（复用 task_manager 的模式）。"""
    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="ai_analyze_note", description_en="Ai Analyze Note")
async def ai_analyze_note(note_id: int, site_id: int, member_id: int, content: str, user_name: str = "") -> int | None:
    """
    AI 应用 6 备注分析：调用百炼 Application API 获取评分。

    CHANGE 2026-03-27 | 打通 AI 应用 6 调用链
    仅执行 App6 评分，不触发 App8 线索整合（后续统一处理）。
    返回 score（1-10），失败返回 None。
    """
    try:
        from app.ai.config import AIConfig
        from app.ai.dashscope_client import DashScopeClient
        import json

        config = AIConfig.from_env()
        client = DashScopeClient(api_key=config.api_key, workspace_id=config.workspace_id)

        # 构建 prompt（简化版，直接传给百炼应用）
        prompt = json.dumps({
            "site_id": site_id,
            "member_id": member_id,
            "note_content": content,
            "noted_by_name": user_name,
        }, ensure_ascii=False)

        result, tokens_used, _ = await client.call_app(config.app_id_6_note, prompt)
        score = result.get("score") if isinstance(result, dict) else None

        if score is not None:
            score = max(1, min(10, int(score)))
            logger.info("App6 备注评分完成: note_id=%d score=%d tokens=%d", note_id, score, tokens_used)

        return score
    except Exception:
        logger.warning("App6 备注评分失败: note_id=%d", note_id, exc_info=True)
        return None


async def _async_ai_score(note_id: int, site_id: int, member_id: int, content: str) -> None:
    """后台异步执行 AI 评分，不阻塞 API 响应。"""
    try:
        ai_score_val = await ai_analyze_note(
            note_id=note_id, site_id=site_id, member_id=member_id, content=content,
        )
        if ai_score_val is not None:
            conn = _get_connection()
            try:
                with conn.cursor() as cur:
                    cur.execute(
                        "UPDATE biz.notes SET ai_score = %s, updated_at = NOW() WHERE id = %s",
                        (ai_score_val, note_id),
                    )
                conn.commit()
                logger.info("AI 评分已写入: note_id=%d ai_score=%d", note_id, ai_score_val)
            finally:
                conn.close()
    except Exception:
        logger.warning("后台 AI 评分失败: note_id=%d", note_id, exc_info=True)


@trace_service("创建备注", "Create note")
async def create_note(
    site_id: int,
    user_id: int,
    target_type: str,
    target_id: int,
    content: str,
    task_id: int | None = None,
    rating_service_willingness: int | None = None,
    rating_revisit_likelihood: int | None = None,
    score: int | None = None,
) -> dict:
    """
    创建备注。

    1. 验证评分范围（1-5 或 NULL），不合法则 422
    2. 确定 note type：
       - 若 task_id 关联的任务 task_type='follow_up_visit' → type='follow_up'
       - 否则 → type='normal'
    3. INSERT INTO biz.notes
    4. 若 type='follow_up'：
       - 保留 AI 占位调用（P5 接入时调用链不变），返回值仅更新 ai_score
       - 不论 ai_score 如何，有备注即标记关联 active 回访任务 completed
    5. 返回创建的备注记录

    注意：星星评分不参与回访完成判定，不参与 AI 分析，仅存储。
    """
    # 验证评分范围
    for label, val in [
        ("再次服务意愿评分", rating_service_willingness),
        ("再来店可能性评分", rating_revisit_likelihood),
        ("备注星星评分", score),
    ]:
        if val is not None and (val < 1 or val > 5):
            raise HTTPException(
                status_code=422, detail=f"{label}必须在 1-5 范围内"
            )

    conn = _get_connection()
    try:
        # 确定 note type
        note_type = "normal"
        task_info = None

        if task_id is not None:
            with conn.cursor() as cur:
                cur.execute(
                    """
                    SELECT id, task_type, status, site_id
                    FROM biz.coach_tasks
                    WHERE id = %s
                    """,
                    (task_id,),
                )
                row = cur.fetchone()

            if row is None:
                raise HTTPException(status_code=404, detail="关联任务不存在")

            task_info = {
                "id": row[0],
                "task_type": row[1],
                "status": row[2],
                "site_id": row[3],
            }

            if task_info["site_id"] != site_id:
                raise HTTPException(status_code=403, detail="权限不足")

            if task_info["task_type"] == "follow_up_visit":
                note_type = "follow_up"

        # INSERT 备注
        with conn.cursor() as cur:
            cur.execute("BEGIN")
            cur.execute(
                """
                INSERT INTO biz.notes
                    (site_id, user_id, target_type, target_id, type,
                     content, rating_service_willingness,
                     rating_revisit_likelihood, task_id, score)
                VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
                RETURNING id, site_id, user_id, target_type, target_id,
                          type, content, rating_service_willingness,
                          rating_revisit_likelihood, task_id,
                          ai_score, ai_analysis, created_at, updated_at, score
                """,
                (
                    site_id, user_id, target_type, target_id, note_type,
                    content, rating_service_willingness,
                    rating_revisit_likelihood, task_id, score,
                ),
            )
            row = cur.fetchone()

            note = {
                "id": row[0],
                "site_id": row[1],
                "user_id": row[2],
                "target_type": row[3],
                "target_id": row[4],
                "type": row[5],
                "content": row[6],
                "rating_service_willingness": row[7],
                "rating_revisit_likelihood": row[8],
                "task_id": row[9],
                "ai_score": row[10],
                "ai_analysis": row[11],
                "created_at": row[12].isoformat() if row[12] else None,
                "updated_at": row[13].isoformat() if row[13] else None,
                "score": row[14],
            }

            # 若 type='follow_up'，标记回访任务完成（不依赖 AI 评分）
            if note_type == "follow_up" and task_id is not None:
                if task_info and task_info["status"] == "active":
                    cur.execute(
                        """
                        UPDATE biz.coach_tasks
                        SET status = 'completed',
                            completed_at = NOW(),
                            completed_task_type = task_type,
                            updated_at = NOW()
                        WHERE id = %s AND status = 'active'
                        """,
                        (task_id,),
                    )
                    _record_history(
                        cur,
                        task_id,
                        action="completed_by_note",
                        old_status="active",
                        new_status="completed",
                        old_task_type=task_info["task_type"],
                        new_task_type=task_info["task_type"],
                        detail={"note_id": note["id"]},
                    )

        conn.commit()

        # CHANGE 2026-03-27 | AI 评分：后台异步执行，不阻塞 API 响应
        # 备注先返回给前端（aiScore=null），AI 评分完成后写入数据库
        # 前端下次加载页面时自动获取最新 aiScore
        import asyncio
        asyncio.create_task(_async_ai_score(note["id"], site_id, target_id, content))

        return note

    except HTTPException:
        conn.rollback()
        raise
    except Exception:
        conn.rollback()
        raise
    finally:
        conn.close()


@trace_service("查询备注列表", "Get notes")
async def get_notes(
    site_id: int, target_type: str, target_id: int
) -> list[dict]:
    """
    查询某目标的备注列表。

    按 created_at DESC 排序，包含星星评分和 AI 评分。
    """
    conn = _get_connection()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """
                SELECT id, site_id, user_id, target_type, target_id,
                       type, content, rating_service_willingness,
                       rating_revisit_likelihood, task_id,
                       ai_score, ai_analysis, created_at, updated_at
                FROM biz.notes
                WHERE site_id = %s
                  AND target_type = %s
                  AND target_id = %s
                ORDER BY created_at DESC
                """,
                (site_id, target_type, target_id),
            )
            rows = cur.fetchall()
        conn.commit()

        return [
            {
                "id": r[0],
                "site_id": r[1],
                "user_id": r[2],
                "target_type": r[3],
                "target_id": r[4],
                "type": r[5],
                "content": r[6],
                "rating_service_willingness": r[7],
                "rating_revisit_likelihood": r[8],
                "task_id": r[9],
                "ai_score": r[10],
                "ai_analysis": r[11],
                "created_at": r[12].isoformat() if r[12] else None,
                "updated_at": r[13].isoformat() if r[13] else None,
            }
            for r in rows
        ]

    finally:
        conn.close()


@trace_service("删除备注", "Delete note")
async def delete_note(note_id: int, user_id: int, site_id: int) -> dict:
    """
    删除备注。

    验证备注归属（user_id + site_id）后执行硬删除。
    - 不存在 → 404
    - 不属于当前用户 → 403
    """
    conn = _get_connection()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """
                SELECT id, user_id, site_id
                FROM biz.notes
                WHERE id = %s
                """,
                (note_id,),
            )
            row = cur.fetchone()

        if not row:
            raise HTTPException(status_code=404, detail="资源不存在")

        if row[2] != site_id or row[1] != user_id:
            raise HTTPException(status_code=403, detail="权限不足")

        with conn.cursor() as cur:
            cur.execute("BEGIN")
            cur.execute(
                "DELETE FROM biz.notes WHERE id = %s",
                (note_id,),
            )
        conn.commit()

        return {"id": note_id, "deleted": True}

    except HTTPException:
        conn.rollback()
        raise
    except Exception:
        conn.rollback()
        raise
    finally:
        conn.close()