oss-stats 0.0.1

data/lib/oss_stats/github_client.rb

@@ -0,0 +1,55 @@
+require 'net/http'
+require 'json'
+require 'uri'
+
+# In general, we use octokit, but there are a few places were
+# we cannot, and this handles those cases
+module OssStats
+  class GitHubClient
+    attr_reader :token
+
+    def initialize(token)
+      @token = token
+      @endpoint = 'https://api.github.com'
+    end
+
+    def pr_statuses(pr)
+      url = pr['statuses_url']
+      return [] unless url
+
+      uri = URI(url)
+      get(uri.path)
+    end
+
+    def recent_prs(org, repo, n = 10)
+      prs_path = "/repos/#{org}/#{repo}/pulls"
+      pr_query_params = {
+        state: 'open', sort: 'updated', direction: 'desc', per_page: n
+      }
+      pr_uri = URI(prs_path)
+      pr_uri.query = URI.encode_www_form(pr_query_params)
+      get(pr_uri.path)
+    rescue StandardError => e
+      log.error("Error fetching PRs for #{repo_url}: #{e.message}")
+      []
+    end
+
+    def get(path)
+      log.trace("github_api_get: Attempting to parse URI with path: '#{path}'")
+      uri = URI("#{@endpoint}#{path}")
+      req = Net::HTTP::Get.new(uri)
+      req['Authorization'] = "Bearer #{@token}"
+      req['Accept'] = 'application/vnd.github+json'
+      req['User-Agent'] = 'private-pipeline-checker'
+
+      res = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
+        http.request(req)
+      end
+      unless res.is_a?(Net::HTTPSuccess)
+        raise "GitHub API error: #{res.code} #{res.body}"
+      end
+
+      JSON.parse(res.body)
+    end
+  end
+end

data/lib/oss_stats/github_token.rb

@@ -0,0 +1,23 @@
+# looks for :github_token in `options`, falling back to
+# $GITHUB_TOKEN, and then gh's auth.
+def get_github_token(options)
+  return options[:github_token] if options[:github_token]
+  return ENV['GITHUB_TOKEN'] if ENV['GITHUB_TOKEN']
+
+  config_path = File.expand_path('~/.config/gh/hosts.yml')
+  if File.exist?(config_path)
+    config = YAML.load_file(config_path)
+    return config.dig('github.com', 'oauth_token')
+  end
+  nil
+end
+
+def get_github_token!(options)
+  token = get_github_token(options)
+  unless token
+    raise ArgumentError,
+      'GitHub token is missing. Please provide a token using ' +
+      '--github-token, or set $GITHUB_TOKEN, or run `gh auth login`'
+  end
+  token
+end

data/lib/oss_stats/log.rb

@@ -0,0 +1,25 @@
+require 'mixlib/log'
+
+module Mixlib
+  module Log
+    class Formatter
+      def call(severity, _time, _progname, msg)
+        if severity == 'INFO'
+          "#{msg2str(msg)}\n"
+        else
+          "#{severity}: #{msg2str(msg)}\n"
+        end
+      end
+    end
+  end
+end
+
+module OssStats
+  class Log
+    extend Mixlib::Log
+  end
+end
+
+def log
+  OssStats::Log
+end