cloud_door 0.0.1 → 0.0.2

data/lib/cloud_door/google_drive_bridge.rb

@@ -0,0 +1,137 @@
+require 'google/api_client'
+require 'yaml'
+require 'mimetype_fu'
+require 'cloud_door/cloud_storage'
+
+module CloudDoor
+  class GoogleDriveBridge
+    def initialize(credentials)
+      client = Google::APIClient.new(application_name: 'cloud_door')
+      client.authorization = credentials
+      drive  = client.discovered_api('drive', 'v2')
+      @gclient = client
+      @gdrive  = drive
+    end
+
+    def request_user
+      result = send_request('@gdrive.about.get')
+      result.data.user
+    end
+
+    def request_root_id
+      result = send_request('@gdrive.about.get')
+      result.data.root_folder_id
+    end
+
+    def request_dir(file_id)
+      dir_files  = Array.new
+      page_token = nil
+      begin
+        parameters = {}
+        parameters['q'] = "'#{file_id}' in parents"
+        if page_token.to_s != ''
+          parameters['pageToken'] = page_token
+        end
+        api_result = send_request('@gdrive.files.list', parameters: parameters)
+        files = api_result.data
+        dir_files.concat(files.items)
+        page_token = files.next_page_token
+      end while page_token.to_s != ''
+      dir_files
+    end
+
+    def request_file(file_id)
+      result = send_request('@gdrive.files.get', {parameters: {'fileId' => file_id}})
+      result.data.to_hash
+    end
+
+    def request_download(file_id)
+      result = @gclient.execute(
+        api_method: @gdrive.files.get,
+        parameters: {'fileId' => file_id}
+      )
+      file = result.data
+      if file.download_url
+        result = @gclient.execute(:uri => file.download_url)
+        if result.status == 200
+          result.body
+        elsif result.status == 401
+          raise UnauthorizedException
+        else
+          raise HttpConnectionException
+        end
+      end
+    end
+
+    def request_upload(file_path, parent_id)
+      mime      = File.mime_type?(File.open(file_path))
+      mime_type = mime[0, mime.index(';')]
+      file = @gdrive.files.insert.request_schema.new({
+        'title' => file_path
+      })
+      if parent_id
+        file.parents = [{'id' => parent_id}]
+      end
+      media = Google::APIClient::UploadIO.new(file_path, mime_type)
+      result = @gclient.execute(
+        :api_method  => @gdrive.files.insert,
+        :body_object => file,
+        :media       => media,
+        :parameters  => {
+          'uploadType' => 'multipart',
+          'alt'        => 'json'
+        }
+      )
+      if result.status == 200
+        result
+      elsif result.status == 401
+        raise UnauthorizedException
+      else
+        raise HttpConnectionException
+      end
+    end
+
+    def request_delete(file_id)
+      send_request('@gdrive.files.delete', {parameters: {'fileId' => file_id}})
+      true
+    rescue => e
+      false
+    end
+
+    def request_mkdir(name, parent_id)
+      file = @gdrive.files.insert.request_schema.new({
+        'title'    => name,
+        'mimeType' => 'application/vnd.google-apps.folder',
+        'parents'  => [{'id' => parent_id}]
+      })
+      result = @gclient.execute(
+        :api_method => @gdrive.files.insert,
+        :body_object => file
+      )
+      if result.status == 200
+        true
+      elsif result.status == 401
+        raise UnauthorizedException
+      else
+        raise HttpConnectionException
+      end
+    end
+
+    private
+
+    def send_request(api_method, parameters = {})
+      requests = {api_method: eval(api_method)}
+      requests.merge!(parameters)
+      result = @gclient.execute(requests)
+      if result.status == 200
+        result
+      elsif result.status == 401
+        raise UnauthorizedException
+      else
+        raise HttpConnectionException
+      end
+    end
+
+  end
+
+end

data/lib/cloud_door/onedrive.rb

@@ -28,11 +28,14 @@ module CloudDoor
       @root_id      = ROOT_ID
       @storage_name = STORAGE_NAME
       @session_id   = session_id
+      @client       = nil
     end
 
     def load_token
       token_file = File.basename(@token.token_file)
-      @token = Token.load_token(token_file, @config.data_path, @session_id)
+      @token     = Token.load_token(token_file, @config.data_path, @session_id)
+      @client    = OneDriveApi.new(@token.access_token)
+      @token
     end
 
     def refresh_token
@@ -45,13 +48,14 @@ module CloudDoor
       handle_exception(e)
     end
 
