"""
Tool call tracking model for Claude Code sessions.
"""

from datetime import datetime
from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey, Boolean
from sqlalchemy.orm import relationship

from app.models.base import Base


class ToolCall(Base):
    """
    Tracks individual tool calls made during Claude Code sessions.
    
    This helps analyze which tools are used most frequently and their success rates.
    """
    __tablename__ = "tool_calls"

    id = Column(Integer, primary_key=True, index=True)
    session_id = Column(String, ForeignKey("sessions.id"), nullable=False, index=True)
    tool_name = Column(String(100), nullable=False, index=True)
    parameters = Column(Text, nullable=True)  # JSON string of tool parameters
    result_status = Column(String(20), nullable=True, index=True)  # success, error, timeout
    error_message = Column(Text, nullable=True)
    execution_time_ms = Column(Integer, nullable=True)  # Tool execution time in milliseconds
    timestamp = Column(DateTime, nullable=False, default=datetime.utcnow, index=True)
    
    # Relationships
    session = relationship("Session", back_populates="tool_calls")

    def __repr__(self):
        return f"<ToolCall(id={self.id}, tool={self.tool_name}, session={self.session_id})>"