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
|
# frozen_string_literal: true
module Import
module GitlabProjects
# Validates the given object's #content_type and #content_length accordingly
# with the Project Import requirements
class RemoteFileValidator < ActiveModel::Validator
ALLOWED_CONTENT_TYPES = [
'application/gzip',
# S3 uses different file types
'application/x-tar',
'application/x-gzip'
].freeze
def validate(record)
validate_content_length(record)
validate_content_type(record)
end
private
def validate_content_length(record)
if record.content_length.to_i <= 0
record.errors.add(:content_length, :size_too_small, file_size: humanize(1.byte))
elsif file_size_limit > 0 && record.content_length > file_size_limit
record.errors.add(:content_length, :size_too_big, file_size: humanize(file_size_limit))
end
end
def humanize(number)
ActiveSupport::NumberHelper.number_to_human_size(number)
end
def validate_content_type(record)
return if ALLOWED_CONTENT_TYPES.include?(record.content_type)
record.errors.add(:content_type, "'%{content_type}' not allowed. (Allowed: %{allowed})" % {
content_type: record.content_type,
allowed: ALLOWED_CONTENT_TYPES.join(', ')
})
end
def file_size_limit
Gitlab::CurrentSettings.current_application_settings.max_import_remote_file_size.megabytes
end
end
end
end
|
