feat: Adds the atlas-list-performance-advisor base tool (#528)

Author: kylelai1

scripts/filter.ts

@@ -41,6 +41,10 @@ function filterOpenapi(openapi: OpenAPIV3_1.Document): OpenAPIV3_1.Document {
         "deleteProjectIpAccessList",
         "listOrganizationProjects",
         "listAlerts",
+        "listDropIndexes",
+        "listClusterSuggestedIndexes",
+        "listSchemaAdvice",
+        "listSlowQueries",
     ];
 
     const filteredPaths = {};

src/common/atlas/apiClient.ts

@@ -429,6 +429,42 @@ export class ApiClient {
         return data;
     }
 
+    // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+    async listDropIndexes(options: FetchOptions<operations["listDropIndexes"]>) {
+        const { data, error, response } = await this.client.GET(
+            "/api/atlas/v2/groups/{groupId}/clusters/{clusterName}/performanceAdvisor/dropIndexSuggestions",
+            options
+        );
+        if (error) {
+            throw ApiClientError.fromError(response, error);
+        }
+        return data;
+    }
+
+    // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+    async listSchemaAdvice(options: FetchOptions<operations["listSchemaAdvice"]>) {
+        const { data, error, response } = await this.client.GET(
+            "/api/atlas/v2/groups/{groupId}/clusters/{clusterName}/performanceAdvisor/schemaAdvice",
+            options
+        );
+        if (error) {
+            throw ApiClientError.fromError(response, error);
+        }
+        return data;
+    }
+
+    // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+    async listClusterSuggestedIndexes(options: FetchOptions<operations["listClusterSuggestedIndexes"]>) {
+        const { data, error, response } = await this.client.GET(
+            "/api/atlas/v2/groups/{groupId}/clusters/{clusterName}/performanceAdvisor/suggestedIndexes",
+            options
+        );
+        if (error) {
+            throw ApiClientError.fromError(response, error);
+        }
+        return data;
+    }
+
     // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
     async listDatabaseUsers(options: FetchOptions<operations["listDatabaseUsers"]>) {
         const { data, error, response } = await this.client.GET(
@@ -508,6 +544,18 @@ export class ApiClient {
         return data;
     }
 
+    // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+    async listSlowQueries(options: FetchOptions<operations["listSlowQueries"]>) {
+        const { data, error, response } = await this.client.GET(
+            "/api/atlas/v2/groups/{groupId}/processes/{processId}/performanceAdvisor/slowQueryLogs",
+            options
+        );
+        if (error) {
+            throw ApiClientError.fromError(response, error);
+        }
+        return data;
+    }
+
     // eslint-disable-next-line @typescript-eslint/explicit-function-return-type
     async listOrganizations(options?: FetchOptions<operations["listOrganizations"]>) {
         const { data, error, response } = await this.client.GET("/api/atlas/v2/orgs", options);

src/common/atlas/cluster.ts

@@ -2,6 +2,7 @@ import type { ClusterDescription20240805, FlexClusterDescription20241113 } from
 import type { ApiClient } from "./apiClient.js";
 import { LogId } from "../logger.js";
 
+const DEFAULT_PORT = "27017";
 export interface Cluster {
     name?: string;
     instanceType: "FREE" | "DEDICATED" | "FLEX";
@@ -96,3 +97,22 @@ export async function inspectCluster(apiClient: ApiClient, projectId: string, cl
         }
     }
 }
+
+export async function getProcessIdFromCluster(
+    apiClient: ApiClient,
+    projectId: string,
+    clusterName: string
+): Promise<string> {
+    try {
+        const cluster = await inspectCluster(apiClient, projectId, clusterName);
+        if (!cluster.connectionString) {
+            throw new Error("No connection string available for cluster");
+        }
+        const url = new URL(cluster.connectionString);
+        return `${url.hostname}:${url.port || DEFAULT_PORT}`;
+    } catch (error) {
+        throw new Error(
+            `Failed to get processId from cluster: ${error instanceof Error ? error.message : String(error)}`
+        );
+    }
+}