mirror of
https://github.com/modelcontextprotocol/servers.git
synced 2026-04-20 00:53:24 +02:00
346c29a086
Added tools to toggle simulated logging and resource updates on and off rather than have them start immediately upon connection
* Updated architecture.md
* In server/index.ts
- remove import of beginSimulatedResourceUpdates and beginSimulatedLogging
- remove clientConnected from createServer factory result
* In tools/index.ts
- import registerToggleLoggingTool and registerToggleSubscriberUpdatesTool
- in registerTools
- call registerToggleLoggingTool and registerToggleSubscriberUpdatesTool
* In logging.ts
- in beginSimulatedLogging
- refactor extract inline interval callback into function sendSimulatedLoggingMessage
- call sendSimulatedLoggingMessage right away to send the first message
- supply sendSimulatedLoggingMessage as interval callback
* In subscriptions.ts
- remove import of Transport
- remove transports map
- in beginSimulatedResourceUpdates()
- change arguments to server and sessionId
- check for the subsUpdateInterval for the session
- remove all transport storage and interaction
- instead use the server to send the notification
- in stopSimulatedResourceUpdates()
- remove management of transports map
* In stdio.ts, sse.ts, and streamableHttp.ts
- remove destructure and calling of clientConnected
* Added tools/toggle-logging.ts
- registers a tool that
- takes no arguments
- tracks clients that have been enabled by session id in a set
- if client isn't enabled,
- calls beginSimulatedLogging
- adds session id to client set
- else
- calls stopSimulatedLogging
- deletes session id from client set
- returns a message explaining what was done including what to expect when logging is enabled
* Added tools/toggle-subscriber-updates.ts
- registers a tool that
- takes no arguments
- tracks clients that have been enabled by session id in a set
- if client isn't enabled,
- calls beginSimulatedResourceUpdates
- adds session id to client set
- else
- calls stopSimulatedResourceUpdates
- deletes session id from client set
- returns a message explaining what was done including what to expect when logging is enabled
41 lines
1.3 KiB
TypeScript
41 lines
1.3 KiB
TypeScript
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
|
|
import {
|
|
beginSimulatedResourceUpdates,
|
|
stopSimulatedResourceUpdates,
|
|
} from "../resources/subscriptions.js";
|
|
|
|
const name = "toggle-subscriber-updates";
|
|
const config = {
|
|
title: "Toggle Subscriber Updates",
|
|
description: "Toggles simulated resource subscription updates on or off.",
|
|
inputSchema: {},
|
|
};
|
|
|
|
const clients: Set<string | undefined> = new Set<string | undefined>();
|
|
|
|
export const registerToggleSubscriberUpdatesTool = (server: McpServer) => {
|
|
server.registerTool(
|
|
name,
|
|
config,
|
|
async (_args, extra): Promise<CallToolResult> => {
|
|
const sessionId = extra?.sessionId;
|
|
|
|
let response: string;
|
|
if (clients.has(sessionId)) {
|
|
stopSimulatedResourceUpdates(sessionId);
|
|
clients.delete(sessionId);
|
|
response = `Stopped simulated resource updates for session ${sessionId}`;
|
|
} else {
|
|
beginSimulatedResourceUpdates(server, sessionId);
|
|
clients.add(sessionId);
|
|
response = `Started simulated resource updated notifications for session ${sessionId} at a 5 second pace. Client will receive updates for any resources the it is subscribed to.`;
|
|
}
|
|
|
|
return {
|
|
content: [{ type: "text", text: `${response}` }],
|
|
};
|
|
}
|
|
);
|
|
};
|