module Bootsnap::CompileCache::YAML

def init!
  require('yaml')
  require('msgpack')
  require('date')
  # MessagePack serializes symbols as strings by default.
  # We want them to roundtrip cleanly, so we use a custom factory.
  # see: https://github.com/msgpack/msgpack-ruby/pull/122
  factory = MessagePack::Factory.new
  factory.register_type(0x00, Symbol)
  factory.register_type(
    MessagePack::Timestamp::TYPE, # or just -1
    Time,
    packer: MessagePack::Time::Packer,
    unpacker: MessagePack::Time::Unpacker
  )
  marshal_fallback = {
    packer: ->(value) { Marshal.dump(value) },
    unpacker: ->(payload) { Marshal.load(payload) },
  }
  {
    Date => 0x01,
    Regexp => 0x02,
  }.each do |type, code|
    factory.register_type(code, type, marshal_fallback)
  end
  self.msgpack_factory = factory
  self.supported_options = []
  params = ::YAML.method(:load).parameters
  if params.include?([:key, :symbolize_names])
    self.supported_options << :symbolize_names
  end
  if params.include?([:key, :freeze])
    if factory.load(factory.dump('yaml'), freeze: true).frozen?
      self.supported_options << :freeze
    end
  end
  self.supported_options.freeze
end