multiversion-8.x-1.0-beta34/src/EventSubscriber/MenuLinkContentMigrateSubscriber.php

<?php

namespace Drupal\multiversion\EventSubscriber;

use Drupal\Core\Database\Connection;
use Drupal\Core\Menu\MenuLinkManagerInterface;
use Drupal\multiversion\Event\MultiversionManagerEvent;
use Drupal\multiversion\Event\MultiversionManagerEvents;
use Drupal\multiversion\MultiversionManager;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;

/**
 * MenuContentLinkMigrateSubscriber class.
 *
 * A menu_tree database table should be rediscovered
 * after enabling/disabling a menu_link_content entity.
 */
class MenuLinkContentMigrateSubscriber implements EventSubscriberInterface {

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

  /**
   * @var \Drupal\Core\Menu\MenuLinkManagerInterface
   */
  protected $menuLinkManager;

  /**
   * Class constructor.
   *
   * @param \Drupal\Core\Database\Connection $connection
   * @param \Drupal\Core\Menu\MenuLinkManagerInterface $menu_link_manager
   */
  public function __construct(Connection $connection, MenuLinkManagerInterface $menu_link_manager) {
    $this->connection = $connection;
    $this->menuLinkManager = $menu_link_manager;
  }

  /**
   * Set rediscover property and rebuild menu trees.
   *
   * @param \Drupal\multiversion\Event\MultiversionManagerEvent $event
   */
  public function onPostMigrateLinks(MultiversionManagerEvent $event) {
    if ($event->getOp() === MultiversionManager::OP_DISABLE && $entity_type = $event->getEntityType('menu_link_content')) {
      $data_table = $entity_type->getDataTable();
      // Truncate 'menu_tree' table before rebuild.
      $this->connection->truncate('menu_tree')->execute();
      // Set a rediscover and rebuild menu_tree table.
      // @see \Drupal\menu_link_content\Plugin\Deriver\MenuLinkContentDeriver
      $this->connection->update($data_table)
        ->fields(['rediscover' => 1])
        ->execute();
      $this->menuLinkManager->rebuild();
    }
  }

  /**
   * {@inheritdoc}
   */
  public static function getSubscribedEvents() {
    return [MultiversionManagerEvents::POST_MIGRATE => ['onPostMigrateLinks']];
  }

}