demarches-normaliennes/app/graphql/extensions/attachment.rb

37 lines
1.2 KiB
Ruby
Raw Normal View History

2018-11-20 22:27:40 +01:00
# references:
# https://evilmartians.com/chronicles/active-storage-meets-graphql-pt-2-exposing-attachment-urls
module Extensions
class Attachment < GraphQL::Schema::FieldExtension
attr_reader :attachment_assoc
def apply
# Here we try to define the attachment name:
# - it could be set explicitly via extension options
# - or we imply that is the same as the field name w/o "_url"
# suffix (e.g., "avatar_url" => "avatar")
attachment = options&.[](:attachment) || field.original_name.to_s.sub(/_url$/, "")
# that's the name of the Active Record association
@attachment_assoc = "#{attachment}_attachment"
end
# This method resolves (as it states) the field itself
# (it's the same as defining a method within a type)
def resolve(object:, **_rest)
Loaders::Association.for(
object.object.class,
attachment_assoc => :blob
).load(object.object)
end
# This method is called if the result of the `resolve`
# is a lazy value (e.g., a Promise like in our case)
def after_resolve(value:, **_rest)
2019-12-11 12:24:50 +01:00
if value&.virus_scanner&.safe? || value&.virus_scanner&.pending?
value
end
2018-11-20 22:27:40 +01:00
end
end
end