Add method parameter types

Author: jimbethancourt

circular-reference-detector/src/main/java/com/ideacrest/parser/JavaProjectParser.java

@@ -3,6 +3,8 @@
 import com.github.javaparser.StaticJavaParser;
 import com.github.javaparser.ast.CompilationUnit;
 import com.github.javaparser.ast.body.FieldDeclaration;
+import com.github.javaparser.ast.body.MethodDeclaration;
+import com.github.javaparser.ast.body.Parameter;
 import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.IOException;
@@ -36,13 +38,14 @@ public Graph<String, DefaultEdge> getClassReferences(String srcDirectory) throws
                 filesStream
                         .filter(path -> path.getFileName().toString().endsWith(".java"))
                         .forEach(path -> {
-                            Set<String> instanceVarTypes = getInstanceVarTypes(classNames, path.toFile());
-                            if (!instanceVarTypes.isEmpty()) {
+                            Set<String> varTypes = getInstanceVarTypes(classNames, path.toFile());
+                            varTypes.addAll(getMethodTypes(classNames, path.toFile()));
+                            if (!varTypes.isEmpty()) {
                                 String className =
                                         getClassName(path.getFileName().toString());
                                 classReferencesGraph.addVertex(className);
-                                instanceVarTypes.forEach(classReferencesGraph::addVertex);
-                                instanceVarTypes.forEach(var -> classReferencesGraph.addEdge(className, var));
+                                varTypes.forEach(classReferencesGraph::addVertex);
+                                varTypes.forEach(var -> classReferencesGraph.addEdge(className, var));
                             }
                         });
             } catch (FileNotFoundException e) {
@@ -54,7 +57,7 @@ public Graph<String, DefaultEdge> getClassReferences(String srcDirectory) throws
 
     /**
      * Get instance variables types of a java source file using java parser
-     * @param classNamesToFilterBy - only add instance variables which have these class names as type
+     * @param classNamesToFilterBy - only add instance variable types which have these class names as type
      * @param file
      * @return
      */
@@ -74,6 +77,31 @@ private Set<String> getInstanceVarTypes(List<String> classNamesToFilterBy, File
         return new HashSet<>();
     }
 
+    /**
+     * Get parameter types of methods declared in a java source file using java parser
+     * @param classNamesToFilterBy - only add types which have these class names as type
+     * @param file
+     * @return
+     */
+    private Set<String> getMethodTypes(List<String> classNamesToFilterBy, File javaSrcFile) {
+        CompilationUnit compilationUnit;
+        try {
+            compilationUnit = StaticJavaParser.parse(javaSrcFile);
+            return compilationUnit.findAll(MethodDeclaration.class).stream()
+                    .flatMap(f -> f.getParameters().stream()
+                            .map(Parameter::getType)
+                            .filter(type -> !type.isPrimitiveType())
+                            .collect(Collectors.toSet())
+                            .stream())
+                    .map(Object::toString)
+                    .filter(classNamesToFilterBy::contains)
+                    .collect(Collectors.toSet());
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+        }
+        return new HashSet<>();
+    }
+
     /**
      * Get all java classes in a source directory
      *