PiAPI-MCP Server

src
tools
base
src/tools/base/tool.ts
import { McpError, ErrorCode } from "@modelcontextprotocol/sdk/types.js";

export interface ToolResponse {
  content: {
    type: "text";
    text: string;
  }[];
  isError: boolean;
  _meta?: Record<string, unknown>;
}

export type ToolParams = {
  [key: string]: unknown;
};

export abstract class BaseTool<T extends ToolParams = ToolParams> {
  abstract name: string;
  abstract description: string;
  abstract inputSchema: {
    type: "object";
    properties: Record<string, unknown>;
    required?: string[];
  };

  abstract execute(params: T): Promise<ToolResponse>;

  protected validateCollection(collection: unknown): string {
    if (typeof collection !== "string") {
      throw new McpError(
        ErrorCode.InvalidRequest,
        `Collection name must be a string, got ${typeof collection}`
      );
    }
    return collection;
  }

  protected validateObject(
    value: unknown,
    name: string
  ): Record<string, unknown> {
    if (!value || typeof value !== "object") {
      throw new McpError(ErrorCode.InvalidRequest, `${name} must be an object`);
    }
    return value as Record<string, unknown>;
  }

  protected handleError(error: unknown): ToolResponse {
    return {
      content: [
        {
          type: "text",
          text: error instanceof Error ? error.message : String(error),
        },
      ],
      isError: true,
    };
  }
}