185 lines
		
	
	
		
			5.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			185 lines
		
	
	
		
			5.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| /**
 | |
|  * @fileoverview A rule to set the maximum number of statements in a function.
 | |
|  * @author Ian Christian Myers
 | |
|  */
 | |
| 
 | |
| "use strict";
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // Requirements
 | |
| //------------------------------------------------------------------------------
 | |
| 
 | |
| const astUtils = require("./utils/ast-utils");
 | |
| const { upperCaseFirst } = require("../shared/string-utils");
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // Rule Definition
 | |
| //------------------------------------------------------------------------------
 | |
| 
 | |
| /** @type {import('../shared/types').Rule} */
 | |
| module.exports = {
 | |
|     meta: {
 | |
|         type: "suggestion",
 | |
| 
 | |
|         docs: {
 | |
|             description: "Enforce a maximum number of statements allowed in function blocks",
 | |
|             recommended: false,
 | |
|             url: "https://eslint.org/docs/latest/rules/max-statements"
 | |
|         },
 | |
| 
 | |
|         schema: [
 | |
|             {
 | |
|                 oneOf: [
 | |
|                     {
 | |
|                         type: "integer",
 | |
|                         minimum: 0
 | |
|                     },
 | |
|                     {
 | |
|                         type: "object",
 | |
|                         properties: {
 | |
|                             maximum: {
 | |
|                                 type: "integer",
 | |
|                                 minimum: 0
 | |
|                             },
 | |
|                             max: {
 | |
|                                 type: "integer",
 | |
|                                 minimum: 0
 | |
|                             }
 | |
|                         },
 | |
|                         additionalProperties: false
 | |
|                     }
 | |
|                 ]
 | |
|             },
 | |
|             {
 | |
|                 type: "object",
 | |
|                 properties: {
 | |
|                     ignoreTopLevelFunctions: {
 | |
|                         type: "boolean"
 | |
|                     }
 | |
|                 },
 | |
|                 additionalProperties: false
 | |
|             }
 | |
|         ],
 | |
|         messages: {
 | |
|             exceed: "{{name}} has too many statements ({{count}}). Maximum allowed is {{max}}."
 | |
|         }
 | |
|     },
 | |
| 
 | |
|     create(context) {
 | |
| 
 | |
|         //--------------------------------------------------------------------------
 | |
|         // Helpers
 | |
|         //--------------------------------------------------------------------------
 | |
| 
 | |
|         const functionStack = [],
 | |
|             option = context.options[0],
 | |
|             ignoreTopLevelFunctions = context.options[1] && context.options[1].ignoreTopLevelFunctions || false,
 | |
|             topLevelFunctions = [];
 | |
|         let maxStatements = 10;
 | |
| 
 | |
|         if (
 | |
|             typeof option === "object" &&
 | |
|             (Object.prototype.hasOwnProperty.call(option, "maximum") || Object.prototype.hasOwnProperty.call(option, "max"))
 | |
|         ) {
 | |
|             maxStatements = option.maximum || option.max;
 | |
|         } else if (typeof option === "number") {
 | |
|             maxStatements = option;
 | |
|         }
 | |
| 
 | |
|         /**
 | |
|          * Reports a node if it has too many statements
 | |
|          * @param {ASTNode} node node to evaluate
 | |
|          * @param {int} count Number of statements in node
 | |
|          * @param {int} max Maximum number of statements allowed
 | |
|          * @returns {void}
 | |
|          * @private
 | |
|          */
 | |
|         function reportIfTooManyStatements(node, count, max) {
 | |
|             if (count > max) {
 | |
|                 const name = upperCaseFirst(astUtils.getFunctionNameWithKind(node));
 | |
| 
 | |
|                 context.report({
 | |
|                     node,
 | |
|                     messageId: "exceed",
 | |
|                     data: { name, count, max }
 | |
|                 });
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /**
 | |
|          * When parsing a new function, store it in our function stack
 | |
|          * @returns {void}
 | |
|          * @private
 | |
|          */
 | |
|         function startFunction() {
 | |
|             functionStack.push(0);
 | |
|         }
 | |
| 
 | |
|         /**
 | |
|          * Evaluate the node at the end of function
 | |
|          * @param {ASTNode} node node to evaluate
 | |
|          * @returns {void}
 | |
|          * @private
 | |
|          */
 | |
|         function endFunction(node) {
 | |
|             const count = functionStack.pop();
 | |
| 
 | |
|             /*
 | |
|              * This rule does not apply to class static blocks, but we have to track them so
 | |
|              * that statements in them do not count as statements in the enclosing function.
 | |
|              */
 | |
|             if (node.type === "StaticBlock") {
 | |
|                 return;
 | |
|             }
 | |
| 
 | |
|             if (ignoreTopLevelFunctions && functionStack.length === 0) {
 | |
|                 topLevelFunctions.push({ node, count });
 | |
|             } else {
 | |
|                 reportIfTooManyStatements(node, count, maxStatements);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /**
 | |
|          * Increment the count of the functions
 | |
|          * @param {ASTNode} node node to evaluate
 | |
|          * @returns {void}
 | |
|          * @private
 | |
|          */
 | |
|         function countStatements(node) {
 | |
|             functionStack[functionStack.length - 1] += node.body.length;
 | |
|         }
 | |
| 
 | |
|         //--------------------------------------------------------------------------
 | |
|         // Public API
 | |
|         //--------------------------------------------------------------------------
 | |
| 
 | |
|         return {
 | |
|             FunctionDeclaration: startFunction,
 | |
|             FunctionExpression: startFunction,
 | |
|             ArrowFunctionExpression: startFunction,
 | |
|             StaticBlock: startFunction,
 | |
| 
 | |
|             BlockStatement: countStatements,
 | |
| 
 | |
|             "FunctionDeclaration:exit": endFunction,
 | |
|             "FunctionExpression:exit": endFunction,
 | |
|             "ArrowFunctionExpression:exit": endFunction,
 | |
|             "StaticBlock:exit": endFunction,
 | |
| 
 | |
|             "Program:exit"() {
 | |
|                 if (topLevelFunctions.length === 1) {
 | |
|                     return;
 | |
|                 }
 | |
| 
 | |
|                 topLevelFunctions.forEach(element => {
 | |
|                     const count = element.count;
 | |
|                     const node = element.node;
 | |
| 
 | |
|                     reportIfTooManyStatements(node, count, maxStatements);
 | |
|                 });
 | |
|             }
 | |
|         };
 | |
| 
 | |
|     }
 | |
| };
 |