module Schemacop
  class NodeSupportingField < NodeWithBlock
    block_method :req?
    block_method :req!
    block_method :req
    block_method :opt?
    block_method :opt!
    block_method :opt

    attr_reader :fields

    def initialize(options = {}, &block)
      @fields = {}
      super
      exec_block(&block)
    end

    def req?(*args, &block)
      field(*args, required: true, allow_nil: true, &block)
    end

    def req!(*args, &block)
      field(*args, required: true, allow_nil: false, &block)
    end

    alias req req!

    def opt?(*args, &block)
      field(*args, required: false, allow_nil: true, &block)
    end

    def opt!(*args, &block)
      field(*args, required: false, allow_nil: false, &block)
    end

    alias opt opt?

    protected

    def field(*args, required:, allow_nil:, &block)
      # name = args.shift
      # options = args.last.is_a?(Hash) ? args.pop : {}
      name = args.shift

      # rubocop: disable Style/IfInsideElse
      if @fields[name]
        @fields[name].type(*args, &block)
      else
        if args.any?
          @fields[name] = FieldNode.new(name, required) do
            type(*args, &block)
          end
        else
          @fields[name] = FieldNode.new(name, required, &block)
        end
      end
      # rubocop: enable Style/IfInsideElse

      @fields[name].type(:nil) if allow_nil
    end
  end
end