github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2024-11-25 09:42:07 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

callback in TopicQuery for plugins to add result set filters

Browse Source
This commit is contained in:
Neil Lalonde 2015-12-21 11:43:17 -05:00
parent a9099f9e23
commit b3198d7a6a
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lib/topic_query.rb
View File

@ -43,6 +43,9 @@ class TopicQuery
'created' => 'created_at' 'created' => 'created_at'
} }
cattr_accessor :results_filter_callbacks
self.results_filter_callbacks = []
def initialize(user=nil, options={}) def initialize(user=nil, options={})
options.assert_valid_keys(VALID_OPTIONS) options.assert_valid_keys(VALID_OPTIONS)
@options = options.dup @options = options.dup
@ -209,12 +212,23 @@ class TopicQuery
result = default_results(options) result = default_results(options)
result = remove_muted_topics(result, @user) unless options && options[:state] == "muted".freeze result = remove_muted_topics(result, @user) unless options && options[:state] == "muted".freeze
result = remove_muted_categories(result, @user, exclude: options[:category]) result = remove_muted_categories(result, @user, exclude: options[:category])
# plugins can remove topics here:
self.class.results_filter_callbacks.each do |filter_callback|
result = filter_callback.call(:latest, result, @user, options)
end
result result
end end
def unread_results(options={}) def unread_results(options={})
result = TopicQuery.unread_filter(default_results(options.reverse_merge(:unordered => true))) result = TopicQuery.unread_filter(default_results(options.reverse_merge(:unordered => true)))
.order('CASE WHEN topics.user_id = tu.user_id THEN 1 ELSE 2 END') .order('CASE WHEN topics.user_id = tu.user_id THEN 1 ELSE 2 END')
self.class.results_filter_callbacks.each do |filter_callback|
result = filter_callback.call(:unread, result, @user, options)
end
suggested_ordering(result, options) suggested_ordering(result, options)
end end
@ -224,6 +238,11 @@ class TopicQuery
result = TopicQuery.new_filter(default_results(options.reverse_merge(:unordered => true)), @user.treat_as_new_topic_start_date) result = TopicQuery.new_filter(default_results(options.reverse_merge(:unordered => true)), @user.treat_as_new_topic_start_date)
result = remove_muted_topics(result, @user) result = remove_muted_topics(result, @user)
result = remove_muted_categories(result, @user, exclude: options[:category]) result = remove_muted_categories(result, @user, exclude: options[:category])
self.class.results_filter_callbacks.each do |filter_callback|
result = filter_callback.call(:new, result, @user, options)
end
suggested_ordering(result, options) suggested_ordering(result, options)
end end