cloud_door 0.0.1

data/lib/cloud_door/cloud_yaml.rb

@@ -0,0 +1,45 @@
+require 'yaml'
+
+module CloudDoor
+  class CloudYaml
+    attr_accessor :file, :items, :storage
+
+    def load_yaml
+      return false unless File.exist?(@file)
+      all_config = YAML.load_file(@file)
+      return false unless all_config.is_a?(Hash)
+      return false unless all_config.key?(@storage)
+      config = all_config[@storage]
+      @items.each do |item|
+        instance_variable_set("@#{item}", config[item]) if config.key?(item)
+      end
+      true
+    end
+
+    def update_yaml(update_params)
+      if File.exist?(@file)
+        all_config = YAML.load_file(@file)
+        if all_config.key?(@storage)
+          config = all_config[@storage]
+        else
+          config = {}
+        end
+      else
+        all_config = {}
+        config     = {}
+      end
+      @items.each do |item|
+        next unless update_params.key?(item)
+        config[item] = update_params[item]
+      end
+      begin
+        all_config[@storage] = config
+        open(@file, 'wb') { |f| YAML.dump(all_config, f) }
+        load_yaml
+        true
+      rescue
+        false
+      end
+    end
+  end
+end

data/lib/cloud_door/config.rb

@@ -0,0 +1,61 @@
+require 'cloud_door/cloud_yaml'
+
+module CloudDoor
+  class Config < CloudYaml
+    attr_accessor :client_id, :client_secret, :redirect_url
+    attr_reader :data_path
+
+    CONFIG_FILE = './cloud_door_config.yml'
+
+    GLOBAL_CONFIG_ITEMS = [
+      'data_path',
+      'sessoin_flg',
+    ]
+
+    CONFIG_ITEMS = [
+      'client_id',
+      'client_secret',
+      'redirect_url'
+    ]
+
+    SESSION_FLG_ON  = '1'
+    SESSION_FLG_OFF = '0'
+
+    def initialize(storage)
+      @storage       = storage
+      @data_path     = './data/'
+      @session_flg   = SESSION_FLG_OFF
+      @file          = CONFIG_FILE
+      @items         = CONFIG_ITEMS
+      @client_id     = ''
+      @client_secret = ''
+      @redirect_url  = ''
+      load_global_config
+      load_yaml
+    end
+
+    def load_global_config
+      return false unless File.exist?(@file)
+      all_config = YAML.load_file(@file)
+      return false unless all_config.is_a?(Hash)
+      return false unless all_config.key?('global')
+      config = all_config['global']
+      GLOBAL_CONFIG_ITEMS.each do |item|
+        instance_variable_set("@#{item}", config[item]) if config.key?(item)
+      end
+      true
+    end
+
+    def init?
+      CONFIG_ITEMS.each do |item|
+        val = instance_variable_get("@#{item}")
+        return false if val.nil? || val.empty?
+      end
+      true
+    end
+
+    def session_use?
+      (@session_flg.to_s == SESSION_FLG_ON.to_s)
+    end
+  end
+end

data/lib/cloud_door/console.rb

