# frozen_string_literal: true

require 'json-schema'

module OpenApi
  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
end