driggsby 0.1.6 → 0.1.7

package/dist/shim/server.js

@@ -1,5 +1,4 @@
 /* eslint-disable @typescript-eslint/no-deprecated */
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
 import { ensureBrokerRunning } from "../broker/launch.js";
@@ -9,6 +8,7 @@ import { KeyringSecretStore } from "../broker/secret-store.js";
 import { retryOperation } from "../lib/retry.js";
 import { buildBrokerInvestigationMessage, errorMessageIncludesReauthenticationCommand, formatRetryWindow, } from "../lib/user-guidance.js";
 import { formatStatusText } from "../cli/format.js";
+import { LifecycleAwareStdioServerTransport } from "./stdio-transport.js";
 const LOCAL_STATUS_TOOL = {
     description: "Report readiness and connectivity for the shared local Driggsby broker.",
     inputSchema: {
@@ -27,7 +27,8 @@ export async function runMcpServerCommand(runtimePaths, entrypointPath) {
         secretStore,
     });
     const server = createLocalShimServer(runtimePaths, secretStore);
-    await server.connect(new StdioServerTransport());
+    const transport = new LifecycleAwareStdioServerTransport();
+    await server.connect(transport);
 }
 export function createLocalShimServer(runtimePaths, secretStore) {
     const server = new Server({

package/dist/shim/stdio-transport.js

@@ -0,0 +1,74 @@
+import process from "node:process";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+export class LifecycleAwareStdioServerTransport {
+    delegate;
+    stdin;
+    closingPromise = null;
+    started = false;
+    onclose;
+    onerror;
+    onmessage;
+    handleInputClosed = () => {
+        void this.close().catch((error) => {
+            this.onerror?.(error instanceof Error
+                ? error
+                : new Error("The Driggsby MCP transport could not close cleanly."));
+        });
+    };
+    constructor(stdin = process.stdin, stdout = process.stdout) {
+        this.stdin = stdin;
+        this.delegate = new StdioServerTransport(stdin, stdout);
+        this.delegate.onclose = () => {
+            this.onclose?.();
+        };
+        this.delegate.onerror = (error) => {
+            this.onerror?.(error);
+        };
+        this.delegate.onmessage = (message) => {
+            this.onmessage?.(message);
+        };
+    }
+    async start() {
+        if (this.started) {
+            throw new Error("LifecycleAwareStdioServerTransport already started.");
+        }
+        this.started = true;
+        this.stdin.once("close", this.handleInputClosed);
+        this.stdin.once("end", this.handleInputClosed);
+        try {
+            await this.delegate.start();
+        }
+        catch (error) {
+            this.removeInputListeners();
+            this.started = false;
+            throw error;
+        }
+    }
+    async close() {
+        if (this.closingPromise !== null) {
+            await this.closingPromise;
+            return;
+        }
+        this.closingPromise = (async () => {
+            if (!this.started) {
+                return;
+            }
+            this.started = false;
+            this.removeInputListeners();
+            await this.delegate.close();
+        })();
+        try {
+            await this.closingPromise;
+        }
+        finally {
+            this.closingPromise = null;
+        }
+    }
+    async send(message) {
+        await this.delegate.send(message);
+    }
+    removeInputListeners() {
+        this.stdin.off("close", this.handleInputClosed);
+        this.stdin.off("end", this.handleInputClosed);
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "driggsby",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "Local MCP broker and CLI for connecting AI clients to Driggsby",
   "license": "UNLICENSED",
   "type": "module",