@hydradb/mcp 1.1.1 → 1.2.1

package/dist/index.js

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { resolveConfig } from "./config.js";
-import { createHydraDBServer } from "./server.js";
+import { logger } from "./logger.js";
+import { awaitInFlight, beginShutdown, createHydraDBServer, inFlightCount, } from "./server.js";
 // Fail fast with a clean message if required config is missing. Honours the
 // canonical HYDRADB_* names (and the deprecated HYDRA_DB_* aliases).
 try {
@@ -11,16 +12,102 @@ catch (error) {
     console.error(`Error: ${error instanceof Error ? error.message : String(error)}`);
     process.exit(1);
 }
-async function main() {
-    try {
-        const server = createHydraDBServer();
-        const transport = new StdioServerTransport();
-        await server.connect(transport);
-    }
-    catch (error) {
-        console.error("Fatal error running server:", error);
+/**
+ * How long to let in-flight work finish after a stop signal.
+ *
+ * SIGTERM is how most MCP hosts and container runtimes stop a server, and an
+ * ingest that has been accepted but not yet answered is the case worth
+ * protecting: dropping it leaves the caller unable to tell whether the write
+ * committed. Short enough that nothing hangs a shutdown.
+ */
+const SHUTDOWN_GRACE_MS = 5000;
+/** Stack included when there is one — this is the last thing logged before exit. */
+function describe(error) {
+    if (error instanceof Error)
+        return error.stack ?? `${error.name}: ${error.message}`;
+    return String(error);
+}
+/**
+ * Wire process-level lifecycle handling.
+ *
+ * Before this the server had none: no SIGINT/SIGTERM handling, no
+ * unhandledRejection, no uncaughtException, and no transport close handling.
+ * SIGTERM terminated the process immediately, dropping any in-flight request
+ * with no log line; and after `connect()` resolved, a stray rejection killed the
+ * process with a raw stack on stderr and no MCP-level notice, so the host simply
+ * saw the server vanish mid-session.
+ *
+ * Everything here writes to stderr only. On stdio transport, stdout is the
+ * JSON-RPC channel and any stray byte on it corrupts the stream.
+ */
+function installLifecycle(server) {
+    let shuttingDown = false;
+    const shutdown = async (signal) => {
+        // A second signal means the operator is impatient; honour that rather
+        // than waiting out the grace period twice.
+        if (shuttingDown) {
+            logger.warn(`${signal} received again — exiting immediately`);
+            process.exit(130);
+        }
+        shuttingDown = true;
+        // Stop accepting NEW calls before draining. Without this, a call arriving
+        // after the drain resolves but before the transport closes is accepted and
+        // then aborted, which is the failure draining exists to prevent.
+        beginShutdown();
+        logger.info(`${signal} received — shutting down`);
+        const timer = setTimeout(() => {
+            logger.warn(`in-flight work did not finish within ${SHUTDOWN_GRACE_MS}ms — exiting anyway`);
+            process.exit(0);
+        }, SHUTDOWN_GRACE_MS);
+        // Do not let the grace timer itself hold the process open.
+        timer.unref();
+        try {
+            // Drain BEFORE closing. `server.close()` tears down the transport; it
+            // does not wait for handlers already running, so closing first would
+            // cut an accepted ingest off mid-write and leave the caller unable to
+            // tell whether it committed.
+            const pending = inFlightCount();
+            if (pending > 0) {
+                logger.info(`waiting for ${pending} in-flight tool call(s)`);
+                await awaitInFlight();
+            }
+            await server.close();
+        }
+        catch (error) {
+            logger.error("error while closing the server", { error: describe(error) });
+        }
+        clearTimeout(timer);
+        process.exit(0);
+    };
+    process.on("SIGINT", () => void shutdown("SIGINT"));
+    process.on("SIGTERM", () => void shutdown("SIGTERM"));
+    // The host closing the pipe is a normal end of session, not a failure.
+    server.onclose = () => {
+        if (shuttingDown)
+            return;
+        logger.info("transport closed — exiting");
+        process.exit(0);
+    };
+    server.onerror = (error) => {
+        logger.error("transport error", { error: describe(error) });
+    };
+    // Node terminates on an unhandled rejection by default, with a raw stack and
+    // no explanation of which server it came from. Log it in our own format, then
+    // exit non-zero so a supervisor restarts rather than silently continuing.
+    process.on("unhandledRejection", (reason) => {
+        logger.error("unhandled promise rejection", { error: describe(reason) });
+        process.exit(1);
+    });
+    process.on("uncaughtException", (error) => {
+        logger.error("uncaught exception", { error: describe(error) });
         process.exit(1);
-    }
+    });
+}
+async function main() {
+    const server = createHydraDBServer();
+    installLifecycle(server);
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
 }
 main().catch((error) => {
     console.error("Fatal error running server:", error);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAElD,4EAA4E;AAC5E,qEAAqE;AACrE,IAAI,CAAC;IACJ,aAAa,EAAE,CAAC;AACjB,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IAChB,OAAO,CAAC,KAAK,CAAC,UAAU,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,IAAI;IAClB,IAAI,CAAC;QACJ,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC;AACF,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACtB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAGA,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EACN,aAAa,EACb,aAAa,EACb,mBAAmB,EACnB,aAAa,GACb,MAAM,aAAa,CAAC;AAErB,4EAA4E;AAC5E,qEAAqE;AACrE,IAAI,CAAC;IACJ,aAAa,EAAE,CAAC;AACjB,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IAChB,OAAO,CAAC,KAAK,CAAC,UAAU,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,iBAAiB,GAAG,IAAK,CAAC;AAEhC,oFAAoF;AACpF,SAAS,QAAQ,CAAC,KAAc;IAC/B,IAAI,KAAK,YAAY,KAAK;QAAE,OAAO,KAAK,CAAC,KAAK,IAAI,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACpF,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACtB,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,SAAS,gBAAgB,CAAC,MAAc;IACvC,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAc,EAAE,EAAE;QACzC,sEAAsE;QACtE,2CAA2C;QAC3C,IAAI,YAAY,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,uCAAuC,CAAC,CAAC;YAC9D,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,YAAY,GAAG,IAAI,CAAC;QACpB,0EAA0E;QAC1E,2EAA2E;QAC3E,iEAAiE;QACjE,aAAa,EAAE,CAAC;QAChB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,2BAA2B,CAAC,CAAC;QAElD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC7B,MAAM,CAAC,IAAI,CACV,wCAAwC,iBAAiB,qBAAqB,CAC9E,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACtB,2DAA2D;QAC3D,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,IAAI,CAAC;YACJ,sEAAsE;YACtE,qEAAqE;YACrE,sEAAsE;YACtE,6BAA6B;YAC7B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;YAChC,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBACjB,MAAM,CAAC,IAAI,CAAC,eAAe,OAAO,yBAAyB,CAAC,CAAC;gBAC7D,MAAM,aAAa,EAAE,CAAC;YACvB,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC5E,CAAC;QACD,YAAY,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IACpD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAEtD,uEAAuE;IACvE,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE;QACrB,IAAI,YAAY;YAAE,OAAO;QACzB,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE;QAC1B,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC;IAEF,6EAA6E;IAC7E,8EAA8E;IAC9E,0EAA0E;IAC1E,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE;QAC3C,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACzE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;QACzC,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,IAAI;IAClB,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;IACrC,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACzB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACtB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}

package/dist/server.d.ts

@@ -1,7 +1,24 @@
 import { HydraDB } from "./hydra/index.js";
+/** Stop accepting tool calls. Idempotent. */
+export declare function beginShutdown(): void;
+/** Test-only: allow a fresh server in the same process after a shutdown test. */
+export declare function __resetShutdown(): void;
+/** Resolves once no tool call is running, or immediately if none is. */
+export declare function awaitInFlight(): Promise<void>;
+/** How many tool calls are currently running. Exported for tests and logging. */
+export declare function inFlightCount(): number;
+/**
+ * Whether the deprecated tool aliases are registered.
+ *
+ * Off by default as of 1.2.0. Anyone whose mcp.json still calls the old names
+ * sets HYDRADB_MCP_LEGACY_TOOLS=1 to restore them — one env var, no code change,
+ * and the opt-in is itself the adoption signal that a later removal needs and
+ * that nothing here could previously collect.
+ */
+export declare function legacyToolsEnabled(env?: NodeJS.ProcessEnv): boolean;
 /** Test-only: reset the once-per-process alias warning dedupe. */
 export declare function __resetAliasWarnings(): void;
-export declare function createHydraDBServer(hydraOverride?: HydraDB): import("@modelcontextprotocol/sdk/server/index.js").Server<{
+export declare function createHydraDBServer(hydraOverride?: HydraDB): import("@modelcontextprotocol/sdk/server").Server<{
     method: string;
     params?: {
         [x: string]: unknown;