opencode-immune 1.0.19 → 1.0.21

package/dist/plugin.js

@@ -902,86 +902,84 @@ function createEventHandler(state) {
 }
 // ═══════════════════════════════════════════════════════════════════════════════
 // HOOK 9: MULTI-CYCLE AUTOMATION (PRE_COMMIT + CYCLE_COMPLETE)
+// Uses experimental.text.complete to detect signal markers in assistant output.
+// chat.message only fires for user messages (output.message is UserMessage),
+// so signal detection MUST use text.complete which fires for assistant text.
 // ═══════════════════════════════════════════════════════════════════════════════
 const MAX_CYCLES = 10;
 const PRE_COMMIT_MARKER = "0-ULTRAWORK: PRE_COMMIT";
 const CYCLE_COMPLETE_MARKER = "0-ULTRAWORK: CYCLE_COMPLETE";
 const NEXT_TASK_PATTERN = /Next task:\s*(.+)/;
-const RATE_LIMIT_MESSAGE_PATTERN = /too many requests|rate_limit|rate limit/i;
 const ALL_CYCLES_COMPLETE_MARKER = "0-ULTRAWORK: ALL_CYCLES_COMPLETE";
 /**
- * chat.message part: scans assistant messages for PRE_COMMIT and CYCLE_COMPLETE markers.
+ * experimental.text.complete: scans completed assistant text for signal markers.
  *
- * PRE_COMMIT → executes /commit via client.session.command()
- * CYCLE_COMPLETE → creates a new session and sends bootstrap prompt
+ * Handles the full cycle-end sequence:
+ * 1. PRE_COMMIT detected → execute /commit → await result
+ * 2. CYCLE_COMPLETE detected → create new session → send AUTO-CYCLE prompt
+ * Steps are sequential: new session only starts AFTER commit completes (or fails).
+ *
+ * ALL_CYCLES_COMPLETE → clears ultrawork marker, no new session.
  */
