cloud_door 0.0.1

data/lib/cloud_door/exceptions.rb

@@ -0,0 +1,153 @@
+module CloudDoor
+  class FileNameEmptyException < StandardError
+    def message
+      'filename is not set.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class DirectoryNameEmptyException < StandardError
+    def message
+      'directory name is not set.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class SetIDException < StandardError
+    def message
+      'not found input file name.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class NotFileException < StandardError
+    def message
+      'target is not file.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class NotDirectoryException < StandardError
+    def message
+      'target is not directory.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class NoDataException < StandardError
+    def message
+      'not found request data.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class TokenClassException < StandardError
+    def message
+      'token is not Token Class.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class FileNotExistsException < StandardError
+  end
+
+  class RequestMethodNotFoundException < StandardError
+  end
+
+  class RequestPropertyNotFoundException < StandardError
+  end
+
+  class AccessCodeNotIncludeException < StandardError
+    def message
+      'access code not include in redirect url.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class UnauthorizedException < StandardError
+    def message
+      'authentication failed.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class HttpConnectionException < StandardError
+    def message
+      'access to storage failed.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class UploadFailedException < StandardError
+    def message
+      'upload file to storage failed.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class DeleteFailedException < StandardError
+    def message
+      'delete file from storage failed.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class MkdirFailedException < StandardError
+    def message
+      'make directory on storage failed.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class AbstractClassException < StandardError
+    def message
+      'this class is abstract class. please make inherited subclass.'
+    end
+
+    def to_s
+      message
+    end
+  end
+
+  class AbstractMethodException < StandardError
+  end
+end

data/lib/cloud_door/file_list.rb

@@ -0,0 +1,164 @@
+module CloudDoor
+  class FileList
+    attr_accessor :list_file, :list_name
+    attr_reader :list
+
+    def initialize(list_name, data_path, id = nil)
+      @list_name = list_name
+      @data_path = data_path
+      if (id.nil?)
+        @list_file = @data_path + @list_name
+      else
+        list_dir = @data_path + "#{id}"
+        unless File.exists?(list_dir)
+          Dir.mkdir(list_dir)
+        end
+        @list_file = "#{list_dir}/#{@list_name}"
+      end
+      @list = nil
+    end
+
+    def set_locate(id)
+      list_dir = @data_path + "#{id}"
+      unless File.exists?(list_dir)
+        Dir.mkdir(list_dir)
+      end
+      @list_file = "#{list_dir}/#{@list_name}"
+    end
+
+    def load_list
+      list = []
+      if File.exist?(@list_file)
+        marshal = File.open(@list_file).read
+        list = Marshal.load(marshal)
+        return false unless list.is_a?(Array)
+      end
+      @list = list
+      true
+    end
+
+    def write_file_list(items, file_id = '', file_name = '')
+      return false if load_list == false
+      if @list.empty?
+        add_list_top(items)
+      elsif file_name =~ CloudStorage::PARENT_DIR_PAT
+        back = file_name.scan(CloudStorage::PARENT_DIR_PAT).size + 1
+        remove_list(back)
+      else
+        if file_name.nil? || file_name.empty?
+          update_list(items)
+        else
+          add_list(items, file_id, file_name)
+        end
+      end
+    end
+
+    def add_list_top(items)
+      return false if items.nil? || !items.is_a?(Hash)
+      @list = []
+      @list << {'id' => 'top', 'name' => 'top', 'items' => items}
+      write_file
+    end
+
+    def add_list(items, file_id, file_name)
+      return false if items.nil? || !items.is_a?(Hash)
+      return false if file_id.nil? || file_id.empty?
+      return false if file_name.nil? || file_name.empty?
+      return false if load_list == false
+      @list << {'id' => file_id, 'name' => file_name, 'items' => items}
+      write_file
+    end
+
+    def update_list(items)
+      return false if items.nil? || !items.is_a?(Hash)
+      return false if load_list == false
+      last_node = @list[-1]
+      @list[-1] = {'id' => last_node['id'], 'name' => last_node['name'], 'items' => items}
+      write_file
+    end
+
+    def remove_list(back)
+      return false if load_list == false
+      size = @list.size
+      return false if back > size
+      last = size - back
+      @list = @list[0..last]
+      write_file
+    end
+
+    def delete_file
+      File.delete(@list_file) if File.exist?(@list_file)
+      true
+    rescue
+      false
+    end
+
+    def pull_parent_id
+      convert_name_to_id('current')
+    end
+
+    def pull_current_dir
+      return false if load_list == false
+      return '/top' if @list.size < 2
+      files = []
+      @list.each { |part| files << part['name'] unless part['name'].nil? }
+      '/' + files.join('/')
+    end
+
+    def pull_file_properties(file_name)
+      if @list.nil?
+        return false if load_list == false
+      end
+      return false if @list.empty?
+      items = @list.last['items']
+      return false if items.empty? || !items.key?(file_name)
+      items[file_name]
+    end
+
+    def convert_name_to_id(mode, file_name = '')
+      return false if load_list == false
+      return false unless %w(current parent target).include?(mode)
+      send("convert_#{mode}_id", file_name)
+    end
+
+    def top?(file_name)
+      back = (file_name.scan(CloudStorage::PARENT_DIR_PAT).size) + 1
+      @list.count < back
+    end
+
+    private
+
+    def write_file
+      marshal = Marshal.dump(@list)
+      open(@list_file, 'wb') { |file| file << marshal }
+      true
+    rescue
+      false
+    end
+
+    def convert_current_id(*)
+      if @list.count < 2
+        nil
+      else
+        @list.last['id']
+      end
+    end
+
+    def convert_parent_id(file_name)
+      back = (file_name.scan(CloudStorage::PARENT_DIR_PAT).size) + 1
+      return false if @list.size < back
+      last = @list.size - back
+      if last == 0
+        nil
+      else
+        @list[list.size - back]['id']
+      end
+    end
+
+    def convert_target_id(file_name)
+      properties = pull_file_properties(file_name)
+      return false unless properties
+      properties['id']
+    end
+  end
+end

