Format FOR .. IN range LOOP

Author: nene

src/syntax/procedural_language.ts

@@ -141,6 +141,7 @@ export const proceduralLanguageMap: Partial<CstToDocMap<AllProceduralNodes>> = {
       hardline,
       print.spaced("endWhileKw"),
     ]),
+  // WHILE .. LOOP .. END LOOP
   while_loop_stmt: (print) =>
     group(print.spaced(["whileKw", "condition", "loop"])),
   // FOR .. IN
@@ -151,6 +152,16 @@ export const proceduralLanguageMap: Partial<CstToDocMap<AllProceduralNodes>> = {
       hardline,
       print.spaced("endForKw"),
     ]),
+  // FOR .. IN range LOOP .. END LOOP
+  for_loop_stmt: (print) =>
+    group(print.spaced(["forKw", "left", "inKw", "right", "loop"])),
+  for_range: (print) =>
+    group([
+      print.spaced(["reverseKw", "from"]),
+      "..",
+      print.spaced(["to", "by"]),
+    ]),
+  for_by_clause: (print) => group(print.spaced(["byKw", "expr"])),
   // BREAK/CONTINUE
   break_stmt: (print) => group(print.spaced(["breakKw", "label", "when"])),
   continue_stmt: (print) =>

test/proc/loops.test.ts

@@ -41,4 +41,28 @@ describe("loops", () => {
       END FOR
     `);
   });
+
+  it(`formats FOR .. IN range LOOP`, async () => {
+    await testPlpgsql(dedent`
+      FOR n IN 1..10 LOOP
+        x = x + n;
+      END LOOP
+    `);
+  });
+
+  it(`formats FOR .. IN REVERSE range LOOP`, async () => {
+    await testPlpgsql(dedent`
+      FOR n IN REVERSE 10..1 LOOP
+        x = x + n;
+      END LOOP
+    `);
+  });
+
+  it(`formats FOR .. IN range BY step LOOP`, async () => {
+    await testPlpgsql(dedent`
+      FOR n IN x..y BY z LOOP
+        x = x + n;
+      END LOOP
+    `);
+  });
 });