metrc-1.0.x-dev/modules/metrc_migrate/src/Plugin/migrate/process/SkipAcceptedPackage.php

<?php

namespace Drupal\metrc_migrate\Plugin\migrate\process;

use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\migrate\Plugin\MigrationInterface;
use Drupal\migrate\MigrateExecutableInterface;
use Drupal\migrate\ProcessPluginBase;
use Drupal\migrate\Row;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Drupal\migrate\MigrateSkipRowException;

/**
 * This plugin checks if a given package has alreeady been sold/transfered aka Accepted by a customer.
 *
 * Example usage with configuration:
 * @code
 *   field_tags:
 *     plugin: skip_accepted_package
 *     source: Id
 * @endcode
 *
 * @MigrateProcessPlugin(
 *  id = "skip_accepted_package"
 * )
 */
class SkipAcceptedPackage extends ProcessPluginBase implements ContainerFactoryPluginInterface {

  /**
   * The entity storage.
   *
   * @var \Drupal\Core\Entity\EntityStorageInterface
   */
  protected $storage;

  /**
   * EntityExists constructor.
   *
   * @param array $configuration
   *   A configuration array containing information about the plugin instance.
   * @param string $plugin_id
   *   The plugin ID.
   * @param mixed $plugin_definition
   *   The plugin implementation definition.
   * @param $storage
   *   The entity storage.
   */
  public function __construct(array $configuration, $plugin_id, $plugin_definition, EntityStorageInterface $storage) {
    parent::__construct($configuration, $plugin_id, $plugin_definition);
    $this->storage = $storage;
  }

  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration = NULL) {
    return new static(
      $configuration,
      $plugin_id,
      $plugin_definition,
      $container->get('entity_type.manager')->getStorage('node')
    );
  }

  /**
   * {@inheritdoc}
   */
  public function transform($value, MigrateExecutableInterface $migrate_executable, Row $row, $destination_property) {
    if (is_array($value)) {
      $value = reset($value);
    }
    $entity = $this->storage->load($value);
    if ($entity instanceof EntityInterface) {
      $pkg_state = $entity->field_package_state->value;
      if ($pkg_state == 'Accepted') {
        // set as unpublished if it's accepted and published still (status =1)
        if ($entity->status->value == '1') {
          // SHITS MESSED UP. CANT HAVE Accepted packages be active!
          $entity->status = FALSE;
          $entity->save();
        }
        // skip it!
        throw new MigrateSkipRowException(" Skipped Accepted (transfered) Package! ");
      }}
    return FALSE;
  }

}