tootlab-mastodon/app/presenters/grouped_notifications_presenter.rb

# frozen_string_literal: true

class GroupedNotificationsPresenter < ActiveModelSerializers::Model
  def initialize(grouped_notifications, options = {})
    super()

    @grouped_notifications = grouped_notifications
    @options = options
  end

  def notification_groups
    @grouped_notifications
  end

  def statuses
    @grouped_notifications.filter_map(&:target_status).uniq(&:id)
  end

  def accounts
    @accounts ||= begin
      if partial_avatars?
        @grouped_notifications.map { |group| group.sample_accounts.first }.uniq(&:id)
      else
        @grouped_notifications.flat_map(&:sample_accounts).uniq(&:id)
      end
    end
  end

  def partial_accounts
    @partial_accounts ||= begin
      if partial_avatars?
        @grouped_notifications.flat_map { |group| group.sample_accounts[1...] }.uniq(&:id).filter { |account| accounts.exclude?(account) }
      else
        []
      end
    end
  end

  private

  def partial_avatars?
    @options[:expand_accounts] == 'partial_avatars'
  end
end