130 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			130 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
/**
 | 
						|
 * @fileoverview Rule to enforce the use of `u` flag on RegExp.
 | 
						|
 * @author Toru Nagashima
 | 
						|
 */
 | 
						|
 | 
						|
"use strict";
 | 
						|
 | 
						|
//------------------------------------------------------------------------------
 | 
						|
// Requirements
 | 
						|
//------------------------------------------------------------------------------
 | 
						|
 | 
						|
const {
 | 
						|
    CALL,
 | 
						|
    CONSTRUCT,
 | 
						|
    ReferenceTracker,
 | 
						|
    getStringIfConstant
 | 
						|
} = require("@eslint-community/eslint-utils");
 | 
						|
const astUtils = require("./utils/ast-utils.js");
 | 
						|
const { isValidWithUnicodeFlag } = require("./utils/regular-expressions");
 | 
						|
 | 
						|
//------------------------------------------------------------------------------
 | 
						|
// Rule Definition
 | 
						|
//------------------------------------------------------------------------------
 | 
						|
 | 
						|
/** @type {import('../shared/types').Rule} */
 | 
						|
module.exports = {
 | 
						|
    meta: {
 | 
						|
        type: "suggestion",
 | 
						|
 | 
						|
        docs: {
 | 
						|
            description: "Enforce the use of `u` or `v` flag on RegExp",
 | 
						|
            recommended: false,
 | 
						|
            url: "https://eslint.org/docs/latest/rules/require-unicode-regexp"
 | 
						|
        },
 | 
						|
 | 
						|
        hasSuggestions: true,
 | 
						|
 | 
						|
        messages: {
 | 
						|
            addUFlag: "Add the 'u' flag.",
 | 
						|
            requireUFlag: "Use the 'u' flag."
 | 
						|
        },
 | 
						|
 | 
						|
        schema: []
 | 
						|
    },
 | 
						|
 | 
						|
    create(context) {
 | 
						|
 | 
						|
        const sourceCode = context.sourceCode;
 | 
						|
 | 
						|
        return {
 | 
						|
            "Literal[regex]"(node) {
 | 
						|
                const flags = node.regex.flags || "";
 | 
						|
 | 
						|
                if (!flags.includes("u") && !flags.includes("v")) {
 | 
						|
                    context.report({
 | 
						|
                        messageId: "requireUFlag",
 | 
						|
                        node,
 | 
						|
                        suggest: isValidWithUnicodeFlag(context.languageOptions.ecmaVersion, node.regex.pattern)
 | 
						|
                            ? [
 | 
						|
                                {
 | 
						|
                                    fix(fixer) {
 | 
						|
                                        return fixer.insertTextAfter(node, "u");
 | 
						|
                                    },
 | 
						|
                                    messageId: "addUFlag"
 | 
						|
                                }
 | 
						|
                            ]
 | 
						|
                            : null
 | 
						|
                    });
 | 
						|
                }
 | 
						|
            },
 | 
						|
 | 
						|
            Program(node) {
 | 
						|
                const scope = sourceCode.getScope(node);
 | 
						|
                const tracker = new ReferenceTracker(scope);
 | 
						|
                const trackMap = {
 | 
						|
                    RegExp: { [CALL]: true, [CONSTRUCT]: true }
 | 
						|
                };
 | 
						|
 | 
						|
                for (const { node: refNode } of tracker.iterateGlobalReferences(trackMap)) {
 | 
						|
                    const [patternNode, flagsNode] = refNode.arguments;
 | 
						|
 | 
						|
                    if (patternNode && patternNode.type === "SpreadElement") {
 | 
						|
                        continue;
 | 
						|
                    }
 | 
						|
                    const pattern = getStringIfConstant(patternNode, scope);
 | 
						|
                    const flags = getStringIfConstant(flagsNode, scope);
 | 
						|
 | 
						|
                    if (!flagsNode || (typeof flags === "string" && !flags.includes("u") && !flags.includes("v"))) {
 | 
						|
                        context.report({
 | 
						|
                            messageId: "requireUFlag",
 | 
						|
                            node: refNode,
 | 
						|
                            suggest: typeof pattern === "string" && isValidWithUnicodeFlag(context.languageOptions.ecmaVersion, pattern)
 | 
						|
                                ? [
 | 
						|
                                    {
 | 
						|
                                        fix(fixer) {
 | 
						|
                                            if (flagsNode) {
 | 
						|
                                                if ((flagsNode.type === "Literal" && typeof flagsNode.value === "string") || flagsNode.type === "TemplateLiteral") {
 | 
						|
                                                    const flagsNodeText = sourceCode.getText(flagsNode);
 | 
						|
 | 
						|
                                                    return fixer.replaceText(flagsNode, [
 | 
						|
                                                        flagsNodeText.slice(0, flagsNodeText.length - 1),
 | 
						|
                                                        flagsNodeText.slice(flagsNodeText.length - 1)
 | 
						|
                                                    ].join("u"));
 | 
						|
                                                }
 | 
						|
 | 
						|
                                                // We intentionally don't suggest concatenating + "u" to non-literals
 | 
						|
                                                return null;
 | 
						|
                                            }
 | 
						|
 | 
						|
                                            const penultimateToken = sourceCode.getLastToken(refNode, { skip: 1 }); // skip closing parenthesis
 | 
						|
 | 
						|
                                            return fixer.insertTextAfter(
 | 
						|
                                                penultimateToken,
 | 
						|
                                                astUtils.isCommaToken(penultimateToken)
 | 
						|
                                                    ? ' "u",'
 | 
						|
                                                    : ', "u"'
 | 
						|
                                            );
 | 
						|
                                        },
 | 
						|
                                        messageId: "addUFlag"
 | 
						|
                                    }
 | 
						|
                                ]
 | 
						|
                                : null
 | 
						|
                        });
 | 
						|
                    }
 | 
						|
                }
 | 
						|
            }
 | 
						|
        };
 | 
						|
    }
 | 
						|
};
 |