dmeremyanin/normalizr

module Normalizr
  module Concern
    def self.included(base)
      base.extend(ClassMethods)
    end

    module ClassMethods
      def normalize(*args, &block)
        options = Normalizr::OptionsParser.new(args, block)

        prepend Module.new {
          options.attributes.each do |method|
            define_method :"#{method}=" do |value, *method_args|
              condition_lambda = -> condition { Proc === condition ? instance_exec(&condition) : send(condition) }

              positive = options.positive_condition.all?(&condition_lambda)
              negative = options.negative_condition.none?(&condition_lambda)

              if positive && negative
                value = Normalizr.normalize(value, *options.before)
                value = Normalizr.normalize(value, *options.after) if options.after.any?
              end

              super(value, *method_args)
            end
          end
        }
      end

      # attribute normalizer compatibility
      alias_method :normalize_attribute,  :normalize
      alias_method :normalize_attributes, :normalize
    end
  end
end