WorkspacesLayoutTempstoreRepository.php

<?php

declare (strict_types=1);
namespace Drupal\workspaces;

use Drupal\layout_builder\LayoutTempstoreRepository;
use Drupal\layout_builder\SectionStorageInterface;

/**
 * Provides a mechanism for loading workspace-specific layout changes.
 */
class WorkspacesLayoutTempstoreRepository extends LayoutTempstoreRepository {
    
    /**
     * The workspace manager.
     */
    protected WorkspaceManagerInterface $workspaceManager;
    
    /**
     * Sets the workspace manager.
     *
     * @param \Drupal\workspaces\WorkspaceManagerInterface $workspace_manager
     *   The workspace manager service.
     */
    public function setWorkspacesManager(WorkspaceManagerInterface $workspace_manager) : static {
        $this->workspaceManager = $workspace_manager;
        return $this;
    }
    
    /**
     * {@inheritdoc}
     */
    protected function getKey(SectionStorageInterface $section_storage) : string {
        $key = parent::getKey($section_storage);
        // Suffix the layout tempstore key with a workspace ID when one is active.
        if ($this->workspaceManager
            ->hasActiveWorkspace()) {
            $key .= '.workspace:' . $this->workspaceManager
                ->getActiveWorkspace()
                ->id();
        }
        return $key;
    }

}