prettier-plugin-solidity/dist/slang-nodes/IfStatement.js

A Prettier Plugin for automatically formatting your Solidity code.

import { doc } from 'prettier';
import { NonterminalKind } from '@nomicfoundation/slang/cst';
import { printSeparatedItem } from '../slang-printers/print-separated-item.js';
import { getNodeMetadata, updateMetadata } from '../slang-utils/metadata.js';
import { isBlockComment } from '../slang-utils/is-comment.js';
import { Expression } from './Expression.js';
import { Statement } from './Statement.js';
import { ElseBranch } from './ElseBranch.js';
const { group, hardline, indent, line } = doc.builders;
export class IfStatement {
    constructor(ast, options) {
        this.kind = NonterminalKind.IfStatement;
        let metadata = getNodeMetadata(ast);
        this.condition = new Expression(ast.condition, options);
        this.body = new Statement(ast.body, options);
        if (ast.elseBranch) {
            this.elseBranch = new ElseBranch(ast.elseBranch, options);
        }
        metadata = updateMetadata(metadata, [
            this.condition,
            this.body,
            this.elseBranch
        ]);
        this.comments = metadata.comments;
        this.loc = metadata.loc;
    }
    print(path, print) {
        return [
            'if (',
            printSeparatedItem(path.call(print, 'condition')),
            ')',
            this.body.variant.kind === NonterminalKind.Block
                ? [' ', path.call(print, 'body')]
                : group(indent([line, path.call(print, 'body')]), {
                    shouldBreak: this.body.variant.kind === NonterminalKind.IfStatement // `if` within `if`
                }),
            this.elseBranch
                ? [
                    this.body.variant.kind !== NonterminalKind.Block || // else on a new line if body is not a block
                        this.body.variant.comments.some((comment) => !isBlockComment(comment) || comment.placement === 'ownLine') // or if body has trailing single line comments or a block comment on a new line
                        ? hardline
                        : ' ',
                    path.call(print, 'elseBranch')
                ]
                : ''
        ];
    }
}
//# sourceMappingURL=IfStatement.js.map