form_mode_manager-8.x-1.x-dev/src/MediaEntityFormModes.php

<?php

namespace Drupal\form_mode_manager;

use Drupal\Core\Routing\RouteMatchInterface;

/**
 * Route controller factory specific for Media (core) entities.
 *
 * This Factory are inherited from ComplexEntityFormModes because this entity,
 * implement some specific things proper to Media like 'type' => 'bundle'.
 */
class MediaEntityFormModes extends ComplexEntityFormModes {

  /**
   * {@inheritdoc}
   */
  public function getEntityFromRouteMatch(RouteMatchInterface $route_match) {
    $entity_type_id = $route_match->getRouteObject()
      ->getOption('_form_mode_manager_entity_type_id');
    $bundle_entity_type_id = $route_match->getRouteObject()
      ->getOption('_form_mode_manager_bundle_entity_type_id');
    $entity = $route_match->getParameter($entity_type_id);

    if (empty($entity)) {
      $entity = $this->entityTypeManager->getStorage($entity_type_id)->create([
        'bundle' => $route_match->getRawParameter($bundle_entity_type_id),
      ]);
    }

    return $entity;
  }

}