add deploymentid to telemetry data for create and delete local atlas tool

Author: jeroenvervaeke

src/telemetry/types.ts

@@ -32,6 +32,7 @@ export type ToolEventProperties = {
     org_id?: string;
     cluster_name?: string;
     is_atlas?: boolean;
+    atlas_local_deployment_id?: string;
 };
 
 export type ToolEvent = TelemetryEvent<ToolEventProperties>;

src/tools/atlasLocal/atlasLocalTool.ts

@@ -6,6 +6,7 @@ import type { Client } from "@mongodb-js-preview/atlas-local";
 
 export abstract class AtlasLocalToolBase extends ToolBase {
     public category: ToolCategory = "atlas-local";
+    protected deploymentId?: string;
 
     protected verifyAllowed(): boolean {
         return this.session.atlasLocalClient !== undefined && super.verifyAllowed();
@@ -38,6 +39,15 @@ please log a ticket here: https://github.com/mongodb-js/mongodb-mcp-server/issue
         return this.executeWithAtlasLocalClient(client, ...args);
     }
 
+    protected async lookupDeploymentIdAndAddToTelemetryMetadata(
+        client: Client,
+        containerId: string,
+
+    ): Promise<void> {
+        const deploymentId = await client.getDeploymentId(containerId);
+        this.deploymentId = deploymentId;
+    }
+
     protected abstract executeWithAtlasLocalClient(
         client: Client,
         ...args: Parameters<ToolCallback<typeof this.argsShape>>
@@ -67,11 +77,9 @@ please log a ticket here: https://github.com/mongodb-js/mongodb-mcp-server/issue
         return super.handleError(error, args);
     }
 
-    protected resolveTelemetryMetadata(
-        ...args: Parameters<ToolCallback<typeof this.argsShape>>
-    ): TelemetryToolMetadata {
-        // TODO: include deployment id in the metadata where possible
-        void args; // this shuts up the eslint rule until we implement the TODO above
-        return {};
+    protected resolveTelemetryMetadata(): TelemetryToolMetadata {
+        return {
+            atlasLocaldeploymentId: this.deploymentId,
+        };
     }
 }

src/tools/atlasLocal/create/createDeployment.ts

@@ -26,6 +26,9 @@ export class CreateDeploymentTool extends AtlasLocalToolBase {
         // Create the deployment
         const deployment = await client.createDeployment(deploymentOptions);
 
+        // Lookup the deployment id and add it to the telemetry metadata
+        await this.lookupDeploymentIdAndAddToTelemetryMetadata(client, deployment.containerId);
+
         return {
             content: [
                 {

src/tools/atlasLocal/delete/deleteDeployment.ts

@@ -16,6 +16,9 @@ export class DeleteDeploymentTool extends AtlasLocalToolBase {
         client: Client,
         { deploymentName }: ToolArgs<typeof this.argsShape>
     ): Promise<CallToolResult> {
+        // Lookup the deployment id and add it to the telemetry metadata
+        await this.lookupDeploymentIdAndAddToTelemetryMetadata(client, deploymentName);
+
         // Delete the deployment
         await client.deleteDeployment(deploymentName);
 

src/tools/tool.ts

@@ -16,6 +16,7 @@ export type ToolCategory = "mongodb" | "atlas" | "atlas-local";
 export type TelemetryToolMetadata = {
     projectId?: string;
     orgId?: string;
+    atlasLocaldeploymentId?: string;
 };
 
 export type ToolConstructor = new (session: Session, config: UserConfig, telemetry: Telemetry) => ToolBase;
@@ -232,6 +233,10 @@ export abstract class ToolBase {
             event.properties.project_id = metadata.projectId;
         }
 
+        if (metadata?.atlasLocaldeploymentId) {
+            event.properties.atlas_local_deployment_id = metadata.atlasLocaldeploymentId;
+        }
+
         await this.telemetry.emitEvents([event]);
     }
 }