Add more data access tools

Author: nirinchev

src/tools/mongodb/collectionIndexes.ts

@@ -1,11 +1,12 @@
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { DbOperationArgs, MongoDBToolBase } from "./mongodbTool.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "./mongodbTool.js";
 import { ToolArgs } from "../tool.js";
 
 export class CollectionIndexesTool extends MongoDBToolBase {
     protected name = "collection-indexes";
     protected description = "Describe the indexes for a collection";
     protected argsShape = DbOperationArgs;
+    protected operationType: DbOperationType = "read";
 
     protected async execute({ database, collection }: ToolArgs<typeof DbOperationArgs>): Promise<CallToolResult> {
         const provider = this.ensureConnected();

src/tools/mongodb/connect.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import { CallToolResult, McpError } from "@modelcontextprotocol/sdk/types.js";
 import { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
-import { MongoDBToolBase } from "./mongodbTool.js";
+import { DbOperationType, MongoDBToolBase } from "./mongodbTool.js";
 import { ToolArgs } from "../tool";
 import { ErrorCodes } from "../../errors.js";
 
@@ -15,6 +15,8 @@ export class ConnectTool extends MongoDBToolBase {
             .describe("MongoDB connection string (in the mongodb:// or mongodb+srv:// format) or cluster name"),
     };
 
+    protected operationType: DbOperationType = "metadata";
+
     protected async execute({
         connectionStringOrClusterName,
     }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {

src/tools/mongodb/create/insertMany.ts

@@ -0,0 +1,40 @@
+import { z } from "zod";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
+import { ToolArgs } from "../../tool.js";
+
+export class InsertManyTool extends MongoDBToolBase {
+    protected name = "insert-many";
+    protected description = "Insert an array of documents into a MongoDB collection";
+    protected argsShape = {
+        ...DbOperationArgs,
+        documents: z
+            .array(z.object({}).passthrough().describe("An individual MongoDB document"))
+            .describe(
+                "The array of documents to insert, matching the syntax of the document argument of db.collection.insertMany()"
+            ),
+    };
+    protected operationType: DbOperationType = "create";
+
+    protected async execute({
+        database,
+        collection,
+        documents,
+    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = this.ensureConnected();
+        const result = await provider.insertMany(database, collection, documents);
+
+        return {
+            content: [
+                {
+                    text: `Inserted \`${result.insertedCount}\` documents into collection \`${collection}\``,
+                    type: "text",
+                },
+                {
+                    text: `Inserted IDs: ${Object.values(result.insertedIds).join(", ")}`,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}

src/tools/mongodb/insert/insertOne.ts renamed to src/tools/mongodb/create/insertOne.ts

@@ -1,14 +1,13 @@
 import { z } from "zod";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { MongoDBToolBase } from "../mongodbTool.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
 import { ToolArgs } from "../../tool.js";
 
 export class InsertOneTool extends MongoDBToolBase {
     protected name = "insert-one";
     protected description = "Insert a document into a MongoDB collection";
     protected argsShape = {
-        collection: z.string().describe("Collection name"),
-        database: z.string().describe("Database name"),
+        ...DbOperationArgs,
         document: z
             .object({})
             .passthrough()
@@ -17,6 +16,8 @@ export class InsertOneTool extends MongoDBToolBase {
             ),
     };
 
+    protected operationType: DbOperationType = "create";
+
     protected async execute({
         database,
         collection,

src/tools/mongodb/createIndex.ts

@@ -1,6 +1,6 @@
 import { z } from "zod";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { DbOperationArgs, MongoDBToolBase } from "./mongodbTool.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "./mongodbTool.js";
 import { ToolArgs } from "../tool.js";
 import { IndexDirection } from "mongodb";
 
@@ -12,6 +12,8 @@ export class CreateIndexTool extends MongoDBToolBase {
         keys: z.record(z.string(), z.custom<IndexDirection>()).describe("The index definition"),
     };
 
+    protected operationType: DbOperationType = "create";
+
     protected async execute({ database, collection, keys }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
         const provider = this.ensureConnected();
         const indexes = await provider.createIndexes(database, collection, [

src/tools/mongodb/delete/deleteMany.ts

@@ -0,0 +1,38 @@
+import { z } from "zod";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
+import { ToolArgs } from "../../tool.js";
+
+export class DeleteManyTool extends MongoDBToolBase {
+    protected name = "delete-many";
+    protected description = "Removes all documents that match the filter from a MongoDB collection";
+    protected argsShape = {
+        ...DbOperationArgs,
+        filter: z
+            .object({})
+            .passthrough()
+            .optional()
+            .describe(
+                "The query filter, specifying the deletion criteria. Matches the syntax of the filter argument of db.collection.deleteMany()"
+            ),
+    };
+    protected operationType: DbOperationType = "delete";
+
+    protected async execute({
+        database,
+        collection,
+        filter,
+    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = this.ensureConnected();
+        const result = await provider.deleteMany(database, collection, filter);
+
+        return {
+            content: [
+                {
+                    text: `Deleted \`${result.deletedCount}\` documents from collection \`${collection}\``,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}

src/tools/mongodb/delete/deleteOne.ts

@@ -0,0 +1,38 @@
+import { z } from "zod";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
+import { ToolArgs } from "../../tool.js";
+
+export class DeleteOneTool extends MongoDBToolBase {
+    protected name = "delete-one";
+    protected description = "Removes a single document that match the filter from a MongoDB collection";
+    protected argsShape = {
+        ...DbOperationArgs,
+        filter: z
+            .object({})
+            .passthrough()
+            .optional()
+            .describe(
+                "The query filter, specifying the deletion criteria. Matches the syntax of the filter argument of db.collection.deleteMany()"
+            ),
+    };
+    protected operationType: DbOperationType = "delete";
+
+    protected async execute({
+        database,
+        collection,
+        filter,
+    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = this.ensureConnected();
+        const result = await provider.deleteOne(database, collection, filter);
+
+        return {
+            content: [
+                {
+                    text: `Deleted \`${result.deletedCount}\` documents from collection \`${collection}\``,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}

src/tools/mongodb/delete/dropCollection.ts

@@ -0,0 +1,27 @@
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
+import { ToolArgs } from "../../tool.js";
+
+export class DropCollectionTool extends MongoDBToolBase {
+    protected name = "drop-collection";
+    protected description =
+        "Removes a collection or view from the database. The method also removes any indexes associated with the dropped collection.";
+    protected argsShape = {
+        ...DbOperationArgs,
+    };
+    protected operationType: DbOperationType = "delete";
+
+    protected async execute({ database, collection }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = this.ensureConnected();
+        const result = await provider.dropCollection(database, collection);
+
+        return {
+            content: [
+                {
+                    text: `${result ? "Successfully dropped" : "Failed to drop"} collection \`${collection}\` from database \`${database}\``,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}

src/tools/mongodb/delete/dropDatabase.ts

@@ -0,0 +1,26 @@
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { DbOperationArgs, DbOperationType, MongoDBToolBase } from "../mongodbTool.js";
+import { ToolArgs } from "../../tool.js";
+
+export class DropDatabaseTool extends MongoDBToolBase {
+    protected name = "drop-database";
+    protected description = "Removes the specified database, deleting the associated data files";
+    protected argsShape = {
+        database: DbOperationArgs.database,
+    };
+    protected operationType: DbOperationType = "delete";
+
+    protected async execute({ database }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
+        const provider = this.ensureConnected();
+        const result = await provider.dropDatabase(database);
+
+        return {
+            content: [
+                {
+                    text: `${result.ok ? "Successfully dropped" : "Failed to drop"} database \`${database}\``,
+                    type: "text",
+                },
+            ],
+        };
+    }
+}

src/tools/mongodb/index.ts

@@ -1,14 +1,26 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { State } from "../../state.js";
 import { ConnectTool } from "./connect.js";
-import { ListCollectionsTool } from "./listCollections.js";
+import { ListCollectionsTool } from "./metadata/listCollections.js";
 import { CollectionIndexesTool } from "./collectionIndexes.js";
-import { ListDatabasesTool } from "./listDatabases.js";
+import { ListDatabasesTool } from "./metadata/listDatabases.js";
 import { MongoDBToolState } from "./mongodbTool.js";
 import { CreateIndexTool } from "./createIndex.js";
-import { CollectionSchemaTool } from "./collectionSchema.js";
-import { InsertOneTool } from "./insert/insertOne.js";
-import { FindTool } from "./find/find.js";
+import { CollectionSchemaTool } from "./metadata/collectionSchema.js";
+import { InsertOneTool } from "./create/insertOne.js";
+import { FindTool } from "./read/find.js";
+import { InsertManyTool } from "./create/insertMany.js";
+import { DeleteManyTool } from "./delete/deleteMany.js";
+import { DeleteOneTool } from "./delete/deleteOne.js";
+import { CollectionStorageSizeTool } from "./metadata/collectionStorageSize.js";
+import { CountTool } from "./read/count.js";
+import { DbStatsTool } from "./metadata/dbStats.js";
+import { AggregateTool } from "./read/aggregate.js";
+import { UpdateOneTool } from "./update/updateOne.js";
+import { UpdateManyTool } from "./update/updateMany.js";
+import { RenameCollectionTool } from "./update/renameCollection.js";
+import { DropDatabaseTool } from "./delete/dropDatabase.js";
+import { DropCollectionTool } from "./delete/dropCollection.js";
 
 export function registerMongoDBTools(server: McpServer, state: State) {
     const mongodbToolState: MongoDBToolState = {};
@@ -22,6 +34,18 @@ export function registerMongoDBTools(server: McpServer, state: State) {
         CollectionSchemaTool,
         InsertOneTool,
         FindTool,
+        InsertManyTool,
+        DeleteManyTool,
+        DeleteOneTool,
+        CollectionStorageSizeTool,
+        CountTool,
+        DbStatsTool,
+        AggregateTool,
+        UpdateOneTool,
+        UpdateManyTool,
+        RenameCollectionTool,
+        DropDatabaseTool,
+        DropCollectionTool,
     ];
 
     for (const tool of tools) {