"""
core/tools.py — บันทึกการเรียก tool ของ LLM ในหนึ่งเทิร์น (ใช้ร่วมกันทุกแบรนด์)
ย้ายมาจาก _tracked / _tool_call_log ของ chatbot_v5.py (ชั้น 5) · พฤติกรรมเท่าเดิม
"""
from __future__ import annotations

import inspect
from typing import Callable


class ToolLog:
    def __init__(self):
        self.calls: list[dict] = []

    def reset(self) -> None:
        self.calls = []

    def track(self, fn: Callable) -> Callable:
        params = list(inspect.signature(fn).parameters.keys())
        log = self

        def wrapper(*args, **kwargs):
            entry = {"tool_name": fn.__name__, "args": {}, "result": "", "success": True}
            try:
                entry["args"] = {**dict(zip(params, args)), **kwargs}
                result = fn(*args, **kwargs)
                entry["result"] = result
                # คืน dict เพื่อให้ Gemini SDK ทำงานได้ (ของเดิม)
                return {"result": result}
            except Exception as e:
                entry["success"] = False
                entry["result"] = str(e)
                raise
            finally:
                log.calls.append(entry)
        wrapper.__name__ = fn.__name__
        wrapper.__doc__ = fn.__doc__
        return wrapper