ShortcutSet.php

<?php

namespace Drupal\shortcut\Entity;

use Drupal\Core\Entity\Attribute\ConfigEntityType;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Drupal\Core\Config\Entity\ConfigEntityBundleBase;
use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\shortcut\Form\SetCustomize;
use Drupal\shortcut\Form\ShortcutSetDeleteForm;
use Drupal\shortcut\ShortcutSetAccessControlHandler;
use Drupal\shortcut\ShortcutSetForm;
use Drupal\shortcut\ShortcutSetInterface;
use Drupal\shortcut\ShortcutSetListBuilder;
use Drupal\shortcut\ShortcutSetStorage;

/**
 * Defines the Shortcut set configuration entity.
 */
class ShortcutSet extends ConfigEntityBundleBase implements ShortcutSetInterface {
  
  /**
   * The machine name for the configuration entity.
   *
   * @var string
   */
  protected $id;
  
  /**
   * The human-readable name of the configuration entity.
   *
   * @var string
   */
  protected $label;
  
  /**
   * {@inheritdoc}
   */
  public function postSave(EntityStorageInterface $storage, $update = TRUE) {
    parent::postSave($storage, $update);
    if (!$update && !$this->isSyncing()) {
      // Save a new shortcut set with links copied from the user's default set.
      $default_set = $storage->getDefaultSet(\Drupal::currentUser());
      // This is the default set, do not copy shortcuts.
      if ($default_set->id() != $this->id()) {
        foreach ($default_set->getShortcuts() as $shortcut) {
          $shortcut = $shortcut->createDuplicate();
          $shortcut->enforceIsNew();
          $shortcut->shortcut_set->target_id = $this->id();
          $shortcut->save();
        }
      }
    }
  }
  
  /**
   * {@inheritdoc}
   */
  public static function preDelete(EntityStorageInterface $storage, array $entities) {
    parent::preDelete($storage, $entities);
    foreach ($entities as $entity) {
      $storage->deleteAssignedShortcutSets($entity);
      // Next, delete the shortcuts for this set.
      $shortcut_ids = \Drupal::entityQuery('shortcut')->accessCheck(FALSE)
        ->condition('shortcut_set', $entity->id(), '=')
        ->execute();
      $controller = \Drupal::entityTypeManager()->getStorage('shortcut');
      $entities = $controller->loadMultiple($shortcut_ids);
      $controller->delete($entities);
    }
  }
  
  /**
   * {@inheritdoc}
   */
  public function resetLinkWeights() {
    $weight = -50;
    foreach ($this->getShortcuts() as $shortcut) {
      $shortcut->setWeight(++$weight);
      $shortcut->save();
    }
    return $this;
  }
  
  /**
   * {@inheritdoc}
   */
  public function getShortcuts() {
    $shortcuts = \Drupal::entityTypeManager()->getStorage('shortcut')
      ->loadByProperties([
      'shortcut_set' => $this->id(),
    ]);
    uasort($shortcuts, [
      '\\Drupal\\shortcut\\Entity\\Shortcut',
      'sort',
    ]);
    return $shortcuts;
  }

}