cloud_door 0.0.1

data/lib/cloud_door/onedrive_api.rb

@@ -0,0 +1,169 @@
+require 'logger'
+require 'open-uri'
+require 'rest_client'
+
+module CloudDoor
+  class OneDriveApi
+    attr_accessor :access_token
+
+    # domain for auth
+    AUTH_BASE = 'https://login.live.com/'
+    # URL for auth
+    AUTH_FORMAT = AUTH_BASE +
+      'oauth20_authorize.srf?client_id=%s&scope=%s&response_type=code&redirect_uri=%s'
+    # URL for get token
+    TOKEN_URL = AUTH_BASE + 'oauth20_token.srf'
+    # domain for action
+    ACTION_BASE = 'https://apis.live.net/v5.0/'
+    # URL for get user info
+    USER_FORMAT =  ACTION_BASE + 'me?access_token=%s'
+    # URL for get directory
+    DIR_FORMAT = ACTION_BASE + '%s/files?access_token=%s'
+    # URL for get file info
+    FILE_FORMAT = ACTION_BASE + '%s?access_token=%s'
+    # URL for download file
+    DOWNLOAD_FORMAT = ACTION_BASE + '%s/content?suppress_redirects=true&access_token=%s'
+    # URL for upload file
+    UPLOAD_FORMAT = ACTION_BASE + '%s/files?access_token=%s'
+    # URL for delete file
+    DELETE_FORMAT = ACTION_BASE + '%s?access_token=%s'
+    # URL for make directory
+    MKDIR_FORMAT = ACTION_BASE + '%s'
+    # update scope
+    UPDATE_SCOPE = 'wl.skydrive_update,wl.offline_access'
+    # log_file
+    LOG_FILE = './log/request.log'
+
+    def initialize(access_token)
+      @access_token = access_token
+    end
+
+    def request_get_token(code, client_id, client_secret, redirect_url)
+      post_body = {
+        client_id:     client_id,
+        client_secret: client_secret,
+        redirect_uri:  redirect_url,
+        code:          code,
+        grant_type:    'authorization_code'
+      }
+      header = {content_type: 'application/x-www-form-urlencoded'}
+      send_request(:post, TOKEN_URL, post_body, header)
+    end
+
+    def request_refresh_token(refresh_token, client_id, client_secret, redirect_url)
+      post_body = {
+        client_id:     client_id,
+        client_secret: client_secret,
+        redirect_uri:  redirect_url,
+        grant_type:    'refresh_token',
+        refresh_token: refresh_token
+      }
+      header = {content_type: 'application/x-www-form-urlencoded'}
+      send_request(:post, TOKEN_URL, post_body, header)
+    end
+
+    def request_user
+      url = USER_FORMAT % @access_token
+      send_request(:get, url)
+    end
+
+    def request_dir(file_id)
+      url = DIR_FORMAT % [file_id, @access_token]
+      send_request(:get, url)
+    end
+
+    def request_file(file_id)
+      url = FILE_FORMAT % [file_id, @access_token]
+      send_request(:get, url)
+    end
+
+    def request_download(file_id)
+      url  = DOWNLOAD_FORMAT % [file_id, @access_token]
+      info = send_request(:get, url)
+      key  = 'location'
+      raise NoDataException if info.nil? || !info.is_a?(Hash) || !info.key?(key)
+      file_url = info[key]
+      open(file_url).read
+    end
+
+    def request_upload(file_path, parent_id)
+      url = UPLOAD_FORMAT % [parent_id, @access_token]
+      send_request(:post_file, url, File.new(file_path, 'rb'))
+    end
+
+    def request_delete(file_id)
+      url = DELETE_FORMAT % [file_id, @access_token]
+      send_request(:delete, url)
+    end
+
+    def request_mkdir(name, parent_id)
+      url = MKDIR_FORMAT % parent_id
+      body = JSON('name' => name)
+      header = {
+        'Authorization' => "Bearer #{@access_token}",
+        'Content-Type'  => 'application/json'
+      }
+      send_request(:post, url, body, header)
+    end
+
+    class << self
+      def make_auth_url(client_id, redirect_url)
+        AUTH_FORMAT % [client_id, UPDATE_SCOPE, redirect_url]
+      end
+    end
+
+    private
+
+    def send_request(method, url, body = '', header = {})
+      # if method == :get
+      #   res = RestClient.get(url) do |response, request, result|
+      #     request_log(response, request, result)
+      #     response
+      #   end
+      # elsif method == :post
+      #   res = RestClient.post(url, body, header) do |response, request, result|
+      #     request_log(response, request, result)
+      #     response
+      #   end
+      # elsif method == :delete
+      #   res = RestClient.delete(url) do |response, request, result|
+      #     request_log(response, request, result)
+      #     response
+      #   end
+      # elsif method == :post_file
+      #   res = RestClient.post(url, :file => body) do |response, request, result|
+      #     request_log(response, request, result)
+      #     response
+      #   end
+      # end
+      if method == :get
+        res = RestClient.get(url)
+      elsif method == :post
+        res = RestClient.post(url, body, header)
+      elsif method == :delete
+        res = RestClient.delete(url)
+      elsif method == :post_file
+        res = RestClient.post(url, file: body)
+      end
+      # raise NoDataException if res.body.nil? || res.body.empty?
+      return if res.body.nil? || res.body.empty?
+      JSON.parse(res.body)
+    rescue => e
+      if e.is_a?(RestClient::Unauthorized)
+        raise UnauthorizedException
+      elsif e.class.to_s.include?('RestClient')
+        raise HttpConnectionException
+      else
+        raise
+      end
+    end
+
+    def request_log(response, request, result)
+      logger = Logger.new(LOG_FILE)
+      log = "request:\n#{request.args.inspect}\n"
+      log << "result:\n#{result.inspect}\n"
+      log << "response:\n#{JSON.parse(response.body).inspect}\n"
+      logger.info(log)
+    end
+  end
+end

