dry-validation/lib/dry/validation/schema/rule.rb

require 'dry/validation/deprecations'

module Dry
  module Validation
    class Schema
      class Rule < BasicObject
        extend ::Dry::Validation::Deprecations

        INVALID_PREDICATES = {
          value: [],
          maybe: [:empty?, :none?],
          filled: [:empty?, :filled?],
        }.freeze

        attr_reader :name, :node, :type, :target, :deps, :options

        def initialize(node, options = {})
          @node = node
          @type = options.fetch(:type, :and)
          @deps = options.fetch(:deps, [])
          @name = options.fetch(:name)
          @target = options.fetch(:target)
          @options = options
        end

        def inspect
          to_ast.inspect
        end
        alias_method :to_s, :inspect

        def schema(other = nil, &block)
          schema = Schema.create_class(target, other, &block)

          if schema.config.type_specs
            target.type_map[name] = schema.type_map
          end

          rule = __send__(type, key(:hash?).and(key(schema)))
          add_rule(rule)
        end

        def schema?
          target.schema?
        end

        def type_map
          target.type_map
        end

        def type_map?
          target.type_map?
        end

        def required(*predicates)
          warn 'required is deprecated - use filled instead.'

          filled(*predicates)
        end

        def filled(*predicates)
          rule = ([key(:filled?)] + infer_predicates(predicates, :filled)).reduce(:and)

          add_rule(__send__(type, rule))
        end

        def value(*predicates)
          ::Kernel.raise ::ArgumentError, "wrong number of arguments (given 0, expected at least 1)" if predicates.empty?

          rule = infer_predicates(predicates, :value).reduce(:and)

          add_rule(__send__(type, rule))
        end

        def maybe(*predicates)
          rule =
            if predicates.size > 0
              key(:none?).or(infer_predicates(predicates, :maybe).reduce(:and))
            else
              key(:none?).or(key(:filled?))
            end

          add_rule(__send__(type, rule))
        end

        def each(*predicates, &block)
          rule = target.each(*predicates, &block)
          add_rule(__send__(type, new([target.type, [name, rule.to_ast]])))
        end

        def add_rule(rule)
          target.add_rule(rule)
        end

        def rules
          target.rules
        end

        def checks
          target.checks
        end

        def rule_ast
          rules.size > 0 ? target.rule_ast : [to_ast]
        end

        def to_ast
          if deps.empty?
            node
          else
            [:guard, [deps, node]]
          end
        end

        def class
          Schema::Rule
        end

        def not
          new([:not, node])
        end

        def and(other)
          new([:and, [node, other.to_ast]])
        end
        alias_method :&, :and

        def or(other)
          new([:or, [node, other.to_ast]])
        end
        alias_method :|, :or

        def xor(other)
          new([:xor, [node, other.to_ast]])
        end
        alias_method :^, :xor

        def then(other)
          new([:implication, [node, other.to_ast]])
        end
        alias_method :>, :then

        def infer_predicates(predicates, macro = nil)
          predicates.map do |predicate|
            name, *args = ::Kernel.Array(predicate).first

            if macro && INVALID_PREDICATES[macro].include?(name)
              ::Kernel.raise InvalidSchemaError, "you can't use #{name} predicate with #{macro} macro"
            else
              key(name, args)
            end
          end
        end

        def with(new_options)
          self.class.new(node, options.merge(new_options))
        end

        private

        def key(predicate, args = [])
          new(target.node(predicate, *args))
        end

        def new(node)
          self.class.new(node, options)
        end
      end
    end
  end
end