GraphQL query for Search API

modules/thunder_gqls/graphql/thunder_menu.base.graphqls

@@ -1,7 +1,6 @@
 type Menu {
   id: String!
   name: String!
-
   items: [MenuItem]
 }
 

modules/thunder_gqls/graphql/thunder_search_api.base.graphqls

@@ -0,0 +1,4 @@
+type SearchApiResult {
+  items: [Page!]
+  total: Int!
+}

modules/thunder_gqls/src/GraphQL/Buffers/SearchApiResultBuffer.php

@@ -0,0 +1,101 @@
+<?php
+
+namespace Drupal\thunder_gqls\GraphQL\Buffers;
+
+use Drupal\Core\Entity\EntityTypeManagerInterface;
+use Drupal\Core\Entity\Plugin\DataType\EntityAdapter;
+use Drupal\graphql\GraphQL\Buffers\BufferBase;
+
+/**
+ * GraphQL Buffer for SearchApi Results.
+ */
+class SearchApiResultBuffer extends BufferBase {
+
+  /**
+   * The entity type manager service.
+   *
+   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
+   */
+  protected EntityTypeManagerInterface $entityTypeManager;
+
+  /**
+   * EntityBuffer constructor.
+   *
+   * @param \Drupal\Core\Entity\EntityTypeManagerInterface $entityTypeManager
+   *   The entity type manager service.
+   */
+  public function __construct(EntityTypeManagerInterface $entityTypeManager) {
+    $this->entityTypeManager = $entityTypeManager;
+  }
+
+  /**
+   * Add an item to the buffer.
+   *
+   * @param string|int|null $index
+   *   The entity type of the given entity ids.
+   * @param array|int $id
+   *   The entity id(s) to load.
+   *
+   * @return \Closure
+   *   The callback to invoke to load the result for this buffer item.
+   */
+  public function add($index, $id) {
+    $item = new \ArrayObject([
+      'index' => $index,
+      'id' => $id,
+    ]);
+
+    return $this->createBufferResolver($item);
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  protected function getBufferId($item) {
+    return $item['index'];
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public function resolveBufferArray(array $buffer) {
+    $index = reset($buffer)['index'];
+    $ids = array_map(function (\ArrayObject $item) {
+      return (array) $item['id'];
+    }, $buffer);
+
+    $ids = call_user_func_array('array_merge', $ids);
+    $ids = array_values(array_unique($ids));
+
+    // Load the buffered entities.
+    /** @var \Drupal\search_api\IndexInterface $index */
+    $index = $this->entityTypeManager
+      ->getStorage('search_api_index')
+      ->load($index);
+
+    $resultSet = $index->loadItemsMultiple($ids);
+    $entities = [];
+
+    foreach ($resultSet as $key => $resultItem) {
+      if ($resultItem instanceof EntityAdapter) {
+        $entities[$key] = $resultItem->getEntity();
+      }
+    }
+
+    return array_map(function ($item) use ($entities) {
+      if (is_array($item['id'])) {
+        return array_reduce($item['id'], static function ($carry, $current) use ($entities) {
+          if (!empty($entities[$current])) {
+            $carry[] = $entities[$current];
+            return $carry;
+          }
+
+          return $carry;
+        }, []);
+      }
+
+      return $entities[$item['id']] ?? NULL;
+    }, $buffer);
+  }
+
+}

modules/thunder_gqls/src/Plugin/GraphQL/DataProducer/EntitiesWithTerm.php

@@ -4,7 +4,6 @@
 
 use Drupal\graphql\GraphQL\Execution\FieldContext;
 use Drupal\taxonomy\TermInterface;
-use Drupal\thunder_gqls\Wrappers\EntityListResponse;
 use Drupal\thunder_gqls\Wrappers\EntityListResponseInterface;
 
 /**
@@ -119,7 +118,7 @@ public function resolve(TermInterface $term, string $type, array $bundles, strin
       $cacheContext
     );
 
-    return new EntityListResponse($query);
+    return $this->entityListResponse($query);
   }
 
   /**

modules/thunder_gqls/src/Plugin/GraphQL/DataProducer/ThunderEntityList.php

@@ -3,7 +3,6 @@
 namespace Drupal\thunder_gqls\Plugin\GraphQL\DataProducer;
 
 use Drupal\graphql\GraphQL\Execution\FieldContext;
-use Drupal\thunder_gqls\Wrappers\EntityListResponse;
 use Drupal\thunder_gqls\Wrappers\EntityListResponseInterface;
 
 /**
@@ -97,7 +96,7 @@ protected function resolve(string $type, array $bundles, int $offset, int $limit
       $cacheContext
     );
 
-    return new EntityListResponse($query);
+    return $this->entityListResponse($query);
   }
 
 }

modules/thunder_gqls/src/Plugin/GraphQL/DataProducer/ThunderEntityListProducerBase.php

@@ -2,12 +2,13 @@
 
 namespace Drupal\thunder_gqls\Plugin\GraphQL\DataProducer;
 
-use Drupal\Core\Entity\EntityTypeManager;
+use Drupal\Core\Entity\EntityTypeManagerInterface;
 use Drupal\Core\Entity\Query\QueryInterface;
 use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
 use Drupal\Core\Session\AccountInterface;
 use Drupal\graphql\GraphQL\Execution\FieldContext;
 use Drupal\graphql\Plugin\GraphQL\DataProducer\DataProducerPluginBase;
+use Drupal\thunder_gqls\Wrappers\EntityListResponse;
 use GraphQL\Error\UserError;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 
@@ -21,54 +22,95 @@ abstract class ThunderEntityListProducerBase extends DataProducerPluginBase impl
   /**
    * The entity type manager service.
    *
-   * @var \Drupal\Core\Entity\EntityTypeManager
+   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
    */
-  protected $entityTypeManager;
+  protected EntityTypeManagerInterface $entityTypeManager;
 
   /**
    * The current user.
    *
    * @var \Drupal\Core\Session\AccountInterface
    */
-  protected $currentUser;
+  protected AccountInterface $currentUser;
+
+  /**
+   * The response wrapper service.
+   *
+   * @var \Drupal\thunder_gqls\Wrappers\EntityListResponse
+   */
+  protected EntityListResponse $responseWrapper;
 
   /**
    * {@inheritdoc}
    */
   public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition): self {
-    return new static(
+    $instance = new static(
       $configuration,
       $plugin_id,
       $plugin_definition,
       $container->get('entity_type.manager'),
       $container->get('current_user')
     );
+
+    $instance->setResponseWrapper($container->get('thunder_gqls.entity_list_response_wrapper'));
+
+    return $instance;
   }
 
   /**
-   * EntityLoad constructor.
+   * ThunderEntityListProducerBase constructor.
    *
    * @param array $configuration
    *   The plugin configuration array.
    * @param string $pluginId
    *   The plugin id.
    * @param array $pluginDefinition
    *   The plugin definition array.
-   * @param \Drupal\Core\Entity\EntityTypeManager $entityTypeManager
+   * @param \Drupal\Core\Entity\EntityTypeManagerInterface $entityTypeManager
    *   The entity type manager service.
-   * @param \Drupal\Core\Session\AccountInterface $current_user
+   * @param \Drupal\Core\Session\AccountInterface $currentUser
    *   The current user.
    */
   public function __construct(
     array $configuration,
     string $pluginId,
     array $pluginDefinition,
-    EntityTypeManager $entityTypeManager,
-    AccountInterface $current_user,
+    EntityTypeManagerInterface $entityTypeManager,
+    AccountInterface $currentUser,
   ) {
     parent::__construct($configuration, $pluginId, $pluginDefinition);
+    $this->setEntityTypeManager($entityTypeManager);
+    $this->setCurrentUser($currentUser);
+  }
+
+  /**
+   * Set the entity type manager service.
+   *
+   * @param \Drupal\Core\Entity\EntityTypeManagerInterface $entityTypeManager
+   *   The entity type manager service.
+   */
+  public function setEntityTypeManager(EntityTypeManagerInterface $entityTypeManager): void {
     $this->entityTypeManager = $entityTypeManager;
-    $this->currentUser = $current_user;
+  }
+
+  /**
+   * Set the current user.
+   *
+   * @param \Drupal\Core\Session\AccountInterface $currentUser
+   *   The current user.
+   */
+  public function setCurrentUser(AccountInterface $currentUser): void {
+    $this->currentUser = $currentUser;
+  }
+
+  /**
+   * Set the response wrapper service.
+   *
+   * @param \Drupal\thunder_gqls\Wrappers\EntityListResponse $responseWrapper
+   *   The response wrapper service.
+   */
+  public function setResponseWrapper(EntityListResponse $responseWrapper): void {
+    $this->responseWrapper = $responseWrapper;
   }
 
   /**
@@ -163,10 +205,7 @@ protected function query(
     $query->range($offset, $limit);
 
     $storage = $this->entityTypeManager->getStorage($type);
-    $entityType = $storage->getEntityType();
-
-    $cacheContext->addCacheTags($entityType->getListCacheTags());
-    $cacheContext->addCacheContexts($entityType->getListCacheContexts());
+    $cacheContext->addCacheableDependency($storage->getEntityType());
     return $query;
   }
 
@@ -202,4 +241,17 @@ protected function createPublishedCondition(string $type, array $conditions) {
     ];
   }
 
+  /**
+   * The entity list response.
+   *
+   * @param \Drupal\Core\Entity\Query\QueryInterface $query
+   *   The entity query.
+   *
+   * @return \Drupal\thunder_gqls\Wrappers\EntityListResponse
+   *   The entity list response.
+   */
+  protected function entityListResponse(QueryInterface $query): EntityListResponse {
+    return $this->responseWrapper->setQuery($query);
+  }
+
 }