merci-8.x-2.x-dev/modules/merci_migration/src/Plugin/migrate/source/d7/MerciResource.php

<?php

namespace Drupal\merci_migration\Plugin\migrate\source\d7;

use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\migrate\Row;
use Drupal\migrate_drupal\Plugin\migrate\source\d7\FieldableEntity;
use Drupal\Core\Database\Query\SelectInterface;
use Drupal\Core\Entity\EntityManagerInterface;
use Drupal\Core\Extension\ModuleHandler;
use Drupal\Core\State\StateInterface;
use Drupal\migrate\Plugin\MigrationInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;


/**
 * Drupal 7 node source from database.
 *
 * @MigrateSource(
 *   id = "merci_resource",
 *   source_module = "node"
 * )
 */
class MerciResource extends FieldableEntity {
  /**
   * The module handler.
   *
   * @var \Drupal\Core\Extension\ModuleHandlerInterface
   */
  protected $moduleHandler;


/**
   * {@inheritdoc}
   */
  public function __construct(array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration, StateInterface $state, EntityManagerInterface $entity_manager, ModuleHandlerInterface $module_handler) {
    parent::__construct($configuration, $plugin_id, $plugin_definition, $migration, $state, $entity_manager);
    $this->moduleHandler = $module_handler;
  }
  
  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration = NULL) {
    return new static($configuration, $plugin_id, $plugin_definition, $migration, $container->get('state'), $container->get('entity.manager'), $container->get('module_handler'));
  }
  

  /**
   * The join options between the node and the node_revisions table.
   */
  const JOIN = 'n.vid = nr.vid';


  
  /**
   * {@inheritdoc}
   */
  public function query() {
    
    // Select node in its last revision.
    $query = $this->select('node_revision', 'nr')->fields('n', [
      'nid',
      'type',
      'language',
      'status',
      'created',
      'changed',
      'comment',
      'promote',
      'sticky',
      'tnid',
      'translate',
    ])->fields('nr', [
      'vid',
      'title',
      'log',
      'timestamp',
    ]);
    $query->addField('n', 'uid', 'node_uid');
    $query->addField('nr', 'uid', 'revision_uid');
    $query->innerJoin('node', 'n', static::JOIN);
    
    // If the content_translation module is enabled, get the source langcode
    // to fill the content_translation_source field.
    if ($this->moduleHandler->moduleExists('content_translation')) {
      $query->leftJoin('node', 'nt', 'n.tnid = nt.nid');
      $query->addField('nt', 'language', 'source_langcode');
    }
    $this->handleTranslations($query);
    if (isset($this->configuration['node_type'])) {
      $query->condition('n.type', $this->configuration['node_type']);
    }
    $query->leftJoin('merci_resource_node', 'mrn', 'n.nid = mrn.nid');
    $query->fields('mrn',
      array(
        'merci_default_availability',
        'merci_sub_type',
        'merci_late_fee_per_hour',
        'merci_rate_per_hour',
        'merci_fee_free_hours',
        'merci_min_cancel_hours',
        'merci_autocheckout',
        'merci_autocheckin',
        'merci_selfcheckout',
        'merci_daypart1_price',
        'merci_daypart2_price',
        'merci_daypart3_price',
        'merci_weekend_price',
      )
    );

    $query->leftJoin('merci_bucket_node', 'mbn', 'n.nid = mbn.nid');
    $query->fields('mbn',
      array(
        'merci_default_availability',
        'merci_sub_type',
        'merci_late_fee_per_hour',
        'merci_rate_per_hour',
        'merci_fee_free_hours',
        'merci_min_cancel_hours',
        'merci_autocheckout',
        'merci_autocheckin',
        'merci_selfcheckout',
        'merci_daypart1_price',
        'merci_daypart2_price',
        'merci_daypart3_price',
        'merci_weekend_price',
      )
    );

    $query->condition(
    $query->orConditionGroup()
      ->condition('mrn.merci_sub_type', '1')
      ->condition('mbn.merci_sub_type', '1')
    );

    $query->join('merci_node_type', 'mnt', 'mnt.type = n.type');
    $query->fields('mnt');
    return $query;
  }

  /**
   * {@inheritdoc}
   */
  public function prepareRow(Row $row) {
    // Get Field API field values.
    $nid = $row->getSourceProperty('nid');
    $vid = $row->getSourceProperty('vid');
    foreach (array_keys($this->getFields('node', $row->getSourceProperty('type'))) as $field) {
      $row->setSourceProperty($field, $this->getFieldValues('node', $field, $nid, $vid));
    }

    // Make sure we always have a translation set.
    if ($row->getSourceProperty('tnid') == 0) {
      $row->setSourceProperty('tnid', $row->getSourceProperty('nid'));
    }

    $query = $this->select('variable', 'v')->fields('v')->condition('name', 'merci_grouping_' . $row->getSourceProperty('type'));

    $result = $query
      ->execute()
      ->fetchAll();


    if (!empty($result)) {
      $tid = unserialize($result[0]['value']);

      if ($tid) {
        $row->setSourceProperty('resource_tree', $tid);
      }
    }

    //Merge merci_resource_node and merci_bucket_node columns.

    $merci_type = $row->getSourceProperty('merci_type_setting');

    $fields = array(
      'merci_default_availability',
      'merci_sub_type',
      'merci_late_fee_per_hour',
      'merci_rate_per_hour',
      'merci_fee_free_hours',
      'merci_min_cancel_hours',
      'merci_autocheckout',
      'merci_autocheckin',
      'merci_selfcheckout',
      'merci_daypart1_price',
      'merci_daypart2_price',
      'merci_daypart3_price',
      'merci_weekend_price',
    );

    if ($merci_type == 'bucket') {
      $alias = 'mbn';
      foreach ($fields as $field) {
        $row->setSourceProperty($field, $row->getSourceProperty($alias . '_' . $field));
      }
    }

    $type = $row->getSourceProperty('type');

    $permissions = $this->select('role_permission', 'rp')
      ->fields('rp', ['rid', 'permission'])
      ->condition('permission', 'edit own ' . $type . ' content')
      ->execute()
      ->fetchCol();
    if (!empty($permissions)) {
      $permissions = $this->select('role_permission', 'rp')
        ->fields('rp', ['rid', 'permission'])
        ->condition('permission', 'delete own ' . $type . ' content')
        ->condition('rid', $permissions, 'IN')
        ->execute()
        ->fetchAll();
    }
    $row->setSourceProperty('permissions', $permissions);

    return parent::prepareRow($row);
  }

