@ridit/lens 0.3.4 → 0.3.6

package/src/components/chat/hooks/useCommandHandlers.ts

@@ -0,0 +1,327 @@
+import type { Message } from "../../../types/chat";
+import {
+  saveChat,
+  loadChat,
+  listChats,
+  deleteChat,
+} from "../../../utils/chatHistory";
+import {
+  clearRepoMemory,
+  addMemory,
+  deleteMemory,
+  listMemories,
+} from "../../../utils/memory";
+
+export const COMMANDS = [
+  { cmd: "/timeline", desc: "browse commit history" },
+  { cmd: "/clear history", desc: "wipe session memory for this repo" },
+  { cmd: "/review", desc: "review current codebase" },
+  { cmd: "/auto", desc: "toggle auto-approve for read/search tools" },
+  {
+    cmd: "/auto --force-all",
+    desc: "auto-approve ALL tools including shell and writes (⚠ dangerous)",
+  },
+  { cmd: "/chat", desc: "chat history commands" },
+  { cmd: "/chat list", desc: "list saved chats for this repo" },
+  { cmd: "/chat load", desc: "load a saved chat by name" },
+  { cmd: "/chat rename", desc: "rename the current chat" },
+  { cmd: "/chat delete", desc: "delete a saved chat by name" },
+  { cmd: "/memory", desc: "memory commands" },
+  { cmd: "/memory list", desc: "list all memories for this repo" },
+  { cmd: "/memory add", desc: "add a memory" },
+  { cmd: "/memory delete", desc: "delete a memory by id" },
+  { cmd: "/memory clear", desc: "clear all memories for this repo" },
+];
+
+type CommandContext = {
+  repoPath: string;
+  allMessages: Message[];
+  autoApprove: boolean;
+  forceApprove: boolean;
+  chatName: string | null;
+  chatNameRef: React.MutableRefObject<string | null>;
+  setShowTimeline: (v: boolean) => void;
+  setShowReview: (v: boolean) => void;
+  setShowForceWarning: (v: boolean) => void;
+  setForceApprove: (v: boolean) => void;
+  setAutoApprove: (v: boolean) => void;
+  setAllMessages: (fn: (prev: Message[]) => Message[]) => void;
+  setCommitted: (fn: (prev: Message[]) => Message[]) => void;
+  setRecentChats: (fn: (prev: string[]) => string[]) => void;
+  updateChatName: (name: string) => void;
+};
+
+import React from "react";
+
+function pushMsg(
+  msg: Message,
+  setCommitted: CommandContext["setCommitted"],
+  setAllMessages: CommandContext["setAllMessages"],
+) {
+  setCommitted((prev) => [...prev, msg]);
+  setAllMessages((prev) => [...prev, msg]);
+}
+
+function makeMsg(content: string): Message {
+  return { role: "assistant", content, type: "text" };
+}
+
+/**
+ * Returns true if the command was handled, false if it should fall through
+ * to the normal message flow.
+ */
+export function handleCommand(text: string, ctx: CommandContext): boolean {
+  const t = text.trim().toLowerCase();
+
+  if (t === "/timeline") {
+    ctx.setShowTimeline(true);
+    return true;
+  }
+
+  if (t === "/review") {
+    ctx.setShowReview(true);
+    return true;
+  }
+
+  if (t === "/auto --force-all") {
+    if (ctx.forceApprove) {
+      ctx.setForceApprove(false);
+      ctx.setAutoApprove(false);
+      pushMsg(
+        makeMsg("Force-all mode OFF — tools will ask for permission again."),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+    } else {
+      ctx.setShowForceWarning(true);
+    }
+    return true;
+  }
+
+  if (t === "/auto") {
+    if (ctx.forceApprove) {
+      ctx.setForceApprove(false);
+      ctx.setAutoApprove(true);
+      pushMsg(
+        makeMsg(
+          "Force-all mode OFF — switched to normal auto-approve (safe tools only).",
+        ),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    const next = !ctx.autoApprove;
+    ctx.setAutoApprove(next);
+    pushMsg(
+      makeMsg(
+        next
+          ? "Auto-approve ON — safe tools (read, search, fetch) will run without asking."
+          : "Auto-approve OFF — all tools will ask for permission.",
+      ),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t === "/clear history") {
+    clearRepoMemory(ctx.repoPath);
+    pushMsg(
+      makeMsg("History cleared for this repo."),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t === "/chat") {
+    pushMsg(
+      makeMsg(
+        "Chat commands: `/chat list` · `/chat load <n>` · `/chat rename <n>` · `/chat delete <n>`",
+      ),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t.startsWith("/chat rename")) {
+    const parts = text.trim().split(/\s+/);
+    const newName = parts.slice(2).join("-");
+    if (!newName) {
+      pushMsg(
+        makeMsg("Usage: `/chat rename <new-name>`"),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    const oldName = ctx.chatNameRef.current;
+    if (oldName) deleteChat(oldName);
+    ctx.updateChatName(newName);
+    saveChat(newName, ctx.repoPath, ctx.allMessages);
+    ctx.setRecentChats((prev) =>
+      [newName, ...prev.filter((n) => n !== newName && n !== oldName)].slice(
+        0,
+        10,
+      ),
+    );
+    pushMsg(
+      makeMsg(`Chat renamed to **${newName}**.`),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t.startsWith("/chat delete")) {
+    const parts = text.trim().split(/\s+/);
+    const name = parts.slice(2).join("-");
+    if (!name) {
+      pushMsg(
+        makeMsg("Usage: `/chat delete <n>`"),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    const deleted = deleteChat(name);
+    if (!deleted) {
+      pushMsg(
+        makeMsg(`Chat **${name}** not found.`),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    if (ctx.chatNameRef.current === name) {
+      ctx.chatNameRef.current = null;
+      ctx.updateChatName("");
+    }
+    ctx.setRecentChats((prev) => prev.filter((n) => n !== name));
+    pushMsg(
+      makeMsg(`Chat **${name}** deleted.`),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t === "/chat list") {
+    const chats = listChats(ctx.repoPath);
+    const content =
+      chats.length === 0
+        ? "No saved chats for this repo yet."
+        : `Saved chats:\n\n${chats
+            .map(
+              (c) =>
+                `- **${c.name}** · ${c.userMessageCount} messages · ${new Date(c.savedAt).toLocaleString()}`,
+            )
+            .join("\n")}`;
+    pushMsg(makeMsg(content), ctx.setCommitted, ctx.setAllMessages);
+    return true;
+  }
+
+  if (t.startsWith("/chat load")) {
+    const parts = text.trim().split(/\s+/);
+    const name = parts.slice(2).join("-");
+    if (!name) {
+      const chats = listChats(ctx.repoPath);
+      const content =
+        chats.length === 0
+          ? "No saved chats found."
+          : `Specify a chat name. Recent chats:\n\n${chats
+              .slice(0, 10)
+              .map((c) => `- **${c.name}**`)
+              .join("\n")}`;
+      pushMsg(makeMsg(content), ctx.setCommitted, ctx.setAllMessages);
+      return true;
+    }
+    const saved = loadChat(name);
+    if (!saved) {
+      pushMsg(
+        makeMsg(
+          `Chat **${name}** not found. Use \`/chat list\` to see saved chats.`,
+        ),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    ctx.updateChatName(name);
+
+    ctx.setAllMessages(() => saved.messages);
+    ctx.setCommitted(() => saved.messages);
+    const notice = makeMsg(
+      `Loaded chat **${name}** · ${saved.userMessageCount} messages · saved ${new Date(saved.savedAt).toLocaleString()}`,
+    );
+    ctx.setCommitted((prev) => [...prev, notice]);
+    ctx.setAllMessages((prev) => [...prev, notice]);
+    return true;
+  }
+
+  if (t === "/memory list" || t === "/memory") {
+    const mems = listMemories(ctx.repoPath);
+    const content =
+      mems.length === 0
+        ? "No memories stored for this repo yet."
+        : `Memories for this repo:\n\n${mems.map((m) => `- [${m.id}] ${m.content}`).join("\n")}`;
+    pushMsg(makeMsg(content), ctx.setCommitted, ctx.setAllMessages);
+    return true;
+  }
+
+  if (t.startsWith("/memory add")) {
+    const content = text.trim().slice("/memory add".length).trim();
+    if (!content) {
+      pushMsg(
+        makeMsg("Usage: `/memory add <content>`"),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    const mem = addMemory(content, ctx.repoPath);
+    pushMsg(
+      makeMsg(`Memory saved **[${mem.id}]**: ${mem.content}`),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t.startsWith("/memory delete")) {
+    const id = text.trim().split(/\s+/)[2];
+    if (!id) {
+      pushMsg(
+        makeMsg("Usage: `/memory delete <id>`"),
+        ctx.setCommitted,
+        ctx.setAllMessages,
+      );
+      return true;
+    }
+    const deleted = deleteMemory(id, ctx.repoPath);
+    pushMsg(
+      makeMsg(
+        deleted
+          ? `Memory **[${id}]** deleted.`
+          : `Memory **[${id}]** not found.`,
+      ),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  if (t === "/memory clear") {
+    clearRepoMemory(ctx.repoPath);
+    pushMsg(
+      makeMsg("All memories cleared for this repo."),
+      ctx.setCommitted,
+      ctx.setAllMessages,
+    );
+    return true;
+  }
+
+  return false;
+}

package/src/components/watch/{WatchRunner.tsx → RunRunner.tsx}

@@ -354,7 +354,7 @@ type ActiveInvestigation = {
   startTime: number;
 };
 
-export function WatchRunner({
+export function RunRunner({
   cmd,
   repoPath,
   clean,
@@ -460,7 +460,6 @@ export function WatchRunner({
           lensContext = `Overview: ${lensFile.overview}
 
 Important folders: ${lensFile.importantFolders.join(", ")}
-${lensFile.securityIssues.length > 0 ? `\nKnown security issues:\n${lensFile.securityIssues.map((s) => `- ${s}`).join("\n")}` : ""}
 ${lensFile.suggestions.length > 0 ? `\nProject suggestions:\n${lensFile.suggestions.map((s) => `- ${s}`).join("\n")}` : ""}`;
         }
       }

package/src/index.tsx

@@ -7,7 +7,7 @@ import { InitCommand } from "./commands/provider";
 import { ReviewCommand } from "./commands/review";
 import { TaskCommand } from "./commands/task";
 import { ChatCommand } from "./commands/chat";
-import { WatchCommand } from "./commands/watch";
+import { RunCommand } from "./commands/run";
 import { TimelineCommand } from "./commands/timeline";
 import { CommitCommand } from "./commands/commit";
 import { registerBuiltins } from "./utils/tools/builtins";
@@ -19,8 +19,7 @@ await loadAddons();
 const program = new Command();
 
 program
-  .command("stalk <url>")
-  .alias("repo")
+  .command("repo <url>")
   .description("Analyze a remote repository")
   .action((url) => {
     render(<RepoCommand url={url} />);
@@ -34,16 +33,14 @@ program
   });
 
 program
-  .command("judge [path]")
-  .alias("review")
+  .command("review [path]")
   .description("Review a local codebase")
   .action((inputPath) => {
     render(<ReviewCommand path={inputPath ?? "."} />);
   });
 
 program
-  .command("cook <text>")
-  .alias("task")
+  .command("task <text>")
   .description("Apply a natural language change to the codebase")
   .option("-p, --path <path>", "Path to the repo", ".")
   .action((text: string, opts: { path: string }) => {
@@ -51,8 +48,7 @@ program
   });
 
 program
-  .command("vibe")
-  .alias("chat")
+  .command("chat")
   .description("Chat with your codebase — ask questions or make changes")
   .option("-p, --path <path>", "Path to the repo", ".")
   .action((opts: { path: string }) => {
@@ -60,8 +56,7 @@ program
   });
 
 program
-  .command("history")
-  .alias("timeline")
+  .command("timeline")
   .description(
     "Explore your code history — see commits, changes, and evolution",
   )
@@ -71,8 +66,7 @@ program
   });
 
 program
-  .command("crimes [files...]")
-  .alias("commit")
+  .command("commit [files...]")
   .description(
     "Generate a smart conventional commit message from staged changes or specific files",
   )
@@ -109,9 +103,10 @@ program
   );
 
 program
-  .command("watch <cmd>")
-  .alias("spy")
-  .description("Watch a dev command and get AI suggestions for errors")
+  .command("run <cmd>")
+  .description(
+    "Run your dev server. Lens detects and fixes errors automatically",
+  )
   .option("-p, --path <path>", "Path to the repo", ".")
   .option("--clean", "Only show AI suggestions, hide raw logs")
   .option("--fix-all", "Auto-apply fixes as errors are detected")
@@ -129,7 +124,7 @@ program
       },
     ) => {
       render(
-        <WatchCommand
+        <RunCommand
           cmd={cmd}
           path={opts.path}
           clean={opts.clean ?? false}

package/src/prompts/system.ts

@@ -1,4 +1,5 @@
 import type { ImportantFile } from "../types/repo";
+import type { Intent } from "../utils/intentClassifier";
 
 export function buildSystemPrompt(
   files: ImportantFile[],
@@ -12,25 +13,25 @@ export function buildSystemPrompt(
   const tools = toolsSection ?? BUILTIN_TOOLS_SECTION;
 
   return `You are an expert software engineer assistant with access to the user's codebase and tools.
-
+ 
 ${tools}
-
+ 
 ## MEMORY OPERATIONS
-
+ 
 You can save and delete memories at any time by emitting these tags alongside your normal response.
 They are stripped before display — the user will not see the raw tags.
-
+ 
 ### memory-add — save something important to long-term memory for this repo
 <memory-add>User prefers TypeScript strict mode in all new files</memory-add>
-
+ 
 ### memory-delete — delete a memory by its ID (shown in brackets like [abc123])
 <memory-delete>abc123</memory-delete>
-
+ 
 Use memory-add when the user asks you to remember something, or when you learn something project-specific that would be useful in future sessions.
 Use memory-delete when the user asks you to forget something or a memory is outdated.
-
+ 
 ## RULES
-
+ 
 1. ONE tool per response — emit the XML tag, then stop. Never chain tools in one response except when scaffolding (see below).
 2. NEVER call a tool more than once for the same path in a session. If write-file or shell returned a result, it succeeded. Move on immediately.
 3. NEVER write the same file twice in one session. One write per file, period. If you already wrote it, it is done.
@@ -48,15 +49,15 @@ Use memory-delete when the user asks you to forget something or a memory is outd
 15. When explaining how to use a tool in text, use [tag] bracket notation — NEVER emit a real XML tool tag as part of an explanation.
 16. NEVER use markdown formatting in plain text responses — no bold, no headings, no bullet points. Only use fenced code blocks when showing actual code.
 17. When scaffolding multiple files, emit ONE write-file tag per response and wait for the result before writing the next file.
-
+ 
 ## ADDON FORMAT
-
+ 
 All addons use defineTool from @ridit/lens-sdk. The ONLY correct format is:
-
+ 
 \`\`\`js
 const { defineTool } = require("@ridit/lens-sdk");
 const { execSync } = require("child_process");
-
+ 
 defineTool({
   name: "tool-name",
   description: "what it does",
@@ -72,26 +73,146 @@ defineTool({
   },
 });
 \`\`\`
-
+ 
 NEVER use module.exports, registerTool, ctx.tools.shell, or any other format. See addons/run-tests.js for a full working example.
-
+ 
 ## SCAFFOLDING
-
+ 
 When creating multiple files, emit ONE write-file per response and wait for each result:
-
+ 
 <write-file>
 {"path": "myapp/package.json", "content": "..."}
 </write-file>
-
+ 
 Wait for result, then emit the next file. Never chain write-file tags when content is complex.
-
+ 
 ## CODEBASE
-
+ 
 ${fileList.length > 0 ? fileList : "(no files indexed)"}
-
+ 
 ${memorySummary}`;
 }
 
+export function buildBuiltinToolsSection(intent: Intent = "any"): string {
+  const isReadonly = intent === "readonly";
+
+  const readTools = `### 1. fetch — load a URL
+<fetch>https://example.com</fetch>
+ 
+### 2. read-file — read a single file from the repo
+<read-file>src/foo.ts</read-file>
+ 
+### 3. read-files — read multiple files at once
+<read-files>
+["src/foo.ts", "src/bar.ts"]
+</read-files>
+ 
+### 4. read-folder — list contents of a folder (one level deep)
+<read-folder>src/components</read-folder>
+ 
+### 5. grep — search for a pattern across files
+<grep>
+{"pattern": "ChatRunner", "glob": "src/**/*.tsx"}
+</grep>
+ 
+### 6. search — search the internet
+<search>how to use React useEffect cleanup</search>`;
+
+  const writeTools = `### 7. shell — run a terminal command (NOT for filesystem inspection)
+<shell>node -v</shell>
+ 
+### 8. write-file — create or overwrite a file (COMPLETE content only)
+<write-file>
+{"path": "data/output.csv", "content": "col1,col2\\nval1,val2"}
+</write-file>
+ 
+### 9. delete-file — permanently delete a single file
+<delete-file>src/old-component.tsx</delete-file>
+ 
+### 10. delete-folder — permanently delete a folder and all its contents
+<delete-folder>src/legacy</delete-folder>
+ 
+### 11. open-url — open a URL in the user's default browser
+<open-url>https://github.com/owner/repo</open-url>
+ 
+### 12. generate-pdf — generate a PDF from markdown-style content
+<generate-pdf>
+{"path": "output/report.pdf", "content": "# Title\\n\\nBody text."}
+</generate-pdf>
+ 
+### 13. clone — clone a GitHub repo
+<clone>https://github.com/owner/repo</clone>
+ 
+### 14. changes — propose code edits shown as a diff for user approval
+<changes>
+{"summary": "what changed and why", "patches": [{"path": "src/foo.ts", "content": "COMPLETE file content", "isNew": false}]}
+</changes>`;
+
+  if (isReadonly) {
+    return `## TOOLS
+ 
+You have 6 tools available for this read-only request. Do NOT attempt to write, delete, or run shell commands — those tools are not available right now.
+ 
+${readTools}`;
+  }
+
+  return `## TOOLS
+ 
+You have exactly 14 tools. To use a tool you MUST wrap it in the exact XML tags shown below — no other format will work.
+ 
+### 1. fetch — load a URL
+<fetch>https://example.com</fetch>
+ 
+### 2. shell — run a terminal command (NOT for filesystem inspection)
+<shell>node -v</shell>
+ 
+### 3. read-file — read a single file from the repo
+<read-file>src/foo.ts</read-file>
+ 
+### 4. read-files — read multiple files at once
+<read-files>
+["src/foo.ts", "src/bar.ts"]
+</read-files>
+ 
+### 5. read-folder — list contents of a folder (one level deep)
+<read-folder>src/components</read-folder>
+ 
+### 6. grep — search for a pattern across files
+<grep>
+{"pattern": "ChatRunner", "glob": "src/**/*.tsx"}
+</grep>
+ 
+### 7. write-file — create or overwrite a file (COMPLETE content only)
+<write-file>
+{"path": "data/output.csv", "content": "col1,col2\\nval1,val2"}
+</write-file>
+ 
+### 8. delete-file — permanently delete a single file
+<delete-file>src/old-component.tsx</delete-file>
+ 
+### 9. delete-folder — permanently delete a folder and all its contents
+<delete-folder>src/legacy</delete-folder>
+ 
+### 10. open-url — open a URL in the user's default browser
+<open-url>https://github.com/owner/repo</open-url>
+ 
+### 11. generate-pdf — generate a PDF from markdown-style content
+<generate-pdf>
+{"path": "output/report.pdf", "content": "# Title\\n\\nBody text."}
+</generate-pdf>
+ 
+### 12. search — search the internet
+<search>how to use React useEffect cleanup</search>
+ 
+### 13. clone — clone a GitHub repo
+<clone>https://github.com/owner/repo</clone>
+ 
+### 14. changes — propose code edits shown as a diff for user approval
+<changes>
+{"summary": "what changed and why", "patches": [{"path": "src/foo.ts", "content": "COMPLETE file content", "isNew": false}]}
+</changes>`;
+}
+
 const BUILTIN_TOOLS_SECTION = `## TOOLS
 
 You have exactly fourteen tools. Use ONLY the XML tags shown below.

package/src/tools/chart.ts

@@ -1,9 +1,3 @@
-//
-
-//
-
-//
-
 import type { Tool } from "@ridit/lens-sdk";
 
 type ChartType = "bar" | "line" | "sparkline";
@@ -13,9 +7,7 @@ interface ChartInput {
   title?: string;
   labels?: string[];
   values: number[];
-  /** For line charts: height in rows. Default 10. */
   height?: number;
-  /** Bar fill character. Default "█" */
   fill?: string;
 }