mirror of
https://github.com/lunaisnotaboy/mastodon.git
synced 2024-12-21 04:39:23 +00:00
32 lines
839 B
Ruby
32 lines
839 B
Ruby
# frozen_string_literal: true
|
|
|
|
class DirectFeed < Feed
|
|
include Redisable
|
|
|
|
def initialize(account)
|
|
@type = :direct
|
|
@id = account.id
|
|
@account = account
|
|
end
|
|
|
|
def get(limit, max_id = nil, since_id = nil, min_id = nil)
|
|
unless redis.exists("account:#{@account.id}:regeneration")
|
|
statuses = super
|
|
return statuses unless statuses.empty?
|
|
end
|
|
from_database(limit, max_id, since_id, min_id)
|
|
end
|
|
|
|
private
|
|
|
|
def from_database(limit, max_id, since_id, min_id)
|
|
loop do
|
|
statuses = Status.as_direct_timeline(@account, limit, max_id, since_id, min_id)
|
|
return statuses if statuses.empty?
|
|
max_id = statuses.last.id
|
|
statuses = statuses.reject { |status| FeedManager.instance.filter?(:direct, status, @account) }
|
|
return statuses unless statuses.empty?
|
|
end
|
|
end
|
|
end
|