57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from typing import Any
|
|
|
|
from duck_core.tools.base import ToolResult
|
|
from duck_core.tools.paths import WorkspacePathError, candidate_path, resolve_workspace_path
|
|
|
|
|
|
class FileWriteTool:
|
|
name = "file_write"
|
|
risk_level = "medium"
|
|
|
|
def __init__(self, workspace: str):
|
|
self.workspace = workspace
|
|
|
|
async def run(self, args: dict[str, Any]) -> ToolResult:
|
|
raw_path = str(args.get("path", ""))
|
|
content = str(args.get("content", ""))
|
|
overwrite = bool(args.get("overwrite", False))
|
|
approved = bool(args.get("_approved"))
|
|
try:
|
|
path = resolve_workspace_path(self.workspace, raw_path, allow_outside=approved)
|
|
except WorkspacePathError as exc:
|
|
path = candidate_path(self.workspace, raw_path)
|
|
return ToolResult(
|
|
ok=False,
|
|
error=f"{exc}. Writing outside workspace requires approval.",
|
|
metadata={
|
|
"path": str(path),
|
|
"requires_approval": True,
|
|
"risk_level": self.risk_level,
|
|
"reason": "Path is outside workspace",
|
|
},
|
|
)
|
|
if path.exists() and not overwrite and not approved:
|
|
return ToolResult(
|
|
ok=False,
|
|
error="Refusing to overwrite existing file without overwrite=true or approval",
|
|
metadata={
|
|
"path": str(path),
|
|
"requires_approval": True,
|
|
"risk_level": self.risk_level,
|
|
"reason": "File exists and overwrite was not requested",
|
|
},
|
|
)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
existed = path.exists()
|
|
path.write_text(content)
|
|
return ToolResult(
|
|
ok=True,
|
|
output=f"Wrote {raw_path}",
|
|
metadata={
|
|
"path": str(path),
|
|
"bytes_written": len(content.encode()),
|
|
"created": not existed,
|
|
"updated": existed,
|
|
},
|
|
)
|