voteable_mongoid 0.2.3 → 0.3.0

data/lib/mongoid/voteable.rb

@@ -2,16 +2,28 @@ module Mongoid
   module Voteable
     extend ActiveSupport::Concern
 
+    VOTE_POINT = {}
+
     included do
       field :up_voter_ids, :type => Array, :default => []
       field :down_voter_ids, :type => Array, :default => []
     
       field :votes_count, :type => Integer, :default => 0
       field :votes_point, :type => Integer, :default => 0
+
     
       index :votes_count
       index :votes_point
       
+      scope :most_voted, order_by(:votes_count.desc)
+      scope :best_voted, order_by(:votes_point.desc)
+      
+      
+      def self.vote_point(klass, options = nil)
+        VOTE_POINT[self] ||= {}
+        VOTE_POINT[self][klass] ||= options
+      end
+            
 
       # We usually need to show current_user his voting value on voteable object
       # voting value can be nil (not voted yet), :up or :down
@@ -28,17 +40,19 @@ module Mongoid
         votee_id = BSON::ObjectID(votee_id) if votee_id.is_a?(String)
         voter_id = BSON::ObjectID(voter_id) if voter_id.is_a?(String)
 
+        value = value.to_sym
+        
         if options[:revote]
-          if value.to_sym == :up
+          if value == :up
             push_field = :up_voter_ids
             pull_field = :down_voter_ids
-            point_delta = +2
+            point_delta = VOTE_POINT[self][self][:up] - VOTE_POINT[self][self][:down]
           else
             push_field = :down_voter_ids
             pull_field = :up_voter_ids
-            point_delta = -2
+            point_delta = -VOTE_POINT[self][self][:up] + VOTE_POINT[self][self][:down]
           end
-
+          
           collection.update({ 
             :_id => votee_id,
             push_field => { '$ne' => voter_id },
@@ -54,10 +68,8 @@ module Mongoid
         else # new vote
           if value.to_sym == :up
             push_field = :up_voter_ids
-            point_delta = +1
           else
             push_field = :down_voter_ids
-            point_delta = -1
           end
 
           collection.update({ 
@@ -68,13 +80,43 @@ module Mongoid
             '$push' => { push_field => voter_id },
             '$inc' => {
               :votes_count => +1,
-              :votes_point => point_delta
+              :votes_point => VOTE_POINT[self][self][value]
+            }
+          })
+        end
+        
+        VOTE_POINT[self].each do |klass, value_point|
+          next unless association = associations[klass.name.underscore]
+          next unless foreign_key = options[association.options[:foreign_key].to_sym]
+          klass.collection.update({ :_id => foreign_key }, { 
+            '$inc' => options[:revote] ? {
+              :votes_point => ( value == :up ? 
+                 value_point[:up] - value_point[:down] : 
+                -value_point[:up] + value_point[:down] )
+            } : {
+              :votes_count => 1,
+              :votes_point => value_point[value]
             }
           })
         end
       end
     end
   
+
+    def vote(options)
+      VOTE_POINT[self.class].each do |klass, value_point|
+        next unless association = self.class.associations[klass.name.underscore]
+        next unless foreign_key = association.options[:foreign_key]
+        options[foreign_key] = read_attribute(foreign_key)
+      end
+      
+      options[:votee_id] ||= _id
+      options[:revote] ||= !vote_value(options[:voter_id]).nil?
+
+      self.class.vote(options)
+    end
+
+    
     def vote_value(x)
       voter_id = case x
       when String

data/lib/mongoid/voter.rb

@@ -18,7 +18,7 @@ module Mongoid
           votee_class = votee.class
           votee_id = votee.id
         else
-          votee_class = options[:votee_type].constantize
+          votee_class = options[:votee_type].classify.constantize
           votee_id = options[:votee_id]
         end
       end
@@ -31,7 +31,7 @@ module Mongoid
       votee = unless options.is_a?(Hash)
         options
       else
-        options[:votee_type].constantize.only(:up_vote_ids, :down_vote_ids).where(
+        options[:votee_type].classify.constantize.only(:up_vote_ids, :down_vote_ids).where(
           :_id => options[:votee_id]
         ).first
       end
@@ -53,7 +53,7 @@ module Mongoid
         options[:votee_id] = votee.id
         votee_class = votee.class
       else
-        votee_class = options[:votee_type].constantize
+        votee_class = options[:votee_type].classify.constantize
       end
       
       options[:revote] = if options.has_key?(:revote)
@@ -64,7 +64,7 @@ module Mongoid
       
       options[:voter_id] = _id
 
-      votee_class.vote(options)
+      ( votee || votee_class ).vote(options)
     end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: voteable_mongoid
 version: !ruby/object:Gem::Version 
-  hash: 17
+  hash: 19
   prerelease: false
   segments: 
   - 0
-  - 2
   - 3
-  version: 0.2.3
+  - 0
+  version: 0.3.0
 platform: ruby
 authors: 
 - Alex Nguyen
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-09-03 00:00:00 +07:00
+date: 2010-09-05 00:00:00 +07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency