@ridit/lens 0.2.1 → 0.2.4

package/src/utils/tools/builtins.ts

@@ -0,0 +1,324 @@
+import type { Tool, ToolContext } from "@ridit/lens-sdk";
+import {
+  fetchUrl,
+  searchWeb,
+  runShell,
+  openUrl,
+  readFile,
+  readFolder,
+  grepFiles,
+  writeFile,
+  deleteFile,
+  deleteFolder,
+  generatePdf,
+} from "../../tools";
+
+// ── fetch ─────────────────────────────────────────────────────────────────────
+
+export const fetchTool: Tool<string> = {
+  name: "fetch",
+  description: "load a URL",
+  safe: true,
+  permissionLabel: "fetch",
+  systemPromptEntry: (i) =>
+    `### ${i}. fetch — load a URL\n<fetch>https://example.com</fetch>`,
+  parseInput: (body) => body.replace(/^<|>$/g, "").trim() || null,
+  summariseInput: (url) => url,
+  execute: async (url) => {
+    try {
+      const value = await fetchUrl(url);
+      return { kind: "text", value };
+    } catch (err) {
+      return {
+        kind: "error",
+        value: `Fetch failed: ${err instanceof Error ? err.message : String(err)}`,
+      };
+    }
+  },
+};
+
+// ── shell ─────────────────────────────────────────────────────────────────────
+
+export const shellTool: Tool<string> = {
+  name: "shell",
+  description: "run a terminal command",
+  safe: false,
+  permissionLabel: "run",
+  systemPromptEntry: (i) =>
+    `### ${i}. shell — run a terminal command\n<shell>node -v</shell>`,
+  parseInput: (body) => body || null,
+  summariseInput: (cmd) => cmd,
+  execute: async (cmd, ctx) => {
+    const value = await runShell(cmd, ctx.repoPath);
+    return { kind: "text", value };
+  },
+};
+
+// ── read-file ─────────────────────────────────────────────────────────────────
+
+export const readFileTool: Tool<string> = {
+  name: "read-file",
+  description: "read a file from the repo",
+  safe: true,
+  permissionLabel: "read",
+  systemPromptEntry: (i) =>
+    `### ${i}. read-file — read a file from the repo\n<read-file>src/foo.ts</read-file>`,
+  parseInput: (body) => body || null,
+  summariseInput: (p) => p,
+  execute: (filePath, ctx) => ({
+    kind: "text",
+    value: readFile(filePath, ctx.repoPath),
+  }),
+};
+
+// ── read-folder ───────────────────────────────────────────────────────────────
+
+export const readFolderTool: Tool<string> = {
+  name: "read-folder",
+  description: "list contents of a folder (files + subfolders, one level deep)",
+  safe: true,
+  permissionLabel: "folder",
+  systemPromptEntry: (i) =>
+    `### ${i}. read-folder — list contents of a folder (files + subfolders, one level deep)\n<read-folder>src/components</read-folder>`,
+  parseInput: (body) => body || null,
+  summariseInput: (p) => p,
+  execute: (folderPath, ctx) => ({
+    kind: "text",
+    value: readFolder(folderPath, ctx.repoPath),
+  }),
+};
+
+// ── grep ──────────────────────────────────────────────────────────────────────
+
+interface GrepInput {
+  pattern: string;
+  glob: string;
+}
+
+export const grepTool: Tool<GrepInput> = {
+  name: "grep",
+  description: "search for a pattern across files in the repo",
+  safe: true,
+  permissionLabel: "grep",
+  systemPromptEntry: (i) =>
+    `### ${i}. grep — search for a pattern across files in the repo (cross-platform, no shell needed)\n<grep>\n{"pattern": "ChatRunner", "glob": "src/**/*.tsx"}\n</grep>`,
+  parseInput: (body) => {
+    try {
+      const parsed = JSON.parse(body) as { pattern: string; glob?: string };
+      return { pattern: parsed.pattern, glob: parsed.glob ?? "**/*" };
+    } catch {
+      return { pattern: body, glob: "**/*" };
+    }
+  },
+  summariseInput: ({ pattern, glob }) => `${pattern} — ${glob}`,
+  execute: ({ pattern, glob }, ctx) => ({
+    kind: "text",
+    value: grepFiles(pattern, glob, ctx.repoPath),
+  }),
+};
+
+// ── write-file ────────────────────────────────────────────────────────────────
+
+interface WriteFileInput {
+  path: string;
+  content: string;
+}
+
+export const writeFileTool: Tool<WriteFileInput> = {
+  name: "write-file",
+  description: "create or overwrite a file",
+  safe: false,
+  permissionLabel: "write",
+  systemPromptEntry: (i) =>
+    `### ${i}. write-file — create or overwrite a file\n<write-file>\n{"path": "data/output.csv", "content": "col1,col2\\nval1,val2"}\n</write-file>`,
+  parseInput: (body) => {
+    try {
+      const parsed = JSON.parse(body) as { path: string; content: string };
+      if (!parsed.path) return null;
+      return parsed;
+    } catch {
+      return null;
+    }
+  },
+  summariseInput: ({ path, content }) => `${path} (${content.length} bytes)`,
+  execute: ({ path: filePath, content }, ctx) => ({
+    kind: "text",
+    value: writeFile(filePath, content, ctx.repoPath),
+  }),
+};
+
+// ── delete-file ───────────────────────────────────────────────────────────────
+
+export const deleteFileTool: Tool<string> = {
+  name: "delete-file",
+  description: "permanently delete a single file",
+  safe: false,
+  permissionLabel: "delete",
+  systemPromptEntry: (i) =>
+    `### ${i}. delete-file — permanently delete a single file\n<delete-file>src/old-component.tsx</delete-file>`,
+  parseInput: (body) => body || null,
+  summariseInput: (p) => p,
+  execute: (filePath, ctx) => ({
+    kind: "text",
+    value: deleteFile(filePath, ctx.repoPath),
+  }),
+};
+
+// ── delete-folder ─────────────────────────────────────────────────────────────
+
+export const deleteFolderTool: Tool<string> = {
+  name: "delete-folder",
+  description: "permanently delete a folder and all its contents",
+  safe: false,
+  permissionLabel: "delete folder",
+  systemPromptEntry: (i) =>
+    `### ${i}. delete-folder — permanently delete a folder and all its contents\n<delete-folder>src/legacy</delete-folder>`,
+  parseInput: (body) => body || null,
+  summariseInput: (p) => p,
+  execute: (folderPath, ctx) => ({
+    kind: "text",
+    value: deleteFolder(folderPath, ctx.repoPath),
+  }),
+};
+
+// ── open-url ──────────────────────────────────────────────────────────────────
+
+export const openUrlTool: Tool<string> = {
+  name: "open-url",
+  description: "open a URL in the user's default browser",
+  safe: true,
+  permissionLabel: "open",
+  systemPromptEntry: (i) =>
+    `### ${i}. open-url — open a URL in the user's default browser\n<open-url>https://github.com/owner/repo</open-url>`,
+  parseInput: (body) => body.replace(/^<|>$/g, "").trim() || null,
+  summariseInput: (url) => url,
+  execute: (url) => ({ kind: "text", value: openUrl(url) }),
+};
+
+// ── generate-pdf ──────────────────────────────────────────────────────────────
+
+interface GeneratePdfInput {
+  filePath: string;
+  content: string;
+}
+
+export const generatePdfTool: Tool<GeneratePdfInput> = {
+  name: "generate-pdf",
+  description: "generate a PDF file from markdown-style content",
+  safe: false,
+  permissionLabel: "pdf",
+  systemPromptEntry: (i) =>
+    `### ${i}. generate-pdf — generate a PDF file from markdown-style content\n<generate-pdf>\n{"path": "output/report.pdf", "content": "# Title\\n\\nSome body text."}\n</generate-pdf>`,
+  parseInput: (body) => {
+    try {
+      const parsed = JSON.parse(body) as {
+        path?: string;
+        filePath?: string;
+        content?: string;
+      };
+      return {
+        filePath: parsed.path ?? parsed.filePath ?? "output.pdf",
+        content: parsed.content ?? "",
+      };
+    } catch {
+      return null;
+    }
+  },
+  summariseInput: ({ filePath }) => filePath,
+  execute: ({ filePath, content }, ctx) => ({
+    kind: "text",
+    value: generatePdf(filePath, content, ctx.repoPath),
+  }),
+};
+
+// ── search ────────────────────────────────────────────────────────────────────
+
+export const searchTool: Tool<string> = {
+  name: "search",
+  description: "search the internet for anything you are unsure about",
+  safe: true,
+  permissionLabel: "search",
+  systemPromptEntry: (i) =>
+    `### ${i}. search — search the internet for anything you are unsure about\n<search>how to use React useEffect cleanup function</search>`,
+  parseInput: (body) => body || null,
+  summariseInput: (q) => `"${q}"`,
+  execute: async (query) => {
+    try {
+      const value = await searchWeb(query);
+      return { kind: "text", value };
+    } catch (err) {
+      return {
+        kind: "error",
+        value: `Search failed: ${err instanceof Error ? err.message : String(err)}`,
+      };
+    }
+  },
+};
+
+// ── clone ─────────────────────────────────────────────────────────────────────
+
+export const cloneTool: Tool<string> = {
+  name: "clone",
+  description: "clone a GitHub repo so you can explore and discuss it",
+  safe: false,
+  permissionLabel: "clone",
+  systemPromptEntry: (i) =>
+    `### ${i}. clone — clone a GitHub repo so you can explore and discuss it\n<clone>https://github.com/owner/repo</clone>`,
+  parseInput: (body) => body.replace(/^<|>$/g, "").trim() || null,
+  summariseInput: (url) => url,
+  // Clone is handled specially by ChatRunner (it triggers a UI flow),
+  // so execute here is just a fallback that should never run.
+  execute: (repoUrl) => ({
+    kind: "text",
+    value: `Clone of ${repoUrl} was handled by the UI.`,
+  }),
+};
+
+// ── changes ───────────────────────────────────────────────────────────────────
+
+export interface ChangesInput {
+  summary: string;
+  patches: { path: string; content: string; isNew: boolean }[];
+}
+
+export const changesTool: Tool<ChangesInput> = {
+  name: "changes",
+  description: "propose code edits (shown as a diff for user approval)",
+  safe: false,
+  permissionLabel: "changes",
+  systemPromptEntry: (i) =>
+    `### ${i}. changes — propose code edits (shown as a diff for user approval)\n<changes>\n{"summary": "what changed and why", "patches": [{"path": "src/foo.ts", "content": "COMPLETE file content", "isNew": false}]}\n</changes>`,
+  parseInput: (body) => {
+    try {
+      return JSON.parse(body) as ChangesInput;
+    } catch {
+      return null;
+    }
+  },
+  summariseInput: ({ summary }) => summary,
+  // changes is handled specially by ChatRunner (diff preview UI).
+  execute: ({ summary }) => ({
+    kind: "text",
+    value: `Changes proposed: ${summary}`,
+  }),
+};
+
+// ── registerBuiltins ──────────────────────────────────────────────────────────
+
+import { registry } from "./registry";
+
+export function registerBuiltins(): void {
+  registry.register(fetchTool);
+  registry.register(shellTool);
+  registry.register(readFileTool);
+  registry.register(readFolderTool);
+  registry.register(grepTool);
+  registry.register(writeFileTool);
+  registry.register(deleteFileTool);
+  registry.register(deleteFolderTool);
+  registry.register(openUrlTool);
+  registry.register(generatePdfTool);
+  registry.register(searchTool);
+  registry.register(cloneTool);
+  registry.register(changesTool);
+}

