Chris@17: <?php
Chris@17: 
Chris@17: namespace Drupal\workspaces\EntityQuery;
Chris@17: 
Chris@17: use Drupal\Core\Database\Connection;
Chris@17: use Drupal\Core\Entity\EntityTypeInterface;
Chris@17: use Drupal\workspaces\WorkspaceManagerInterface;
Chris@17: 
Chris@17: /**
Chris@17:  * Provides workspaces-specific helpers for altering entity queries.
Chris@17:  */
Chris@17: trait QueryTrait {
Chris@17: 
Chris@17:   /**
Chris@17:    * The workspace manager.
Chris@17:    *
Chris@17:    * @var \Drupal\workspaces\WorkspaceManagerInterface
Chris@17:    */
Chris@17:   protected $workspaceManager;
Chris@17: 
Chris@17:   /**
Chris@17:    * Constructs a Query object.
Chris@17:    *
Chris@17:    * @param \Drupal\Core\Entity\EntityTypeInterface $entity_type
Chris@17:    *   The entity type definition.
Chris@17:    * @param string $conjunction
Chris@17:    *   - AND: all of the conditions on the query need to match.
Chris@17:    *   - OR: at least one of the conditions on the query need to match.
Chris@17:    * @param \Drupal\Core\Database\Connection $connection
Chris@17:    *   The database connection to run the query against.
Chris@17:    * @param array $namespaces
Chris@17:    *   List of potential namespaces of the classes belonging to this query.
Chris@17:    * @param \Drupal\workspaces\WorkspaceManagerInterface $workspace_manager
Chris@17:    *   The workspace manager.
Chris@17:    */
Chris@17:   public function __construct(EntityTypeInterface $entity_type, $conjunction, Connection $connection, array $namespaces, WorkspaceManagerInterface $workspace_manager) {
Chris@17:     parent::__construct($entity_type, $conjunction, $connection, $namespaces);
Chris@17: 
Chris@17:     $this->workspaceManager = $workspace_manager;
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * {@inheritdoc}
Chris@17:    */
Chris@17:   public function prepare() {
Chris@17:     parent::prepare();
Chris@17: 
Chris@17:     // Do not alter entity revision queries.
Chris@17:     // @todo How about queries for the latest revision? Should we alter them to
Chris@17:     //   look for the latest workspace-specific revision?
Chris@17:     if ($this->allRevisions) {
Chris@17:       return $this;
Chris@17:     }
Chris@17: 
Chris@17:     // Only alter the query if the active workspace is not the default one and
Chris@17:     // the entity type is supported.
Chris@17:     $active_workspace = $this->workspaceManager->getActiveWorkspace();
Chris@17:     if (!$active_workspace->isDefaultWorkspace() && $this->workspaceManager->isEntityTypeSupported($this->entityType)) {
Chris@17:       $this->sqlQuery->addMetaData('active_workspace_id', $active_workspace->id());
Chris@17:       $this->sqlQuery->addMetaData('simple_query', FALSE);
Chris@17: 
Chris@17:       // LEFT JOIN 'workspace_association' to the base table of the query so we
Chris@17:       // can properly include live content along with a possible workspace
Chris@17:       // revision.
Chris@17:       $id_field = $this->entityType->getKey('id');
Chris@17:       $this->sqlQuery->leftJoin('workspace_association', 'workspace_association', "%alias.target_entity_type_id = '{$this->entityTypeId}' AND %alias.target_entity_id = base_table.$id_field AND %alias.workspace = '{$active_workspace->id()}'");
Chris@17:     }
Chris@17: 
Chris@17:     return $this;
Chris@17:   }
Chris@17: 
Chris@18:   /**
Chris@18:    * {@inheritdoc}
Chris@18:    */
Chris@18:   public function isSimpleQuery() {
Chris@18:     // We declare that this is not a simple query in
Chris@18:     // \Drupal\workspaces\EntityQuery\QueryTrait::prepare(), but that's not
Chris@18:     // enough because the parent method can return TRUE in some circumstances.
Chris@18:     if ($this->sqlQuery->getMetaData('active_workspace_id')) {
Chris@18:       return FALSE;
Chris@18:     }
Chris@18: 
Chris@18:     return parent::isSimpleQuery();
Chris@18:   }
Chris@18: 
Chris@17: }