Better Description for Stats and added "Used Directly By"

square · Sep 30, 2024 · 8f76b28 · 8f76b28
1 parent 10d52aa
commit 8f76b28
Show file tree

Hide file tree

Showing 3 changed files with 236 additions and 205 deletions.
diff --git a/invert-report/src/jsMain/kotlin/com/squareup/invert/common/ReportDataRepo.kt b/invert-report/src/jsMain/kotlin/com/squareup/invert/common/ReportDataRepo.kt
@@ -203,6 +203,17 @@ class ReportDataRepo(
       }
     }
 
+
+  fun moduleDirectlyUsedBy(path: ModulePath): Flow<Map<ModulePath, List<ConfigurationName>>?> =
+    collectedDataRepo.directDependenciesData.mapLatest { directDependenciesData ->
+      val directDependencies = directDependenciesData?.directDependencies
+      directDependencies?.entries?.filter { (modulePath, configurationToDependencyIds) ->
+        configurationToDependencyIds.any { it.value.contains(path) }
+      }?.associate { (modulePath, configurationToDependencyIds) ->
+        modulePath to configurationToDependencyIds.keys.toList()
+      }
+    }
+
   fun moduleTransitivelyUsedBy(path: ModulePath): Flow<Map<ModulePath, List<ConfigurationName>>?> =
     allInvertedDependencies.mapLatest { allInvertedDependenciesMap: Map<DependencyId, Map<ModulePath, List<ConfigurationName>>>? ->
       if (allInvertedDependenciesMap != null) {
@@ -214,8 +225,8 @@ class ReportDataRepo(
 
 
   fun directDependenciesOf(modulePath: ModulePath?): Flow<Map<ConfigurationName, Set<DependencyId>>?> =
-    collectedDataRepo.directDependenciesData.mapLatest { directDependenciesData ->
-      directDependenciesData?.directDependencies?.get(modulePath)
+    allDirectDependencies.mapLatest { directDependenciesData ->
+      directDependenciesData?.get(modulePath)
     }
 
   fun statsForKey(statKey: StatKey): Flow<MutableList<ModuleOwnerAndCodeReference>> =