"""大周天内驱力 prompt 构建（纯函数，无 IO）。

复用 ``EvaluationSnapshot``（态势快照），产出 ``(system, user)`` 消息对，喂给
模型网关。模型返回的 JSON 由 ``extract`` 模块解析。

枚举取值以 [内驱力生成器设计](../../docs/domain/core-domains.md) §5 与
[数据库 schema](../../docs/contracts/contract-glossary.md) §22 为准。
"""

from app.modules.big_cycle.prompts import EvaluationSnapshot

DRIVE_SOURCES = (
    "world_uncertainty",
    "memory_gap",
    "perception_change",
    "goal_tension",
    "reflection",
    "time_rhythm",
)
DRIVE_LEVELS = ("L0", "L1", "L2", "L3", "L4")


def build_drive_extraction_messages(
    snapshot: EvaluationSnapshot,
) -> tuple[str, str]:
    system = (
        "你是魂镜的内驱力生成器。基于本轮自我评估的态势快照，识别「匮乏感」——"
        "世界模型不确定性、记忆缺口、目标张力等——生成好奇问题与可审核的探索性次生目标。"
        "不生成已有确信答案的问题；不为满足好奇而越权（摄像头 / 麦克风 / 敏感读取仍需授权）。"
        f"source 取自：{', '.join(DRIVE_SOURCES)}。"
        f"risk_level 取自：{', '.join(DRIVE_LEVELS)}"
        "（探索风险等级，L4 必须人工确认）。"
        '以 JSON 数组返回，每个元素是一个好奇问题：{"question": str, "source": str, '
        '"expected_value": 0~1, "urgency": 0~1, "risk_level": str, '
        '"related_entities": [实体名], "subgoals": [{"title": str, "description": str, '
        '"expected_information_gain": 0~1, "interruption_cost": 0~1, "risk_level": str}]}。'
        "subgoals 为该好奇问题转化的可执行探索目标，可空数组。"
        "仅在态势确有缺口时生成，宁缺毋滥；无缺口返回 []。"
        "只返回 JSON 数组，不要解释或 markdown。"
    )
    user = (
        f"记忆总数：{snapshot.memory_count}\n"
        f"世界实体数：{snapshot.world_entity_count}\n"
        f"自我评估结论：{snapshot.conclusion or '（无明显态势）'}\n"
        f"当前人格摘要：\n{snapshot.personality_summary or '（尚未形成）'}\n"
        f"近期记忆摘要：\n{_format_memories(snapshot.recent_memories)}"
    )
    return system, user


def _format_memories(memories: list[tuple[str, str]]) -> str:
    if not memories:
        return "（暂无）"
    lines = [f"[{idx}] {title}：{content}" for idx, (title, content) in enumerate(memories)]
    return "\n".join(lines)