gspush 0.0.1 → 0.0.2

data/bin/gspush

@@ -1,5 +1,5 @@
 #!/usr/bin/env ruby
 
-require File.dirname(__FILE__) + '/../lib/gspush'
+require File.expand_path(File.dirname(__FILE__) + '/../lib/gspush')
 
 Gspush::CLI.execute(ARGV)

data/bin/gspush_generate

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+require File.expand_path(File.dirname(__FILE__) + '/../lib/gspush/oauth2')
+
+# FIXME: give file path
+Gspush::Oauth2.generate
+

data/gspush.gemspec

@@ -16,5 +16,8 @@ Gem::Specification.new do |gem|
   gem.version       = Gspush::VERSION
 
   gem.add_dependency "google_drive"
-  gem.add_development_dependency "rspec"
+
+  ["rspec", "fakefs"].each do |g|
+    gem.add_development_dependency g
+  end
 end

data/lib/gspush.rb

@@ -1,7 +1,11 @@
 require 'optparse'
 require 'google_drive'
+require 'oauth2'
+require 'yaml'
+require 'highline'
 
 require_relative './gspush/version'
+require_relative './gspush/oauth2'
 
 class Gspush
   class WorksheetNotFound < StandardError; end
@@ -16,6 +20,9 @@ class Gspush
     @username  = options[:username]
     @password  = options[:password]
 
+    @oauth2 = options[:oauth2]
+    @oauth_token_file = options[:oauth_token_file]
+
     @prepend_timestamp = options[:prepend_timestamp]
     @sheet_title = options[:sheet_title]
 
@@ -40,7 +47,7 @@ class Gspush
   end
 
   def save
-    spreadsheet = open(@url, @username, @password)
+    spreadsheet = open(@url)
 
     sheet = select_worksheet(spreadsheet)
     raise WorksheetNotFound if sheet.nil?
@@ -53,9 +60,14 @@ class Gspush
 
   private
 
-  def open(url, username, password)
-    session     = GoogleDrive.login(username, password)
-    spreadsheet = session.spreadsheet_by_url(url)
+  def open(url)
+    session = if @oauth2
+                oauth = Gspush::Oauth2.new(@oauth_token_file)
+                GoogleDrive.login_with_oauth(oauth.token)
+              else
+                GoogleDrive.login(username, password)
+              end
+    session.spreadsheet_by_url(url)
   end
 
   def select_worksheet(spreadsheet)
@@ -89,6 +101,8 @@ class Gspush
       opt.on('-d delim', 'input delimiter') {|v| options[:delimiter] = v }
       opt.on('-u username', 'Google Drive username(email)') {|v| options[:username] = v }
       opt.on('-p password', 'user password') {|v| options[:password] = v } # XXX how should i get this?
+      opt.on('-o', 'using oauth2 authorize') {|v| options[:oauth2] = v }
+      opt.on('-f oauth_token_file', 'oauth2 token file path') {|v| options[:oauth_token_file] = v }
       opt.on('-s sheet_title', 'worksheet title (default: first worksheet)') {|v| options[:sheet_title] = v }
       opt.on('-t', 'prepend timestamp cell') {|v| options[:prepend_timestamp] = v }
 
@@ -97,7 +111,7 @@ class Gspush
 
       argv = opt.parse(argv_original)
 
-      if argv.empty?
+      if argv.empty? && !options[:generate_token_file]
         puts opt.banner
         exit 1
       end

data/lib/gspush/oauth2.rb

