module ActiveStorage::Reflection::ReflectionExtension

def reflection_class_for(macro)
  case macro
  when :has_one_attached
    HasOneAttachedReflection
  when :has_many_attached
    HasManyAttachedReflection
  else
    super
  end
end