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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
# frozen_string_literal: true
module Gitlab
class GlRepository
class ContainerClassMismatchError < StandardError
def initialize(container_class, repo_type)
@container_class = container_class
@repo_type = repo_type
end
def message
"Expected container class to be #{@repo_type.container_class} for " \
"repo type #{@repo_type.name}, but found #{@container_class} instead."
end
end
class RepoType
attr_reader :name,
:access_checker_class,
:repository_resolver,
:container_class,
:project_resolver,
:guest_read_ability,
:suffix
def initialize(
name:,
access_checker_class:,
repository_resolver:,
container_class: default_container_class,
project_resolver: nil,
guest_read_ability: :download_code,
suffix: nil)
@name = name
@access_checker_class = access_checker_class
@repository_resolver = repository_resolver
@container_class = container_class
@project_resolver = project_resolver
@guest_read_ability = guest_read_ability
@suffix = suffix
end
def identifier_for_container(container)
"#{name}-#{container.id}"
end
def wiki?
name == :wiki
end
def project?
name == :project
end
def snippet?
name == :snippet
end
def design?
name == :design
end
def path_suffix
suffix ? ".#{suffix}" : ''
end
def repository_for(container)
check_container(container)
return unless container
repository_resolver.call(container)
end
def project_for(container)
return container unless project_resolver
project_resolver.call(container)
end
def valid?(repository_path)
repository_path.end_with?(path_suffix) &&
(
!snippet? ||
repository_path.match?(Gitlab::PathRegex.full_snippets_repository_path_regex)
)
end
private
def default_container_class
Project
end
def check_container(container)
# Don't check container for wiki or project because these repo types
# accept several container types.
return if wiki? || project?
return unless container.present? && container_class.present?
return if container.is_a?(container_class)
raise ContainerClassMismatchError.new(container.class.name, self)
end
end
end
end
Gitlab::GlRepository::RepoType.prepend_mod_with('Gitlab::GlRepository::RepoType')
|