105 lines
2.3 KiB
Ruby
105 lines
2.3 KiB
Ruby
class Vote < ActiveRecord::Base
|
|
belongs_to :photo
|
|
belongs_to :user
|
|
|
|
validates_presence_of :photo_id
|
|
validate :photo_id_cannot_be_zero
|
|
validates_uniqueness_of :user_id, :scope => :photo_id
|
|
validates_uniqueness_of :session_id, :scope => [ :photo_id, :user_id ]
|
|
validates_presence_of :session_id, :if => lambda { |x| x.user_id.to_s.empty? }
|
|
validates_presence_of :user_id, :if => lambda { |x| x.session_id.to_s.empty? }
|
|
|
|
attr_protected :user_id
|
|
attr_protected :session_id
|
|
|
|
after_create :update_photo_stats
|
|
after_destroy :update_photo_stats
|
|
|
|
##
|
|
# Checks if this vote is anonymous, or not an authenticated User vote.
|
|
#
|
|
def anonymous?
|
|
self.user_id.nil?
|
|
end
|
|
|
|
##
|
|
# Convert this Vote to a number.
|
|
#
|
|
def to_i
|
|
self.vote? ? 1 : 0
|
|
end
|
|
|
|
##
|
|
# Is this a 'no' vote.
|
|
#
|
|
def zero?
|
|
self.to_i == 0
|
|
end
|
|
|
|
##
|
|
# Is this a 'yes' vote.
|
|
#
|
|
def one?
|
|
self.to_i == 1
|
|
end
|
|
|
|
##
|
|
# Checks if a user has voted for a Photo. If you pass a User for user it will
|
|
# check for an authenticated user, else it will look for an anonymous vote.
|
|
#
|
|
def self.voted_for?(photo, user)
|
|
c = [ 'photo_id = :pid' ]
|
|
v = { :pid => (photo.respond_to?('id') ? photo.id : photo) }
|
|
if user.respond_to?('user_name')
|
|
c << 'user_id = :uid'
|
|
v[:uid] = user.id
|
|
else
|
|
c << 'session_id = :uid'
|
|
v[:uid] = user
|
|
end
|
|
self.find :first, :conditions => [ c.join(' AND '), v ]
|
|
end
|
|
|
|
##
|
|
# Does a quick find and collect on the cast votes so you can find all voted
|
|
# photo ids.
|
|
#
|
|
def self.voted_photo_ids(user)
|
|
c = if user.respond_to?('user_name')
|
|
"votes.user_id = #{user.id}"
|
|
else
|
|
"votes.session_id = '#{user}'"
|
|
end
|
|
self.find(:all, :conditions => c, :select => 'photo_id').collect { |v| v.photo_id }
|
|
end
|
|
|
|
protected
|
|
|
|
##
|
|
# Checks to make sure the photo_id is not zero. There will never be a zero
|
|
# id of a Photo.
|
|
#
|
|
def photo_id_cannot_be_zero
|
|
self.errors.add :photo_id, 'cannot be zero' if self.photo_id.to_i == 0
|
|
end
|
|
|
|
##
|
|
# Recalc the stats in the Photo for specific stats don't have to query the
|
|
# database.
|
|
#
|
|
def update_photo_stats
|
|
v = if self.frozen?
|
|
-1
|
|
else
|
|
1
|
|
end
|
|
self.photo.votes_count += v
|
|
if self.zero?
|
|
self.photo.zero_votes += v
|
|
else
|
|
self.photo.one_votes += v
|
|
end
|
|
self.photo.save
|
|
end
|
|
end
|