debian-mirror-gitlab/app/finders/alert_management/alerts_finder.rb

# 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)
      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_if_ee('EE::AlertManagement::AlertsFinder')