diff options
Diffstat (limited to 'app/models/direct_feed.rb')
-rw-r--r-- | app/models/direct_feed.rb | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/app/models/direct_feed.rb b/app/models/direct_feed.rb new file mode 100644 index 000000000..689a735b3 --- /dev/null +++ b/app/models/direct_feed.rb @@ -0,0 +1,32 @@ +# 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 |