#45 working on CBO scanner

Author: jimbethancourt

change-proneness-ranker/src/main/java/org/hjug/git/GitLogReader.java

@@ -47,7 +47,7 @@ public File getGitDir(File basedir) {
     // and
     // https://github.com/centic9/jgit-cookbook/blob/master/src/main/java/org/dstadler/jgit/api/ReadFileFromCommit.java
     @Override
-    public Map<String, ByteArrayOutputStream> listRepositoryContentsAtHEAD(Repository repository) throws IOException {
+    public Map<String, ByteArrayOutputStream> (Repository repository) throws IOException {
         Ref head = repository.exactRef("HEAD");
         // a RevWalk allows us to walk over commits based on some filtering that is defined
         RevWalk walk = new RevWalk(repository);
@@ -56,7 +56,7 @@ public Map<String, ByteArrayOutputStream> listRepositoryContentsAtHEAD(Repositor
 
         TreeWalk treeWalk = new TreeWalk(repository);
         treeWalk.addTree(tree);
-        treeWalk.setRecursive(false);
+        treeWalk.setRecursive(false);listRepositoryContentsAtHEAD
 
         // TODO: extract rest of this method to test it
         Map<String, ByteArrayOutputStream> fileContentsCollection = new HashMap<>();

cost-benefit-calculator/src/main/java/org/hjug/cbc/CostBenefitCalculator.java

@@ -13,9 +13,7 @@
 import org.hjug.git.GitLogReader;
 import org.hjug.git.RepositoryLogReader;
 import org.hjug.git.ScmLogInfo;
-import org.hjug.metrics.GodClass;
-import org.hjug.metrics.GodClassRanker;
-import org.hjug.metrics.PMDGodClassRuleRunner;
+import org.hjug.metrics.*;
 
 @Slf4j
 public class CostBenefitCalculator {
@@ -31,7 +29,10 @@ public List<RankedDisharmony> calculateCostBenefitValues(String repositoryPath)
             log.error("Failure to access Git repository", e);
         }
 
-        List<GodClass> godClasses = getGodClasses(repositoryLogReader, repository);
+        Map<String, ByteArrayOutputStream> filesToScan = getFilesToScan(repositoryLogReader, repository);
+
+        List<GodClass> godClasses = getGodClasses(filesToScan);
+        List<CBOClass> cboClasses = getCBOClasses(filesToScan);
 
         List<ScmLogInfo> scmLogInfos = getRankedChangeProneness(repositoryLogReader, repository, godClasses);
 
@@ -67,17 +68,10 @@ List<ScmLogInfo> getRankedChangeProneness(
         return scmLogInfos;
     }
 
-    private List<GodClass> getGodClasses(RepositoryLogReader repositoryLogReader, Repository repository) {
-        Map<String, ByteArrayOutputStream> filesToScan = new HashMap<>();
-        log.info("Identifying God Classes from files in repository");
-        try {
-            filesToScan = repositoryLogReader.listRepositoryContentsAtHEAD(repository);
-        } catch (IOException e) {
-            log.error("Error reading Git repository contents", e);
-        }
-
+    private List<GodClass> getGodClasses(Map<String, ByteArrayOutputStream> filesToScan) {
         PMDGodClassRuleRunner ruleRunner = new PMDGodClassRuleRunner();
 
+        log.info("Identifying God Classes from files in repository");
         List<GodClass> godClasses = new ArrayList<>();
         for (Map.Entry<String, ByteArrayOutputStream> entry : filesToScan.entrySet()) {
             String filePath = entry.getKey();
@@ -92,4 +86,32 @@ private List<GodClass> getGodClasses(RepositoryLogReader repositoryLogReader, Re
         godClassRanker.rankGodClasses(godClasses);
         return godClasses;
     }
+
+    private List<CBOClass> getCBOClasses(Map<String, ByteArrayOutputStream> filesToScan) {
+
+        CBORuleRunner ruleRunner = new CBORuleRunner();
+        
+        log.info("Identifying highly coupled classes from files in repository");
+        List<CBOClass> cboClasses = new ArrayList<>();
+        for (Map.Entry<String, ByteArrayOutputStream> entry : filesToScan.entrySet()) {
+            String filePath = entry.getKey();
+            ByteArrayOutputStream value = entry.getValue();
+
+            ByteArrayInputStream inputStream = new ByteArrayInputStream(value.toByteArray());
+            Optional<CBOClass> godClassOptional = ruleRunner.runCBOClassRule(filePath, inputStream);
+            godClassOptional.ifPresent(cboClasses::add);
+        }
+
+        return cboClasses;
+    }
+
+    private Map<String, ByteArrayOutputStream> getFilesToScan(RepositoryLogReader repositoryLogReader, Repository repository) {
+        Map<String, ByteArrayOutputStream> filesToScan = new HashMap<>();
+        try {
+            filesToScan = repositoryLogReader.listRepositoryContentsAtHEAD(repository);
+        } catch (IOException e) {
+            log.error("Error reading Git repository contents", e);
+        }
+        return filesToScan;
+    }
 }

cost-benefit-calculator/src/main/java/org/hjug/cbc/RankedDisharmony.java

@@ -16,6 +16,8 @@ public class RankedDisharmony {
     private final Integer changePronenessRank;
     private final Integer priority;
 
+    private Integer couplingCount;
+
     private final Integer wmc;
     private final Integer wmcRank;
     private final Integer atfd;