12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879 |
- /*
- MIT License http://www.opensource.org/licenses/mit-license.php
- Author Tobias Koppers @sokra
- */
- "use strict";
- /** @typedef {import("eslint-scope").Reference} Reference */
- /** @typedef {import("eslint-scope").Variable} Variable */
- /** @typedef {import("../javascript/JavascriptParser").AnyNode} AnyNode */
- /** @typedef {import("../javascript/JavascriptParser").Program} Program */
- /**
- * @param {Variable} variable variable
- * @returns {Reference[]} references
- */
- const getAllReferences = variable => {
- let set = variable.references;
- // Look for inner scope variables too (like in class Foo { t() { Foo } })
- const identifiers = new Set(variable.identifiers);
- for (const scope of variable.scope.childScopes) {
- for (const innerVar of scope.variables) {
- if (innerVar.identifiers.some(id => identifiers.has(id))) {
- set = set.concat(innerVar.references);
- break;
- }
- }
- }
- return set;
- };
- /**
- * @param {Program | Program[]} ast ast
- * @param {AnyNode} node node
- * @returns {undefined | AnyNode[]} result
- */
- const getPathInAst = (ast, node) => {
- if (ast === node) {
- return [];
- }
- const nr = node.range;
- const enterNode = n => {
- if (!n) return undefined;
- const r = n.range;
- if (r) {
- if (r[0] <= nr[0] && r[1] >= nr[1]) {
- const path = getPathInAst(n, node);
- if (path) {
- path.push(n);
- return path;
- }
- }
- }
- return undefined;
- };
- if (Array.isArray(ast)) {
- for (let i = 0; i < ast.length; i++) {
- const enterResult = enterNode(ast[i]);
- if (enterResult !== undefined) return enterResult;
- }
- } else if (ast && typeof ast === "object") {
- const keys = Object.keys(ast);
- for (let i = 0; i < keys.length; i++) {
- const value = ast[keys[i]];
- if (Array.isArray(value)) {
- const pathResult = getPathInAst(value, node);
- if (pathResult !== undefined) return pathResult;
- } else if (value && typeof value === "object") {
- const enterResult = enterNode(value);
- if (enterResult !== undefined) return enterResult;
- }
- }
- }
- };
- module.exports = { getAllReferences, getPathInAst };
|