# frozen_string_literal: true

$LOAD_PATH << 'lib'
require 'imagen'
require 'rainbow'
require 'bigdecimal'
require 'forwardable'

require 'undercover/lcov_parser'
require 'undercover/result'
require 'undercover/cli'
require 'undercover/changeset'
require 'undercover/formatter'
require 'undercover/options'

module Undercover
  class Report
    extend Forwardable
    def_delegators :changeset, :validate

    attr_reader :changeset,
                :lcov,
                :results

    # TODO: pass merge base as cli argument
    # add dependecy on "options" for all opts (dirs, git_dir, etc)
    def initialize(lcov_report_path, code_dir, git_dir: '.git', compare: nil)
      @lcov = LcovParser.parse(File.open(lcov_report_path))
      @changeset = Changeset.new(File.join(code_dir, git_dir), compare).update
      @results = Hash.new { |hsh, key| hsh[key] = [] }
    end

    def build
      each_result_arg do |filename, coverage, imagen_node|
        results[filename] << Result.new(imagen_node, coverage, filename)
      end
      self
    end

    # TODO: this is experimental and might be incorrect!
    # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
    def build_warnings
      flagged_results = Set.new
      changeset.each_changed_line do |filepath, line_no|
        dist_from_line_no = lambda do |res|
          return BigDecimal::INFINITY if line_no < res.first_line
          line_no - res.first_line
        end
        dist_from_line_no_sorter = lambda do |res1, res2|
          dist_from_line_no[res1] <=> dist_from_line_no[res2]
        end

        res = results[filepath].min(&dist_from_line_no_sorter)
        flagged_results << res if res&.uncovered?(line_no)
      end
      flagged_results
    end
    # rubocop:enable Metrics/MethodLength, Metrics/AbcSize

    def all_results
      results.values.flatten
    end

    def inspect
      "#<Undercover::Report:#{object_id} results: #{results.size}>"
    end
    alias to_s inspect

    private

    # TODO: some of this could be moved to the imagen gem
    # TODO: should that start from changeset.file_paths?
    # this way we could report things that weren't even loaded in any spec,
    # so is this still good idea? (Rakefile, .gemspec etc)
    def each_result_arg
      match_all = ->(_) { true }
      lcov.source_files.each do |filename, coverage|
        ast = Imagen::Node::Root.new
        Imagen::Visitor.traverse(Parser::CurrentRuby.parse_file(filename), ast)
        ast.children[0].find_all(match_all).each do |node|
          yield(filename, coverage, node)
        end
      end
    end
  end
end