data/lib/cloud_door/token.rb

@@ -0,0 +1,67 @@
+module CloudDoor
+  class Token
+    attr_accessor :token_file, :token_type, :expires_in, :scope, :access_token,
+                  :refresh_token, :user_id, :token_name
+    attr_reader :data_path
+
+    TOKEN_ITEMS = [
+      'token_type',
+      'expires_in',
+      'scope',
+      'access_token',
+      'refresh_token',
+      'user_id'
+    ]
+
+    def initialize(token_name, data_path, id = nil)
+      @token_name = token_name
+      @data_path  = data_path
+      if id.nil?
+        @token_file = @data_path + @token_name
+      else
+        token_dir = @data_path + "#{id}"
+        unless File.exists?(token_dir)
+          Dir.mkdir(token_dir)
+        end
+        @token_file = "#{token_dir}/#{@token_name}"
+      end
+    end
+
+    def set_locate(id)
+      token_dir = @data_path + "#{id}"
+      unless File.exists?(token_dir)
+        Dir.mkdir(token_dir)
+      end
+      @token_file = "#{token_dir}/#{@token_name}"
+    end
+
+    def set_attributes(attributes)
+      TOKEN_ITEMS.each do |item|
+        instance_variable_set("@#{item}", attributes[item]) if attributes.key?(item)
+      end
+    end
+
+    def write_token
+      marshal = Marshal.dump(self)
+      open(@token_file, 'wb') { |file| file << marshal }
+      true
+    rescue
+      false
+    end
+
+    def self.load_token(token_name, data_path, id = nil)
+      if id.nil?
+        token_file = data_path + token_name
+      else
+        token_file = data_path + "#{id}/#{token_name}"
+      end
+      return nil unless File.exist?(token_file)
+      marshal = File.open(token_file).read
+      token = Marshal.load(marshal)
+      return nil unless token.is_a?(Token)
+      token
+    rescue
+      nil
+    end
+  end
+end

data/lib/cloud_door/version.rb

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

data/spec/cloud_door/account_spec.rb

