pi-review-loop 0.2.1 → 0.3.0

package/README.md

@@ -44,18 +44,8 @@ The pattern is the same: keep reviewing until there's genuinely nothing left to
 
 ## Install
 
-**npm** (recommended):
-
 ```bash
-npm install -g pi-review-loop
-```
-
-The extension is automatically copied to `~/.pi/agent/extensions/pi-review-loop/`.
-
-**Or clone directly:**
-
-```bash
-git clone https://github.com/nicobailon/pi-review-loop ~/.pi/agent/extensions/pi-review-loop
+pi install npm:pi-review-loop
 ```
 
 Restart pi to load the extension. On activation, you'll see status in the footer:
@@ -253,9 +243,13 @@ The loop exits when:
 | Command | Description |
 |---------|-------------|
 | `/review-start` | Activate and send review prompt immediately |
+| `/review-start "focus on X"` | Start review with custom focus appended to prompt |
+| `/review-plan` | Activate and review plans/specs/PRDs (uses `double-check-plan` template) |
+| `/review-plan "focus on X"` | Review plan with custom focus appended to prompt |
 | `/review-exit` | Exit review mode |
 | `/review-max <n>` | Set max iterations (session only) |
 | `/review-auto [on\|off]` | Toggle auto-trigger from keywords (session only) |
+| `/review-auto "focus on X"` | Enable auto-trigger AND start review with custom focus |
 | `/review-status` | Show current state |
 
 ## Tool API
@@ -272,6 +266,9 @@ review_loop({ start: true })
 // Start with custom max iterations
 review_loop({ start: true, maxIterations: 5 })
 
+// Start with custom focus
+review_loop({ start: true, focus: "focus on error handling and edge cases" })
+
 // Stop review mode
 review_loop({ stop: true })
 
@@ -290,6 +287,7 @@ review_loop({ autoTrigger: false })
   "currentIteration": 2,
   "maxIterations": 7,
   "autoTrigger": false,
