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);
 | |
|                         }
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|         };
 | |
|     }
 | |
| };
 |