tmcp 1.0.2 → 1.1.0

package/package.json

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

package/src/index.d.ts

@@ -7,6 +7,13 @@ export class McpServer<StandardSchema extends StandardSchemaV1> {
      * @param {ServerOptions<StandardSchema>} options
      */
     constructor(server_info: ServerInfo, options: ServerOptions<StandardSchema>);
+    /**
+     * @template {keyof McpEvents} TEvent
+     * @param {TEvent} event
+     * @param {McpEvents[TEvent]} callback
+     * @param {AddEventListenerOptions} [options]
+     */
+    on<TEvent extends keyof McpEvents>(event: TEvent, callback: McpEvents[TEvent], options?: AddEventListenerOptions): void;
     /**
      * @template {StandardSchema | undefined} [TSchema=undefined]
      * @param {{ name: string; description: string; schema?: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema> extends Record<string, unknown> ? TSchema : never }} options
@@ -52,11 +59,19 @@ export class McpServer<StandardSchema extends StandardSchemaV1> {
      * @returns {ReturnType<JSONRPCServer['receive']>}
      */
     receive(request: JSONRPCRequest): ReturnType<JSONRPCServer["receive"]>;
+    /**
+     * Send a notification for subscriptions
+     * @param {SubscriptionsKeys} what
+     * @param {string} id
+     */
+    changed(what: SubscriptionsKeys, id: string): void;
     #private;
 }
 import type { StandardSchemaV1 } from "@standard-schema/spec";
+import type { McpEvents } from "./internal/internal.js";
 import type { Completion } from "./internal/internal.js";
 import type { JSONRPCRequest } from "json-rpc-2.0";
 import { JSONRPCServer } from 'json-rpc-2.0';
+import type { SubscriptionsKeys } from "./internal/internal.js";
 import type { ServerInfo } from "./internal/internal.js";
 import type { ServerOptions } from "./internal/internal.js";

package/src/index.js

@@ -1,7 +1,7 @@
 /**
  * @import { StandardSchemaV1 } from "@standard-schema/spec";
  * @import { JSONRPCRequest, JSONRPCParams } from "json-rpc-2.0";
- * @import { Tool, Completion, Prompt, Resource, ServerOptions, ServerInfo } from "./internal/internal.js";
+ * @import { Tool, Completion, Prompt, Resource, ServerOptions, ServerInfo, SubscriptionsKeys, McpEvents } from "./internal/internal.js";
  */
 import { JSONRPCServer, JSONRPCClient } from 'json-rpc-2.0';
 import { UriTemplateMatcher } from 'uri-template-matcher';
@@ -36,17 +36,25 @@ export class McpServer {
 		'ref/prompt': new Map(),
 		'ref/resource': new Map(),
 	};
+
+	#event_target = new EventTarget();
+
+	/**
+	 * @type {Record<SubscriptionsKeys, Set<string>>}
+	 */
+	#subscriptions = {
+		/**
+		 * @type {Set<string>}
+		 */
+		resource: new Set(),
+	};
+
 	/**
 	 * @param {ServerInfo} server_info
 	 * @param {ServerOptions<StandardSchema>} options
 	 */
 	constructor(server_info, options) {
 		this.#options = options;
-		if (options.send) {
-			this.#client = new JSONRPCClient((payload) => {
-				options.send?.(payload);
-			});
-		}
 		this.#server.addMethod('initialize', ({ protocolVersion }) => {
 			return {
 				protocolVersion,
@@ -66,6 +74,30 @@ export class McpServer {
 		this.#init_completion();
 	}
 
+	/**
+	 * @template {keyof McpEvents} TEvent
+	 * @param {TEvent} event
+	 * @param {McpEvents[TEvent]} callback
+	 * @param {AddEventListenerOptions} [options]
+	 */
+	on(event, callback, options) {
+		if (event === 'send' && !this.#client) {
+			this.#client = new JSONRPCClient((payload) => {
+				this.#event_target.dispatchEvent(
+					new CustomEvent('send', { detail: payload }),
+				);
+			});
+		}
+
+		this.#event_target.addEventListener(
+			event,
+			(e) => {
+				callback(/** @type {CustomEvent} */ (e).detail);
+			},
+			options,
+		);
+	}
+
 	/**
 	 * @param {string} method
 	 * @param {JSONRPCParams} [params]
@@ -190,6 +222,14 @@ export class McpServer {
 	 */
 	#init_resources() {
 		if (!this.#options.capabilities?.resources) return;
+
+		if (this.#options.capabilities?.resources?.subscribe) {
+			this.#server.addMethod('resources/subscribe', async ({ uri }) => {
+				this.#subscriptions.resource.add(uri);
+				return {};
+			});
+		}
+
 		this.#server.addMethod('resources/list', async () => {
 			return {
 				resources: [...this.#resources].reduce(
@@ -339,4 +379,20 @@ export class McpServer {
 	receive(request) {
 		return this.#server.receive(request);
 	}
+
+	/**
+	 * Send a notification for subscriptions
+	 * @param {SubscriptionsKeys} what
+	 * @param {string} id
+	 */
+	changed(what, id) {
+		if (this.#subscriptions[what].has(id)) {
+			const resource = this.#resources.get(id);
+			if (!resource) return;
+			this.#notify(`notifications/resources/updated`, {
+				uri: id,
+				title: resource.name,
+			});
+		}
+	}
 }

package/src/internal/internal.d.ts

@@ -31,11 +31,10 @@ export type Resource = {
 };
 
 export type ServerOptions<TSchema extends StandardSchemaV1> = {
-	send?: (payload: JSONRPCRequest) => void;
 	capabilities?: {
 		tools?: { listChanged?: boolean };
 		resources?: { listChanged?: boolean; subscribe?: boolean };
-		prompts?: { listChanged?: boolean; subscribe?: boolean };
+		prompts?: { listChanged?: boolean };
 	};
 	instructions?: string;
 	adapter: JsonSchemaAdapter<TSchema>;
@@ -46,3 +45,9 @@ export type ServerInfo = {
 	version: string;
 	description: string;
 };
+
+export type SubscriptionsKeys = 'resource';
+
+export type McpEvents = {
+	send: (message: JSONRPCRequest) => void;
+};