@modelcontextprotocol/server-everything
Version:
MCP server that exercises all the features of the MCP protocol
178 lines (177 loc) • 7.6 kB
JavaScript
import { z } from "zod";
import { ResourceNotFoundError, ResourceTemplate, completable, } from "@modelcontextprotocol/server";
import { RESOURCE_TEMPLATE_ICONS } from "../server/icons.js";
// Resource types
export const RESOURCE_TYPE_TEXT = "Text";
export const RESOURCE_TYPE_BLOB = "Blob";
export const RESOURCE_TYPES = [
RESOURCE_TYPE_TEXT,
RESOURCE_TYPE_BLOB,
];
/**
* A completer function for resource types.
*
* This variable provides functionality to perform autocompletion for the resource types based on user input.
* It uses a schema description to validate the input and filters through a predefined list of resource types
* to return suggestions that start with the given input.
*
* The input value is expected to be a string representing the type of resource to fetch.
* The completion logic matches the input against available resource types.
*/
export const resourceTypeCompleter = completable(
// Prompt arguments carry no JSON Schema, only a description, so the
// description is the only place a client can learn the allowed values (#3985)
z
.string()
.describe(`Type of resource to fetch. Must be "${RESOURCE_TYPE_TEXT}" or "${RESOURCE_TYPE_BLOB}".`), (value) => {
return RESOURCE_TYPES.filter((t) => t.startsWith(value));
});
/**
* A completer function for resource IDs as strings.
*
* The `resourceIdCompleter` accepts a string input representing the ID of a text resource
* and validates whether the provided value corresponds to an integer resource ID.
*
* NOTE: Currently, prompt arguments can only be strings since type is not field of `PromptArgument`
* Consequently, we must define it as a string and convert the argument to number before using it
* https://modelcontextprotocol.io/specification/2025-11-25/schema#promptargument
*
* If the value is a valid integer, it returns the value within an array.
* Otherwise, it returns an empty array.
*
* The input string is first transformed into a number and checked to ensure it is an integer.
* This helps validate and suggest appropriate resource IDs.
*/
export const resourceIdForPromptCompleter = completable(z
.string()
.describe("ID of the resource to fetch. Must be a positive integer (1 or greater)."), (value) => {
const resourceId = Number(value);
return Number.isInteger(resourceId) && resourceId > 0 ? [value] : [];
});
/**
* A callback function that acts as a completer for resource ID values, validating and returning
* the input value as part of a resource template.
*
* @typedef {CompleteResourceTemplateCallback}
* @param {string} value - The input string value to be evaluated as a resource ID.
* @returns {string[]} Returns an array containing the input value if it represents a positive
* integer resource ID, otherwise returns an empty array.
*/
export const resourceIdForResourceTemplateCompleter = (value) => {
const resourceId = Number(value);
return Number.isInteger(resourceId) && resourceId > 0 ? [value] : [];
};
const uriBase = "demo://resource/dynamic";
const textUriBase = `${uriBase}/text`;
const blobUriBase = `${uriBase}/blob`;
const textUriTemplate = `${textUriBase}/{resourceId}`;
const blobUriTemplate = `${blobUriBase}/{resourceId}`;
/**
* Create a dynamic text resource
* - Exposed for use by embedded resource prompt example
* @param uri
* @param resourceId
*/
export const textResource = (uri, resourceId) => {
const timestamp = new Date().toLocaleTimeString();
return {
uri: uri.toString(),
mimeType: "text/plain",
text: `Resource ${resourceId}: This is a plaintext resource created at ${timestamp}`,
};
};
/**
* Create a dynamic blob resource
* - Exposed for use by embedded resource prompt example
* @param uri
* @param resourceId
*/
export const blobResource = (uri, resourceId) => {
const timestamp = new Date().toLocaleTimeString();
const resourceText = Buffer.from(`Resource ${resourceId}: This is a base64 blob created at ${timestamp}`).toString("base64");
return {
uri: uri.toString(),
mimeType: "application/octet-stream",
blob: resourceText,
};
};
/**
* Create a dynamic text resource URI
* - Exposed for use by embedded resource prompt example
* @param resourceId
*/
export const textResourceUri = (resourceId) => new URL(`${textUriBase}/${resourceId}`);
/**
* Create a dynamic blob resource URI
* - Exposed for use by embedded resource prompt example
* @param resourceId
*/
export const blobResourceUri = (resourceId) => new URL(`${blobUriBase}/${resourceId}`);
/**
* Parses the resource identifier from the provided variables and validates
* that it is a positive integer.
*
* The SDK only routes URIs that match the registered template to the
* resource handler, so by the time we get here the URI prefix is already
* known to be one of `textUriBase` / `blobUriBase`. Only the resourceId
* variable still needs validating.
*
* @param {URL} uri - The URI of the resource (used in the error message).
* @param {Record<string, unknown>} variables - Context variables including resourceId.
* @returns {number} The parsed and validated resource identifier as a positive integer.
* @throws {ResourceNotFoundError} If the resourceId is not a finite positive
* integer: no such resource exists, which the protocol reports as -32602
* with the URI in the error's data (#4852).
*/
const parseResourceId = (uri, variables) => {
const idxStr = String(variables.resourceId ?? "");
const idx = Number(idxStr);
if (Number.isFinite(idx) && Number.isInteger(idx) && idx > 0) {
return idx;
}
throw new ResourceNotFoundError(uri.toString());
};
/**
* Register resource templates with the MCP server.
* - Text and blob resources, dynamically generated from the URI {resourceId} variable
* - Any finite positive integer is acceptable for the resourceId variable
* - List resources method will not return these resources
* - These are only accessible via template URIs
* - Both blob and text resources:
* - have content that is dynamically generated, including a timestamp
* - have different template URIs
* - Blob: "demo://resource/dynamic/blob/{resourceId}"
* - Text: "demo://resource/dynamic/text/{resourceId}"
*
* @param server
*/
export const registerResourceTemplates = (server) => {
// Register the text resource template
server.registerResource("Dynamic Text Resource", new ResourceTemplate(textUriTemplate, {
list: undefined,
complete: { resourceId: resourceIdForResourceTemplateCompleter },
}), {
mimeType: "text/plain",
icons: RESOURCE_TEMPLATE_ICONS,
description: "Plaintext dynamic resource fabricated from the {resourceId} variable, which must be an integer.",
}, async (uri, variables) => {
const resourceId = parseResourceId(uri, variables);
return {
contents: [textResource(uri, resourceId)],
};
});
// Register the blob resource template
server.registerResource("Dynamic Blob Resource", new ResourceTemplate(blobUriTemplate, {
list: undefined,
complete: { resourceId: resourceIdForResourceTemplateCompleter },
}), {
mimeType: "application/octet-stream",
icons: RESOURCE_TEMPLATE_ICONS,
description: "Binary (base64) dynamic resource fabricated from the {resourceId} variable, which must be an integer.",
}, async (uri, variables) => {
const resourceId = parseResourceId(uri, variables);
return {
contents: [blobResource(uri, resourceId)],
};
});
};