binaryattraction/app/models/vote.rb

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, :if => lambda { |x| x.session_id.to_s.empty? }
  validates_uniqueness_of :session_id, :scope => :photo_id, :if => lambda { |x| !x.session_id.to_s.empty? }
  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