support class assigned to var

ng-annotate-main.js

@@ -771,7 +771,7 @@ function judgeInjectArraySuspect(path, ctx) {
     path = jumpOverIife(path);
     node = path.node;
 
-    if (t.isClass(node)){
+    if (t.isClass(node) || t.isClassExpression(node)){
         if (!node.id) {
             node.id = path.scope.generateUidIdentifier('ngInjectAnonymousClass');
         }

nginject.js

@@ -274,6 +274,11 @@ function inspectClassMethod(path, ctx){
       addSuspect(ancestor, ctx, !annotation);
       return;
     }
+
+    if (ancestor.isVariableDeclaration()) {
+      addSuspect(ancestor, ctx, !annotation);
+      return;
+    }
   }
 }
 

tests/es6.js

@@ -146,6 +146,27 @@ module.exports = {
     _ngInjectAnonymousClass.$inject = ["$timeout"];
     `
   },
+  {
+    name: "class assigned to a var",
+    explicit: true,
+    noES5: true,
+    input: `
+    let Foo = class Foo {
+      constructor($element) {
+        'ngInject';
+      }
+    };
+  `,
+    expected: `
+    let Foo = class Foo {
+      constructor($element) {
+        'ngInject';
+      }
+    };
+
+    Foo.$inject = ["$element"];
+  `
+  },
   {
     name: "annotated constructor",
     explicit: true,