values.rb

# frozen_string_literal: true

require "dry/schema/path"
require "dry/validation/constants"

module Dry
  module Validation
    # A convenient wrapper for data processed by schemas
    #
    # Values are available within the rule blocks. They act as hash-like
    # objects and expose a convenient API for accessing data.
    #
    # @api public
    class Values
      include ::Enumerable
      include ::Dry::Equalizer(:data)

      # Schema's result output
      #
      # @return [Hash]
      #
      # @api private
      attr_reader :data

      # @api private
      def initialize(data)
        @data = data
      end

      # Read from the provided key
      #
      # @example
      #   rule(:age) do
      #     key.failure('must be > 18') if values[:age] <= 18
      #   end
      #
      # @param args [Symbol, String, Hash, Array<Symbol>] If given as a single
      #   Symbol, String, Array or Hash, build a key array using
      #   {Dry::Schema::Path} digging for data. If given as positional
      #   arguments, use these with Hash#dig on the data directly.
      #
      # @return [Object]
      #
      # @api public
      def [](*args)
        return data.dig(*args) if args.size > 1

        case (key = args[0])
        when ::Symbol, ::String, ::Array, ::Hash
          keys = ::Dry::Schema::Path[key].to_a

          return data.dig(*keys) unless keys.last.is_a?(Array)

          last = keys.pop
          vals = self.class.new(data.dig(*keys))
          vals.fetch_values(*last) { nil }
        else
          raise ::ArgumentError, "+key+ must be a valid path specification"
        end
      end

      # @api public
      # rubocop: disable Metrics/PerceivedComplexity
      def key?(key, hash = data)
        return hash.key?(key) if key.is_a?(::Symbol)

        Schema::Path[key].reduce(hash) do |a, e|
          if e.is_a?(::Array)
            return e.all? { |k| key?(k, a) }
          elsif (e.is_a?(::Symbol) && a.is_a?(::Array)) || a.nil? || a.is_a?(::String)
            return false
          else
            return false unless a.is_a?(::Array) ? (e >= 0 && e < a.size) : a.key?(e)
          end

          a[e]
        end
        true
      end
      # rubocop: enable Metrics/PerceivedComplexity

      # @api private
      def respond_to_missing?(meth, include_private = false)
        super || data.respond_to?(meth, include_private)
      end

      private

      # @api private
      def method_missing(meth, ...)
        if data.respond_to?(meth)
          data.public_send(meth, ...)
        else
          super
        end
      end
    end
  end
end