Extend active record concern for mm0.9.x

dlobraico · Jun 3, 2011 · ce26218 · ce26218
1 parent e28dab1
commit ce26218
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/lib/versionable.rb b/lib/versionable.rb
@@ -2,10 +2,4 @@
 require 'mongo_mapper'
 require 'versionable/plugins/versionable'
 
-module VersionablePlugin
-  def self.included(model)
-    model.plugin Versionable
-  end
-end
-
-MongoMapper::Document.append_inclusions(VersionablePlugin)
+MongoMapper::Document.plugin(Versionable)
diff --git a/lib/versionable/plugins/versionable.rb b/lib/versionable/plugins/versionable.rb
@@ -1,6 +1,8 @@
 require 'versionable/models/version'
 
 module Versionable
+  extend ActiveSupport::Concern
+
   module InstanceMethods
     # Save new versions but only if the data changes
     def save(options={})
@@ -70,7 +72,7 @@ def enable_versioning(opts={})
         self.version_number = version.pos
         self
       end
-      
+
       define_method(:rollback!) do |pos = nil|
         self.rollback(pos)
 
@@ -110,7 +112,7 @@ def enable_versioning(opts={})
           index = self.versions.index {|v| v.pos == pos}
           version = self.versions[index] if index
           version ||= Version.first(:user_id => self._id.to_s, :pos => pos)
-          version 
+          version
         end
       end
     end