141 lines
5.0 KiB
JavaScript
141 lines
5.0 KiB
JavaScript
|
/**
|
||
|
* @fileoverview enforce "for" loop update clause moving the counter in the right direction.(for-direction)
|
||
|
* @author Aladdin-ADD<hh_2013@foxmail.com>
|
||
|
*/
|
||
|
|
||
|
"use strict";
|
||
|
|
||
|
//------------------------------------------------------------------------------
|
||
|
// Requirements
|
||
|
//------------------------------------------------------------------------------
|
||
|
|
||
|
const { getStaticValue } = require("@eslint-community/eslint-utils");
|
||
|
|
||
|
//------------------------------------------------------------------------------
|
||
|
// Rule Definition
|
||
|
//------------------------------------------------------------------------------
|
||
|
|
||
|
/** @type {import('../shared/types').Rule} */
|
||
|
module.exports = {
|
||
|
meta: {
|
||
|
type: "problem",
|
||
|
|
||
|
docs: {
|
||
|
description: "Enforce \"for\" loop update clause moving the counter in the right direction",
|
||
|
recommended: true,
|
||
|
url: "https://eslint.org/docs/latest/rules/for-direction"
|
||
|
},
|
||
|
|
||
|
fixable: null,
|
||
|
schema: [],
|
||
|
|
||
|
messages: {
|
||
|
incorrectDirection: "The update clause in this loop moves the variable in the wrong direction."
|
||
|
}
|
||
|
},
|
||
|
|
||
|
create(context) {
|
||
|
const { sourceCode } = context;
|
||
|
|
||
|
/**
|
||
|
* report an error.
|
||
|
* @param {ASTNode} node the node to report.
|
||
|
* @returns {void}
|
||
|
*/
|
||
|
function report(node) {
|
||
|
context.report({
|
||
|
node,
|
||
|
messageId: "incorrectDirection"
|
||
|
});
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* check the right side of the assignment
|
||
|
* @param {ASTNode} update UpdateExpression to check
|
||
|
* @param {int} dir expected direction that could either be turned around or invalidated
|
||
|
* @returns {int} return dir, the negated dir, or zero if the counter does not change or the direction is not clear
|
||
|
*/
|
||
|
function getRightDirection(update, dir) {
|
||
|
const staticValue = getStaticValue(update.right, sourceCode.getScope(update));
|
||
|
|
||
|
if (staticValue && ["bigint", "boolean", "number"].includes(typeof staticValue.value)) {
|
||
|
const sign = Math.sign(Number(staticValue.value)) || 0; // convert NaN to 0
|
||
|
|
||
|
return dir * sign;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* check UpdateExpression add/sub the counter
|
||
|
* @param {ASTNode} update UpdateExpression to check
|
||
|
* @param {string} counter variable name to check
|
||
|
* @returns {int} if add return 1, if sub return -1, if nochange, return 0
|
||
|
*/
|
||
|
function getUpdateDirection(update, counter) {
|
||
|
if (update.argument.type === "Identifier" && update.argument.name === counter) {
|
||
|
if (update.operator === "++") {
|
||
|
return 1;
|
||
|
}
|
||
|
if (update.operator === "--") {
|
||
|
return -1;
|
||
|
}
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* check AssignmentExpression add/sub the counter
|
||
|
* @param {ASTNode} update AssignmentExpression to check
|
||
|
* @param {string} counter variable name to check
|
||
|
* @returns {int} if add return 1, if sub return -1, if nochange, return 0
|
||
|
*/
|
||
|
function getAssignmentDirection(update, counter) {
|
||
|
if (update.left.name === counter) {
|
||
|
if (update.operator === "+=") {
|
||
|
return getRightDirection(update, 1);
|
||
|
}
|
||
|
if (update.operator === "-=") {
|
||
|
return getRightDirection(update, -1);
|
||
|
}
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
return {
|
||
|
ForStatement(node) {
|
||
|
|
||
|
if (node.test && node.test.type === "BinaryExpression" && node.update) {
|
||
|
for (const counterPosition of ["left", "right"]) {
|
||
|
if (node.test[counterPosition].type !== "Identifier") {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
const counter = node.test[counterPosition].name;
|
||
|
const operator = node.test.operator;
|
||
|
const update = node.update;
|
||
|
|
||
|
let wrongDirection;
|
||
|
|
||
|
if (operator === "<" || operator === "<=") {
|
||
|
wrongDirection = counterPosition === "left" ? -1 : 1;
|
||
|
} else if (operator === ">" || operator === ">=") {
|
||
|
wrongDirection = counterPosition === "left" ? 1 : -1;
|
||
|
} else {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (update.type === "UpdateExpression") {
|
||
|
if (getUpdateDirection(update, counter) === wrongDirection) {
|
||
|
report(node);
|
||
|
}
|
||
|
} else if (update.type === "AssignmentExpression" && getAssignmentDirection(update, counter) === wrongDirection) {
|
||
|
report(node);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
}
|
||
|
};
|