tmcp 1.8.2 → 1.9.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tmcp",
-  "version": "1.8.2",
+  "version": "1.9.1",
   "description": "The main tmcp library",
   "main": "src/index.js",
   "type": "module",

package/src/index.js

@@ -4,7 +4,7 @@
  * @import { JSONRPCRequest, JSONRPCParams } from "json-rpc-2.0";
  * @import { ExtractURITemplateVariables } from "./internal/uri-template.js";
  * @import { CallToolResult, ReadResourceResult, GetPromptResult, ClientCapabilities as ClientCapabilitiesType, JSONRPCRequest as JSONRPCRequestType, JSONRPCResponse, CreateMessageRequestParams, CreateMessageResult, Resource, LoggingLevel, ToolAnnotations } from "./validation/index.js";
- * @import { Tool, Completion, Prompt, StoredResource, ServerOptions, ServerInfo, SubscriptionsKeys, McpEvents } from "./internal/internal.js";
+ * @import { Tool, Completion, Prompt, StoredResource, ServerOptions, ServerInfo, SubscriptionsKeys, McpEvents, Context as ContextType } from "./internal/internal.js";
  */
 import { JSONRPCClient, JSONRPCServer } from 'json-rpc-2.0';
 import { AsyncLocalStorage } from 'node:async_hooks';
@@ -29,6 +29,25 @@ import {
 } from './validation/version.js';
 import { should_version_negotiation_fail } from './validation/version.js';
 
+/**
+ * Information about a validated access token, provided to request handlers.
+ * @typedef {Object} AuthInfo
+ * @property {string} token - The access token.
+ * @property {string} clientId - The client ID associated with this token.
+ * @property {string[]} scopes - Scopes associated with this token.
+ * @property {number} [expiresAt] - When the token expires (in seconds since epoch).
+ * @property {URL} [resource] - The RFC 8707 resource server identifier for which this token is valid.
+ *   If set, this MUST match the MCP server's resource identifier (minus hash fragment).
+ * @property {Record<string, unknown>} [extra] - Additional data associated with the token.
+ *   This field should be used for any additional data that needs to be attached to the auth info.
+ */
+
+/**
+ * @typedef {Object} Context
+ * @property {string} [sessionId]
+ * @property {AuthInfo} [auth]
+ */
+
 /**
  * @type {SqidsType | undefined}
  */
@@ -110,9 +129,9 @@ export class McpServer {
 	};
 
 	/**
-	 * @type {AsyncLocalStorage<string | undefined>}
+	 * @type {AsyncLocalStorage<Context>}
 	 */
-	#session_storage = new AsyncLocalStorage();
+	#ctx_storage = new AsyncLocalStorage();
 
 	/**
 	 * @type {Map<string|undefined, ClientCapabilities>}
@@ -236,7 +255,11 @@ export class McpServer {
 	}
 
 	get #session_id() {
-		return this.#session_storage.getStore();
+		return this.#ctx_storage.getStore()?.sessionId;
+	}
+
+	get ctx() {
+		return this.#ctx_storage.getStore() ?? {};
 	}
 
 	get #client_capabilities() {
@@ -577,7 +600,7 @@ export class McpServer {
 	#init_completion() {
 		this.#server.addMethod(
 			'completion/complete',
-			({ argument, ref, context }) => {
+			async ({ argument, ref, context }) => {
 				const completions = this.#completions[ref.type];
 				if (!completions) return null;
 				const complete = completions.get(ref.uri ?? ref.name);
@@ -586,7 +609,7 @@ export class McpServer {
 				if (!actual_complete) return null;
 				return v.parse(
 					CompleteResultSchema,
-					actual_complete(argument.value, context),
+					await actual_complete(argument.value, context),
 				);
 			},
 		);
@@ -690,10 +713,10 @@ export class McpServer {
 	}
 	/**
 	 * @param {JSONRPCResponse | JSONRPCRequest} message
-	 * @param {string} [session_id]
+	 * @param {Context} [ctx]
 	 * @returns {ReturnType<JSONRPCServer['receive']> | ReturnType<JSONRPCClient['receive'] | undefined>}
 	 */
