View source
<?php
namespace Drupal\metatag\Plugin\migrate\source\d7;
use Drupal\Core\Entity\EntityTypeBundleInfoInterface;
use Drupal\migrate\Plugin\MigrationInterface;
use Drupal\migrate_drupal\Plugin\migrate\source\DrupalSqlBase;
use Symfony\Component\DependencyInjection\ContainerInterface;
class MetatagFieldInstance extends DrupalSqlBase {
protected $entityTypeBundleInfo;
public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration = NULL) {
$source = parent::create($container, $configuration, $plugin_id, $plugin_definition, $migration);
$source
->setEntityTypeBundleInfo($container
->get('entity_type.bundle.info'));
return $source;
}
public function setEntityTypeBundleInfo(EntityTypeBundleInfoInterface $entity_type_bundle_info) {
$this->entityTypeBundleInfo = $entity_type_bundle_info;
}
public function query() {
$base_query = $this
->select('metatag', 'm')
->fields('m', [
'entity_type',
])
->groupBy('entity_type');
if (isset($this->configuration['entity_type_id'])) {
$entity_type_id = $this->configuration['entity_type_id'];
$base_query
->condition('m.entity_type', $entity_type_id);
if (isset($this->configuration['bundle'])) {
$bundle = $this->configuration['bundle'];
switch ($entity_type_id) {
case 'node':
$base_query
->join('node', 'n', "n.nid = m.entity_id");
$base_query
->condition('n.type', $bundle);
$base_query
->addField('n', 'type', 'bundle');
$base_query
->groupBy('bundle');
break;
case 'taxonomy_term':
$base_query
->join('taxonomy_term_data', 'ttd', "ttd.tid = m.entity_id");
$base_query
->fields('ttd', [
'vid',
]);
$base_query
->join('taxonomy_vocabulary', 'tv', 'ttd.vid = tv.vid');
$base_query
->condition('tv.machine_name', $bundle);
$base_query
->addField('tv', 'machine_name', 'bundle');
$base_query
->groupBy('ttd.vid');
$base_query
->groupBy('bundle');
break;
}
}
}
return $base_query;
}
public function fields() {
return [
'entity_type' => $this
->t('Entity type'),
'bundle' => $this
->t('Bundle'),
];
}
public function initializeIterator() {
$bundles = [];
foreach (parent::initializeIterator() as $instance) {
if (isset($instance['bundle'])) {
$bundles[] = $instance;
continue;
}
$bundle_info = $this->entityTypeBundleInfo
->getBundleInfo($instance['entity_type']);
foreach (array_keys($bundle_info) as $bundle) {
$bundles[] = [
'entity_type' => $instance['entity_type'],
'bundle' => $bundle,
];
}
}
return new \ArrayIterator($bundles);
}
public function getIds() {
$ids['entity_type']['type'] = 'string';
$ids['bundle']['type'] = 'string';
return $ids;
}
public function count($refresh = FALSE) {
return $this
->initializeIterator()
->count();
}
}