graphql_compose-1.0.0-beta20/modules/graphql_compose_layout_builder/src/Plugin/GraphQL/DataProducer/LayoutBuilderContexts.php

<?php

declare(strict_types=1);

namespace Drupal\graphql_compose_layout_builder\Plugin\GraphQL\DataProducer;

use Drupal\Core\Entity\Entity\EntityViewDisplay;
use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\Core\Plugin\Context\Context;
use Drupal\Core\Plugin\Context\ContextDefinition;
use Drupal\Core\Plugin\Context\ContextRepositoryInterface;
use Drupal\Core\Plugin\Context\EntityContext;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Drupal\graphql\Attribute\DataProducer;
use Drupal\graphql\GraphQL\Execution\FieldContext;
use Drupal\graphql\Plugin\GraphQL\DataProducer\DataProducerPluginBase;
use Drupal\graphql_compose_layout_builder\EnabledBundlesTrait;
use Drupal\layout_builder\SectionStorage\SectionStorageManagerInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Layout Builder section contexts.
 */
#[DataProducer(
  id: "layout_builder_contexts",
  name: new TranslatableMarkup("Layout Builder contexts"),
  description: new TranslatableMarkup("Get layout builder contexts."),
  produces: new ContextDefinition(
    data_type: "any",
    label: new TranslatableMarkup("Section"),
  ),
  consumes: [
    "entity" => new ContextDefinition(
      data_type: "entity",
      label: new TranslatableMarkup("Entity with layouts enabled"),
    ),
    "view_mode" => new ContextDefinition(
      data_type: "string",
      label: new TranslatableMarkup("View mode name"),
      required: FALSE,
    ),
  ],
)]
class LayoutBuilderContexts extends DataProducerPluginBase implements ContainerFactoryPluginInterface {

  use EnabledBundlesTrait;

  /**
   * The layout builder section storage manager.
   *
   * @var \Drupal\layout_builder\SectionStorage\SectionStorageManagerInterface
   */
  protected SectionStorageManagerInterface $sectionStorageManager;

  /**
   * The context repository.
   *
   * @var \Drupal\Core\Plugin\Context\ContextRepositoryInterface
   */
  protected ContextRepositoryInterface $contextRepository;

  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
    $instance = new static(
      $configuration,
      $plugin_id,
      $plugin_definition
    );

    $instance->sectionStorageManager = $container->get('plugin.manager.layout_builder.section_storage');
    $instance->contextRepository = $container->get('context.repository');

    return $instance;
  }

  /**
   * Resolves the request to the requested values.
   *
   * @param \Drupal\Core\Entity\EntityInterface|null $entity
   *   The field the comment references are attached to.
   * @param string|null $view_mode
   *   The view mode to load.
   * @param \Drupal\graphql\GraphQL\Execution\FieldContext $context
   *   Cacheability context for this request.
   *
   * @return array|null
   *   An array of contexts for the requested entity and view_mode.
   */
  public function resolve(?EntityInterface $entity, ?string $view_mode, FieldContext $context) {

    // If access was denied to the field, $field_list will be null.
    if (!$entity) {
      return NULL;
    }

    $context->addCacheableDependency($entity);

    // Check if view entity is enabled.
    $view_modes_enabled = $this->getLayoutBuilderViewDisplays(
      $entity->getEntityTypeId(),
      $entity->bundle()
    );

    $view_mode = $view_mode ?: 'default';
    $view_mode = array_key_exists($view_mode, $view_modes_enabled) ? $view_mode : 'default';

    $context->addCacheableDependency($view_modes_enabled[$view_mode]);

    $display = EntityViewDisplay::collectRenderDisplay($entity, $view_mode);

    $available_context_ids = array_keys($this->contextRepository->getAvailableContexts());

    return $this->contextRepository->getRuntimeContexts($available_context_ids) + [
      'view_mode' => new Context(ContextDefinition::create('string'), $display->getMode()),
      'entity' => EntityContext::fromEntity($entity),
      'display' => EntityContext::fromEntity($display),
    ];
  }

}