@@ -0,0 +1,96 @@
+require 'spec_helper'
+
+describe 'Account' do
+  let(:account_file) { './data/test.yml' }
+  describe 'isset_account?' do
+    subject { account.isset_account? }
+    let(:account) { Fabricate.build(:account, storage: 'onedrive', file: account_file) }
+    context 'account & password setted' do
+      before(:each) do
+        account.login_account  = 'login@onedrive.com'
+        account.login_password = 'onedrive.com'
+      end
+      it { is_expected.to be_truthy }
+    end
+    context 'account not setted' do
+      before(:each) do
+        account.login_account  = 'login@onedrive.com'
+        account.login_password = ''
+      end
+      it { is_expected.to be_falsey }
+    end
+    context 'password not setted' do
+      before(:each) do
+        account.login_account  = ''
+        account.login_password = 'onedrive.com'
+      end
+      it { is_expected.to be_falsey }
+    end
+  end
+
+  describe 'load_yaml' do
+    before(:all) do
+      accounts = {
+        'onedrive' => {
+          'login_account'  => 'login@onedrive.com',
+          'login_password' => 'onedrive'
+        },
+        'dropbox' => {
+          'login_account'  => 'login@dropbox.com',
+          'login_password' => 'dropbox'
+        }
+      }
+      open('./data/test.yml', 'wb') { |file| YAML.dump(accounts, file) }
+    end
+    before(:each) { account.load_yaml }
+    context 'onedrive' do
+      let(:account) { Fabricate.build(:account, storage: 'onedrive', file: account_file) }
+      it { expect(account.login_account).to eq 'login@onedrive.com' }
+      it { expect(account.login_password).to eq 'onedrive' }
+    end
+    context 'dropbox' do
+      let(:account) { Fabricate.build(:account, storage: 'dropbox', file: account_file) }
+      it { expect(account.login_account).to eq 'login@dropbox.com' }
+      it { expect(account.login_password).to eq 'dropbox' }
+    end
+    after(:all) do
+      File.delete('./data/test.yml') if File.exist?('./data/test.yml')
+    end
+  end
+
+  describe 'update_yaml' do
+    before(:each) do
+      File.delete(account_file) if File.exist?(account_file)
+      account.update_yaml(update_params)
+    end
+    context 'onedrive' do
+      let(:account) { Fabricate.build(:account, storage: 'onedrive', file: account_file) }
+      let(:update_params) do
+        {
+          'login_account'  => 'login@onedrive.com',
+          'login_password' => 'onedrive'
+        }
+      end
+      let(:load_file) { YAML.load_file(account_file)['onedrive'] }
+      it { expect(File.exist?(account_file)).to be_truthy }
+      it { expect(load_file['login_account']).to eq 'login@onedrive.com' }
+      it { expect(load_file['login_password']).to eq 'onedrive' }
+    end
+    context 'dropbox' do
+      let(:account) { Fabricate.build(:account, storage: 'dropbox', file: account_file) }
+      let(:update_params) do
+        {
+          'login_account'  => 'login@dropbox.com',
+          'login_password' => 'dropbox'
+        }
+      end
+      let(:load_file) { YAML.load_file(account_file)['dropbox'] }
+      it { expect(File.exist?(account_file)).to be_truthy }
+      it { expect(load_file['login_account']).to eq 'login@dropbox.com' }
+      it { expect(load_file['login_password']).to eq 'dropbox' }
+    end
+    after(:each) do
+      File.delete(account_file) if File.exist?(account_file)
+    end
+  end
+end

data/spec/cloud_door/cloud_storage_spec.rb

@@ -0,0 +1,10 @@
+require 'spec_helper'
+
+# CloudStorage class is abstract class.
+# then, test inherited subclass.
+describe 'CloudStorage' do
+  describe 'initialize' do
+    subject { CloudDoor::CloudStorage.new }
+    it { expect { subject }.to raise_error(CloudDoor::AbstractClassException) }
+  end
+end

