nexforge-cli 0.1.0__py3-none-any.whl

nexforge/tools/search_files.py

@@ -0,0 +1,48 @@
+"""search_files 工具:按文件名匹配(子串或 glob)。"""
+
+from __future__ import annotations
+
+from nexforge.tools._paths import resolve, truncate
+from nexforge.tools.base import BaseTool, ToolResult
+
+_SKIP_DIRS = {".git", "__pycache__", ".venv", "node_modules", ".mypy_cache",
+              ".pytest_cache", "dist", "build", ".nexforge"}
+
+
+class SearchFilesTool(BaseTool):
+    name = "search_files"
+    description = "按文件名查找(不区分大小写的子串匹配)。返回匹配的相对路径列表。"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "pattern": {"type": "string", "description": "文件名子串"},
+            "path": {"type": "string", "description": "起始目录,默认当前目录"},
+        },
+        "required": ["pattern"],
+    }
+
+    def preview(self, args: dict) -> str:
+        return f"search_files({args.get('pattern', '')!r})"
+
+    def run(self, pattern: str, path: str = ".") -> ToolResult:
+        base = resolve(path)
+        if not base.is_dir():
+            return ToolResult(ok=False, error=f"目录不存在:{path}")
+        needle = pattern.lower()
+        matches: list[str] = []
+        for fp in base.rglob("*"):
+            if not fp.is_file():
+                continue
+            if any(part in _SKIP_DIRS for part in fp.parts):
+                continue
+            if needle in fp.name.lower():
+                matches.append(fp.relative_to(base).as_posix())
+                if len(matches) >= 500:
+                    break
+        if not matches:
+            return ToolResult(ok=True, output="(无匹配)", summary=f"{pattern} — 0")
+        return ToolResult(
+            ok=True,
+            output=truncate("\n".join(sorted(matches))),
+            summary=f"{pattern} — {len(matches)} 个",
+        )

nexforge/tools/shell.py

@@ -0,0 +1,60 @@
+"""shell 工具:执行终端命令(带超时与输出截断)。"""
+
+from __future__ import annotations
+
+import subprocess
+
+from nexforge.tools._paths import truncate
+from nexforge.tools.base import BaseTool, ToolResult
+
+# 简单危险命令拦截(M2 会做更完善的沙箱)
+_DANGER = ("rm -rf /", "rm -rf /*", ":(){", "mkfs", "dd if=", "> /dev/sda")
+
+
+class ShellTool(BaseTool):
+    name = "run_shell"
+    description = "在当前工作目录执行一条 shell 命令,返回 stdout+stderr。默认超时 30s。"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "command": {"type": "string", "description": "要执行的命令"},
+            "timeout": {"type": "integer", "description": "超时秒数,默认 30"},
+        },
+        "required": ["command"],
+    }
+    requires_approval = True
+    parallel_safe = False
+
+    def preview(self, args: dict) -> str:
+        cmd = args.get("command", "")
+        return f"run_shell($ {cmd[:60]})"
+
+    def run(self, command: str, timeout: int = 30) -> ToolResult:
+        low = command.strip().lower()
+        if any(d in low for d in _DANGER):
+            return ToolResult(ok=False, error="命令被安全策略拦截(高危操作)")
+        try:
+            proc = subprocess.run(
+                command,
+                shell=True,
+                stdin=subprocess.DEVNULL,
+                capture_output=True,
+                text=True,
+                timeout=timeout,
+                encoding="utf-8",
+                errors="replace",
+            )
+        except subprocess.TimeoutExpired:
+            return ToolResult(ok=False, error=f"命令超时({timeout}s)")
+        except Exception as exc:
+            return ToolResult(ok=False, error=f"执行失败:{exc}")
+
+        out = (proc.stdout or "") + (proc.stderr or "")
+        out = truncate(out.strip()) or "(无输出)"
+        body = f"[exit {proc.returncode}]\n{out}"
+        return ToolResult(
+            ok=proc.returncode == 0,
+            output=body,
+            error="" if proc.returncode == 0 else f"退出码 {proc.returncode}\n{out}",
+            summary=f"$ {command[:50]} → exit {proc.returncode}",
+        )

nexforge/tools/web.py

@@ -0,0 +1,40 @@
+"""web_fetch 工具:HTTP GET(用 httpx,截断)。"""
+
+from __future__ import annotations
+
+from nexforge.tools._paths import truncate
+from nexforge.tools.base import BaseTool, ToolResult
+
+
+class WebFetchTool(BaseTool):
+    name = "web_fetch"
+    description = "HTTP GET 抓取一个 URL,返回响应正文(截断)。用于读文档/API。"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "url": {"type": "string", "description": "http(s) URL"},
+            "timeout": {"type": "integer", "description": "超时秒数,默认 20"},
+        },
+        "required": ["url"],
+    }
+
+    def preview(self, args: dict) -> str:
+        return f"web_fetch({args.get('url', '')[:50]})"
+
+    def run(self, url: str, timeout: int = 20) -> ToolResult:
+        if not url.startswith(("http://", "https://")):
+            return ToolResult(ok=False, error="仅支持 http(s) URL")
+        try:
+            import httpx
+
+            resp = httpx.get(url, timeout=timeout, follow_redirects=True)
+        except Exception as exc:
+            return ToolResult(ok=False, error=f"请求失败:{exc}")
+        ctype = resp.headers.get("content-type", "")
+        body = resp.text
+        return ToolResult(
+            ok=resp.status_code < 400,
+            output=truncate(f"[{resp.status_code} {ctype}]\n{body}"),
+            error="" if resp.status_code < 400 else f"HTTP {resp.status_code}",
+            summary=f"GET {url[:40]} → {resp.status_code}",
+        )

nexforge/tools/write_file.py

@@ -0,0 +1,40 @@
+"""write_file 工具:整文件覆写 / 新建。"""
+
+from __future__ import annotations
+
+from nexforge.tools._paths import resolve
+from nexforge.tools.base import BaseTool, ToolResult
+
+
+class WriteFileTool(BaseTool):
+    name = "write_file"
+    description = "把内容整体写入文件(覆盖已有内容或新建)。谨慎使用,会覆盖原文件。"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "path": {"type": "string", "description": "目标文件路径"},
+            "content": {"type": "string", "description": "文件完整内容"},
+        },
+        "required": ["path", "content"],
+    }
+    requires_approval = True
+    parallel_safe = False
+
+    def preview(self, args: dict) -> str:
+        return f"write_file({args.get('path', '')})"
+
+    def run(self, path: str, content: str) -> ToolResult:
+        fp = resolve(path)
+        existed = fp.exists()
+        try:
+            fp.parent.mkdir(parents=True, exist_ok=True)
+            fp.write_text(content, encoding="utf-8")
+        except Exception as exc:
+            return ToolResult(ok=False, error=f"写入失败:{exc}")
+        verb = "覆写" if existed else "新建"
+        lines = content.count("\n") + 1
+        return ToolResult(
+            ok=True,
+            output=f"已{verb} {path}({lines} 行)",
+            summary=f"{verb} {path}",
+        )

nexforge/ui/__init__.py

@@ -0,0 +1 @@
+"""表现层:Textual 组件与屏幕。"""

nexforge/ui/screens/__init__.py

@@ -0,0 +1 @@
+"""Textual 屏幕。"""