mirror of https://github.com/Siphonay/mastodon
33 lines
881 B
Ruby
33 lines
881 B
Ruby
# frozen_string_literal: true
|
|
|
|
class DirectFeed < Feed
|
|
include Redisable
|
|
|
|
def initialize(account)
|
|
@account = account
|
|
super(:direct, account.id)
|
|
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
|
|
|
|
# TODO: _min_id is not actually handled by `as_direct_timeline`
|
|
def from_database(limit, max_id, since_id, _min_id)
|
|
loop do
|
|
statuses = Status.as_direct_timeline(@account, limit, max_id, since_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
|