save

Author: edgarw19

src/tools/mongodb/create/createSearchIndex.ts

@@ -1,20 +1,28 @@
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { DbOperationArgs, MongoDBToolBase, SearchIndexArgs } from "../mongodbTool.js";
+import { OperationType, ToolArgs } from "../../tool.js";
 
 export class CreateSearchIndexTool extends MongoDBToolBase {
-    constructor() {
-        super(...arguments);
-        this.name = "create-search-index";
-        this.description = "Create an Atlas Search index for a collection";
-        this.argsShape = {
-            ...DbOperationArgs,
-            name: SearchIndexArgs.name,
-            type: SearchIndexArgs.type,
-            analyzer: SearchIndexArgs.analyzer,
-            mappings: SearchIndexArgs.mappings,
-        };
-        this.operationType = "create";
-    }
-    async execute({ database, collection, name, type, analyzer, mappings, }) {
+    protected name = "create-search-index";
+    protected description = "Create an Atlas Search index for a collection";
+    protected argsShape = {
+        ...DbOperationArgs,
+        name: SearchIndexArgs.name,
+        type: SearchIndexArgs.type,
+        analyzer: SearchIndexArgs.analyzer,
+        mappings: SearchIndexArgs.mappings,
+    };
+
+    protected operationType: OperationType = "create";
+
+    protected async execute({
+        database,
+        collection,
+        name,
+        type,
+        analyzer,
+        mappings,
+    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
         const provider = await this.ensureConnected();
         const indexes = await provider.createSearchIndexes(database, collection, [
             {
@@ -26,6 +34,7 @@ export class CreateSearchIndexTool extends MongoDBToolBase {
                 },
             },
         ]);
+
         return {
             content: [
                 {
@@ -35,4 +44,4 @@ export class CreateSearchIndexTool extends MongoDBToolBase {
             ],
         };
     }
-}
+}

src/tools/mongodb/update/updateSearchIndex.ts

@@ -0,0 +1,47 @@
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, MongoDBToolBase, SearchIndexArgs } from "../mongodbTool.js";
+import { OperationType, ToolArgs } from "../../tool.js";
+import logger, { LogId } from "../../../logger.js";
+
+export class UpdateSearchIndexTool extends MongoDBToolBase {
+    protected name = "update-search-index";
+    protected description = "Updates a search index for a collection";
+    protected argsShape = {
+        ...DbOperationArgs,
+        name: SearchIndexArgs.name,
+        analyzer: SearchIndexArgs.analyzer,
+        mappings: SearchIndexArgs.mappings,
+    };
+
+    protected operationType: OperationType = "update";
+
+    protected async execute({
+        database,
+        collection,
+        name,
+        analyzer,
+        mappings,
+    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = await this.ensureConnected();
+        logger.info(
+            LogId.toolExecute,
+            "server",
+            `Server started with transport ${JSON.stringify({ definition: { analyzer, mappings } })}`
+        );
+        // @ts-expect-error: Interface expects a SearchIndexDefinition. However,
+        // passing analyzer/mappings at the root for the definition is necessary for the call to succeed.
+        await provider.updateSearchIndex(database, collection, name, {
+            analyzer,
+            mappings,
+        });
+
+        return {
+            content: [
+                {
+                    text: `Successfully updated index "${name}" on collection "${collection}" in database "${database}"`,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}