-   def login(login_account, login_password)
+    def login(login_account, login_password)
       @account.login_account  = login_account
       @account.login_password = login_password
       url  = login_browser
       info = request_get_token(url)
       raise NoDataException if info.nil?
       @session_id = reset_token(info)
+      @client = OneDriveApi.new(@token.access_token)
       items = pull_files
       @file_list.delete_file
       @file_list.write_file_list(items)
@@ -92,41 +96,34 @@ module CloudDoor
     end
 
     def request_user
-      api = OneDriveApi.new(@token.access_token)
-      api.request_user
+      @client.request_user
     end
 
     def request_dir
       file_id = @parent_id || @file_id || ROOT_ID
-      api     = OneDriveApi.new(@token.access_token)
-      api.request_dir(file_id)
+      @client.request_dir(file_id)
     end
 
     def request_file
-      api = OneDriveApi.new(@token.access_token)
-      api.request_file(@file_id)
+      @client.request_file(@file_id)
     end
 
     def request_download
-      api     = OneDriveApi.new(@token.access_token)
-      contens = api.request_download(@file_id)
+      contens = @client.request_download(@file_id)
       open("#{@file_name}", 'wb') { |file| file << contens }
     end
 
     def request_upload(file_path)
-      api = OneDriveApi.new(@token.access_token)
-      api.request_upload(file_path, @parent_id)
+      @client.request_upload(file_path, @parent_id)
     end
 
     def request_delete
-      api = OneDriveApi.new(@token.access_token)
-      api.request_delete(@file_id)
+      @client.request_delete(@file_id)
     end
 
     def request_mkdir
       parent_id = @parent_id || ROOT_ID
-      api       = OneDriveApi.new(@token.access_token)
-      api.request_mkdir(@mkdir_name, parent_id)
+      @client.request_mkdir(@mkdir_name, parent_id)
     end
 
     def pull_files

data/lib/cloud_door/token.rb

@@ -1,7 +1,7 @@
 module CloudDoor
   class Token
     attr_accessor :token_file, :token_type, :expires_in, :scope, :access_token,
-                  :refresh_token, :user_id, :token_name
+                  :refresh_token, :user_id, :token_name, :credentials
     attr_reader :data_path
 
     TOKEN_ITEMS = [
@@ -10,7 +10,8 @@ module CloudDoor
       'scope',
       'access_token',
       'refresh_token',
-      'user_id'
+      'user_id',
+      'credentials'
     ]
 
     def initialize(token_name, data_path, id = nil)

data/lib/cloud_door/version.rb

@@ -1,3 +1,3 @@
 module CloudDoor
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/cloud_door.rb

@@ -6,6 +6,8 @@ require 'cloud_door/console'
 require 'cloud_door/dropbox'
 require 'cloud_door/exceptions'
 require 'cloud_door/file_list'
+require 'cloud_door/google_drive'
+require 'cloud_door/google_drive_bridge'
 require 'cloud_door/onedrive'
 require 'cloud_door/onedrive_api'
 require 'cloud_door/token'

data/spec/cloud_door/console_spec.rb

