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



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

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
54
 
# 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
  token_type = target_audience ? :id_token : :access_token
  fetch_access_token! opts if send(token_type).nil? || expires_within?(60)
  a_hash[AUTH_METADATA_KEY] = "Bearer #{send token_type}"
end

#build_default_connection ⇒ Object 



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

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 



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

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 



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

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

#on_refresh(&block) ⇒ Object 



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

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

#orig_fetch_access_token! ⇒ Object 



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

alias orig_fetch_access_token! fetch_access_token!

#retry_with_error(max_retry_count = 5) ⇒ Object 



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

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



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

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