@ridit/lens 0.2.0 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ridit/lens",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Know Your Codebase.",
   "author": "Ridit Jangra <riditjangra09@gmail.com> (https://ridit.space)",
   "license": "MIT",

package/src/components/chat/ChatMessage.tsx

@@ -98,6 +98,20 @@ function MessageBody({ content }: { content: string }) {
   );
 }
 
+function summarizeToolContent(toolName: string, content: string): string {
+  // For write-file, extract just the path
+  if (toolName === "write-file" || toolName === "read-file") {
+    const pathMatch = content.match(/"path"\s*:\s*"([^"]+)"/);
+    if (pathMatch) return pathMatch[1]!;
+  }
+  // For changes blocks, just say what changed
+  if (content.includes('"summary"')) {
+    const summaryMatch = content.match(/"summary"\s*:\s*"([^"]+)"/);
+    if (summaryMatch) return summaryMatch[1]!;
+  }
+  return content.length > 120 ? content.slice(0, 120) + "…" : content;
+}
+
 export function StaticMessage({ msg }: { msg: Message }) {
   if (msg.role === "user") {
     return (
@@ -130,7 +144,7 @@ export function StaticMessage({ msg }: { msg: Message }) {
         ? msg.content
         : msg.toolName === "search"
           ? `"${msg.content}"`
-          : msg.content;
+          : summarizeToolContent(msg.toolName, msg.content);
 
     return (
       <Box flexDirection="column" marginBottom={1}>

package/src/components/chat/ChatOverlays.tsx

@@ -1,8 +1,7 @@
 import React from "react";
-import { Box, Text, Static } from "ink";
+import { Box, Static, Text } from "ink";
 import Spinner from "ink-spinner";
 import TextInput from "ink-text-input";
-import figures from "figures";
 import { ACCENT } from "../../colors";
 import { DiffViewer } from "../repo/DiffViewer";
 import { StaticMessage } from "./ChatMessage";
@@ -25,61 +24,89 @@ function Hint({ text }: { text: string }) {
   );
 }
 
+// ── PermissionPrompt ──────────────────────────────────────────────────────────
+//
+// Works with both the old explicit ToolCall union and the new generic
+// { type, _label, _display } shape produced by the plugin system.
+
 export function PermissionPrompt({
   tool,
   onDecide,
 }: {
-  tool: ToolCall;
+  tool: ToolCall | { type: string; _label: string; _display: string };
   onDecide: (approved: boolean) => void;
 }) {
   let icon: string;
   let label: string;
   let value: string;
 
-  if (tool.type === "shell") {
-    icon = "$";
-    label = "run";
-    value = tool.command;
-  } else if (tool.type === "fetch") {
-    icon = "~>";
-    label = "fetch";
-    value = tool.url;
-  } else if (tool.type === "read-file") {
-    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";
-    value = `${tool.filePath} (${tool.fileContent.length} bytes)`;
+  // Generic plugin tool shape
+  if ("_label" in tool) {
+    const iconMap: Record<string, string> = {
+      run: "$",
+      fetch: "~>",
+      read: "r",
+      write: "w",
+      delete: "x",
+      "delete folder": "X",
+      open: "↗",
+      pdf: "P",
+      search: "?",
+      folder: "d",
+      grep: "/",
+      clone: "↓",
+      query: "⌗",
+    };
+    icon = iconMap[tool._label] ?? "·";
+    label = tool._label;
+    value = tool._display;
   } else {
-    icon = "?";
-    label = "search";
-    value = tool.query;
+    // Legacy explicit ToolCall union
+    if (tool.type === "shell") {
+      icon = "$";
+      label = "run";
+      value = tool.command;
+    } else if (tool.type === "fetch") {
+      icon = "~>";
+      label = "fetch";
+      value = tool.url;
+    } else if (tool.type === "read-file") {
+      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";
+      value = `${tool.filePath} (${tool.fileContent.length} bytes)`;
+    } else {
+      icon = "?";
+      label = "search";
+      value = (tool as any).query ?? "";
+    }
   }
 
   return (
@@ -100,10 +127,12 @@ export function InputBox({
   value,
   onChange,
   onSubmit,
+  inputKey,
 }: {
   value: string;
   onChange: (v: string) => void;
   onSubmit: (v: string) => void;
+  inputKey?: number;
 }) {
   return (
     <Box
@@ -117,7 +146,12 @@ export function InputBox({
     >
       <Box gap={1}>
         <Text color={ACCENT}>{">"}</Text>
-        <TextInput value={value} onChange={onChange} onSubmit={onSubmit} />
+        <TextInput
+          key={inputKey}
+          value={value}
+          onChange={onChange}
+          onSubmit={onSubmit}
+        />
       </Box>
     </Box>
   );