class ActiveFedora::Associations::BasicContainsAssociation

def configure_datastream(record)
  # If you called has_metadata with a block, pass the block into the File class
  if reflection.options[:block].class == Proc
    reflection.options[:block].call(record)
  end
  if record.new_record? && reflection.options[:autocreate]
    record.datastream_will_change!
  end
end

def find_target
  reflection.build_association(target_uri) do |record|
    configure_datastream(record) if reflection.options[:block]
  end
end

def new_record(method, attributes)
  record = super
  configure_datastream(record)
  record
end

def raise_on_type_mismatch(record)
  return if record.is_a? LoadableFromJson::SolrBackedMetadataFile
  super
end

def reader(force_reload = false)
  super || build
end

def replace(record)
  if record
    raise_on_type_mismatch(record)
    @updated = true
  end
  self.target = record
end

def target_uri
  "#{owner.uri}/#{reflection.name}"
end