spark-sql-language-server 0.0.1-beta.5 → 0.2.0-beta.0

package/out-tsc/formatter/strategies/procedure/index.js

@@ -0,0 +1,63 @@
+"use strict";
+/**
+ * Procedure Strategies Index
+ *
+ * Exports all stored procedure formatting strategies:
+ * - CompoundStatementStrategy: BEGIN END blocks
+ * - IfElseStrategy: IF ELSEIF ELSE statements
+ * - WhileStrategy: WHILE loops
+ * - RepeatStrategy: REPEAT loops
+ * - LoopStrategy: LOOP statements
+ * - ForStrategy: FOR loops
+ * - CaseStatementStrategy: CASE statements
+ * - DeclareHandlerStrategy: DECLARE HANDLER
+ * - DeclareConditionStrategy: DECLARE CONDITION
+ * - LeaveIterateStrategy: LEAVE/ITERATE statements
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.procedureStrategies = exports.LeaveIterateStrategy = exports.DeclareConditionStrategy = exports.DeclareHandlerStrategy = exports.CaseStatementStrategy = exports.ForStrategy = exports.LoopStrategy = exports.RepeatStrategy = exports.WhileStrategy = exports.IfElseStrategy = exports.CompoundStatementItemStrategy = exports.CompoundBodyStrategy = exports.CompoundStatementStrategy = exports.SingleCompoundStatementStrategy = void 0;
+var compound_statement_strategy_1 = require("./compound-statement.strategy");
+Object.defineProperty(exports, "SingleCompoundStatementStrategy", { enumerable: true, get: function () { return compound_statement_strategy_1.SingleCompoundStatementStrategy; } });
+Object.defineProperty(exports, "CompoundStatementStrategy", { enumerable: true, get: function () { return compound_statement_strategy_1.CompoundStatementStrategy; } });
+Object.defineProperty(exports, "CompoundBodyStrategy", { enumerable: true, get: function () { return compound_statement_strategy_1.CompoundBodyStrategy; } });
+Object.defineProperty(exports, "CompoundStatementItemStrategy", { enumerable: true, get: function () { return compound_statement_strategy_1.CompoundStatementItemStrategy; } });
+var if_else_strategy_1 = require("./if-else.strategy");
+Object.defineProperty(exports, "IfElseStrategy", { enumerable: true, get: function () { return if_else_strategy_1.IfElseStrategy; } });
+var loop_strategy_1 = require("./loop.strategy");
+Object.defineProperty(exports, "WhileStrategy", { enumerable: true, get: function () { return loop_strategy_1.WhileStrategy; } });
+Object.defineProperty(exports, "RepeatStrategy", { enumerable: true, get: function () { return loop_strategy_1.RepeatStrategy; } });
+Object.defineProperty(exports, "LoopStrategy", { enumerable: true, get: function () { return loop_strategy_1.LoopStrategy; } });
+Object.defineProperty(exports, "ForStrategy", { enumerable: true, get: function () { return loop_strategy_1.ForStrategy; } });
+var case_statement_strategy_1 = require("./case-statement.strategy");
+Object.defineProperty(exports, "CaseStatementStrategy", { enumerable: true, get: function () { return case_statement_strategy_1.CaseStatementStrategy; } });
+var handler_strategy_1 = require("./handler.strategy");
+Object.defineProperty(exports, "DeclareHandlerStrategy", { enumerable: true, get: function () { return handler_strategy_1.DeclareHandlerStrategy; } });
+Object.defineProperty(exports, "DeclareConditionStrategy", { enumerable: true, get: function () { return handler_strategy_1.DeclareConditionStrategy; } });
+var leave_iterate_strategy_1 = require("./leave-iterate.strategy");
+Object.defineProperty(exports, "LeaveIterateStrategy", { enumerable: true, get: function () { return leave_iterate_strategy_1.LeaveIterateStrategy; } });
+// Import for strategy registry
+const case_statement_strategy_2 = require("./case-statement.strategy");
+const compound_statement_strategy_2 = require("./compound-statement.strategy");
+const handler_strategy_2 = require("./handler.strategy");
+const if_else_strategy_2 = require("./if-else.strategy");
+const leave_iterate_strategy_2 = require("./leave-iterate.strategy");
+const loop_strategy_2 = require("./loop.strategy");
+/**
+ * All procedure formatting strategies
+ */
+exports.procedureStrategies = [
+    new compound_statement_strategy_2.SingleCompoundStatementStrategy(),
+    new compound_statement_strategy_2.CompoundStatementStrategy(),
+    new compound_statement_strategy_2.CompoundBodyStrategy(),
+    new compound_statement_strategy_2.CompoundStatementItemStrategy(),
+    new if_else_strategy_2.IfElseStrategy(),
+    new loop_strategy_2.WhileStrategy(),
+    new loop_strategy_2.RepeatStrategy(),
+    new loop_strategy_2.LoopStrategy(),
+    new loop_strategy_2.ForStrategy(),
+    new case_statement_strategy_2.CaseStatementStrategy(),
+    new handler_strategy_2.DeclareHandlerStrategy(),
+    new handler_strategy_2.DeclareConditionStrategy(),
+    new leave_iterate_strategy_2.LeaveIterateStrategy()
+];
+//# sourceMappingURL=index.js.map