data/lib/cloud_door/onedrive.rb

@@ -0,0 +1,180 @@
+require 'cloud_door/cloud_storage'
+require 'cloud_door/onedrive_api'
+
+module CloudDoor
+  class OneDrive < CloudStorage
+    attr_accessor :token, :file_list, :file_id, :file_name,
+                  :up_file_name, :mkdir_name, :parent_id
+    attr_reader :config, :account
+
+    # user root file_id
+    ROOT_ID = 'me/skydrive'
+    # onedrive login site components
+    LOGIN_COMPONENTS = {
+      'account_text_name'  => 'login',
+      'password_text_name' => 'passwd',
+      'signin_button_id'   => 'idSIButton9',
+      'accept_button_id'   => 'idBtn_Accept'
+    }
+    TIME_PROPERTY_PAT = /_time$/
+    STORAGE_NAME = 'OneDrive'
+
+    def initialize(session_id = nil)
+      @config       = Config.new('onedrive')
+      @account      = Account.new('onedrive', @config.data_path)
+      @token        = Token.new('onedrive_token', @config.data_path, session_id)
+      @file_list    = FileList.new('onedrive_list', @config.data_path, session_id)
+      @file_id      = nil
+      @root_id      = ROOT_ID
+      @storage_name = STORAGE_NAME
+      @session_id   = session_id
+    end
+
+    def load_token
+      token_file = File.basename(@token.token_file)
+      @token = Token.load_token(token_file, @config.data_path, @session_id)
+    end
+
+    def refresh_token
+      raise TokenClassException unless @token.is_a?(Token)
+      info = request_refresh_token
+      raise NoDataException if info.nil?
+      @token.set_attributes(info)
+      @token.write_token
+    rescue => e
+      handle_exception(e)
+    end
+
+   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)
+      items = pull_files
+      @file_list.delete_file
+      @file_list.write_file_list(items)
+      if @config.session_use?
+        @session_id
+      else
+        true
+      end
+    rescue => e
+      handle_exception(e)
+    end
+
+    private
+
+    def request_get_token(url)
+      query  = URI.parse(url).query
+      raise AccessCodeNotIncludeException if query.nil?
+      params = CGI.parse(query)
+      raise AccessCodeNotIncludeException unless params.key?('code')
+      code = params['code'][0]
+      api  = OneDriveApi.new(@token.access_token)
+      api.request_get_token(
+        code,
+        @config.client_id,
+        @config.client_secret,
+        @config.redirect_url
+      )
+    end
+
+    def request_refresh_token
+      api = OneDriveApi.new(@token.access_token)
+      api.request_refresh_token(
+        @token.refresh_token,
+        @config.client_id,
+        @config.client_secret,
+        @config.redirect_url
+      )
+    end
+
+    def request_user
+      api = OneDriveApi.new(@token.access_token)
+      api.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)
+    end
+
+    def request_file
+      api = OneDriveApi.new(@token.access_token)
+      api.request_file(@file_id)
+    end
+
+    def request_download
+      api     = OneDriveApi.new(@token.access_token)
+      contens = api.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)
+    end
+
+    def request_delete
+      api = OneDriveApi.new(@token.access_token)
+      api.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)
+    end
+
+    def pull_files
+      dir = pick_cloud_info('request_dir', 'data')
+      return {} if dir.nil? || !dir.is_a?(Array) || dir.count == 0
+      items = {}
+      dir.each do |item|
+        type = get_type_from_id(item['id'])
+        items[item['name']] = {'id' => item['id'], 'type' => type}
+      end
+      items
+    end
+
+    def format_property(info)
+      items = {}
+      info.each do |key, val|
+        if key =~ TIME_PROPERTY_PAT
+          items[key] = DateTime.parse(info[key]).strftime('%Y-%m-%d %H:%M:%S')
+        else
+          items[key] = info[key]
+        end
+      end
+      items
+    end
+
+    def get_type_from_id(file_id)
+      file_id.split('.')[0]
+    end
+
+    def login_browser
+      auth_url = OneDriveApi.make_auth_url(@config.client_id, @config.redirect_url)
+      browser  = Watir::Browser.new :phantomjs
+      # input account
+      browser.goto(auth_url)
+      browser.wait
+      browser.text_field(:name, LOGIN_COMPONENTS['account_text_name']).set @account.login_account
+      browser.text_field(:name, LOGIN_COMPONENTS['password_text_name']).set @account.login_password
+      browser.button(:id, LOGIN_COMPONENTS['signin_button_id']).click
+      browser.wait
+      # allow access
+      browser.button(:id, LOGIN_COMPONENTS['accept_button_id']).click
+      browser.wait
+      # get redirect url
+      url = browser.url
+      browser.close
+      url
+    rescue => e
+      handle_exception(e)
+    end
+  end
+end