Make use of getContainer

github · May 4, 2020 · dcd81b5 · dcd81b5
1 parent d90fca3
commit dcd81b5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/queries/undeclared-action-input.ql b/queries/undeclared-action-input.ql
@@ -36,14 +36,16 @@ class ActionDeclaration extends File {
 }
 
 Expr getAFunctionChildExpr(Function f) {
-  result = f.getBody().getAChildStmt*().getAChildExpr*()
+  result.getContainer() = f
 }
 
 /*
  * Result is a function that is called from the body of the given function `f`
  */
 Function calledBy(Function f) {
   result = getAFunctionChildExpr(f).(InvokeExpr).getResolvedCallee()
+  or
+  result.getEnclosingContainer() = f // assume outer function causes inner function to be called
 }
 
 class GetInputMethodCallExpr extends MethodCallExpr {