prettier 0.21.0 → 1.0.1

data/src/nodes/conditionals.js

@@ -9,7 +9,7 @@ const {
   softline
 } = require("../prettier");
 
-const { containsAssignment } = require("../utils");
+const { containsAssignment, isEmptyStmts } = require("../utils");
 const inlineEnsureParens = require("../utils/inlineEnsureParens");
 
 const printWithAddition = (keyword, path, print, { breaking = false } = {}) =>
@@ -80,36 +80,50 @@ const printTernary = (path, _opts, print) => {
 // Prints an `if_mod` or `unless_mod` node. Because it was previously in the
 // modifier form, we're guaranteed to not have an additional node, so we can
 // just work with the predicate and the body.
-const printSingle = (keyword) => (path, { inlineConditionals }, print) => {
-  const multiline = concat([
-    `${keyword} `,
-    align(keyword.length + 1, path.call(print, "body", 0)),
-    indent(concat([softline, path.call(print, "body", 1)])),
-    concat([softline, "end"])
-  ]);
+function printSingle(keyword, modifier = false) {
+  return function printSingleWithKeyword(path, { rubyModifier }, print) {
+    const [_predicateNode, statementsNode] = path.getValue().body;
+    const predicateDoc = path.call(print, "body", 0);
+    const statementsDoc = path.call(print, "body", 1);
 
-  const [_predicate, stmts] = path.getValue().body;
-  const hasComments =
-    stmts.type === "stmts" &&
-    stmts.body.some((stmt) => stmt.type === "@comment");
+    const multilineParts = [
+      `${keyword} `,
+      align(keyword.length + 1, predicateDoc),
+      indent(concat([softline, statementsDoc])),
+      softline,
+      "end"
+    ];
 
-  if (!inlineConditionals || hasComments) {
-    return multiline;
-  }
+    // If we do not allow modifier form conditionals or there are comments
+    // inside of the body of the conditional, then we must print in the
+    // multiline form.
+    if (!rubyModifier || (!modifier && statementsNode.body[0].comments)) {
+      return concat([concat(multilineParts), breakParent]);
+    }
 
-  const inline = concat(
-    inlineEnsureParens(path, [
-      path.call(print, "body", 1),
-      ` ${keyword} `,
-      path.call(print, "body", 0)
-    ])
-  );
+    const inline = concat(
+      inlineEnsureParens(path, [
+        path.call(print, "body", 1),
+        ` ${keyword} `,
+        path.call(print, "body", 0)
+      ])
+    );
 
-  return group(ifBreak(multiline, inline));
-};
+    // An expression with a conditional modifier (expression if true), the
+    // conditional body is parsed before the predicate expression, meaning that
+    // if the parser encountered a variable declaration, it would initialize
+    // that variable first before evaluating the predicate expression. That
+    // parse order means the difference between a NameError or not. #591
+    // https://docs.ruby-lang.org/en/2.0.0/syntax/control_expressions_rdoc.html#label-Modifier+if+and+unless
+    if (modifier && containsAssignment(statementsNode)) {
+      return inline;
+    }
+
+    return group(ifBreak(concat(multilineParts), inline));
+  };
+}
 
 const noTernary = [
-  "@comment",
   "alias",
   "assign",
   "break",
@@ -176,7 +190,7 @@ const canTernary = (path) => {
 };
 
 // A normalized print function for both `if` and `unless` nodes.
-const printConditional = (keyword) => (path, { inlineConditionals }, print) => {
+const printConditional = (keyword) => (path, { rubyModifier }, print) => {
   if (canTernary(path)) {
     let ternaryParts = [path.call(print, "body", 0), " ? "].concat(
       printTernaryClauses(
@@ -205,7 +219,7 @@ const printConditional = (keyword) => (path, { inlineConditionals }, print) => {
 
   // If the body of the conditional is empty, then we explicitly have to use the
   // block form.
-  if (statements.type === "stmts" && statements.body[0].type === "void_stmt") {
+  if (isEmptyStmts(statements) && !statements.body[0].comments) {
     return concat([
       `${keyword} `,
       align(keyword.length + 1, path.call(print, "body", 0)),
@@ -225,7 +239,7 @@ const printConditional = (keyword) => (path, { inlineConditionals }, print) => {
     ]);
   }
 
-  return printSingle(keyword)(path, { inlineConditionals }, print);
+  return printSingle(keyword)(path, { rubyModifier }, print);
 };
 
 module.exports = {
@@ -260,7 +274,7 @@ module.exports = {
   },
   if: printConditional("if"),
   ifop: printTernary,
-  if_mod: printSingle("if"),
+  if_mod: printSingle("if", true),
   unless: printConditional("unless"),
-  unless_mod: printSingle("unless")
+  unless_mod: printSingle("unless", true)
 };

data/src/nodes/constants.js

@@ -1,25 +1,43 @@
 const { concat, group, indent, join, softline } = require("../prettier");
-const { first, makeCall, prefix } = require("../utils");
+const { makeCall } = require("../utils");
+
+function printConstPath(path, opts, print) {
+  return join("::", path.map(print, "body"));
+}
+
+function printConstRef(path, opts, print) {
+  return path.call(print, "body", 0);
+}
+
+function printDefined(path, opts, print) {
+  return group(
+    concat([
+      "defined?(",
+      indent(concat([softline, path.call(print, "body", 0)])),
+      concat([softline, ")"])
+    ])
+  );
+}
+
+function printField(path, opts, print) {
+  return group(
+    concat([
+      path.call(print, "body", 0),
+      concat([makeCall(path, opts, print), path.call(print, "body", 2)])
+    ])
+  );
+}
+
+function printTopConst(path, opts, print) {
+  return concat(["::", path.call(print, "body", 0)]);
+}
 
 module.exports = {
-  const_path_field: (path, opts, print) => join("::", path.map(print, "body")),
-  const_path_ref: (path, opts, print) => join("::", path.map(print, "body")),
-  const_ref: first,
-  defined: (path, opts, print) =>
-    group(
-      concat([
-        "defined?(",
-        indent(concat([softline, path.call(print, "body", 0)])),
-        concat([softline, ")"])
-      ])
-    ),
-  field: (path, opts, print) =>
-    group(
-      concat([
-        path.call(print, "body", 0),
-        concat([makeCall(path, opts, print), path.call(print, "body", 2)])
-      ])
-    ),
-  top_const_field: prefix("::"),
-  top_const_ref: prefix("::")
+  const_path_field: printConstPath,
+  const_path_ref: printConstPath,
+  const_ref: printConstRef,
+  defined: printDefined,
+  field: printField,
+  top_const_field: printTopConst,
+  top_const_ref: printTopConst
 };

data/src/nodes/flow.js

@@ -1,5 +1,15 @@
 const { concat, join } = require("../prettier");
-const { literal, nodeDive } = require("../utils");
+const { literal } = require("../utils");
+
+const nodeDive = (node, steps) => {
+  let current = node;
+
+  steps.forEach((step) => {
+    current = current[step];
+  });
+
+  return current;
+};
 
 const unskippableParens = [
   "if_mod",

data/src/nodes/hashes.js

@@ -1,14 +1,11 @@
-const {
-  concat,
-  group,
-  ifBreak,
-  indent,
-  join,
-  line,
-  literalline
-} = require("../prettier");
+const { concat, group, ifBreak, indent, join, line } = require("../prettier");
 
-const { nodeDive, prefix, skipAssignIndent } = require("../utils");
+const {
+  getTrailingComma,
+  prefix,
+  printEmptyCollection,
+  skipAssignIndent
+} = require("../utils");
 
 // When attempting to convert a hash rocket into a hash label, you need to take
 // care because only certain patterns are allowed. Ruby source says that they
@@ -20,128 +17,112 @@ const { nodeDive, prefix, skipAssignIndent } = require("../utils");
 //
 // This function represents that check, as it determines if it can convert the
 // symbol node into a hash label.
-const isValidHashLabel = (symbolLiteral) => {
-  const label = symbolLiteral.body[0].body[0].body;
+function isValidHashLabel(symbolLiteral) {
+  const label = symbolLiteral.body[0].body;
   return label.match(/^[_A-Za-z]/) && !label.endsWith("=");
-};
+}
 
-const makeLabel = (path, { preferHashLabels }, print, steps) => {
-  const labelNode = nodeDive(path.getValue(), steps);
-  const labelDoc = path.call.apply(path, [print].concat(steps));
+function canUseHashLabels(contentsNode) {
+  return contentsNode.body.every((assocNode) => {
+    if (assocNode.type === "assoc_splat") {
+      return true;
+    }
 
-  switch (labelNode.type) {
-    case "@label":
-      if (preferHashLabels) {
-        return labelDoc;
-      }
-      return `:${labelDoc.slice(0, labelDoc.length - 1)} =>`;
-    case "symbol_literal": {
-      if (preferHashLabels && isValidHashLabel(labelNode)) {
-        const symbolSteps = steps.concat("body", 0, "body", 0);
-
-        return concat([
-          path.call.apply(path, [print].concat(symbolSteps)),
-          ":"
-        ]);
-      }
-      return concat([labelDoc, " =>"]);
+    switch (assocNode.body[0].type) {
+      case "@label":
+        return true;
+      case "symbol_literal":
+        return isValidHashLabel(assocNode.body[0]);
+      case "dyna_symbol":
+        return true;
+      default:
+        return false;
     }
+  });
+}
+
+function printHashKeyLabel(path, print) {
+  const node = path.getValue();
+
+  switch (node.type) {
+    case "@label":
+      return print(path);
+    case "symbol_literal":
+      return concat([path.call(print, "body", 0), ":"]);
     case "dyna_symbol":
-      if (preferHashLabels) {
-        return concat(labelDoc.parts.slice(1).concat(":"));
-      }
-      return concat([labelDoc, " =>"]);
-    default:
-      return concat([labelDoc, " =>"]);
+      return concat(print(path).parts.slice(1).concat(":"));
   }
-};
+}
 
-function printHash(path, { addTrailingCommas }, print) {
+function printHashKeyRocket(path, print) {
+  const node = path.getValue();
+  const doc = print(path);
+
+  if (node.type === "@label") {
+    return `:${doc.slice(0, doc.length - 1)} =>`;
+  }
+
+  return concat([doc, " =>"]);
+}
+
+function printAssocNew(path, opts, print) {
+  const { keyPrinter } = path.getParentNode();
+
+  const parts = [path.call((keyPath) => keyPrinter(keyPath, print), "body", 0)];
+  const valueDoc = path.call(print, "body", 1);
+
+  if (skipAssignIndent(path.getValue().body[1])) {
+    parts.push(" ", valueDoc);
+  } else {
+    parts.push(indent(concat([line, valueDoc])));
+  }
+
+  return group(concat(parts));
+}
+
+function printHashContents(path, opts, print) {
+  const node = path.getValue();
+
+  // First determine which key printer we're going to use, so that the child
+  // nodes can reference it when they go to get printed.
+  node.keyPrinter =
+    opts.rubyHashLabel && canUseHashLabels(path.getValue())
+      ? printHashKeyLabel
+      : printHashKeyRocket;
+
+  return join(concat([",", line]), path.map(print, "body"));
+}
+
+function printHash(path, opts, print) {
   const hashNode = path.getValue();
 
   // Hashes normally have a single assoclist_from_args child node. If it's
   // missing, then it means we're dealing with an empty hash, so we can just
   // exit here and print.
   if (hashNode.body[0] === null) {
-    return "{}";
-  }
-
-  // Here we get a reference to the printed assoclist_from_args child node,
-  // which handles printing all of the key-value pairs of the hash. We're
-  // wrapping it in an array in case we need to append a trailing comma.
-  const assocDocs = [path.call(print, "body", 0)];
-
-  // Here we get a reference to the last key-value pair's value node, in order
-  // to check if we're dealing with a heredoc. If we are, then the trailing
-  // comma printing is handled from within the assoclist_from_args node
-  // printing, because the trailing comma has to go after the heredoc
-  // declaration.
-  const assocNodes = hashNode.body[0].body[0];
-  const lastAssocValueNode = assocNodes[assocNodes.length - 1].body[1];
-
-  // If we're adding a trailing comma and the last key-value pair's value node
-  // is not a heredoc node, then we can safely append the extra comma if the
-  // hash ends up getting printed on multiple lines.
-  if (addTrailingCommas && lastAssocValueNode.type !== "heredoc") {
-    assocDocs.push(ifBreak(",", ""));
+    return printEmptyCollection(path, opts, "{", "}");
   }
 
   return group(
     concat([
       "{",
-      indent(concat([line, concat(assocDocs)])),
-      concat([line, "}"])
+      indent(
+        concat([
+          line,
+          path.call(print, "body", 0),
+          getTrailingComma(opts) ? ifBreak(",", "") : ""
+        ])
+      ),
+      line,
+      "}"
     ])
   );
 }
 
 module.exports = {
-  assoc_new: (path, opts, print) => {
-    const valueDoc = path.call(print, "body", 1);
-    const parts = [makeLabel(path, opts, print, ["body", 0])];
-
-    if (skipAssignIndent(path.getValue().body[1])) {
-      parts.push(" ", valueDoc);
-    } else {
-      parts.push(indent(concat([line, valueDoc])));
-    }
-
-    return group(concat(parts));
-  },
+  assoc_new: printAssocNew,
   assoc_splat: prefix("**"),
-  assoclist_from_args: (path, opts, print) => {
-    const { addTrailingCommas } = opts;
-
-    const assocNodes = path.getValue().body[0];
-    const assocDocs = [];
-
-    assocNodes.forEach((assocNode, index) => {
-      const isInner = index !== assocNodes.length - 1;
-      const valueNode = assocNode.body[1];
-
-      if (valueNode && valueNode.type === "heredoc") {
-        assocDocs.push(
-          makeLabel(path, opts, print, ["body", 0, index, "body", 0]),
-          " ",
-          valueNode.beging,
-          isInner || addTrailingCommas ? "," : "",
-          literalline,
-          concat(path.map(print, "body", 0, index, "body", 1, "body")),
-          valueNode.ending,
-          isInner ? line : ""
-        );
-      } else {
-        assocDocs.push(path.call(print, "body", 0, index));
-
-        if (isInner) {
-          assocDocs.push(concat([",", line]));
-        }
-      }
-    });
-
-    return group(concat(assocDocs));
-  },
-  bare_assoc_hash: (path, opts, print) =>
-    group(join(concat([",", line]), path.map(print, "body", 0))),
+  assoclist_from_args: printHashContents,
+  bare_assoc_hash: printHashContents,
   hash: printHash
 };

data/src/nodes/heredocs.js

@@ -0,0 +1,34 @@
+const { concat, group, lineSuffix, join } = require("../prettier");
+const { literalLineNoBreak } = require("../utils");
+
+function printHeredoc(path, opts, print) {
+  const { body, ending } = path.getValue();
+
+  const parts = body.map((part, index) => {
+    if (part.type !== "@tstring_content") {
+      // In this case, the part of the string is an embedded expression
+      return path.call(print, "body", index);
+    }
+
+    // In this case, the part of the string is just regular string content
+    return join(literalLineNoBreak, part.body.split("\n"));
+  });
+
+  // We use a literalline break because matching indentation is required
+  // for the heredoc contents and ending. If the line suffix contains a
+  // break-parent, all ancestral groups are broken, and heredocs automatically
+  // break lines in groups they appear in. We prefer them to appear in-line if
+  // possible, so we use a literalline without the break-parent.
+  return group(
+    concat([
+      path.call(print, "beging"),
+      lineSuffix(
+        group(concat([literalLineNoBreak].concat(parts).concat(ending)))
+      )
+    ])
+  );
+}
+
+module.exports = {
+  heredoc: printHeredoc
+};