Extract Blacklight::Document::ActiveModelShim from Blacklight::Document

Author: cbeer

lib/blacklight/document.rb

@@ -13,6 +13,7 @@
 # transformation formats.
 #
 module Blacklight::Document
+  autoload :ActiveModelShim, 'blacklight/document/active_model_shim'
   autoload :SchemaOrg, 'blacklight/document/schema_org'
   autoload :DublinCore, 'blacklight/document/dublin_core'
   autoload :Email, 'blacklight/document/email'
@@ -42,14 +43,6 @@ def initialize(source_doc={}, response=nil)
     apply_extensions
   end
 
-  def to_model
-    self
-  end
-
-  def persisted?
-    true
-  end
-
   # the wrapper method to the @_source object.
   # If a method is missing, it gets sent to @_source
   # with all of the original params and block
@@ -65,14 +58,6 @@ def respond_to_missing? *args
     _source_responds_to?(*args) || super
   end
 
-  def [] *args
-    _source.send :[], *args
-  end
-
-  def _read_attribute(attr)
-    self[attr]
-  end
-
   # Helper method to check if value/multi-values exist for a given key.
   # The value can be a string, or a RegExp
   # Multiple "values" can be given; only one needs to match.
@@ -138,30 +123,10 @@ def first key
     Array(self[key]).first
   end
 
-  def id
-    self[self.class.unique_key]
-  end
-
-  def to_param
-    id.to_s
-  end
-
-  def as_json(options = nil)
-    _source.as_json(options)
-  end
-
   def to_partial_path
     'catalog/document'
   end
 
-  def destroyed?
-    false
-  end
-  
-  def new_record?
-    false
-  end
-  
   def has_highlight_field? k
     false
   end
@@ -186,13 +151,6 @@ def unique_key
       @unique_key ||= 'id' 
     end
 
-    def primary_key
-      unique_key
-    end
-    
-    def base_class
-      self
-    end  
   end
 
   private

lib/blacklight/document/active_model_shim.rb

@@ -0,0 +1,62 @@
+require 'active_model/conversion'
+
+module Blacklight::Document
+  module ActiveModelShim
+    extend ActiveSupport::Concern
+
+    include ::ActiveModel::Conversion
+
+    module ClassMethods
+      def primary_key
+        unique_key
+      end
+
+      def base_class
+        self
+      end
+    end
+    
+    ##
+    # Unique ID for the document
+    def id
+      self[self.class.unique_key]
+    end
+
+    ##
+    # accessors for reading attribute
+    def [] *args
+      _source.send :[], *args
+    end
+    
+    def _read_attribute(attr)
+      self[attr]
+    end
+
+    def as_json(options = nil)
+      _source.as_json(options)
+    end
+    
+    ##
+    # ActiveRecord::Persistence method stubs to get non-AR objects to
+    # play nice with e.g. Blacklight's bookmarks
+    def persisted?
+      true
+    end
+
+    def destroyed?
+      false
+    end
+    
+    def new_record?
+      false
+    end
+
+    ##
+    # #to_partial_path is also defined in Blacklight::Document, but
+    # ActiveModel::Conversion (included above) will overwrite that..
+    def to_partial_path
+      'catalog/document'
+    end
+
+  end
+end

lib/blacklight/solr/document.rb

@@ -26,6 +26,7 @@ module Blacklight::Solr::Document
 
   extend ActiveSupport::Concern
   include Blacklight::Document
+  include Blacklight::Document::ActiveModelShim
   include Blacklight::Solr::Document::MoreLikeThis
 
   def has_highlight_field? k