brew/Library/Homebrew/dev-cmd/contributions.rb

# typed: true
# frozen_string_literal: true

require "cli/parser"

module Homebrew
  extend T::Sig

  module_function

  SUPPORTED_REPOS = (
      %w[brew core cask] +
      OFFICIAL_CMD_TAPS.keys.map { |t| t.delete_prefix("homebrew/") } +
      OFFICIAL_CASK_TAPS
    ).freeze

  sig { returns(CLI::Parser) }
  def contributions_args
    Homebrew::CLI::Parser.new do
      usage_banner <<~EOS
        `contributions [email]`

        Contributions to Homebrew repos for a user.
      EOS

      flag "--from=",
           description: "Date (ISO-8601 format) to start searching contributions."

      flag "--to=",
           description: "Date (ISO-8601 format) to stop searching contributions."

      flag "--all",
           description: "Show contributions across all official Homebrew formula, cask and command repos."

      comma_array "--repos=",
                  description: "The Homebrew repositories to search for contributions in. " \
                               "Comma separated. Supported repos: #{SUPPORTED_REPOS.join(", ")}."

      conflicts "--all", "--repos"

      named_args :email, number: 1
    end
  end

  sig { returns(NilClass) }
  def contributions
    args = contributions_args.parse

    commits = 0
    coauthorships = 0

    repos = args[:repos] || SUPPORTED_REPOS
    repos.each do |repo|
      if !args[:all] && SUPPORTED_REPOS.exclude?(repo)
        return ofail "Unsupported repo: #{repo}. Try one of #{SUPPORTED_REPOS.join(", ")}."
      end

      repo_path = find_repo_path_for_repo(repo)
      if !repo_path.exist?
        next if repo == "versions" # This tap is deprecated, tapping it will error.

        opoo "Couldn't find repo #{repo} locally. Tapping it now..."
        Utils.safe_system("brew", "tap", repo_path) # TODO: Figure out why `exit code 1` happens here.
      end

      commits += git_log_author_cmd(T.must(repo_path), args)
      coauthorships += git_log_coauthor_cmd(T.must(repo_path), args)
    end

    sentence = "Person #{args.named.first} directly authored #{commits} commits " \
               "and co-authored #{coauthorships} commits " \
               "to #{args[:all] ? "all Homebrew repos" : repos.join(", ")}"
    sentence += if args[:from] && args[:to]
      " between #{args[:from]} and #{args[:to]}"
    elsif args[:from]
      " after #{args[:from]}"
    elsif args[:to]
      " before #{args[:to]}"
    else
      " in all time"
    end
    sentence += "."

    puts sentence
  end

  sig { params(repo: String).returns(Pathname) }
  def find_repo_path_for_repo(repo)
    return HOMEBREW_REPOSITORY if repo == "brew"

    Tap.fetch("homebrew", repo).path
  end

  sig { params(repo_path: Pathname, args: Homebrew::CLI::Args).returns(Integer) }
  def git_log_author_cmd(repo_path, args)
    cmd = ["git", "-C", repo_path, "log", "--oneline", "--author=#{args.named.first}"]
    cmd << "--before=#{args[:to]}" if args[:to]
    cmd << "--after=#{args[:from]}" if args[:from]

    Utils.safe_popen_read(*cmd).lines.count
  end

  sig { params(repo_path: Pathname, args: Homebrew::CLI::Args).returns(Integer) }
  def git_log_coauthor_cmd(repo_path, args)
    cmd = ["git", "-C", repo_path, "log", "--oneline"]
    cmd << "--format='%(trailers:key=Co-authored-by:)'"
    cmd << "--before=#{args[:to]}" if args[:to]
    cmd << "--after=#{args[:from]}" if args[:from]

    Utils.safe_popen_read(*cmd).lines.select { |l| l.include?(args.named.first) }.length
  end
end