summaryrefslogtreecommitdiff
path: root/lib/license/management/python/pypi.rb
blob: 0397532fd75efa47f14028dd84345744a394d306 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# frozen_string_literal: true

require 'net/hippie'

module License
  module Management
    class PyPI
      include Loggable

      def initialize(http)
        @http = http
      end

      def definition_for(name, version)
        uri = "https://pypi.org/pypi/#{name}/#{version}/json"
        process(@http.with_retry { |client| client.get(uri) }).tap do |definition|
          log_info([name, version, definition["license"]].inspect)
        end
      rescue *Net::Hippie::CONNECTION_ERRORS
        {}
      end

      class << self
        def definition(name, version)
          @pypi ||= new(License::Management.http)
          @pypi.definition_for(name, version)
        end
      end

      private

      def process(response)
        return JSON.parse(response.body).fetch('info', {}) if ok?(response)

        log_error([response.class, response.code, response.body].inspect)
        {}
      end

      def ok?(response)
        response.is_a?(Net::HTTPSuccess)
      end
    end
  end
end