@orkestrel/mcp 0.0.35 → 0.0.37

package/dist/src/core/index.cjs

@@ -249,6 +249,8 @@ var JSONRPC_INTERNAL_ERROR = -32603;
 * Retained for the legacy branch alone. A modern fault answers
 * {@link JSONRPC_INTERNAL_ERROR}; this code survives only where an old-wire peer was
 * already characterized against it.
+* A `handshake` hook that rejects with an `MCPError` answers the legacy `initialize`
+* with that error's own code and data.
 */
 var JSONRPC_SERVER_ERROR = -32e3;
 /**
@@ -4113,6 +4115,26 @@ function isInitializeRequest(value) {
 	return owned.success && isJSONRPCInvocation(owned.value) && owned.value["method"] === "initialize";
 }
 /**
+* Determines whether a parsed value is a legacy MCP `ping` request.
+*
+* @remarks
+* Requires a valid request id and excludes the modern per-request wire shape.
+* Notifications and hostile or malformed values return false.
+*
+* @param value - The already-parsed value to test
+* @returns True if `value` is a legacy `ping` request; false otherwise
+*
+* @example
+* ```ts
+* isPingRequest({ jsonrpc: '2.0', method: 'ping', id: 1 }) // true
+* isPingRequest({ jsonrpc: '2.0', method: 'ping' }) // false
+* ```
+*/
+function isPingRequest(value) {
+	const owned = (0, _orkestrel_contract.attempt)(() => (0, _orkestrel_contract.cloneJSONRecord)(value));
+	return owned.success && isJSONRPCRequest(owned.value) && owned.value.method === "ping" && !isModernRequest(owned.value);
+}
+/**
 * Determines whether a JSON-RPC invocation uses the modern per-request MCP wire shape.
 *
 * @remarks
@@ -4752,6 +4774,23 @@ var MCPLegacy = class {
 		}) === void 0) return buildJSONRPCError(void 0, JSONRPC_INVALID_REQUEST, "Invalid Request");
 		switch (invocation.method) {
 			case "initialize": {
+				if (this.#options.handshake !== void 0) {
+					const closure = new AbortController();
+					const resolved = buildMethodOptions(options ?? {}, closure.signal);
+					try {
+						await this.#options.handshake(resolved);
+					} catch (error) {
+						if (resolved.signal.aborted) throw error;
+						if (isMCPError(error)) {
+							const context = (0, _orkestrel_contract.attempt)(() => (0, _orkestrel_contract.parseJSON)(JSON.stringify(error.context)));
+							return buildJSONRPCError(id, error.code, error.message, context.success ? context.value : void 0);
+						}
+						this.emitter.emit("error", error);
+						return buildJSONRPCError(id, JSONRPC_INTERNAL_ERROR, "Server error");
+					} finally {
+						closure.abort();
+					}
+				}
 				const requested = invocation.params?.["protocolVersion"];
 				return buildJSONRPCResult(id, buildInitializeResult(this.#options.identity.name, this.#options.identity.version, (0, _orkestrel_contract.isString)(requested) ? requested : void 0));
 			}
@@ -5111,6 +5150,9 @@ var MCPServer = class {
 	get identity() {
 		return this.#options.identity;
 	}
+	get handshake() {
+		return this.#options.handshake;
+	}
 	get methods() {
 		return this.#methods;
 	}
@@ -5527,7 +5569,7 @@ var MCPServer = class {
 		if (!isBoundedString(verified, this.#limits.state) || verified.length === 0) return this.#contain(/* @__PURE__ */ new Error("Continuation port opened a value outside the configured state bound"), id);
 		const state = parseMCPInputState(verified);
 		if (state === void 0) return this.#contain(/* @__PURE__ */ new Error("Continuation port opened a malformed protected payload"), id);
-		if (state.expiry <= Date.now() || state.id === id || state.version !== context.version || state.method !== request.method || state.name !== name || state.digest !== digest) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
+		if (state.expiry <= (configured.clock ?? Date.now)() || state.id === id || state.version !== context.version || state.method !== request.method || state.name !== name || state.digest !== digest) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
 		const responses = this.#checkAnswers(state.requests, inputResponses);
 		if (responses === void 0) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: an input response is missing or malformed");
 		const principal = await configured.principal(request, options);
@@ -5539,7 +5581,7 @@ var MCPServer = class {
 			responses,
 			...state.state !== void 0 ? { state: state.state } : {}
 		}, options);
-		if (state.expiry <= Date.now()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
+		if (state.expiry <= (configured.clock ?? Date.now)()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
 		if (selected === void 0) return void 0;
 		const round = this.#ownRound(selected);
 		if (round === void 0) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: input policy returned an invalid round or continuation context");
@@ -5580,8 +5622,8 @@ var MCPServer = class {
 			depth: this.#limits.depth
 		});
 		if (configured === void 0 || context === void 0 || !(0, _orkestrel_contract.isString)(principal) || principal.length === 0 || !(0, _orkestrel_contract.isFiniteNumber)(configured.ttl) || configured.ttl <= 0) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: input policy returned an invalid round or continuation context");
-		if (previous !== void 0 && previous <= Date.now()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
-		const expiry = Date.now() + configured.ttl;
+		if (previous !== void 0 && previous <= (configured.clock ?? Date.now)()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
+		const expiry = (configured.clock ?? Date.now)() + configured.ttl;
 		const protectedState = {
 			principal,
 			expiry,
@@ -5604,7 +5646,7 @@ var MCPServer = class {
 		if (serialized === void 0) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state exceeds the configured limit");
 		const requestState = await configured.continuation.seal(serialized);
 		if (!isBoundedString(requestState, this.#limits.state) || requestState.length === 0) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state exceeds the configured limit");
-		if (expiry <= Date.now() || previous !== void 0 && previous <= Date.now()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, previous === void 0 ? "Invalid params: request state expired before it could be issued" : "Invalid params: request state could not be verified for this retry");
+		if (expiry <= (configured.clock ?? Date.now)() || previous !== void 0 && previous <= (configured.clock ?? Date.now)()) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, previous === void 0 ? "Invalid params: request state expired before it could be issued" : "Invalid params: request state could not be verified for this retry");
 		return buildJSONRPCResult(id, {
 			resultType: "input_required",
 			inputRequests: round.requests,
@@ -6753,7 +6795,8 @@ function createMCPServer(options) {
 function createMCPLegacy(server) {
 	return new MCPLegacy({
 		dispatcher: server,
-		identity: server.identity
+		identity: server.identity,
+		...server.handshake === void 0 ? {} : { handshake: server.handshake }
 	});
 }
 /**
@@ -7028,6 +7071,7 @@ exports.isMCPTextResource = isMCPTextResource;
 exports.isMCPToolAnnotations = isMCPToolAnnotations;
 exports.isMCPVersion = isMCPVersion;
 exports.isModernRequest = isModernRequest;
+exports.isPingRequest = isPingRequest;
 exports.isRFC3339Date = isRFC3339Date;
 exports.isRFC3339DateTime = isRFC3339DateTime;
 exports.isStandardBase64 = isStandardBase64;