@rawwee/prettier-plugin-twig-melody/src/print/UnarySubclass.js

Prettier Plugin for Twig/Melody (Enhanced Fork)

const prettier = require("prettier");
const { concat, softline, indent, group } = prettier.doc.builders;
const { Node } = require("melody-types");
const {
    firstValueInAncestorChain,
    findParentNode,
    isMultipartExpression,
    IS_ROOT_LOGICAL_EXPRESSION,
    GROUP_TOP_LEVEL_LOGICAL
} = require("../util");

const argumentNeedsParentheses = node => isMultipartExpression(node);

const isLogicalOperator = operator => operator === "not";

const printLogicalExpression = (node, path, print) => {
    const foundRootAbove = firstValueInAncestorChain(
        path,
        IS_ROOT_LOGICAL_EXPRESSION,
        false
    );
    if (!foundRootAbove) {
        node[IS_ROOT_LOGICAL_EXPRESSION] = true;
    }
    const parentNode = findParentNode(path);
    const shouldGroupOnTopLevel = parentNode[GROUP_TOP_LEVEL_LOGICAL] !== false;

    const parts = [node.operator, " "];
    const needsParentheses = argumentNeedsParentheses(node.argument);
    const printedArgument = path.call(print, "argument");
    if (needsParentheses) {
        parts.push(
            "(",
            indent(concat([softline, printedArgument])),
            concat([softline, ")"])
        );
    } else {
        parts.push(printedArgument);
    }
    const result = concat(parts);
    const shouldCreateTopLevelGroup = !foundRootAbove && shouldGroupOnTopLevel;

    return shouldCreateTopLevelGroup ? group(result) : result;
};

const p = (node, path, print) => {
    const parts = [];
    // Example: a is not same as ... => Here, the "not" is printed "inline"
    // Therefore, we do not output it here
    const hasTestExpressionArgument = Node.isTestExpression(node.argument);
    if (isLogicalOperator(node.operator) && !hasTestExpressionArgument) {
        return printLogicalExpression(node, path, print);
    }
    if (!hasTestExpressionArgument) {
        parts.push(node.operator, " ");
    }
    parts.push(path.call(print, "argument"));
    return concat(parts);
};

module.exports = {
    printUnarySubclass: p
};