prettier-plugin-java 2.6.1 → 2.6.2

package/dist/printers/blocks-and-statements.js

@@ -0,0 +1,432 @@
+import { builders } from "prettier/doc";
+import { concat, group, indent, join } from "./prettier-builder.js";
+import { printTokenWithComments } from "./comments/format-comments.js";
+import { hasLeadingLineComments, hasTrailingLineComments } from "./comments/comments-utils.js";
+import { displaySemicolon, getBlankLinesSeparator, isStatementEmptyStatement, putIntoBraces, rejectAndConcat, rejectAndJoin, rejectAndJoinSeps, rejectSeparators, sortModifiers } from "./printer-utils.js";
+import { BaseCstPrettierPrinter } from "../base-cst-printer.js";
+const { line, softline, hardline } = builders;
+export class BlocksAndStatementPrettierVisitor extends BaseCstPrettierPrinter {
+    block(ctx) {
+        const blockStatements = this.visit(ctx.blockStatements);
+        return putIntoBraces(blockStatements, hardline, ctx.LCurly[0], ctx.RCurly[0]);
+    }
+    blockStatements(ctx) {
+        const blockStatement = this.mapVisit(ctx.blockStatement);
+        const separators = rejectSeparators(getBlankLinesSeparator(ctx.blockStatement), blockStatement);
+        return rejectAndJoinSeps(separators, blockStatement);
+    }
+    blockStatement(ctx) {
+        return this.visitSingle(ctx);
+    }
+    localVariableDeclarationStatement(ctx) {
+        const localVariableDeclaration = this.visit(ctx.localVariableDeclaration);
+        return rejectAndConcat([localVariableDeclaration, ctx.Semicolon[0]]);
+    }
+    localVariableDeclaration(ctx) {
+        const modifiers = sortModifiers(ctx.variableModifier);
+        const firstAnnotations = this.mapVisit(modifiers[0]);
+        const finalModifiers = this.mapVisit(modifiers[1]);
+        const localVariableType = this.visit(ctx.localVariableType);
+        const variableDeclaratorList = this.visit(ctx.variableDeclaratorList);
+        return rejectAndJoin(hardline, [
+            rejectAndJoin(hardline, firstAnnotations),
+            rejectAndJoin(" ", [
+                rejectAndJoin(" ", finalModifiers),
+                localVariableType,
+                variableDeclaratorList
+            ])
+        ]);
+    }
+    localVariableType(ctx) {
+        if (ctx.unannType) {
+            return this.visitSingle(ctx);
+        }
+        return printTokenWithComments(this.getSingle(ctx));
+    }
+    statement(ctx, params) {
+        // handling Labeled statements comments
+        if (ctx.labeledStatement !== undefined) {
+            const newLabelStatement = Object.assign({}, ctx.labeledStatement[0]);
+            const newColon = Object.assign({}, ctx.labeledStatement[0].children.Colon[0]);
+            const newStatement = Object.assign({}, ctx.labeledStatement[0].children.statement[0]);
+            const labeledStatementLeadingComments = [];
+            if (newColon.trailingComments !== undefined) {
+                labeledStatementLeadingComments.push(...newColon.trailingComments);
+                delete newColon.trailingComments;
+            }
+            if (newStatement.leadingComments !== undefined) {
+                labeledStatementLeadingComments.push(...newStatement.leadingComments);
+                delete newStatement.leadingComments;
+            }
+            if (labeledStatementLeadingComments.length !== 0) {
+                newLabelStatement.leadingComments = labeledStatementLeadingComments;
+            }
+            newLabelStatement.children.Colon[0] = newColon;
+            newLabelStatement.children.statement[0] = newStatement;
+            return this.visit([newLabelStatement]);
+        }
+        return this.visitSingle(ctx, params);
+    }
+    statementWithoutTrailingSubstatement(ctx, params) {
+        return this.visitSingle(ctx, params);
+    }
+    emptyStatement(ctx, params) {
+        return displaySemicolon(ctx.Semicolon[0], params);
+    }
+    labeledStatement(ctx) {
+        const identifier = ctx.Identifier[0];
+        const statement = this.visit(ctx.statement);
+        return rejectAndJoin(ctx.Colon[0], [identifier, statement]);
+    }
+    expressionStatement(ctx) {
+        const statementExpression = this.visit(ctx.statementExpression);
+        return rejectAndConcat([statementExpression, ctx.Semicolon[0]]);
+    }
+    statementExpression(ctx) {
+        return this.visitSingle(ctx);
+    }
+    ifStatement(ctx) {
+        var _a;
+        const expression = this.visit(ctx.expression);
+        const ifStatement = this.visit(ctx.statement[0], {
+            allowEmptyStatement: true
+        });
+        const ifSeparator = isStatementEmptyStatement(ifStatement) ? "" : " ";
+        let elsePart = "";
+        if (ctx.Else !== undefined) {
+            const elseStatement = this.visit(ctx.statement[1], {
+                allowEmptyStatement: true
+            });
+            const elseSeparator = isStatementEmptyStatement(elseStatement) ? "" : " ";
+            const elseOnSameLine = hasTrailingLineComments(ctx.statement[0]) ||
+                hasLeadingLineComments(ctx.Else[0]) ||
+                !((_a = ctx.statement[0].children.statementWithoutTrailingSubstatement) === null || _a === void 0 ? void 0 : _a[0].children.block)
+                ? hardline
+                : " ";
+            elsePart = rejectAndJoin(elseSeparator, [
+                concat([elseOnSameLine, ctx.Else[0]]),
+                elseStatement
+            ]);
+        }
+        return rejectAndConcat([
+            rejectAndJoin(" ", [
+                ctx.If[0],
+                concat([
+                    putIntoBraces(expression, softline, ctx.LBrace[0], ctx.RBrace[0]),
+                    ifSeparator
+                ])
+            ]),
+            ifStatement,
+            elsePart
+        ]);
+    }
+    assertStatement(ctx) {
+        const expressions = this.mapVisit(ctx.expression);
+        const colon = ctx.Colon ? ctx.Colon[0] : ":";
+        return rejectAndConcat([
+            concat([ctx.Assert[0], " "]),
+            rejectAndJoin(concat([" ", colon, " "]), expressions),
+            ctx.Semicolon[0]
+        ]);
+    }
+    switchStatement(ctx) {
+        const expression = this.visit(ctx.expression);
+        const switchBlock = this.visit(ctx.switchBlock);
+        return rejectAndJoin(" ", [
+            ctx.Switch[0],
+            putIntoBraces(expression, softline, ctx.LBrace[0], ctx.RBrace[0]),
+            switchBlock
+        ]);
+    }
+    switchBlock(ctx) {
+        const switchCases = ctx.switchBlockStatementGroup !== undefined
+            ? this.mapVisit(ctx.switchBlockStatementGroup)
+            : this.mapVisit(ctx.switchRule);
+        return putIntoBraces(rejectAndJoin(hardline, switchCases), hardline, ctx.LCurly[0], ctx.RCurly[0]);
+    }
+    switchBlockStatementGroup(ctx) {
+        var _a, _b, _c;
+        const switchLabel = this.visit(ctx.switchLabel);
+        const blockStatements = this.visit(ctx.blockStatements);
+        const statements = (_a = ctx.blockStatements) === null || _a === void 0 ? void 0 : _a[0].children.blockStatement;
+        const hasSingleStatementBlock = (statements === null || statements === void 0 ? void 0 : statements.length) === 1 &&
+            ((_c = (_b = statements[0].children.statement) === null || _b === void 0 ? void 0 : _b[0].children.statementWithoutTrailingSubstatement) === null || _c === void 0 ? void 0 : _c[0].children.block) !== undefined;
+        return concat([
+            switchLabel,
+            ctx.Colon[0],
+            hasSingleStatementBlock
+                ? concat([" ", blockStatements])
+                : blockStatements && indent([hardline, blockStatements])
+        ]);
+    }
+    switchLabel(ctx) {
+        var _a, _b, _c;
+        const Case = (_a = ctx.Case) === null || _a === void 0 ? void 0 : _a[0];
+        const commas = (_b = ctx.Comma) === null || _b === void 0 ? void 0 : _b.map(elt => concat([elt, line]));
+        if (ctx.caseConstant || ctx.Null) {
+            const caseConstants = ctx.Null
+                ? [ctx.Null[0], (_c = ctx.Default) === null || _c === void 0 ? void 0 : _c[0]]
+                : this.mapVisit(ctx.caseConstant);
+            return group(indent(join(" ", [Case, rejectAndJoinSeps(commas, caseConstants)])));
+        }
+        else if (ctx.casePattern) {
+            const casePatterns = this.mapVisit(ctx.casePattern);
+            const guard = this.visit(ctx.guard);
+            const multiplePatterns = ctx.casePattern.length > 1;
+            const separator = multiplePatterns ? line : " ";
+            const contents = join(separator, [
+                Case,
+                rejectAndJoinSeps(commas, casePatterns)
+            ]);
+            return group(rejectAndJoin(separator, [
+                multiplePatterns ? indent(contents) : contents,
+                guard
+            ]));
+        }
+        return printTokenWithComments(ctx.Default[0]);
+    }
+    switchRule(ctx) {
+        const switchLabel = this.visit(ctx.switchLabel);
+        let caseInstruction;
+        if (ctx.throwStatement !== undefined) {
+            caseInstruction = this.visit(ctx.throwStatement);
+        }
+        else if (ctx.block !== undefined) {
+            caseInstruction = this.visit(ctx.block);
+        }
+        else {
+            caseInstruction = concat([this.visit(ctx.expression), ctx.Semicolon[0]]);
+        }
+        return concat([switchLabel, " ", ctx.Arrow[0], " ", caseInstruction]);
+    }
+    caseConstant(ctx) {
+        return this.visitSingle(ctx);
+    }
+    casePattern(ctx) {
+        return this.visitSingle(ctx);
+    }
+    whileStatement(ctx) {
+        const expression = this.visit(ctx.expression);
+        const statement = this.visit(ctx.statement[0], {
+            allowEmptyStatement: true
+        });
+        const statementSeparator = isStatementEmptyStatement(statement) ? "" : " ";
+        return rejectAndJoin(" ", [
+            ctx.While[0],
+            rejectAndJoin(statementSeparator, [
+                putIntoBraces(expression, softline, ctx.LBrace[0], ctx.RBrace[0]),
+                statement
+            ])
+        ]);
+    }
+    doStatement(ctx) {
+        const statement = this.visit(ctx.statement[0], {
+            allowEmptyStatement: true
+        });
+        const statementSeparator = isStatementEmptyStatement(statement) ? "" : " ";
+        const expression = this.visit(ctx.expression);
+        return rejectAndJoin(" ", [
+            rejectAndJoin(statementSeparator, [ctx.Do[0], statement]),
+            ctx.While[0],
+            rejectAndConcat([
+                putIntoBraces(expression, softline, ctx.LBrace[0], ctx.RBrace[0]),
+                ctx.Semicolon[0]
+            ])
+        ]);
+    }
+    forStatement(ctx) {
+        return this.visitSingle(ctx);
+    }
+    basicForStatement(ctx) {
+        const forInit = this.visit(ctx.forInit);
+        const expression = this.visit(ctx.expression);
+        const forUpdate = this.visit(ctx.forUpdate);
+        const statement = this.visit(ctx.statement[0], {
+            allowEmptyStatement: true
+        });
+        const statementSeparator = isStatementEmptyStatement(statement) ? "" : " ";
+        return rejectAndConcat([
+            rejectAndJoin(" ", [
+                ctx.For[0],
+                putIntoBraces(rejectAndConcat([
+                    forInit,
+                    rejectAndJoin(line, [ctx.Semicolon[0], expression]),
+                    rejectAndJoin(line, [ctx.Semicolon[1], forUpdate])
+                ]), softline, ctx.LBrace[0], ctx.RBrace[0])
+            ]),
+            statementSeparator,
+            statement
+        ]);
+    }
+    forInit(ctx) {
+        return this.visitSingle(ctx);
+    }
+    forUpdate(ctx) {
+        return this.visitSingle(ctx);
+    }
+    statementExpressionList(ctx) {
+        const statementExpressions = this.mapVisit(ctx.statementExpression);
+        const commas = ctx.Comma
+            ? ctx.Comma.map(elt => {
+                return concat([printTokenWithComments(elt), " "]);
+            })
+            : [];
+        return rejectAndJoinSeps(commas, statementExpressions);
+    }
+    enhancedForStatement(ctx) {
+        const localVariableDeclaration = this.visit(ctx.localVariableDeclaration);
+        const expression = this.visit(ctx.expression);
+        const statement = this.visit(ctx.statement[0], {
+            allowEmptyStatement: true
+        });
+        const statementSeparator = isStatementEmptyStatement(statement) ? "" : " ";
+        return rejectAndConcat([
+            rejectAndJoin(" ", [ctx.For[0], ctx.LBrace[0]]),
+            localVariableDeclaration,
+            concat([" ", ctx.Colon[0], " "]),
+            expression,
+            concat([ctx.RBrace[0], statementSeparator]),
+            statement
+        ]);
+    }
+    breakStatement(ctx) {
+        if (ctx.Identifier) {
+            const identifier = ctx.Identifier[0];
+            return rejectAndConcat([
+                concat([ctx.Break[0], " "]),
+                identifier,
+                ctx.Semicolon[0]
+            ]);
+        }
+        return concat([ctx.Break[0], ctx.Semicolon[0]]);
+    }
+    continueStatement(ctx) {
+        if (ctx.Identifier) {
+            const identifier = ctx.Identifier[0];
+            return rejectAndConcat([
+                concat([ctx.Continue[0], " "]),
+                identifier,
+                ctx.Semicolon[0]
+            ]);
+        }
+        return rejectAndConcat([ctx.Continue[0], ctx.Semicolon[0]]);
+    }
+    returnStatement(ctx) {
+        if (ctx.expression) {
+            const expression = this.visit(ctx.expression, {
+                addParenthesisToWrapStatement: true
+            });
+            return rejectAndConcat([
+                concat([ctx.Return[0], " "]),
+                expression,
+                ctx.Semicolon[0]
+            ]);
+        }
+        return rejectAndConcat([ctx.Return[0], ctx.Semicolon[0]]);
+    }
+    throwStatement(ctx) {
+        const expression = this.visit(ctx.expression);
+        return rejectAndConcat([
+            concat([ctx.Throw[0], " "]),
+            expression,
+            ctx.Semicolon[0]
+        ]);
+    }
+    synchronizedStatement(ctx) {
+        const expression = this.visit(ctx.expression);
+        const block = this.visit(ctx.block);
+        return rejectAndConcat([
+            join(" ", [
+                ctx.Synchronized[0],
+                concat([
+                    putIntoBraces(expression, softline, ctx.LBrace[0], ctx.RBrace[0]),
+                    " "
+                ])
+            ]),
+            block
+        ]);
+    }
+    tryStatement(ctx) {
+        if (ctx.tryWithResourcesStatement) {
+            return this.visit(ctx.tryWithResourcesStatement);
+        }
+        const block = this.visit(ctx.block);
+        const catches = this.visit(ctx.catches);
+        const finallyBlock = this.visit(ctx.finally);
+        return rejectAndJoin(" ", [ctx.Try[0], block, catches, finallyBlock]);
+    }
+    catches(ctx) {
+        const catchClauses = this.mapVisit(ctx.catchClause);
+        return rejectAndJoin(" ", catchClauses);
+    }
+    catchClause(ctx) {
+        const catchFormalParameter = this.visit(ctx.catchFormalParameter);
+        const block = this.visit(ctx.block);
+        return rejectAndConcat([
+            group(rejectAndConcat([
+                rejectAndJoin(" ", [ctx.Catch[0], ctx.LBrace[0]]),
+                indent(rejectAndConcat([softline, catchFormalParameter])),
+                softline,
+                concat([ctx.RBrace[0], " "])
+            ])),
+            block
+        ]);
+    }
+    catchFormalParameter(ctx) {
+        const variableModifiers = this.mapVisit(ctx.variableModifier);
+        const catchType = this.visit(ctx.catchType);
+        const variableDeclaratorId = this.visit(ctx.variableDeclaratorId);
+        return rejectAndJoin(" ", [
+            rejectAndJoin(" ", variableModifiers),
+            catchType,
+            variableDeclaratorId
+        ]);
+    }
+    catchType(ctx) {
+        const unannClassType = this.visit(ctx.unannClassType);
+        const classTypes = this.mapVisit(ctx.classType);
+        const ors = ctx.Or ? ctx.Or.map(elt => concat([line, elt, " "])) : [];
+        return group(rejectAndJoinSeps(ors, [unannClassType, ...classTypes]));
+    }
+    finally(ctx) {
+        const block = this.visit(ctx.block);
+        return rejectAndJoin(" ", [ctx.Finally[0], block]);
+    }
+    tryWithResourcesStatement(ctx) {
+        const resourceSpecification = this.visit(ctx.resourceSpecification);
+        const block = this.visit(ctx.block);
+        const catches = this.visit(ctx.catches);
+        const finallyBlock = this.visit(ctx.finally);
+        return rejectAndJoin(" ", [
+            ctx.Try[0],
+            resourceSpecification,
+            block,
+            catches,
+            finallyBlock
+        ]);
+    }
+    resourceSpecification(ctx) {
+        const resourceList = this.visit(ctx.resourceList);
+        const optionalSemicolon = ctx.Semicolon ? ctx.Semicolon[0] : "";
+        return putIntoBraces(rejectAndConcat([resourceList, optionalSemicolon]), softline, ctx.LBrace[0], ctx.RBrace[0]);
+    }
+    resourceList(ctx) {
+        const resources = this.mapVisit(ctx.resource);
+        const semicolons = ctx.Semicolon
+            ? ctx.Semicolon.map(elt => {
+                return concat([elt, line]);
+            })
+            : [""];
+        return rejectAndJoinSeps(semicolons, resources);
+    }
+    resource(ctx) {
+        return this.visitSingle(ctx);
+    }
+    yieldStatement(ctx) {
+        const expression = this.visit(ctx.expression);
+        return join(" ", [ctx.Yield[0], concat([expression, ctx.Semicolon[0]])]);
+    }
+    variableAccess(ctx) {
+        return this.visitSingle(ctx);
+    }
+}