voteable_mongoid 0.4.4 → 0.4.5

data/lib/voteable_mongoid/railtie.rb

@@ -0,0 +1,17 @@
+module Rails #:nodoc:
+  module VoteableMongoid #:nodoc:
+    class Railtie < Rails::Railtie #:nodoc:
+
+      initializer "preload all application models" do |app|
+        config.to_prepare do
+          ::Rails::Mongoid.load_models(app)
+        end
+      end
+
+      rake_tasks do
+        load 'voteable_mongoid/railties/database.rake'
+      end
+
+    end
+  end
+end

data/lib/voteable_mongoid/railties/database.rake

@@ -3,7 +3,7 @@ namespace :db do
     namespace :voteable do
       desc 'Update up_votes_count, down_votes_count, votes_count and votes_point'
       task :remake_stats => :environment do
-       Mongoid::Voteable::Stats.remake_stats
+        Mongoid::Voteable::Stats.remake(:log)
       end
     end
   end

data/lib/voteable_mongoid/version.rb

@@ -1,3 +1,3 @@
 module VoteableMongoid
-  VERSION = "0.4.4"
+  VERSION = '0.4.5'
 end

data/lib/voteable_mongoid/voteable/stats.rb

@@ -15,15 +15,55 @@ module Mongoid
         scope :best_voted, order_by(['voteable.votes_point', :desc])
       end
     
-      def self.remake_stats
+      # Get the number of up votes
+      def up_votes_count
+        voteable.try(:[], 'up_votes_count') || 0
+      end
+    
+      # Get the number of down votes
+      def down_votes_count
+        voteable.try(:[], 'down_votes_count') || 0
+      end
+    
+      # Get the number of votes
+      def votes_count
+        voteable.try(:[], 'votes_count') || 0
+      end
+    
+      # Get the votes point
+      def votes_point
+        voteable.try(:[], 'votes_point') || 0
+      end
+
+      def self.remake(log = false)
+        remake_stats(log)
+        update_parent_stats(log)
+      end
+      
+      def self.remake_stats(log)
         Mongoid::Voteable::VOTE_POINT.each do |class_name, value_point|
           klass = class_name.constantize
           klass_value_point = value_point[class_name]
+          puts "Generating stats for #{class_name}" if log
           klass.voteable_related.each{ |doc|
             doc.remake_stats(klass_value_point)
           }
         end
+      end
+      
+      def remake_stats(value_point)
+        up_count = up_voter_ids.length
+        down_count = down_voter_ids.length
 
+        update_attributes(
+          'voteable.up_votes_count' => up_count,
+          'voteable.down_votes_count' => down_count,          
+          'voteable.votes_count' => up_count + down_count,
+          'voteable.votes_point' => value_point[:up]*up_count + value_point[:down]*down_count
+        )
+      end
+    
+      def self.update_parent_stats(log)
         VOTE_POINT.each do |class_name, value_point|
           klass = class_name.constantize
           value_point.each do |parent_class_name, parent_value_point|
@@ -31,6 +71,7 @@ module Mongoid
             if relation_metadata
               parent_class = parent_class_name.constantize
               foreign_key = relation_metadata.foreign_key
+              puts "Updating stats for #{class_name} > #{parent_class_name}" if log
               klass.voteable_related.each{ |doc|
                 doc.update_parent_stats(parent_class, foreign_key, parent_value_point)
               }
@@ -38,31 +79,19 @@ module Mongoid
           end
         end
       end
-    
-      def remake_stats(value_point)
-        up_count = up_voter_ids.length
-        down_count = down_voter_ids.length
-        # puts "#{self.class}(#{id}): #{up_count}, #{down_count}" # DEBUG
-        update_attributes(
-          'voteable.up_votes_count' => up_count,
-          'voteable.down_votes_count' => down_count,          
-          'voteable.votes_count' => up_count + down_count,
-          'voteable.votes_point' => value_point[:up]*up_count + value_point[:down]*down_count
-        )
-      end
-    
+      
       def update_parent_stats(parent_class, foreign_key, value_point)
         parent_id = read_attribute(foreign_key.to_sym)
         if parent_id
           up_count = up_voter_ids.length
           down_count = down_voter_ids.length
-          
+        
           return if up_count == 0 && down_count == 0
 
           inc_options = {
             'voteable.votes_point' => value_point[:up]*up_count + value_point[:down]*down_count
           }
-        
+      
           unless value_point[:update_counters] == false
             inc_options.merge!(
               'voteable.votes_count' => up_count + down_count,
@@ -70,35 +99,14 @@ module Mongoid
               'voteable.down_votes_count' => down_count
             )
           end
-          
-          # puts inc_options.inspect # DEBUG          
+        
           parent_class.collection.update(
             { :_id => parent_id }, 
             { '$inc' =>  inc_options }
           )
         end
       end
-    
-      # Get the number of up votes
-      def up_votes_count
-        voteable.try(:[], 'up_votes_count') || 0
-      end
-    
-      # Get the number of down votes
-      def down_votes_count
-        voteable.try(:[], 'down_votes_count') || 0
-      end
-    
-      # Get the number of votes
-      def votes_count
-        voteable.try(:[], 'votes_count') || 0
-      end
-    
-      # Get the votes point
-      def votes_point
-        voteable.try(:[], 'votes_point') || 0
-      end
-    
+      
     end
   end
 end

