Allow specifying what types of violations to output on get_info

lib/packs/cli.rb

@@ -130,11 +130,17 @@ def update
 
     desc 'get_info [ packs/my_pack packs/my_other_pack ]', 'Get info about size and violations for packs'
     option :format, type: :string, default: 'detail', aliases: :f, banner: 'Specify the output format (detail, csv)'
+    option :types, type: :string, default: 'privacy,dependency', aliases: :t, banner: 'List of validation types to include (privacy,dependency,architecture)'
     sig { params(pack_names: String).void }
     def get_info(*pack_names)
+      selected_types = options[:types].to_s.downcase.split(',')
+      invalid_types = selected_types - POSIBLE_TYPES
+      raise StandardError, "Invalid type(s): #{invalid_types.join(', ')}. Possible types are: #{POSIBLE_TYPES.join(', ')}" unless invalid_types.empty?
+
       Private.get_info(
         packs: parse_pack_names(pack_names),
-        format: options[:format].to_sym
+        format: options[:format].to_sym,
+        types: selected_types.map(&:to_sym)
       )
       exit_successfully
     end

lib/packs/private.rb

@@ -482,19 +482,19 @@ def self.packwerk_package_to_pack(package)
     sig do
       params(
         packs: T::Array[Packs::Pack],
-        format: Symbol
+        format: Symbol,
+        types: T::Array[Symbol]
       ).void
     end
-    def self.get_info(packs: Packs.all, format: :detail)
+    def self.get_info(packs: Packs.all, format: :detail, types: %i[privacy dependency architecture])
       require 'csv' if format == :csv
 
-      violation_types = %i[privacy dependency architecture]
       violations = {
         inbound: {},
         outbound: {}
       }
       directions = violations.keys
-      dir_x_types = directions.product(violation_types)
+      dir_x_types = directions.product(types)
 
       ParsePackwerk.all.each do |p|
         p.violations.each do |violation|

lib/packs/private/interactive_cli/use_cases/get_info.rb

@@ -28,9 +28,18 @@ def perform!(prompt)
 
             format = prompt.select('What output format do you want?', %w[Detail CSV])
 
+            types = prompt.multi_select(
+              'What violation types do you want stats for?',
+              %w[Privacy Dependency Architecture]
+            )
+
             puts "You've selected #{selected_packs.count} packs. Wow! Here's all the info."
 
-            Private.get_info(packs: selected_packs, format: format.downcase.to_sym)
+            Private.get_info(
+              packs: selected_packs,
+              format: format.downcase.to_sym,
+              types: types.map(&:downcase).map(&:to_sym)
+            )
           end
         end
       end