package/out-tsc/formatter/strategies/procedure/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/formatter/strategies/procedure/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH,6EAKuC;AAJrC,8IAAA,+BAA+B,OAAA;AAC/B,wIAAA,yBAAyB,OAAA;AACzB,mIAAA,oBAAoB,OAAA;AACpB,4IAAA,6BAA6B,OAAA;AAG/B,uDAAoD;AAA3C,kHAAA,cAAc,OAAA;AAEvB,iDAA2F;AAAlF,8GAAA,aAAa,OAAA;AAAE,+GAAA,cAAc,OAAA;AAAE,6GAAA,YAAY,OAAA;AAAE,4GAAA,WAAW,OAAA;AAEjE,qEAAkE;AAAzD,gIAAA,qBAAqB,OAAA;AAE9B,uDAAsF;AAA7E,0HAAA,sBAAsB,OAAA;AAAE,4HAAA,wBAAwB,OAAA;AAEzD,mEAAgE;AAAvD,8HAAA,oBAAoB,OAAA;AAE7B,+BAA+B;AAC/B,uEAAkE;AAClE,+EAKuC;AACvC,yDAAsF;AACtF,yDAAoD;AACpD,qEAAgE;AAChE,mDAA2F;AAG3F;;GAEG;AACU,QAAA,mBAAmB,GAAyB;IACvD,IAAI,6DAA+B,EAAE;IACrC,IAAI,uDAAyB,EAAE;IAC/B,IAAI,kDAAoB,EAAE;IAC1B,IAAI,2DAA6B,EAAE;IACnC,IAAI,iCAAc,EAAE;IACpB,IAAI,6BAAa,EAAE;IACnB,IAAI,8BAAc,EAAE;IACpB,IAAI,4BAAY,EAAE;IAClB,IAAI,2BAAW,EAAE;IACjB,IAAI,+CAAqB,EAAE;IAC3B,IAAI,yCAAsB,EAAE;IAC5B,IAAI,2CAAwB,EAAE;IAC9B,IAAI,6CAAoB,EAAE;CAC3B,CAAC"}

package/out-tsc/formatter/strategies/procedure/leave-iterate.strategy.d.ts

@@ -0,0 +1,19 @@
+/**
+ * LEAVE/ITERATE Statement Strategy
+ *
+ * Format rules:
+ * - Single line statement
+ * - When standalone (not in compoundBody): add semicolon
+ * - When in compoundBody: no semicolon (compoundBody adds it)
+ */
+import { ParserRuleContext } from 'antlr4ts/ParserRuleContext';
+import { BaseFormattingStrategy, FormattingBridge } from '../../core';
+/**
+ * Strategy for formatting LEAVE/ITERATE statements
+ */
+export declare class LeaveIterateStrategy extends BaseFormattingStrategy {
+    readonly id = "leave-iterate";
+    canHandle(ctx: ParserRuleContext): boolean;
+    format(ctx: ParserRuleContext, bridge: FormattingBridge): void;
+}
+//# sourceMappingURL=leave-iterate.strategy.d.ts.map

package/out-tsc/formatter/strategies/procedure/leave-iterate.strategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"leave-iterate.strategy.d.ts","sourceRoot":"","sources":["../../../../src/formatter/strategies/procedure/leave-iterate.strategy.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAG/D,OAAO,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAgBtE;;GAEG;AACH,qBAAa,oBAAqB,SAAQ,sBAAsB;IAC9D,QAAQ,CAAC,EAAE,mBAAmB;IAE9B,SAAS,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO;IAI1C,MAAM,CAAC,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;CAsB/D"}

package/out-tsc/formatter/strategies/procedure/leave-iterate.strategy.js

@@ -0,0 +1,54 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LeaveIterateStrategy = void 0;
+const SparkSQLParser_1 = require("../../../lib/SparkSQLParser");
+const core_1 = require("../../core");
+/**
+ * Check if context is inside a compoundBody
+ */
+function isInCompoundBody(ctx) {
+    let parent = ctx.parent;
+    while (parent) {
+        if (parent instanceof SparkSQLParser_1.CompoundBodyContext) {
+            return true;
+        }
+        parent = parent.parent;
+    }
+    return false;
+}
+/**
+ * Strategy for formatting LEAVE/ITERATE statements
+ */
+class LeaveIterateStrategy extends core_1.BaseFormattingStrategy {
+    constructor() {
+        super(...arguments);
+        this.id = 'leave-iterate';
+    }
+    canHandle(ctx) {
+        return ctx instanceof SparkSQLParser_1.LeaveStatementContext || ctx instanceof SparkSQLParser_1.IterateStatementContext;
+    }
+    format(ctx, bridge) {
+        // Check if we're inside a compoundBody
+        // compoundBody adds SEMICOLON after each statement, so we shouldn't add one
+        // But if standalone (not in compoundBody), we need to add semicolon
+        const inCompoundBody = isInCompoundBody(ctx);
+        if (!inCompoundBody) {
+            // Standalone - add semicolon
+            if (ctx instanceof SparkSQLParser_1.LeaveStatementContext) {
+                const identifier = ctx.strictIdentifier();
+                if (identifier) {
+                    bridge.updateBeforeNextTokenByContext(identifier, ';');
+                }
+            }
+            else if (ctx instanceof SparkSQLParser_1.IterateStatementContext) {
+                const identifier = ctx.strictIdentifier();
+                if (identifier) {
+                    bridge.updateBeforeNextTokenByContext(identifier, ';');
+                }
+            }
+        }
+        // If in compoundBody, do nothing - let compoundBody handle the semicolon
+    }
+}
+exports.LeaveIterateStrategy = LeaveIterateStrategy;
+//# sourceMappingURL=leave-iterate.strategy.js.map

package/out-tsc/formatter/strategies/procedure/leave-iterate.strategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"leave-iterate.strategy.js","sourceRoot":"","sources":["../../../../src/formatter/strategies/procedure/leave-iterate.strategy.ts"],"names":[],"mappings":";;;AAUA,gEAAkH;AAClH,qCAAsE;AAEtE;;GAEG;AACH,SAAS,gBAAgB,CAAC,GAAsB;IAC9C,IAAI,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;IACxB,OAAO,MAAM,EAAE,CAAC;QACd,IAAI,MAAM,YAAY,oCAAmB,EAAE,CAAC;YAC1C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,MAAa,oBAAqB,SAAQ,6BAAsB;IAAhE;;QACW,OAAE,GAAG,eAAe,CAAC;IA4BhC,CAAC;IA1BC,SAAS,CAAC,GAAsB;QAC9B,OAAO,GAAG,YAAY,sCAAqB,IAAI,GAAG,YAAY,wCAAuB,CAAC;IACxF,CAAC;IAED,MAAM,CAAC,GAAsB,EAAE,MAAwB;QACrD,uCAAuC;QACvC,4EAA4E;QAC5E,oEAAoE;QACpE,MAAM,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,6BAA6B;YAC7B,IAAI,GAAG,YAAY,sCAAqB,EAAE,CAAC;gBACzC,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,CAAC,8BAA8B,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;iBAAM,IAAI,GAAG,YAAY,wCAAuB,EAAE,CAAC;gBAClD,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,CAAC,8BAA8B,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;QACH,CAAC;QACD,yEAAyE;IAC3E,CAAC;CACF;AA7BD,oDA6BC"}

