Fix issue with adding pantheon document reference field

Author: rkunjappan

src/Plugin/EntityReferenceSelection/PantheonDocumentSelection.php

@@ -0,0 +1,60 @@
+<?php
+
+namespace Drupal\pantheon_content_publisher\Plugin\EntityReferenceSelection;
+
+use Drupal\Core\Entity\Plugin\EntityReferenceSelection\DefaultSelection;
+
+/**
+ * Provides a selection plugin for Pantheon documents.
+ *
+ * @EntityReferenceSelection(
+ *   id = "pantheon_document_selection",
+ *   label = @Translation("Pantheon Document Selection"),
+ *   group = "default",
+ *   weight = 0
+ * )
+ */
+class PantheonDocumentSelection extends DefaultSelection {
+
+  /**
+   * {@inheritdoc}
+   */
+   protected function buildEntityQuery($match = NULL, $match_operator = 'CONTAINS'): QueryInterface {
+    $configuration = $this->getConfiguration();
+    $target_type = 'pantheon_document';
+    $entity_type = $this->entityTypeManager->getDefinition($target_type);
+
+    /** @var \Drupal\Core\Entity\Query\QueryInterface $query */
+    $query = $this->entityTypeManager->getStorage($target_type)->getQuery();
+    $query->accessCheck(TRUE);
+
+    if (isset($match) && $label_key = $entity_type->getKey('label')) {
+      $query->condition($label_key, $match, $match_operator);
+    }
+
+    $collection_ids = $configuration['target_bundles'] ?? [];
+    if (!empty($collection_ids)) {
+      $query->condition('collection', $collection_ids, 'IN');
+    }
+
+    // Add entity-access tag.
+    $query->addTag($target_type . '_access');
+    $query->addTag('entity_reference');
+    $query->addMetaData('entity_reference_selection_handler', $this);
+
+    // Add the sort option.
+    if (!empty($configuration['sort']) && $configuration['sort']['field'] !== '_none') {
+      $query->sort($configuration['sort']['field'], $configuration['sort']['direction']);
+    }
+
+    return $query;
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public function validateReferenceableEntities(array $ids): array {
+    return parent::validateReferenceableEntities($ids);
+  }
+
+}

src/Query/Query.php

@@ -23,23 +23,33 @@ public function __construct(EntityTypeInterface $entity_type, $conjunction, arra
   public function execute() {
     $conditions = &$this->condition->conditions();
     $collections = NULL;
+    $target_ids = NULL;
     foreach ($conditions as $key => $condition) {
-      if ($condition['field'] === 'collection' && in_array(($condition['operator'] ?? '='), ['=', 'IN'])) {
+      if ($condition['field'] === 'collection' && in_array($condition['operator'] ?? '=', ['=', 'IN'])) {
         $collections = (array) $condition['value'];
         unset($conditions[$key]);
-        break;
+      }
+
+      if ($condition['field'] === 'id' && in_array($condition['operator'] ?? '=', ['=', 'IN'])) {
+        $target_ids = (array) $condition['value'];
+        unset($conditions[$key]);
       }
     }
+
     $collections = PantheonDocumentCollection::loadMultiple($collections);
     $records = [];
     foreach ($collections as $collection) {
       foreach ($collection->getGraphQL()->getArticles() as $pantheon_record) {
         $key = PantheonDocumentStorage::getEntityId($collection, $pantheon_record['id']);
+
+        if ($target_ids !== NULL && !in_array($key, $target_ids)) {
+            continue;
+        }
+
         $records[$key] = $this->converter->convert($pantheon_record, $collection->id());
       }
     }
 
-    // Copy-paste from
     // Drupal\Core\Entity\KeyValueStore\Query\Query::execute().
     $result = $this->condition->compile($records);
 
@@ -68,3 +78,4 @@ public function execute() {
   }
 
 }
+