consul/nairobi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Customize resolvers to only permit allowed records

Browse Source
This commit is contained in:
Alberto Miedes Garcés
2016-12-29 14:17:44 +01:00
parent 90130f20e7
commit 0aed5338f6
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/graph_ql/type_creator.rb
View File

@@ -28,10 +28,22 @@ module GraphQL
field(field_name.to_s, TYPES_CONVERSION[model.columns_hash[field_name.to_s].type])
else
association = type_creator.class.association?(model, field_name)
target_model = association.klass
public_elements = target_model.respond_to?(:public_for_api) ? target_model.public_for_api : target_model.all
if type_creator.class.needs_pagination?(association)
connection association.name, -> { type_creator.created_types[association.klass].connection_type }
connection(association.name, -> { type_creator.created_types[target_model].connection_type }) do
resolve -> (object, arguments, context) do
object.send(association.name).all & public_elements.all
end
end
else
field association.name, -> { type_creator.created_types[association.klass] }
field(association.name, -> { type_creator.created_types[target_model] }) do
resolve -> (object, arguments, context) do
linked_element = object.send(field_name)
public_elements.include?(linked_element) ? linked_element : nil
end
end
end
end
end