class EntityDisplayRebuilder

class EntityDisplayRebuilder implements ContainerInjectionInterface {
  
  /**
   * Constructs a new EntityDisplayRebuilder.
   *
   * @param \Drupal\Core\Entity\EntityTypeManagerInterface $entityTypeManager
   *   The entity manager.
   * @param \Drupal\Core\Entity\EntityDisplayRepositoryInterface $entityDisplayRepository
   *   The entity display repository.
   * @param \Drupal\Core\Entity\EntityTypeBundleInfoInterface $entityTypeBundleInfo
   *   The entity type bundle info.
   */
  public function __construct(protected EntityTypeManagerInterface $entityTypeManager, protected EntityDisplayRepositoryInterface $entityDisplayRepository, protected EntityTypeBundleInfoInterface $entityTypeBundleInfo) {
  }
  
  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container) {
    return new static($container->get('entity_type.manager'), $container->get('entity_display.repository'), $container->get('entity_type.bundle.info'));
  }
  
  /**
   * Rebuild displays for single Entity Type.
   *
   * @param string $entity_type_id
   *   The entity type machine name.
   * @param string $bundle
   *   The bundle we need to rebuild.
   */
  public function rebuildEntityTypeDisplays($entity_type_id, $bundle) {
    // Get the displays.
    $view_modes = $this->entityDisplayRepository
      ->getViewModeOptions($entity_type_id);
    $form_modes = $this->entityDisplayRepository
      ->getFormModeOptions($entity_type_id);
    // Save view mode displays.
    $view_mode_ids = array_map(function ($view_mode) use ($entity_type_id, $bundle) {
      return "{$entity_type_id}.{$bundle}.{$view_mode}";
    }, array_keys($view_modes));
    foreach ($this->entityTypeManager
      ->getStorage('entity_view_display')
      ->loadMultiple($view_mode_ids) as $display) {
      $display->save();
    }
    // Save form mode displays.
    $form_mode_ids = array_map(function ($form_mode) use ($entity_type_id, $bundle) {
      return "{$entity_type_id}.{$bundle}.{$form_mode}";
    }, array_keys($form_modes));
    foreach ($this->entityTypeManager
      ->getStorage('entity_form_display')
      ->loadMultiple($form_mode_ids) as $display) {
      $display->save();
    }
  }

}