Restforce monkey patch

restforce.rb

module Restforce
  class Mash < Hashie::Mash

    class << self

      def klass(val)
        if val.key? 'records'
          # When the hash has a records key, it should be considered a collection
          # of sobject records.
          Restforce::Collection
        elsif val.key? 'attributes'
          case val.dig('attributes', 'type')
          when "Attachment"
            Restforce::Attachment
          when "Document"
            Restforce::Document
          else
            # When the hash contains an attributes key, it should be considered an
            # sobject record
            Restforce::SObject
          end
        else
          # Fallback to a standard Restforce::Mash for everything else
          Restforce::Mash
        end
      end

    end

  end
end