debian-mirror-gitlab/app/finders/alert_management/alerts_finder.rb
2021-06-08 01:23:25 +05:30

70 lines
1.9 KiB
Ruby

# frozen_string_literal: true
module AlertManagement
class AlertsFinder
# @return [Hash<Symbol,Integer>] Mapping of status id to count
# ex) { triggered: 6, ...etc }
def self.counts_by_status(current_user, project, params = {})
new(current_user, project, params).execute.counts_by_status
end
def initialize(current_user, project, params)
@current_user = current_user
@project = project
@params = params
end
def execute
return AlertManagement::Alert.none unless authorized?
collection = project.alert_management_alerts
collection = by_domain(collection)
collection = by_status(collection)
collection = by_iid(collection)
collection = by_assignee(collection)
collection = by_search(collection)
sort(collection)
end
private
attr_reader :current_user, :project, :params
def by_domain(collection)
return collection if params[:iid].present?
collection.with_operations_alerts
end
def by_iid(collection)
return collection unless params[:iid]
collection.for_iid(params[:iid])
end
def by_status(collection)
values = AlertManagement::Alert.status_names & Array(params[:status])
values.present? ? collection.for_status(values) : collection
end
def by_search(collection)
params[:search].present? ? collection.search(params[:search]) : collection
end
def sort(collection)
params[:sort] ? collection.sort_by_attribute(params[:sort]) : collection
end
def by_assignee(collection)
params[:assignee_username].present? ? collection.for_assignee_username(params[:assignee_username]) : collection
end
def authorized?
Ability.allowed?(current_user, :read_alert_management_alert, project)
end
end
end
AlertManagement::AlertsFinder.prepend_mod_with('AlertManagement::AlertsFinder')