1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# frozen_string_literal: true
require 'active_support/inflector'
module InjectEnterpriseEditionModule
def prepend_mod_with(constant_name, namespace: Object, with_descendants: false)
each_extension_for(constant_name, namespace) do |constant|
prepend_module(constant, with_descendants)
end
end
def extend_mod_with(constant_name, namespace: Object)
each_extension_for(constant_name, namespace) do |constant|
extend constant
end
end
def include_mod_with(constant_name, namespace: Object)
each_extension_for(constant_name, namespace) do |constant|
include constant
end
end
def prepend_mod(with_descendants: false)
prepend_mod_with(name, with_descendants: with_descendants) # rubocop: disable Cop/InjectEnterpriseEditionModule
end
def extend_mod
extend_mod_with(name) # rubocop: disable Cop/InjectEnterpriseEditionModule
end
def include_mod
include_mod_with(name) # rubocop: disable Cop/InjectEnterpriseEditionModule
end
def gitlab_extensions
extensions = [self]
each_extension_for(name, Object) { |c| extensions << c }
extensions
end
private
def prepend_module(mod, with_descendants)
prepend(mod)
if with_descendants
descendants.each { |descendant| descendant.prepend(mod) }
end
end
def each_extension_for(constant_name, namespace)
GitlabEdition.extensions.each do |extension_name|
extension_namespace =
const_get_maybe_false(namespace, extension_name.upcase)
extension_module =
const_get_maybe_false(extension_namespace, constant_name)
yield(extension_module) if extension_module
end
end
def const_get_maybe_false(mod, name)
mod && mod.const_defined?(name, false) && mod.const_get(name, false)
end
end
Module.prepend(InjectEnterpriseEditionModule)
|
