"""文件写入工具"""

from pathlib import Path
from typing import Optional, Type

from anyio import Path as AsyncPath
from pydantic import BaseModel, Field

from app.agent.tools.base import MoviePilotTool
from app.log import logger


class WriteFileInput(BaseModel):
    """Input parameters for write file tool"""
    file_path: str = Field(..., description="The absolute path of the file to write")
    content: str = Field(..., description="The content to write into the file")


class WriteFileTool(MoviePilotTool):
    name: str = "write_file"
    description: str = "Write full content to a file. If the file already exists, it will be overwritten. Automatically creates parent directories if they don't exist."
    args_schema: Type[BaseModel] = WriteFileInput

    def get_tool_message(self, **kwargs) -> Optional[str]:
        """根据参数生成友好的提示消息"""
        file_path = kwargs.get("file_path", "")
        file_name = Path(file_path).name if file_path else "未知文件"
        return f"正在写入文件: {file_name}"

    async def run(self, file_path: str, content: str, **kwargs) -> str:
        logger.info(f"执行工具: {self.name}, 参数: file_path={file_path}")

        try:
            path = AsyncPath(file_path)
            
            if await path.exists() and not await path.is_file():
                return f"错误：{file_path} 路径已存在但不是一个文件"

            # 自动创建父目录
            await path.parent.mkdir(parents=True, exist_ok=True)
            
            # 写入文件
            await path.write_text(content, encoding="utf-8")
                
            logger.info(f"成功写入文件 {file_path}")
            return f"成功写入文件 {file_path}"

        except PermissionError:
            return f"错误：没有权限写入 {file_path}"
        except Exception as e:
            logger.error(f"写入文件 {file_path} 时发生错误: {str(e)}", exc_info=True)
            return f"操作失败: {str(e)}"