fix(EntityDefinition): Fix incorrect context type for "field_types" (drupal-graphql#991)

Author: rthideaway

src/Plugin/GraphQL/DataProducer/EntityDefinition/EntityDefinition.php

@@ -24,8 +24,8 @@
  *       label = @Translation("Bundle"),
  *       required = FALSE
  *     ),
- *     "field_types" = @ContextDefinition("FieldTypes",
- *       label = @Translation("Field types"),
+ *     "field_types" = @ContextDefinition("string",
+ *       label = @Translation("Field types (ALL, BASE_FIELDS, FIELD_CONFIG)"),
  *       default = "ALL",
  *       required = FALSE
  *     )

src/Plugin/GraphQL/DataProducer/EntityDefinition/Fields.php

@@ -29,7 +29,7 @@
  *       label = @Translation("Bundle context"),
  *       required = FALSE,
  *     ),
- *     "field_types_context" = @ContextDefinition("any",
+ *     "field_types_context" = @ContextDefinition("string",
  *       label = @Translation("Field types context"),
  *       required = FALSE,
  *     )
@@ -110,15 +110,15 @@ public function __construct(
    *   The entity type definition.
    * @param array|null $bundle_context
    *   Bundle context.
-   * @param array|null $field_types_context
+   * @param string|null $field_types_context
    *   Field types context.
    * @param \Drupal\graphql\GraphQL\Execution\FieldContext $field_context
    *   Field context.
    */
   public function resolve(
     EntityTypeInterface $entity_definition,
     ?array $bundle_context = NULL,
-    ?array $field_types_context = NULL,
+    ?string $field_types_context = NULL,
     FieldContext $field_context
   ): \Iterator {
 
@@ -140,14 +140,13 @@ public function resolve(
       }
 
       if ($field_types_context) {
-        $field_types = $field_types_context['key'];
         foreach ($fields as $field) {
-          if ($field_types === 'BASE_FIELDS') {
+          if ($field_types_context === 'BASE_FIELDS') {
             if ($field instanceof BaseFieldDefinition) {
               yield $field;
             }
           }
-          elseif ($field_types === 'FIELD_CONFIG') {
+          elseif ($field_types_context === 'FIELD_CONFIG') {
             if ($field instanceof FieldConfig || $field instanceof BaseFieldOverride) {
               yield $field;
             }