Skip ahead to relevant key -- broken

client/src/components/Tool/extractEmbeddedJs.test.ts

@@ -42,4 +42,18 @@ describe("extractEmbeddedJs", () => {
         const result = extractEmbeddedJs(input);
         expect(result).toEqual([]);
     });
+
+    it("should skip shell_command prefix", () => {
+        const input = "shell_command: $(first)";
+        const regex = /shell_command:\s+/;
+        const result = extractEmbeddedJs(input, regex);
+        expect(result).toEqual([{ fragment: "$(first)", start: 15 }]);
+    });
+
+    it("should skip shell_command prefix and stop at next line", () => {
+        const input = "shell_command: $(first)\n$(second)";
+        const regex = /shell_command:(.*)$/m;
+        const result = extractEmbeddedJs(input, regex);
+        expect(result).toEqual([{ fragment: "$(first)", start: 15 }]);
+    });
 });

client/src/components/Tool/extractEmbeddedJs.ts

@@ -1,14 +1,22 @@
-export function extractEmbeddedJs(inputString: string): { fragment: string; start: number }[] {
-    const matches: { fragment: string; start: number }[] = [];
+export function extractEmbeddedJs(inputString: string, fieldRegex?: RegExp): { fragment: string; start: number }[] {
     let i = 0;
+    let end = inputString.length;
+    if (fieldRegex) {
+        const fieldMatch = fieldRegex.exec(inputString);
+        if (fieldMatch && fieldMatch.length) {
+            i = fieldMatch.index;
+            end = i + fieldMatch[0].length;
+        }
+    }
+    const matches: { fragment: string; start: number }[] = [];
 
-    while (i < inputString.length) {
+    while (i < end) {
         if (inputString[i] === "$" && inputString[i + 1] === "(") {
             let depth = 0;
             const startIndex = i;
             i += 2; // Skip past "$("
 
-            while (i < inputString.length) {
+            while (i < end) {
                 if (inputString[i] === "(") {
                     depth++;
                 } else if (inputString[i] === ")") {