sobjectmodel 0.1.2

data/lib/sobject_model/rest/client.rb

@@ -0,0 +1,60 @@
+require 'uri'
+require 'net/http'
+require 'json'
+require 'cgi'
+
+require_relative 'http'
+require_relative 'query_result'
+require_relative 'errors'
+
+module SObjectModel
+  module Rest
+    class Client
+      attr_reader :instance_url, :access_token, :api_version
+
+      def initialize(instance_url:, access_token:, api_version:)
+        @instance_url = instance_url
+        @access_token = access_token
+        @api_version = api_version
+        @http = Http.new(instance_url, access_token)
+      end
+
+      def query(soql)
+        response = http.get "/services/data/v#{api_version}/query?q=#{CGI.escape(soql)}"
+        QueryResult.new(response)
+      end
+
+      def describe(object_type)
+        response = http.get "/services/data/v#{api_version}/sobjects/#{object_type.to_sym}/describe"
+        JSON.parse(response)
+      end
+
+      def find(object_type, id, fields: [])
+        query = fields.empty? ? '' : %|?fields=#{fields.map(&:to_s).join(',')}|
+        response = http.get "/services/data/v#{api_version}/sobjects/#{object_type}/#{id}#{query}"
+        JSON.parse(response)
+      end
+
+      def create(object_type, values)
+        response = http.post "/services/data/v#{api_version}/sobjects/#{object_type}/", values
+        JSON.parse(response)['id']
+      end
+
+      def update(object_type, id, values)
+        http.patch "/services/data/v#{api_version}/sobjects/#{object_type}/#{id}/", values
+        id
+      end
+
+      def delete(object_type, id)
+        http.delete "/services/data/v#{api_version}/sobjects/#{object_type}/#{id}/"
+        id
+      end
+
+      private
+
+      def http
+        @http
+      end
+    end
+  end
+end

data/lib/sobject_model/rest/errors.rb

@@ -0,0 +1,19 @@
+module SObjectModel
+  module Rest
+    class RecordNotFoundError < StandardError
+    end
+
+    class RequestError < StandardError
+      attr_reader :error_code, :error_message
+      def initialize(code, msg)
+        @error_code = code
+        @error_message  = msg
+        super(to_s)
+      end
+
+      def to_s
+        %|[#{@code}] #{@msg}|
+      end
+    end
+  end
+end

data/lib/sobject_model/rest/http.rb

@@ -0,0 +1,61 @@
+require 'uri'
+require 'net/http'
+require 'json'
+
+module SObjectModel
+  module Rest
+    class Http
+      attr_reader :instance_url, :access_token
+
+      def initialize(instance_url, access_token)
+        @instance_url = instance_url
+        @access_token = access_token
+      end
+
+      def get(path)
+        url = URI.parse(instance_url + path)
+        res = Net::HTTP.get(url, headers)
+
+        if res =~/errorCode/
+          err = JSON.parse(res).first
+          if err['errorCode'] == 'NOT_FOUND'
+            raise RecordNotFoundError.new
+          else
+            raise RequestError.new(err['errorCode'], err['message'])
+          end
+        end
+
+        res
+      end
+
+      def post(path, data)
+        response = post_request(path, data)
+        response.body
+      end
+
+      def patch(path, data)
+        response = post_request(path + '?_HttpMethod=PATCH', data)
+        response.code
+      end
+
+      def delete(path)
+        response = post_request(path + '?_HttpMethod=DELETE', '')
+        response.code
+      end
+
+      private
+
+      def headers
+        { 'Authorization' => ('Bearer %{token}' % {token: access_token}), 'Content-Type'  => 'application/json' }
+      end
+
+      def post_request(path, data)
+        url = URI.parse(instance_url + path)
+        response = Net::HTTP.post(url, JSON.dump(data), headers)
+        raise RequestError.new(response.code, response.message) if response.is_a?(Net::HTTPClientError) || response.is_a?(Net::HTTPServerError)
+
+        response
+      end
+    end
+  end
+end

data/lib/sobject_model/rest/query_result.rb

@@ -0,0 +1,66 @@
+module SObjectModel
+  module Rest
+    class QueryResult
+      def initialize(api_response)
+        @response = JSON.parse(api_response)
+      end
+
+      def total_size
+        @total_size ||= response['totalSize']
+      end
+
+      def done?
+        @done ||= response['done']
+      end
+
+      alias completed? done?
+
+      def next_records_url
+        @next_records_url ||= response['nextRecordsUrl']
+      end
+
+      def to_records(model_class: nil)
+        records.each_with_object([]) do |h, a|
+          record = prepare_record(h)
+          a << (model_class ? model_class.new(**record) : record)
+        end
+      end
+
+      private
+
+      def response
+        @response
+      end
+
+      def records
+        @records ||= response['records']
+      end
+
+      def prepare_record(hash)
+        hash.delete 'attributes'
+
+        hash.keys.each do |k|
+          if parent?(hash[k])
+            hash[k] = prepare_record(hash[k])
+          elsif children?(hash[k])
+            hash[k] = hash[k]['records'].map{|h| prepare_record(h)}
+          end
+        end
+
+        hash
+      end
+
+      def children?(h)
+        return false unless h.instance_of?(Hash)
+
+        h.has_key? 'records'
+      end
+
+      def parent?(h)
+        return false unless h.instance_of?(Hash)
+
+        h.has_key?('records') == false
+      end
+    end
+  end
+end