@@ -1,19 +1,24 @@
 | 
			
		||||
# frozen_string_literal: true
 | 
			
		||||
 | 
			
		||||
class StatusRelationshipsPresenter
 | 
			
		||||
  attr_reader :reblogs_map, :favourites_map, :mutes_map
 | 
			
		||||
  attr_reader :reblogs_map, :favourites_map, :mutes_map, :pins_map
 | 
			
		||||
 | 
			
		||||
  def initialize(statuses, current_account_id = nil, reblogs_map: {}, favourites_map: {}, mutes_map: {})
 | 
			
		||||
  def initialize(statuses, current_account_id = nil, options = {})
 | 
			
		||||
    if current_account_id.nil?
 | 
			
		||||
      @reblogs_map    = {}
 | 
			
		||||
      @favourites_map = {}
 | 
			
		||||
      @mutes_map      = {}
 | 
			
		||||
      @pins_map       = {}
 | 
			
		||||
    else
 | 
			
		||||
      status_ids       = statuses.compact.flat_map { |s| [s.id, s.reblog_of_id] }.uniq
 | 
			
		||||
      conversation_ids = statuses.compact.map(&:conversation_id).compact.uniq
 | 
			
		||||
      @reblogs_map     = Status.reblogs_map(status_ids, current_account_id).merge(reblogs_map)
 | 
			
		||||
      @favourites_map  = Status.favourites_map(status_ids, current_account_id).merge(favourites_map)
 | 
			
		||||
      @mutes_map       = Status.mutes_map(conversation_ids, current_account_id).merge(mutes_map)
 | 
			
		||||
      statuses            = statuses.compact
 | 
			
		||||
      status_ids          = statuses.flat_map { |s| [s.id, s.reblog_of_id] }.uniq
 | 
			
		||||
      conversation_ids    = statuses.map(&:conversation_id).compact.uniq
 | 
			
		||||
      pinnable_status_ids = statuses.map(&:proper).select { |s| s.account_id == current_account_id && %w(public unlisted).include?(s.visibility) }.map(&:id)
 | 
			
		||||
 | 
			
		||||
      @reblogs_map     = Status.reblogs_map(status_ids, current_account_id).merge(options[:reblogs_map] || {})
 | 
			
		||||
      @favourites_map  = Status.favourites_map(status_ids, current_account_id).merge(options[:favourites_map] || {})
 | 
			
		||||
      @mutes_map       = Status.mutes_map(conversation_ids, current_account_id).merge(options[:mutes_map] || {})
 | 
			
		||||
      @pins_map        = Status.pins_map(pinnable_status_ids, current_account_id).merge(options[:pins_map] || {})
 | 
			
		||||
    end
 | 
			
		||||
  end
 | 
			
		||||
end
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user