@ridit/lens 0.1.6 → 0.1.8

package/src/components/chat/ChatOverlays.tsx

@@ -48,6 +48,30 @@ export function PermissionPrompt({
     icon = "r";
     label = "read";
     value = tool.filePath;
+  } else if (tool.type === "read-folder") {
+    icon = "d";
+    label = "folder";
+    value = tool.folderPath;
+  } else if (tool.type === "grep") {
+    icon = "/";
+    label = "grep";
+    value = `${tool.pattern}  ${tool.glob}`;
+  } else if (tool.type === "delete-file") {
+    icon = "x";
+    label = "delete";
+    value = tool.filePath;
+  } else if (tool.type === "delete-folder") {
+    icon = "X";
+    label = "delete folder";
+    value = tool.folderPath;
+  } else if (tool.type === "open-url") {
+    icon = "↗";
+    label = "open";
+    value = tool.url;
+  } else if (tool.type === "generate-pdf") {
+    icon = "P";
+    label = "pdf";
+    value = tool.filePath;
   } else if (tool.type === "write-file") {
     icon = "w";
     label = "write";
@@ -128,12 +152,15 @@ export function TypewriterText({
   return <Text color={color}>{displayed}</Text>;
 }
 
-export function ShortcutBar() {
+export function ShortcutBar({ autoApprove }: { autoApprove?: boolean }) {
   return (
     <Box gap={3} marginTop={0}>
       <Text color="gray" dimColor>
         enter send · ^v paste · ^c exit
       </Text>
+      <Text color={autoApprove ? "green" : "gray"} dimColor={!autoApprove}>
+        {autoApprove ? "⚡ auto" : "/auto"}
+      </Text>
     </Box>
   );
 }

package/src/components/chat/ChatRunner.tsx

@@ -20,6 +20,12 @@ import {
   runShell,
   fetchUrl,
   readFile,
+  readFolder,
+  grepFiles,
+  deleteFile,
+  deleteFolder,
+  openUrl,
+  generatePdf,
   writeFile,
   buildSystemPrompt,
   parseResponse,
@@ -55,6 +61,7 @@ const COMMANDS = [
   { cmd: "/timeline", desc: "browse commit history" },
   { cmd: "/clear history", desc: "wipe session memory for this repo" },
   { cmd: "/review", desc: "review current codebsae" },
+  { cmd: "/auto", desc: "toggle auto-approve for read/search tools" },
 ];
 
 function CommandPalette({
@@ -98,6 +105,15 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
   const [clonedUrls, setClonedUrls] = useState<Set<string>>(new Set());
   const [showTimeline, setShowTimeline] = useState(false);
   const [showReview, setShowReview] = useState(false);
+  const [autoApprove, setAutoApprove] = useState(false);
+
+  // Abort controller for the currently in-flight API call.
+  // Pressing ESC while thinking aborts the request and drops the response.
+  const abortControllerRef = useRef<AbortController | null>(null);
+
+  // Cache of tool results within a single conversation turn to prevent
+  // the model from re-calling tools it already ran with the same args
+  const toolResultCache = useRef<Map<string, string>>(new Map());
 
   const inputBuffer = useRef("");
   const flushTimer = useRef<ReturnType<typeof setTimeout> | null>(null);
@@ -119,6 +135,11 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
   };
 
   const handleError = (currentAll: Message[]) => (err: unknown) => {
+    // Silently drop aborted requests — user pressed ESC intentionally
+    if (err instanceof Error && err.name === "AbortError") {
+      setStage({ type: "idle" });
+      return;
+    }
     const errMsg: Message = {
       role: "assistant",
       content: `Error: ${err instanceof Error ? err.message : "Something went wrong"}`,
@@ -129,7 +150,17 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
     setStage({ type: "idle" });
   };
 
-  const processResponse = (raw: string, currentAll: Message[]) => {
+  const processResponse = (
+    raw: string,
+    currentAll: Message[],
+    signal: AbortSignal,
+  ) => {
+    // If ESC was pressed before we got here, silently drop the response
+    if (signal.aborted) {
+      setStage({ type: "idle" });
+      return;
+    }
+
     const parsed = parseResponse(raw);
 
     if (parsed.kind === "changes") {
@@ -169,7 +200,13 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
       parsed.kind === "shell" ||
       parsed.kind === "fetch" ||
       parsed.kind === "read-file" ||
+      parsed.kind === "read-folder" ||
+      parsed.kind === "grep" ||
       parsed.kind === "write-file" ||
+      parsed.kind === "delete-file" ||
+      parsed.kind === "delete-folder" ||
+      parsed.kind === "open-url" ||
+      parsed.kind === "generate-pdf" ||
       parsed.kind === "search"
     ) {
       let tool: Parameters<typeof PermissionPrompt>[0]["tool"];
@@ -179,6 +216,22 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
         tool = { type: "fetch", url: parsed.url };
       } else if (parsed.kind === "read-file") {
         tool = { type: "read-file", filePath: parsed.filePath };
+      } else if (parsed.kind === "read-folder") {
+        tool = { type: "read-folder", folderPath: parsed.folderPath };
+      } else if (parsed.kind === "grep") {
+        tool = { type: "grep", pattern: parsed.pattern, glob: parsed.glob };
+      } else if (parsed.kind === "delete-file") {
+        tool = { type: "delete-file", filePath: parsed.filePath };
+      } else if (parsed.kind === "delete-folder") {
+        tool = { type: "delete-folder", folderPath: parsed.folderPath };
+      } else if (parsed.kind === "open-url") {
+        tool = { type: "open-url", url: parsed.url };
+      } else if (parsed.kind === "generate-pdf") {
+        tool = {
+          type: "generate-pdf",
+          filePath: parsed.filePath,
+          content: parsed.pdfContent,
+        };
       } else if (parsed.kind === "search") {
         tool = { type: "search", query: parsed.query };
       } else {
@@ -199,13 +252,32 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
         setCommitted((prev) => [...prev, preambleMsg]);
       }
 
-      setStage({
-        type: "permission",
-        tool,
-        pendingMessages: currentAll,
-        resolve: async (approved: boolean) => {
-          let result = "(denied by user)";
-          if (approved) {
+      // Safe tools that can be auto-approved (no side effects)
+      const isSafeTool =
+        parsed.kind === "read-file" ||
+        parsed.kind === "read-folder" ||
+        parsed.kind === "grep" ||
+        parsed.kind === "fetch" ||
+        parsed.kind === "open-url" ||
+        parsed.kind === "search";
+
+      const executeAndContinue = async (approved: boolean) => {
+        let result = "(denied by user)";
+        if (approved) {
+          const cacheKey =
+            parsed.kind === "read-file"
+              ? `read-file:${parsed.filePath}`
+              : parsed.kind === "read-folder"
+                ? `read-folder:${parsed.folderPath}`
+                : parsed.kind === "grep"
+                  ? `grep:${parsed.pattern}:${parsed.glob}`
+                  : null;
+
+          if (cacheKey && toolResultCache.current.has(cacheKey)) {
+            result =
+              toolResultCache.current.get(cacheKey)! +
+              "\n\n[NOTE: This result was already retrieved earlier. Do not request it again.]";
+          } else {
             try {
               setStage({ type: "thinking" });
               if (parsed.kind === "shell") {
@@ -214,6 +286,22 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
                 result = await fetchUrl(parsed.url);
               } else if (parsed.kind === "read-file") {
                 result = readFile(parsed.filePath, repoPath);
+              } else if (parsed.kind === "read-folder") {
+                result = readFolder(parsed.folderPath, repoPath);
+              } else if (parsed.kind === "grep") {
+                result = grepFiles(parsed.pattern, parsed.glob, repoPath);
+              } else if (parsed.kind === "delete-file") {
+                result = deleteFile(parsed.filePath, repoPath);
+              } else if (parsed.kind === "delete-folder") {
+                result = deleteFolder(parsed.folderPath, repoPath);
+              } else if (parsed.kind === "open-url") {
+                result = openUrl(parsed.url);
+              } else if (parsed.kind === "generate-pdf") {
+                result = generatePdf(
+                  parsed.filePath,
+                  parsed.pdfContent,
+                  repoPath,
+                );
               } else if (parsed.kind === "write-file") {
                 result = writeFile(
                   parsed.filePath,
@@ -223,72 +311,133 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
               } else if (parsed.kind === "search") {
                 result = await searchWeb(parsed.query);
               }
+              if (cacheKey) {
+                toolResultCache.current.set(cacheKey, result);
+              }
             } catch (err: unknown) {
               result = `Error: ${err instanceof Error ? err.message : "failed"}`;
             }
           }
+        }
 
-          if (approved && !result.startsWith("Error:")) {
-            const kindMap = {
-              shell: "shell-run",
-              fetch: "url-fetched",
-              "read-file": "file-read",
-              "write-file": "file-written",
-              search: "url-fetched",
-            } as const;
-            appendHistory({
-              kind: kindMap[parsed.kind as keyof typeof kindMap] ?? "shell-run",
-              detail:
-                parsed.kind === "shell"
-                  ? parsed.command
-                  : parsed.kind === "fetch"
-                    ? parsed.url
-                    : parsed.kind === "search"
-                      ? parsed.query
-                      : parsed.filePath,
-              summary: result.split("\n")[0]?.slice(0, 120) ?? "",
-              repoPath,
-            });
-          }
-
-          const toolName =
-            parsed.kind === "shell"
-              ? "shell"
-              : parsed.kind === "fetch"
-                ? "fetch"
-                : parsed.kind === "read-file"
-                  ? "read-file"
+        if (approved && !result.startsWith("Error:")) {
+          const kindMap = {
+            shell: "shell-run",
+            fetch: "url-fetched",
+            "read-file": "file-read",
+            "read-folder": "file-read",
+            grep: "file-read",
+            "delete-file": "file-written",
+            "delete-folder": "file-written",
+            "open-url": "url-fetched",
+            "generate-pdf": "file-written",
+            "write-file": "file-written",
+            search: "url-fetched",
+          } as const;
+          appendHistory({
+            kind: kindMap[parsed.kind as keyof typeof kindMap] ?? "shell-run",
+            detail:
+              parsed.kind === "shell"
+                ? parsed.command
+                : parsed.kind === "fetch"
+                  ? parsed.url
                   : parsed.kind === "search"
-                    ? "search"
-                    : "write-file";
-
-          const toolContent =
-            parsed.kind === "shell"
-              ? parsed.command
-              : parsed.kind === "fetch"
-                ? parsed.url
-                : parsed.kind === "search"
-                  ? parsed.query
-                  : parsed.filePath;
-
-          const toolMsg: Message = {
-            role: "assistant",
-            type: "tool",
-            toolName,
-            content: toolContent,
-            result,
-            approved,
-          };
+                    ? parsed.query
+                    : parsed.kind === "read-folder"
+                      ? parsed.folderPath
+                      : parsed.kind === "grep"
+                        ? `${parsed.pattern} ${parsed.glob}`
+                        : parsed.kind === "delete-file"
+                          ? parsed.filePath
+                          : parsed.kind === "delete-folder"
+                            ? parsed.folderPath
+                            : parsed.kind === "open-url"
+                              ? parsed.url
+                              : parsed.kind === "generate-pdf"
+                                ? parsed.filePath
+                                : parsed.filePath,
+            summary: result.split("\n")[0]?.slice(0, 120) ?? "",
+            repoPath,
+          });
+        }
+
+        const toolName =
+          parsed.kind === "shell"
+            ? "shell"
+            : parsed.kind === "fetch"
+              ? "fetch"
+              : parsed.kind === "read-file"
+                ? "read-file"
+                : parsed.kind === "read-folder"
+                  ? "read-folder"
+                  : parsed.kind === "grep"
+                    ? "grep"
+                    : parsed.kind === "delete-file"
+                      ? "delete-file"
+                      : parsed.kind === "delete-folder"
+                        ? "delete-folder"
+                        : parsed.kind === "open-url"
+                          ? "open-url"
+                          : parsed.kind === "generate-pdf"
+                            ? "generate-pdf"
+                            : parsed.kind === "search"
+                              ? "search"
+                              : "write-file";
+
+        const toolContent =
+          parsed.kind === "shell"
+            ? parsed.command
+            : parsed.kind === "fetch"
+              ? parsed.url
+              : parsed.kind === "search"
+                ? parsed.query
+                : parsed.kind === "read-folder"
+                  ? parsed.folderPath
+                  : parsed.kind === "grep"
+                    ? `${parsed.pattern} — ${parsed.glob}`
+                    : parsed.kind === "delete-file"
+                      ? parsed.filePath
+                      : parsed.kind === "delete-folder"
+                        ? parsed.folderPath
+                        : parsed.kind === "open-url"
+                          ? parsed.url
+                          : parsed.kind === "generate-pdf"
+                            ? parsed.filePath
+                            : parsed.filePath;
+
+        const toolMsg: Message = {
+          role: "assistant",
+          type: "tool",
+          toolName,
+          content: toolContent,
+          result,
+          approved,
+        };
+
+        const withTool = [...currentAll, toolMsg];
+        setAllMessages(withTool);
+        setCommitted((prev) => [...prev, toolMsg]);
 
-          const withTool = [...currentAll, toolMsg];
-          setAllMessages(withTool);
-          setCommitted((prev) => [...prev, toolMsg]);
+        // Create a fresh abort controller for the follow-up call
+        const nextAbort = new AbortController();
+        abortControllerRef.current = nextAbort;
 
-          setStage({ type: "thinking" });
-          callChat(provider!, systemPrompt, withTool)
-            .then((r: string) => processResponse(r, withTool))
-            .catch(handleError(withTool));
-        },
+        setStage({ type: "thinking" });
+        callChat(provider!, systemPrompt, withTool, nextAbort.signal)
+          .then((r: string) => processResponse(r, withTool, nextAbort.signal))
+          .catch(handleError(withTool));
+      };
+
+      if (autoApprove && isSafeTool) {
+        executeAndContinue(true);
+        return;
+      }
+
+      setStage({
+        type: "permission",
+        tool,
+        pendingMessages: currentAll,
+        resolve: executeAndContinue,
       });
       return;
     }
@@ -349,6 +498,21 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
       return;
     }
 
+    if (text.trim().toLowerCase() === "/auto") {
+      const next = !autoApprove;
+      setAutoApprove(next);
+      const msg: Message = {
+        role: "assistant",
+        content: next
+          ? "Auto-approve ON — read, search, grep and folder tools will run without asking. Write and code changes still require approval."
+          : "Auto-approve OFF — all tools will ask for permission.",
+        type: "text",
+      };
+      setCommitted((prev) => [...prev, msg]);
+      setAllMessages((prev) => [...prev, msg]);
+      return;
+    }
+
     if (text.trim().toLowerCase() === "/clear history") {
       clearRepoHistory(repoPath);
       const clearedMsg: Message = {
@@ -365,15 +529,29 @@ export const ChatRunner = ({ repoPath }: { repoPath: string }) => {
     const nextAll = [...allMessages, userMsg];
     setCommitted((prev) => [...prev, userMsg]);
     setAllMessages(nextAll);
+    toolResultCache.current.clear();
+
+    // Create a fresh abort controller for this request
+    const abort = new AbortController();
+    abortControllerRef.current = abort;
+
     setStage({ type: "thinking" });
-    callChat(provider, systemPrompt, nextAll)
-      .then((raw: string) => processResponse(raw, nextAll))
+    callChat(provider, systemPrompt, nextAll, abort.signal)
+      .then((raw: string) => processResponse(raw, nextAll, abort.signal))
       .catch(handleError(nextAll));
   };
 
   useInput((input, key) => {
     if (showTimeline) return;
 
+    // ESC while thinking → abort the in-flight request and go idle
+    if (stage.type === "thinking" && key.escape) {
+      abortControllerRef.current?.abort();
+      abortControllerRef.current = null;
+      setStage({ type: "idle" });
+      return;
+    }
+
     if (stage.type === "idle") {
       if (key.ctrl && input === "c") {
         process.exit(0);
@@ -699,6 +877,9 @@ Suggestions: ${lensFile.suggestions.slice(0, 3).join("; ")}`
         <Box gap={1}>
           <Text color={ACCENT}>●</Text>
           <TypewriterText text={thinkingPhrase} />
+          <Text color="gray" dimColor>
+            · esc cancel
+          </Text>
         </Box>
       )}
 
@@ -724,7 +905,7 @@ Suggestions: ${lensFile.suggestions.slice(0, 3).join("; ")}`
               setInputValue("");
             }}
           />
-          <ShortcutBar />
+          <ShortcutBar autoApprove={autoApprove} />
         </Box>
       )}
     </Box>

package/src/index.tsx

@@ -1,60 +1,60 @@
-import React from "react";
-import { render } from "ink";
-import { Command } from "commander";
-import { RepoCommand } from "./commands/repo";
-import { InitCommand } from "./commands/provider";
-import { ReviewCommand } from "./commands/review";
-import { TaskCommand } from "./commands/task";
-import { ChatCommand } from "./commands/chat";
-import { TimelineCommand } from "./commands/timeline";
-
-const program = new Command();
-
-program
-  .command("repo <url>")
-  .description("Analyze a remote repository")
-  .action((url) => {
-    render(<RepoCommand url={url} />);
-  });
-
-program
-  .command("provider")
-  .description("Configure AI providers")
-  .action(() => {
-    render(<InitCommand />);
-  });
-
-program
-  .command("review [path]")
-  .description("Review a local codebase")
-  .action((inputPath) => {
-    render(<ReviewCommand path={inputPath ?? "."} />);
-  });
-
-program
-  .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 }) => {
-    render(<TaskCommand prompt={text} path={opts.path} />);
-  });
-
-program
-  .command("chat")
-  .description("Chat with your codebase — ask questions or make changes")
-  .option("-p, --path <path>", "Path to the repo", ".")
-  .action((opts: { path: string }) => {
-    render(<ChatCommand path={opts.path} />);
-  });
-
-program
-  .command("timeline")
-  .description(
-    "Explore your code history — see commits, changes, and evolution",
-  )
-  .option("-p, --path <path>", "Path to the repo", ".")
-  .action((opts: { path: string }) => {
-    render(<TimelineCommand path={opts.path} />);
-  });
-
-program.parse(process.argv);
+import React from "react";
+import { render } from "ink";
+import { Command } from "commander";
+import { RepoCommand } from "./commands/repo";
+import { InitCommand } from "./commands/provider";
+import { ReviewCommand } from "./commands/review";
+import { TaskCommand } from "./commands/task";
+import { ChatCommand } from "./commands/chat";
+import { TimelineCommand } from "./commands/timeline";
+
+const program = new Command();
+
+program
+  .command("repo <url>")
+  .description("Analyze a remote repository")
+  .action((url) => {
+    render(<RepoCommand url={url} />);
+  });
+
+program
+  .command("provider")
+  .description("Configure AI providers")
+  .action(() => {
+    render(<InitCommand />);
+  });
+
+program
+  .command("review [path]")
+  .description("Review a local codebase")
+  .action((inputPath) => {
+    render(<ReviewCommand path={inputPath ?? "."} />);
+  });
+
+program
+  .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 }) => {
+    render(<TaskCommand prompt={text} path={opts.path} />);
+  });
+
+program
+  .command("chat")
+  .description("Chat with your codebase — ask questions or make changes")
+  .option("-p, --path <path>", "Path to the repo", ".")
+  .action((opts: { path: string }) => {
+    render(<ChatCommand path={opts.path} />);
+  });
+
+program
+  .command("timeline")
+  .description(
+    "Explore your code history — see commits, changes, and evolution",
+  )
+  .option("-p, --path <path>", "Path to the repo", ".")
+  .action((opts: { path: string }) => {
+    render(<TimelineCommand path={opts.path} />);
+  });
+
+program.parse(process.argv);

package/src/types/chat.ts

@@ -6,7 +6,13 @@ export type ToolCall =
   | { type: "shell"; command: string }
   | { type: "fetch"; url: string }
   | { type: "read-file"; filePath: string }
+  | { type: "read-folder"; folderPath: string }
+  | { type: "grep"; pattern: string; glob: string }
   | { type: "write-file"; filePath: string; fileContent: string }
+  | { type: "delete-file"; filePath: string }
+  | { type: "delete-folder"; folderPath: string }
+  | { type: "open-url"; url: string }
+  | { type: "generate-pdf"; filePath: string; content: string }
   | { type: "search"; query: string };
 
 // ── Messages ──────────────────────────────────────────────────────────────────
@@ -16,7 +22,18 @@ export type Message =
   | {
       role: "assistant";
       type: "tool";
-      toolName: "shell" | "fetch" | "read-file" | "write-file" | "search";
+      toolName:
+        | "shell"
+        | "fetch"
+        | "read-file"
+        | "read-folder"
+        | "grep"
+        | "write-file"
+        | "delete-file"
+        | "delete-folder"
+        | "open-url"
+        | "generate-pdf"
+        | "search";
       content: string;
       result: string;
       approved: boolean;