66 lines
1.3 KiB
Ruby
66 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class EventFilter
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
attr_accessor :filter
|
|
|
|
ALL = 'all'
|
|
PUSH = 'push'
|
|
MERGED = 'merged'
|
|
ISSUE = 'issue'
|
|
COMMENTS = 'comments'
|
|
TEAM = 'team'
|
|
WIKI = 'wiki'
|
|
DESIGNS = 'designs'
|
|
|
|
def initialize(filter)
|
|
# Split using comma to maintain backward compatibility Ex/ "filter1,filter2"
|
|
filter = filter.to_s.split(',')[0].to_s
|
|
@filter = filters.include?(filter) ? filter : ALL
|
|
end
|
|
|
|
def active?(key)
|
|
filter == key.to_s
|
|
end
|
|
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
def apply_filter(events)
|
|
case filter
|
|
when PUSH
|
|
events.pushed_action
|
|
when MERGED
|
|
events.merged_action
|
|
when COMMENTS
|
|
events.commented_action
|
|
when TEAM
|
|
events.where(action: [:joined, :left, :expired])
|
|
when ISSUE
|
|
events.where(action: [:created, :updated, :closed, :reopened], target_type: 'Issue')
|
|
when WIKI
|
|
wiki_events(events)
|
|
when DESIGNS
|
|
design_events(events)
|
|
else
|
|
events
|
|
end
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
|
|
private
|
|
|
|
def wiki_events(events)
|
|
events.for_wiki_page
|
|
end
|
|
|
|
def design_events(events)
|
|
events.for_design
|
|
end
|
|
|
|
def filters
|
|
[ALL, PUSH, MERGED, ISSUE, COMMENTS, TEAM, WIKI, DESIGNS]
|
|
end
|
|
end
|
|
|
|
EventFilter.prepend_mod_with('EventFilter')
|