fix(stepfunctions): nested arrays are not serialized correctly

packages/aws-cdk-lib/aws-stepfunctions/lib/private/json-path.ts

@@ -183,18 +183,25 @@ function recurseArray(key: string, arr: any[], handlers: FieldHandlers, visited:
   }
 
   return {
-    [key]: arr.map(value => {
-      if ((typeof value === 'string' && jsonPathString(value) !== undefined)
+    [key]: resolveArray(arr, handlers, visited),
+  };
+}
+
+function resolveArray(arr: any[], handlers: FieldHandlers, visited: object[] = []): any[] {
+  return arr.map(value => {
+    if ((typeof value === 'string' && jsonPathString(value) !== undefined)
         || (typeof value === 'number' && jsonPathNumber(value) !== undefined)
         || (isStringArray(value) && jsonPathStringList(value) !== undefined)) {
-        throw new Error('Cannot use JsonPath fields in an array, they must be used in objects');
-      }
-      if (typeof value === 'object' && value !== null) {
-        return recurseObject(value, handlers, visited);
-      }
-      return value;
-    }),
-  };
+      throw new Error('Cannot use JsonPath fields in an array, they must be used in objects');
+    }
+    if (Array.isArray(value)) {
+      return resolveArray(value, handlers, visited);
+    }
+    if (typeof value === 'object' && value !== null) {
+      return recurseObject(value, handlers, visited);
+    }
+    return value;
+  });
 }
 
 function isStringArray(x: any): x is string[] {

packages/aws-cdk-lib/aws-stepfunctions/test/fields.test.ts

@@ -470,6 +470,48 @@ describe('intrinsics constructors', () => {
       'Field.$': 'States.JsonToString($.Obj)',
     });
   });
+
+  test('correctly serialize a nested array', () => {
+    expect(
+      FieldUtils.renderObject({
+        nestedArray: [
+          [
+            [123, 123],
+            [456, 456],
+          ],
+        ],
+      }),
+    ).toStrictEqual({
+      nestedArray: [
+        [
+          [123, 123],
+          [456, 456],
+        ],
+      ],
+    });
+  });
+
+  test('deep replace correctly handles fields in nested arrays', () => {
+    expect(
+      FieldUtils.renderObject({
+        deep: [
+          [
+            {
+              deepField: JsonPath.numberAt('$.numField'),
+            },
+          ],
+        ],
+      }),
+    ).toStrictEqual({
+      deep: [
+        [
+          {
+            'deepField.$': '$.numField',
+          },
+        ],
+      ],
+    });
+  });
 });
 
 test('find task token even if nested in intrinsic functions', () => {