import { McpServer, ResourceTemplate } from "@modelcontextprotocol/sdk/server/mcp.js";
import { HASSClient } from "../hass/hassClient";
import { ConfigMcpDef } from "../hass/types";
import { ZodRawShape } from "zod";
export abstract class BaseMcp {
public server: McpServer;
public refClient: { ref: HASSClient, ensureConnection: () => Promise<void> };
public options: ConfigMcpDef;
abstract register(): void;
constructor(
server: McpServer,
refClient: { ref: HASSClient, ensureConnection: () => Promise<void> },
options: ConfigMcpDef
) {
this.server = server;
this.refClient = refClient;
this.options = options;
}
ensureConnection = () => this.refClient.ensureConnection();
get client() {
return this.refClient.ref;
}
// eslint-disable-next-line @typescript-eslint/no-explicit-any
private transformResourceResultToTool(name: string, result: any, isStructuredContent = false) {
if (this.options.DEBUG) {
console.error(`Raw resource ${name}:`, JSON.stringify(result, undefined, 0));
}
const content = result.contents[0] || {};
const noContent = !content.text;
if (noContent) {
content.text = "No data found";
} else if (result.contents.length > 1) {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
content.text = `[${result.contents.map((c: any) => c.text).join(",")}]`;
}
content.type = "text";
delete content.uri;
delete content.mimeType;
const final = isStructuredContent ? {
structuredContent: typeof content.text === "string" ? JSON.parse(content.text) : content.text,
content: [content]
} : {
content: [content]
};
if (this.options.DEBUG) {
console.error(`Resource to Tool ${name}:`, JSON.stringify(final, undefined, 0));
}
return final;
}
registerResourceOrTool<InputArgs extends ZodRawShape, OutputArgs extends ZodRawShape>(
name: string,
templateOrUri: string | ResourceTemplate,
options: {
title: string;
mimeType: string;
description: string;
inputSchema?: InputArgs;
outputSchema?: OutputArgs;
},
// eslint-disable-next-line @typescript-eslint/no-explicit-any
handler: (...args: any[]) => any
) {
if (this.options.LIMIT_RESOURCES === 0) {
console.error(`Warning: Limit reached for resources, cannot register ${name}`);
return;
} else if (this.options.LIMIT_RESOURCES > 0) {
this.options.LIMIT_RESOURCES--;
if (this.options.LIMIT_RESOURCES === 1) {
console.error(`Warning: Limit reached for resources, this is the last one registered : ${name}`);
}
}
if (this.options.RESOURCES_TO_TOOLS) {
// Translate resource registration to tool registration
this.server.registerTool(
name,
{
title: options.title,
description: options.description,
inputSchema: options.inputSchema,
outputSchema: options.outputSchema,
annotations: {
readOnlyHint: true
}
},
// eslint-disable-next-line @typescript-eslint/no-explicit-any
(async (args: any) => {
const firstArg = typeof templateOrUri === "string" ? undefined : new URL("http://dummy/" + name);
const result = await handler(firstArg, args);
return this.transformResourceResultToTool(name, result, !!options.outputSchema);
// eslint-disable-next-line @typescript-eslint/no-explicit-any
}) as any
);
} else {
this.server.registerResource(name, templateOrUri as string, options, handler);
}
}
}