-function createMultiCycleHandler(state) {
+function createTextCompleteHandler(state) {
     return async (input, output) => {
         const sessionID = input.sessionID;
-        // Extract text content from parts
-        const parts = output.parts ?? [];
-        let messageContent = "";
-        for (const p of parts) {
-            if ("type" in p && p.type === "text" && "text" in p) {
-                messageContent += " " + p.text;
-            }
-        }
-        messageContent = messageContent.trim();
-        if (!messageContent)
+        const text = output.text ?? "";
+        if (!text)
             return;
-        const messageRole = output.message?.role;
-        // ── ALL_CYCLES_COMPLETE: clear ultrawork marker (checked before managed session guard) ──
-        if (messageContent.includes(ALL_CYCLES_COMPLETE_MARKER)) {
+        // ── ALL_CYCLES_COMPLETE: clear ultrawork marker ──
+        if (text.includes(ALL_CYCLES_COMPLETE_MARKER)) {
             await clearUltraworkMarker(state);
-            console.log("[opencode-immune] Multi-Cycle: ALL_CYCLES_COMPLETE detected, ultrawork marker cleared.");
+            console.log("[opencode-immune] Multi-Cycle: ALL_CYCLES_COMPLETE detected in text.complete, ultrawork marker cleared.");
+            return;
         }
-        // ── PRE_COMMIT: execute /commit (checked before managed session guard) ──
-        if (messageContent.includes(PRE_COMMIT_MARKER) && !state.commitPending && sessionID) {
-            state.commitPending = true;
-            console.log("[opencode-immune] Multi-Cycle: PRE_COMMIT detected, executing /commit...");
-            // Small delay to let the message finish rendering
-            setTimeout(async () => {
+        const hasPRECOMMIT = text.includes(PRE_COMMIT_MARKER);
+        const hasCYCLECOMPLETE = text.includes(CYCLE_COMPLETE_MARKER);
+        if (!hasPRECOMMIT && !hasCYCLECOMPLETE)
+            return;
+        // Run the full sequence in one async flow (no timers):
+        // commit → wait → new session
+        // Use setTimeout(0) only to not block the hook return.
+        setTimeout(async () => {
+            // ── Step 1: PRE_COMMIT → execute /commit and await result ──
+            if (hasPRECOMMIT && !state.commitPending && sessionID) {
+                state.commitPending = true;
+                console.log("[opencode-immune] Multi-Cycle: PRE_COMMIT detected, executing /commit...");
                 try {
-                    await state.input.client.session.command({
+                    const commitResult = await state.input.client.session.command({
                         body: {
                             command: "/commit",
                             arguments: "",
                         },
                         path: { id: sessionID },
                     });
-                    console.log("[opencode-immune] Multi-Cycle: /commit executed successfully.");
+                    console.log("[opencode-immune] Multi-Cycle: /commit completed.", commitResult?.response?.status ?? "");
                 }
                 catch (err) {
-                    console.error("[opencode-immune] Multi-Cycle: /commit failed:", err);
+                    console.error("[opencode-immune] Multi-Cycle: /commit failed (continuing to next cycle anyway):", err);
                 }
                 finally {
                     state.commitPending = false;
                 }
-            }, 2_000);
-        }
-        // ── CYCLE_COMPLETE: create new session (checked before managed session guard) ──
-        if (messageContent.includes(CYCLE_COMPLETE_MARKER)) {
-            state.cycleCount++;
-            if (state.cycleCount >= MAX_CYCLES) {
-                console.log(`[opencode-immune] Multi-Cycle: MAX_CYCLES (${MAX_CYCLES}) reached. Not creating new session.`);
-                await clearUltraworkMarker(state);
-                return;
             }
-            // Extract next task description
-            const taskMatch = messageContent.match(NEXT_TASK_PATTERN);
-            const nextTask = taskMatch?.[1]?.trim() ?? "Continue processing task backlog";
-            console.log(`[opencode-immune] Multi-Cycle: CYCLE_COMPLETE detected (cycle ${state.cycleCount}/${MAX_CYCLES}). ` +
-                `Creating new session for: "${nextTask}"`);
-            // Delay to let commit finish
-            setTimeout(async () => {
+            // ── Step 2: CYCLE_COMPLETE → create new session ──
+            if (hasCYCLECOMPLETE) {
+                state.cycleCount++;
+                if (state.cycleCount >= MAX_CYCLES) {
+                    console.log(`[opencode-immune] Multi-Cycle: MAX_CYCLES (${MAX_CYCLES}) reached. Not creating new session.`);
+                    await clearUltraworkMarker(state);
+                    return;
+                }
+                const taskMatch = text.match(NEXT_TASK_PATTERN);
+                const nextTask = taskMatch?.[1]?.trim() ?? "Continue processing task backlog";
+                console.log(`[opencode-immune] Multi-Cycle: CYCLE_COMPLETE detected (cycle ${state.cycleCount}/${MAX_CYCLES}). ` +
+                    `Creating new session for: "${nextTask}"`);
                 try {
-                    // Create a new session
                     const createResult = await state.input.client.session.create({
                         body: {
                             title: `Ultrawork Cycle ${state.cycleCount + 1}`,
                         },
                     });
-                    // Extract new session ID from the response
                     const newSessionData = createResult?.data;
                     const newSessionID = newSessionData?.id;
                     if (!newSessionID) {
@@ -990,7 +988,6 @@ function createMultiCycleHandler(state) {
                     }
                     console.log(`[opencode-immune] Multi-Cycle: New session created: ${newSessionID}`);
                     await addManagedUltraworkSession(state, newSessionID);
-                    // Send bootstrap prompt to the new session
                     await state.input.client.session.promptAsync({
                         body: {
                             agent: ULTRAWORK_AGENT,
@@ -1008,14 +1005,35 @@ function createMultiCycleHandler(state) {
                 catch (err) {
                     console.error("[opencode-immune] Multi-Cycle: Failed to create new session or send prompt:", err);
                 }
-            }, 8_000); // 8s delay: let /commit finish first
+            }
+        }, 0);
+    };
+}
+/**
+ * chat.message part: handles rate-limit text detection in managed root sessions.
+ * NOTE: chat.message fires for user messages only (output.message = UserMessage).
+ * Signal detection (PRE_COMMIT/CYCLE_COMPLETE) moved to experimental.text.complete.
+ */
+function createMultiCycleHandler(state) {
+    return async (input, output) => {
+        const sessionID = input.sessionID;
+        // Extract text content from parts
+        const parts = output.parts ?? [];
+        let messageContent = "";
+        for (const p of parts) {
+            if ("type" in p && p.type === "text" && "text" in p) {
+                messageContent += " " + p.text;
+            }
         }
+        messageContent = messageContent.trim();
+        if (!messageContent)
+            return;
         if (!isManagedRootUltraworkSession(state, sessionID))
             return;
         const managedSession = getManagedSession(state, sessionID);
         // Rate-limit message detection (only for managed root sessions)
+        const RATE_LIMIT_MESSAGE_PATTERN = /too many requests|rate_limit|rate limit/i;
         if (sessionID &&
-            messageRole === "assistant" &&
             RATE_LIMIT_MESSAGE_PATTERN.test(messageContent)) {
             if (managedSession && !managedSession.fallbackModel) {
                 await setSessionFallbackModel(state, sessionID, RATE_LIMIT_FALLBACK_MODEL);
@@ -1166,6 +1184,7 @@ async function server(input) {
         "tool.execute.after": withErrorBoundary("tool.execute.after", compositeToolAfter(toolAfterHandlers)),
         "experimental.chat.system.transform": withErrorBoundary("experimental.chat.system.transform", createSystemTransform(state)),
         "experimental.session.compacting": withErrorBoundary("experimental.session.compacting", createCompactionHandler(state)),
+        "experimental.text.complete": withErrorBoundary("experimental.text.complete", createTextCompleteHandler(state)),
     };
 }
 exports.default = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-immune",
-  "version": "1.0.19",
+  "version": "1.0.21",
   "description": "OpenCode plugin: session recovery, auto-retry, multi-cycle automation, context monitoring",
   "exports": {
     "./server": "./dist/plugin.js"