-	receive(message, session_id) {
+	receive(message, ctx) {
 		// Validate the message first
 		const validated_message = v.safeParse(
 			v.union([JSONRPCRequestSchema, JSONRPCNotificationSchema]),
@@ -707,15 +730,15 @@ export class McpServer {
 			// to each client that didn't explicitly set a log level too
 			if (
 				!!this.#options.capabilities?.logging &&
-				!this.#session_log_levels.has(session_id)
+				!this.#session_log_levels.has(ctx?.sessionId)
 			) {
 				this.#session_log_levels.set(
-					session_id,
+					ctx?.sessionId,
 					this.#options.logging?.default ?? 'info',
 				);
 			}
-			return this.#session_storage.run(
-				session_id,
+			return this.#ctx_storage.run(
+				ctx ?? {},
 				async () =>
 					await this.#server.receive(validated_message.output),
 			);
@@ -723,7 +746,7 @@ export class McpServer {
 		// It's a response - handle with client
 		const validated_response = v.parse(JSONRPCResponseSchema, message);
 		this.#lazyily_create_client();
-		return this.#session_storage.run(session_id, async () =>
+		return this.#ctx_storage.run(ctx ?? {}, async () =>
 			this.#client?.receive(validated_response),
 		);
 	}

package/src/internal/internal.d.ts

@@ -26,7 +26,7 @@ export type Tool<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
 export type Completion = (
 	query: string,
 	context: { arguments: Record<string, string> },
-) => CompleteResult;
+) => CompleteResult | Promise<CompleteResult>;
 
 export type Prompt<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
 	description: string;
@@ -90,4 +90,4 @@ export type McpEvents = {
 		};
 	}) => void;
 	initialize: (initialize_request: InitializeRequestParams) => void;
-};
+};

package/src/types/index.d.ts

@@ -11,6 +11,7 @@ declare module 'tmcp' {
 			uri: string;
 			name?: string;
 		}>;
+		get ctx(): Context;
 		currentClientCapabilities(): {
 			experimental?: {} | undefined;
 			sampling?: {} | undefined;
@@ -54,7 +55,7 @@ declare module 'tmcp' {
 			list?: () => Promise<Array<Resource>> | Array<Resource>;
 		}, execute: (uri: string, params: Record<TVariables, string | string[]>) => Promise<ReadResourceResult> | ReadResourceResult): void;
 		
-		receive(message: JSONRPCResponse | JSONRPCRequest, session_id?: string): ReturnType<JSONRPCServer["receive"]> | ReturnType<JSONRPCClient["receive"] | undefined>;
+		receive(message: JSONRPCResponse | JSONRPCRequest, ctx?: Context): ReturnType<JSONRPCServer["receive"]> | ReturnType<JSONRPCClient["receive"] | undefined>;
 		/**
 		 * Send a notification for subscriptions
 		 * */
@@ -76,11 +77,46 @@ declare module 'tmcp' {
 		log(level: LoggingLevel, data: unknown, logger?: string): void;
 		#private;
 	}
