databasedotcom_additions 0.5.1

data/lib/databasedotcom_additions/client.rb

@@ -0,0 +1,95 @@
+require 'rforce'
+require 'digest/md5'
+require 'rforce_proxy_patch'
+require 'proxy_patch'
+require 'persistent_hash'
+
+module Databasedotcom
+  class Client
+
+    attr_accessor :query_result_cache_enabled
+    attr_accessor :login_failed_user_message
+
+    alias_method :query_orig, :query
+
+    # opt username, password, sandbox
+    def self.username_password_login(opt)
+      client = Databasedotcom::Client.new :verify_mode => OpenSSL::SSL::VERIFY_NONE
+      client.host = opt[:sandbox] ? "test.salesforce.com" : "login.salesforce.com"
+      begin
+        binding = RForce::Binding.new("https://#{client.host}/services/Soap/u/20.0", nil)
+        response = binding.login(opt[:username], opt[:password])
+        token = response.loginResponse.result.sessionId
+        serverUrl = response.loginResponse.result.serverUrl
+        uri = URI(serverUrl)
+        client.username = opt[:username]
+        client.password = opt[:password]
+        client.version = '22.0'
+        client.oauth_token = token
+        client.instance_url = "#{uri.scheme}://#{uri.host}"
+      rescue => e
+        client.login_failed_user_message = e.message
+      end
+      client
+    end
+
+    def cache
+
+      if ! defined?(@global_cache)
+        cache_file_name = 'cache/global-cache.json'
+        @global_cache ||= UtilityPack::PersistentHash.new cache_file_name
+      end
+
+      @global_cache
+
+    end
+
+    # support for count query - SELECT COUNT() FROM Account_vod__c
+    def count(soql_expr)
+      result = http_get("/services/data/v#{self.version}/query", :q => soql_expr)
+      JSON.parse(result.body)['totalSize']
+    end
+
+    def query(soql)
+      resp = query_orig(soql)
+      resp
+    end
+
+    def query_raw(soql)
+
+      if query_result_cache_enabled
+        cache_key = "#{self.username}:#{soql}"
+        if cache[cache_key]
+          puts "cache get - #{cache_key}"
+          return cache[cache_key]
+        end
+      end
+
+
+      result = http_get("/services/data/v#{self.version}/query", :q => soql)
+      resp = JSON.parse(result.body)
+
+      if query_result_cache_enabled
+        puts "cache put - #{cache_key}"
+        cache[cache_key] = resp
+      end
+      resp
+    end
+
+    def next_page_raw(path)
+      result = http_get(path)
+      JSON.parse(result.body)
+    end
+
+    def query_all(soql_expr)
+      all_records = []
+      records = query(soql_expr)
+      while records.total_size > 0
+        all_records.concat(records)
+        records = records.next_page
+      end
+      all_records
+    end
+
+  end
+end

data/lib/databasedotcom_additions/persistent_hash.rb

@@ -0,0 +1,35 @@
+require 'rubygems'
+require 'json'
+
+module UtilityPack
+
+  class PersistentHash
+    def initialize path
+      @path = path
+    end
+
+    def json_parse_file(path)
+      JSON.parse(File.open(path).read)
+    end
+
+    def load
+      File.exists?(@path) ? json_parse_file(@path) : {}
+    end
+
+    def save h
+      File.open(@path, 'w') {|f| f.write(h.to_json)}
+    end
+
+    def [](key)
+      load[key]
+    end
+
+    def []=(key, value)
+      h = load
+      h[key] = value
+      save(h)
+    end
+
+  end
+
+end

data/lib/databasedotcom_additions/proxy_patch.rb

@@ -0,0 +1,29 @@
+require 'databasedotcom'
+require 'set_proxy_if_needed'
+
+# patch for using a proxy
+module Databasedotcom
+  class Client
+
+    def proxy
+      http_proxy = ENV["http_proxy"]
+      URI.parse(http_proxy) rescue nil
+    end
+
+    def https_request(host=nil)
+
+      if proxy
+        http = Net::HTTP::Proxy(proxy.host, proxy.port).new(host || URI.parse(self.instance_url).host, 443)
+      else
+        http = Net::HTTP.new(host || URI.parse(self.instance_url).host, 443)
+      end
+
+      http.tap do |http|
+        http.use_ssl = true
+        http.ca_file = self.ca_file if self.ca_file
+        http.verify_mode = self.verify_mode if self.verify_mode
+      end
+    end
+
+  end
+end

data/lib/databasedotcom_additions/rforce_proxy_patch.rb

@@ -0,0 +1,33 @@
+require 'rubygems'
+require 'rforce'
+require 'set_proxy_if_needed'
+
+module RForce
+
+  class Binding
+
+    def init_server(url)
+      @url = URI.parse(url)
+
+      if proxy
+        @server = Net::HTTP::Proxy(proxy.host, proxy.port).new(@url.host, 443)
+      else
+        @server = Net::HTTP.new(@url.host, 443)
+      end
+
+      #@server = Net::HTTP.new(@url.host, @url.port)
+      @server.use_ssl = @url.scheme == 'https'
+      @server.verify_mode = OpenSSL::SSL::VERIFY_NONE
+
+      # run ruby with -d or env variable SHOWSOAP=true to see SOAP wiredumps.
+      @server.set_debug_output $stderr if show_debug
+    end
+
+    def proxy
+      http_proxy = ENV['http_proxy']
+      URI.parse(http_proxy) rescue nil
+    end
+
+  end
+
+end

data/lib/databasedotcom_additions/set_proxy_if_needed.rb

@@ -0,0 +1,14 @@
+def on_work_network
+  if RUBY_PLATFORM =~ /32/ # windows
+    %x{ipconfig /all}.split.grep(/54\./).count > 0
+  else
+    %x{ifconfig | grep -i "inet 54." > /dev/null}
+    $? == 0
+  end
+end
+
+if on_work_network
+  ENV["http_proxy"] = "http://webproxy.merck.com:8080"
+else
+  ENV.delete("http_proxy")
+end

data/lib/databasedotcom_additions.rb

@@ -0,0 +1 @@
+require 'databasedotcom_additions/client.rb'

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification
+name: databasedotcom_additions
+version: !ruby/object:Gem::Version
+  version: 0.5.1
+  prerelease: 
+platform: ruby
+authors:
+- Brian Pfeil
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: databasedotcom
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rforce
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: databasedotcom additions
+email: brian.pfeil@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/databasedotcom_additions/client.rb
+- lib/databasedotcom_additions/persistent_hash.rb
+- lib/databasedotcom_additions/proxy_patch.rb
+- lib/databasedotcom_additions/rforce_proxy_patch.rb
+- lib/databasedotcom_additions/set_proxy_if_needed.rb
+- lib/databasedotcom_additions.rb
+homepage: http://rubygems.org/gems/databasedotcom_additions
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: databasedotcom additions
+test_files: []