30 lines
642 B
Ruby
30 lines
642 B
Ruby
class API::V2::Context < GraphQL::Query::Context
|
|
def has_fragment?(name)
|
|
if self["has_fragment_#{name}"]
|
|
true
|
|
else
|
|
visitor = HasFragment.new(self.query.selected_operation, name)
|
|
visitor.visit
|
|
self["has_fragment_#{name}"] = visitor.found
|
|
self["has_fragment_#{name}"]
|
|
end
|
|
end
|
|
|
|
class HasFragment < GraphQL::Language::Visitor
|
|
def initialize(document, name)
|
|
super(document)
|
|
@name = name.to_s
|
|
@found = false
|
|
end
|
|
|
|
attr_reader :found
|
|
|
|
def on_inline_fragment(node, parent)
|
|
if node.type.name == @name
|
|
@found = true
|
|
end
|
|
|
|
super
|
|
end
|
|
end
|
|
end
|