@@ -0,0 +1,109 @@
+require 'highline'
+require 'oauth2'
+
+class Gspush
+  class Oauth2
+    attr_accessor :access_token, :refresh_token, :expires_at
+    attr_accessor :client_id, :client_secret, :redirect_uri
+
+		def initialize(file_path = nil)
+      @file_path = file_path || "#{ENV["HOME"]}/.gspushrc"
+      @redirect_uri = "urn:ietf:wg:oauth:2.0:oob"
+    end
+
+		def token
+      self.load or raise "First you need gspush_generate to create oauth2 configuration"
+      token = OAuth2::AccessToken.from_hash(client, to_hash)
+
+      if expired?
+        token.refresh!
+        save
+      end
+
+      token
+		end
+
+    def self.generate
+      puts "Input your application information"
+      puts "(see https://code.google.com/apis/console )"
+
+			oauth2 = new
+
+      hi = HighLine.new
+      oauth2.client_id = hi.ask("Client ID? > ")
+      oauth2.client_secret = hi.ask("Client Secret? > ")
+
+      oauth2.get_access_token
+      oauth2.save
+    end
+
+    def save
+			File.open(@file_path, 'w') do |f|
+				f.print to_hash.to_yaml
+			end
+			puts "ok saved in #{@file_path}"
+    end
+
+    def load
+      if FileTest.exist?(@file_path)
+        o = YAML.load(File.read(@file_path))
+        @access_token = o[:access_token]
+        @refresh_token = o[:refresh_token]
+        @expires_at = o[:expires_at]
+        @client_id = o[:client_id]
+        @client_secret = o[:client_secret]
+        @redirect_uri = o[:redirect_uri]
+        true
+      else
+        false
+      end
+    end
+
+    def client
+      OAuth2::Client.new(
+        @client_id, @client_secret,
+        :site => "https://accounts.google.com",
+        :token_url => "/o/oauth2/token",
+        :authorize_url => "/o/oauth2/auth")
+    end
+
+    def get_access_token
+      auth_url = client.auth_code.authorize_url(
+        :redirect_uri => @redirect_uri,
+        :scope =>
+          ["https://docs.google.com/feeds/",
+           "https://docs.googleusercontent.com/",
+           "https://spreadsheets.google.com/feeds/"].join(" "))
+      hi = HighLine.new
+
+      puts "Access in your browser: #{auth_url}"
+      authorization_code = hi.ask("And please input authorization code: ")
+
+      auth_token = client.auth_code.get_token(
+        authorization_code, :redirect_uri => @redirect_uri)
+
+      @access_token  = auth_token.token
+      @refresh_token = auth_token.refresh_token
+      @expires_at    = auth_token.expires_at
+
+      @access_token
+    end
+
+    def expired?
+      @expires_at && Time.now.to_i > @expires_at
+    end
+
+    def to_hash
+      {
+        :client_id => client_id,
+        :client_secret => client_secret,
+        :redirect_uri => redirect_uri,
+        :access_token => access_token,
+        :refresh_token => refresh_token,
+        :expires_at => expires_at,
+      }
+    end
+  end
+end
+
+

data/lib/gspush/version.rb

@@ -1,3 +1,3 @@
 class Gspush
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/gspush_spec.rb

@@ -1,4 +1,4 @@
-require 'gspush'
+require 'spec_helper'
 
 describe Gspush do
   let(:url) { 'http://example.com/ '} # FIXME

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+require 'rspec'
+require 'gspush'
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gspush
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-09-28 00:00:00.000000000 Z
+date: 2012-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: google_drive
@@ -43,11 +43,28 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: fakefs
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Push command to Google Spreadsheet
 email:
 - holysugar@gmail.com
 executables:
 - gspush
+- gspush_generate
 extensions: []
 extra_rdoc_files: []
 files:
@@ -57,10 +74,13 @@ files:
 - README.md
 - Rakefile
 - bin/gspush
+- bin/gspush_generate
 - gspush.gemspec
 - lib/gspush.rb
+- lib/gspush/oauth2.rb
 - lib/gspush/version.rb
 - spec/gspush_spec.rb
+- spec/spec_helper.rb
 homepage: https://github.com/holysugar/gspush
 licenses: []
 post_install_message: 
@@ -87,3 +107,4 @@ specification_version: 3
 summary: Phsh command to Google Spreadsheet
 test_files:
 - spec/gspush_spec.rb
+- spec/spec_helper.rb