Contents

require 'base64'
require 'octokit'

module Blakey
  module Source
    class Github < Base
      VULNERABILITY_ALERTS_PREVIEW_HEADER = 'application/vnd.github.dorian-preview+json'.freeze

      def initialize(access_token: nil, repo_path: nil)
        @access_token = (access_token || ENV['BLAKEY_SOURCE_GITHUB_ACCESS_TOKEN'])
        @octokit_client ||= ::Octokit::Client.new(access_token: @access_token)
        @repo_path = repo_path
      end

      attr_reader :repo_path

      def read_file(file_path)
        begin
          encoded_file_content = octokit_client.contents(repo_path, path: file_path).content
        rescue Octokit::NotFound => e
          raise FileNotFound.new(e)
        end

        Base64.decode64(encoded_file_content).chomp
      end

      def repository_overview
        {
          open_issues_count: repository.open_issues_count,
          open_pull_requests_count: open_pull_requests_count,
          language: repository.language,
          visibility: repository.private? ? 'private' : 'public',
          url: repository.html_url,
          updated_at: repository.updated_at,
          created_at: repository.created_at,
          last_pushed_at: repository.pushed_at,
          vulnerability_alerts_enabled: vulnerability_alerts_enabled?
        }
      end

      private

      def vulnerability_alerts_enabled?
        octokit_client.vulnerability_alerts_enabled?(repo_path, accept: VULNERABILITY_ALERTS_PREVIEW_HEADER)
      end

      def open_pull_requests_count
        search_query = "repo:#{repo_path} is:pr is:open"
        search_issues_response = octokit_client.search_issues(search_query, per_page: 1)
        search_issues_response.total_count
      end

      def repository
        @repository ||= octokit_client.repo(repo_path)
      end

      attr_reader :octokit_client
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
blakey-0.4.0	lib/blakey/source/github.rb