172 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			172 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| 'use strict';
 | |
| 
 | |
| var acorn = require('acorn');
 | |
| var walk = require('acorn-walk');
 | |
| 
 | |
| function isScope(node) {
 | |
|   return node.type === 'FunctionExpression' || node.type === 'FunctionDeclaration' || node.type === 'ArrowFunctionExpression' || node.type === 'Program';
 | |
| }
 | |
| function isBlockScope(node) {
 | |
|   return node.type === 'BlockStatement' || isScope(node);
 | |
| }
 | |
| 
 | |
| function declaresArguments(node) {
 | |
|   return node.type === 'FunctionExpression' || node.type === 'FunctionDeclaration';
 | |
| }
 | |
| 
 | |
| function declaresThis(node) {
 | |
|   return node.type === 'FunctionExpression' || node.type === 'FunctionDeclaration';
 | |
| }
 | |
| 
 | |
| function reallyParse(source, options) {
 | |
|   var parseOptions = Object.assign({}, options,
 | |
|     {
 | |
|       allowReturnOutsideFunction: true,
 | |
|       allowImportExportEverywhere: true,
 | |
|       allowHashBang: true
 | |
|     }
 | |
|   );
 | |
|   return acorn.parse(source, parseOptions);
 | |
| }
 | |
| module.exports = findGlobals;
 | |
| module.exports.parse = reallyParse;
 | |
| function findGlobals(source, options) {
 | |
|   options = options || {};
 | |
|   var globals = [];
 | |
|   var ast;
 | |
|   // istanbul ignore else
 | |
|   if (typeof source === 'string') {
 | |
|     ast = reallyParse(source, options);
 | |
|   } else {
 | |
|     ast = source;
 | |
|   }
 | |
|   // istanbul ignore if
 | |
|   if (!(ast && typeof ast === 'object' && ast.type === 'Program')) {
 | |
|     throw new TypeError('Source must be either a string of JavaScript or an acorn AST');
 | |
|   }
 | |
|   var declareFunction = function (node) {
 | |
|     var fn = node;
 | |
|     fn.locals = fn.locals || {};
 | |
|     node.params.forEach(function (node) {
 | |
|       declarePattern(node, fn);
 | |
|     });
 | |
|     if (node.id) {
 | |
|       fn.locals[node.id.name] = true;
 | |
|     }
 | |
|   };
 | |
|   var declarePattern = function (node, parent) {
 | |
|     switch (node.type) {
 | |
|       case 'Identifier':
 | |
|         parent.locals[node.name] = true;
 | |
|         break;
 | |
|       case 'ObjectPattern':
 | |
|         node.properties.forEach(function (node) {
 | |
|           declarePattern(node.value || node.argument, parent);
 | |
|         });
 | |
|         break;
 | |
|       case 'ArrayPattern':
 | |
|         node.elements.forEach(function (node) {
 | |
|           if (node) declarePattern(node, parent);
 | |
|         });
 | |
|         break;
 | |
|       case 'RestElement':
 | |
|         declarePattern(node.argument, parent);
 | |
|         break;
 | |
|       case 'AssignmentPattern':
 | |
|         declarePattern(node.left, parent);
 | |
|         break;
 | |
|       // istanbul ignore next
 | |
|       default:
 | |
|         throw new Error('Unrecognized pattern type: ' + node.type);
 | |
|     }
 | |
|   };
 | |
|   var declareModuleSpecifier = function (node, parents) {
 | |
|     ast.locals = ast.locals || {};
 | |
|     ast.locals[node.local.name] = true;
 | |
|   };
 | |
|   walk.ancestor(ast, {
 | |
|     'VariableDeclaration': function (node, parents) {
 | |
|       var parent = null;
 | |
|       for (var i = parents.length - 1; i >= 0 && parent === null; i--) {
 | |
|         if (node.kind === 'var' ? isScope(parents[i]) : isBlockScope(parents[i])) {
 | |
|           parent = parents[i];
 | |
|         }
 | |
|       }
 | |
|       parent.locals = parent.locals || {};
 | |
|       node.declarations.forEach(function (declaration) {
 | |
|         declarePattern(declaration.id, parent);
 | |
|       });
 | |
|     },
 | |
|     'FunctionDeclaration': function (node, parents) {
 | |
|       var parent = null;
 | |
|       for (var i = parents.length - 2; i >= 0 && parent === null; i--) {
 | |
|         if (isScope(parents[i])) {
 | |
|           parent = parents[i];
 | |
|         }
 | |
|       }
 | |
|       parent.locals = parent.locals || {};
 | |
|       if (node.id) {
 | |
|         parent.locals[node.id.name] = true;
 | |
|       }
 | |
|       declareFunction(node);
 | |
|     },
 | |
|     'Function': declareFunction,
 | |
|     'ClassDeclaration': function (node, parents) {
 | |
|       var parent = null;
 | |
|       for (var i = parents.length - 2; i >= 0 && parent === null; i--) {
 | |
|         if (isBlockScope(parents[i])) {
 | |
|           parent = parents[i];
 | |
|         }
 | |
|       }
 | |
|       parent.locals = parent.locals || {};
 | |
|       if (node.id) {
 | |
|         parent.locals[node.id.name] = true;
 | |
|       }
 | |
|     },
 | |
|     'TryStatement': function (node) {
 | |
|       if (node.handler === null) return;
 | |
|       node.handler.locals = node.handler.locals || {};
 | |
|       node.handler.locals[node.handler.param.name] = true;
 | |
|     },
 | |
|     'ImportDefaultSpecifier': declareModuleSpecifier,
 | |
|     'ImportSpecifier': declareModuleSpecifier,
 | |
|     'ImportNamespaceSpecifier': declareModuleSpecifier
 | |
|   });
 | |
|   function identifier(node, parents) {
 | |
|     var name = node.name;
 | |
|     if (name === 'undefined') return;
 | |
|     for (var i = 0; i < parents.length; i++) {
 | |
|       if (name === 'arguments' && declaresArguments(parents[i])) {
 | |
|         return;
 | |
|       }
 | |
|       if (parents[i].locals && name in parents[i].locals) {
 | |
|         return;
 | |
|       }
 | |
|     }
 | |
|     node.parents = parents.slice();
 | |
|     globals.push(node);
 | |
|   }
 | |
|   walk.ancestor(ast, {
 | |
|     'VariablePattern': identifier,
 | |
|     'Identifier': identifier,
 | |
|     'ThisExpression': function (node, parents) {
 | |
|       for (var i = 0; i < parents.length; i++) {
 | |
|         if (declaresThis(parents[i])) {
 | |
|           return;
 | |
|         }
 | |
|       }
 | |
|       node.parents = parents.slice();
 | |
|       globals.push(node);
 | |
|     }
 | |
|   });
 | |
|   var groupedGlobals = {};
 | |
|   globals.forEach(function (node) {
 | |
|     var name = node.type === 'ThisExpression' ? 'this' : node.name;
 | |
|     groupedGlobals[name] = (groupedGlobals[name] || []);
 | |
|     groupedGlobals[name].push(node);
 | |
|   });
 | |
|   return Object.keys(groupedGlobals).sort().map(function (name) {
 | |
|     return {name: name, nodes: groupedGlobals[name]};
 | |
|   });
 | |
| }
 |