Class: FileSizeValidator

Inherits:
ActiveModel::EachValidator
  • Object
show all
Defined in:
lib/file_size_validator.rb

Defined Under Namespace

Classes: Helper

Constant Summary collapse

MESSAGES =
{ is: :wrong_size, minimum: :size_too_small, maximum: :size_too_big }.freeze
CHECKS =
{ is: :==, minimum: :>=, maximum: :<= }.freeze
DEFAULT_TOKENIZER =
-> (value) { value.split(//) }.freeze
RESERVED_OPTIONS =
[:minimum, :maximum, :within, :is, :tokenizer, :too_short, :too_long].freeze

Instance Method Summary collapse

Constructor Details

#initialize(options) ⇒ FileSizeValidator

Returns a new instance of FileSizeValidator


8
9
10
11
12
13
14
15
16
# File 'lib/file_size_validator.rb', line 8

def initialize(options)
  if range = (options.delete(:in) || options.delete(:within))
    raise ArgumentError, ":in and :within must be a Range" unless range.is_a?(Range)
    options[:minimum], options[:maximum] = range.begin, range.end
    options[:maximum] -= 1 if range.exclude_end?
  end

  super
end

Instance Method Details

#check_validity!Object


18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# File 'lib/file_size_validator.rb', line 18

def check_validity!
  keys = CHECKS.keys & options.keys

  if keys.empty?
    raise ArgumentError, 'Range unspecified. Specify the :within, :maximum, :minimum, or :is option.'
  end

  keys.each do |key|
    value = options[key]

    unless (value.is_a?(Integer) && value >= 0) || value.is_a?(Symbol)
      raise ArgumentError, ":#{key} must be a nonnegative Integer or symbol"
    end
  end
end

#helpObject


65
66
67
# File 'lib/file_size_validator.rb', line 65

def help
  Helper.instance
end

#validate_each(record, attribute, value) ⇒ Object

Raises:

  • (ArgumentError)

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
# File 'lib/file_size_validator.rb', line 34

def validate_each(record, attribute, value)
  raise(ArgumentError, "A CarrierWave::Uploader::Base object was expected") unless value.kind_of? CarrierWave::Uploader::Base

  value = (options[:tokenizer] || DEFAULT_TOKENIZER).call(value) if value.kind_of?(String)

  CHECKS.each do |key, validity_check|
    next unless check_value = options[key]

    check_value =
      case check_value
      when Integer
        check_value
      when Symbol
        record.send(check_value)
      end

    value ||= [] if key == :maximum

    value_size = value.size
    next if value_size.send(validity_check, check_value)

    errors_options = options.except(*RESERVED_OPTIONS)
    errors_options[:file_size] = help.number_to_human_size check_value

    default_message = options[MESSAGES[key]]
    errors_options[:message] ||= default_message if default_message

    record.errors.add(attribute, MESSAGES[key], errors_options)
  end
end