@ridit/lens 0.4.6 → 0.4.7

package/dist/index.mjs

@@ -78205,16 +78205,42 @@ function getLastDeniedAction(messages) {
 async function runHeadless(opts) {
   const repoPath = opts.path;
   let session = opts.sessionId ? loadSession(opts.sessionId) ?? createSessionWithId(opts.sessionId, repoPath) : opts.single ? getLatestSession(repoPath) ?? createSession(repoPath) : createSession(repoPath);
-  let prompt = opts.prompt;
-  if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
-    const pending = getLastDeniedAction(getMessages(session));
-    if (pending) {
-      prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+  if (opts.resume) {
+    const msgs = getMessages(session);
+    let trimAt = -1;
+    for (let i = msgs.length - 1;i >= 0; i--) {
+      const msg = msgs[i];
+      if (!msg || msg.role !== "tool")
+        continue;
+      const content = Array.isArray(msg.content) ? msg.content : [];
+      const isDenied = content.some((p) => typeof p === "object" && p !== null && ("type" in p) && p.type === "tool-result" && ("result" in p) && typeof p.result === "string" && p.result.includes("Permission denied"));
+      if (isDenied) {
+        for (let j = i - 1;j >= 0; j--) {
+          if (msgs[j]?.role === "assistant") {
+            trimAt = j;
+            break;
+          }
+        }
+        break;
+      }
+    }
+    if (trimAt >= 0) {
+      session = { ...session, messages: msgs.slice(0, trimAt) };
+    }
+    if (!opts.single || opts.sessionId)
+      saveSession(session);
+  } else {
+    let prompt = opts.prompt;
+    if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
+      const pending = getLastDeniedAction(getMessages(session));
+      if (pending) {
+        prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+      }
     }
+    session = addMessage(session, "user", prompt);
+    if (!opts.single || opts.sessionId)
+      saveSession(session);
   }
-  session = addMessage(session, "user", prompt);
-  if (!opts.single || opts.sessionId)
-    saveSession(session);
   const toolLog = [];
   const denied = [];
   const runtimeToolNames = new Set;
@@ -78264,10 +78290,10 @@ async function runHeadless(opts) {
   });
 }
 var program = new Command().enablePositionalOptions();
