|
2 | 2 |
|
3 | 3 | namespace Drupal\graphql\Plugin\GraphQL\DataProducer\EntityDefinition; |
4 | 4 |
|
5 | | -use Drupal\Core\Entity\ContentEntityType; |
| 5 | +use Drupal\Core\Entity\ContentEntityTypeInterface; |
6 | 6 | use Drupal\Core\Entity\EntityFieldManager; |
7 | 7 | use Drupal\Core\Entity\EntityTypeInterface; |
8 | 8 | use Drupal\Core\Entity\EntityTypeManager; |
@@ -121,24 +121,22 @@ public function resolve( |
121 | 121 | ?array $field_types_context = NULL, |
122 | 122 | FieldContext $field_context |
123 | 123 | ): \Iterator { |
124 | | - $entity_definition->getBundleEntityType(); |
125 | | - if ($entity_definition instanceof ContentEntityType) { |
| 124 | + |
| 125 | + if ($entity_definition instanceof ContentEntityTypeInterface) { |
| 126 | + $entity_type_id = $entity_definition->id(); |
126 | 127 | if ($bundle_context) { |
127 | 128 | $key = $bundle_context['key']; |
128 | | - $id = $entity_definition->id(); |
129 | | - $fields = $this->entityFieldManager->getFieldDefinitions($id, $key); |
| 129 | + $fields = $this->entityFieldManager->getFieldDefinitions($entity_type_id, $key); |
130 | 130 |
|
131 | 131 | // Set entity form default display as context. |
132 | | - $entity_id = $id . '.' . $key . '.default'; |
133 | | - /** @var \Drupal\Core\Config\Entity\ConfigEntityStorage $form_display_context */ |
| 132 | + $form_display_id = $entity_type_id . '.' . $key . '.default'; |
134 | 133 | $form_display_context = $this->entityTypeManager |
135 | 134 | ->getStorage('entity_form_display') |
136 | | - ->load($entity_id); |
| 135 | + ->load($form_display_id); |
137 | 136 | $field_context->setContextValue('entity_form_display', $form_display_context); |
138 | 137 | } |
139 | 138 | else { |
140 | | - $id = $entity_definition->id(); |
141 | | - $fields = $this->entityFieldManager->getFieldDefinitions($id, $id); |
| 139 | + $fields = $this->entityFieldManager->getFieldDefinitions($entity_type_id, $entity_type_id); |
142 | 140 | } |
143 | 141 |
|
144 | 142 | if ($field_types_context) { |
|
0 commit comments