status_relationships_presenter.rb 1.3 KB

123456789101112131415161718192021222324252627
  1. # frozen_string_literal: true
  2. class StatusRelationshipsPresenter
  3. attr_reader :reblogs_map, :favourites_map, :mutes_map, :pins_map,
  4. :bookmarks_map
  5. def initialize(statuses, current_account_id = nil, **options)
  6. if current_account_id.nil?
  7. @reblogs_map = {}
  8. @favourites_map = {}
  9. @bookmarks_map = {}
  10. @mutes_map = {}
  11. @pins_map = {}
  12. else
  13. statuses = statuses.compact
  14. status_ids = statuses.flat_map { |s| [s.id, s.reblog_of_id] }.uniq.compact
  15. conversation_ids = statuses.map(&:conversation_id).compact.uniq
  16. pinnable_status_ids = statuses.map(&:proper).select { |s| s.account_id == current_account_id && %w(public unlisted).include?(s.visibility) }.map(&:id)
  17. @reblogs_map = Status.reblogs_map(status_ids, current_account_id).merge(options[:reblogs_map] || {})
  18. @favourites_map = Status.favourites_map(status_ids, current_account_id).merge(options[:favourites_map] || {})
  19. @bookmarks_map = Status.bookmarks_map(status_ids, current_account_id).merge(options[:bookmarks_map] || {})
  20. @mutes_map = Status.mutes_map(conversation_ids, current_account_id).merge(options[:mutes_map] || {})
  21. @pins_map = Status.pins_map(pinnable_status_ids, current_account_id).merge(options[:pins_map] || {})
  22. end
  23. end
  24. end