127 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			127 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # Copyright 2010 Google Inc.
 | |
| #
 | |
| # Licensed under the Apache License, Version 2.0 (the "License");
 | |
| # you may not use this file except in compliance with the License.
 | |
| # You may obtain a copy of the License at
 | |
| #
 | |
| #      http://www.apache.org/licenses/LICENSE-2.0
 | |
| #
 | |
| # Unless required by applicable law or agreed to in writing, software
 | |
| # distributed under the License is distributed on an "AS IS" BASIS,
 | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| # See the License for the specific language governing permissions and
 | |
| # limitations under the License.
 | |
| 
 | |
| require 'jwt'
 | |
| require 'signet/oauth_2/client'
 | |
| require 'delegate'
 | |
| 
 | |
| module Google
 | |
|   class APIClient
 | |
|     ##
 | |
|     # Generates access tokens using the JWT assertion profile. Requires a
 | |
|     # service account & access to the private key.
 | |
|     #
 | |
|     # @example Using Signet
 | |
|     #
 | |
|     #   key = Google::APIClient::KeyUtils.load_from_pkcs12('client.p12', 'notasecret')
 | |
|     #   client.authorization = Signet::OAuth2::Client.new(
 | |
|     #     :token_credential_uri => 'https://accounts.google.com/o/oauth2/token',
 | |
|     #     :audience => 'https://accounts.google.com/o/oauth2/token',
 | |
|     #     :scope => 'https://www.googleapis.com/auth/prediction',
 | |
|     #     :issuer => '123456-abcdef@developer.gserviceaccount.com',
 | |
|     #     :signing_key => key)
 | |
|     #   client.authorization.fetch_access_token!
 | |
|     #   client.execute(...)
 | |
|     #
 | |
|     # @deprecated
 | |
|     #  Service accounts are now supported directly in Signet
 | |
|     # @see https://developers.google.com/accounts/docs/OAuth2ServiceAccount
 | |
|     class JWTAsserter
 | |
|       # @return [String] ID/email of the issuing party
 | |
|       attr_accessor :issuer
 | |
|       # @return [Fixnum] How long, in seconds, the assertion is valid for
 | |
|       attr_accessor :expiry
 | |
|       # @return [Fixnum] Seconds to expand the issued at/expiry window to account for clock skew
 | |
|       attr_accessor :skew
 | |
|       # @return [String] Scopes to authorize
 | |
|       attr_reader :scope
 | |
|       # @return [String,OpenSSL::PKey] key for signing assertions
 | |
|       attr_writer :key
 | |
|       # @return [String] Algorithm used for signing
 | |
|       attr_accessor :algorithm
 | |
|       
 | |
|       ##
 | |
|       # Initializes the asserter for a service account.
 | |
|       #
 | |
|       # @param [String] issuer
 | |
|       #    Name/ID of the client issuing the assertion
 | |
|       # @param [String, Array] scope
 | |
|       #   Scopes to authorize. May be a space delimited string or array of strings
 | |
|       # @param [String,OpenSSL::PKey] key
 | |
|       #   Key for signing assertions
 | |
|       # @param [String] algorithm
 | |
|       #   Algorithm to use, either 'RS256' for RSA with SHA-256 
 | |
|       #   or 'HS256' for HMAC with SHA-256
 | |
|       def initialize(issuer, scope, key, algorithm = "RS256")
 | |
|         self.issuer = issuer
 | |
|         self.scope = scope
 | |
|         self.expiry = 60 # 1 min default 
 | |
|         self.skew = 60      
 | |
|         self.key = key
 | |
|         self.algorithm = algorithm
 | |
|       end
 | |
| 
 | |
|       ##
 | |
|       # Set the scopes to authorize
 | |
|       #
 | |
|       # @param [String, Array] new_scope
 | |
|       #   Scopes to authorize. May be a space delimited string or array of strings
 | |
|       def scope=(new_scope)
 | |
|         case new_scope
 | |
|         when Array
 | |
|           @scope = new_scope.join(' ')
 | |
|         when String
 | |
|           @scope = new_scope
 | |
|         when nil
 | |
|           @scope = ''
 | |
|         else
 | |
|           raise TypeError, "Expected Array or String, got #{new_scope.class}"
 | |
|         end
 | |
|       end
 | |
|       
 | |
|       ##
 | |
|       # Request a new access token.
 | |
|       # 
 | |
|       # @param [String] person
 | |
|       #   Email address of a user, if requesting a token to act on their behalf
 | |
|       # @param [Hash] options
 | |
|       #   Pass through to Signet::OAuth2::Client.fetch_access_token
 | |
|       # @return [Signet::OAuth2::Client] Access token 
 | |
|       #
 | |
|       # @see Signet::OAuth2::Client.fetch_access_token!
 | |
|       def authorize(person = nil, options={})
 | |
|         authorization = self.to_authorization(person)
 | |
|         authorization.fetch_access_token!(options)
 | |
|         return authorization
 | |
|       end
 | |
|       
 | |
|       ##
 | |
|       # Builds a Signet OAuth2 client
 | |
|       #
 | |
|       # @return [Signet::OAuth2::Client] Access token 
 | |
|       def to_authorization(person = nil)
 | |
|         return Signet::OAuth2::Client.new(
 | |
|           :token_credential_uri => 'https://accounts.google.com/o/oauth2/token',
 | |
|           :audience => 'https://accounts.google.com/o/oauth2/token',
 | |
|           :scope => self.scope,
 | |
|           :issuer => @issuer,
 | |
|           :signing_key => @key,
 | |
|           :signing_algorithm => @algorithm,
 | |
|           :person => person
 | |
|         )
 | |
|       end      
 | |
|     end
 | |
|   end
 | |
| end
 |