stats.rb

# frozen_string_literal: true

module Msf
  module Modules
    module Metadata
      module Stats
        attr_reader :module_counts

        def update_stats
          @metadata = get_metadata
          map_types_to_metadata!

          @module_counts = {
            exploit: @stats_by_type['exploit'].size,
            auxiliary: @stats_by_type['auxiliary'].size,
            post: @stats_by_type['post'].size,
            payload: @stats_by_type['payload'].size,
            encoder: @stats_by_type['encoder'].size,
            nop: @stats_by_type['nop'].size,
            evasion: @stats_by_type['evasion'].size,
            total: @metadata.size
          }
        end

        private

        def map_types_to_metadata!
          @stats_by_type = Hash.new { |h, k| h[k] = [] }

          @metadata.each do |module_metadata|
            @stats_by_type[module_metadata.type] << module_metadata
          end
        end
      end
    end
  end
end