+	/**
+	 * Information about a validated access token, provided to request handlers.
+	 */
+	export type AuthInfo = {
+		/**
+		 * - The access token.
+		 */
+		token: string;
+		/**
+		 * - The client ID associated with this token.
+		 */
+		clientId: string;
+		/**
+		 * - Scopes associated with this token.
+		 */
+		scopes: string[];
+		/**
+		 * - When the token expires (in seconds since epoch).
+		 */
+		expiresAt?: number | undefined;
+		/**
+		 * - The RFC 8707 resource server identifier for which this token is valid.
+		 * If set, this MUST match the MCP server's resource identifier (minus hash fragment).
+		 */
+		resource?: URL | undefined;
+		/**
+		 * - Additional data associated with the token.
+		 * This field should be used for any additional data that needs to be attached to the auth info.
+		 */
+		extra?: Record<string, unknown> | undefined;
+	};
+	export type Context = {
+		sessionId?: string | undefined;
+		auth?: AuthInfo | undefined;
+	};
 	export type ClientCapabilities = ClientCapabilities_1;
 	type Completion = (
 		query: string,
 		context: { arguments: Record<string, string> },
-	) => CompleteResult;
+	) => CompleteResult | Promise<CompleteResult>;
 
 	type ServerOptions<TSchema extends StandardSchemaV1> = {
 		capabilities?: ServerCapabilities;

package/src/types/index.d.ts.map

@@ -56,6 +56,6 @@
 		null,
 		null
 	],
-	"mappings": ";;;;;cAmEaA,SAASA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aCo4CsCC,kBAAkBA;MC96ClEC,UAAUA;;;;;MAsCVC,aAAaA;;;;;;;;;;;;;MAabC,UAAUA;;;;;;MAMVC,iBAAiBA;;MAEjBC,SAASA;;;;;;;;;;;;ODgBRC,qBAAqBA;;;;;;;;;;;;;;OAuGrBC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;OA6BxBC,6BAA6BA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAuB7BC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmMxBC,cAAcA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4GdC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8PxBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BrBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2GrBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoEpBC,kBAAkBA;OAuGlBC,gCAAgCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8ChCC,yBAAyBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+LzBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;MA8H2BC,kBAAkBA;MASbC,uBAAuBA;MAGhCC,cAAcA;MAGVC,kBAAkBA;MAGrBC,eAAeA;MAGhBC,cAAcA;MAGFC,0BAA0BA;MAGjCC,mBAAmBA;MAY9BC,QAAQA;MAMDC,eAAeA;MAGlBC,YAAYA;MAGTC,eAAeA;;ME5/CnEC,IAAIA;;;;;;;MAOJC,cAAcA;;;;;;;MAOdC,cAAcA;;;;;MAKdC,qBAAqBA;;;;;;;;;;;;;;;;;MAiBrBC,4BAA4BA;;;;;;MAMrBC,2BAA2BA;;;;;;;OCjC1BC,iBAAiBA;;;;;;;;;;;;;;;;cAAjBA,iBAAiBA",
+	"mappings": ";;;;;cAsFaA,SAASA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aCi3CsCC,kBAAkBA;MC96ClEC,UAAUA;;;;;MAsCVC,aAAaA;;;;;;;;;;;;;MAabC,UAAUA;;;;;;MAMVC,iBAAiBA;;MAEjBC,SAASA;;;;;;;;;;;;ODgBRC,qBAAqBA;;;;;;;;;;;;;;OAuGrBC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;OA6BxBC,6BAA6BA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAuB7BC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmMxBC,cAAcA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4GdC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8PxBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BrBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2GrBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoEpBC,kBAAkBA;OAuGlBC,gCAAgCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8ChCC,yBAAyBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+LzBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;MA8H2BC,kBAAkBA;MASbC,uBAAuBA;MAGhCC,cAAcA;MAGVC,kBAAkBA;MAGrBC,eAAeA;MAGhBC,cAAcA;MAGFC,0BAA0BA;MAGjCC,mBAAmBA;MAY9BC,QAAQA;MAMDC,eAAeA;MAGlBC,YAAYA;MAGTC,eAAeA;;ME5/CnEC,IAAIA;;;;;;;MAOJC,cAAcA;;;;;;;MAOdC,cAAcA;;;;;MAKdC,qBAAqBA;;;;;;;;;;;;;;;;;MAiBrBC,4BAA4BA;;;;;;MAMrBC,2BAA2BA;;;;;;;OCjC1BC,iBAAiBA;;;;;;;;;;;;;;;;cAAjBA,iBAAiBA",
 	"ignoreList": []
 }