data/spec/cloud_door/cloud_yaml_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+describe 'CloudYaml' do
+  describe 'load_yaml' do
+    subject { account.load_yaml }
+    account_file = './data/test.yml'
+    before(:all) do
+      accounts = {
+        'onedrive' => {
+          'login_account'  => 'login@onedrive.com',
+          'login_password' => 'onedrive'
+        }
+      }
+      open(account_file, 'wb') { |file| YAML.dump(accounts, file) }
+    end
+    context 'file & storage exists' do
+      let(:account) { Fabricate.build(:cloud_yaml, file: account_file) }
+      it { is_expected.to be_truthy }
+    end
+    context 'file not exists' do
+      let(:account) { Fabricate.build(:cloud_yaml, file: './data/example.yml') }
+      it { is_expected.to be_falsey }
+    end
+    context 'storage exists' do
+      let(:account) { Fabricate.build(:cloud_yaml, file: account_file, storage: 'example') }
+      it { is_expected.to be_falsey }
+    end
+    after(:all) do
+      File.delete(account_file) if File.exist?(account_file)
+    end
+  end
+end

data/spec/cloud_door/config_spec.rb

@@ -0,0 +1,95 @@
+require 'spec_helper'
+
+describe 'Config' do
+  describe 'load_yaml' do
+    let(:config_file) { './data/test.yml' }
+    before(:all) do
+      configs = {
+        'onedrive' => {
+          'client_id'     => '1234',
+          'client_secret' => 'abcd',
+          'redirect_url'  => 'onedrive'
+        },
+        'dropbox' => {
+          'client_id'     => '5678',
+          'client_secret' => 'efgh',
+          'redirect_url'  => 'dropbox'
+        }
+      }
+      open('./data/test.yml', 'wb') { |file| YAML.dump(configs, file) }
+    end
+    before(:each) { config.load_yaml }
+    context 'onedrive' do
+      let(:config) { Fabricate.build(:config, storage: 'onedrive', file: config_file) }
+      it { expect(config.client_id).to eq '1234' }
+      it { expect(config.client_secret).to eq 'abcd' }
+      it { expect(config.redirect_url).to eq 'onedrive' }
+    end
+    context 'dropbox' do
+      let(:config) { Fabricate.build(:config, storage: 'dropbox', file: config_file) }
+      it { expect(config.client_id).to eq '5678' }
+      it { expect(config.client_secret).to eq 'efgh' }
+      it { expect(config.redirect_url).to eq 'dropbox' }
+    end
+    after(:all) do
+      File.delete('./data/test.yml') if File.exist?('./data/test.yml')
+    end
+  end
+
+  describe 'update_yaml' do
+    let(:config_file) { './data/test.yml' }
+    before(:each) do
+      File.delete(config_file) if File.exist?(config_file)
+      account.update_yaml(update_params)
+    end
+    context 'onedrive' do
+      let(:account) { Fabricate.build(:config, storage: 'onedrive', file: config_file) }
+      let(:update_params) do
+        {
+          'client_id'     => '1234',
+          'client_secret' => 'abcd',
+          'redirect_url'  => 'onedrive'
+        }
+      end
+      let(:load_file) { YAML.load_file(config_file)['onedrive'] }
+      it { expect(File.exist?(config_file)).to be_truthy }
+      it { expect(load_file['client_id']).to eq '1234' }
+      it { expect(load_file['client_secret']).to eq 'abcd' }
+      it { expect(load_file['redirect_url']).to eq 'onedrive' }
+    end
+    context 'dropbox' do
+      let(:account) { Fabricate.build(:config, storage: 'dropbox', file: config_file) }
+      let(:update_params) do
+        {
+          'client_id'     => '5678',
+          'client_secret' => 'efgh',
+          'redirect_url'  => 'dropbox'
+        }
+      end
+      let(:load_file) { YAML.load_file(config_file)['dropbox'] }
+      it { expect(File.exist?(config_file)).to be_truthy }
+      it { expect(load_file['client_id']).to eq '5678' }
+      it { expect(load_file['client_secret']).to eq 'efgh' }
+      it { expect(load_file['redirect_url']).to eq 'dropbox' }
+    end
+    after(:each) do
+      File.delete(config_file) if File.exist?(config_file)
+    end
+  end
+
+  describe 'init?' do
+    subject { config.init? }
+    let(:config) { Fabricate.build(:config) }
+    context 'initialized' do
+      before(:each) do
+        config.client_id     = '1234'
+        config.client_secret = 'abcd'
+        config.redirect_url  = 'onedrive'
+      end
+      it { is_expected.to be_truthy }
+    end
+    context 'not initialized' do
+      it { is_expected.to be_falsey }
+    end
+  end
+end