support module names that are in variables

index.js

@@ -14,19 +14,25 @@ function parseOptions(opts) {
 function findDependencies(source, opts) {
   opts = parseOptions(opts);
 
+  var potentialModuleNameVariable = {};
   var rootDeps = [];
   var modules = {};
 
   estraverse.traverse(esprima.parse(source, {sourceType: "module"}), {
     leave: function(node, parent) {
+      if (canBeModuleNameVariable(node)) {
+        potentialModuleNameVariable[node.id.name] = node.init.value;
+      }
+
       if (!isAngularModuleStatement(node)) {
         if (isNgModuleDeclaration(node)) {
           modules['ng'] = [];
         }
         return;
       }
 
-      var moduleName = parent.arguments[0].value;
+      var moduleNameArg = parent.arguments[0];
+      var moduleName = moduleNameArg.value || potentialModuleNameVariable[moduleNameArg.name];
       if (parent.arguments[1]) {
         // if already declared, will reset dependencies, like how angular behaves (latest declaration wins)
         modules[moduleName] = _.map(parent.arguments[1].elements, 'value');
@@ -64,6 +70,10 @@ function isNgModuleDeclaration(node) {
   return node.type === 'CallExpression' && node.callee.name === 'angularModule' && node.arguments.length > 0 && node.arguments[0].value === 'ng';
 }
 
+function canBeModuleNameVariable(node) {
+  return node.type === 'VariableDeclarator' && typeof node.init.value === 'string';
+}
+
 module.exports = findDependencies;
 module.exports.isAngularModuleStatement = isAngularModuleStatement;
 module.exports.isNgModuleDeclaration = isNgModuleDeclaration;

test/index_test.js

@@ -101,4 +101,16 @@ describe('lookup', function () {
     };
     lookup(source).should.eql(deps);
   });
+
+  it('should capture simple module declaration using a variable instead of a literal module name', function () {
+    var source = 'var moduleName = "test"; angular.module(moduleName, []);';
+    var deps = {
+      dependencies: ['ng'],
+      modules: {
+        'test': []
+      }
+    };
+    lookup(source).should.eql(deps);
+  });
+
 });