graphql_compose-1.0.0-beta20/modules/graphql_compose_menus/src/Plugin/GraphQL/DataProducer/MenuLinkEntity.php

<?php

declare(strict_types=1);

namespace Drupal\graphql_compose_menus\Plugin\GraphQL\DataProducer;

use Drupal\Component\Uuid\Uuid;
use Drupal\Core\Menu\MenuLinkInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\Core\Plugin\Context\ContextDefinition;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Drupal\graphql\Attribute\DataProducer;
use Drupal\graphql\GraphQL\Buffers\EntityUuidBuffer;
use Drupal\graphql\GraphQL\Execution\FieldContext;
use Drupal\graphql\Plugin\GraphQL\DataProducer\DataProducerPluginBase;
use Drupal\graphql_compose\EntityTranslateTrait;
use Drupal\menu_link_content\Plugin\Menu\MenuLinkContent;
use GraphQL\Deferred;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Returns the menu link content entity of a menu link.
 */
#[DataProducer(
  id: "menu_link_entity",
  name: new TranslatableMarkup("Menu link content entity"),
  description: new TranslatableMarkup("Returns the menu link content of a menu link."),
  produces: new ContextDefinition(
    data_type: "any",
    label: new TranslatableMarkup("Menu Link Content entity"),
  ),
  consumes: [
    "link" => new ContextDefinition(
      data_type: "any",
      label: new TranslatableMarkup("Menu link tree element")
    ),
  ],
)]
class MenuLinkEntity extends DataProducerPluginBase implements ContainerFactoryPluginInterface {

  use EntityTranslateTrait;

  /**
   * Create the menu link translated URL resolver.
   *
   * @param array $configuration
   *   The plugin configuration.
   * @param string $plugin_id
   *   The plugin id.
   * @param mixed $plugin_definition
   *   The plugin definition.
   * @param \Drupal\graphql\GraphQL\Buffers\EntityUuidBuffer $entityBuffer
   *   The entity buffer service.
   */
  public function __construct(
    array $configuration,
    $plugin_id,
    $plugin_definition,
    protected EntityUuidBuffer $entityBuffer,
  ) {
    parent::__construct($configuration, $plugin_id, $plugin_definition);
  }

  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
    return new static(
      $configuration,
      $plugin_id,
      $plugin_definition,
      $container->get('graphql.buffer.entity_uuid'),
    );
  }

  /**
   * Resolve the language of the menu item.
   *
   * @param \Drupal\Core\Menu\MenuLinkInterface $link
   *   The menu link plugin to resolve the entity.
   * @param \Drupal\graphql\GraphQL\Execution\FieldContext $context
   *   Cacheability metadata for this request.
   *
   * @return \GraphQL\Deferred|null
   *   The menu link content entity or null.
   */
  public function resolve(MenuLinkInterface $link, FieldContext $context): ?Deferred {

    if (!$link instanceof MenuLinkContent) {
      return NULL;
    }

    $derivative_id = $link->getDerivativeId();

    if (!Uuid::isValid($derivative_id)) {
      return NULL;
    }

    $resolver = $this->entityBuffer->add('menu_link_content', $derivative_id);

    // Get the current or preferred language.
    $langcode = $this->getCurrentLanguage($context);

    return new Deferred(function () use ($resolver, $langcode) {
      if ($entity = $resolver()) {
        return $this->getTranslation($entity, $langcode);
      }

      return NULL;
    });
  }

}