Class: Signet::OAuth2::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/googleauth/signet.rb

Overview

Signet::OAuth2::Client creates an OAuth2 client

This reopens Client to add #apply and #apply! methods which update a hash with the fetched authentication token.

Instance Method Summary collapse

Instance Method Details

#apply(a_hash, opts = {}) ⇒ Object

Returns a clone of a_hash updated with the authentication token



56
57
58
59
60
 
# File 'lib/googleauth/signet.rb', line 56

def apply a_hash, opts = {}
  a_copy = a_hash.clone
  apply! a_copy, opts
  a_copy
end

#apply!(a_hash, opts = {}) ⇒ Object

Updates a_hash updated with the authentication token



48
49
50
51
52
53
 
# File 'lib/googleauth/signet.rb', line 48

def apply! a_hash, opts = {}
  # fetch the access token there is currently not one, or if the client
  # has expired
  fetch_access_token! opts if access_token.nil? || expires_within?(60)
  a_hash[AUTH_METADATA_KEY] = "Bearer #{access_token}"
end

#build_default_connection ⇒ Object 



93
94
95
96
97
98
99
100
101
 
# File 'lib/googleauth/signet.rb', line 93

def build_default_connection
  if !defined?(@connection_info)
    nil
  elsif @connection_info.respond_to? :call
    @connection_info.call
  else
    @connection_info
  end
end

#configure_connection(options) ⇒ Object 



41
42
43
44
45
 
# File 'lib/googleauth/signet.rb', line 41

def configure_connection options
  @connection_info =
    options[:connection_builder] || options[:default_connection]
  self
end

#fetch_access_token!(options = {}) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/googleauth/signet.rb', line 74

def fetch_access_token! options = {}
  unless options[:connection]
    connection = build_default_connection
    options = options.merge connection: connection if connection
  end
  info = retry_with_error do
    orig_fetch_access_token! options
  end
  notify_refresh_listeners
  info
end

#notify_refresh_listeners ⇒ Object 



86
87
88
89
90
91
 
# File 'lib/googleauth/signet.rb', line 86

def notify_refresh_listeners
  listeners = defined?(@refresh_listeners) ? @refresh_listeners : []
  listeners.each do |block|
    block.call self
  end
end

#on_refresh(&block) ⇒ Object 



68
69
70
71
 
# File 'lib/googleauth/signet.rb', line 68

def on_refresh &block
  @refresh_listeners = [] unless defined? @refresh_listeners
  @refresh_listeners << block
end

#orig_fetch_access_token! ⇒ Object 



73
 
# File 'lib/googleauth/signet.rb', line 73

alias orig_fetch_access_token! fetch_access_token!

#retry_with_error(max_retry_count = 5) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
 
# File 'lib/googleauth/signet.rb', line 103

def retry_with_error max_retry_count = 5
  retry_count = 0

  begin
    yield
  rescue StandardError => e
    raise e if e.is_a?(Signet::AuthorizationError) || e.is_a?(Signet::ParseError)

    if retry_count < max_retry_count
      retry_count += 1
      sleep retry_count * 0.3
      retry
    else
      msg = "Unexpected error: #{e.inspect}"
      raise Signet::AuthorizationError, msg
    end
  end
end

#updater_proc ⇒ Object

Returns a reference to the #apply method, suitable for passing as a closure



64
65
66
 
# File 'lib/googleauth/signet.rb', line 64

def updater_proc
  lambda(&method(:apply))
end