voteable_mongoid 0.3.5 → 0.4.0

data/lib/mongoid/voteable.rb

@@ -22,8 +22,7 @@ module Mongoid
       def self.vote_point(klass = self, options = nil)
         VOTE_POINT[self.name] ||= {}
         VOTE_POINT[self.name][klass.name] ||= options
-      end
-            
+      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
@@ -103,6 +102,48 @@ module Mongoid
           })
         end
       end
+      
+      def self.unvote(options)
+        options.symbolize_keys!
+                
+        voter_id = options[:voter_id]
+        voter_id = BSON::ObjectId(voter_id) if voter_id.is_a?(String)
+        votee_id = options[:votee_id]
+        votee_id = BSON::ObjectId(votee_id) if votee_id.is_a?(String)
+        value = options[:value].to_sym
+        return unless value  # not voted yet!
+        
+        value_point = VOTE_POINT[self.name][self.name]
+        
+        pull_field = if value == :up
+            :up_voter_ids
+        else
+            :down_voter_ids
+        end
+        
+        collection.update({ 
+          :_id => votee_id,
+        }, {
+          '$pull' => { pull_field => voter_id },
+          '$inc' => {
+            :votes_count => -1,
+            :votes_point => -value_point[value]
+          }
+        })
+        
+        VOTE_POINT[self.name].each do |class_name, value_point|
+          next unless relation_metadata = relations[class_name.underscore]
+          next unless foreign_key_value = options[relation_metadata.foreign_key.to_sym]
+          foreign_key_value = BSON::ObjectId(foreign_key_value) if foreign_key_value.is_a?(String)
+          
+          class_name.constantize.collection.update({ :_id => foreign_key_value }, {
+            '$inc' => {
+              :votes_count => value_point[:not_increase_votes_count] ? 0 : -1,
+              :votes_point => -value_point[value]
+            }
+          })
+        end
+      end
     end
   
 
@@ -118,7 +159,18 @@ module Mongoid
 
       self.class.vote(options)
     end
-
+    
+    def unvote(options)  
+      VOTE_POINT[self.class.name].each do |class_name, value_point|
+        next unless relation_metadata = relations[class_name.underscore]
+        next unless foreign_key = relation_metadata.foreign_key
+        options[foreign_key.to_sym] = read_attribute(foreign_key)
+      end
+      
+      options[:votee_id] ||= _id
+      options[:value] = vote_value(options[:voter_id])
+      self.class.unvote(options)
+    end
 
     def vote_value(x)
       voter_id = case x

data/lib/mongoid/voter.rb

@@ -25,7 +25,10 @@ module Mongoid
       
       votees(votee_class).where(:_id => votee_id).count == 1
     end
-
+    
+    def unvote(votee)
+      votee.unvote(:voter_id => _id)
+    end
 
     def vote_value(options)
       votee = unless options.is_a?(Hash)

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: voteable_mongoid
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 15
   prerelease: 
   segments: 
   - 0
-  - 3
-  - 5
-  version: 0.3.5
+  - 4
+  - 0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Alex Nguyen
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-17 00:00:00 +08:00
+date: 2011-02-23 00:00:00 +08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -173,8 +173,36 @@ dependencies:
   type: :development
   requirement: *id011
 - !ruby/object:Gem::Dependency 
-  name: mongoid
+  name: bson_ext
   version_requirements: &id012 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  prerelease: false
+  type: :development
+  requirement: *id012
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  version_requirements: &id013 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  prerelease: false
+  type: :development
+  requirement: *id013
+- !ruby/object:Gem::Dependency 
+  name: mongoid
+  version_requirements: &id014 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -188,10 +216,10 @@ dependencies:
         version: 2.0.0.rc
   prerelease: false
   type: :runtime
-  requirement: *id012
+  requirement: *id014
 - !ruby/object:Gem::Dependency 
   name: bson_ext
-  version_requirements: &id013 !ruby/object:Gem::Requirement 
+  version_requirements: &id015 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ">="
@@ -202,10 +230,10 @@ dependencies:
         version: "0"
   prerelease: false
   type: :development
-  requirement: *id013
+  requirement: *id015
 - !ruby/object:Gem::Dependency 
   name: rspec
-  version_requirements: &id014 !ruby/object:Gem::Requirement 
+  version_requirements: &id016 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ">="
@@ -216,7 +244,7 @@ dependencies:
         version: "0"
   prerelease: false
   type: :development
-  requirement: *id014
+  requirement: *id016
 description: Add Up / Down Voting for Mongoid
 email: alex@vinova.sg
 executables: []