github_commit_graph.rb

module Plugins
  class GithubCommitGraph < Base
    def locals
      { username:, contributions:, stats: }
    end

    private

    def username = settings['username']

    def contributions
      @contributions ||= begin
        query = "query($userName:String!) {
          user(login: $userName){
            contributionsCollection {
              contributionCalendar {
                totalContributions
                weeks {
                  contributionDays {
                    contributionCount
                    date
                  }
                }
              }
            }
          }
        }"
        body = {
          query: query,
          variables: { userName: settings['username'] }
        }

        url = 'https://api.github.com/graphql'
        resp = HTTParty.post(url, body: body.to_json, headers: headers)
        data = resp.dig('data', 'user', 'contributionsCollection', 'contributionCalendar')

        {
          total: data['totalContributions'],
          commits: data['weeks']
        }
      end
    end

    def headers
      { 'authorization' => "Bearer #{Rails.application.credentials.plugins.github_commit_graph_token}" }
    end

    def stats
      days = contributions[:commits].flat_map { |week| week['contributionDays'] }
      sorted_days = days.sort_by { |day| Date.parse(day['date']) }

      {
        longest_streak: longest_streak(sorted_days),
        current_streak: current_streak(sorted_days),
        max_contributions: days.map { |day| day['contributionCount'] }.max,
        average_contributions: average_contributions(days)
      }
    end

    def average_contributions(days)
      total_contributions = days.sum { |day| day['contributionCount'] }
      (total_contributions.to_f / days.size).round(2)
    end

    def longest_streak(days)
      longest = current = 0
      days.each do |day|
        if (day['contributionCount']).positive?
          current += 1
          longest = [longest, current].max
        else
          current = 0
        end
      end
      longest
    end

    def current_streak(days)
      streak = 0
      
      # The current day can count towards the streak but it shouldn't break the streak
      streak += 1 if days.last[:contributionCount].positive?
      
      days[0..-2].reverse_each do |day|
        break if (day['contributionCount']).zero?

        streak += 1
      end
      streak
    end
  end
end