WorkspacesEntityRepository.php

<?php

namespace Drupal\workspaces;

use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\EntityRepositoryInterface;

/**
 * Provides workspace-specific mechanisms for retrieving entities.
 */
class WorkspacesEntityRepository implements EntityRepositoryInterface {
  public function __construct(protected EntityRepositoryInterface $inner, protected WorkspaceManagerInterface $workspaceManager) {
  }
  
  /**
   * {@inheritdoc}
   */
  public function loadEntityByUuid($entity_type_id, $uuid) {
    return $this->inner
      ->loadEntityByUuid($entity_type_id, $uuid);
  }
  
  /**
   * {@inheritdoc}
   */
  public function loadEntityByConfigTarget($entity_type_id, $target) {
    return $this->inner
      ->loadEntityByConfigTarget($entity_type_id, $target);
  }
  
  /**
   * {@inheritdoc}
   */
  public function getTranslationFromContext(EntityInterface $entity, $langcode = NULL, $context = []) {
    return $this->inner
      ->getTranslationFromContext($entity, $langcode, $context);
  }
  
  /**
   * {@inheritdoc}
   */
  public function getActive($entity_type_id, $entity_id, ?array $contexts = NULL) {
    // When there's no active workspace, the active entity variant is the
    // canonical one.
    if (!$this->workspaceManager
      ->hasActiveWorkspace()) {
      return $this->inner
        ->getCanonical($entity_type_id, $entity_id, $contexts);
    }
    return $this->inner
      ->getActive($entity_type_id, $entity_id, $contexts);
  }
  
  /**
   * {@inheritdoc}
   */
  public function getActiveMultiple($entity_type_id, array $entity_ids, ?array $contexts = NULL) {
    // When there's no active workspace, the active entity variant is the
    // canonical one.
    if (!$this->workspaceManager
      ->hasActiveWorkspace()) {
      return $this->inner
        ->getCanonicalMultiple($entity_type_id, $entity_ids, $contexts);
    }
    return $this->inner
      ->getActiveMultiple($entity_type_id, $entity_ids, $contexts);
  }
  
  /**
   * {@inheritdoc}
   */
  public function getCanonical($entity_type_id, $entity_id, ?array $contexts = NULL) {
    return $this->inner
      ->getCanonical($entity_type_id, $entity_id, $contexts);
  }
  
  /**
   * {@inheritdoc}
   */
  public function getCanonicalMultiple($entity_type_id, array $entity_ids, ?array $contexts = NULL) {
    return $this->inner
      ->getCanonicalMultiple($entity_type_id, $entity_ids, $contexts);
  }

}