localgov_publications-1.0.14/src/Token/Hooks.php

<?php

namespace Drupal\localgov_publications\Token;

use Drupal\Core\DependencyInjection\ContainerInjectionInterface;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Render\BubbleableMetadata;
use Drupal\localgov_publications\Service\PublicationManager;
use Drupal\node\NodeInterface;
use Drupal\path_alias\AliasManagerInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * A class for hook implementations.
 */
class Hooks implements ContainerInjectionInterface {

  /**
   * Constructor.
   */
  public function __construct(
    private PublicationManager $publicationManager,
    private AliasManagerInterface $aliasManager,
    private EntityTypeManagerInterface $entityTypeManager,
  ) {
  }

  /**
   * {@inheritDoc}
   */
  public static function create(ContainerInterface $container) {
    return new static(
      $container->get('localgov_publications.publication_manager'),
      $container->get('path_alias.manager'),
      $container->get('entity_type.manager'),
    );
  }

  /**
   * Implements hook_tokens_alter().
   *
   * See localgov_publications_tokens_alter().
   */
  public function tokensAlter(array &$replacements, array $context, BubbleableMetadata $bubbleable_metadata) {

    if ($context['type'] !== 'node' || !isset($context['data']['node'])) {
      return;
    }

    /** @var \Drupal\node\NodeInterface $node */
    $node = $context['data']['node'];

    if (!isset($node->book['bid']) || $node->book['bid'] === 0) {
      return;
    }

    $nodeStorage = $this->entityTypeManager->getStorage('node');

    if (isset($context['tokens']['localgov-publication-cover-page-alias'])) {

      $coverPage = $this->publicationManager->getCoverPage($node->book['bid']);
      if ($coverPage instanceof NodeInterface) {
        $bubbleable_metadata->addCacheableDependency($coverPage);
        $coverPageAlias = $this->aliasManager->getAliasByPath('/node/' . $coverPage->id());
        $replacements[$context['tokens']['localgov-publication-cover-page-alias']] = $coverPageAlias;
      }
    }

    if (isset($context['tokens']['localgov-publication-path'])) {

      $pathElements = [];

      if ($node->book['bid'] === $node->id()) {
        // Add the cover page alias if we are on the root node and there's a
        // cover page.
        $coverPage = $this->publicationManager->getCoverPage($node->book['bid']);
        if ($coverPage instanceof NodeInterface) {
          $bubbleable_metadata->addCacheableDependency($coverPage);
          $coverPageAlias = trim($this->aliasManager->getAliasByPath('/node/' . $coverPage->id()), '/');
          $pathElements[] = $coverPageAlias;
        }
      }
      else {
        // Add the root node's URL alias in if we're not on the root node now:
        $rootNode = $nodeStorage->load($node->book['bid']);
        if ($rootNode instanceof NodeInterface) {
          $pathElements[] = $this->aliasManager->getAliasByPath('/node/' . $rootNode->id());
        }
      }

      foreach ($this->bookParents($node) as $parent) {
        $pathElements[] = $parent;
      }

      $replacements[$context['tokens']['localgov-publication-path']] = implode('/', $pathElements);
    }
  }

  /**
   * Loads all the parents of the book page.
   *
   * Doesn't include the current node or the root node.
   */
  private function bookParents(NodeInterface $argNode) {

    $nodeStorage = $this->entityTypeManager->getStorage('node');

    // Re-load the node, to ensure it's got all the book data on it.
    $node = $nodeStorage->load($argNode->id());

    if (empty($node->book['nid'])) {
      return [];
    }

    $parents = [];

    // Skip the first level of the book, as we add this elsewhere.
    $i = 2;
    while (isset($node->book["p$i"]) && ($node->book["p$i"] != $node->book['nid'])) {
      $node = $nodeStorage->load($node->book["p$i"]);
      if ($node instanceof NodeInterface) {
        $parents[] = $node->getTitle();
      }
      $i++;
    }

    return $parents;
  }

}