WorkspacesAliasRepository.php

<?php

namespace Drupal\workspaces;

use Drupal\path_alias\AliasRepository;

/**
 * Provides workspace-specific path alias lookup queries.
 */
class WorkspacesAliasRepository extends AliasRepository {
    
    /**
     * The workspace manager.
     *
     * @var \Drupal\workspaces\WorkspaceManagerInterface
     */
    protected $workspaceManager;
    
    /**
     * Sets the workspace manager.
     *
     * @param \Drupal\workspaces\WorkspaceManagerInterface $workspace_manager
     *   The workspace manager service.
     *
     * @return $this
     */
    public function setWorkspacesManager(WorkspaceManagerInterface $workspace_manager) {
        $this->workspaceManager = $workspace_manager;
        return $this;
    }
    
    /**
     * {@inheritdoc}
     */
    protected function getBaseQuery() {
        // Don't alter any queries if we're not in a workspace context.
        if (!$this->workspaceManager
            ->hasActiveWorkspace()) {
            return parent::getBaseQuery();
        }
        $active_workspace = $this->workspaceManager
            ->getActiveWorkspace();
        $query = $this->connection
            ->select('path_alias', 'base_table_2');
        $wa_join = $query->leftJoin('workspace_association', NULL, "[%alias].[target_entity_type_id] = 'path_alias' AND [%alias].[target_entity_id] = [base_table_2].[id] AND [%alias].[workspace] = :active_workspace_id", [
            ':active_workspace_id' => $active_workspace->id(),
        ]);
        $query->innerJoin('path_alias_revision', 'base_table', "[%alias].[revision_id] = COALESCE([{$wa_join}].[target_entity_revision_id], [base_table_2].[revision_id])");
        return $query;
    }

}