package/src/utils/tools/registry.ts

@@ -0,0 +1,63 @@
+// ── Tool Plugin System ────────────────────────────────────────────────────────
+//
+// To create a new tool:
+//
+//   1. Implement the Tool interface
+//   2. Call registry.register(myTool) before the app starts
+//
+// External addon example:
+//
+//   import { registry } from "lens/tools/registry";
+//   registry.register({ name: "my-tool", ... });
+
+import type { Tool } from "@ridit/lens-sdk";
+
+// ── Registry ──────────────────────────────────────────────────────────────────
+
+class ToolRegistry {
+  private tools = new Map<string, Tool<unknown>>();
+
+  register<T>(tool: Tool<T>): void {
+    if (this.tools.has(tool.name)) {
+      console.warn(`[ToolRegistry] Overwriting existing tool: "${tool.name}"`);
+    }
+    this.tools.set(tool.name, tool as Tool<unknown>);
+  }
+
+  unregister(name: string): void {
+    this.tools.delete(name);
+  }
+
+  get(name: string): Tool<unknown> | undefined {
+    return this.tools.get(name);
+  }
+
+  all(): Tool<unknown>[] {
+    return Array.from(this.tools.values());
+  }
+
+  names(): string[] {
+    return Array.from(this.tools.keys());
+  }
+
+  /**
+   * Build the TOOLS section of the system prompt from all registered tools.
+   */
+  buildSystemPromptSection(): string {
+    const lines: string[] = ["## TOOLS\n"];
+    lines.push(
+      "You have exactly " +
+        this.tools.size +
+        " tools. To use a tool you MUST wrap it in the exact XML tags shown below — no other format will work.\n",
+    );
+    let i = 1;
+    for (const tool of this.tools.values()) {
+      lines.push(tool.systemPromptEntry(i++));
+    }
+    return lines.join("\n");
+  }
+}
+
+export const registry = new ToolRegistry();
+
+(globalThis as any).__lens_registry = registry;

