3 files changed, 50 insertions, 0 deletions

@@ -1,5 +1,6 @@
 class Blog < ApplicationRecord
   include Recordable
+  include Votable
 
   acts_as_taggable
 
@@ -0,0 +1,43 @@
+module Votable
+  extend ActiveSupport::Concern
+
+  included do
+    has_many :votes, as: :votable
+
+    def already_upvoted?(ip)
+      !votes.where(ip: ip, upvote: 1).empty?
+    end
+
+    def already_downvoted?(ip)
+      !votes.where(ip: ip, upvote: 0).empty?
+    end
+
+    def upvote!(ip)
+      return false if already_upvoted?(ip)
+
+      if already_downvoted?(ip)
+        votes.where(ip: ip, upvote: 0).first.delete
+        self.downvotes -= 1
+        save!
+      else
+        votes.create(ip: ip, upvote: 1).save
+        self.upvotes += 1
+        save!
+      end
+    end
+
+    def downvote!(ip)
+      return false if already_downvoted?(ip)
+
+      if already_upvoted?(ip)
+        votes.where(ip: ip, upvote: 1).first.delete
+        self.upvotes -= 1
+        save!
+      else
+        votes.create(ip: ip, upvote: 0).save
+        self.downvotes += 1
+        save!
+      end
+    end
+  end
+end
@@ -0,0 +1,6 @@
+class Vote < ApplicationRecord
+  belongs_to :votable, polymorphic: true
+
+  validates :upvote, presence: true, inclusion: { in: [0, 1] }
+  validates :ip, presence: true
+end