mongo_mapper_acts_as_versioned 0.0.10 → 0.0.11

data/lib/acts_as_versioned.rb

@@ -1,15 +1,16 @@
 module MongoMapper
   module Acts
     module Versioned
-      VERSION   = '0.0.10'
-      CALLBACKS = [:save_version, :save_version?]
+      VERSION   = '0.0.11'
+      CALLBACKS = [:save_version, :save_version?, :clear_old_versions]
 
       def self.configure(model)
         model.class_eval do
-          cattr_accessor :versioned_class_name, :non_versioned_keys
+          cattr_accessor :versioned_class_name, :non_versioned_keys, :max_version_limit
 
           self.versioned_class_name = :Version
-          self.non_versioned_keys   = %w(
+          self.max_version_limit = 0
+          self.non_versioned_keys = %w(
             _id _type created_at updated_at creator_id updater_id version
           )
 
@@ -21,27 +22,39 @@ module MongoMapper
           end
 
           many :versions, :class => "#{self}::#{versioned_class_name}".constantize do
-            def [](given_version)
-              detect {|version| version.version.to_s == given_version.to_s }
+            def [](version)
+              detect { |doc| doc.version.to_s == version.to_s }
             end
           end
         end
 
         model.key :version, Integer
         model.before_save :save_version
+        model.before_save :clear_old_versions
       end
 
       module InstanceMethods
         def save_version
           if new_record? || save_version?
             self.version = next_version
+
             rev = self.class.versioned_class.new
             clone_attributes(self, rev)
             rev.version = version
+
             self.versions << rev
           end
         end
 
+        def clear_old_versions
+          return if self.class.max_version_limit == 0
+          excess_bagage = version.to_i - self.class.max_version_limit
+
+          if excess_bagage > 0
+            versions.reject! { |v| v.version.to_i <= excess_bagage }
+          end
+        end
+
         def revert_to(version)
           if version.is_a?(self.class.versioned_class)
             return false if version.new_record?
@@ -56,7 +69,7 @@ module MongoMapper
         end
 
         def revert_to!(version)
-          revert_to(version) ? save_without_revision : false
+          revert_to(version) and save_without_revision or false
         end
 
         def save_without_revision
@@ -67,9 +80,7 @@ module MongoMapper
         end
 
         def save_without_revision!
-          without_revision do
-            save!
-          end
+          without_revision { save! }
         end
 
         def clone_attributes(orig_model, new_model)

data/spec/acts_as_versioned_spec.rb

@@ -33,6 +33,23 @@ describe MongoMapper::Acts::Versioned do
       l.versions.first.should be_a(Landmark.versioned_class)
     end
 
+    it 'should clear old versions when a limit is set' do
+      Landmark.max_version_limit = 3
+
+      l = Landmark.create(:title => 'title')
+      (2..10).each do |i|
+        l = Landmark.first
+        l.update_attributes(:title => "title#{i}")
+      end
+
+      l = l.reload
+      l.versions.size.should == 3
+      l.versions.first.version.should == 8
+      l.versions.last.version.should == 10
+
+      Landmark.max_version_limit = 0
+    end
+
     it 'should save without revision' do
       l = Landmark.create(:title => 'title')
       l.version.should == 1

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 10
-  version: 0.0.10
+  - 11
+  version: 0.0.11
 platform: ruby
 authors: 
 - Gigamo
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-09-12 00:00:00 +02:00
+date: 2010-09-22 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency