Create agent/fs.py

agent/fs.py

@@ -0,0 +1,62 @@
+import os
+
+BASE_DIR = "/workspace/agent"
+
+def safe_path(path: str) -> str:
+    full = os.path.abspath(os.path.join(BASE_DIR, path))
+    if not full.startswith(BASE_DIR):
+        raise PermissionError("Access denied")
+    return full
+
+
+def init_workspace():
+    os.makedirs(safe_path("projects"), exist_ok=True)
+    os.makedirs(safe_path("temp"), exist_ok=True)
+    os.makedirs(safe_path("logs"), exist_ok=True)
+
+
+def list_projects():
+    base = safe_path("projects")
+    return [
+        d for d in os.listdir(base)
+        if os.path.isdir(os.path.join(base, d))
+    ]
+
+
+def create_project(name):
+    path = safe_path(f"projects/{name}")
+    os.makedirs(path, exist_ok=True)
+    return {"created": name}
+
+
+def read_file(path, start=0, limit=2000):
+    file_path = safe_path(path)
+    with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
+        f.seek(start)
+        return f.read(limit)
+
+
+def write_file(path, content):
+    file_path = safe_path(path)
+    os.makedirs(os.path.dirname(file_path), exist_ok=True)
+    with open(file_path, "w", encoding="utf-8") as f:
+        f.write(content)
+
+
+def append_file(path, content):
+    file_path = safe_path(path)
+    with open(file_path, "a", encoding="utf-8") as f:
+        f.write(content)
+
+
+def tree(path="projects", depth=2):
+    root = safe_path(path)
+    result = {}
+
+    for base, dirs, files in os.walk(root):
+        level = base.replace(root, "").count(os.sep)
+        if level >= depth:
+            continue
+        result[base.replace(BASE_DIR, "")] = files
+
+    return result