require "awesome_print"

module Nicht
  class Stats
    def initialize(path, search = nil)
      @path = "#{path}/**/Gemfile"
      @search = search
      @stats = { per_project: {}, per_gem: {} }
    end

    def render
      Dir[@path].each do |path|
        scan_project path
      end
      output
    end

    private

    def scan_record(line, project_name)
      candidate = line.split(',').shift.split ' '
      return unless candidate.first == 'gem'
      gem_name = candidate.last
      return if %w[* #].include?(gem_name[0])
      add_stat(gem_name, project_name)
    end

    def add_stat(gem_name, project_name)
      gem_name.delete!('\'')
      gem_name.delete!('"')
      @stats[:per_project][project_name] << gem_name
      @stats[:per_gem][gem_name] ||= []
      @stats[:per_gem][gem_name] << project_name
    end

    def scan_project(path)
      project_name = path.split('/')
      project_name = project_name[project_name.size - 2]
      File.open(path, 'r') do |f|
        @stats[:per_project][project_name] = []
        f.each_line do |line|
          scan_record(line, project_name)
        end
      end
    end

    def output
      @search.nil? ? dump_per_project : dump_per_gem
    end

    def dump_per_project
      @stats[:per_project].each_key do |key|
        @stats[:per_project][key].sort!
      end
      ap @stats[:per_project]
    end

    def dump_per_gem
      @stats[:per_gem] = @stats[:per_gem].sort.to_h
      dump_gems = @stats[:per_gem][@search] ? @stats[:per_gem][@search] : @stats[:per_gem].select { |k, v| k.include? @search }
      if dump_gems.empty?
        puts "No matches found for '#{@search}'"
      else
        ap dump_gems
      end
    end
  end
end