slipsquare 0.0.1

data/lib/slipsquare/middleware/account.rb

@@ -0,0 +1,14 @@
+require 'table_print'
+
+module Slipsquare
+  module Middleware
+    class Account < Base
+      def call(env)
+
+        tp env['dropbox-client'].account
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/ask_for_credentials.rb

@@ -0,0 +1,39 @@
+require "dropbox-api"
+require "cgi"
+
+module Slipsquare
+  module Middleware
+    # Ask for user credentials from the command line, then write them out.
+    class AskForCredentials < Base
+      def call(env)
+
+        say "If you dont know your app and secret key, try `slipsquare get-keys`"
+
+        app_key = ask "Enter your App key:"
+        secret_key = ask "Enter your Secret key:"
+
+        Dropbox::API::Config.app_key    = app_key
+        Dropbox::API::Config.app_secret = secret_key
+
+        consumer = Dropbox::API::OAuth.consumer(:authorize)
+        request_token = consumer.get_request_token
+        say "Go to this url and click 'Authorize' to get the token:"
+        say request_token.authorize_url, :green
+        query  = request_token.authorize_url.split('?').last
+        params = CGI.parse(query)
+        token  = params['oauth_token'].first
+        yes? "Once you authorize the app on Dropbox, type yes... "
+        access_token  = request_token.get_access_token(:oauth_verifier => token)
+
+        # Write the config file.
+        env['config'].create_config_file(app_key, secret_key, access_token.token, access_token.secret)
+        env['config'].reload!
+
+        say "Credentials saved to ~/.slipsquare"
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/slipsquare/middleware/base.rb

@@ -0,0 +1,28 @@
+module Slipsquare
+  module Middleware
+    # A base middleware class to initalize.
+    class Base
+      # Some colors for making things pretty.
+      CLEAR      = "\e[0m"
+      RED        = "\e[31m"
+      GREEN      = "\e[32m"
+      YELLOW     = "\e[33m"
+
+      # We want access to all of the fun thor cli helper methods,
+      # like say, yes?, ask, etc.
+      include Thor::Shell
+
+      def initialize(app)
+        @app = app
+        # This resets the color to "clear" on the user's terminal.
+        say "", :clear, false
+      end
+
+      def call(env)
+        @app.call(env)
+      end
+
+    end
+  end
+end
+

data/lib/slipsquare/middleware/check_configuration.rb

@@ -0,0 +1,18 @@
+module Slipsquare
+  module Middleware
+    # Check if the client has set-up configuration yet.
+    class CheckConfiguration < Base
+      def call(env)
+        config = env["config"]
+
+        if !config || !config.data || !config.app_key || !config.secret_key
+          say "You must run `slipsquare authorize` in order to connect to Dropbox", :red
+          exit 1
+        end
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/slipsquare/middleware/check_credentials.rb

@@ -0,0 +1,27 @@
+require 'thor'
+require 'dropbox-api'
+require 'cgi'
+
+module Slipsquare
+  module Middleware
+    class CheckCredentials < Base
+      def call(env)
+
+        say "Checking credentials with Dropbox..."
+
+        begin
+            env['dropbox-client'].ls
+        rescue Dropbox::API::Error => e
+            say "Connection to Dropbox failed (#{e})", :red
+            say "Check your .slipsquare file, and double check your credentials are correct", :yellow
+            exit 1
+        end
+
+        say "Connection to dropbox successful!", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/slipsquare/middleware/chunked_upload.rb

@@ -0,0 +1,29 @@
+require 'ruby-progressbar'
+
+module Slipsquare
+  module Middleware
+    class ChunkedUpload < Base
+      def call(env)
+        say "Starting chunked upload"
+        
+        file_name = env['chunked_upload_file_name']
+        contents  = File.open(env['chunked_upload_file_name'])
+        total_size = File.size(env['chunked_upload_file_name'])
+
+        say "Total Size: #{total_size} bytes"
+        upload_progress_bar = ProgressBar.create(:title => "Upload progress", 
+          :format => '%a <%B> %p%% %t',
+          :starting_at => 0, 
+          :total => total_size)
+
+        response = env['dropbox-client'].chunked_upload file_name, contents do |offset, upload|          
+          upload_progress_bar.progress += offset          
+        end        
+
+        say "File uploaded successfully!"
+        
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/delete_file.rb

@@ -0,0 +1,13 @@
+module Slipsquare
+  module Middleware
+    class DeleteFile < Base
+      def call(env)
+        say "Deleting file on Dropbox..."
+
+        say "To be implemented...", :yellow
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/download_file.rb

@@ -0,0 +1,25 @@
+module Slipsquare
+  module Middleware
+    class DownloadFile < Base
+      def call(env)
+        say "Download file..."
+
+        begin
+          contents = env['dropbox-client'].download env['download_file_name']
+        rescue Dropbox::API::Error::NotFound => e
+          say "File Not Found! Could not download", :red
+          exit 1
+        rescue Dropbox::API::Error => e
+          say "Connection to Dropbox failed (#{e})", :red
+          exit 1
+        end
+
+        File.open(env['download_file_name'], 'w') {|f| f.write(contents) }
+
+        say "File '#{env['download_file_name']}' downloaded successfully!"
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/inject_client.rb

