module Jimmy
  class SchemaCreation

    @handlers = {}

    class << self

      attr_reader :handlers

      def apply_to(klass, &handler)
        @handlers[klass] = handler
        %i(one all any).each do |condition|
          klass.__send__ :define_method, :"#{condition}_of" do |*args, &inner_block|
            Combination.new(condition, domain).tap do |combo|
              combo.evaluate inner_block
              instance_exec combo, *args, &handler
            end
          end
        end
        klass.include ResolveMissingToCustomType
      end
    end

    module ResolveMissingToCustomType

      def respond_to_missing?(method, *)
        SchemaTypes.key?(method) || domain.types.key?(method) || super
      end

      def method_missing(method, *args, &block)
        if SchemaTypes.key? method
          handler = SchemaCreation.handlers[self.class]
          self.class.__send__ :define_method, method do |*inner_args, &inner_block|
            handler_args = handler && inner_args.shift(handler.arity - 1)
            schema = Schema.new(method, domain, *inner_args, &inner_block)
            instance_exec schema, *handler_args, &handler if handler
            schema.dsl
          end
          return __send__ method, *args, &block
        end

        domain.autoload_type method

        if domain.types.key? method
          return instance_exec method, *args, &SchemaCreation.handlers[self.class]
        end

        super method, *args, &block
      end

    end

  end
end