pr_work_done.rb

#!/usr/bin/env ruby
# frozen_string_literal: true

# This script for every week will calculate:
# the number of closed prs
# the number of merged prs
# the number of comments made on prs

require 'optparse'
require 'csv'
require_relative 'octokit_utils'

options = {}
options[:oauth] = ENV['GITHUB_COMMUNITY_TOKEN'] if ENV['GITHUB_COMMUNITY_TOKEN']
parser = OptionParser.new do |opts|
  opts.banner = 'Usage: pr_work_done.rb [options]'
  opts.on('-f', '--file NAME', String, 'Module file list') { |v| options[:file] = v }
  opts.on('-t', '--oauth-token TOKEN', 'OAuth token. Required.') { |v| options[:oauth] = v }
end

parser.parse!

options[:file] = 'modules.json' if options[:file].nil?

missing = []
missing << '-t' if options[:oauth].nil?
unless missing.empty?
  puts "Missing options: #{missing.join(', ')}"
  puts parser
  exit
end

util = OctokitUtils.new(options[:oauth])
parsed = util.load_module_list(options[:file])

number_of_weeks_to_show = 10

def date_of_next(day)
  date  = Date.parse(day)
  delta = date > Date.today ? 0 : 7
  date + delta
end

# find next wedenesday, set boundries
right_bound = date_of_next 'Wednesday'
left_bound = right_bound - 7
since = (right_bound - (number_of_weeks_to_show * 7))

all_merged_pulls = []
all_closed_pulls = []
comments = []
members_of_organisation = {}

# gather all commments / merges / closed in our time range (since)
parsed.each do |m|
  closed_pr_information_cache = util.fetch_async("#{m['github_namespace']}/#{m['repo_name']}", { state: 'closed' }, [:issue_comments], attribute: 'closed_at', date: since)
  # closed prs
  all_closed_pulls.concat(util.fetch_unmerged_pull_requests(closed_pr_information_cache))
  # merged prs
  all_merged_pulls.concat(util.fetch_merged_pull_requests(closed_pr_information_cache))
  # find organisation members, if we havent already
  if members_of_organisation.empty?
    members_of_organisation = util.puppet_organisation_members(all_merged_pulls) unless all_merged_pulls.size.zero?
  end
  # all comments made by organisation members
  closed_pr_information_cache.each do |pull|
    next if pull[:issue_comments].empty?

    pull[:issue_comments].each do |comment|
      comments.push(comment) if members_of_organisation.key?(comment.user.login)
    end
  end

  puts "repo #{m}"
end

week_data = []
# for gathered comments / merges / closed, which week does it belong to
(0..number_of_weeks_to_show - 1).each do |_week_number|
  # find closed
  closed = 0
  all_closed_pulls.each do |pull|
    closed += 1 if (pull[:closed_at] < right_bound.to_time) && (pull[:closed_at] > left_bound.to_time)
  end
  # find merged
  merged = 0
  all_merged_pulls.each do |pull|
    merged += 1 if (pull[:closed_at] < right_bound.to_time) && (pull[:closed_at] > left_bound.to_time)
  end
  # find commments from puppet
  comment_count = 0
  comments.each do |iter|
    comment_count += 1 if (iter[:created_at] < right_bound.to_time) && (iter[:created_at] > left_bound.to_time)
  end

  row = { 'week ending on' => right_bound, 'closed' => closed, 'commented' => comment_count, 'merged' => merged }
  week_data.push(row)
  # move boundries
  right_bound = left_bound
  left_bound = right_bound - 7
end

# reverse week_data to give it in chronological order
week_data = week_data.reverse

CSV.open('pr_work_done.csv', 'wb') do |csv|
  csv << ['week ending on', 'closed', 'commented', 'merged']
  week_data.each do |week|
    csv << [week['week ending on'], week['closed'], week['commented'], week['merged']]
  end
end