require 'avro_turf'
require 'avro_turf/schema_registry'

module Avromatic
  module CacheableSchemaRegistration
    # Override register to first check if a schema is registered by fingerprint
    def register(subject, schema)
      return super unless Avromatic.use_cacheable_schema_registration

      schema_object = if schema.is_a?(String)
                        Avro::Schema.parse(schema)
                      else
                        schema
                      end

      registered = false
      data = begin
        get("/subjects/#{subject}/fingerprints/#{schema_object.sha256_fingerprint.to_s(16)}")
      rescue
        registered = true
        post("/subjects/#{subject}/versions", body: { schema: schema.to_s }.to_json)
      end

      id = data.fetch('id')

      @logger.info("#{registered ? 'Registered' : 'Found'} schema for subject `#{subject}`; id = #{id}")

      id
    end
  end
end

AvroTurf::SchemaRegistry.prepend(Avromatic::CacheableSchemaRegistration)