@shakecodeslikecray/whiterose 0.2.2 → 0.2.3

package/dist/cli/index.js

@@ -886,8 +886,23 @@ Set "survived": true if you CANNOT disprove it (it's a real bug).`;
     const candidates = [];
     let depth = 0;
     let start = -1;
+    let inString = false;
+    let escapeNext = false;
     for (let i = 0; i < text3.length; i++) {
       const char = text3[i];
+      if (escapeNext) {
+        escapeNext = false;
+        continue;
+      }
+      if (char === "\\" && inString) {
+        escapeNext = true;
+        continue;
+      }
+      if (char === '"' && !escapeNext) {
+        inString = !inString;
+        continue;
+      }
+      if (inString) continue;
       if (char === "{") {
         if (depth === 0) {
           start = i;
@@ -1082,13 +1097,58 @@ Set "survived": true if you CANNOT disprove it (it's a real bug).`;
     if (codeBlockMatch) {
       return codeBlockMatch[1].trim();
     }
-    const arrayMatch = text3.match(/\[[\s\S]*\]/);
-    if (arrayMatch) {
-      return arrayMatch[0];
+    return this.findBalancedJson(text3);
+  }
+  // Find the first balanced JSON object or array in text
+  findBalancedJson(text3) {
+    const objectStart = text3.indexOf("{");
+    const arrayStart = text3.indexOf("[");
+    let start = -1;
+    let openChar = "{";
+    let closeChar = "}";
+    if (objectStart === -1 && arrayStart === -1) {
+      return null;
+    } else if (objectStart === -1) {
+      start = arrayStart;
+      openChar = "[";
+      closeChar = "]";
+    } else if (arrayStart === -1) {
+      start = objectStart;
+    } else {
+      if (arrayStart < objectStart) {
+        start = arrayStart;
+        openChar = "[";
+        closeChar = "]";
+      } else {
+        start = objectStart;
+      }
     }
-    const objectMatch = text3.match(/\{[\s\S]*\}/);
-    if (objectMatch) {
-      return objectMatch[0];
+    let depth = 0;
+    let inString = false;
+    let escapeNext = false;
+    for (let i = start; i < text3.length; i++) {
+      const char = text3[i];
+      if (escapeNext) {
+        escapeNext = false;
+        continue;
+      }
+      if (char === "\\" && inString) {
+        escapeNext = true;
+        continue;
+      }
+      if (char === '"' && !escapeNext) {
+        inString = !inString;
+        continue;
+      }
+      if (inString) continue;
+      if (char === openChar) {
+        depth++;
+      } else if (char === closeChar) {
+        depth--;
+        if (depth === 0) {
+          return text3.slice(start, i + 1);
+        }
+      }
     }
     return null;
   }
@@ -4233,7 +4293,7 @@ ${chalk.red(" \u255A\u2550\u2550\u255D\u255A\u2550\u2550\u255D \u255A\u2550\u255
 ${chalk.dim(`  "I've been staring at your code for a long time."`)}
 `;
 var program = new Command();
-program.name("whiterose").description("AI-powered bug hunter that uses your existing LLM subscription").version("0.2.2").hook("preAction", () => {
+program.name("whiterose").description("AI-powered bug hunter that uses your existing LLM subscription").version("0.2.3").hook("preAction", () => {
   const args = process.argv.slice(2);
   if (!args.includes("--help") && !args.includes("-h") && args.length > 0) {
     console.log(BANNER);