data/lib/voteable_mongoid.rb

@@ -2,3 +2,7 @@ require 'voteable_mongoid/voteable/stats'
 require 'voteable_mongoid/voteable'
 require 'voteable_mongoid/voter'
 
+# add railtie
+if defined?(Rails)
+  require 'voteable_mongoid/railtie'
+end

data/spec/voteable_mongoid/voteable_spec.rb

@@ -60,7 +60,7 @@ describe Mongoid::Voteable do
   context 'user1 vote up post1 the first time' do
     before :all do    
       Post.vote(:votee_id => @post1.id, :voter_id => @user1.id, :value => :up)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post1.reload
     end
     
@@ -93,7 +93,7 @@ describe Mongoid::Voteable do
   context 'user2 vote down post1 the first time' do
     before :all do
       Post.vote(:votee_id => @post1.id, :voter_id => @user2.id, :value => :down)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post1.reload
     end
     
@@ -114,7 +114,7 @@ describe Mongoid::Voteable do
   context 'user1 change vote on post1 from up to down' do
     before :all do
       Post.vote(:revote => true, :votee_id => @post1.id, :voter_id => @user1.id, :value => :down)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post1.reload
     end
     
@@ -135,7 +135,7 @@ describe Mongoid::Voteable do
   context 'user1 vote down post2 the first time' do
     before :all do
       Post.vote(:votee_id => @post2.id, :voter_id => @user1.id, :value => :down)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post2.reload
     end
     
@@ -155,7 +155,7 @@ describe Mongoid::Voteable do
   context 'user1 change vote on post2 from down to up' do
     before :all do
       Post.vote(:revote => true, :votee_id => @post2.id.to_s, :voter_id => @user1.id.to_s, :value => :up)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post2.reload
     end
     
@@ -176,7 +176,7 @@ describe Mongoid::Voteable do
   context 'user1 vote up post2 comment the first time' do
     before :all do
       @comment.vote(:voter_id => @user1.id, :value => :up)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @comment.reload
       @post2.reload
     end
@@ -198,7 +198,7 @@ describe Mongoid::Voteable do
   context 'user1 revote post2 comment from up to down' do
     before :all do
       @user1.vote(:votee => @comment, :value => :down)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @comment.reload
       @post2.reload
     end
@@ -233,7 +233,7 @@ describe Mongoid::Voteable do
   context "user1 unvote on post1" do
     before(:all) do
       @post1.vote(:voter_id => @user1.id, :votee_id => @post1.id, :unvote => true)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @post1.reload
     end
     
@@ -276,7 +276,7 @@ describe Mongoid::Voteable do
   context "user1 unvote on comment" do
     before(:all) do
       @user1.unvote(@comment)
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
       @comment.reload
       @post2.reload
     end
@@ -296,7 +296,7 @@ describe Mongoid::Voteable do
   
   context 'final' do
     it "test remake stats" do
-      Mongoid::Voteable::Stats.remake_stats
+      Mongoid::Voteable::Stats.remake
 
       @post1.up_votes_count.should == 0
       @post1.down_votes_count.should == 1

data/voteable_mongoid.gemspec

@@ -8,11 +8,11 @@ Gem::Specification.new do |s|
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Alex Nguyen"]
   s.email       = ["alex@vinova.sg"]
-  s.homepage    = "http://vinova.sg"
+  s.homepage    = "http://vinova.sg/"
   s.summary     = %q{Add Up / Down Voting for Mongoid}
   s.description = %q{Add Up / Down Voting for Mongoid (built for speed by using one Mongodb update-in-place for each collection when provided enough information)}
 
-  s.add_dependency 'mongoid', '~> 2.0.0.rc'
+  s.add_dependency 'mongoid', '~> 2.0.0'
   s.add_development_dependency 'rspec'
   s.add_development_dependency 'bson_ext'
   s.add_development_dependency 'watchr'

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: voteable_mongoid
 version: !ruby/object:Gem::Version 
-  hash: 7
+  hash: 5
   prerelease: 
   segments: 
   - 0
   - 4
-  - 4
-  version: 0.4.4
+  - 5
+  version: 0.4.5
 platform: ruby
 authors: 
 - Alex Nguyen
@@ -26,13 +26,12 @@ dependencies:
     requirements: 
     - - ~>
       - !ruby/object:Gem::Version 
-        hash: 7712058
+        hash: 15
         segments: 
         - 2
         - 0
         - 0
-        - rc
-        version: 2.0.0.rc
+        version: 2.0.0
   type: :runtime
   version_requirements: *id001
 - !ruby/object:Gem::Dependency 
@@ -113,6 +112,7 @@ files:
 - doc/method_list.html
 - doc/top-level-namespace.html
 - lib/voteable_mongoid.rb
+- lib/voteable_mongoid/railtie.rb
 - lib/voteable_mongoid/railties/database.rake
 - lib/voteable_mongoid/version.rb
 - lib/voteable_mongoid/voteable.rb
@@ -127,7 +127,7 @@ files:
 - spec/voteable_mongoid/voter_spec.rb
 - voteable_mongoid.gemspec
 has_rdoc: true
-homepage: http://vinova.sg
+homepage: http://vinova.sg/
 licenses: []
 
 post_install_message: