Class: B2::APIConnection

Inherits:
Object
  • Object
show all
Defined in:
lib/b2/api_connection.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(key_id, secret) ⇒ APIConnection

Returns a new instance of APIConnection.



6
7
8
9
 
# File 'lib/b2/api_connection.rb', line 6

def initialize(key_id, secret)
  @key_id = key_id
  @key_secret = secret
end

Instance Attribute Details

#download_url ⇒ Object (readonly)

Returns the value of attribute download_url.



4
5
6
 
# File 'lib/b2/api_connection.rb', line 4

def download_url
  @download_url
end

#key_id ⇒ Object (readonly)

Returns the value of attribute key_id.



4
5
6
 
# File 'lib/b2/api_connection.rb', line 4

def key_id
  @key_id
end

#key_secret ⇒ Object (readonly)

Returns the value of attribute key_secret.



4
5
6
 
# File 'lib/b2/api_connection.rb', line 4

def key_secret
  @key_secret
end

Instance Method Details

#account_id ⇒ Object 



40
41
42
43
44
45
 
# File 'lib/b2/api_connection.rb', line 40

def 
  return @account_id if !@account_id.nil?
  
  connect!
  @account_id
end

#active? ⇒ Boolean

Returns:

  • (Boolean) 


66
67
68
 
# File 'lib/b2/api_connection.rb', line 66

def active?
  !@connection.nil? && @connection.active?
end

#authorization_token ⇒ Object 



59
60
61
62
63
64
 
# File 'lib/b2/api_connection.rb', line 59

def authorization_token
  if @auth_token_expires_at.nil? || @auth_token_expires_at <= Time.now.to_i
    reconnect!
  end
  @auth_token
end

#connect! ⇒ Object 



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
 
# File 'lib/b2/api_connection.rb', line 11

def connect!
  conn = Net::HTTP.new('api.backblazeb2.com', 443)
  conn.use_ssl = true
  
  req = Net::HTTP::Get.new('/b2api/v2/b2_authorize_account')
  req.basic_auth(@key_id, @key_secret)

  key_expiration = Time.now.to_i + 86_400 #24hr expiry
  resp = conn.start { |http| http.request(req) }
  if resp.is_a?(Net::HTTPSuccess)
    resp = JSON.parse(resp.body)
  else
    raise "Error connecting to B2 API"
  end

  uri = URI.parse(resp['apiUrl'])
  @connection = Net::HTTP.new(uri.host, uri.port)
  @connection.use_ssl = uri.scheme == 'https'
  @connection.start

  @auth_token_expires_at = key_expiration
  @account_id = resp['accountId']
  @minimum_part_size = resp['absoluteMinimumPartSize']
  @recommended_part_size = resp['recommendedPartSize']
  @auth_token = resp['authorizationToken']
  @download_url = resp['downloadUrl']
  @buckets_cache = []
end

#connection ⇒ Object 



70
71
72
73
 
# File 'lib/b2/api_connection.rb', line 70

def connection
  reconnect! if !active?
  @connection
end

#disconnect! ⇒ Object 



47
48
49
50
51
52
 
# File 'lib/b2/api_connection.rb', line 47

def disconnect!
  if @connection
    @connection.finish if @connection.active?
    @connection = nil
  end
end

#reconnect! ⇒ Object 



54
55
56
57
 
# File 'lib/b2/api_connection.rb', line 54

def reconnect!
  disconnect!
  connect!
end

#send_request(request, body = nil, &block) ⇒ Object 



75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/b2/api_connection.rb', line 75

def send_request(request, body=nil, &block)
  request['Authorization'] = authorization_token
  request.body = (body.is_a?(String) ? body : JSON.generate(body)) if body
  
  return_value = nil
  close_connection = false
  connection.request(request) do |response|
    close_connection = response['Connection'] == 'close'
    
    case response
    when Net::HTTPSuccess
      if block_given?
        return_value = yield(response)
      else
        return_value = JSON.parse(response.body)
      end
    else
      raise "Error connecting to B2 API #{response.body}"
    end
  end
  disconnect! if close_connection

  return_value
end