visualize_packs 0.5.12 → 0.5.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8017d316d5d179efe32200f2152bb0abb1c8a3b3299ba8f777a604c07165660e
-  data.tar.gz: 5f306bd08bdc5cb70eee69ea23a9233ba03ad8d13bd7c6c75c92a39a18f3f1f6
+  metadata.gz: fd8d04886b359c5fe0ee721ba30aa63a82cf15cbbd49f4f22aae9bb122d746da
+  data.tar.gz: f3e546e753e571226f782fe8aa2ae9cbee845c3629964cd8eedaf1527fc77f90
 SHA512:
-  metadata.gz: e9c0773394b6f17466e75e00c825dfcca7a1d359922988b257a08e59fb1ce76e2c373cfe968783d87f3ec8da5b9f500d53d64dcfc74946a3ac804376e22bdeb3
-  data.tar.gz: aff8ae19a21ca89a91331c2c2679fc64e15c5151e5f59a72a7dc35b7d5df72680e0627d6462ec6b5cc519098263210c60994445a563fc6aa95149be7a4d4c684
+  metadata.gz: 23b4d5aeb2ac339bb16a731d66733b47dd4d3583adbd099064a3bcd2c2cd77d56ded62cd64128cf7e992108227f91fa32667622e461e42ed49dcee60cf16fc7d
+  data.tar.gz: 3c07f47b89313c37d062550289ca28591a0d2a2023c7fb3f23f401a27710cad28eff6df42775108429af7193374088d0ad9c690599c64c2ad8b0352e921c076c

data/lib/visualize_packs.rb

@@ -12,13 +12,12 @@ module VisualizePacks
   sig { params(options: Options, raw_config: T::Hash[String, T.untyped], packages: T::Array[ParsePackwerk::Package]).returns(String) }
   def self.package_graph!(options, raw_config, packages)
     all_packages = filtered(packages, options).compact.sort_by {|x| x.name }
-    all_package_names = all_packages.map &:name
-
     all_packages = remove_nested_packs(all_packages, options)
+    all_package_names = all_packages.map &:name
 
     show_edge = show_edge_builder(options, all_package_names)
     node_color = node_color_builder()
-    max_todo_count = max_todo_count(all_packages, show_edge)
+    max_todo_count = max_todo_count(all_packages, show_edge, options)
 
     title = diagram_title(options, max_todo_count)
 
@@ -130,17 +129,21 @@ module VisualizePacks
     end
   end
 
-  sig { params(all_packages: T::Array[ParsePackwerk::Package], show_edge: T.proc.params(arg0: String, arg1: String).returns(T::Boolean)).returns(T.nilable(Integer)) }
-  def self.max_todo_count(all_packages, show_edge)
+  sig { params(all_packages: T::Array[ParsePackwerk::Package], show_edge: T.proc.params(arg0: String, arg1: String).returns(T::Boolean), options: Options).returns(T.nilable(Integer)) }
+  def self.max_todo_count(all_packages, show_edge, options)
     todo_counts = {}
-    all_packages.each do |package|
-      todos_by_package = package.violations&.group_by(&:to_package_name)
-      todos_by_package&.keys&.each do |todos_to_package|
-        todo_types = todos_by_package&& todos_by_package[todos_to_package]&.group_by(&:type)
-        todo_types&.keys&.each do |todo_type|
-          if show_edge.call(package.name, todos_to_package)
-            key = "#{package.name}->#{todos_to_package}:#{todo_type}"
-            todo_counts[key] = todo_types && todo_types[todo_type]&.count
+    if options.show_todos
+      all_packages.each do |package|
+        todos_by_package = package.violations&.group_by(&:to_package_name)
+        todos_by_package&.keys&.each do |todos_to_package|
+          todo_types = todos_by_package&& todos_by_package[todos_to_package]&.group_by(&:type)
+          todo_types&.keys&.each do |todo_type|
+            if options.only_todo_types.empty? || options.only_todo_types.include?(todo_type)
+              if show_edge.call(package.name, todos_to_package)
+                key = "#{package.name}->#{todos_to_package}:#{todo_type}"
+                todo_counts[key] = todo_types && todo_types[todo_type]&.count
+              end
+            end
           end
         end
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: visualize_packs
 version: !ruby/object:Gem::Version
-  version: 0.5.12
+  version: 0.5.13
 platform: ruby
 authors:
 - Gusto Engineers
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-09-02 00:00:00.000000000 Z
+date: 2023-09-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler