Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\node\Plugin\migrate;
Chris@0: 
Chris@0: use Drupal\Component\Plugin\Derivative\DeriverBase;
Chris@0: use Drupal\Core\Database\DatabaseExceptionWrapper;
Chris@0: use Drupal\Core\Plugin\Discovery\ContainerDeriverInterface;
Chris@0: use Drupal\migrate\Exception\RequirementsException;
Chris@0: use Drupal\migrate\Plugin\MigrationDeriverTrait;
Chris@18: use Drupal\migrate_drupal\FieldDiscoveryInterface;
Chris@0: use Symfony\Component\DependencyInjection\ContainerInterface;
Chris@0: 
Chris@0: /**
Chris@0:  * Deriver for Drupal 6 node and node revision migrations based on node types.
Chris@0:  */
Chris@0: class D6NodeDeriver extends DeriverBase implements ContainerDeriverInterface {
Chris@0:   use MigrationDeriverTrait;
Chris@0: 
Chris@0:   /**
Chris@0:    * The base plugin ID this derivative is for.
Chris@0:    *
Chris@0:    * @var string
Chris@0:    */
Chris@0:   protected $basePluginId;
Chris@0: 
Chris@0:   /**
Chris@0:    * Whether or not to include translations.
Chris@0:    *
Chris@0:    * @var bool
Chris@0:    */
Chris@0:   protected $includeTranslations;
Chris@0: 
Chris@0:   /**
Chris@18:    * The migration field discovery service.
Chris@18:    *
Chris@18:    * @var \Drupal\migrate_drupal\FieldDiscoveryInterface
Chris@18:    */
Chris@18:   protected $fieldDiscovery;
Chris@18: 
Chris@18:   /**
Chris@0:    * D6NodeDeriver constructor.
Chris@0:    *
Chris@0:    * @param string $base_plugin_id
Chris@0:    *   The base plugin ID for the plugin ID.
Chris@0:    * @param bool $translations
Chris@0:    *   Whether or not to include translations.
Chris@18:    * @param \Drupal\migrate_drupal\FieldDiscoveryInterface $field_discovery
Chris@18:    *   The migration field discovery service.
Chris@0:    */
Chris@18:   public function __construct($base_plugin_id, $translations, FieldDiscoveryInterface $field_discovery) {
Chris@0:     $this->basePluginId = $base_plugin_id;
Chris@0:     $this->includeTranslations = $translations;
Chris@18:     $this->fieldDiscovery = $field_discovery;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public static function create(ContainerInterface $container, $base_plugin_id) {
Chris@0:     // Translations don't make sense unless we have content_translation.
Chris@0:     return new static(
Chris@0:       $base_plugin_id,
Chris@18:       $container->get('module_handler')->moduleExists('content_translation'),
Chris@18:       $container->get('migrate_drupal.field_discovery')
Chris@0:     );
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@18:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function getDerivativeDefinitions($base_plugin_definition) {
Chris@0:     if ($base_plugin_definition['id'] == 'd6_node_translation' && !$this->includeTranslations) {
Chris@0:       // Refuse to generate anything.
Chris@0:       return $this->derivatives;
Chris@0:     }
Chris@0: 
Chris@0:     $node_types = static::getSourcePlugin('d6_node_type');
Chris@0:     try {
Chris@0:       $node_types->checkRequirements();
Chris@0:     }
Chris@0:     catch (RequirementsException $e) {
Chris@0:       // If the d6_node_type requirements failed, that means we do not have a
Chris@0:       // Drupal source database configured - there is nothing to generate.
Chris@0:       return $this->derivatives;
Chris@0:     }
Chris@0: 
Chris@0:     try {
Chris@0:       foreach ($node_types as $row) {
Chris@0:         $node_type = $row->getSourceProperty('type');
Chris@0:         $values = $base_plugin_definition;
Chris@0: 
Chris@0:         $values['label'] = t("@label (@type)", [
Chris@0:           '@label' => $values['label'],
Chris@0:           '@type' => $node_type,
Chris@0:         ]);
Chris@0:         $values['source']['node_type'] = $node_type;
Chris@0:         $values['destination']['default_bundle'] = $node_type;
Chris@0: 
Chris@0:         // If this migration is based on the d6_node_revision migration or
Chris@0:         // is for translations of nodes, it should explicitly depend on the
Chris@0:         // corresponding d6_node variant.
Chris@0:         if (in_array($base_plugin_definition['id'], ['d6_node_revision', 'd6_node_translation'])) {
Chris@0:           $values['migration_dependencies']['required'][] = 'd6_node:' . $node_type;
Chris@0:         }
Chris@0: 
Chris@0:         /** @var \Drupal\migrate\Plugin\Migration $migration */
Chris@0:         $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($values);
Chris@18:         $this->fieldDiscovery->addBundleFieldProcesses($migration, 'node', $node_type);
Chris@0:         $this->derivatives[$node_type] = $migration->getPluginDefinition();
Chris@0:       }
Chris@0:     }
Chris@0:     catch (DatabaseExceptionWrapper $e) {
Chris@0:       // Once we begin iterating the source plugin it is possible that the
Chris@0:       // source tables will not exist. This can happen when the
Chris@0:       // MigrationPluginManager gathers up the migration definitions but we do
Chris@0:       // not actually have a Drupal 6 source database.
Chris@0:     }
Chris@0: 
Chris@0:     return $this->derivatives;
Chris@0:   }
Chris@0: 
Chris@0: }