fix: the cycle between modules is poorly depicted (#19)

heowc · Aug 4, 2024 · 816ebea · 816ebea
1 parent 9bfefa1
commit 816ebea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/heo-core/src/main/java/dev/heowc/heo/core/reporting/AnalysisReportService.java b/heo-core/src/main/java/dev/heowc/heo/core/reporting/AnalysisReportService.java
@@ -33,7 +33,10 @@ private static String createDotFormatted(DependencyAnalysisResult result) {
         exporter.setEdgeAttributeProvider(edge -> {
             final Map<String, Attribute> attributes = new LinkedHashMap<>();
             if (cycled.contains(edge.getTarget()) && cycled.contains(edge.getSource())) {
-                attributes.put("color", DefaultAttribute.createAttribute("red"));
+                final Set<String> targets = result.getCycleDetector().findCyclesContainingVertex(edge.getSource());
+                if (targets.contains(edge.getTarget())) {
+                    attributes.put("color", DefaultAttribute.createAttribute("red"));
+                }
             }
             return attributes;
         });