-program.command("chat").description("Chat with your codebase — ask questions or make changes").option("-p, --path <path>", "Path to the repo", ".").option("-d, --dev", "Output structured JSON (no UI)").option("--single", "Single-shot: run one message then exit").option("--session <id>", "Resume session by ID, or create one with that ID").option("--id <id>", "Alias for --session").option("--force-all", "Auto-approve all tools").option("--prompt <text>", "Run a prompt non-interactively").option("--runtime-tools <path>", "path to runtime tools JSON file").action((opts) => {
+program.command("chat").description("Chat with your codebase — ask questions or make changes").option("-p, --path <path>", "Path to the repo", ".").option("-d, --dev", "Output structured JSON (no UI)").option("--single", "Single-shot: run one message then exit").option("--session <id>", "Resume session by ID, or create one with that ID").option("--id <id>", "Alias for --session").option("--force-all", "Auto-approve all tools").option("--prompt <text>", "Run a prompt non-interactively").option("--resume", "Resume from last permission-denied tool call (no new prompt needed)").option("--runtime-tools <path>", "path to runtime tools JSON file").action((opts) => {
   const sessionId = opts.session ?? opts.id;
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll, runtimeTools: opts.runtimeTools });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll ?? opts.resume, runtimeTools: opts.runtimeTools, resume: opts.resume });
     return;
   }
   render(/* @__PURE__ */ jsxDEV21(ChatCommand, {
@@ -78365,13 +78391,13 @@ program.command("run <cmd>").description("Run your dev server. Lens watches and
 });
 var firstArg = process.argv[2];
 if (!firstArg || firstArg.startsWith("-")) {
-  const defaultFlags = new Command().option("-p, --path <path>", "Path to the repo", ".").option("--session <id>", "Resume session by ID").option("--single", "Single-shot mode").option("--prompt <text>", "Run a prompt").option("-d, --dev", "Output JSON (no UI)").option("--force-all", "Auto-approve all tools").allowUnknownOption().exitOverride();
+  const defaultFlags = new Command().option("-p, --path <path>", "Path to the repo", ".").option("--session <id>", "Resume session by ID").option("--single", "Single-shot mode").option("--prompt <text>", "Run a prompt").option("-d, --dev", "Output JSON (no UI)").option("--force-all", "Auto-approve all tools").option("--resume", "Resume from last permission-denied tool call").allowUnknownOption().exitOverride();
   try {
     defaultFlags.parse(process.argv);
   } catch {}
   const opts = defaultFlags.opts();
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll ?? opts.resume, resume: opts.resume });
   } else {
     render(/* @__PURE__ */ jsxDEV21(ChatCommand, {
       path: opts.path ?? ".",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ridit/lens",
-  "version": "0.4.6",
+  "version": "0.4.7",
   "description": "Understand your codebase.",
   "author": "Ridit Jangra <riditjangra09@gmail.com> (https://ridit.space)",
   "license": "MIT",

package/src/index.tsx

@@ -72,11 +72,12 @@ function getLastDeniedAction(messages: ReturnType<typeof getMessages>): { tool:
 
 async function runHeadless(opts: {
   path: string;
-  prompt: string;
+  prompt?: string;
   sessionId?: string;
   single?: boolean;
   forceAll?: boolean;
   runtimeTools?: string;
+  resume?: boolean;
 }) {
   const repoPath = opts.path;
 
@@ -86,18 +87,54 @@ async function runHeadless(opts: {
       ? (getLatestSession(repoPath) ?? createSession(repoPath))
       : createSession(repoPath);
 
-  // if user is approving a prior denial, make the intent unambiguous
-  let prompt = opts.prompt;
-  if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
-    const pending = getLastDeniedAction(getMessages(session));
-    if (pending) {
-      prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+  if (opts.resume) {
+    // Rewind the session: strip the last denied tool-call (assistant message),
+    // its "Permission denied" tool result, and the assistant's text response after it.
+    // This leaves the session ending at the last successful state so the agent
+    // can re-attempt the denied tool with forceAll: true.
+    const msgs = getMessages(session);
+    let trimAt = -1;
+    for (let i = msgs.length - 1; i >= 0; i--) {
+      const msg = msgs[i];
+      if (!msg || msg.role !== "tool") continue;
+      const content = Array.isArray(msg.content) ? msg.content : [];
+      const isDenied = content.some(
+        (p: unknown) =>
+          typeof p === "object" && p !== null &&
+          "type" in p && (p as { type: string }).type === "tool-result" &&
+          "result" in p && typeof (p as { result: unknown }).result === "string" &&
+          ((p as { result: string }).result).includes("Permission denied"),
+      );
+      if (isDenied) {
+        // Find the assistant message immediately before this tool result (the tool-call message)
+        for (let j = i - 1; j >= 0; j--) {
+          if (msgs[j]?.role === "assistant") {
+            trimAt = j;
+            break;
+          }
+        }
+        break;
+      }
+    }
+    if (trimAt >= 0) {
+      session = { ...session, messages: msgs.slice(0, trimAt) };
+    }
+    // Save trimmed session so context is clean for this run
+    if (!opts.single || opts.sessionId) saveSession(session);
+  } else {
+    // if user is approving a prior denial, make the intent unambiguous
+    let prompt = opts.prompt!;
+    if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
+      const pending = getLastDeniedAction(getMessages(session));
+      if (pending) {
+        prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+      }
     }
-  }
 
-  session = addMessage(session, "user", prompt);
-  // save now so context is available on follow-up messages even if we exit early
-  if (!opts.single || opts.sessionId) saveSession(session);
+    session = addMessage(session, "user", prompt);
+    // save now so context is available on follow-up messages even if we exit early
+    if (!opts.single || opts.sessionId) saveSession(session);
+  }
 
   const toolLog: { tool: string; args: unknown; result: unknown }[] = [];
   const denied: { tool: string; description: string }[] = [];
@@ -170,6 +207,7 @@ program
   .option("--id <id>", "Alias for --session")
   .option("--force-all", "Auto-approve all tools")
   .option("--prompt <text>", "Run a prompt non-interactively")
+  .option("--resume", "Resume from last permission-denied tool call (no new prompt needed)")
   .option("--runtime-tools <path>", "path to runtime tools JSON file")
   .action(
     (opts: {
@@ -180,12 +218,13 @@ program
       id?: string;
       forceAll?: boolean;
       prompt?: string;
+      resume?: boolean;
       runtimeTools?: string;
     }) => {
       const sessionId = opts.session ?? opts.id;
-      // headless: dev+prompt or single+prompt → no UI, output JSON and exit
-      if (opts.prompt && (opts.dev || opts.single)) {
-        runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll, runtimeTools: opts.runtimeTools });
+      // headless: dev+prompt, single+prompt, or --resume → no UI, output JSON and exit
+      if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+        runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll ?? opts.resume, runtimeTools: opts.runtimeTools, resume: opts.resume });
         return;
       }
       render(
@@ -367,6 +406,7 @@ if (!firstArg || firstArg.startsWith("-")) {
     .option("--prompt <text>", "Run a prompt")
     .option("-d, --dev", "Output JSON (no UI)")
     .option("--force-all", "Auto-approve all tools")
+    .option("--resume", "Resume from last permission-denied tool call")
     .allowUnknownOption()
     .exitOverride();
 
@@ -379,10 +419,11 @@ if (!firstArg || firstArg.startsWith("-")) {
     prompt?: string;
     dev?: boolean;
     forceAll?: boolean;
+    resume?: boolean;
   }>();
 
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll ?? opts.resume, resume: opts.resume });
   } else {
     render(
       <ChatCommand