@@ -0,0 +1,334 @@
+require 'highline/import'
+require 'cloud_door'
+
+module CloudDoor
+  class Console
+    attr_accessor :drive
+    attr_reader :storage_name
+
+    def initialize(drive, session_id = nil)
+      @drive = CloudDoor.new(drive, session_id)
+      @storage_name = @drive.show_storage_name
+    end
+
+    def config(show = false)
+      if show
+        show_configuration(@drive.show_configuration)
+        exit
+      end
+      configs = ask_configuration
+      result  = @drive.update_configuration(configs)
+      show_result_message(result, 'update configuration')
+    end
+
+    def account(show = false)
+      if show
+        show_account(@drive.show_account)
+        exit
+      end
+      accounts = ask_account
+      result   = @drive.update_account(accounts)
+      show_result_message(result, 'update account')
+    end
+
+    def login(default = false)
+      unless @drive.configuration_init?
+        say(make_message(:not_configuration, @storage_name.downcase))
+        exit
+      end
+      exit unless agree(make_message(:agree_access, @storage_name))
+      account = login_set_account(default)
+      say(make_message(:start_connection, @storage_name))
+      if @drive.login(account['login_account'], account['login_password'])
+        user      = @drive.show_user
+        user_name = user['name'] || user['display_name']
+        say(make_message(:result_success, 'login'))
+        say(make_message(:login_as, user_name))
+        say("\n")
+        pwd  = @drive.show_current_directory
+        list = @drive.show_files
+        show_file_list(pwd, list)
+      else
+        say(make_message(:result_fail, 'login'))
+      end
+    rescue => e
+      show_exception(e)
+    end
+
+    def ls(file_name)
+      @drive.load_token
+      fullname = make_fullname(file_name)
+      if !file_name.nil? && !file_name.empty?
+        unless @drive.file_exist?(file_name)
+          say(make_message(:file_not_exists, fullname, @storage_name))
+          exit
+        end
+      end
+      list = @drive.show_files(file_name)
+      show_file_list(fullname, list)
+    rescue => e
+      show_exception(e)
+    end
+
+    def cd(file_name)
+      if (file_name.nil? || file_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      fullname = make_fullname(file_name)
+      unless @drive.file_exist?(file_name)
+        say(make_message(:file_not_exists, fullname, @storage_name))
+        exit
+      end
+      say(make_message(:move_to, fullname))
+      list = @drive.change_directory(file_name)
+      show_file_list(fullname, list)
+    rescue => e
+      show_exception(e)
+    end
+
+    def info(file_name)
+      if (file_name.nil? || file_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      fullname = make_fullname(file_name)
+      unless @drive.file_exist?(file_name)
+        say(make_message(:file_not_exists, fullname, @storage_name))
+        exit
+      end
+      info = @drive.show_property(file_name)
+      unless (info.empty?)
+        say(make_message(:show_information, fullname))
+        max = info.max { |a, b| a[0].length <=> b[0].length }
+        max_len = max[0].length
+        info.each do |key, value|
+          say("  #{key.ljust(max_len)} : #{value}")
+        end
+      end
+    rescue => e
+      show_exception(e)
+    end
+
+    def pwd
+      @drive.load_token
+      say(@drive.show_current_directory)
+    end
+
+    def download(file_name)
+      if (file_name.nil? || file_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      fullname = make_fullname(file_name)
+      unless @drive.file_exist?(file_name)
+        say(make_message(:file_not_exists, fullname, @storage_name))
+        exit
+      end
+      if File.exist?(file_name)
+        say(make_message(:same_file_exist, file_name, 'local'))
+        exit unless agree(make_message(:agree_overwrite, file_name))
+      end
+      result = @drive.download_file(file_name)
+      show_result_message(result, "'#{file_name}' download")
+    rescue => e
+      show_exception(e)
+    end
+
+    def upload(file_name)
+      if (file_name.nil? || file_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      unless File.exists?(file_name)
+        say(make_message(:file_not_exists, file_name, 'local'))
+        exit
+      end
+      if File.directory?(file_name)
+        say(make_message(:is_directory, file_name))
+        say(make_message(:compress_to, file_name))
+        say("\n")
+      end
+      up_file  = @drive.assign_upload_file_name(file_name)
+      fullname = make_fullname(up_file)
+      if @drive.file_exist?(up_file)
+        say(make_message(:same_file_exist, fullname, @storage_name))
+        exit unless agree(make_message(:agree_overwrite, fullname))
+      end
+      result = @drive.upload_file(file_name)
+      show_result_message(result, "'#{fullname}' upload")
+    rescue => e
+      show_exception(e)
+    end
+
+    def rm(file_name)
+      if (file_name.nil? || file_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      fullname = make_fullname(file_name)
+      exit unless agree(make_message(:agree_delete, fullname))
+      unless @drive.file_exist?(file_name)
+        say(make_message(:file_not_exists, fullname, @storage_name))
+        exit
+      end
+      if @drive.has_file?(file_name)
+        say(make_message(:has_files, fullname))
+        # exit unless agree("Do you want to delete these files (Y/N)?")
+        exit
+      end
+      result = @drive.delete_file(file_name)
+      show_result_message(result, "'#{fullname}' delete")
+    rescue => e
+      show_exception(e)
+    end
+
+    def mkdir(mkdir_name)
+      if (mkdir_name.nil? || mkdir_name.empty?)
+        say(make_message(:wrong_parameter, 'file name'))
+        exit
+      end
+      @drive.load_token
+      fullname = make_fullname(mkdir_name)
+      if @drive.file_exist?(mkdir_name)
+        say(make_message(:same_file_exist, fullname, @storage_name))
+        exit
+      end
+      result = @drive.make_directory(mkdir_name)
+      show_result_message(result, "make '#{fullname}' directory")
+    rescue => e
+      show_exception(e)
+    end
+
+    private
+
+    def make_message(type, *args)
+      messages = {
+        show_configuration: "#{args[0]} configuration are",
+        show_account:       "#{args[0]} account are",
+        ask_configuration:  "please enter the #{args[0]} configuration.",
+        ask_account:        "please enter the #{args[0]} account.",
+        not_configuration:  "config is not found. please execute './#{args[0]} config' before.",
+        account_found:      "found defaulut account '#{args[0]}'. use this account.",
+        account_not_found:  "default account is not found. please execute './#{args[0]} account' before.",
+        start_connection:   "start a connection to the #{args[0]}. please wait a few seconds.",
+        login_as:           "login as #{args[0]}.",
+        show_files:         "you have these files on '#{args[0]}'.",
+        show_no_file:       "you have no file on '#{args[0]}'.",
+        show_information:   "information of '#{args[0]}'.",
+        move_to:            "move to '#{args[0]}'.",
+        is_directory:       "'#{args[0]}' is a directory.",
+        compress_to:        "upload as '#{args[0]}.zip'.",
+        has_files:          "'#{args[0]}' has files.",
+        wrong_parameter:    "this command needs #{args[0]}.",
+        file_not_exists:    "'#{args[0]}' not exists in #{args[1]}.",
+        same_file_exist:    "'#{args[0]}' already exists in #{args[1]}.",
+        agree_access:       "do you want to allow access to the #{args[0]} from this system(Y/N)?",
+        agree_overwrite:    "do you want to overwrite '#{args[0]}' (Y/N)?",
+        agree_delete:       "do you want to delete '#{args[0]}' (Y/N)?",
+        result_success:     "#{args[0]} success.",
+        result_fail:        "#{args[0]} fail.",
+        result_exception:   "command fail.",
+        login_fail:         "login fail.",
+        unauthorized:       "please execute #{args[0]} auth' command."
+      }
+      if messages.key?(type)
+        messages[type]
+      end
+    end
+
+    def make_fullname(file_name)
+      pwd = @drive.show_current_directory
+      if file_name.nil? || file_name.empty?
+        pwd
+      else
+        "#{pwd}/#{file_name}"
+      end
+    end
+
+    def show_configuration(config)
+      say(make_message(:show_configuration, @storage_name))
+      say("  client_id    : #{config.client_id}\n")
+      say("  client_secret: #{config.client_secret}\n")
+      say("  redirect_url : #{config.redirect_url}\n")
+    end
+
+    def ask_configuration
+      say(make_message(:ask_configuration, @storage_name))
+      configs = {}
+      configs['client_id']      = ask('client_id     : ')
+      configs['client_secret']  = ask('client_secret : ')
+      configs['redirect_url']   = ask('redirect_url  : ')
+      configs.each { |key, val| configs[key] = val.to_s }
+    end
+
+    def show_account(account)
+      say(make_message(:show_account, @storage_name))
+      say("  login_account : #{account.login_account}\n")
+      say("  login_password: #{account.login_password}\n")
+    end
+
+    def ask_account
+      say(make_message(:ask_account, @storage_name))
+      accounts = {}
+      accounts['login_account']  = ask('login_account : ')
+      accounts['login_password'] = ask('login_password: ') { |q| q.echo = '*' }
+      accounts.each { |key, val| accounts[key] = val.to_s }
+    end
+
+    def login_set_account(default)
+      account = Hash.new
+      if default
+        if @drive.isset_account?
+          default_account = @drive.show_account
+          account['login_account']  = default_account.login_account
+          account['login_password'] = default_account.login_password
+          say(make_message(:account_found, account['login_account']))
+        else
+          say(make_message(:account_not_found, @storage_name.downcase))
+          exit
+        end
+      else
+        say(make_message(:ask_account, @storage_name))
+        login_account  = ask('login_account : ')
+        login_password = ask('login_password: ') { |q| q.echo = '*' }
+        account['login_account']  = login_account
+        account['login_password'] = login_password
+      end
+      account
+    end
+
+    def show_file_list(pwd, list)
+      if list.count > 0
+        say(make_message(:show_files, pwd))
+        list.each do |name, properties|
+          type = properties['type'].ljust(6)
+          say("[#{type}] #{name}")
+        end
+      else
+        say(make_message(:show_no_file, pwd))
+      end
+    end
+
+    def show_result_message(result, target)
+      if result
+        say(make_message(:result_success, target))
+      else
+        say(make_message(:result_fail, target))
+      end
+    end
+
+    def show_exception(e)
+      say(make_message(:result_exception))
+      say(e.message)
+      if e.is_a?(UnauthorizedException)
+        say(make_message(:unauthorized, @storage_name.downcase))
+      end
+    end
+  end
+end

data/lib/cloud_door/dropbox.rb

@@ -0,0 +1,166 @@
+require 'dropbox_sdk'
+require 'cloud_door/cloud_storage'
+
+module CloudDoor
+  class Dropbox < CloudStorage
+    attr_accessor :token, :file_list, :file_id, :file_name,
+                  :up_file_name, :mkdir_name, :parent_id
+    attr_reader :config, :account
+
+    # dropbox login site components
+    LOGIN_COMPONENTS = {
+      'account_text_name'   => 'login_email',
+      'password_text_name'  => 'login_password',
+      'signin_button_class' => 'login-button',
+      'accept_button_name'  => 'allow_access',
+      'auth_code_id'        => 'auth-code'
+    }
+
+    TIME_PROPERTIES = %w(client_mtime modified)
+    CONTENTS_KEY    = 'contents'
+    ROOT_ID         = '/'
+    STORAGE_NAME    = 'Dropbox'
+
+    def initialize(session_id = nil)
+      @config       = Config.new('dropbox')
+      @account      = Account.new('dropbox', @config.data_path)
+      @token        = Token.new('dropbox_token', @config.data_path, session_id)
+      @file_list    = FileList.new('dropbox_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 login(login_account, login_password)
+      @account.login_account  = login_account
+      @account.login_password = login_password
+      flow = DropboxOAuth2FlowNoRedirect.new(@config.client_id, @config.client_secret)
+      code = login_browser(flow.start())
+      access_token, user_id = flow.finish(code)
+      raise NoDataException if access_token.nil?
+      @session_id = reset_token({'access_token' => access_token})
+      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_user
+      client = DropboxClient.new(@token.access_token)
+      info   = client.account_info()
+      raise NoDataException if info.nil? || info.empty?
+      info
+    end
+
+    def request_dir
+      file_id = @parent_id || @file_id || ROOT_ID
+      client  = DropboxClient.new(@token.access_token)
+      client.metadata(file_id)
+    end
+
+    def request_file
+      client = DropboxClient.new(@token.access_token)
+      client.metadata(@file_id)
+    end
+
+    def request_download
+      client = DropboxClient.new(@token.access_token)
+      contents, metadata = client.get_file_and_metadata(@file_id)
+      open(@file_name, 'w') {|f| f.puts contents }
+      metadata
+    end
+
+    def request_upload(file_path)
+      client = DropboxClient.new(@token.access_token)
+      if @parent_id == ROOT_ID
+        to_path = @parent_id + file_path
+      else
+        to_path = "#{@parent_id}/#{file_path}"
+      end
+      response = client.put_file(to_path, open(file_path))
+      raise UploadFailedException if response.nil? || response.empty?
+      response
+    end
+
+    def request_delete
+      client = DropboxClient.new(@token.access_token)
+      response = client.file_delete(@file_id)
+      raise DeleteFailedException if response.nil? || response.empty?
+      response
+    end
+
+    def request_mkdir
+      client = DropboxClient.new(@token.access_token)
+      if @parent_id == ROOT_ID
+        path = @parent_id + @mkdir_name
+      else
+        path = "#{@parent_id}/#{@mkdir_name}"
+      end
+      response = client.file_create_folder(path)
+      raise MkdirFailedException if response.nil? || response.empty?
+      response
+    end
+
+    def pull_files
+      dir = pick_cloud_info('request_dir', 'contents')
+      return {} if dir.nil? || !dir.is_a?(Array) || dir.count == 0
+      items = {}
+      dir.each do |item|
+        path, name = File.split(item['path'])
+        type = item['is_dir'] ? 'folder' : 'file'
+        items[name] = {'id' => item['path'], 'type' => type}
+      end
+      items
+    end
+
+    def format_property(info)
+      path, name = File.split(info['path'])
+      items = {}
+      items['name'] = name
+      info.each do |key, val|
+        if TIME_PROPERTIES.include?(key)
+          items[key] = DateTime.parse(val).strftime('%Y-%m-%d %H:%M:%S')
+        elsif key == CONTENTS_KEY
+          items['count'] = val.count
+        else
+          items[key] = val
+        end
+      end
+      items
+    end
+
+    def login_browser(auth_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(:class, LOGIN_COMPONENTS['signin_button_class']).click
+      browser.button(:name => LOGIN_COMPONENTS['accept_button_name']).wait_until_present
+      # allow access
+      browser.button(:name, LOGIN_COMPONENTS['accept_button_name']).click
+      browser.wait
+      # get code
+      code = browser.div(:id, LOGIN_COMPONENTS['auth_code_id']).text
+      browser.close
+      code
+    rescue => e
+      handle_exception(e)
+    end
+  end
+end