fix: Flush streams before exit

packages/opencode/src/cli/cmd/generate.ts

@@ -1,18 +1,11 @@
 import { Server } from "../../server/server"
 import type { CommandModule } from "yargs"
+import { EOL } from "os"
 
 export const GenerateCommand = {
   command: "generate",
   handler: async () => {
     const specs = await Server.openapi()
-    const json = JSON.stringify(specs, null, 2)
-
-    // Wait for stdout to finish writing before process.exit() is called
-    await new Promise<void>((resolve, reject) => {
-      process.stdout.write(json, (err) => {
-        if (err) reject(err)
-        else resolve()
-      })
-    })
+    process.stdout.write(JSON.stringify(specs, null, 2) + EOL)
   },
 } satisfies CommandModule

packages/opencode/src/index.ts

@@ -144,5 +144,20 @@ try {
   // Most notably, some docker-container-based MCP servers don't handle such signals unless
   // run using `docker run --init`.
   // Explicitly exit to avoid any hanging subprocesses.
-  process.exit()
+
+  /** Make sure all output operations have finished on the given stream */
+  async function awaitStream(stream: NodeJS.WriteStream) {
+    return new Promise<void>((resolve, reject) => {
+      stream.write("", (err) => {
+        err ? resolve() : reject(err)
+      })
+    })
+  }
+
+  await Promise.all([
+    awaitStream(process.stdout),
+    awaitStream(process.stderr),
+  ])
+
+  setImmediate(() => process.exit())
 }