optmiizes code chunk detection #20

Author: ran-codes

src/utils/findSections.ts

@@ -23,12 +23,15 @@ export function findSections(text: string, languageId?: string): SectionMatch[]
 
   // For Markdown/Quarto files, find code block ranges to exclude from parsing
   const codeBlocks: { start: number; end: number }[] = [];
+  let lineStartIndices: number[] = [];
   if (isMarkdownOrQuarto) {
     const lines = text.split('\n');
     let inCodeBlock = false;
     let codeBlockStart = 0;
-    
+    let charIndex = 0;
+    lineStartIndices = [];
     lines.forEach((line, index) => {
+      lineStartIndices.push(charIndex);
       if (line.trim().startsWith('```')) {
         if (!inCodeBlock) {
           inCodeBlock = true;
@@ -41,6 +44,7 @@ export function findSections(text: string, languageId?: string): SectionMatch[]
           });
         }
       }
+      charIndex += line.length + 1; // +1 for '\n'
     });
     // Edge case: unmatched opening code block at end of file
     if (inCodeBlock) {
@@ -51,10 +55,18 @@ export function findSections(text: string, languageId?: string): SectionMatch[]
   // Helper function to check if a match index is inside a code block
   const isInCodeBlock = (matchIndex: number): boolean => {
     if (!isMarkdownOrQuarto) return false;
-    
-    const lines = text.substring(0, matchIndex).split('\n');
-    const matchLineNumber = lines.length - 1;
-    
+    // Binary search to find line number for matchIndex
+    let left = 0, right = lineStartIndices.length - 1;
+    let matchLineNumber = 0;
+    while (left <= right) {
+      const mid = Math.floor((left + right) / 2);
+      if (lineStartIndices[mid] <= matchIndex) {
+        matchLineNumber = mid;
+        left = mid + 1;
+      } else {
+        right = mid - 1;
+      }
+    }
     return codeBlocks.some(block => 
       matchLineNumber >= block.start && matchLineNumber <= block.end
     );