package/out-tsc/formatter/strategies/procedure/loop.strategy.d.ts

@@ -0,0 +1,71 @@
+/**
+ * Loop Statement Strategy - WHILE/REPEAT/LOOP/FOR Formatting
+ *
+ * Format rules (from documentation):
+ *
+ * WHILE:
+ * - [label:] on its own line
+ * - WHILE condition DO on same line
+ * - Body indented on new line after DO
+ * - END WHILE [label] on its own line
+ *
+ * REPEAT:
+ * - [label:] on its own line
+ * - REPEAT on its own line
+ * - Body indented
+ * - UNTIL condition on its own line
+ * - END REPEAT [label] on its own line
+ *
+ * LOOP:
+ * - [label:] on its own line
+ * - LOOP on its own line
+ * - Body indented
+ * - END LOOP [label] on its own line
+ *
+ * FOR:
+ * - [label:] on its own line
+ * - FOR on its own line
+ * - query indented
+ * - DO on its own line
+ * - Body indented
+ * - END FOR [label] on its own line
+ */
+import { ParserRuleContext } from 'antlr4ts/ParserRuleContext';
+import { BaseFormattingStrategy, FormattingBridge } from '../../core';
+/**
+ * Strategy for formatting WHILE statements
+ */
+export declare class WhileStrategy extends BaseFormattingStrategy {
+    readonly id = "while";
+    canHandle(ctx: ParserRuleContext): boolean;
+    format(ctx: ParserRuleContext, bridge: FormattingBridge): void;
+    private formatWhile;
+}
+/**
+ * Strategy for formatting REPEAT statements
+ */
+export declare class RepeatStrategy extends BaseFormattingStrategy {
+    readonly id = "repeat";
+    canHandle(ctx: ParserRuleContext): boolean;
+    format(ctx: ParserRuleContext, bridge: FormattingBridge): void;
+    private formatRepeat;
+}
+/**
+ * Strategy for formatting LOOP statements
+ */
+export declare class LoopStrategy extends BaseFormattingStrategy {
+    readonly id = "loop";
+    canHandle(ctx: ParserRuleContext): boolean;
+    format(ctx: ParserRuleContext, bridge: FormattingBridge): void;
+    private formatLoop;
+}
+/**
+ * Strategy for formatting FOR statements
+ */
+export declare class ForStrategy extends BaseFormattingStrategy {
+    readonly id = "for";
+    canHandle(ctx: ParserRuleContext): boolean;
+    format(ctx: ParserRuleContext, bridge: FormattingBridge): void;
+    private formatFor;
+}
+//# sourceMappingURL=loop.strategy.d.ts.map

package/out-tsc/formatter/strategies/procedure/loop.strategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"loop.strategy.d.ts","sourceRoot":"","sources":["../../../../src/formatter/strategies/procedure/loop.strategy.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAW/D,OAAO,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAgBtE;;GAEG;AACH,qBAAa,aAAc,SAAQ,sBAAsB;IACvD,QAAQ,CAAC,EAAE,WAAW;IAEtB,SAAS,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO;IAI1C,MAAM,CAAC,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAM9D,OAAO,CAAC,WAAW;CAiCpB;AAED;;GAEG;AACH,qBAAa,cAAe,SAAQ,sBAAsB;IACxD,QAAQ,CAAC,EAAE,YAAY;IAEvB,SAAS,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO;IAI1C,MAAM,CAAC,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAM9D,OAAO,CAAC,YAAY;CAqCrB;AAED;;GAEG;AACH,qBAAa,YAAa,SAAQ,sBAAsB;IACtD,QAAQ,CAAC,EAAE,UAAU;IAErB,SAAS,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO;IAI1C,MAAM,CAAC,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAM9D,OAAO,CAAC,UAAU;CA+BnB;AAED;;GAEG;AACH,qBAAa,WAAY,SAAQ,sBAAsB;IACrD,QAAQ,CAAC,EAAE,SAAS;IAEpB,SAAS,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO;IAI1C,MAAM,CAAC,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAM9D,OAAO,CAAC,SAAS;CAiDlB"}

