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

<?php

declare(strict_types=1);

namespace Drupal\graphql_compose_menus\Plugin\GraphQL\DataProducer;

use Drupal\Core\Plugin\Context\ContextDefinition;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Drupal\Core\Url;
use Drupal\graphql\Attribute\DataProducer;
use Drupal\graphql\Plugin\GraphQL\DataProducer\DataProducerPluginBase;
use Drupal\menu_link_content\Entity\MenuLinkContent;

/**
 * Returns the translated URL object of a menu link.
 */
#[DataProducer(
  id: "menu_link_url_override",
  name: new TranslatableMarkup("Menu link translated url"),
  description: new TranslatableMarkup("Returns the translated URL of a menu link."),
  produces: new ContextDefinition(
    data_type: "any",
    label: new TranslatableMarkup("URL"),
  ),
  consumes: [
    "entity" => new ContextDefinition(
      data_type: "any",
      label: new TranslatableMarkup("Menu link content entity"),
    ),
  ],
)]
class MenuLinkUrlOverride extends DataProducerPluginBase {

  /**
   * Resolve the translated menu link url.
   *
   * @param \Drupal\menu_link_content\Entity\MenuLinkContent $entity
   *   The menu link content entity to resolve the url off of.
   *
   * @return \Drupal\Core\Url
   *   The Url.
   */
  public function resolve(MenuLinkContent $entity): Url {
    if ($entity->hasField('link_override')) {
      /** @var \Drupal\link\LinkItemInterface|null $link_override */
      $link_override = $entity->get('link_override')->first();

      if ($link_override && !$link_override->isEmpty()) {
        return $link_override->getUrl();
      }
    }

    return $entity->getUrlObject();
  }

}