Files
danbooru/app/models/post_flag.rb
2013-02-19 12:27:17 -05:00

83 lines
1.8 KiB
Ruby

class PostFlag < ActiveRecord::Base
class Error < Exception ; end
belongs_to :creator, :class_name => "User"
belongs_to :post
validates_presence_of :reason, :creator_id, :creator_ip_addr
validate :validate_creator_is_not_limited
validate :validate_post_is_active
before_validation :initialize_creator, :on => :create
validates_uniqueness_of :creator_id, :scope => :post_id, :message => "have already flagged this post"
before_save :update_post
module SearchMethods
def resolved
where("is_resolved = ?", true)
end
def unresolved
where("is_resolved = ?", false)
end
def old
where("created_at <= ?", 3.days.ago)
end
def search(params)
q = scoped
return q if params.blank?
if params[:creator_id].present?
q = q.where("creator_id = ?", params[:creator_id].to_i)
end
if params[:creator_name].present?
q = q.where("creator_id = (select _.id from users _ where lower(_.name) = ?)", params[:creator_name].downcase)
end
if params[:post_id].present?
q = q.where("post_id = ?", params[:post_id].to_i)
end
q
end
end
extend SearchMethods
def update_post
post.update_column(:is_flagged, true)
end
def validate_creator_is_not_limited
if flag_count_for_creator >= 10
errors[:creator] << "can flag 10 posts a day"
false
else
true
end
end
def validate_post_is_active
if post.is_deleted?
errors[:post] << "is deleted"
false
else
true
end
end
def initialize_creator
self.creator_id = CurrentUser.id
self.creator_ip_addr = CurrentUser.ip_addr
end
def resolve!
update_column(:is_resolved, true)
end
def flag_count_for_creator
PostAppeal.for_user(creator_id).recent.count
end
end