create_one_of_many.rb

# frozen_string_literal: true

module ActiveStorage
  class Attached::Changes::CreateOneOfMany < Attached::Changes::CreateOne # :nodoc:
    private
      def find_attachment
        if blob.persisted?
          record.public_send("#{name}_attachments").detect { |attachment| attachment.blob_id == blob.id }
        else
          blob.attachments.find { |attachment| attachment.record == record }
        end
      end
  end
end