from pydantic import BaseModel, Field

from crewai.agents.cache.cache_handler import CacheHandler
from crewai.tools.structured_tool import CrewStructuredTool


class CacheTools(BaseModel):
    """Default tools to hit the cache."""

    name: str = "Hit Cache"
    cache_handler: CacheHandler = Field(
        description="Cache Handler for the crew",
        default_factory=CacheHandler,
    )

    def tool(self) -> CrewStructuredTool:
        return CrewStructuredTool.from_function(
            func=self.hit_cache,
            name=self.name,
            description="Reads directly from the cache",
        )

    def hit_cache(self, key: str) -> str:
        import json

        split = key.split("tool:")
        tool = split[1].split("|input:")[0].strip()
        tool_input_str = split[1].split("|input:")[1].strip()
        tool_input = json.loads(tool_input_str) if tool_input_str else None
        result = self.cache_handler.read(tool, tool_input)
        return result if result is not None else ""