moebooru/app/models/post.rb

Dir["#{RAILS_ROOT}/app/models/post/**/*.rb"].each {|x| require_dependency x}

class Post < ActiveRecord::Base
  STATUSES = %w(active pending flagged deleted)
  
  define_callbacks :after_delete
  define_callbacks :after_undelete
  has_many :notes, :order => "id desc"
  has_one :flag_detail, :class_name => "FlaggedPostDetail"
  belongs_to :user
  before_validation_on_create :set_random!
  before_create :set_index_timestamp!
  belongs_to :approver, :class_name => "User"
  attr_accessor :updater_ip_addr, :updater_user_id
  attr_accessor :metatag_flagged
  has_many :avatars, :class_name => "User", :foreign_key => "avatar_post_id"
  after_delete :clear_avatars
  after_save :commit_flag
  
  include PostSqlMethods
  include PostCommentMethods
  include PostImageStoreMethods
  include PostVoteMethods
  include PostTagMethods
  include PostCountMethods
  include PostCacheMethods if CONFIG["enable_caching"]
  include PostParentMethods if CONFIG["enable_parent_posts"]
  include PostFileMethods
  include PostChangeSequenceMethods
  include PostRatingMethods
  include PostStatusMethods
  include PostApiMethods
  include PostMirrorMethods
  
  def self.destroy_with_reason(id, reason, current_user)
    post = Post.find(id)
    post.flag!(reason, current_user.id)
    if post.flag_detail
      post.flag_detail.update_attributes(:is_resolved => true)
    end

    post.delete
  end

  def delete
    self.update_attributes(:status => "deleted")
    self.run_callbacks(:after_delete)
  end

  def undelete
    return if self.status == "active"
    self.update_attributes(:status => "active")
    self.run_callbacks(:after_undelete)
  end
  
  def can_user_delete?(user)
    if not user.has_permission?(self)
      return false
    end

    if not user.is_mod_or_higher? and Time.now - self.created_at > 1.day
      return false
    end

    return true
  end

  def clear_avatars
    User.clear_avatars(self.id)
  end

  def set_random!
    self.random = rand;
  end

  def set_index_timestamp!
    self.index_timestamp = self.created_at
  end

  def flag!(reason, creator_id)
    transaction do
      update_attributes(:status => "flagged")
      
      if flag_detail
        flag_detail.update_attributes(:reason => reason, :user_id => creator_id, :created_at => Time.now)
      else
        FlaggedPostDetail.create!(:post_id => id, :reason => reason, :user_id => creator_id, :is_resolved => false)
      end
    end
  end
  
  # If the flag_post metatag was used and the current user has access, flag the post.
  def commit_flag
    return if self.metatag_flagged.nil?
    return if not Thread.current["danbooru-user"].is_mod_or_higher?
    return if self.status != "active"

    self.flag!(self.metatag_flagged, Thread.current["danbooru-user"].id)
  end

  def approve!(approver_id)
    if flag_detail
      flag_detail.update_attributes(:is_resolved => true)
    end
    
    update_attributes(:status => "active", :approver_id => approver_id)
  end
  
  def voted_by
    # Cache results
    if @voted_by.nil?
      @voted_by = {}
      (1..3).each { |v|
        @voted_by[v] = User.find(:all, :joins => "JOIN post_votes v ON v.user_id = users.id", :select => "users.name, users.id", :conditions => ["v.post_id = ? and v.score = ?", self.id, v], :order => "v.updated_at DESC") || []
      }
    end

    return @voted_by
  end

  def favorited_by
    return voted_by[3]
  end

  def author
    return User.find_name(user_id)
  end
  
  def delete_from_database
    delete_file
    execute_sql("DELETE FROM posts WHERE id = ?", id)
  end
  
  def active_notes
    notes.select {|x| x.is_active?}
  end
  
  STATUSES.each do |x|
    define_method("is_#{x}?") do
      return status == x
    end
  end
  
  def can_be_seen_by?(user, options={})
    if not options[:show_deleted] and self.status == 'deleted'
      return false
    end
    CONFIG["can_see_post"].call(user, self)
  end
  
  def self.new_deleted?(user)
    conds = []
    conds += ["creator_id <> %d" % [user.id]] unless user.is_anonymous?

    newest_topic = ForumPost.find(:first, :order => "id desc", :limit => 1, :select => "created_at", :conditions => conds)
    return false if newest_topic == nil
    return newest_topic.created_at > user.last_forum_topic_read_at
  end

  def normalized_source
    if source =~ /pixiv\.net\/img\//
      img_id = source[/(\d+)\.\w+$/, 1]
      "http://www.pixiv.net/member_illust.php?mode=medium&illust_id=#{img_id}"
    else
      source
    end
  end
end