class EntityTypeInfo

class EntityTypeInfo {
  public function __construct(protected WorkspaceInformationInterface $workspaceInfo) {
  }
  
  /**
   * Implements hook_entity_type_build().
   *
   * Adds workspace support info to eligible entity types.
   */
  public function entityTypeBuild(array &$entity_types) : void {
    foreach ($entity_types as $entity_type) {
      if ($entity_type->hasHandlerClass('workspace')) {
        continue;
      }
      // Revisionable and publishable entity types are always supported.
      if ($entity_type->entityClassImplements(EntityPublishedInterface::class) && $entity_type->isRevisionable()) {
        $entity_type->setHandlerClass('workspace', DefaultWorkspaceHandler::class);
        // Support for custom blocks has to be determined on a per-entity
        // basis.
        if ($entity_type->id() === 'block_content') {
          $entity_type->setHandlerClass('workspace', BlockContentWorkspaceHandler::class);
        }
      }
      // The 'file' entity type is allowed to perform CRUD operations inside a
      // workspace without being tracked.
      if ($entity_type->id() === 'file') {
        $entity_type->setHandlerClass('workspace', IgnoredWorkspaceHandler::class);
      }
      // Internal entity types are allowed to perform CRUD operations inside a
      // workspace.
      if ($entity_type->isInternal()) {
        $entity_type->setHandlerClass('workspace', IgnoredWorkspaceHandler::class);
      }
    }
  }
  
  /**
   * Implements hook_entity_type_alter().
   *
   * Adds workspace configuration to appropriate entity types.
   */
  public function entityTypeAlter(array &$entity_types) : void {
    foreach ($entity_types as $entity_type) {
      if (!$this->workspaceInfo
        ->isEntityTypeSupported($entity_type)) {
        continue;
      }
      // Workspace-support status has been declared in the "build" phase, now we
      // can use that information and add additional configuration in the
      // "alter" phase.
      $entity_type->addConstraint('EntityWorkspaceConflict');
      $entity_type->setRevisionMetadataKey('workspace', 'workspace');
      // Non-default workspaces display the active revision on the canonical
      // route of an entity, so the latest version route is no longer needed.
      $link_templates = $entity_type->get('links');
      unset($link_templates['latest-version']);
      $entity_type->set('links', $link_templates);
    }
  }
  
  /**
   * Implements hook_field_info_alter().
   */
  public function fieldInfoAlter(array &$definitions) : void {
    if (isset($definitions['entity_reference'])) {
      $definitions['entity_reference']['constraints']['EntityReferenceSupportedNewEntities'] = [];
    }
    // Allow path aliases to be changed in workspace-specific pending revisions.
    if (isset($definitions['path'])) {
      unset($definitions['path']['constraints']['PathAlias']);
    }
  }
  
  /**
   * Implements hook_entity_base_field_info().
   */
  public function entityBaseFieldInfo(EntityTypeInterface $entity_type) : array {
    if ($this->workspaceInfo
      ->isEntityTypeSupported($entity_type)) {
      $field_name = $entity_type->getRevisionMetadataKey('workspace');
      $fields[$field_name] = BaseFieldDefinition::create('entity_reference')->setLabel(new TranslatableMarkup('Workspace'))
        ->setDescription(new TranslatableMarkup('Indicates the workspace that this revision belongs to.'))
        ->setSetting('target_type', 'workspace')
        ->setInternal(TRUE)
        ->setTranslatable(FALSE)
        ->setRevisionable(TRUE);
      return $fields;
    }
    return [];
  }

}