follower.rb in mongoid_follow-0.2.2

- old
+ new

@@ -8,11 +8,11 @@
     end
 
     # follow a model
     #
     # Example:
-    # >> @bonnie.follow(@clyde)
+    # => @bonnie.follow(@clyde)
     def follow(model)
       if self.id != model.id && !self.follows?(model)
 
         model.before_followed_by(self) if model.respond_to?('before_followed_by')
         model.followers.create!(:ff_type => self.class.name, :ff_id => self.id)
@@ -30,11 +30,11 @@
     end
 
     # unfollow a model
     #
     # Example:
-    # >> @bonnie.unfollow(@clyde)
+    # => @bonnie.unfollow(@clyde)
     def unfollow(model)
       if self.id != model.id && self.follows?(model)
 
         model.before_unfollowed_by(self) if model.respond_to?('before_unfollowed_by')
         model.followers.where(:ff_type => self.class.name, :ff_id => self.id).destroy
@@ -52,49 +52,80 @@
     end
 
     # know if self is already following model
     #
     # Example:
-    # >> @bonnie.follows?(@clyde)
+    # => @bonnie.follows?(@clyde)
     # => true
     def follows?(model)
       0 < self.followees.find(:all, conditions: {ff_id: model.id}).limit(1).count
     end
 
     # get followees count
+    # Note: this is a cache counter
     #
     # Example:
-    # >> @bonnie.followees_count
+    # => @bonnie.followees_count
     # => 1
     def followees_count
       self.ffeec
     end
 
+    # get followees count by model
+    #
+    # Example:
+    # => @bonnie.followees_count_by_model(User)
+    # => 1
+    def followees_count_by_model(model)
+      self.followees.where(:ff_type => model.to_s).count
+    end
+
     # view all selfs followees
     #
     # Example:
-    # >> @alec.all_followees
+    # => @alec.all_followees
     # => [@bonnie]
     def all_followees
       get_followees_of(self)
     end
 
+    # view all selfs followees by model
+    #
+    # Example:
+    # => @clyde.all_followees_by_model
+    # => [@bonnie]
+    def all_followees_by_model(model)
+      get_followees_of(self, model)
+    end
+
     # view all common followees of self against model
     #
     # Example:
-    # >> @clyde.common_followees_with(@gang)
+    # => @clyde.common_followees_with(@gang)
     # => [@bonnie, @alec]
     def common_followees_with(model)
       model_followees = get_followees_of(model)
       self_followees = get_followees_of(self)
 
       self_followees & model_followees
     end
 
     private
-    def get_followees_of(me)
-      me.followees.collect do |f|
+    def get_followees_of(me, model = nil)
+      followees = !model ? me.followees : me.followees.where(:ff_type => model.to_s)
+
+      followees.collect do |f|
         f.ff_type.constantize.find(f.ff_id)
+      end
+    end
+
+    def method_missing(missing_method, *args, &block)
+      if missing_method.to_s =~ /^(.+)_followees_count$/
+        followees_count_by_model($1.camelize)
+      elsif missing_method.to_s =~ /^all_(.+)_followees$/
+        all_followees_by_model($1.camelize)
+      else
+        super
       end
     end
   end
 end