@@ -191,6 +191,11 @@ EOF
   describe 'ls' do
     subject { console.ls(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'have files' do
       let(:file_name) { nil }
       let(:posit) do
@@ -259,11 +264,19 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'cd' do
     subject { console.cd(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'file_name not input' do
       let(:file_name) { nil }
       it do
@@ -330,11 +343,19 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'info' do
     subject { console.info(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'file_name not input' do
       let(:file_name) { nil }
       it do
@@ -377,12 +398,20 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'pwd' do
     subject { console.pwd }
     let(:console) { create_console }
     let(:posit) { '/top' }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     it do
       expect_any_instance_of(CloudDoor::OneDrive).to receive(:show_current_directory)
         .and_return(posit)
@@ -390,11 +419,19 @@ EOF
       expects = '/top'
       expect($terminal.output.string).to include(expects)
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'download' do
     subject { console.download(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'file_name not input' do
       let(:file_name) { nil }
       it do
@@ -447,11 +484,19 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'upload' do
     subject { console.upload(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'file_name not input' do
       let(:file_name) { nil }
       it do
@@ -534,11 +579,19 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'rm' do
     subject { console.rm(file_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'file_name not input' do
       let(:file_name) { nil }
       it do
@@ -591,11 +644,19 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 
   describe 'mkdir' do
     subject { console.mkdir(mkdir_name) }
     let(:console) { create_console }
+    let(:token) { console.drive.storage.token }
+    let(:token_file) { console.drive.storage.token.token_file }
+    before(:each) do
+      open(token_file, 'wb') { |file| file << Marshal.dump(token) }
+    end
     context 'mkdir_name not input' do
       let(:mkdir_name) { nil }
       it do
@@ -629,5 +690,8 @@ EOF
         expect($terminal.output.string).to include(expects)
       end
     end
+    after(:each) do
+      File.delete(token_file) if File.exist?(token_file)
+    end
   end
 end

data/spec/cloud_door/dropbox_spec.rb

@@ -1,6 +1,12 @@
 require 'spec_helper'
 
 describe 'Dropbox' do
+  let(:storage) do
+    storage = create_storage(CloudDoor::Dropbox)
+    storage.load_token
+    storage
+  end
+
   describe 'reset_token' do
     subject { storage.reset_token(token_value) }
     let(:storage) { create_storage(CloudDoor::Dropbox) }
@@ -23,7 +29,6 @@ describe 'Dropbox' do
 
   describe 'show_user' do
     subject { storage.show_user }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     context 'success' do
       let(:posit) { {'name' => 'dropbox'} }
       it do
@@ -36,7 +41,6 @@ describe 'Dropbox' do
 
   describe 'show_files' do
     subject { storage.show_files(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     context 'success' do
       let(:file_name) { nil }
@@ -90,7 +94,6 @@ describe 'Dropbox' do
 
   describe 'change_directory' do
     subject { storage.change_directory(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     context 'success' do
       let(:file_name) { 'folder1' }
@@ -153,13 +156,11 @@ describe 'Dropbox' do
 
   describe 'show_current_directory' do
     subject { storage.show_current_directory }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     it { is_expected.to eq('/top') }
   end
 
   describe 'show_property' do
     subject { storage.show_property(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     before(:each) do
       list = [{'items' => {'file1' => {'id' => '/file1', 'type' => 'file'}}}]
@@ -290,7 +291,6 @@ describe 'Dropbox' do
 
   describe 'download_file' do
     subject { storage.download_file(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     context 'success' do
       let(:file_name) { 'file1' }
@@ -334,7 +334,6 @@ describe 'Dropbox' do
 
   describe 'upload_file' do
     subject { storage.upload_file(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     let(:up_file) { 'upload' }
     context 'success' do
@@ -373,7 +372,6 @@ describe 'Dropbox' do
 
   describe 'delete_file' do
     subject { storage.delete_file(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     context 'success' do
       let(:file_name) { 'file1' }
@@ -412,7 +410,6 @@ describe 'Dropbox' do
 
   describe 'make_directory' do
     subject { storage.make_directory(mkdir_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     context 'success' do
       let(:mkdir_name) { 'folder1' }
@@ -443,7 +440,6 @@ describe 'Dropbox' do
 
   describe 'assign_upload_file_name' do
     subject { storage.assign_upload_file_name(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     context 'file' do
       let(:file_name) { 'testfile' }
       it { is_expected.to eq file_name }
@@ -462,7 +458,6 @@ describe 'Dropbox' do
 
   describe 'file_exist?' do
     subject { storage.file_exist?(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     before(:each) do
       list = [{'items' => {'file1' => {'id' => '/file1', 'type' => 'file'}}}]
@@ -502,7 +497,6 @@ describe 'Dropbox' do
 
   describe 'has_file?' do
     subject { storage.has_file?(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     before(:each) do
       file1   = {'id' => '/file1', 'type' => 'file'}
@@ -515,7 +509,12 @@ describe 'Dropbox' do
     context 'return true' do
       context 'count > 0' do
         let(:file_name) { 'folder1' }
-        let(:posit) { {'path' => '/foler1', 'count' => 5} }
+        let(:posit) do
+          {'contents' => [
+            {'path' => '/file1', 'name' => 'file1', 'is_dir' => false},
+            {'path' => '/folder1', 'name' => 'folder1', 'is_dir' => true},
+          ]}
+        end
         before(:each) do
           storage.stub(:file_exist?)
             .with(file_name)
@@ -536,7 +535,7 @@ describe 'Dropbox' do
       end
       context 'count == 0' do
         let(:file_name) { 'folder1' }
-        let(:posit) { {'path' => '/foler1', 'count' => 0} }
+        let(:posit) { {'contents' => []} }
         before(:each) do
           storage.stub(:file_exist?)
             .with(file_name)
@@ -574,7 +573,6 @@ describe 'Dropbox' do
 
   describe 'file?' do
     subject { storage.file?(file_name) }
-    let(:storage) { create_storage(CloudDoor::Dropbox) }
     let(:list_file) { storage.file_list.list_file }
     let(:access_token) { storage.token.access_token }
     before(:each) do