mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:20:44 +08:00
oliver
d8bc08ee0d
Refuse openpyxl/pandas-to_excel via run_command, auto-write content-only files under tmp/, and limit WhatsApp ops short intents to 8 tool rounds by default. Co-authored-by: Cursor <cursoragent@cursor.com>
93 lines
3.8 KiB
Python
93 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
from typing import Any
|
|
|
|
from runtime.tools.base import ToolSpec
|
|
from runtime.tools.path_guard import resolve_workspace_path
|
|
|
|
|
|
def _resolve_write_path(args: dict[str, Any]) -> str:
|
|
for key in ("path", "file", "filename", "file_path", "filepath", "name"):
|
|
raw = str(args.get(key) or "").strip().strip('"').strip("'")
|
|
if raw:
|
|
return raw
|
|
return ""
|
|
|
|
|
|
def _default_write_path() -> str:
|
|
stamp = datetime.now(timezone.utc).strftime("%Y%m%dT%H%M%SZ")
|
|
return f"tmp/write_{stamp}.txt"
|
|
|
|
|
|
def write_file_tool() -> ToolSpec:
|
|
def _handler(args: dict[str, Any]) -> dict[str, Any]:
|
|
raw = _resolve_write_path(args)
|
|
content = args.get("content")
|
|
if content is None:
|
|
content = args.get("text")
|
|
if content is None:
|
|
content = args.get("body")
|
|
auto_path = False
|
|
if not raw:
|
|
if content is None:
|
|
return {
|
|
"ok": False,
|
|
"error": "path_required",
|
|
"hint": "Pass path (or file/filename) relative to workspace root.",
|
|
"example": {"path": "tmp/notes.txt", "content": "hello", "mode": "overwrite"},
|
|
}
|
|
# Production WA agents often omit path; default under tmp/ instead of schema-fail looping.
|
|
raw = _default_write_path()
|
|
auto_path = True
|
|
content_s = "" if content is None else str(content)
|
|
mode = str(args.get("mode") or "overwrite").strip().lower()
|
|
try:
|
|
p = resolve_workspace_path(raw)
|
|
except ValueError as exc:
|
|
return {"ok": False, "error": str(exc)}
|
|
p.parent.mkdir(parents=True, exist_ok=True)
|
|
if mode not in ("overwrite", "append"):
|
|
return {"ok": False, "error": "invalid_mode", "allowed": ["overwrite", "append"]}
|
|
if mode == "append":
|
|
p.write_text(p.read_text(encoding="utf-8", errors="replace") + content_s, encoding="utf-8")
|
|
else:
|
|
p.write_text(content_s, encoding="utf-8")
|
|
out: dict[str, Any] = {"ok": True, "path": str(p), "bytes": p.stat().st_size}
|
|
if auto_path:
|
|
out["auto_path"] = True
|
|
out["hint"] = f"No path provided; wrote content to {raw}."
|
|
return out
|
|
|
|
return ToolSpec(
|
|
name="write_file",
|
|
description=(
|
|
"Write text content to a workspace file (overwrite or append). "
|
|
"Path aliases: file, filename, file_path. "
|
|
"If path is omitted but content is provided, writes to tmp/write_<timestamp>.txt."
|
|
),
|
|
parameters={
|
|
"type": "object",
|
|
"properties": {
|
|
"path": {"type": "string", "description": "File path, relative to workspace root."},
|
|
"file": {"type": "string", "description": "Alias for path."},
|
|
"filename": {"type": "string", "description": "Alias for path."},
|
|
"file_path": {"type": "string", "description": "Alias for path."},
|
|
"filepath": {"type": "string", "description": "Alias for path."},
|
|
"name": {"type": "string", "description": "Alias for path (when it looks like a relative file path)."},
|
|
"content": {"type": "string", "description": "Full text content to write."},
|
|
"text": {"type": "string", "description": "Alias for content."},
|
|
"body": {"type": "string", "description": "Alias for content."},
|
|
"mode": {"type": "string", "enum": ["overwrite", "append"], "default": "overwrite"},
|
|
},
|
|
"required": [],
|
|
"additionalProperties": False,
|
|
},
|
|
handler=_handler,
|
|
tags=frozenset({"public", "workspace", "write"}),
|
|
risk_level="high",
|
|
read_only=False,
|
|
)
|
|
|
|
|
|
__all__ = ["write_file_tool"]
|