#!/usr/bin/env ruby

# frozen_string_literal: true

require 'undercover'

module Undercover
  class Changeset
    # Rugged merge_base complains when graft/shallow
    # (https://github.com/libgit2/rugged/issues/846)
    #
    # So we assume we provide the merge-base ourself. Modified from
    # https://github.com/grodowski/undercover/blob/32e62f66682ee566032b5970437ed2934ef29701/lib/undercover/changeset.rb#L74-L78
    def compare_base_obj
      return unless compare_base

      repo.lookup(compare_base.to_s)
    end
  end

  class Result
    # Undercover does not show branch undercoverage when there no `DA` line
    # coverage in the lcov file.
    # https://github.com/grodowski/undercover/issues/167#issuecomment-3737218228
    #
    # Patch it to show branch coverage even when it's `hits: n/a`.
    # rubocop:disable Metrics/AbcSize -- Based on upstream code, with minimal modifications
    # rubocop:disable Style/NumericPredicate -- Based on upstream code, with minimal modifications
    def pretty_print
      pad = node.last_line.to_s.length
      pretty_print_lines.map do |covered, (num, line)|
        formatted_line = "#{num.to_s.rjust(pad)}: #{line}"
        if line.strip.empty?
          Rainbow(formatted_line).darkgray.dark
        elsif skipped?(file_path, num)
          Rainbow(formatted_line).darkgray.dark +
            Rainbow(' skipped with :nocov:').italic.darkgray.dark
        elsif covered.nil?
          Rainbow(formatted_line).darkgray.dark +
            Rainbow(' hits: n/a').italic.darkgray.dark +
            count_covered_branches(num)
        elsif covered.positive?
          Rainbow(formatted_line).green +
            Rainbow(" hits: #{covered}").italic.darkgray.dark +
            count_covered_branches(num)
        elsif covered.zero?
          Rainbow(formatted_line).red +
            Rainbow(" hits: #{covered}").italic.darkgray.dark +
            count_covered_branches(num)
        end
      end.join("\n")
    end
    # rubocop:enable Style/NumericPredicate
    # rubocop:enable Metrics/AbcSize
  end
end

compare_base = ARGV[0]
compare_base ||= IO.popen(%w[git merge-base origin/master HEAD]) { |p| p.read.chomp }
coverage_file_path = 'coverage/lcov/gitlab.lcov'

# By default undercover includes Ruby related files and only excludes common paths
# like test, spec, *_spec.rb
exclude_file_globs = Undercover::Options::DEFAULT_FILE_EXCLUDE_GLOBS.dup

# We need to exclude more folders:
# EE/JH counterparts for default globs
extension_globs = Undercover::Options::DEFAULT_FILE_EXCLUDE_GLOBS.flat_map do |glob|
  %W[jh/#{glob} ee/#{glob}] unless glob.start_with?('*')
end.compact

exclude_file_globs.concat extension_globs

# These have own specs
exclude_file_globs.concat %w[qa/* gems/* vendor/*]

result = if File.exist?(coverage_file_path)
           Undercover::CLI.run(%W[-c #{compare_base} --exclude-files #{exclude_file_globs.join(',')}])
         else
           warn "#{coverage_file_path} doesn't exist"
           0
         end

exit result