#🎯 小项目③：问答 / 任务交互

# brain.py —— 小莫的交互中枢：既答问题（Service），又做长任务（Action）
import pyarrow as pa
from dora import Node

# 一个简单的"知识库"，用于回答问题
KNOWLEDGE = {
    "你是谁": "我是小莫，你的机器人伙伴！",
    "几点了": "我还没装时钟，第八章见～",
    "会什么": "我正在学习四种沟通方式！",
}


def main():
    node = Node()

    active_tasks = {}      # 进行中的任务：goal_id -> 还剩几步

    for event in node:
        if event["type"] == "INPUT":

            # ===== 角色一：Service 服务端，回答提问 =====
            if event["id"] == "question":
                q = event["value"][0].as_py()
                answer = KNOWLEDGE.get(q, "这个问题我还答不上来～")
                # 关键：把 request_id 原样贴回答复
                node.send_output(
                    "answer",
                    pa.array([answer]),
                    metadata=event["metadata"],
                )
                print(f"[brain] 收到提问「{q}」，已回答", flush=True)

            # ===== 角色二：Action 服务端，接收长任务 =====
            elif event["id"] == "task":
                steps = event["value"][0].as_py()          # 任务需要几步
                goal_id = event["metadata"]["goal_id"]
                active_tasks[goal_id] = steps
                print(f"[brain] 接到任务 {goal_id[:8]}：共 {steps} 步", flush=True)

            # ===== 定时器：推进所有进行中的任务各一步 =====
            elif event["id"] == "tick":
                for goal_id in list(active_tasks.keys()):
                    remaining = active_tasks[goal_id] - 1
                    if remaining > 0:
                        active_tasks[goal_id] = remaining
                        node.send_output(
                            "feedback",
                            pa.array([remaining]),
                            metadata={"goal_id": goal_id},
                        )
                    else:
                        del active_tasks[goal_id]
                        node.send_output(
                            "result",
                            pa.array([0]),
                            metadata={"goal_id": goal_id, "goal_status": "succeeded"},
                        )

        elif event["type"] == "STOP":
            break


if __name__ == "__main__":
    main()

# user.py —— 模拟用户：轮流提问和派任务，接收所有回复
import uuid
import pyarrow as pa
from dora import Node

QUESTIONS = ["你是谁", "会什么", "几点了"]


def main():
    node = Node()

    pending_q = {}       # 未收到答复的提问：request_id -> 问题
    round_num = 0

    for event in node:
        if event["type"] == "INPUT":

            if event["id"] == "tick":
                # 偶数轮提问（Service），奇数轮派任务（Action）
                if round_num % 2 == 0:
                    q = QUESTIONS[(round_num // 2) % len(QUESTIONS)]
                    req_id = str(uuid.uuid4())
                    pending_q[req_id] = q
                    node.send_output("question", pa.array([q]),
                                     metadata={"request_id": req_id})
                    print(f"[user] 提问：{q}", flush=True)
                else:
                    goal_id = str(uuid.uuid4())
                    node.send_output("task", pa.array([3]),      # 派一个 3 步的任务
                                     metadata={"goal_id": goal_id})
                    print(f"[user] 派任务：请做 3 步（{goal_id[:8]}）", flush=True)
                round_num += 1

            # 收到 Service 答复
            elif event["id"] == "answer":
                req_id = event["metadata"]["request_id"]
                ans = event["value"][0].as_py()
                if req_id in pending_q:
                    q = pending_q.pop(req_id)
                    print(f"[user]   答复「{q}」→ {ans}", flush=True)

            # 收到 Action 反馈
            elif event["id"] == "feedback":
                remaining = event["value"][0].as_py()
                print(f"[user]   任务进度：还剩 {remaining} 步", flush=True)

            # 收到 Action 结果
            elif event["id"] == "result":
                status = event["metadata"]["goal_status"]
                print(f"[user]   任务完成：{status} 🎉", flush=True)

        elif event["type"] == "STOP":
            break


if __name__ == "__main__":
    main()