connection.rb 1.87 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
module Bitbucket
  class Connection
    DEFAULT_API_VERSION = '2.0'
    DEFAULT_BASE_URI    = 'https://api.bitbucket.org/'
    DEFAULT_QUERY       = {}

    def initialize(options = {})
      @api_version = options.fetch(:api_version, DEFAULT_API_VERSION)
      @base_uri    = options.fetch(:base_uri, DEFAULT_BASE_URI)
      @query       = options.fetch(:query, DEFAULT_QUERY)

      @token         = options.fetch(:token)
      @expires_at    = options.fetch(:expires_at)
      @expires_in    = options.fetch(:expires_in)
      @refresh_token = options.fetch(:refresh_token)
Stan Hu's avatar
Stan Hu committed
16
    end
17

Stan Hu's avatar
Stan Hu committed
18
19
20
21
22
23
    def client
      @client ||= OAuth2::Client.new(provider.app_id, provider.app_secret, options)
    end

    def connection
      @connection ||= OAuth2::AccessToken.new(client, @token, refresh_token: @refresh_token, expires_at: @expires_at, expires_in: @expires_in)
24
25
26
    end

    def query(params = {})
Stan Hu's avatar
Stan Hu committed
27
      @query.merge!(params)
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
    end

    def get(path, query = {})
      refresh! if expired?

      response = connection.get(build_url(path), params: @query.merge(query))
      response.parsed
    end

    def expired?
      connection.expired?
    end

    def refresh!
      response = connection.refresh!

      @token         = response.token
      @expires_at    = response.expires_at
      @expires_in    = response.expires_in
      @refresh_token = response.refresh_token
48
      @connection = nil
49
50
51
52
    end

    private

Stan Hu's avatar
Stan Hu committed
53
    attr_reader :expires_at, :expires_in, :refresh_token, :token
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

    def build_url(path)
      return path if path.starts_with?(root_url)

      "#{root_url}#{path}"
    end

    def root_url
      @root_url ||= "#{@base_uri}#{@api_version}"
    end

    def provider
      Gitlab.config.omniauth.providers.find { |provider| provider.name == 'bitbucket' }
    end

    def options
      OmniAuth::Strategies::Bitbucket.default_options[:client_options].deep_symbolize_keys
    end
  end
end