+  "focus": "focus on error handling",
   "message": "Review mode active: iteration 2/7"
 }
 ```

package/index.ts

@@ -6,6 +6,7 @@ export default function (pi: ExtensionAPI) {
   let settings: ReviewerLoopSettings = loadSettings();
   let reviewModeActive = false;
   let currentIteration = 0;
+  let customPromptSuffix = "";
 
   function updateStatus(ctx: ExtensionContext) {
     if (reviewModeActive) {
@@ -18,9 +19,25 @@ export default function (pi: ExtensionAPI) {
     }
   }
 
+  function buildReviewPrompt(promptConfig: { type: "inline" | "file" | "template"; value: string }): string {
+    const basePrompt = getReviewPrompt(promptConfig);
+    if (customPromptSuffix) {
+      return `${basePrompt}\n\n**Additional focus:** ${customPromptSuffix}`;
+    }
+    return basePrompt;
+  }
+
+  function parseCustomText(args: string): string {
+    // Match quoted strings: "text" or 'text' (must use matching quotes)
+    const match = args.match(/^"(.+)"$/s) || args.match(/^'(.+)'$/s) ||
+                  args.match(/"(.+?)"/s) || args.match(/'(.+?)'/s);
+    return match ? match[1].trim() : "";
+  }
+
   function exitReviewMode(ctx: ExtensionContext, reason: string) {
     reviewModeActive = false;
     currentIteration = 0;
+    customPromptSuffix = "";
     updateStatus(ctx);
     ctx.ui.notify(`Review mode ended: ${reason}`, "info");
   }
@@ -103,19 +120,33 @@ export default function (pi: ExtensionAPI) {
     }
 
     updateStatus(ctx);
-    pi.sendUserMessage(getReviewPrompt(settings.reviewPromptConfig), {
+    pi.sendUserMessage(buildReviewPrompt(settings.reviewPromptConfig), {
       deliverAs: "followUp",
     });
   });
 
   pi.registerCommand("review-start", {
-    description: "Activate review loop and send review prompt immediately",
-    handler: async (_args, ctx) => {
+    description: "Activate review loop and send review prompt immediately. Optional: add custom focus with quotes.",
+    handler: async (args, ctx) => {
       if (reviewModeActive) {
         ctx.ui.notify("Review mode is already active", "info");
       } else {
+        customPromptSuffix = parseCustomText(args);
         enterReviewMode(ctx);
-        pi.sendUserMessage(getReviewPrompt(settings.reviewPromptConfig));
+        pi.sendUserMessage(buildReviewPrompt(settings.reviewPromptConfig));
+      }
+    },
+  });
+
+  pi.registerCommand("review-plan", {
+    description: "Activate review loop for plans/specs/PRDs. Optional: add custom focus with quotes.",
+    handler: async (args, ctx) => {
+      if (reviewModeActive) {
+        ctx.ui.notify("Review mode is already active", "info");
+      } else {
+        customPromptSuffix = parseCustomText(args);
+        enterReviewMode(ctx);
+        pi.sendUserMessage(buildReviewPrompt({ type: "template", value: "double-check-plan" }));
       }
     },
   });
@@ -162,17 +193,36 @@ export default function (pi: ExtensionAPI) {
   });
 
   pi.registerCommand("review-auto", {
-    description: "Toggle auto-trigger from keywords (off by default)",
+    description: "Toggle auto-trigger, or start review with custom focus: /review-auto \"focus on X\"",
     handler: async (args, ctx) => {
-      const arg = args.trim().toLowerCase();
-      if (arg === "on" || arg === "true" || arg === "1") {
+      const arg = args.trim();
+      const argLower = arg.toLowerCase();
+      
+      // Check for quoted custom text first
+      const customText = parseCustomText(arg);
+      if (customText) {
+        // Custom text provided: enable auto-trigger and start review with custom focus
+        settings.autoTrigger = true;
+        customPromptSuffix = customText;
+        if (reviewModeActive) {
+          ctx.ui.notify(`Auto-trigger enabled, focus updated for next iteration`, "info");
+        } else {
+          enterReviewMode(ctx);
+          pi.sendUserMessage(buildReviewPrompt(settings.reviewPromptConfig));
+          ctx.ui.notify(`Auto-trigger enabled, review started with custom focus`, "info");
+        }
+        return;
+      }
+      
+      // Standard on/off/toggle behavior
+      if (argLower === "on" || argLower === "true" || argLower === "1") {
         settings.autoTrigger = true;
-      } else if (arg === "off" || arg === "false" || arg === "0") {
+      } else if (argLower === "off" || argLower === "false" || argLower === "0") {
         settings.autoTrigger = false;
       } else if (arg === "") {
         settings.autoTrigger = !settings.autoTrigger;
       } else {
-        ctx.ui.notify("Usage: /review-auto [on|off]", "error");
+        ctx.ui.notify("Usage: /review-auto [on|off] or /review-auto \"custom focus\"", "error");
         return;
       }
       ctx.ui.notify(
@@ -208,6 +258,11 @@ export default function (pi: ExtensionAPI) {
           minimum: 1,
         })
       ),
+      focus: Type.Optional(
+        Type.String({
+          description: "Custom focus/instructions to append to the review prompt (e.g., \"focus on error handling\")",
+        })
+      ),
     }),
 
     async execute(_toolCallId, params, _onUpdate, ctx) {
@@ -225,6 +280,11 @@ export default function (pi: ExtensionAPI) {
         );
       }
 
+      // Update custom focus if provided
+      if (typeof params.focus === "string") {
+        customPromptSuffix = params.focus.trim();
+      }
+
       // Mode: start > stop > status
       if (params.start) {
         if (reviewModeActive) {
@@ -237,6 +297,7 @@ export default function (pi: ExtensionAPI) {
                   currentIteration,
                   maxIterations: settings.maxIterations,
                   autoTrigger: settings.autoTrigger,
+                  focus: customPromptSuffix || undefined,
                   message: "Review mode is already active",
                 }),
               },
@@ -245,7 +306,7 @@ export default function (pi: ExtensionAPI) {
         }
 
         enterReviewMode(ctx);
-        pi.sendUserMessage(getReviewPrompt(settings.reviewPromptConfig));
+        pi.sendUserMessage(buildReviewPrompt(settings.reviewPromptConfig));
 
         return {
           content: [
@@ -256,7 +317,10 @@ export default function (pi: ExtensionAPI) {
                 currentIteration,
                 maxIterations: settings.maxIterations,
                 autoTrigger: settings.autoTrigger,
-                message: "Review mode started. Review prompt sent.",
+                focus: customPromptSuffix || undefined,
+                message: customPromptSuffix
+                  ? `Review mode started with custom focus. Review prompt sent.`
+                  : "Review mode started. Review prompt sent.",
               }),
             },
           ],
@@ -309,6 +373,7 @@ export default function (pi: ExtensionAPI) {
               currentIteration,
               maxIterations: settings.maxIterations,
               autoTrigger: settings.autoTrigger,
+              focus: customPromptSuffix || undefined,
               message: reviewModeActive
                 ? `Review mode active: iteration ${currentIteration}/${settings.maxIterations}`
                 : "Review mode inactive",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-review-loop",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Automated code review loop extension for pi coding agent",
   "scripts": {
     "postinstall": "node scripts/install.js",