mongo_watchable 0.1.2 → 0.2.0

data/lib/mongo_watchable/proxy.rb

@@ -1,11 +1,12 @@
 module MongoWatchable
   class Proxy
     attr_reader :array_key
+    attr_reader :array_count_key
     attr_reader :model
     attr_reader :target_class
     
-    def initialize(model, array_key, target_class)
-      @model, @array_key, @target_class = model, array_key, target_class
+    def initialize(model, array_key, array_count_key, target_class)
+      @model, @array_key, @array_count_key, @target_class = model, array_key, array_count_key, target_class
     end
     
     def to_a
@@ -47,11 +48,12 @@ module MongoWatchable
     
     def << (entry)
       array << entry.id
+      increment_array_count
       @fetch ? @fetch << entry : fetch_all
     end
     
     def delete(entry)
-      array.delete entry.id
+      decrement_array_count if array.delete(entry.id)
       @fetch ? @fetch.delete(entry) : fetch_all
     end
     
@@ -67,5 +69,13 @@ module MongoWatchable
     def array
       model.send(array_key)
     end
+    
+    def increment_array_count
+      model.send("#{array_count_key}=", model.send(array_count_key) + 1)
+    end
+    
+    def decrement_array_count
+      model.send("#{array_count_key}=", model.send(array_count_key) - 1)
+    end
   end
 end

data/lib/mongo_watchable/watchable.rb

@@ -4,9 +4,10 @@ module MongoWatchable
       watchable.class_eval do
         MongoWatchable.watchers.each do |watcher|
           key :"#{watcher.name.underscore}_watcher_ids", Array
+          key :"#{watcher.name.underscore}_watchers_count", Integer, :default => 0, :index => true
           
           define_method :"#{watcher.name.underscore}_watchers" do
-            MongoWatchable::Proxy.new(self, :"#{watcher.name.underscore}_watcher_ids", watcher)
+            MongoWatchable::Proxy.new(self, :"#{watcher.name.underscore}_watcher_ids", :"#{watcher.name.underscore}_watchers_count", watcher)
           end
         end
       end
@@ -14,9 +15,10 @@ module MongoWatchable
       MongoWatchable.watchers.each do |watcher|
         watcher.class_eval do
           key :"#{watchable.name.underscore}_watching_ids", Array
+          key :"#{watchable.name.underscore}_watchings_count", Integer, :default => 0, :index => true
           
           define_method :"#{watchable.name.underscore}_watchings" do
-            MongoWatchable::Proxy.new(self, :"#{watchable.name.underscore}_watching_ids", watchable)
+            MongoWatchable::Proxy.new(self, :"#{watchable.name.underscore}_watching_ids", :"#{watchable.name.underscore}_watchings_count", watchable)
           end
         end
       end

data/lib/mongo_watchable/watcher.rb

@@ -4,9 +4,10 @@ module MongoWatchable
       watcher.class_eval do
         MongoWatchable.watchables.each do |watchable|
           key :"#{watchable.name.underscore}_watching_ids", Array
+          key :"#{watchable.name.underscore}_watchings_count", Integer, :default => 0, :index => true
           
           define_method :"#{watchable.name.underscore}_watchings" do
-            MongoWatchable::Proxy.new(self, :"#{watchable.name.underscore}_watching_ids", watchable)
+            MongoWatchable::Proxy.new(self, :"#{watchable.name.underscore}_watching_ids", :"#{watchable.name.underscore}_watchings_count", watchable)
           end
         end
       end
@@ -14,9 +15,10 @@ module MongoWatchable
       MongoWatchable.watchables.each do |watchable|
         watchable.class_eval do
           key :"#{watcher.name.underscore}_watcher_ids", Array
+          key :"#{watcher.name.underscore}_watchers_count", Integer, :default => 0, :index => true
           
           define_method :"#{watcher.name.underscore}_watchers" do
-            MongoWatchable::Proxy.new(self, :"#{watcher.name.underscore}_watcher_ids", watcher)
+            MongoWatchable::Proxy.new(self, :"#{watcher.name.underscore}_watcher_ids", :"#{watcher.name.underscore}_watchers_count", watcher)
           end
         end
       end
@@ -75,6 +77,14 @@ module MongoWatchable
       end
       send("#{klass.name.underscore}_watchings")
     end
+    
+    def watchings_count_proxy_for(watchable)
+      klass = watchable.class
+      while klass.superclass && klass.superclass.include?(MongoWatchable::Watchable)
+        klass = klass.superclass
+      end
+      send("#{klass.name.underscore}_watchings")
+    end
   
   private
     def root_watching_class(instance)

data/test/mongo_watchable_test.rb

@@ -38,11 +38,15 @@ class MongoWatchableTest < ActiveSupport::TestCase
       @widget.reload
     end
     
-    should "should only have 1 widget in widget_watchings array" do
+    should "only have 1 widget in widget_watchings array" do
       assert_equal 1, @user.widget_watchings.size
     end
     