package/hello.py

@@ -1,51 +0,0 @@
-# hello.py
-# A simple script that reads a text file, makes an HTTP request,
-# and prints the result in a friendly way.
-
-import json
-import pathlib
-import sys
-from urllib.request import urlopen
-
-def read_local_file(path: str) -> str:
-    """Read the contents of a file and return it as a string."""
-    file_path = pathlib.Path(path)
-    if not file_path.is_file():
-        raise FileNotFoundError(f"❌ File not found: {path}")
-    return file_path.read_text(encoding="utf-8")
-
-def fetch_json(url: str) -> dict:
-    """Fetch JSON from a URL and decode it into a Python dict."""
-    with urlopen(url) as response:
-        if response.status != 200:
-            raise RuntimeError(f"❌ HTTP {response.status} from {url}")
-        data = response.read()
-        return json.loads(data)
-
-def main():
-    # 1️⃣ Read a local file (optional – you can comment this out)
-    try:
-        local_content = read_local_file("example.txt")
-        print("📄 Contents of example.txt:")
-        print(local_content)
-    except FileNotFoundError:
-        print("⚠️ example.txt not found – skipping that step.")
-
-    # 2️⃣ Fetch some JSON data from a public API
-    url = "https://api.github.com/repos/python/cpython"
-    print(f"\n🌐 Fetching data from {url} …")
-    repo_info = fetch_json(url)
-
-    # 3️⃣ Extract a couple of useful fields and display them
-    name = repo_info.get("name")
-    stars = repo_info.get("stargazers_count")
-    description = repo_info.get("description")
-    print("\n🔎 Repository info:")
-    print(f"• Name: {name}")
-    print(f"• Stars: {stars:,}")          # adds commas for readability
-    print(f"• Description: {description}")
-
-    return 0
-
-if __name__ == "__main__":
-    sys.exit(main())