2011-05-12 17:12:09 +00:00
|
|
|
$:.unshift('lib')
|
|
|
|
require 'rubygems'
|
|
|
|
require 'sinatra'
|
|
|
|
require 'datamapper'
|
|
|
|
require 'google/api_client'
|
2011-01-06 14:49:58 +00:00
|
|
|
|
2011-05-12 17:12:09 +00:00
|
|
|
use Rack::Session::Pool, :expire_after => 86400 # 1 day
|
2011-01-06 14:49:58 +00:00
|
|
|
|
2011-05-12 17:12:09 +00:00
|
|
|
# Set up our token store
|
|
|
|
DataMapper.setup(:default, 'sqlite::memory:')
|
|
|
|
class TokenPair
|
|
|
|
include DataMapper::Resource
|
2011-01-06 14:49:58 +00:00
|
|
|
|
2011-05-12 17:12:09 +00:00
|
|
|
property :id, Serial
|
|
|
|
property :refresh_token, String
|
|
|
|
property :access_token, String
|
2011-05-21 00:05:05 +00:00
|
|
|
property :expires_in, Integer
|
|
|
|
property :issued_at, Integer
|
2011-01-06 14:49:58 +00:00
|
|
|
|
2011-05-12 17:12:09 +00:00
|
|
|
def update_token!(object)
|
|
|
|
self.refresh_token = object.refresh_token
|
|
|
|
self.access_token = object.access_token
|
2011-05-21 00:05:05 +00:00
|
|
|
self.expires_in = object.expires_in
|
|
|
|
self.issued_at = object.issued_at
|
2011-05-12 17:12:09 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def to_hash
|
|
|
|
return {
|
|
|
|
:refresh_token => refresh_token,
|
|
|
|
:access_token => access_token,
|
2011-05-21 00:05:05 +00:00
|
|
|
:expires_in => expires_in,
|
|
|
|
:issued_at => Time.at(issued_at)
|
2011-05-12 17:12:09 +00:00
|
|
|
}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
TokenPair.auto_migrate!
|
|
|
|
|
|
|
|
before do
|
|
|
|
@client = Google::APIClient.new
|
|
|
|
@client.authorization.client_id = '245083617981.apps.googleusercontent.com'
|
|
|
|
@client.authorization.client_secret = 'pYelZCRjSa+iMezYENVScXFk'
|
|
|
|
@client.authorization.scope = 'https://www.googleapis.com/auth/buzz'
|
|
|
|
@client.authorization.redirect_uri = to('/oauth2callback')
|
|
|
|
@client.authorization.code = params[:code] if params[:code]
|
|
|
|
if session[:token_id]
|
|
|
|
# Load the access token here if it's available
|
|
|
|
token_pair = TokenPair.get(session[:token_id])
|
|
|
|
@client.authorization.update_token!(token_pair.to_hash)
|
|
|
|
end
|
2011-05-21 00:05:05 +00:00
|
|
|
if @client.authorization.refresh_token && @client.authorization.expired?
|
|
|
|
@client.authorization.fetch_access_token!
|
|
|
|
end
|
2011-05-12 17:12:09 +00:00
|
|
|
@buzz = @client.discovered_api('buzz')
|
|
|
|
unless @client.authorization.access_token || request.path_info =~ /^\/oauth2/
|
|
|
|
redirect to('/oauth2authorize')
|
|
|
|
end
|
2011-01-06 14:49:58 +00:00
|
|
|
end
|
|
|
|
|
2011-05-12 17:12:09 +00:00
|
|
|
get '/oauth2authorize' do
|
|
|
|
redirect @client.authorization.authorization_uri.to_s, 303
|
|
|
|
end
|
|
|
|
|
|
|
|
get '/oauth2callback' do
|
|
|
|
@client.authorization.fetch_access_token!
|
|
|
|
# Persist the token here
|
|
|
|
token_pair = if session[:token_id]
|
|
|
|
TokenPair.get(session[:token_id])
|
|
|
|
else
|
|
|
|
TokenPair.new
|
|
|
|
end
|
|
|
|
token_pair.update_token!(@client.authorization)
|
|
|
|
token_pair.save()
|
|
|
|
session[:token_id] = token_pair.id
|
|
|
|
redirect to('/')
|
|
|
|
end
|
|
|
|
|
|
|
|
get '/' do
|
|
|
|
response = @client.execute(
|
|
|
|
@buzz.activities.list,
|
|
|
|
'userId' => '@me', 'scope' => '@consumption', 'alt'=> 'json'
|
|
|
|
)
|
|
|
|
status, headers, body = response
|
|
|
|
[status, {'Content-Type' => 'application/json'}, body]
|
|
|
|
end
|