PiAPI-MCP Server

src
tools
indexes
import { BaseTool, ToolParams } from "../base/tool.js";
import { IndexDirection } from "mongodb";
import { db } from "../../mongodb/client.js";

interface CreateIndexParams extends ToolParams {
  collection: string;
  indexSpec: { [key: string]: IndexDirection };
  [key: string]: unknown;
}

export class CreateIndexTool extends BaseTool<CreateIndexParams> {
  name = "createIndex";
  description = "Create a new index on a collection";
  inputSchema = {
    type: "object" as const,
    properties: {
      collection: {
        type: "string",
        description: "Name of the collection",
      },
      indexSpec: {
        type: "object",
        description:
          "Index specification (e.g., { field: 1 } for ascending index)",
      },
    },
    required: ["collection", "indexSpec"],
  };

  async execute(params: CreateIndexParams) {
    try {
      const collection = this.validateCollection(params.collection);
      const indexName = await db
        .collection(collection)
        .createIndex(params.indexSpec);

      return {
        content: [
          {
            type: "text" as const,
            text: JSON.stringify({ indexName }, null, 2),
          },
        ],
        isError: false,
      };
    } catch (error) {
      return this.handleError(error);
    }
  }
}