/**
|
* @fileoverview enforce the location of arrow function bodies
|
* @author Sharmila Jesupaul
|
*/
|
"use strict";
|
|
const { isCommentToken, isNotOpeningParenToken } = require("./utils/ast-utils");
|
|
//------------------------------------------------------------------------------
|
// Rule Definition
|
//------------------------------------------------------------------------------
|
module.exports = {
|
meta: {
|
type: "layout",
|
|
docs: {
|
description: "enforce the location of arrow function bodies",
|
category: "Stylistic Issues",
|
recommended: false,
|
url: "https://eslint.org/docs/rules/implicit-arrow-linebreak"
|
},
|
|
fixable: "whitespace",
|
|
schema: [
|
{
|
enum: ["beside", "below"]
|
}
|
],
|
messages: {
|
expected: "Expected a linebreak before this expression.",
|
unexpected: "Expected no linebreak before this expression."
|
}
|
},
|
|
create(context) {
|
const sourceCode = context.getSourceCode();
|
const option = context.options[0] || "beside";
|
|
/**
|
* Validates the location of an arrow function body
|
* @param {ASTNode} node The arrow function body
|
* @returns {void}
|
*/
|
function validateExpression(node) {
|
if (node.body.type === "BlockStatement") {
|
return;
|
}
|
|
const arrowToken = sourceCode.getTokenBefore(node.body, isNotOpeningParenToken);
|
const firstTokenOfBody = sourceCode.getTokenAfter(arrowToken);
|
|
if (arrowToken.loc.end.line === firstTokenOfBody.loc.start.line && option === "below") {
|
context.report({
|
node: firstTokenOfBody,
|
messageId: "expected",
|
fix: fixer => fixer.insertTextBefore(firstTokenOfBody, "\n")
|
});
|
} else if (arrowToken.loc.end.line !== firstTokenOfBody.loc.start.line && option === "beside") {
|
context.report({
|
node: firstTokenOfBody,
|
messageId: "unexpected",
|
fix(fixer) {
|
if (sourceCode.getFirstTokenBetween(arrowToken, firstTokenOfBody, { includeComments: true, filter: isCommentToken })) {
|
return null;
|
}
|
|
return fixer.replaceTextRange([arrowToken.range[1], firstTokenOfBody.range[0]], " ");
|
}
|
});
|
}
|
}
|
|
//----------------------------------------------------------------------
|
// Public
|
//----------------------------------------------------------------------
|
return {
|
ArrowFunctionExpression: node => validateExpression(node)
|
};
|
}
|
};
|
