Migration.php 5.16 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
<?php

/**
 * @file
 * Contains \Drupal\migrate\Plugin\migrate\process\Migration.
 */


namespace Drupal\migrate\Plugin\migrate\process;

11
use Drupal\Core\Entity\EntityStorageInterface;
12
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
13
use Drupal\migrate\MigrateSkipProcessException;
14
15
16
use Drupal\migrate\Plugin\MigratePluginManager;
use Drupal\migrate\ProcessPluginBase;
use Drupal\migrate\Entity\MigrationInterface;
17
use Drupal\migrate\MigrateExecutableInterface;
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
use Drupal\migrate\Row;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Calculates the value of a property based on a previous migration.
 *
 * @MigrateProcessPlugin(
 *   id = "migration"
 * )
 */
class Migration extends ProcessPluginBase implements ContainerFactoryPluginInterface {

  /**
   * @var \Drupal\migrate\Plugin\MigratePluginManager
   */
  protected $processPluginManager;

  /**
36
   * @var \Drupal\Core\Entity\EntityStorageInterface
37
   */
38
  protected $migrationStorage;
39
40
41
42

  /**
   * {@inheritdoc}
   */
43
  public function __construct(array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration, EntityStorageInterface $storage, MigratePluginManager $process_plugin_manager) {
44
    parent::__construct($configuration, $plugin_id, $plugin_definition);
45
    $this->migrationStorage = $storage;
46
47
48
49
50
51
52
    $this->migration = $migration;
    $this->processPluginManager = $process_plugin_manager;
  }

  /**
   * {@inheritdoc}
   */
53
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration = NULL) {
54
55
56
57
58
    return new static(
      $configuration,
      $plugin_id,
      $plugin_definition,
      $migration,
59
      $container->get('entity.manager')->getStorage('migration'),
60
61
62
63
64
65
66
      $container->get('plugin.manager.migrate.process')
    );
  }

  /**
   * {@inheritdoc}
   */
67
  public function transform($value, MigrateExecutableInterface $migrate_executable, Row $row, $destination_property) {
68
69
70
71
72
73
74
75
76
    $migration_ids = $this->configuration['migration'];
    if (!is_array($migration_ids)) {
      $migration_ids = array($migration_ids);
    }
    $scalar = FALSE;
    if (!is_array($value)) {
      $scalar = TRUE;
      $value = array($value);
    }
77
    $this->skipOnEmpty($value);
78
79
    $self = FALSE;
    /** @var \Drupal\migrate\Entity\MigrationInterface[] $migrations */
80
    $migrations = $this->migrationStorage->loadMultiple($migration_ids);
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
    $destination_ids = NULL;
    $source_id_values = array();
    foreach ($migrations as $migration_id => $migration) {
      if ($migration_id == $this->migration->id()) {
        $self = TRUE;
      }
      if (isset($this->configuration['source_ids'][$migration_id])) {
        $configuration = array('source' => $this->configuration['source_ids'][$migration_id]);
        $source_id_values[$migration_id] = $this->processPluginManager
          ->createInstance('get', $configuration, $this->migration)
          ->transform(NULL, $migrate_executable, $row, $destination_property);
      }
      else {
        $source_id_values[$migration_id] = $value;
      }
      // Break out of the loop as soon as a destination ID is found.
      if ($destination_ids = $migration->getIdMap()->lookupDestinationID($source_id_values[$migration_id])) {
        break;
      }
    }
101

102
    if (!$destination_ids && ($self || isset($this->configuration['stub_id']) || count($migrations) == 1)) {
103
104
105
106
107
108
109
110
111
112
113
      // If the lookup didn't succeed, figure out which migration will do the
      // stubbing.
      if ($self) {
        $migration = $this->migration;
      }
      elseif (isset($this->configuration['stub_id'])) {
        $migration = $migrations[$this->configuration['stub_id']];
      }
      else {
        $migration = reset($migrations);
      }
114
      $destination_plugin = $migration->getDestinationPlugin(TRUE);
115
      // Only keep the process necessary to produce the destination ID.
116
      $process = $migration->get('process');
117

118
      // We already have the source ID values but need to key them for the Row
119
120
121
122
123
124
      // constructor.
      $source_ids = $migration->getSourcePlugin()->getIds();
      $values = array();
      foreach (array_keys($source_ids) as $index => $source_id) {
        $values[$source_id] = $source_id_values[$migration->id()][$index];
      }
125

126
127
      $stub_row = new Row($values + $migration->get('source'), $source_ids, TRUE);

128
129
130
131
132
133
      // Do a normal migration with the stub row.
      $migrate_executable->processRow($stub_row, $process);
      $destination_ids = array();
      try {
        $destination_ids = $destination_plugin->import($stub_row);
      }
134
135
      catch (\Exception $e) {
        $migrate_executable->saveMessage($e->getMessage());
136
137
138
139
140
141
142
143
144
145
146
147
148
149
      }
    }
    if ($destination_ids) {
      if ($scalar) {
        if (count($destination_ids) == 1) {
          return reset($destination_ids);
        }
      }
      else {
        return $destination_ids;
      }
    }
  }

150
151
152
153
154
155
156
157
158
159
160
161
162
163
  /**
   * Skip the migration process entirely if the value is FALSE.
   *
   * @param mixed $value
   *   The incoming value to transform.
   *
   * @throws \Drupal\migrate\MigrateSkipProcessException
   */
  protected function skipOnEmpty($value) {
    if (!array_filter($value)) {
      throw new MigrateSkipProcessException();
    }
  }

164
}