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

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

namespace Drupal\migrate\Plugin\migrate\process;

10
use Drupal\Core\Entity\EntityStorageInterface;
11
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
12
use Drupal\migrate\MigrateSkipProcessException;
13
14
15
use Drupal\migrate\Plugin\MigratePluginManager;
use Drupal\migrate\ProcessPluginBase;
use Drupal\migrate\Entity\MigrationInterface;
16
use Drupal\migrate\MigrateExecutableInterface;
17
18
19
20
21
22
23
24
25
26
27
28
29
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 {

  /**
30
31
   * The process plugin manager.
   *
32
33
34
35
36
   * @var \Drupal\migrate\Plugin\MigratePluginManager
   */
  protected $processPluginManager;

  /**
37
38
   * The entity storage manager.
   *
39
   * @var \Drupal\Core\Entity\EntityStorageInterface
40
   */
41
  protected $migrationStorage;
42
43
44
45

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

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

  /**
   * {@inheritdoc}
   */
70
  public function transform($value, MigrateExecutableInterface $migrate_executable, Row $row, $destination_property) {
71
72
73
74
75
76
77
78
79
    $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);
    }
80
    $this->skipOnEmpty($value);
81
82
    $self = FALSE;
    /** @var \Drupal\migrate\Entity\MigrationInterface[] $migrations */
83
    $migrations = $this->migrationStorage->loadMultiple($migration_ids);
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
    $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;
      }
    }
104

105
    if (!$destination_ids && ($self || isset($this->configuration['stub_id']) || count($migrations) == 1)) {
106
107
108
109
110
111
112
113
114
115
116
      // 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);
      }
117
      $destination_plugin = $migration->getDestinationPlugin(TRUE);
118
      // Only keep the process necessary to produce the destination ID.
119
      $process = $migration->get('process');
120

121
      // We already have the source ID values but need to key them for the Row
122
123
124
125
126
127
      // 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];
      }
128

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

131
132
133
134
135
136
      // 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);
      }
137
138
      catch (\Exception $e) {
        $migrate_executable->saveMessage($e->getMessage());
139
140
141
142
143
144
145
146
147
148
149
150
151
152
      }
    }
    if ($destination_ids) {
      if ($scalar) {
        if (count($destination_ids) == 1) {
          return reset($destination_ids);
        }
      }
      else {
        return $destination_ids;
      }
    }
  }

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

167
}