Make entity query bundle optional and entity type default to node

This commit is contained in:
benjamin melançon 2023-11-16 10:04:03 -05:00
parent 097611f89a
commit e845aa76aa

View file

@ -3,18 +3,25 @@
/** /**
* Entity query helper. * 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 */ /** @var \Drupal\Core\Entity\EntityTypeManagerInterface $entity_type_manager */
$entity_type_manager = \Drupal::service('entity_type.manager'); $entity_type_manager = \Drupal::service('entity_type.manager');
$entity_storage = $entity_type_manager->getStorage($entity_type); $entity_storage = $entity_type_manager->getStorage($entity_type);
$bundle_key = $entity_storage->getEntityType()->getKey('bundle');
$entity_query = $entity_storage->getQuery(); $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. // Check that field BOTH is present on entity bundle AND has a value saved.
$entity_query->exists($field); $entity_query->exists($field);
} }