package/out-tsc/formatter/strategies/procedure/loop.strategy.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ForStrategy = exports.LoopStrategy = exports.RepeatStrategy = exports.WhileStrategy = void 0;
+const TerminalNode_1 = require("antlr4ts/tree/TerminalNode");
+const SparkSQLParser_1 = require("../../../lib/SparkSQLParser");
+const core_1 = require("../../core");
+function formatBeginLabel(ctx, bridge) {
+    const colon = ctx.children?.find(c => c.text === ':');
+    if (colon instanceof TerminalNode_1.TerminalNode) {
+        bridge.updateBeforeNextTokenByNode(colon, '\n');
+    }
+}
+function formatEndLabel(ctx, bridge) {
+    const identifier = ctx.strictIdentifier();
+    if (identifier) {
+        bridge.updateAfterPreviousTokenByContext(ctx, ' ');
+    }
+}
+/**
+ * Strategy for formatting WHILE statements
+ */
+class WhileStrategy extends core_1.BaseFormattingStrategy {
+    constructor() {
+        super(...arguments);
+        this.id = 'while';
+    }
+    canHandle(ctx) {
+        return ctx instanceof SparkSQLParser_1.WhileStatementContext;
+    }
+    format(ctx, bridge) {
+        if (ctx instanceof SparkSQLParser_1.WhileStatementContext) {
+            this.formatWhile(ctx, bridge);
+        }
+    }
+    formatWhile(ctx, bridge) {
+        const beginLabel = ctx.beginLabel();
+        const doNode = ctx.DO();
+        const body = ctx.compoundBody();
+        const endNode = ctx.END();
+        const endLabel = ctx.endLabel();
+        if (beginLabel) {
+            formatBeginLabel(beginLabel, bridge);
+        }
+        // Add indent for body
+        bridge.addIndent();
+        // WHILE condition DO - DO should be on same line, then newline+indent
+        if (doNode && body) {
+            bridge.updateBeforeNextTokenByNode(doNode, bridge.newlineIndent);
+        }
+        if (body) {
+            bridge.visit(body);
+        }
+        // END WHILE - on its own line
+        bridge.minusIndent();
+        if (endNode) {
+            bridge.updateAfterPreviousTokenByNode(endNode, '\n');
+        }
+        if (endLabel) {
+            formatEndLabel(endLabel, bridge);
+        }
+    }
+}
+exports.WhileStrategy = WhileStrategy;
+/**
+ * Strategy for formatting REPEAT statements
+ */
+class RepeatStrategy extends core_1.BaseFormattingStrategy {
+    constructor() {
+        super(...arguments);
+        this.id = 'repeat';
+    }
+    canHandle(ctx) {
+        return ctx instanceof SparkSQLParser_1.RepeatStatementContext;
+    }
+    format(ctx, bridge) {
+        if (ctx instanceof SparkSQLParser_1.RepeatStatementContext) {
+            this.formatRepeat(ctx, bridge);
+        }
+    }
+    formatRepeat(ctx, bridge) {
+        const beginLabel = ctx.beginLabel();
+        const repeatNode = ctx.REPEAT(0);
+        const body = ctx.compoundBody();
+        const untilNode = ctx.UNTIL();
+        const endNode = ctx.END();
+        const endLabel = ctx.endLabel();
+        if (beginLabel) {
+            formatBeginLabel(beginLabel, bridge);
+        }
+        // REPEAT - on its own line, then newline+indent
+        bridge.addIndent();
+        if (repeatNode && body) {
+            bridge.updateBeforeNextTokenByNode(repeatNode, bridge.newlineIndent);
+        }
+        if (body) {
+            bridge.visit(body);
+        }
+        // UNTIL condition - on its own line (no indent)
+        bridge.minusIndent();
+        if (untilNode) {
+            bridge.updateAfterPreviousTokenByNode(untilNode, '\n');
+        }
+        // END REPEAT - on its own line
+        if (endNode) {
+            bridge.updateAfterPreviousTokenByNode(endNode, '\n');
+        }
+        if (endLabel) {
+            formatEndLabel(endLabel, bridge);
+        }
+    }
+}
+exports.RepeatStrategy = RepeatStrategy;
+/**
+ * Strategy for formatting LOOP statements
+ */
+class LoopStrategy extends core_1.BaseFormattingStrategy {
+    constructor() {
+        super(...arguments);
+        this.id = 'loop';
+    }
+    canHandle(ctx) {
+        return ctx instanceof SparkSQLParser_1.LoopStatementContext;
+    }
+    format(ctx, bridge) {
+        if (ctx instanceof SparkSQLParser_1.LoopStatementContext) {
+            this.formatLoop(ctx, bridge);
+        }
+    }
+    formatLoop(ctx, bridge) {
+        const beginLabel = ctx.beginLabel();
+        const loopNode = ctx.LOOP(0);
+        const body = ctx.compoundBody();
+        const endNode = ctx.END();
+        const endLabel = ctx.endLabel();
+        if (beginLabel) {
+            formatBeginLabel(beginLabel, bridge);
+        }
+        // LOOP - on its own line, then newline+indent
+        bridge.addIndent();
+        if (loopNode && body) {
+            bridge.updateBeforeNextTokenByNode(loopNode, bridge.newlineIndent);
+        }
+        if (body) {
+            bridge.visit(body);
+        }
+        // END LOOP - on its own line
+        bridge.minusIndent();
+        if (endNode) {
+            bridge.updateAfterPreviousTokenByNode(endNode, '\n');
+        }
+        if (endLabel) {
+            formatEndLabel(endLabel, bridge);
+        }
+    }
+}
+exports.LoopStrategy = LoopStrategy;
+/**
+ * Strategy for formatting FOR statements
+ */
+class ForStrategy extends core_1.BaseFormattingStrategy {
+    constructor() {
+        super(...arguments);
+        this.id = 'for';
+    }
+    canHandle(ctx) {
+        return ctx instanceof SparkSQLParser_1.ForStatementContext;
+    }
+    format(ctx, bridge) {
+        if (ctx instanceof SparkSQLParser_1.ForStatementContext) {
+            this.formatFor(ctx, bridge);
+        }
+    }
+    formatFor(ctx, bridge) {
+        const beginLabel = ctx.beginLabel();
+        const forNode = ctx.FOR(0);
+        const query = ctx.query();
+        const doNode = ctx.DO();
+        const body = ctx.compoundBody();
+        const endNode = ctx.END();
+        const endLabel = ctx.endLabel();
+        if (beginLabel) {
+            formatBeginLabel(beginLabel, bridge);
+        }
+        // FOR - on its own line, then newline+indent for query
+        bridge.addIndent();
+        if (forNode && query) {
+            bridge.updateBeforeNextTokenByNode(forNode, bridge.newlineIndent);
+        }
+        if (query) {
+            bridge.visit(query);
+        }
+        // DO - on its own line (remove indent first)
+        bridge.minusIndent();
+        if (doNode) {
+            bridge.updateAfterPreviousTokenByNode(doNode, '\n');
+        }
+        // Body - indented
+        bridge.addIndent();
+        if (doNode && body) {
+            bridge.updateBeforeNextTokenByNode(doNode, bridge.newlineIndent);
+        }
+        if (body) {
+            bridge.visit(body);
+        }
+        // END FOR - on its own line
+        bridge.minusIndent();
+        if (endNode) {
+            bridge.updateAfterPreviousTokenByNode(endNode, '\n');
+        }
+        if (endLabel) {
+            formatEndLabel(endLabel, bridge);
+        }
+    }
+}
+exports.ForStrategy = ForStrategy;
+//# sourceMappingURL=loop.strategy.js.map

