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

51 lines
1.5 KiB
Ruby
Raw Normal View History

2020-05-24 23:13:21 +05:30
# frozen_string_literal: true
module Limitable
extend ActiveSupport::Concern
2020-06-23 00:09:42 +05:30
GLOBAL_SCOPE = :limitable_global_scope
2020-05-24 23:13:21 +05:30
included do
class_attribute :limit_scope
class_attribute :limit_name
2021-06-08 01:23:25 +05:30
class_attribute :limit_feature_flag
2020-05-24 23:13:21 +05:30
self.limit_name = self.name.demodulize.tableize
validate :validate_plan_limit_not_exceeded, on: :create
end
private
def validate_plan_limit_not_exceeded
2020-06-23 00:09:42 +05:30
if GLOBAL_SCOPE == limit_scope
validate_global_plan_limit_not_exceeded
else
validate_scoped_plan_limit_not_exceeded
end
end
def validate_scoped_plan_limit_not_exceeded
2020-05-24 23:13:21 +05:30
scope_relation = self.public_send(limit_scope) # rubocop:disable GitlabSecurity/PublicSend
return unless scope_relation
2021-06-08 01:23:25 +05:30
return if limit_feature_flag && ::Feature.disabled?(limit_feature_flag, scope_relation, default_enabled: :yaml)
2020-05-24 23:13:21 +05:30
relation = self.class.where(limit_scope => scope_relation)
2020-06-23 00:09:42 +05:30
limits = scope_relation.actual_limits
2020-05-24 23:13:21 +05:30
2020-06-23 00:09:42 +05:30
check_plan_limit_not_exceeded(limits, relation)
end
def validate_global_plan_limit_not_exceeded
relation = self.class.all
limits = Plan.default.actual_limits
check_plan_limit_not_exceeded(limits, relation)
end
def check_plan_limit_not_exceeded(limits, relation)
return unless limits.exceeded?(limit_name, relation)
errors.add(:base, _("Maximum number of %{name} (%{count}) exceeded") %
{ name: limit_name.humanize(capitalize: false), count: limits.public_send(limit_name) }) # rubocop:disable GitlabSecurity/PublicSend
2020-05-24 23:13:21 +05:30
end
end