Contents

# frozen_string_literal: true

class FileExtensionValidator < BaseValidator

  def validate_each(record, attribute, value)
    assert_valid_include_or_exclude!
    assign_attr_readers(record, attribute, value)
    valid?
  end

  private

  def assert_valid_include_or_exclude!
    return if options.key?(:include) || options.key?(:exclude)

    raise ArgumentError, 'Missing ":include" or ":exclude" attribute for comparison.'
  end

  def check
    options.key?(:exclude) ? :exclude : :include
  end

  def include_or_exclude
    values = options[:include] || options[:exclude]
    Array(values)
  end

  def valid_attr?
    extension = File.extname(value.original_filename).tr(".", "")
    return false if extension.empty?

    included = include_or_exclude.any? { |ext| ext.to_s == extension }
    included = !included if check == :exclude
    return if included

    record.errors.add(attribute, *error_message)
  end

end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
lite-validators-1.8.0	lib/lite/validators/file_extension_validator.rb