package/out-tsc/formatter/strategies/procedure/loop.strategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loop.strategy.js","sourceRoot":"","sources":["../../../../src/formatter/strategies/procedure/loop.strategy.ts"],"names":[],"mappings":";;;AAiCA,6DAA0D;AAE1D,gEAOqC;AACrC,qCAAsE;AAEtE,SAAS,gBAAgB,CAAC,GAAsB,EAAE,MAAwB;IACxE,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC;IACtD,IAAI,KAAK,YAAY,2BAAY,EAAE,CAAC;QAClC,MAAM,CAAC,2BAA2B,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,GAAoB,EAAE,MAAwB;IACpE,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1C,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,CAAC,iCAAiC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAa,aAAc,SAAQ,6BAAsB;IAAzD;;QACW,OAAE,GAAG,OAAO,CAAC;IA6CxB,CAAC;IA3CC,SAAS,CAAC,GAAsB;QAC9B,OAAO,GAAG,YAAY,sCAAqB,CAAC;IAC9C,CAAC;IAED,MAAM,CAAC,GAAsB,EAAE,MAAwB;QACrD,IAAI,GAAG,YAAY,sCAAqB,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,GAA0B,EAAE,MAAwB;QACtE,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;QAEhC,IAAI,UAAU,EAAE,CAAC;YACf,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,sBAAsB;QACtB,MAAM,CAAC,SAAS,EAAE,CAAC;QAEnB,sEAAsE;QACtE,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YACnB,MAAM,CAAC,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,8BAA8B;QAC9B,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,8BAA8B,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;CACF;AA9CD,sCA8CC;AAED;;GAEG;AACH,MAAa,cAAe,SAAQ,6BAAsB;IAA1D;;QACW,OAAE,GAAG,QAAQ,CAAC;IAiDzB,CAAC;IA/CC,SAAS,CAAC,GAAsB;QAC9B,OAAO,GAAG,YAAY,uCAAsB,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,GAAsB,EAAE,MAAwB;QACrD,IAAI,GAAG,YAAY,uCAAsB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,GAA2B,EAAE,MAAwB;QACxE,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;QAEhC,IAAI,UAAU,EAAE,CAAC;YACf,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,gDAAgD;QAChD,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,CAAC,2BAA2B,CAAC,UAAU,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,gDAAgD;QAChD,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,CAAC,8BAA8B,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;QAED,+BAA+B;QAC/B,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,8BAA8B,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;CACF;AAlDD,wCAkDC;AAED;;GAEG;AACH,MAAa,YAAa,SAAQ,6BAAsB;IAAxD;;QACW,OAAE,GAAG,MAAM,CAAC;IA2CvB,CAAC;IAzCC,SAAS,CAAC,GAAsB;QAC9B,OAAO,GAAG,YAAY,qCAAoB,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,GAAsB,EAAE,MAAwB;QACrD,IAAI,GAAG,YAAY,qCAAoB,EAAE,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,GAAyB,EAAE,MAAwB;QACpE,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;QAEhC,IAAI,UAAU,EAAE,CAAC;YACf,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,8CAA8C;QAC9C,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,CAAC,2BAA2B,CAAC,QAAQ,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,6BAA6B;QAC7B,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,8BAA8B,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;CACF;AA5CD,oCA4CC;AAED;;GAEG;AACH,MAAa,WAAY,SAAQ,6BAAsB;IAAvD;;QACW,OAAE,GAAG,KAAK,CAAC;IA6DtB,CAAC;IA3DC,SAAS,CAAC,GAAsB;QAC9B,OAAO,GAAG,YAAY,oCAAmB,CAAC;IAC5C,CAAC;IAED,MAAM,CAAC,GAAsB,EAAE,MAAwB;QACrD,IAAI,GAAG,YAAY,oCAAmB,EAAE,CAAC;YACvC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,GAAwB,EAAE,MAAwB;QAClE,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;QAEhC,IAAI,UAAU,EAAE,CAAC;YACf,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,uDAAuD;QACvD,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;YACrB,MAAM,CAAC,2BAA2B,CAAC,OAAO,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;QAED,6CAA6C;QAC7C,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,8BAA8B,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACtD,CAAC;QAED,kBAAkB;QAClB,MAAM,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YACnB,MAAM,CAAC,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,4BAA4B;QAC5B,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,8BAA8B,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;CACF;AA9DD,kCA8DC"}