| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213 | /** * @fileoverview A rule to set the maximum number of line of code in a function. * @author Pete Ward <peteward44@gmail.com> */"use strict";//------------------------------------------------------------------------------// Requirements//------------------------------------------------------------------------------const astUtils = require("./utils/ast-utils");const { upperCaseFirst } = require("../shared/string-utils");//------------------------------------------------------------------------------// Constants//------------------------------------------------------------------------------const OPTIONS_SCHEMA = {    type: "object",    properties: {        max: {            type: "integer",            minimum: 0        },        skipComments: {            type: "boolean"        },        skipBlankLines: {            type: "boolean"        },        IIFEs: {            type: "boolean"        }    },    additionalProperties: false};const OPTIONS_OR_INTEGER_SCHEMA = {    oneOf: [        OPTIONS_SCHEMA,        {            type: "integer",            minimum: 1        }    ]};/** * Given a list of comment nodes, return a map with numeric keys (source code line numbers) and comment token values. * @param {Array} comments An array of comment nodes. * @returns {Map<string, Node>} A map with numeric keys (source code line numbers) and comment token values. */function getCommentLineNumbers(comments) {    const map = new Map();    comments.forEach(comment => {        for (let i = comment.loc.start.line; i <= comment.loc.end.line; i++) {            map.set(i, comment);        }    });    return map;}//------------------------------------------------------------------------------// Rule Definition//------------------------------------------------------------------------------/** @type {import('../shared/types').Rule} */module.exports = {    meta: {        type: "suggestion",        docs: {            description: "Enforce a maximum number of lines of code in a function",            recommended: false,            url: "https://eslint.org/docs/rules/max-lines-per-function"        },        schema: [            OPTIONS_OR_INTEGER_SCHEMA        ],        messages: {            exceed: "{{name}} has too many lines ({{lineCount}}). Maximum allowed is {{maxLines}}."        }    },    create(context) {        const sourceCode = context.getSourceCode();        const lines = sourceCode.lines;        const option = context.options[0];        let maxLines = 50;        let skipComments = false;        let skipBlankLines = false;        let IIFEs = false;        if (typeof option === "object") {            maxLines = typeof option.max === "number" ? option.max : 50;            skipComments = !!option.skipComments;            skipBlankLines = !!option.skipBlankLines;            IIFEs = !!option.IIFEs;        } else if (typeof option === "number") {            maxLines = option;        }        const commentLineNumbers = getCommentLineNumbers(sourceCode.getAllComments());        //--------------------------------------------------------------------------        // Helpers        //--------------------------------------------------------------------------        /**         * Tells if a comment encompasses the entire line.         * @param {string} line The source line with a trailing comment         * @param {number} lineNumber The one-indexed line number this is on         * @param {ASTNode} comment The comment to remove         * @returns {boolean} If the comment covers the entire line         */        function isFullLineComment(line, lineNumber, comment) {            const start = comment.loc.start,                end = comment.loc.end,                isFirstTokenOnLine = start.line === lineNumber && !line.slice(0, start.column).trim(),                isLastTokenOnLine = end.line === lineNumber && !line.slice(end.column).trim();            return comment &&                (start.line < lineNumber || isFirstTokenOnLine) &&                (end.line > lineNumber || isLastTokenOnLine);        }        /**         * Identifies is a node is a FunctionExpression which is part of an IIFE         * @param {ASTNode} node Node to test         * @returns {boolean} True if it's an IIFE         */        function isIIFE(node) {            return (node.type === "FunctionExpression" || node.type === "ArrowFunctionExpression") && node.parent && node.parent.type === "CallExpression" && node.parent.callee === node;        }        /**         * Identifies is a node is a FunctionExpression which is embedded within a MethodDefinition or Property         * @param {ASTNode} node Node to test         * @returns {boolean} True if it's a FunctionExpression embedded within a MethodDefinition or Property         */        function isEmbedded(node) {            if (!node.parent) {                return false;            }            if (node !== node.parent.value) {                return false;            }            if (node.parent.type === "MethodDefinition") {                return true;            }            if (node.parent.type === "Property") {                return node.parent.method === true || node.parent.kind === "get" || node.parent.kind === "set";            }            return false;        }        /**         * Count the lines in the function         * @param {ASTNode} funcNode Function AST node         * @returns {void}         * @private         */        function processFunction(funcNode) {            const node = isEmbedded(funcNode) ? funcNode.parent : funcNode;            if (!IIFEs && isIIFE(node)) {                return;            }            let lineCount = 0;            for (let i = node.loc.start.line - 1; i < node.loc.end.line; ++i) {                const line = lines[i];                if (skipComments) {                    if (commentLineNumbers.has(i + 1) && isFullLineComment(line, i + 1, commentLineNumbers.get(i + 1))) {                        continue;                    }                }                if (skipBlankLines) {                    if (line.match(/^\s*$/u)) {                        continue;                    }                }                lineCount++;            }            if (lineCount > maxLines) {                const name = upperCaseFirst(astUtils.getFunctionNameWithKind(funcNode));                context.report({                    node,                    messageId: "exceed",                    data: { name, lineCount, maxLines }                });            }        }        //--------------------------------------------------------------------------        // Public API        //--------------------------------------------------------------------------        return {            FunctionDeclaration: processFunction,            FunctionExpression: processFunction,            ArrowFunctionExpression: processFunction        };    }};
 |