| /** | |
| * @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) | |
| }; | |
| } | |
| }; |