Merge branch 'fix/js-jest30-loop-runner' of github.com:codeflash-ai/codeflash into fix/js-jest30-loop-runner

Author: mohammedahmed18

codeflash/languages/javascript/instrument.py

@@ -323,17 +323,24 @@ def _find_balanced_parens(self, code: str, open_paren_pos: int) -> tuple[str | N
         in_string = False
         string_char = None
 
-        while pos < len(code) and depth > 0:
-            char = code[pos]
+        s = code  # local alias for speed
+        s_len = len(s)
+        quotes = "\"'`"
+
+        while pos < s_len and depth > 0:
+            char = s[pos]
 
             # Handle string literals
-            if char in "\"'`" and (pos == 0 or code[pos - 1] != "\\"):
-                if not in_string:
-                    in_string = True
-                    string_char = char
-                elif char == string_char:
-                    in_string = False
-                    string_char = None
+            # Note: preserve original escaping semantics (only checks immediate preceding char)
+            if char in quotes:
+                prev_char = s[pos - 1] if pos > 0 else None
+                if prev_char != "\\":
+                    if not in_string:
+                        in_string = True
+                        string_char = char
+                    elif char == string_char:
+                        in_string = False
+                        string_char = None
             elif not in_string:
                 if char == "(":
                     depth += 1
@@ -345,7 +352,8 @@ def _find_balanced_parens(self, code: str, open_paren_pos: int) -> tuple[str | N
         if depth != 0:
             return None, -1
 
-        return code[open_paren_pos + 1 : pos - 1], pos
+        # slice once
+        return s[open_paren_pos + 1 : pos - 1], pos
 
     def _parse_bracket_standalone_call(self, code: str, match: re.Match) -> StandaloneCallMatch | None:
         """Parse a complete standalone obj['func'](...) call with bracket notation."""
@@ -367,10 +375,12 @@ def _parse_bracket_standalone_call(self, code: str, match: re.Match) -> Standalo
         # Check for trailing semicolon
         end_pos = close_pos
         # Skip whitespace
-        while end_pos < len(code) and code[end_pos] in " \t":
+        s = code
+        s_len = len(s)
+        while end_pos < s_len and s[end_pos] in " \t":
             end_pos += 1
 
-        has_trailing_semicolon = end_pos < len(code) and code[end_pos] == ";"
+        has_trailing_semicolon = end_pos < s_len and s[end_pos] == ";"
         if has_trailing_semicolon:
             end_pos += 1