Check for class expressions in inspectDeclarator and inspectClassMethod

schmod · Nov 20, 2019 · 373e292 · 373e292
1 parent 70152ff
commit 373e292
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/nginject.js b/nginject.js
@@ -227,7 +227,7 @@ function inspectAssignment(path, ctx){
 
 function inspectDeclarator(path, ctx){
   const node = path.node;
-  if(!isFunctionExpressionOrArrow(node.init)){
+  if(!isFunctionExpressionOrArrow(node.init) && !t.isClassExpression(node.init)){
     return;
   }
 
@@ -270,10 +270,17 @@ function inspectClassMethod(path, ctx){
   const ancestry = path.getAncestry();
   for(var i=0; i < ancestry.length; i++){
     let ancestor = ancestry[i];
+
     if(ancestor.isClassDeclaration()){
       addSuspect(ancestor, ctx, !annotation);
       return;
     }
+
+	if (ancestor.isClassExpression()) {
+		// Add the variable declaration of the class as suspect
+		addSuspect(ancestor.parentPath.parentPath, ctx, !annotation);
+		return;
+	}
   }
 }