cypriss/mutations

module Mutations
  class ArrayFilter < InputFilter
    def self.register_additional_filter(type_class, type_name)
      define_method(type_name) do |options = {}, &block|
        @element_filter = type_class.new(options, &block)
      end
    end

    @default_options = {
      :nils => false,            # true allows an explicit nil to be valid. Overrides any other options
      :class => nil,             # A constant or string indicates that each element of the array needs to be one of these classes
      :arrayize => false,        # true will convert "hi" to ["hi"]. "" converts to []
      :min_length => nil,        # Can be a number like 5, meaning 5 objects are required
      :max_length => nil         # Can be a number like 20, meaning no more than 20 objects
    }

    def initialize(name, opts = {}, &block)
      super(opts)

      @name = name
      @element_filter = nil

      if block_given?
        instance_eval(&block)
      end

      raise ArgumentError.new("Can't supply both a class and a filter") if @element_filter && self.options[:class]
    end

    def hash(options = {}, &block)
      @element_filter = HashFilter.new(options, &block)
    end

    def model(name, options = {})
      @element_filter = ModelFilter.new(name.to_sym, options)
    end

    def array(options = {}, &block)
      @element_filter = ArrayFilter.new(nil, options, &block)
    end

    def initialize_constants!
      @initialize_constants ||= begin
        if options[:class]
          options[:class] = options[:class].constantize if options[:class].is_a?(String)
        end

        true
      end

      unless Mutations.cache_constants?
        options[:class] = options[:class].to_s.constantize if options[:class]
      end
    end

    def filter(data)
      initialize_constants!

      # Handle nil case
      if data.nil?
        return [nil, nil] if options[:nils]
        return [nil, :nils]
      end

      if !data.is_a?(Array) && options[:arrayize]
        return [[], nil] if data == ""
        data = Array(data)
      end

      if data.is_a?(Array)
        errors = ErrorArray.new
        filtered_data = []
        found_error = false

        return [data, :min_length] if options[:min_length] && data.length < options[:min_length]
        return [data, :max_length] if options[:max_length] && data.length > options[:max_length]
        data.each_with_index do |el, i|
          el_filtered, el_error = filter_element(el)
          el_error = ErrorAtom.new(@name, el_error, :index => i) if el_error.is_a?(Symbol)
          errors << el_error
          if el_error
            found_error = true
          else
            filtered_data << el_filtered
          end
        end

        if found_error && !(@element_filter && @element_filter.discard_invalid?)
          [data, errors]
        else
          [filtered_data, nil]
        end
      else
        return [data, :array]
      end
    end

    # Returns [filtered, errors]
    def filter_element(data)
      if @element_filter
        data, el_errors = @element_filter.filter(data)
        return [data, el_errors] if el_errors
      elsif options[:class]
        if !data.is_a?(options[:class])
          return [data, :class]
        end
      end

      [data, nil]
    end
  end
end