View source
<?php
namespace Drupal\migrate_plus\Plugin\migrate\process;
use Drupal\Core\Entity\EntityManagerInterface;
use Drupal\Core\Entity\EntityReferenceSelection\SelectionPluginManagerInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\migrate\Plugin\MigrationInterface;
use Drupal\migrate\MigrateException;
use Drupal\migrate\MigrateExecutableInterface;
use Drupal\migrate\ProcessPluginBase;
use Drupal\migrate\Row;
use Symfony\Component\DependencyInjection\ContainerInterface;
class EntityLookup extends ProcessPluginBase implements ContainerFactoryPluginInterface {
protected $entityManager;
protected $migration;
protected $selectionPluginManager;
protected $destinationEntityType;
protected $destinationBundleKey;
protected $lookupValueKey;
protected $lookupBundleKey;
protected $lookupBundle;
protected $lookupEntityType;
protected $destinationProperty;
public function __construct(array $configuration, $pluginId, $pluginDefinition, MigrationInterface $migration, EntityManagerInterface $entityManager, SelectionPluginManagerInterface $selectionPluginManager) {
parent::__construct($configuration, $pluginId, $pluginDefinition);
$this->migration = $migration;
$this->entityManager = $entityManager;
$this->selectionPluginManager = $selectionPluginManager;
$pluginIdParts = explode(':', $this->migration
->getDestinationPlugin()
->getPluginId());
$this->destinationEntityType = empty($pluginIdParts[1]) ?: $pluginIdParts[1];
$this->destinationBundleKey = !$this->destinationEntityType ?: $this->entityManager
->getDefinition($this->destinationEntityType)
->getKey('bundle');
}
public static function create(ContainerInterface $container, array $configuration, $pluginId, $pluginDefinition, MigrationInterface $migration = NULL) {
return new static($configuration, $pluginId, $pluginDefinition, $migration, $container
->get('entity.manager'), $container
->get('plugin.manager.entity_reference_selection'));
}
public function transform($value, MigrateExecutableInterface $migrateExecutable, Row $row, $destinationProperty) {
$parts = explode('/', $destinationProperty);
$destinationProperty = reset($parts);
$this
->determineLookupProperties($destinationProperty);
$this->destinationProperty = isset($this->configuration['destination_field']) ? $this->configuration['destination_field'] : NULL;
return $this
->query($value);
}
protected function determineLookupProperties($destinationProperty) {
if (!empty($this->configuration['value_key'])) {
$this->lookupValueKey = $this->configuration['value_key'];
}
if (!empty($this->configuration['bundle_key'])) {
$this->lookupBundleKey = $this->configuration['bundle_key'];
}
if (!empty($this->configuration['bundle'])) {
$this->lookupBundle = $this->configuration['bundle'];
}
if (!empty($this->configuration['entity_type'])) {
$this->lookupEntityType = $this->configuration['entity_type'];
}
if (empty($this->lookupValueKey) || empty($this->lookupBundleKey) || empty($this->lookupBundle) || empty($this->lookupEntityType)) {
if (!empty($this->migration
->getProcess()[$this->destinationBundleKey][0]['default_value'])) {
$destinationEntityBundle = $this->migration
->getProcess()[$this->destinationBundleKey][0]['default_value'];
$fieldConfig = $this->entityManager
->getFieldDefinitions($this->destinationEntityType, $destinationEntityBundle)[$destinationProperty]
->getConfig($destinationEntityBundle);
switch ($fieldConfig
->getType()) {
case 'entity_reference':
if (empty($this->lookupBundle)) {
$handlerSettings = $fieldConfig
->getSetting('handler_settings');
$bundles = array_filter((array) $handlerSettings['target_bundles']);
if (count($bundles) == 1) {
$this->lookupBundle = reset($bundles);
}
elseif (!empty($handlerSettings['auto_create']) && !empty($handlerSettings['auto_create_bundle'])) {
$this->lookupBundle = reset($handlerSettings['auto_create_bundle']);
}
}
$this->lookupEntityType = $this->lookupEntityType ?: reset($this->selectionPluginManager
->createInstance($fieldConfig
->getSetting('handler'))
->getPluginDefinition()['entity_types']);
$this->lookupValueKey = $this->lookupValueKey ?: $this->entityManager
->getDefinition($this->lookupEntityType)
->getKey('label');
$this->lookupBundleKey = $this->lookupBundleKey ?: $this->entityManager
->getDefinition($this->lookupEntityType)
->getKey('bundle');
break;
case 'file':
case 'image':
$this->lookupEntityType = 'file';
$this->lookupValueKey = $this->lookupValueKey ?: 'uri';
break;
default:
throw new MigrateException('Destination field type ' . $fieldConfig
->getType() . 'is not a recognized reference type.');
}
}
}
if (empty($this->lookupValueKey)) {
throw new MigrateException('The entity_lookup plugin requires a value_key, none located.');
}
if (!empty($this->lookupBundleKey) && empty($this->lookupBundle)) {
throw new MigrateException('The entity_lookup plugin found no bundle but destination entity requires one.');
}
if (empty($this->lookupEntityType)) {
throw new MigrateException('The entity_lookup plugin requires a entity_type, none located.');
}
}
protected function query($value) {
$ignoreCase = !empty($this->configuration['ignore_case']) ?: FALSE;
$multiple = is_array($value);
$query = $this->entityManager
->getStorage($this->lookupEntityType)
->getQuery()
->condition($this->lookupValueKey, $value, $multiple ? 'IN' : NULL);
if ($this->lookupBundleKey) {
$query
->condition($this->lookupBundleKey, $this->lookupBundle);
}
$results = $query
->execute();
if (empty($results)) {
return NULL;
}
if (!$ignoreCase) {
foreach ($results as $k => $identifier) {
$result_value = $this->entityManager
->getStorage($this->lookupEntityType)
->load($identifier)->{$this->lookupValueKey}->value;
if ($multiple && !in_array($result_value, $value, TRUE) || !$multiple && $result_value !== $value) {
unset($results[$k]);
}
}
}
if ($multiple && !empty($this->destinationProperty)) {
array_walk($results, function (&$value) {
$value = [
$this->destinationProperty => $value,
];
});
}
return $multiple ? array_values($results) : reset($results);
}
}