Software Público Brasileiro

# monkey patch to add comments on action_tracker

Rails.configuration.to_prepare do
  ActionTracker::Record.module_eval do

    has_many :comments, :class_name => 'Comment', :foreign_key => 'source_id', :dependent => :destroy, :finder_sql => 'SELECT * FROM comments WHERE #{conditions_for_comments} ORDER BY created_at ASC', :counter_sql => 'SELECT * FROM comments WHERE #{conditions_for_comments}'

    def conditions_for_comments
      type, id = (self.target_type == 'Article' ? ['Article', self.target_id] : [self.class.to_s, self.id])
      "source_type = '#{type}' AND source_id = '#{id}'"
    end

    def comments_as_thread
      result = {}
      root = []
      self.comments.each do |c|
        c.replies = []
        result[c.id] ||= c
        c.reply_of_id.nil? ? root << c : result[c.reply_of_id].replies << c
      end
      root
    end

  end
end