ReturnStatement.js

"use strict";

var _tk = require('rocambole-token');
var _ws = require('rocambole-whitespace');

var expressionParentheses = require('./expressionParentheses');


exports.format = function ReturnStatement(node) {
  // need to make sure we only remove line breaks inside the node itself
  // because of ASI (see #29)
  var nonEmpty = _tk.findInBetween(node.startToken.next, node.endToken, _tk.isNotEmpty);
  // XXX: we want to remove line breaks and white spaces inside the node, not
  // using _br.limitAfter to avoid changing the program behavior (ASI)
  if (nonEmpty) _tk.removeEmptyInBetween(node.startToken, nonEmpty);

  _ws.limitAfter(node.startToken, 1);
  if (_tk.isSemiColon(node.endToken)) {
    // XXX: we want semicolon to be on same line and no whitespaces for now.
    _tk.removeEmptyInBetween(_tk.findPrevNonEmpty(node.endToken), node.endToken);
  }

  if (node.argument) {
    expressionParentheses.addSpaceInside(node.argument);
  }
};


exports.getIndentEdges = function(node, opts) {
  // we bypass indentation if argument already adds indentation
  if (
    !node.argument ||
    opts[node.argument.type] ||
    node.argument.type === 'UnaryExpression'
  ) {
    return false;
  }

  // handle agument wrapped in parenthesis
  var argumentStart = _tk.findNextNonEmpty(node.startToken);
  if (argumentStart.value === '(') {
    return {
      startToken: argumentStart,
      endToken: node.endToken.value === ')' ?
        node.endToken :
        _tk.findPrevNonEmpty(_tk.findPrev(node.endToken, ')'))
    };
  }

  return {
    startToken: node.startToken.next,
    endToken: _tk.isEmpty(node.endToken) ?
      _tk.findPrevNonEmpty(node.endToken) :
      node.endToken
  };
};