comparison core/modules/workspaces/src/EntityOperations.php @ 17:129ea1e6d783

Update, including to Drupal core 8.6.10

+<?php
+
+namespace Drupal\workspaces;
+
+use Drupal\Core\DependencyInjection\ContainerInjectionInterface;
+use Drupal\Core\Entity\EntityInterface;
+use Drupal\Core\Entity\EntityTypeManagerInterface;
+use Drupal\Core\Entity\RevisionableInterface;
+use Drupal\Core\Form\FormStateInterface;
+use Drupal\Core\StringTranslation\StringTranslationTrait;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Defines a class for reacting to entity events.
+ *
+ * @internal
+ */
+class EntityOperations implements ContainerInjectionInterface {
+
+  use StringTranslationTrait;
+
+  /**
+   * The entity type manager service.
+   *
+   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
+   */
+  protected $entityTypeManager;
+
+  /**
+   * The workspace manager service.
+   *
+   * @var \Drupal\workspaces\WorkspaceManagerInterface
+   */
+  protected $workspaceManager;
+
+  /**
+   * Constructs a new EntityOperations instance.
+   *
+   * @param \Drupal\Core\Entity\EntityTypeManagerInterface $entity_type_manager
+   *   The entity type manager service.
+   * @param \Drupal\workspaces\WorkspaceManagerInterface $workspace_manager
+   *   The workspace manager service.
+   */
+  public function __construct(EntityTypeManagerInterface $entity_type_manager, WorkspaceManagerInterface $workspace_manager) {
+    $this->entityTypeManager = $entity_type_manager;
+    $this->workspaceManager = $workspace_manager;
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public static function create(ContainerInterface $container) {
+    return new static(
+      $container->get('entity_type.manager'),
+      $container->get('workspaces.manager')
+    );
+  }
+
+  /**
+   * Acts on entities when loaded.
+   *
+   * @see hook_entity_load()
+   */
+  public function entityLoad(array &$entities, $entity_type_id) {
+    // Only run if the entity type can belong to a workspace and we are in a
+    // non-default workspace.
+    if (!$this->workspaceManager->shouldAlterOperations($this->entityTypeManager->getDefinition($entity_type_id))) {
+      return;
+    }
+
+    // Get a list of revision IDs for entities that have a revision set for the
+    // current active workspace. If an entity has multiple revisions set for a
+    // workspace, only the one with the highest ID is returned.
+    $entity_ids = array_keys($entities);
+    $max_revision_id = 'max_target_entity_revision_id';
+    $results = $this->entityTypeManager
+      ->getStorage('workspace_association')
+      ->getAggregateQuery()
+      ->accessCheck(FALSE)
+      ->allRevisions()
+      ->aggregate('target_entity_revision_id', 'MAX', NULL, $max_revision_id)
+      ->groupBy('target_entity_id')
+      ->condition('target_entity_type_id', $entity_type_id)
+      ->condition('target_entity_id', $entity_ids, 'IN')
+      ->condition('workspace', $this->workspaceManager->getActiveWorkspace()->id())
+      ->execute();
+
+    // Since hook_entity_load() is called on both regular entity load as well as
+    // entity revision load, we need to prevent infinite recursion by checking
+    // whether the default revisions were already swapped with the workspace
+    // revision.
+    // @todo This recursion protection should be removed when
+    //   https://www.drupal.org/project/drupal/issues/2928888 is resolved.
+    if ($results) {
+      $results = array_filter($results, function ($result) use ($entities, $max_revision_id) {
+        return $entities[$result['target_entity_id']]->getRevisionId() != $result[$max_revision_id];
+      });
+    }
+
+    if ($results) {
+      /** @var \Drupal\Core\Entity\RevisionableStorageInterface $storage */
+      $storage = $this->entityTypeManager->getStorage($entity_type_id);
+
+      // Swap out every entity which has a revision set for the current active
+      // workspace.
+      $swap_revision_ids = array_column($results, $max_revision_id);
+      foreach ($storage->loadMultipleRevisions($swap_revision_ids) as $revision) {
+        $entities[$revision->id()] = $revision;
+      }
+    }
+  }
+
+  /**
+   * Acts on an entity before it is created or updated.
+   *
+   * @param \Drupal\Core\Entity\EntityInterface $entity
+   *   The entity being saved.
+   *
+   * @see hook_entity_presave()
+   */
+  public function entityPresave(EntityInterface $entity) {
+    $entity_type = $entity->getEntityType();
+
+    // Only run if this is not an entity type provided by the Workspaces module
+    // and we are in a non-default workspace
+    if ($entity_type->getProvider() === 'workspaces' || $this->workspaceManager->getActiveWorkspace()->isDefaultWorkspace()) {
+      return;
+    }
+
+    // Disallow any change to an unsupported entity when we are not in the
+    // default workspace.
+    if (!$this->workspaceManager->isEntityTypeSupported($entity_type)) {
+      throw new \RuntimeException('This entity can only be saved in the default workspace.');
+    }
+
+    /** @var \Drupal\Core\Entity\RevisionableInterface|\Drupal\Core\Entity\EntityPublishedInterface $entity */
+    if (!$entity->isNew() && !isset($entity->_isReplicating)) {
+      // Force a new revision if the entity is not replicating.
+      $entity->setNewRevision(TRUE);
+
+      // All entities in the non-default workspace are pending revisions,
+      // regardless of their publishing status. This means that when creating
+      // a published pending revision in a non-default workspace it will also be
+      // a published pending revision in the default workspace, however, it will
+      // become the default revision only when it is replicated to the default
+      // workspace.
+      $entity->isDefaultRevision(FALSE);
+    }
+
+    // When a new published entity is inserted in a non-default workspace, we
+    // actually want two revisions to be saved:
+    // - An unpublished default revision in the default ('live') workspace.
+    // - A published pending revision in the current workspace.
+    if ($entity->isNew() && $entity->isPublished()) {
+      // Keep track of the publishing status in a dynamic property for
+      // ::entityInsert(), then unpublish the default revision.
+      // @todo Remove this dynamic property once we have an API for associating
+      //   temporary data with an entity: https://www.drupal.org/node/2896474.
+      $entity->_initialPublished = TRUE;
+      $entity->setUnpublished();
+    }
+  }
+
+  /**
+   * Responds to the creation of a new entity.
+   *
+   * @param \Drupal\Core\Entity\EntityInterface $entity
+   *   The entity that was just saved.
+   *
+   * @see hook_entity_insert()
+   */
+  public function entityInsert(EntityInterface $entity) {
+    /** @var \Drupal\Core\Entity\RevisionableInterface|\Drupal\Core\Entity\EntityPublishedInterface $entity */
+    // Only run if the entity type can belong to a workspace and we are in a
+    // non-default workspace.
+    if (!$this->workspaceManager->shouldAlterOperations($entity->getEntityType())) {
+      return;
+    }
+
+    $this->trackEntity($entity);
+
+    // When an entity is newly created in a workspace, it should be published in
+    // that workspace, but not yet published on the live workspace. It is first
+    // saved as unpublished for the default revision, then immediately a second
+    // revision is created which is published and attached to the workspace.
+    // This ensures that the published version of the entity does not 'leak'
+    // into the live site. This differs from edits to existing entities where
+    // there is already a valid default revision for the live workspace.
+    if (isset($entity->_initialPublished)) {
+      // Operate on a clone to avoid changing the entity prior to subsequent
+      // hook_entity_insert() implementations.
+      $pending_revision = clone $entity;
+      $pending_revision->setPublished();
+      $pending_revision->isDefaultRevision(FALSE);
+      $pending_revision->save();
+    }
+  }
+
+  /**
+   * Responds to updates to an entity.
+   *
+   * @param \Drupal\Core\Entity\EntityInterface $entity
+   *   The entity that was just saved.
+   *
+   * @see hook_entity_update()
+   */
+  public function entityUpdate(EntityInterface $entity) {
+    // Only run if the entity type can belong to a workspace and we are in a
+    // non-default workspace.
+    if (!$this->workspaceManager->shouldAlterOperations($entity->getEntityType())) {
+      return;
+    }
+
+    // Only track new revisions.
+    /** @var \Drupal\Core\Entity\RevisionableInterface $entity */
+    if ($entity->getLoadedRevisionId() != $entity->getRevisionId()) {
+      $this->trackEntity($entity);
+    }
+  }
+
+  /**
+   * Acts on an entity before it is deleted.
+   *
+   * @param \Drupal\Core\Entity\EntityInterface $entity
+   *   The entity being deleted.
+   *
+   * @see hook_entity_predelete()
+   */
+  public function entityPredelete(EntityInterface $entity) {
+    $entity_type = $entity->getEntityType();
+
+    // Only run if this is not an entity type provided by the Workspaces module
+    // and we are in a non-default workspace
+    if ($entity_type->getProvider() === 'workspaces' || $this->workspaceManager->getActiveWorkspace()->isDefaultWorkspace()) {
+      return;
+    }
+
+    // Disallow any change to an unsupported entity when we are not in the
+    // default workspace.
+    if (!$this->workspaceManager->isEntityTypeSupported($entity_type)) {
+      throw new \RuntimeException('This entity can only be deleted in the default workspace.');
+    }
+  }
+
+  /**
+   * Updates or creates a WorkspaceAssociation entity for a given entity.
+   *
+   * If the passed-in entity can belong to a workspace and already has a
+   * WorkspaceAssociation entity, then a new revision of this will be created with
+   * the new information. Otherwise, a new WorkspaceAssociation entity is created to
+   * store the passed-in entity's information.
+   *
+   * @param \Drupal\Core\Entity\EntityInterface $entity
+   *   The entity to update or create from.
+   */
+  protected function trackEntity(EntityInterface $entity) {
+    /** @var \Drupal\Core\Entity\RevisionableInterface|\Drupal\Core\Entity\EntityPublishedInterface $entity */
+    // If the entity is not new, check if there's an existing
+    // WorkspaceAssociation entity for it.
+    $workspace_association_storage = $this->entityTypeManager->getStorage('workspace_association');
+    if (!$entity->isNew()) {
+      $workspace_associations = $workspace_association_storage->loadByProperties([
+        'target_entity_type_id' => $entity->getEntityTypeId(),
+        'target_entity_id' => $entity->id(),
+      ]);
+
+      /** @var \Drupal\Core\Entity\ContentEntityInterface $workspace_association */
+      $workspace_association = reset($workspace_associations);
+    }
+
+    // If there was a WorkspaceAssociation entry create a new revision,
+    // otherwise create a new entity with the type and ID.
+    if (!empty($workspace_association)) {
+      $workspace_association->setNewRevision(TRUE);
+    }
+    else {
+      $workspace_association = $workspace_association_storage->create([
+        'target_entity_type_id' => $entity->getEntityTypeId(),
+        'target_entity_id' => $entity->id(),
+      ]);
+    }
+
+    // Add the revision ID and the workspace ID.
+    $workspace_association->set('target_entity_revision_id', $entity->getRevisionId());
+    $workspace_association->set('workspace', $this->workspaceManager->getActiveWorkspace()->id());
+
+    // Save without updating the tracked content entity.
+    $workspace_association->save();
+  }
+
+  /**
+   * Alters entity forms to disallow concurrent editing in multiple workspaces.
+   *
+   * @param array $form
+   *   An associative array containing the structure of the form.
+   * @param \Drupal\Core\Form\FormStateInterface $form_state
+   *   The current state of the form.
+   * @param string $form_id
+   *   The form ID.
+   *
+   * @see hook_form_alter()
+   */
+  public function entityFormAlter(array &$form, FormStateInterface $form_state, $form_id) {
+    /** @var \Drupal\Core\Entity\EntityInterface $entity */
+    $entity = $form_state->getFormObject()->getEntity();
+    if (!$this->workspaceManager->isEntityTypeSupported($entity->getEntityType())) {
+      return;
+    }
+
+    // For supported entity types, signal the fact that this form is safe to use
+    // in a non-default workspace.
+    // @see \Drupal\workspaces\FormOperations::validateForm()
+    $form_state->set('workspace_safe', TRUE);
+
+    // Add an entity builder to the form which marks the edited entity object as
+    // a pending revision. This is needed so validation constraints like
+    // \Drupal\path\Plugin\Validation\Constraint\PathAliasConstraintValidator
+    // know in advance (before hook_entity_presave()) that the new revision will
+    // be a pending one.
+    $active_workspace = $this->workspaceManager->getActiveWorkspace();
+    if (!$active_workspace->isDefaultWorkspace()) {
+      $form['#entity_builders'][] = [get_called_class(), 'entityFormEntityBuild'];
+    }
+
+    /** @var \Drupal\workspaces\WorkspaceAssociationStorageInterface $workspace_association_storage */
+    $workspace_association_storage = $this->entityTypeManager->getStorage('workspace_association');
+    if ($workspace_ids = $workspace_association_storage->getEntityTrackingWorkspaceIds($entity)) {
+      // An entity can only be edited in one workspace.
+      $workspace_id = reset($workspace_ids);
+
+      if ($workspace_id !== $active_workspace->id()) {
+        $workspace = $this->entityTypeManager->getStorage('workspace')->load($workspace_id);
+
+        $form['#markup'] = $this->t('The content is being edited in the %label workspace.', ['%label' => $workspace->label()]);
+        $form['#access'] = FALSE;
+      }
+    }
+  }
+
+  /**
+   * Entity builder that marks all supported entities as pending revisions.
+   */
+  public static function entityFormEntityBuild($entity_type_id, RevisionableInterface $entity, &$form, FormStateInterface &$form_state) {
+    // Set the non-default revision flag so that validation constraints are also
+    // aware that a pending revision is about to be created.
+    $entity->isDefaultRevision(FALSE);
+  }
+
+}