-    should "should have widget in widget watchings array" do
+    should "increment widget_watchings_count" do
+      assert_equal 1, @user.widget_watchings_count
+    end
+    
+    should "have widget in widget watchings array" do
       assert_equal @widget, @user.widget_watchings.first
     end
     
@@ -50,6 +54,10 @@ class MongoWatchableTest < ActiveSupport::TestCase
       assert_equal 1, @widget.user_watchers.size
     end
     
+    should "increment user_watchers_count" do
+      assert_equal 1, @widget.user_watchers_count
+    end
+    
     should "be the watcher in widgets user_watchers array" do
       assert_equal @user, @widget.user_watchers.first
     end
@@ -84,21 +92,29 @@ class MongoWatchableTest < ActiveSupport::TestCase
         assert @return
       end
     
-      should "should only have 2 widgets in widget_watchings array" do
+      should "only have 2 widgets in widget_watchings array" do
         assert_equal 2, @user.widget_watchings.size
       end
+    
+      should "increment widget_watchings_count" do
+        assert_equal 2, @user.widget_watchings_count
+      end
       
-      should "should have widget in widget watchings array" do
+      should "have widget in widget watchings array" do
         assert_equal @widget, @user.widget_watchings.first
       end
       
-      should "should have widget2 in widget watchings array" do
+      should "have widget2 in widget watchings array" do
         assert_equal @widget2, @user.widget_watchings.last
       end
     
       should "have only one watcher in widget's user_watchers array" do
         assert_equal 1, @widget.user_watchers.size
       end
+    
+      should "not increment user_watchers_count" do
+        assert_equal 1, @widget.user_watchers_count
+      end
       
       should "be the watcher in widget's user_watchers array" do
         assert_equal @user, @widget.user_watchers.first
@@ -127,6 +143,14 @@ class MongoWatchableTest < ActiveSupport::TestCase
       should "have no user_watchers in widget" do
         assert @widget.user_watchers.empty?
       end
+      
+      should "have 0 widget_watchings_count" do
+        assert_equal 0, @user.widget_watchings_count
+      end
+      
+      should "have 0 user_watchers_count" do
+        assert_equal 0, @widget.user_watchers_count
+      end
     
       should "return with true" do
         assert @return
@@ -144,11 +168,15 @@ class MongoWatchableTest < ActiveSupport::TestCase
       @widget.reload
     end
     
-    should "should only have 1 widget in widget_watchings array" do
+    should "only have 1 widget in widget_watchings array" do
       assert_equal 1, @user.widget_watchings.size
     end
     
-    should "should have widget in widget watchings array" do
+    should "increment widget_watchings_count" do
+      assert_equal 1, @user.widget_watchings_count
+    end
+    
+    should "have widget in widget watchings array" do
       assert_equal @widget, @user.widget_watchings.first
     end
     
@@ -156,6 +184,10 @@ class MongoWatchableTest < ActiveSupport::TestCase
       assert_equal 1, @widget.user_watchers.size
     end
     
+    should "increment user_watchers_count" do
+      assert_equal 1, @widget.user_watchers_count
+    end
+    
     should "be the watcher in widgets user_watchers array" do
       assert_equal @user, @widget.user_watchers.first
     end
@@ -172,21 +204,29 @@ class MongoWatchableTest < ActiveSupport::TestCase
         @widget2.reload
       end
     
-      should "should only have 2 widgets in widget_watchings array" do
+      should "have 2 widgets in widget_watchings array" do
         assert_equal 2, @user.widget_watchings.size
       end
+    
+      should "increment widget_watchings_count" do
+        assert_equal 2, @user.widget_watchings_count
+      end
       
-      should "should have widget in widget watchings array" do
+      should "have widget in widget watchings array" do
         assert_equal @widget, @user.widget_watchings.first
       end
       
-      should "should have widget2 in widget watchings array" do
+      should "have widget2 in widget watchings array" do
         assert_equal @widget2, @user.widget_watchings.last
       end
     
-      should "have only one watcher in widget's user_watchers array" do
+      should "have one watcher in widget's user_watchers array" do
         assert_equal 1, @widget.user_watchers.size
       end
+    
+      should "not increment user_watchers_count" do
+        assert_equal 1, @widget.user_watchers_count
+      end
       
       should "be the watcher in widget's user_watchers array" do
         assert_equal @user, @widget.user_watchers.first
@@ -223,6 +263,14 @@ class MongoWatchableTest < ActiveSupport::TestCase
       should "have no user_watchers in widget" do
         assert @widget.user_watchers.empty?
       end
+      
+      should "have 0 widget_watchings_count" do
+        assert_equal 0, @user.widget_watchings_count
+      end
+      
+      should "have 0 user_watchers_count" do
+        assert_equal 0, @widget.user_watchers_count
+      end
     end
   end
   

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
   - 2
-  version: 0.1.2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Jonathan Bell
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-25 00:00:00 -05:00
+date: 2010-06-14 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency