gitlab-org--gitlab-foss/app/models/protectable_dropdown.rb

45 lines
1.1 KiB
Ruby

# frozen_string_literal: true
class ProtectableDropdown
REF_TYPES = %i[branches tags].freeze
REF_NAME_METHODS = {
branches: :branch_names,
tags: :tag_names
}.freeze
def initialize(project, ref_type)
raise ArgumentError, "invalid ref type `#{ref_type}`" unless ref_type.in?(REF_TYPES)
@project = project
@ref_type = ref_type
end
# Tags/branches which are yet to be individually protected
def protectable_ref_names
return [] if @project.empty_repo?
@protectable_ref_names ||= ref_names - non_wildcard_protected_ref_names
end
def hash
protectable_ref_names.map { |ref_name| { text: ref_name, id: ref_name, title: ref_name } }
end
private
def ref_names
@project.repository.public_send(ref_name_method) # rubocop:disable GitlabSecurity/PublicSend
end
def ref_name_method
REF_NAME_METHODS[@ref_type]
end
def protections
@project.public_send("protected_#{@ref_type}") # rubocop:disable GitlabSecurity/PublicSend
end
def non_wildcard_protected_ref_names
protections.reject(&:wildcard?).map(&:name)
end
end