@@ -0,0 +1,27 @@
+require 'thor'
+require "dropbox-api"
+
+module Slipsquare
+  module Middleware
+    class InjectClient < Base
+        def call(env)
+
+          Dropbox::API::Config.app_key    = env['config'].app_key
+          Dropbox::API::Config.app_secret = env['config'].secret_key
+          Dropbox::API::Config.mode       = 'dropbox'
+
+          begin
+            client = Dropbox::API::Client.new(:token  => env['config'].app_token, :secret =>  env['config'].app_secret)
+          rescue Dropbox::API::Error => e
+            say "Connection to Dropbox failed (#{e})", :red
+            exit 1
+          end
+
+          env['dropbox-client'] = client
+
+          @app.call(env)
+        end
+    end
+  end
+end
+

data/lib/slipsquare/middleware/inject_configuration.rb

@@ -0,0 +1,15 @@
+module Slipsquare
+  module Middleware
+    # Check if the client has set-up configuration yet.
+    class InjectConfiguration < Base
+      def call(env)
+        config = Slipsquare::Configuration.instance
+
+        env["config"] = config
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/slipsquare/middleware/list_files.rb

@@ -0,0 +1,19 @@
+require 'table_print'
+
+module Slipsquare
+  module Middleware
+    class ListFiles < Base
+      def call(env)
+        say "Listing files from Dropbox..."
+
+        if env["list_files_path"].nil?
+          tp env['dropbox-client'].ls, 'path', 'mime_type', 'size'
+        else
+          tp env['dropbox-client'].ls(env["list_files_path"]), 'path', 'mime_type', 'size'
+        end
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/make_directory.rb

@@ -0,0 +1,25 @@
+require 'table_print'
+
+module Slipsquare
+  module Middleware
+    class MakeDirectory < Base
+      def call(env)
+        say "Creating directory on Dropbox..."
+
+        friendly_path = env["mkdir_path"]
+
+        if friendly_path =~ (/[\\\:\?\*\<\>\"\|]+/)
+          friendly_path = friendly_path.gsub(/[\\\:\?\*\<\>\"\|]+/, '')
+          say "Illegal character found! Escaping...", :yellow
+          say "Escaped directory path: #{friendly_path}"
+        end
+
+        new_folder_response = env['dropbox-client'].mkdir(env["mkdir_path"])
+
+        say "Directory `#{friendly_path}` created successfully", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/middleware/upload_file.rb

@@ -0,0 +1,17 @@
+module Slipsquare
+  module Middleware
+    class UploadFile < Base
+      def call(env)
+        say "Uploading file..."
+
+        contents = File.read(env['upload_file_name'])
+
+        env['dropbox-client'].upload env['upload_file_name'], contents
+
+        say "File '#{env['upload_file_name']}' uploaded successfully"
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/slipsquare/version.rb

@@ -0,0 +1,3 @@
+module Slipsquare
+  VERSION = "0.0.1"
+end

data/slipsquare.gemspec

@@ -0,0 +1,34 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'slipsquare/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "slipsquare"
+  gem.version       = Slipsquare::VERSION
+  gem.authors       = ["Peter Souter"]
+  gem.email         = ["p.morsou@gmail.com"]
+  gem.description   = %q{A command line tool for uploading to dropbox.}
+  gem.summary       = %q{Slipsquare is a CLI for Dropbox, for downloading and uploading files.}
+  gem.homepage      = "https://github.com/petems/slipsquare"
+  gem.license       = 'MIT'
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "thor", "~> 0.18.1"
+  gem.add_dependency "middleware" , "~> 0.1.0"
+
+  gem.add_dependency "dropbox-api-petems", "0.4.2"
+  gem.add_dependency "table_print", "~> 1.1.5"
+  gem.add_dependency "ruby-progressbar", "~> 1.2.0"
+
+  gem.add_development_dependency "rake", "~> 10.1.0"
+  gem.add_development_dependency "rspec-core", "~> 2.13.0"
+  gem.add_development_dependency "rspec-expectations", "~> 2.13.0"
+  gem.add_development_dependency "rspec-mocks", "~> 2.13.0"
+  gem.add_development_dependency "webmock", "~> 1.11.0"
+
+end

data/spec/cli/account.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe Slipsquare::CLI do
+  include_context "spec"
+
+  describe "account" do
+    it "returns account info" do
+
+      stub_request(:get, "https://api.dropbox.com/1/account/info?").
+         with(:headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_consumer_key="#{app_key}", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_token="#{app_token}", oauth_version="1.0"/, 'User-Agent'=>'OAuth gem v0.4.7'}).
+         to_return(:status => 200, :body => fixture('account'))
+
+      @cli.account
+      expect($stdout.string).to eq <<-eos
+REFERRAL_LINK                  | DISPLAY_NAME | UID      | COUNTRY | QUOTA_INFO                    
+-------------------------------|--------------|----------|---------|-------------------------------
+https://www.dropbox.com/ref... | John P. User | 12345678 | US      | #<Dropbox::API::Object norm...
+      eos
+    end
+
+  end
+
+end
+

data/spec/cli/authorize_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+describe Slipsquare::CLI do
+  include_context "spec"
+
+  describe "authorize" do
+    before do
+      stub_request(:post, "https://www.dropbox.com/1/oauth/request_token").
+      with(:headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_body_hash="\S+", oauth_callback="oob", oauth_consumer_key="foo", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_version="1.0"/, 'Content-Length'=>'0', 'User-Agent'=>'OAuth gem v0.4.7'}).
+      to_return(:status => 200, :body => "oauth_token_secret=b9q1n5il4lcc&oauth_token=mh7an9dkrg59")
+
+      stub_request(:post, "https://www.dropbox.com/1/oauth/access_token").
+      with(:headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_body_hash="\S+", oauth_consumer_key="#{app_key}", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_token="\S+", oauth_verifier="\S+", oauth_version="1.0"/, 'Content-Length'=>'0', 'User-Agent'=>'OAuth gem v0.4.7'}).
+      to_return(:status => 200, :body => "oauth_token_secret=95grkd9na7hm&oauth_token=ccl4li5n1q9b&uid=100")
+
+      stub_request(:get, "https://api.dropbox.com/1/metadata/dropbox/?").
+      with(:headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_consumer_key="#{app_key}", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_token="\S+", oauth_version="1.0"/, 'User-Agent'=>'OAuth gem v0.4.7'}).
+      to_return(:status => 200, :body => fixture("verify_success"))
+    end
+
+    it "asks the right questions and saves then checks credentials" do
+
+      $stdout.should_receive(:print).exactly(6).times
+      $stdout.should_receive(:print).with("Enter your App key: ")
+      $stdin.should_receive(:gets).and_return(app_key)
+      $stdout.should_receive(:print).with("Enter your Secret key: ")
+      $stdin.should_receive(:gets).and_return(secret_key)
+      $stdout.should_receive(:print).with("Once you authorize the app on Dropbox, type yes...  ")
+      $stdin.should_receive(:gets).and_return('yes')
+
+      @cli.authorize
+
+      expect($stdout.string).to eq <<-eos
+If you dont know your app and secret key, try `slipsquare get-keys`
+Go to this url and click 'Authorize' to get the token:
+https://www.dropbox.com/1/oauth/authorize?oauth_token=mh7an9dkrg59
+Credentials saved to ~/.slipsquare
+Checking credentials with Dropbox...
+Connection to dropbox successful!
+      eos
+   end
+
+ end
+
+end
+

data/spec/cli/chunked_upload_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+describe Slipsquare::CLI do
+  include_context "spec"
+
+  before :each do
+    File.open("foo.txt", 'w') {|f| f.write("Laugh and grow fat...") }
+  end
+
+  after :each do
+    File.delete("foo.txt") if File.exist?("foo.txt")
+  end
+
+  describe "upload-chunked" do
+
+    it "uploads a larger file and gives progress" do
+
+      stub_request(:put, "https://api-content.dropbox.com/1/chunked_upload?offset=0").
+         with(:body => "Laugh and grow fat...",
+              :headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_body_hash="\S+", oauth_consumer_key="#{app_key}", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_token="#{app_token}", oauth_version="1.0"/, 'Content-Length'=>'21', 'Content-Type'=>'application/octet-stream', 'User-Agent'=>'OAuth gem v0.4.7'}).
+         to_return(:status => 200, :body => fixture('chunked_upload'))
+
+      stub_request(:post, "https://api-content.dropbox.com/1/commit_chunked_upload/dropbox/foo.txt?upload_id=v0k84B0AT9fYkfMUp0sBTA").
+         with(:headers => {'Accept'=>'*/*', 'Authorization'=>/OAuth oauth_body_hash="\S+", oauth_consumer_key="#{app_key}", oauth_nonce="\S+", oauth_signature="\S+", oauth_signature_method="HMAC-SHA1", oauth_timestamp="\S+", oauth_token="#{app_token}", oauth_version="1.0"/, 'Content-Length'=>'0', 'Content-Type'=>'application/octet-stream', 'User-Agent'=>'OAuth gem v0.4.7'}).
+         to_return(:status => 200, :body => fixture('commit_chunked_upload'), :headers => {})
+
+      @cli.chunked_upload('foo.txt')
+      expect($stdout.string).to eq <<-eos
+Starting chunked upload
+Total Size: 21 bytes
+Time: 00:00:00 <==========================================> 100% Upload progress
+File uploaded successfully!
+      eos
+    end
+
+  end
+
+end
+