mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 04:40:45 +08:00
oliver
e06184cc2f
将 run_command 在未传 cwd 时改为默认 data/workspace，并将 write_file 调整为相对路径写入 data/workspace、绝对路径直写（经路径守卫校验），同步更新测试与文档。 Made-with: Cursor
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from oclaw.runtime.tools.base import ToolSpec
|
|
from oclaw.runtime.tools.path_guard import resolve_workspace_path
|
|
|
|
|
|
def write_file_tool() -> ToolSpec:
|
|
def _normalize_write_path(path: str) -> str:
|
|
raw = str(path or "").strip().strip('"').strip("'")
|
|
if not raw:
|
|
raise ValueError("path_required")
|
|
p = Path(raw)
|
|
if p.is_absolute():
|
|
return raw
|
|
rel = raw.lstrip("./\\")
|
|
if not rel:
|
|
raise ValueError("path_required")
|
|
return str(Path("data") / "workspace" / rel)
|
|
|
|
def _handler(args: dict[str, Any]) -> dict[str, Any]:
|
|
path = str(args.get("path") or "").strip()
|
|
content = str(args.get("content") or "")
|
|
mode = str(args.get("mode") or "overwrite").strip().lower()
|
|
try:
|
|
normalized = _normalize_write_path(path)
|
|
except ValueError as exc:
|
|
return {"ok": False, "error": str(exc)}
|
|
p = resolve_workspace_path(normalized)
|
|
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, encoding="utf-8")
|
|
else:
|
|
p.write_text(content, encoding="utf-8")
|
|
return {"ok": True, "path": str(p), "bytes": p.stat().st_size}
|
|
|
|
return ToolSpec(
|
|
name="write_file",
|
|
description="Write text content to a workspace file (overwrite or append).",
|
|
parameters={
|
|
"type": "object",
|
|
"properties": {
|
|
"path": {"type": "string", "description": "File path, relative to workspace root."},
|
|
"content": {"type": "string", "description": "Full text content to write."},
|
|
"mode": {"type": "string", "enum": ["overwrite", "append"], "default": "overwrite"},
|
|
},
|
|
"required": ["path", "content"],
|
|
"additionalProperties": False,
|
|
},
|
|
handler=_handler,
|
|
tags=frozenset({"public", "workspace", "write"}),
|
|
risk_level="high",
|
|
read_only=False,
|
|
)
|
|
|
|
|
|
__all__ = ["write_file_tool"]
|