smart_date-3.1.0-beta1/src/Commands/SmartDateDrushCommands.php

<?php

namespace Drupal\smart_date\Commands;

use Drupal\Core\Config\ConfigFactoryInterface;
use Drupal\Core\Database\Connection;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drush\Commands\DrushCommands;

/**
 * A drush command file.
 *
 * @package Drupal\smart_date\Commands
 */
class SmartDateDrushCommands extends DrushCommands {

  /**
   * The config factory.
   *
   * @var \Drupal\Core\Config\ConfigFactoryInterface
   */
  protected $configFactory;

  /**
   * The database connection.
   *
   * @var \Drupal\Core\Database\Connection
   */
  protected $database;

  /**
   * Information about the entity type.
   *
   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
   */
  protected $entityTypeManager;

  /**
   * {@inheritdoc}
   */
  public function __construct(ConfigFactoryInterface $configFactory, Connection $database, EntityTypeManagerInterface $entityTypeManager) {
    $this->configFactory = $configFactory;
    $this->database = $database;
    $this->entityTypeManager = $entityTypeManager;
  }

  /**
   * Drush command to migrate core fields to Smart Date fields.
   *
   * @param string $bundle
   *   Content type or other bundle whose data will be used.
   * @param string $dest
   *   Field data will be copied to.
   * @param string $source_start
   *   Field to copy data from.
   * @param string $source_end
   *   Field to copy data from.
   * @param string $source_all_day
   *   Field to copy data from.
   * @param array $options
   *   Additional configuration options.
   *
   * @command smart_date:migrate
   * @aliases sdm
   * @option clear
   *   Clear any data in the destination field.
   * @option entity
   *   Which entity to use as the destination.
   * @option default_duration
   *   If no end date, assumed duration.
   * @option langcode
   *   Language code to store.
   */
  public function migrate(
    $bundle,
    $dest,
    $source_start,
    $source_end = NULL,
    $source_all_day = NULL,
    array $options = [
      'clear' => FALSE,
      'entity' => 'node',
      'default_duration' => 0,
      'langcode' => NULL,
    ],
  ): void {
    // @todo Sanitize provide input.
    $entity = $options['entity'];
    $dest_table = $entity . '__' . $dest;
    $duration = $def_duration = (int) $options['default_duration'];
    $site_tz_name = $this->configFactory->get('system.date')->get('timezone.default');

    $connection = $this->database;
    if ($options['clear']) {
      $this->output()->writeln('Clearing existing values.');
      $connection->truncate($dest_table)->execute();
    }
    $this->output()->writeln('Starting date migration.');

    $definition = $this->entityTypeManager->getDefinition($options['entity']);
    $bundle_key = $definition->getKey('bundle');

    // Get all events.
    $events = $this->entityTypeManager->getStorage($entity)
      ->loadByProperties([$bundle_key => $bundle]);

    $utc = new \DateTimeZone('UTC');

    foreach ($events as $event) {
      $dates = $event->get($source_start)->getValue();
      $all_day_set = [];
      if ($source_all_day) {
        $all_day_set = $event->get($source_all_day)->getValue();
      }
      $end_dates_set = [];
      if ($source_end && $source_start != $source_end) {
        $end_dates_set = $event->get($source_end)->getValue();
      }
      $fallback_langcode = $event->get($source_start)->getLangcode();
      // Hardcoded last resort langcode value of 'und'.
      if (empty($fallback_langcode)) {
        $fallback_langcode = 'und';
      }
      $langcode = $options['langcode'] ?? $fallback_langcode;
      foreach ($dates as $delta => $date) {
        $start_date = $date['value'];
        // Only store the timezone for all day events.
        $timezone = NULL;
        // If a field was provided to check for all day, check it.
        if ($all_day_set) {
          $all_day = $all_day_set[$delta]['value'];
        }
        elseif (strlen($start_date) == 10) {
          $all_day = TRUE;
        }
        else {
          $all_day = FALSE;
        }

        $end_date = NULL;

        if ($end_dates_set && isset($end_dates_set[$delta]['value'])) {
          $end_date = $end_dates_set[$delta]['value'];
        }
        else {
          // Assume a datetime range, so look for the end_value.
          if (!empty($date['end_value'])) {
            $end_date = $date['end_value'];
          }
        }

        if (!empty($all_day)) {
          // Store the timezone for all day events.
          $timezone = $site_tz_name;
          $tz_obj = new \DateTimeZone($site_tz_name);
          $date = new \DateTime(substr($start_date, 0, 10) . ' 00:00:00', $tz_obj);
          $date = $date->format('U');

          $start_date = $date;
          if (!empty($end_date)) {
            $end_date = new \DateTime(substr($end_date, 0, 10) . ' 11:59:00', $tz_obj);
            $end_date = $end_date->format('U');

            // If valid end date, set duration. Otherwise make a new end date.
            if ($start_date < $end_date) {
              $duration = round(($end_date - $start_date) / 60);
            }
            else {
              $end_date = NULL;
            }
          }

          if (!$end_date) {
            // If the end date is bogus, make all day for the day it starts.
            $end_date = $date + 86340;
            $duration = 1439;
          }
        }
        else {
          $start_date = new \DateTime($start_date, $utc);
          $start_date = $start_date->format('U');

          // Remove any seconds from the incoming value.
          $start_date -= $start_date % 60;

          if (!empty($end_date)) {
            $end_date = new \DateTime($end_date, $utc);
            $end_date = $end_date->format('U');

            // Remove any seconds from the incoming value.
            $end_date -= $end_date % 60;

            // If valid end date, set duration. Otherwise make a new end date.
            if ($start_date < $end_date) {
              $duration = round(($end_date - $start_date) / 60);
            }
            else {
              $end_date = NULL;
            }
          }

          if (!$end_date) {
            // If the end date is bogus, use default duration, set earlier.
            $end_date = $start_date + ($def_duration * 60);
          }
        }

        // Insert the resulting data.
        $connection->insert($dest_table)
          ->fields([
            'bundle' => $bundle,
            'deleted' => 0,
            'entity_id' => $event->id(),
            'revision_id' => $event->getRevisionId() ?? $event->id(),
            'langcode' => $langcode,
            'delta' => $delta,
            $dest . '_value' => $start_date,
            $dest . '_end_value' => $end_date,
            $dest . '_duration' => $duration,
            $dest . '_rrule' => NULL,
            $dest . '_rrule_index' => NULL,
            $dest . '_timezone' => $timezone,
          ])
          ->execute();
      }
    }

    $this->output()->writeln('Finished date migration, flushing caches.');
    drupal_flush_all_caches();
  }

}