dry-rb/dry-schema

View on GitHub
lib/dry/schema/extensions/struct.rb

Summary

Maintainability
A
35 mins
Test Coverage
# frozen_string_literal: true

require "dry/struct"
require "dry/schema/predicate_inferrer"
require "dry/schema/primitive_inferrer"
require "dry/schema/macros/dsl"
require "dry/schema/macros/hash"

module Dry
  module Schema
    module Macros
      Hash.prepend(::Module.new {
        def call(*args)
          if args.size >= 1 && args[0].is_a?(::Class) && args[0] <= ::Dry::Struct
            if block_given?
              raise ArgumentError, "blocks are not supported when using "\
                                   "a struct class (#{name.inspect} => #{args[0]})"
            end

            super(args[0].schema, *args.drop(1))
            type(schema_dsl.types[name].constructor(args[0]))
          else
            super
          end
        end
      })
    end

    PredicateInferrer::Compiler.send(:alias_method, :visit_struct, :visit_hash)
    PrimitiveInferrer::Compiler.send(:alias_method, :visit_struct, :visit_hash)
  end
end