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