/**
   * {@inheritdoc}
   */
  public function fields() {
    $fields = [
      'nid' => $this->t('Node ID'),
      'type' => $this->t('Type'),
      'title' => $this->t('Title'),
      'node_uid' => $this->t('Node authored by (uid)'),
      'revision_uid' => $this->t('Revision authored by (uid)'),
      'created' => $this->t('Created timestamp'),
      'changed' => $this->t('Modified timestamp'),
      'status' => $this->t('Published'),
      'promote' => $this->t('Promoted to front page'),
      'sticky' => $this->t('Sticky at top of lists'),
      'revision' => $this->t('Create new revision'),
      'language' => $this->t('Language (fr, en, ...)'),
      'tnid' => $this->t('The translation set id for this node'),
      'timestamp' => $this->t('The timestamp the latest revision of this node was created.'),
    ];
    return $fields;
  }
  
  /**
   * {@inheritdoc}
   */
  public function getIds() {
    $ids['nid']['type'] = 'integer';
    $ids['nid']['alias'] = 'n';
    return $ids;
  }
  
  /**
   * Adapt our query for translations.
   *
   * @param \Drupal\Core\Database\Query\SelectInterface $query
   *   The generated query.
   */
  protected function handleTranslations(SelectInterface $query) {
    
    // Check whether or not we want translations.
    if (empty($this->configuration['translations'])) {
      
      // No translations: Yield untranslated nodes, or default translations.
      $query->where('n.tnid = 0 OR n.tnid = n.nid');
    }
    else {
      
      // Translations: Yield only non-default translations.
      $query->where('n.tnid <> 0 AND n.tnid <> n.nid');
    }
  }


}