@mu-crew/mu 3.8.0 → 3.9.0

package/dist/extension/mu-pi.js

@@ -501,7 +501,7 @@ ${task}` : task,
       };
     } finally {
       freeSlot();
-      if (queuedAs) drain();
+      drain();
     }
   }
   pi.registerTool({
@@ -711,25 +711,31 @@ function readSkill() {
   }
   return void 0;
 }
+var ENV_ASSIGN = /^[A-Za-z_][A-Za-z0-9_]*=/;
+function muArgs(words) {
+  let at = 0;
+  while (at < words.length && ENV_ASSIGN.test(words[at] ?? "")) at++;
+  const cmd = words[at];
+  if (cmd !== "mu" && !cmd?.endsWith("/mu")) return void 0;
+  return words.slice(at + 1);
+}
 function dispatchedWorkstreams(command) {
   const found = /* @__PURE__ */ new Set();
   for (const words of shellSegments(command)) {
-    const at = words.findIndex((w) => w === "mu" || w.endsWith("/mu"));
-    if (at < 0) continue;
-    const [noun, verb] = [words[at + 1], words[at + 2]];
-    const isSendSpawn = noun === "agent" && (verb === "send" || verb === "spawn");
-    const isClaimFor = noun === "task" && verb === "claim" && words.includes("--for");
-    if (!isSendSpawn && !isClaimFor) continue;
+    const args = muArgs(words);
+    if (args === void 0) continue;
+    const claim = claimOf(args);
+    if (claim !== void 0) {
+      if (claim.ws !== "scratch") found.add(claim.ws);
+      continue;
+    }
+    if (args[0] !== "agent" || args[1] !== "send" && args[1] !== "spawn") continue;
     let ws = "";
-    for (let k = at + 3; k < words.length; k++) {
-      const w = words[k] ?? "";
-      if (w === "-w" || w === "--workstream") ws = words[k + 1] ?? "";
+    for (let k = 2; k < args.length; k++) {
+      const w = args[k] ?? "";
+      if (w === "-w" || w === "--workstream") ws = args[k + 1] ?? "";
       else if (w.startsWith("--workstream=")) ws = w.slice("--workstream=".length);
     }
-    if (ws === "" && isClaimFor) {
-      const ref = words[at + 3] ?? "";
-      if (ref.includes("/")) ws = ref.split("/")[0] ?? "";
-    }
     if (ws !== "scratch") found.add(ws);
   }
   return found.size > 0 ? [...found] : void 0;
@@ -739,6 +745,8 @@ function shellSegments(command) {
   const segs = [];
   let words = [];
   let word;
+  const heredocs = [];
+  let arithEnd = -1;
   const endWord = () => {
     if (word !== void 0) words.push(word);
     word = void 0;
@@ -748,6 +756,18 @@ function shellSegments(command) {
     if (words.length > 0) segs.push(words);
     words = [];
   };
+  const skipHeredocs = (i) => {
+    for (const h of heredocs.splice(0)) {
+      while (i < command.length) {
+        const eol = command.indexOf("\n", i + 1);
+        const end = eol < 0 ? command.length : eol;
+        const line = command.slice(i + 1, end);
+        i = end;
+        if ((h.strip ? line.replace(/^\t+/, "") : line) === h.delim) break;
+      }
+    }
+    return i;
+  };
   for (let i = 0; i < command.length; i++) {
     const c = command[i] ?? "";
     if (c === "'" || c === '"') {
@@ -755,11 +775,34 @@ function shellSegments(command) {
       const end = close < 0 ? command.length : close;
       word = (word ?? "") + command.slice(i + 1, end);
       i = end;
+    } else if (c === "\\" && command[i + 1] === "\n") {
+      i++;
     } else if (c === "\\" && i + 1 < command.length) {
       word = (word ?? "") + command[++i];
+    } else if (c === "#" && word === void 0) {
+      const eol = command.indexOf("\n", i);
+      i = (eol < 0 ? command.length : eol) - 1;
+    } else if (c === "(" && command[i + 1] === "(" && i > arithEnd) {
+      arithEnd = arithmeticEnd(command, i);
+      word = `${word ?? ""}((`;
+      i++;
+    } else if (command.startsWith("<<<", i)) {
+      endWord();
+      i += 2;
+    } else if (c === "<" && command[i + 1] === "<" && i > arithEnd) {
+      endWord();
+      const m = /^<<(-?)[ \t]*('([^']*)'|"([^"]*)"|[^\s;&|<>()]+)/.exec(command.slice(i));
+      if (!m) {
+        i++;
+        continue;
+      }
+      const delim = m[3] ?? m[4] ?? (m[2] ?? "").replace(/\\/g, "");
+      heredocs.push({ delim, strip: m[1] === "-" });
+      i += m[0].length - 1;
     } else if (c === ";" || c === "|" || c === "\n" || c === "&" && command[i + 1] === "&") {
       if (c !== ";" && c !== "\n" && command[i + 1] === c) i++;
       endSeg();
+      if (c === "\n" && heredocs.length > 0) i = skipHeredocs(i);
     } else if (/\s/.test(c)) {
       endWord();
     } else {
@@ -769,30 +812,53 @@ function shellSegments(command) {
   endSeg();
   return segs;
 }
+function arithmeticEnd(command, open) {
+  let depth = 0;
+  for (let i = open + 2; i < command.length; i++) {
+    const c = command[i];
+    if (c === "'" || c === '"') {
+      const close = command.indexOf(c, i + 1);
+      if (close < 0) return -1;
+      i = close;
+    } else if (c === "\\") {
+      i++;
+    } else if (c === "(") {
+      depth++;
+    } else if (c === ")") {
+      if (depth > 0) depth--;
+      else return command[i + 1] === ")" ? i + 1 : -1;
+    }
+  }
+  return -1;
+}
+function claimOf(args) {
+  if (args[0] !== "task" || args[1] !== "claim") return void 0;
+  let ws = "";
+  let id;
+  let isFor = false;
+  for (let k = 2; k < args.length; k++) {
+    const w = args[k] ?? "";
+    const [opt, eq] = w.startsWith("--") && w.includes("=") ? w.split("=", 2) : [w, void 0];
+    if (opt === "--for" || opt === "-f") isFor = true;
+    if (opt === "-w" || opt === "--workstream") ws = eq ?? args[k + 1] ?? "";
+    if (CLAIM_VALUE_OPTS.has(opt ?? "")) {
+      if (eq === void 0) k++;
+    } else if (id === void 0 && !w.startsWith("-")) id = w;
+  }
+  if (!isFor || id === void 0) return void 0;
+  const slash = id.indexOf("/");
+  if (slash >= 0) {
+    if (ws === "") ws = id.slice(0, slash);
+    id = id.slice(slash + 1);
+  }
+  return { ws, id };
+}
 function dispatchedTasks(command) {
   const out = [];
   for (const words of shellSegments(command)) {
-    if (words[0] !== "mu" && !words[0]?.endsWith("/mu")) continue;
-    if (words[1] !== "task" || words[2] !== "claim") continue;
-    let ws = "";
-    let id;
-    let isFor = false;
-    for (let k = 3; k < words.length; k++) {
-      const w = words[k] ?? "";
-      const [opt, eq] = w.startsWith("--") && w.includes("=") ? w.split("=", 2) : [w, void 0];
-      if (opt === "--for" || opt === "-f") isFor = true;
-      if (opt === "-w" || opt === "--workstream") ws = eq ?? words[k + 1] ?? "";
-      if (CLAIM_VALUE_OPTS.has(opt ?? "")) {
-        if (eq === void 0) k++;
-      } else if (id === void 0 && !w.startsWith("-")) id = w;
-    }
-    if (!isFor || id === void 0) continue;
-    const slash = id.indexOf("/");
-    if (slash >= 0) {
-      if (ws === "") ws = id.slice(0, slash);
-      id = id.slice(slash + 1);
-    }
-    if (ws !== "scratch") out.push({ ws, id });
+    const args = muArgs(words);
+    const claim = args === void 0 ? void 0 : claimOf(args);
+    if (claim !== void 0 && claim.ws !== "scratch") out.push(claim);
   }
   return out;
 }
@@ -994,7 +1060,7 @@ function registerRefuteNudge(pi, run = defaultRunner(), nested = notNested) {
 }
 
 // extension/mu-pi.ts
-var EXT_VERSION = true ? "3.8.0" : void 0;
+var EXT_VERSION = true ? "3.9.0" : void 0;
 var FRESH_COMMAND = "mu-fresh";
 var FRESH_TIMEOUT_MS = 3e4;
 function commandName(name) {