75 lines
2.0 KiB
Ruby
75 lines
2.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ActiveStorage
|
|
module Reflection
|
|
class HasAttachedReflection < ActiveRecord::Reflection::MacroReflection # :nodoc:
|
|
def variant(name, transformations)
|
|
variants[name] = transformations
|
|
end
|
|
|
|
def variants
|
|
@variants ||= {}
|
|
end
|
|
end
|
|
|
|
# Holds all the metadata about a has_one_attached attachment as it was
|
|
# specified in the Active Record class.
|
|
class HasOneAttachedReflection < HasAttachedReflection # :nodoc:
|
|
def macro
|
|
:has_one_attached
|
|
end
|
|
end
|
|
|
|
# Holds all the metadata about a has_many_attached attachment as it was
|
|
# specified in the Active Record class.
|
|
class HasManyAttachedReflection < HasAttachedReflection # :nodoc:
|
|
def macro
|
|
:has_many_attached
|
|
end
|
|
end
|
|
|
|
module ReflectionExtension # :nodoc:
|
|
def add_attachment_reflection(model, name, reflection)
|
|
model.attachment_reflections = model.attachment_reflections.merge(name.to_s => reflection)
|
|
end
|
|
|
|
private
|
|
def reflection_class_for(macro)
|
|
case macro
|
|
when :has_one_attached
|
|
HasOneAttachedReflection
|
|
when :has_many_attached
|
|
HasManyAttachedReflection
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|
|
|
|
module ActiveRecordExtensions
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
class_attribute :attachment_reflections, instance_writer: false, default: {}
|
|
end
|
|
|
|
module ClassMethods
|
|
# Returns an array of reflection objects for all the attachments in the
|
|
# class.
|
|
def reflect_on_all_attachments
|
|
attachment_reflections.values
|
|
end
|
|
|
|
# Returns the reflection object for the named +attachment+.
|
|
#
|
|
# User.reflect_on_attachment(:avatar)
|
|
# # => the avatar reflection
|
|
#
|
|
def reflect_on_attachment(attachment)
|
|
attachment_reflections[attachment.to_s]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|