25 lines
972 B
Ruby
25 lines
972 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
# Custom validator to ensure that a field using ActiveStorage
|
||
|
|
# * matches the given formats, specified with regex or by a list (leave empty to allow any)
|
||
|
|
# * is less than the given maximum (if none is given, the default is 500kb)
|
||
|
|
class AttachmentValidator < ActiveModel::EachValidator
|
||
|
|
def validate_each(record, attribute, value)
|
||
|
|
return unless value&.attached?
|
||
|
|
|
||
|
|
allowed_formats = options[:allowed_formats]
|
||
|
|
maximum_size = options[:maximum_size] || 500.kilobytes
|
||
|
|
|
||
|
|
case allowed_formats
|
||
|
|
when Regexp
|
||
|
|
record.errors.add(attribute, :invalid_format) unless allowed_formats.match?(value.content_type)
|
||
|
|
when Array
|
||
|
|
record.errors.add(attribute, :invalid_format) unless allowed_formats.include?(value.content_type)
|
||
|
|
end
|
||
|
|
|
||
|
|
record.errors.add(attribute, :too_large, maximum_size: maximum_size.to_fs(:human_size)) unless value.blob.byte_size < maximum_size
|
||
|
|
|
||
|
|
value.purge if record.errors[attribute].any?
|
||
|
|
end
|
||
|
|
end
|