src/plugins/agents/builtin/opencode.ts (1)

390-439: Flush the trailing buffer on process end to avoid dropping the final JSONL record.

If stdout ends without a newline, the last buffered line is never parsed, so the final message is lost. Add an onEnd wrapper (and reuse a shared line handler) to flush the remaining buffer before delegating to the original callback.

🔧 Suggested fix

   override execute(
     prompt: string,
     files?: AgentFileContext[],
     options?: AgentExecuteOptions
   ): AgentExecutionHandle {
     let buffer = '';
 
+    const handleLine = (line: string) => {
+      const trimmed = line.trim();
+      if (!trimmed) return;
+
+      // Parse raw JSONL lines and forward to onJsonlMessage for subagent tracing
+      if (options?.onJsonlMessage) {
+        if (trimmed.startsWith('{')) {
+          try {
+            const parsed = JSON.parse(trimmed);
+            options.onJsonlMessage(parsed);
+          } catch {
+            // Not valid JSON, skip
+          }
+        }
+      }
+
+      // Process for display events
+      const events = parseOpenCodeJsonLine(trimmed);
+      if (events.length > 0) {
+        if (options?.onStdoutSegments) {
+          const segments = processAgentEventsToSegments(events);
+          if (segments.length > 0) {
+            options.onStdoutSegments(segments);
+          }
+        }
+        if (options?.onStdout) {
+          const parsed = processAgentEvents(events);
+          if (parsed.length > 0) {
+            options.onStdout(parsed);
+          }
+        }
+      }
+    };
+
+    const flushBuffer = () => {
+      if (!buffer) return;
+      handleLine(buffer);
+      buffer = '';
+    };
+
     // Wrap callbacks to parse JSON events
     const parsedOptions: AgentExecuteOptions = {
       ...options,
       onStdout: (options?.onStdout || options?.onStdoutSegments || options?.onJsonlMessage)
         ? (data: string) => {
             buffer += data;
 
             // If no newline, wait for more data
             if (!buffer.includes('\n')) return;
 
             const lines = buffer.split('\n');
             buffer = lines.pop() ?? ''; // Keep the last partial line (or empty string) in buffer
 
             for (const line of lines) {
-              const trimmed = line.trim();
-              if (!trimmed) continue;
-
-              // Parse raw JSONL lines and forward to onJsonlMessage for subagent tracing
-              if (options?.onJsonlMessage) {
-                if (trimmed.startsWith('{')) {
-                  try {
-                    const parsed = JSON.parse(trimmed);
-                    options.onJsonlMessage(parsed);
-                  } catch {
-                    // Not valid JSON, skip
-                  }
-                }
-              }
-
-              // Process for display events
-              const events = parseOpenCodeJsonLine(trimmed);
-              if (events.length > 0) {
-                // Call TUI-native segments callback if provided
-                if (options?.onStdoutSegments) {
-                  const segments = processAgentEventsToSegments(events);
-                  if (segments.length > 0) {
-                    options.onStdoutSegments(segments);
-                  }
-                }
-                // Also call legacy string callback if provided
-                if (options?.onStdout) {
-                  const parsed = processAgentEvents(events);
-                  if (parsed.length > 0) {
-                    options.onStdout(parsed);
-                  }
-                }
-              }
+              handleLine(line);
             }
           }
         : undefined,
+      onEnd: (result) => {
+        flushBuffer();
+        options?.onEnd?.(result);
+      },
     };
 
     return super.execute(prompt, files, parsedOptions);
   }