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;
    }

}