debian-mirror-gitlab/app/models/concerns/issue_available_features.rb

# frozen_string_literal: true

# Verifies features availability based on issue type.
# This can be used, for example, for hiding UI elements or blocking specific
# quick actions for particular issue types;
module IssueAvailableFeatures
  extend ActiveSupport::Concern

  class_methods do
    # EE only features are listed on EE::IssueAvailableFeatures
    def available_features_for_issue_types
      {
        assignee: %w(issue incident),
        confidentiality: %(issue incident)
      }.with_indifferent_access
    end
  end

  included do
    scope :with_feature, ->(feature) { where(issue_type: available_features_for_issue_types[feature]) }
  end

  def issue_type_supports?(feature)
    unless self.class.available_features_for_issue_types.has_key?(feature)
      raise ArgumentError, 'invalid feature'
    end

    self.class.available_features_for_issue_types[feature].include?(issue_type)
  end
end

IssueAvailableFeatures.prepend_if_ee('EE::IssueAvailableFeatures')
IssueAvailableFeatures::ClassMethods.prepend_if_ee('EE::IssueAvailableFeatures::ClassMethods')