From e845aa76aa2ffb6bd4bc1bbd91776c72c0a271d6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?benjamin=20melan=C3=A7on?= <ben@agaric.coop>
Date: Thu, 16 Nov 2023 10:04:03 -0500
Subject: [PATCH] Make entity query bundle optional and entity type default to
 node

---
 migration_helpers.module | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/migration_helpers.module b/migration_helpers.module
index 9a6f597..c99ed2b 100644
--- a/migration_helpers.module
+++ b/migration_helpers.module
@@ -3,18 +3,25 @@
 /**
  * Entity query helper.
  */
-function migration_helpers_entity_query($entity_type, $bundle, $field = '') {
+function migration_helpers_entity_query(
+  $entity_type = 'node',
+  $bundle = NULL,
+  $field = NULL,
+) {
 
   /** @var \Drupal\Core\Entity\EntityTypeManagerInterface $entity_type_manager */
   $entity_type_manager = \Drupal::service('entity_type.manager');
 
   $entity_storage = $entity_type_manager->getStorage($entity_type);
-  $bundle_key = $entity_storage->getEntityType()->getKey('bundle');
 
   $entity_query = $entity_storage->getQuery();
-  $entity_query->condition($bundle_key, $bundle);
 
-  if (!empty($field)) {
+  if ($bundle) {
+    $bundle_key = $entity_storage->getEntityType()->getKey('bundle');
+    $entity_query->condition($bundle_key, $bundle);
+  }
+
+  if ($field) {
     // Check that field BOTH is present on entity bundle AND has a value saved.
     $entity_query->exists($field);
   }