prettierx/src/language-js/print/member.js

prettierX - a less opinionated fork of the Prettier code formatter

"use strict";

const {
  // [prettierx] --computed-property-spacing option support
  builders: { softline, line, group, indent, label },
} = require("../../document");
const {
  isNumericLiteral,
  isMemberExpression,
  isCallExpression,
} = require("../utils");
const { printOptionalToken } = require("./misc");

function printMemberExpression(path, options, print) {
  const node = path.getValue();

  const parent = path.getParentNode();
  let firstNonMemberParent;
  let i = 0;
  do {
    firstNonMemberParent = path.getParentNode(i);
    i++;
  } while (
    firstNonMemberParent &&
    (isMemberExpression(firstNonMemberParent) ||
      firstNonMemberParent.type === "TSNonNullExpression")
  );

  const objectDoc = print("object");
  const lookupDoc = printMemberLookup(path, options, print);

  const shouldInline =
    (firstNonMemberParent &&
      (firstNonMemberParent.type === "NewExpression" ||
        firstNonMemberParent.type === "BindExpression" ||
        (firstNonMemberParent.type === "AssignmentExpression" &&
          firstNonMemberParent.left.type !== "Identifier"))) ||
    node.computed ||
    (node.object.type === "Identifier" &&
      node.property.type === "Identifier" &&
      !isMemberExpression(parent)) ||
    ((parent.type === "AssignmentExpression" ||
      parent.type === "VariableDeclarator") &&
      ((isCallExpression(node.object) && node.object.arguments.length > 0) ||
        (node.object.type === "TSNonNullExpression" &&
          isCallExpression(node.object.expression) &&
          node.object.expression.arguments.length > 0) ||
        objectDoc.label === "member-chain"));

  return label(objectDoc.label === "member-chain" ? "member-chain" : "member", [
    objectDoc,
    shouldInline ? lookupDoc : group(indent([softline, lookupDoc])),
  ]);
}

function printMemberLookup(path, options, print) {
  const property = print("property");
  const node = path.getValue();
  const optional = printOptionalToken(path);

  // [prettierx] --computed-property-spacing option support
  const computedPropertySpace = options.computedPropertySpacing ? " " : "";
  const computedPropertyLine = options.computedPropertySpacing
    ? line
    : softline;

  if (!node.computed) {
    return [optional, ".", property];
  }

  if (!node.property || isNumericLiteral(node.property)) {
    // [prettierx] --computed-property-spacing option support
    return [
      optional,
      "[",
      computedPropertySpace,
      property,
      computedPropertySpace,
      "]",
    ];
  }

  // [prettierx] --computed-property-spacing option support
  return group([
    optional,
    "[",
    // [prettierx] --computed-property-spacing option support
    indent([computedPropertyLine, property]),
    computedPropertyLine,
    "]",
  ]);
}

module.exports = { printMemberExpression, printMemberLookup };