feat: add currentIp option atlas-create-access-list tool

Author: fmenezes

src/common/atlas/apiClient.ts

@@ -252,6 +252,30 @@ export class ApiClient {
         }
     }
 
+    async getIpInfo() {
+        await this.validateToken();
+
+        const endpoint = "api/private/ipinfo";
+        const url = new URL(endpoint, config.apiBaseUrl);
+        const response = await fetch(url, {
+            method: "GET",
+            headers: {
+                Accept: "application/json",
+                Authorization: `Bearer ${this.token?.access_token}`,
+                "User-Agent": config.userAgent,
+            },
+        });
+
+        if (!response.ok) {
+            throw await ApiClientError.fromResponse(response);
+        }
+
+        const responseBody = await response.json();
+        return responseBody as {
+            currentIpv4Address: string;
+        };
+    }
+
     async listProjects(options?: FetchOptions<operations["listProjects"]>) {
         const { data } = await this.client.GET(`/api/atlas/v2/groups`, options);
         return data;

src/tools/atlas/createAccessList.ts

@@ -15,6 +15,7 @@ export class CreateAccessListTool extends AtlasToolBase {
             .describe("IP addresses to allow access from")
             .optional(),
         cidrBlocks: z.array(z.string().cidr()).describe("CIDR blocks to allow access from").optional(),
+        currentIpAddress: z.boolean().describe("Add the current IP address").default(false),
         comment: z.string().describe("Comment for the access list entries").default(DEFAULT_COMMENT).optional(),
     };
 
@@ -23,11 +24,12 @@ export class CreateAccessListTool extends AtlasToolBase {
         ipAddresses,
         cidrBlocks,
         comment,
+        currentIpAddress,
     }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
         await this.ensureAuthenticated();
 
-        if (!ipAddresses?.length && !cidrBlocks?.length) {
-            throw new Error("Either ipAddresses or cidrBlocks must be provided.");
+        if (!ipAddresses?.length && !cidrBlocks?.length && !currentIpAddress) {
+            throw new Error("One of  ipAddresses, cidrBlocks, currentIpAddress must be provided.");
         }
 
         const ipInputs = (ipAddresses || []).map((ipAddress) => ({
@@ -36,6 +38,16 @@ export class CreateAccessListTool extends AtlasToolBase {
             comment: comment || DEFAULT_COMMENT,
         }));
 
+        if (currentIpAddress) {
+            const currentIp = await this.apiClient!.getIpInfo();
+            const input = {
+                groupId: projectId,
+                ipAddress: currentIp.currentIpv4Address,
+                comment: comment || DEFAULT_COMMENT,
+            };
+            ipInputs.push(input);
+        }
+
         const cidrInputs = (cidrBlocks || []).map((cidrBlock) => ({
             groupId: projectId,
             cidrBlock,

src/tools/atlas/tools.ts

@@ -11,6 +11,7 @@ import { CreateAccessListTool } from "./createAccessList.js";
 import { InspectAccessListTool } from "./inspectAccessList.js";
 import { ListDBUsersTool } from "./listDBUsers.js";
 import { CreateDBUserTool } from "./createDBUser.js";
+import { InspectCurrentIPTool } from "./inspectCurrentIP.js";
 
 export function registerAtlasTools(server: McpServer, state: State, apiClient: ApiClient) {
     const tools: ToolBase[] = [
@@ -23,6 +24,7 @@ export function registerAtlasTools(server: McpServer, state: State, apiClient: A
         new InspectAccessListTool(state, apiClient),
         new ListDBUsersTool(state, apiClient),
         new CreateDBUserTool(state, apiClient),
+        new InspectCurrentIPTool(state, apiClient),
     ];
 
     for (const tool of tools) {