mongoid_vote 0.0.1.3 → 0.0.1.4

data/lib/mongoid_vote/version.rb

@@ -1,3 +1,3 @@
 module MongoidVote
-  VERSION = "0.0.1.3"
+  VERSION = "0.0.1.4"
 end

data/lib/mongoid_vote/voteable.rb

@@ -20,6 +20,14 @@ module MongoidVote
         vote_adjust(current_vote(user), 'up', user)
       end
 
+      def upvote_toggle(user)
+        vote_adjust(current_vote(user), 'up', user, 1)
+      end
+
+      def downvote_toggle(user)
+        vote_adjust(current_vote(user), 'down', user, 1)
+      end
+
       def downvote(user)
         vote_adjust(current_vote(user), 'down', user)
       end
@@ -54,8 +62,18 @@ module MongoidVote
 
     private
 
-      def vote_adjust(current, vote, user)
-        unless current.to_s == vote
+      def vote_adjust(current, vote, user, toggle = 0)
+        if current.to_s == vote
+          if current == :up && vote == "up" && toggle == 1
+            self.votes["up_count"] -= 1
+            self.votes["count"] -= 1
+            self.votes["up"].delete_if {|x| x == user.id.to_s}
+          elsif current == :down && vote == "down" && toggle == 1
+            self.votes["down_count"] += 1
+            self.votes["count"] += 1
+            self.votes["down"].delete_if {|x| x == user.id.to_s}
+          end
+        else
           if !current && vote == "up"
             self.votes["up_count"] += 1
             self.votes["count"] += 1

data/readme.rdoc

@@ -21,17 +21,20 @@ Then add this to the document model you'd like to add voting capabilites to:
 == Usage Examples
 
 
-  @comment.upvote(@user)   # up vote
-  @comment.downvote(@user) # down vote
+  @comment.upvote(@user)           # up vote
+  @comment.downvote(@user)         # down vote
 
-  @comment.votecount       # vote count
-  @comment.upcount         # up vote count
-  @comment.downcount       # down vote count
+  @comment.upvote_toggle(@user)    # acts as upvote but if up vote is already present, removes vote instead. (i.e. reddit style)
+  @comment.downvote_toggle(@user)  # acts as upvote but if up vote is already present, removes vote instead. (i.e. reddit style)
 
-  @comment.upvoters        # up voters 
-  @comment.downvoters      # down voters
+  @comment.votecount               # vote count
+  @comment.upcount                 # up vote count
+  @comment.downcount               # down vote count
 
-  @comment.current_vote(@user)  # returns :up / :down / nil
+  @comment.upvoters                # up voters 
+  @comment.downvoters              # down voters
+
+  @comment.current_vote(@user)     # returns :up / :down / nil
 
 == Notes
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongoid_vote
 version: !ruby/object:Gem::Version
-  version: 0.0.1.3
+  version: 0.0.1.4
   prerelease: 
 platform: ruby
 authors: