fix(cloudformation): add pagination to list-stack-resources, describe-stacks, and describe-stack-events routes

Author: waleedlatif1

apps/sim/app/api/tools/cloudformation/describe-stack-events/route.ts

@@ -1,4 +1,8 @@
-import { CloudFormationClient, DescribeStackEventsCommand } from '@aws-sdk/client-cloudformation'
+import {
+  CloudFormationClient,
+  DescribeStackEventsCommand,
+  type StackEvent,
+} from '@aws-sdk/client-cloudformation'
 import { createLogger } from '@sim/logger'
 import { type NextRequest, NextResponse } from 'next/server'
 import { z } from 'zod'
@@ -35,15 +39,21 @@ export async function POST(request: NextRequest) {
       },
     })
 
-    const command = new DescribeStackEventsCommand({
-      StackName: validatedData.stackName,
-    })
-
     const limit = validatedData.limit ?? 50
 
-    const response = await client.send(command)
+    const allEvents: StackEvent[] = []
+    let nextToken: string | undefined
+    do {
+      const command = new DescribeStackEventsCommand({
+        StackName: validatedData.stackName,
+        ...(nextToken && { NextToken: nextToken }),
+      })
+      const response = await client.send(command)
+      allEvents.push(...(response.StackEvents ?? []))
+      nextToken = allEvents.length >= limit ? undefined : response.NextToken
+    } while (nextToken)
 
-    const events = (response.StackEvents ?? []).slice(0, limit).map((e) => ({
+    const events = allEvents.slice(0, limit).map((e) => ({
       stackId: e.StackId ?? '',
       eventId: e.EventId ?? '',
       stackName: e.StackName ?? '',

apps/sim/app/api/tools/cloudformation/describe-stacks/route.ts

@@ -1,4 +1,8 @@
-import { CloudFormationClient, DescribeStacksCommand } from '@aws-sdk/client-cloudformation'
+import {
+  CloudFormationClient,
+  DescribeStacksCommand,
+  type Stack,
+} from '@aws-sdk/client-cloudformation'
 import { createLogger } from '@sim/logger'
 import { type NextRequest, NextResponse } from 'next/server'
 import { z } from 'zod'
@@ -31,13 +35,19 @@ export async function POST(request: NextRequest) {
       },
     })
 
-    const command = new DescribeStacksCommand({
-      ...(validatedData.stackName && { StackName: validatedData.stackName }),
-    })
-
-    const response = await client.send(command)
+    const allStacks: Stack[] = []
+    let nextToken: string | undefined
+    do {
+      const command = new DescribeStacksCommand({
+        ...(validatedData.stackName && { StackName: validatedData.stackName }),
+        ...(nextToken && { NextToken: nextToken }),
+      })
+      const response = await client.send(command)
+      allStacks.push(...(response.Stacks ?? []))
+      nextToken = response.NextToken
+    } while (nextToken)
 
-    const stacks = (response.Stacks ?? []).map((s) => ({
+    const stacks = allStacks.map((s) => ({
       stackName: s.StackName ?? '',
       stackId: s.StackId ?? '',
       stackStatus: s.StackStatus ?? 'UNKNOWN',

apps/sim/app/api/tools/cloudformation/list-stack-resources/route.ts

@@ -1,4 +1,8 @@
-import { CloudFormationClient, ListStackResourcesCommand } from '@aws-sdk/client-cloudformation'
+import {
+  CloudFormationClient,
+  ListStackResourcesCommand,
+  type StackResourceSummary,
+} from '@aws-sdk/client-cloudformation'
 import { createLogger } from '@sim/logger'
 import { type NextRequest, NextResponse } from 'next/server'
 import { z } from 'zod'
@@ -31,13 +35,19 @@ export async function POST(request: NextRequest) {
       },
     })
 
-    const command = new ListStackResourcesCommand({
-      StackName: validatedData.stackName,
-    })
-
-    const response = await client.send(command)
+    const allSummaries: StackResourceSummary[] = []
+    let nextToken: string | undefined
+    do {
+      const command = new ListStackResourcesCommand({
+        StackName: validatedData.stackName,
+        ...(nextToken && { NextToken: nextToken }),
+      })
+      const response = await client.send(command)
+      allSummaries.push(...(response.StackResourceSummaries ?? []))
+      nextToken = response.NextToken
+    } while (nextToken)
 
-    const resources = (response.StackResourceSummaries ?? []).map((r) => ({
+    const resources = allSummaries.map((r) => ({
       logicalResourceId: r.LogicalResourceId ?? '',
       physicalResourceId: r.PhysicalResourceId,
       resourceType: r.ResourceType ?? '',