deleteOne.ts

import { z } from "zod";
import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
import { DbOperationArgs, MongoDBToolBase } from "../mongodbTool.js";
import { ToolArgs, OperationType } 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: OperationType = "delete";

    protected async execute({
        database,
        collection,
        filter,
    }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
        const provider = await this.ensureConnected();
        const result = await provider.deleteOne(database, collection, filter);

        return {
            content: [
                {
                    text: `Deleted \`${result.deletedCount}\` documents from collection \`${collection}\``,
                    type: "text",
                },
            ],
        };
    }
}