You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
486 lines
19 KiB
486 lines
19 KiB
/** |
|
* @fileoverview Rule to flag statements without curly braces |
|
* @author Nicholas C. Zakas |
|
*/ |
|
"use strict"; |
|
|
|
//------------------------------------------------------------------------------ |
|
// Requirements |
|
//------------------------------------------------------------------------------ |
|
|
|
const astUtils = require("./utils/ast-utils"); |
|
|
|
//------------------------------------------------------------------------------ |
|
// Rule Definition |
|
//------------------------------------------------------------------------------ |
|
|
|
/** @type {import('../shared/types').Rule} */ |
|
module.exports = { |
|
meta: { |
|
type: "suggestion", |
|
|
|
docs: { |
|
description: "Enforce consistent brace style for all control statements", |
|
recommended: false, |
|
url: "https://eslint.org/docs/rules/curly" |
|
}, |
|
|
|
schema: { |
|
anyOf: [ |
|
{ |
|
type: "array", |
|
items: [ |
|
{ |
|
enum: ["all"] |
|
} |
|
], |
|
minItems: 0, |
|
maxItems: 1 |
|
}, |
|
{ |
|
type: "array", |
|
items: [ |
|
{ |
|
enum: ["multi", "multi-line", "multi-or-nest"] |
|
}, |
|
{ |
|
enum: ["consistent"] |
|
} |
|
], |
|
minItems: 0, |
|
maxItems: 2 |
|
} |
|
] |
|
}, |
|
|
|
fixable: "code", |
|
|
|
messages: { |
|
missingCurlyAfter: "Expected { after '{{name}}'.", |
|
missingCurlyAfterCondition: "Expected { after '{{name}}' condition.", |
|
unexpectedCurlyAfter: "Unnecessary { after '{{name}}'.", |
|
unexpectedCurlyAfterCondition: "Unnecessary { after '{{name}}' condition." |
|
} |
|
}, |
|
|
|
create(context) { |
|
|
|
const multiOnly = (context.options[0] === "multi"); |
|
const multiLine = (context.options[0] === "multi-line"); |
|
const multiOrNest = (context.options[0] === "multi-or-nest"); |
|
const consistent = (context.options[1] === "consistent"); |
|
|
|
const sourceCode = context.getSourceCode(); |
|
|
|
//-------------------------------------------------------------------------- |
|
// Helpers |
|
//-------------------------------------------------------------------------- |
|
|
|
/** |
|
* Determines if a given node is a one-liner that's on the same line as it's preceding code. |
|
* @param {ASTNode} node The node to check. |
|
* @returns {boolean} True if the node is a one-liner that's on the same line as it's preceding code. |
|
* @private |
|
*/ |
|
function isCollapsedOneLiner(node) { |
|
const before = sourceCode.getTokenBefore(node); |
|
const last = sourceCode.getLastToken(node); |
|
const lastExcludingSemicolon = astUtils.isSemicolonToken(last) ? sourceCode.getTokenBefore(last) : last; |
|
|
|
return before.loc.start.line === lastExcludingSemicolon.loc.end.line; |
|
} |
|
|
|
/** |
|
* Determines if a given node is a one-liner. |
|
* @param {ASTNode} node The node to check. |
|
* @returns {boolean} True if the node is a one-liner. |
|
* @private |
|
*/ |
|
function isOneLiner(node) { |
|
if (node.type === "EmptyStatement") { |
|
return true; |
|
} |
|
|
|
const first = sourceCode.getFirstToken(node); |
|
const last = sourceCode.getLastToken(node); |
|
const lastExcludingSemicolon = astUtils.isSemicolonToken(last) ? sourceCode.getTokenBefore(last) : last; |
|
|
|
return first.loc.start.line === lastExcludingSemicolon.loc.end.line; |
|
} |
|
|
|
/** |
|
* Determines if the given node is a lexical declaration (let, const, function, or class) |
|
* @param {ASTNode} node The node to check |
|
* @returns {boolean} True if the node is a lexical declaration |
|
* @private |
|
*/ |
|
function isLexicalDeclaration(node) { |
|
if (node.type === "VariableDeclaration") { |
|
return node.kind === "const" || node.kind === "let"; |
|
} |
|
|
|
return node.type === "FunctionDeclaration" || node.type === "ClassDeclaration"; |
|
} |
|
|
|
/** |
|
* Checks if the given token is an `else` token or not. |
|
* @param {Token} token The token to check. |
|
* @returns {boolean} `true` if the token is an `else` token. |
|
*/ |
|
function isElseKeywordToken(token) { |
|
return token.value === "else" && token.type === "Keyword"; |
|
} |
|
|
|
/** |
|
* Determines whether the given node has an `else` keyword token as the first token after. |
|
* @param {ASTNode} node The node to check. |
|
* @returns {boolean} `true` if the node is followed by an `else` keyword token. |
|
*/ |
|
function isFollowedByElseKeyword(node) { |
|
const nextToken = sourceCode.getTokenAfter(node); |
|
|
|
return Boolean(nextToken) && isElseKeywordToken(nextToken); |
|
} |
|
|
|
/** |
|
* Determines if a semicolon needs to be inserted after removing a set of curly brackets, in order to avoid a SyntaxError. |
|
* @param {Token} closingBracket The } token |
|
* @returns {boolean} `true` if a semicolon needs to be inserted after the last statement in the block. |
|
*/ |
|
function needsSemicolon(closingBracket) { |
|
const tokenBefore = sourceCode.getTokenBefore(closingBracket); |
|
const tokenAfter = sourceCode.getTokenAfter(closingBracket); |
|
const lastBlockNode = sourceCode.getNodeByRangeIndex(tokenBefore.range[0]); |
|
|
|
if (astUtils.isSemicolonToken(tokenBefore)) { |
|
|
|
// If the last statement already has a semicolon, don't add another one. |
|
return false; |
|
} |
|
|
|
if (!tokenAfter) { |
|
|
|
// If there are no statements after this block, there is no need to add a semicolon. |
|
return false; |
|
} |
|
|
|
if (lastBlockNode.type === "BlockStatement" && lastBlockNode.parent.type !== "FunctionExpression" && lastBlockNode.parent.type !== "ArrowFunctionExpression") { |
|
|
|
/* |
|
* If the last node surrounded by curly brackets is a BlockStatement (other than a FunctionExpression or an ArrowFunctionExpression), |
|
* don't insert a semicolon. Otherwise, the semicolon would be parsed as a separate statement, which would cause |
|
* a SyntaxError if it was followed by `else`. |
|
*/ |
|
return false; |
|
} |
|
|
|
if (tokenBefore.loc.end.line === tokenAfter.loc.start.line) { |
|
|
|
// If the next token is on the same line, insert a semicolon. |
|
return true; |
|
} |
|
|
|
if (/^[([/`+-]/u.test(tokenAfter.value)) { |
|
|
|
// If the next token starts with a character that would disrupt ASI, insert a semicolon. |
|
return true; |
|
} |
|
|
|
if (tokenBefore.type === "Punctuator" && (tokenBefore.value === "++" || tokenBefore.value === "--")) { |
|
|
|
// If the last token is ++ or --, insert a semicolon to avoid disrupting ASI. |
|
return true; |
|
} |
|
|
|
// Otherwise, do not insert a semicolon. |
|
return false; |
|
} |
|
|
|
/** |
|
* Determines whether the code represented by the given node contains an `if` statement |
|
* that would become associated with an `else` keyword directly appended to that code. |
|
* |
|
* Examples where it returns `true`: |
|
* |
|
* if (a) |
|
* foo(); |
|
* |
|
* if (a) { |
|
* foo(); |
|
* } |
|
* |
|
* if (a) |
|
* foo(); |
|
* else if (b) |
|
* bar(); |
|
* |
|
* while (a) |
|
* if (b) |
|
* if(c) |
|
* foo(); |
|
* else |
|
* bar(); |
|
* |
|
* Examples where it returns `false`: |
|
* |
|
* if (a) |
|
* foo(); |
|
* else |
|
* bar(); |
|
* |
|
* while (a) { |
|
* if (b) |
|
* if(c) |
|
* foo(); |
|
* else |
|
* bar(); |
|
* } |
|
* |
|
* while (a) |
|
* if (b) { |
|
* if(c) |
|
* foo(); |
|
* } |
|
* else |
|
* bar(); |
|
* @param {ASTNode} node Node representing the code to check. |
|
* @returns {boolean} `true` if an `if` statement within the code would become associated with an `else` appended to that code. |
|
*/ |
|
function hasUnsafeIf(node) { |
|
switch (node.type) { |
|
case "IfStatement": |
|
if (!node.alternate) { |
|
return true; |
|
} |
|
return hasUnsafeIf(node.alternate); |
|
case "ForStatement": |
|
case "ForInStatement": |
|
case "ForOfStatement": |
|
case "LabeledStatement": |
|
case "WithStatement": |
|
case "WhileStatement": |
|
return hasUnsafeIf(node.body); |
|
default: |
|
return false; |
|
} |
|
} |
|
|
|
/** |
|
* Determines whether the existing curly braces around the single statement are necessary to preserve the semantics of the code. |
|
* The braces, which make the given block body, are necessary in either of the following situations: |
|
* |
|
* 1. The statement is a lexical declaration. |
|
* 2. Without the braces, an `if` within the statement would become associated with an `else` after the closing brace: |
|
* |
|
* if (a) { |
|
* if (b) |
|
* foo(); |
|
* } |
|
* else |
|
* bar(); |
|
* |
|
* if (a) |
|
* while (b) |
|
* while (c) { |
|
* while (d) |
|
* if (e) |
|
* while(f) |
|
* foo(); |
|
* } |
|
* else |
|
* bar(); |
|
* @param {ASTNode} node `BlockStatement` body with exactly one statement directly inside. The statement can have its own nested statements. |
|
* @returns {boolean} `true` if the braces are necessary - removing them (replacing the given `BlockStatement` body with its single statement content) |
|
* would change the semantics of the code or produce a syntax error. |
|
*/ |
|
function areBracesNecessary(node) { |
|
const statement = node.body[0]; |
|
|
|
return isLexicalDeclaration(statement) || |
|
hasUnsafeIf(statement) && isFollowedByElseKeyword(node); |
|
} |
|
|
|
/** |
|
* Prepares to check the body of a node to see if it's a block statement. |
|
* @param {ASTNode} node The node to report if there's a problem. |
|
* @param {ASTNode} body The body node to check for blocks. |
|
* @param {string} name The name to report if there's a problem. |
|
* @param {{ condition: boolean }} opts Options to pass to the report functions |
|
* @returns {Object} a prepared check object, with "actual", "expected", "check" properties. |
|
* "actual" will be `true` or `false` whether the body is already a block statement. |
|
* "expected" will be `true` or `false` if the body should be a block statement or not, or |
|
* `null` if it doesn't matter, depending on the rule options. It can be modified to change |
|
* the final behavior of "check". |
|
* "check" will be a function reporting appropriate problems depending on the other |
|
* properties. |
|
*/ |
|
function prepareCheck(node, body, name, opts) { |
|
const hasBlock = (body.type === "BlockStatement"); |
|
let expected = null; |
|
|
|
if (hasBlock && (body.body.length !== 1 || areBracesNecessary(body))) { |
|
expected = true; |
|
} else if (multiOnly) { |
|
expected = false; |
|
} else if (multiLine) { |
|
if (!isCollapsedOneLiner(body)) { |
|
expected = true; |
|
} |
|
|
|
// otherwise, the body is allowed to have braces or not to have braces |
|
|
|
} else if (multiOrNest) { |
|
if (hasBlock) { |
|
const statement = body.body[0]; |
|
const leadingCommentsInBlock = sourceCode.getCommentsBefore(statement); |
|
|
|
expected = !isOneLiner(statement) || leadingCommentsInBlock.length > 0; |
|
} else { |
|
expected = !isOneLiner(body); |
|
} |
|
} else { |
|
|
|
// default "all" |
|
expected = true; |
|
} |
|
|
|
return { |
|
actual: hasBlock, |
|
expected, |
|
check() { |
|
if (this.expected !== null && this.expected !== this.actual) { |
|
if (this.expected) { |
|
context.report({ |
|
node, |
|
loc: body.loc, |
|
messageId: opts && opts.condition ? "missingCurlyAfterCondition" : "missingCurlyAfter", |
|
data: { |
|
name |
|
}, |
|
fix: fixer => fixer.replaceText(body, `{${sourceCode.getText(body)}}`) |
|
}); |
|
} else { |
|
context.report({ |
|
node, |
|
loc: body.loc, |
|
messageId: opts && opts.condition ? "unexpectedCurlyAfterCondition" : "unexpectedCurlyAfter", |
|
data: { |
|
name |
|
}, |
|
fix(fixer) { |
|
|
|
/* |
|
* `do while` expressions sometimes need a space to be inserted after `do`. |
|
* e.g. `do{foo()} while (bar)` should be corrected to `do foo() while (bar)` |
|
*/ |
|
const needsPrecedingSpace = node.type === "DoWhileStatement" && |
|
sourceCode.getTokenBefore(body).range[1] === body.range[0] && |
|
!astUtils.canTokensBeAdjacent("do", sourceCode.getFirstToken(body, { skip: 1 })); |
|
|
|
const openingBracket = sourceCode.getFirstToken(body); |
|
const closingBracket = sourceCode.getLastToken(body); |
|
const lastTokenInBlock = sourceCode.getTokenBefore(closingBracket); |
|
|
|
if (needsSemicolon(closingBracket)) { |
|
|
|
/* |
|
* If removing braces would cause a SyntaxError due to multiple statements on the same line (or |
|
* change the semantics of the code due to ASI), don't perform a fix. |
|
*/ |
|
return null; |
|
} |
|
|
|
const resultingBodyText = sourceCode.getText().slice(openingBracket.range[1], lastTokenInBlock.range[0]) + |
|
sourceCode.getText(lastTokenInBlock) + |
|
sourceCode.getText().slice(lastTokenInBlock.range[1], closingBracket.range[0]); |
|
|
|
return fixer.replaceText(body, (needsPrecedingSpace ? " " : "") + resultingBodyText); |
|
} |
|
}); |
|
} |
|
} |
|
} |
|
}; |
|
} |
|
|
|
/** |
|
* Prepares to check the bodies of a "if", "else if" and "else" chain. |
|
* @param {ASTNode} node The first IfStatement node of the chain. |
|
* @returns {Object[]} prepared checks for each body of the chain. See `prepareCheck` for more |
|
* information. |
|
*/ |
|
function prepareIfChecks(node) { |
|
const preparedChecks = []; |
|
|
|
for (let currentNode = node; currentNode; currentNode = currentNode.alternate) { |
|
preparedChecks.push(prepareCheck(currentNode, currentNode.consequent, "if", { condition: true })); |
|
if (currentNode.alternate && currentNode.alternate.type !== "IfStatement") { |
|
preparedChecks.push(prepareCheck(currentNode, currentNode.alternate, "else")); |
|
break; |
|
} |
|
} |
|
|
|
if (consistent) { |
|
|
|
/* |
|
* If any node should have or already have braces, make sure they |
|
* all have braces. |
|
* If all nodes shouldn't have braces, make sure they don't. |
|
*/ |
|
const expected = preparedChecks.some(preparedCheck => { |
|
if (preparedCheck.expected !== null) { |
|
return preparedCheck.expected; |
|
} |
|
return preparedCheck.actual; |
|
}); |
|
|
|
preparedChecks.forEach(preparedCheck => { |
|
preparedCheck.expected = expected; |
|
}); |
|
} |
|
|
|
return preparedChecks; |
|
} |
|
|
|
//-------------------------------------------------------------------------- |
|
// Public |
|
//-------------------------------------------------------------------------- |
|
|
|
return { |
|
IfStatement(node) { |
|
const parent = node.parent; |
|
const isElseIf = parent.type === "IfStatement" && parent.alternate === node; |
|
|
|
if (!isElseIf) { |
|
|
|
// This is a top `if`, check the whole `if-else-if` chain |
|
prepareIfChecks(node).forEach(preparedCheck => { |
|
preparedCheck.check(); |
|
}); |
|
} |
|
|
|
// Skip `else if`, it's already checked (when the top `if` was visited) |
|
}, |
|
|
|
WhileStatement(node) { |
|
prepareCheck(node, node.body, "while", { condition: true }).check(); |
|
}, |
|
|
|
DoWhileStatement(node) { |
|
prepareCheck(node, node.body, "do").check(); |
|
}, |
|
|
|
ForStatement(node) { |
|
prepareCheck(node, node.body, "for", { condition: true }).check(); |
|
}, |
|
|
|
ForInStatement(node) { |
|
prepareCheck(node, node.body, "for-in").check(); |
|
}, |
|
|
|
ForOfStatement(node) { |
|
prepareCheck(node, node.body, "for-of").check(); |
|
} |
|
}; |
|
} |
|
};
|
|
|