98 lines
2.5 KiB
TypeScript
98 lines
2.5 KiB
TypeScript
import {
|
|
registerAppResource,
|
|
registerAppTool,
|
|
RESOURCE_MIME_TYPE,
|
|
} from "@modelcontextprotocol/ext-apps/server";
|
|
import {
|
|
McpServer,
|
|
type CallToolResult,
|
|
type ReadResourceResult,
|
|
} from "@modelcontextprotocol/server";
|
|
import fs from "node:fs/promises";
|
|
import path from "node:path";
|
|
import { z } from "zod";
|
|
// Works both from source (server.ts) and compiled (dist/server.js)
|
|
const DIST_DIR = import.meta.filename.endsWith(".ts")
|
|
? path.join(import.meta.dirname, "dist")
|
|
: import.meta.dirname;
|
|
const RESOURCE_URI = "ui://get-time/mcp-app.html";
|
|
const SAMPLE_DOWNLOAD_URI = "resource:///sample-report.txt";
|
|
|
|
/**
|
|
* Creates a new MCP server instance with tools and resources registered.
|
|
*/
|
|
export function createServer(): McpServer {
|
|
const server = new McpServer({
|
|
name: "Integration Test Server",
|
|
version: "1.0.0",
|
|
});
|
|
|
|
registerAppTool(
|
|
server,
|
|
"get-time",
|
|
{
|
|
title: "Get Time",
|
|
description: "Returns the current server time.",
|
|
inputSchema: z.object({}),
|
|
outputSchema: z.object({
|
|
time: z.string(),
|
|
}),
|
|
_meta: { ui: { resourceUri: RESOURCE_URI } },
|
|
},
|
|
async (): Promise<CallToolResult> => {
|
|
const time = new Date().toISOString();
|
|
return {
|
|
content: [
|
|
{
|
|
type: "text",
|
|
text: JSON.stringify({ time }),
|
|
},
|
|
],
|
|
structuredContent: { time },
|
|
};
|
|
},
|
|
);
|
|
|
|
registerAppResource(
|
|
server,
|
|
RESOURCE_URI,
|
|
RESOURCE_URI,
|
|
{ mimeType: RESOURCE_MIME_TYPE },
|
|
async (): Promise<ReadResourceResult> => {
|
|
const html = await fs.readFile(
|
|
path.join(DIST_DIR, "mcp-app.html"),
|
|
"utf-8",
|
|
);
|
|
return {
|
|
contents: [
|
|
{ uri: RESOURCE_URI, mimeType: RESOURCE_MIME_TYPE, text: html },
|
|
],
|
|
};
|
|
},
|
|
);
|
|
|
|
// Sample downloadable resource — used to demo ResourceLink in ui/download-file
|
|
server.registerResource(
|
|
SAMPLE_DOWNLOAD_URI,
|
|
SAMPLE_DOWNLOAD_URI,
|
|
{
|
|
mimeType: "text/plain",
|
|
},
|
|
async (): Promise<ReadResourceResult> => {
|
|
const content = [
|
|
"Integration Test Server — Sample Report",
|
|
`Generated: ${new Date().toISOString()}`,
|
|
"",
|
|
"This file was downloaded via MCP ResourceLink.",
|
|
"The host resolved it by calling resources/read on the server.",
|
|
].join("\n");
|
|
return {
|
|
contents: [
|
|
{ uri: SAMPLE_DOWNLOAD_URI, mimeType: "text/plain", text: content },
|
|
],
|
|
};
|
|
},
|
|
);
|
|
|
|
return server;
|
|
}
|