Contents

const { concat, group, hardline, indent, line } = require("../../prettier");
const { isEmptyBodyStmt } = require("../../utils");

function printMethod(offset) {
  return function printMethodWithOffset(path, opts, print) {
    const [_name, params, bodystmt] = path.getValue().body.slice(offset);
    const declaration = ["def "];

    // In this case, we're printing a method that's defined as a singleton, so
    // we need to include the target and the operator
    if (offset > 0) {
      declaration.push(
        path.call(print, "body", 0),
        path.call(print, "body", 1)
      );
    }

    // In case there are no parens but there are arguments
    const parens = params.type === "params" && params.body.some((type) => type);

    declaration.push(
      path.call(print, "body", offset),
      parens ? "(" : "",
      path.call(print, "body", offset + 1),
      parens ? ")" : ""
    );

    if (isEmptyBodyStmt(bodystmt)) {
      return group(concat(declaration.concat("; end")));
    }

    return group(
      concat([
        group(concat(declaration)),
        indent(concat([hardline, path.call(print, "body", offset + 2)])),
        group(concat([hardline, "end"]))
      ])
    );
  };
}

function printSingleLineMethod(path, opts, print) {
  let parensNode = path.getValue().body[1];
  let paramsDoc = "";

  if (parensNode) {
    const paramsNode = parensNode.body[0];

    if (paramsNode.body.some((type) => type)) {
      paramsDoc = path.call(print, "body", 1);
    }
  }

  return group(
    concat([
      "def ",
      path.call(print, "body", 0),
      paramsDoc,
      " =",
      indent(group(concat([line, path.call(print, "body", 2)])))
    ])
  );
}

function printAccessControl(path, opts, print) {
  return path.call(print, "body", 0);
}

module.exports = {
  access_ctrl: printAccessControl,
  def: printMethod(0),
  defs: printMethod(2),
  defsl: printSingleLineMethod
};

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
prettier-1.6.1	src/ruby/nodes/methods.js
prettier-1.6.0	src/ruby/nodes/methods.js
prettier-1.5.5	src/ruby/nodes/methods.js
prettier-1.5.4	src/ruby/nodes/methods.js
prettier-1.5.3	src/ruby/nodes/methods.js
prettier-1.5.2	src/ruby/nodes/methods.js
prettier-1.5.1	src/ruby/nodes/methods.js
prettier-1.5.0	src/ruby/nodes/methods.js