Format FOREACH loop

Author: nene

src/syntax/procedural_language.ts

@@ -162,6 +162,19 @@ export const proceduralLanguageMap: Partial<CstToDocMap<AllProceduralNodes>> = {
       print.spaced(["to", "by"]),
     ]),
   for_by_clause: (print) => group(print.spaced(["byKw", "expr"])),
+  // FOREACH
+  foreach_stmt: (print) =>
+    group([
+      print.spaced([
+        "foreachKw",
+        "left",
+        "slice",
+        "inArrayKw",
+        "right",
+        "loop",
+      ]),
+    ]),
+  foreach_slice: (print) => group(print.spaced(["sliceKw", "count"])),
   // BREAK/CONTINUE
   break_stmt: (print) => group(print.spaced(["breakKw", "label", "when"])),
   continue_stmt: (print) =>

test/proc/loops.test.ts

@@ -81,4 +81,20 @@ describe("loops", () => {
       END LOOP
     `);
   });
+
+  it(`formats FOREACH .. LOOP`, async () => {
+    await testPlpgsql(dedent`
+      FOREACH n IN ARRAY arr LOOP
+        x = x + n;
+      END LOOP
+    `);
+  });
+
+  it(`formats FOREACH .. SLICE .. LOOP`, async () => {
+    await testPlpgsql(dedent`
+      FOREACH n SLICE 8 IN ARRAY arr LOOP
+        x = x + n;
+      END LOOP
+    `);
+  });
 });