aboutsummaryrefslogtreecommitdiffstats
path: root/lib/active_storage/attached/macros.rb
blob: 11f88f16e5e1aaba9f408fc21c9c632a088b4c37 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# Provides the class-level DSL for declaring that an Active Record model has attached blobs.
module ActiveStorage::Attached::Macros
  # Specifies the relation between a single attachment and the model.
  #
  #   class User < ActiveRecord::Base
  #     has_one_attached :avatar
  #   end
  #
  # There is no column defined on the model side, Active Storage takes
  # care of the mapping between your records and the attachment.
  #
  # If the +:dependent+ option isn't set, the attachment will be purged
  # (i.e. destroyed) whenever the record is destroyed.
  def has_one_attached(name, dependent: :purge_later)
    define_method(name) do
      instance_variable_get("@active_storage_attached_#{name}") ||
        instance_variable_set("@active_storage_attached_#{name}", ActiveStorage::Attached::One.new(name, self))
    end

    has_one :"#{name}_attachment", -> { where(name: name) }, class_name: "ActiveStorage::Attachment", as: :record
    has_one :"#{name}_blob", through: :"#{name}_attachment", class_name: "ActiveStorage::Blob", source: :blob

    if dependent == :purge_later
      before_destroy { public_send(name).purge_later }
    end
  end

  # Specifies the relation between multiple attachments and the model.
  #
  #   class Gallery < ActiveRecord::Base
  #     has_many_attached :photos
  #   end
  #
  # There are no columns defined on the model side, Active Storage takes
  # care of the mapping between your records and the attachments.
  #
  # To avoid N+1 queries, you can include the attached blobs in your query like so:
  #
  #   Gallery.where(user: Current.user).with_attached_photos
  #
  # If the +:dependent+ option isn't set, all the attachments will be purged
  # (i.e. destroyed) whenever the record is destroyed.
  def has_many_attached(name, dependent: :purge_later)
    define_method(name) do
      instance_variable_get("@active_storage_attached_#{name}") ||
        instance_variable_set("@active_storage_attached_#{name}", ActiveStorage::Attached::Many.new(name, self))
    end

    has_many :"#{name}_attachments", -> { where(name: name) }, as: :record, class_name: "ActiveStorage::Attachment"
    has_many :"#{name}_blobs", through: :"#{name}_attachments", class_name: "ActiveStorage::Blob", source: :blob

    scope :"with_attached_#{name}", -> { includes("#{name}_attachments": :blob) }

    if dependent == :purge_later
      before_destroy { public_send(name).purge_later }
    end
  end
end