rswag/rswag-specs/lib/rswag/specs/extended_schema.rb

# frozen_string_literal: true

require 'json-schema'

module Rswag
  module Specs
    class ExtendedSchema < JSON::Schema::Draft4
      def initialize
        super
        @attributes['type'] = ExtendedTypeAttribute
        @uri = URI.parse('http://tempuri.org/rswag/specs/extended_schema')
        @names = ['http://tempuri.org/rswag/specs/extended_schema']
      end
    end

    class ExtendedTypeAttribute < JSON::Schema::TypeV4Attribute
      def self.validate(current_schema, data, fragments, processor, validator, options = {})
        return if data.nil? && (current_schema.schema['nullable'] == true || current_schema.schema['x-nullable'] == true)

        super
      end
    end

    JSON::Validator.register_validator(ExtendedSchema.new)
  end
end