Class: Wakatime::Session

Inherits:
Object
  • Object
show all
Defined in:
lib/wakatime/session.rb

Instance Method Summary collapse

Constructor Details

#initialize(opts = {}, backoff = 0.1) ⇒ Session

Returns a new instance of Session.



3
4
5
6
 
# File 'lib/wakatime/session.rb', line 3

def initialize(opts = {}, backoff = 0.1)
  @backoff = backoff # try not to excessively hammer API.
  @api_key = opts[:api_key]
end

Instance Method Details

#get(url, raw = false) ⇒ Object 



8
9
10
11
12
 
# File 'lib/wakatime/session.rb', line 8

def get(url, raw = false)
  uri = URI.parse(url)
  request = Net::HTTP::Get.new(uri.request_uri)
  resp = request(uri, request, raw)
end

#handle_errors(response, raw) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/wakatime/session.rb', line 34

def handle_errors(response, raw)
  status = response.code.to_i
  body = response.body
  begin
    parsed_body = JSON.parse(body)
  rescue
    msg = body.nil? || body.empty? ? 'no data returned' : body
    parsed_body = { 'errors' =>  msg }
  end
  # status is used to determine whether
  # we need to refresh the access token.
  parsed_body['status'] = status

  case status / 100
  when 3
    # 302 Found. We should return the url
    parsed_body['location'] = response['Location'] if status == 302
  when 4
    fail(Wakatime::AuthError.new(parsed_body, status, body), parsed_body['errors']) if status == 401
    fail(Wakatime::RequestError.new(parsed_body, status, body), parsed_body['errors'])
  when 5
    fail(Wakatime::ServerError.new(parsed_body, status, body), parsed_body['errors'])
  end
  raw ? body : parsed_body
end

#request(uri, request, raw = false, _retries = 0) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/wakatime/session.rb', line 14

def request(uri, request, raw = false, _retries = 0)
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true

  # http.set_debug_output($stdout)
  request.add_field('Content-Type', 'application/json')

  if @api_key
    request.add_field('Authorization', "Basic #{Base64.strict_encode64(@api_key)}") #
  end

  response = http.request(request)

  fail Wakatime::ObjectNotFound if response.is_a? Net::HTTPNotFound

  sleep(@backoff) # try not to excessively hammer API.

  handle_errors(response, raw)
end