Contents

require 'dry/data'
require 'dry/data/compiler'

module Dry
  module Validation
    class InputTypeCompiler
      attr_reader :type_compiler

      TYPES = {
        default: 'string',
        none?: 'form.nil',
        bool?: 'form.bool',
        str?: 'string',
        int?: 'form.int',
        float?: 'form.float',
        decimal?: 'form.decimal',
        date?: 'form.date',
        date_time?: 'form.date_time',
        time?: 'form.time'
      }.freeze

      DEFAULT_TYPE_NODE = [[:type, 'string']].freeze

      def initialize
        @type_compiler = Dry::Data::Compiler.new(Dry::Data)
      end

      def call(ast)
        schema = ast.map { |node| visit(node) }
        type_compiler.([:type, ['hash', [:symbolized, schema]]])
      end

      def visit(node, *args)
        send(:"visit_#{node[0]}", node[1], *args)
      end

      def visit_or(node, *args)
        left, right = node
        [:sum, [visit(left, *args), visit(right, *args)]]
      end

      def visit_and(node, first = true)
        if first
          name, type = node.map { |n| visit(n, false) }.uniq
          [:key, [name, type]]
        else
          result = node.map { |n| visit(n, first) }.uniq

          if result.size == 1
            result.first
          else
            (result - DEFAULT_TYPE_NODE).first
          end
        end
      end

      def visit_implication(node)
        key, types = node
        [:key, [visit(key), visit(types, false)]]
      end

      def visit_key(node, *args)
        node[0].to_s
      end

      def visit_val(node, *args)
        visit(node[1], false)
      end

      def visit_set(node, *args)
        [:type, ['form.hash', [:symbolized, node[1].map { |n| visit(n) }]]]
      end

      def visit_each(node, *args)
        [:type, ['form.array', visit(node[1], *args)]]
      end

      def visit_predicate(node, *args)
        [:type, TYPES[node[0]] || TYPES[:default]]
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
dry-validation-0.6.0	lib/dry/validation/input_type_compiler.rb
dry-validation-0.5.0	lib/dry/validation/input_type_compiler.rb