View source  
  <?php
namespace Drupal\Core\Entity;
use Drupal\Component\Plugin\Exception\InvalidPluginDefinitionException;
use Drupal\Component\Plugin\Exception\PluginNotFoundException;
use Drupal\Core\Cache\CacheBackendInterface;
use Drupal\Core\DependencyInjection\ClassResolverInterface;
use Drupal\Core\Entity\Exception\InvalidLinkTemplateException;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\Core\Plugin\DefaultPluginManager;
use Drupal\Core\Plugin\Discovery\AnnotatedClassDiscovery;
use Drupal\Core\StringTranslation\TranslationInterface;
use Symfony\Component\DependencyInjection\ContainerAwareInterface;
use Symfony\Component\DependencyInjection\ContainerAwareTrait;
class EntityTypeManager extends DefaultPluginManager implements EntityTypeManagerInterface, ContainerAwareInterface {
  use ContainerAwareTrait;
  
  protected $handlers = [];
  
  protected $stringTranslation;
  
  protected $classResolver;
  
  protected $entityLastInstalledSchemaRepository;
  
  protected $activeDefinitions;
  
  public function __construct(\Traversable $namespaces, ModuleHandlerInterface $module_handler, CacheBackendInterface $cache, TranslationInterface $string_translation, ClassResolverInterface $class_resolver, EntityLastInstalledSchemaRepositoryInterface $entity_last_installed_schema_repository) {
    parent::__construct('Entity', $namespaces, $module_handler, 'Drupal\\Core\\Entity\\EntityInterface');
    $this
      ->setCacheBackend($cache, 'entity_type', [
      'entity_types',
    ]);
    $this
      ->alterInfo('entity_type');
    $this->discovery = new AnnotatedClassDiscovery('Entity', $namespaces, 'Drupal\\Core\\Entity\\Annotation\\EntityType');
    $this->stringTranslation = $string_translation;
    $this->classResolver = $class_resolver;
    $this->entityLastInstalledSchemaRepository = $entity_last_installed_schema_repository;
  }
  
  public function processDefinition(&$definition, $plugin_id) {
    
    parent::processDefinition($definition, $plugin_id);
    
    foreach ((array) $definition
      ->getLinkTemplates() as $link_relation_name => $link_template) {
      if ($link_template[0] != '/') {
        throw new InvalidLinkTemplateException("Link template '{$link_relation_name}' for entity type '{$plugin_id}' must start with a leading slash, the current link template is '{$link_template}'");
      }
    }
  }
  
  protected function findDefinitions() {
    $definitions = $this
      ->getDiscovery()
      ->getDefinitions();
    
    foreach ($this->moduleHandler
      ->getImplementations('entity_type_build') as $module) {
      $function = $module . '_entity_type_build';
      $function($definitions);
    }
    foreach ($definitions as $plugin_id => $definition) {
      $this
        ->processDefinition($definition, $plugin_id);
    }
    $this
      ->alterDefinitions($definitions);
    return $definitions;
  }
  
  public function getDefinition($entity_type_id, $exception_on_invalid = TRUE) {
    if (($entity_type = parent::getDefinition($entity_type_id, FALSE)) && class_exists($entity_type
      ->getClass())) {
      return $entity_type;
    }
    elseif (!$exception_on_invalid) {
      return NULL;
    }
    throw new PluginNotFoundException($entity_type_id, sprintf('The "%s" entity type does not exist.', $entity_type_id));
  }
  
  public function getActiveDefinition($entity_type_id) {
    if (!isset($this->activeDefinitions[$entity_type_id])) {
      $this->activeDefinitions[$entity_type_id] = $this->entityLastInstalledSchemaRepository
        ->getLastInstalledDefinition($entity_type_id);
    }
    return $this->activeDefinitions[$entity_type_id] ?: $this
      ->getDefinition($entity_type_id);
  }
  
  public function clearCachedDefinitions() {
    parent::clearCachedDefinitions();
    $this->activeDefinitions = [];
    $this->handlers = [];
  }
  
  public function useCaches($use_caches = FALSE) {
    parent::useCaches($use_caches);
    if (!$use_caches) {
      $this->activeDefinitions = [];
      $this->handlers = [];
      $this->container
        ->get('entity.memory_cache')
        ->reset();
    }
  }
  
  public function hasHandler($entity_type_id, $handler_type) {
    if ($definition = $this
      ->getDefinition($entity_type_id, FALSE)) {
      return $definition
        ->hasHandlerClass($handler_type);
    }
    return FALSE;
  }
  
  public function getStorage($entity_type_id) {
    return $this
      ->getHandler($entity_type_id, 'storage');
  }
  
  public function getListBuilder($entity_type_id) {
    return $this
      ->getHandler($entity_type_id, 'list_builder');
  }
  
  public function getFormObject($entity_type_id, $operation) {
    if (!($class = $this
      ->getDefinition($entity_type_id, TRUE)
      ->getFormClass($operation))) {
      throw new InvalidPluginDefinitionException($entity_type_id, sprintf('The "%s" entity type did not specify a "%s" form class.', $entity_type_id, $operation));
    }
    $form_object = $this->classResolver
      ->getInstanceFromDefinition($class);
    return $form_object
      ->setStringTranslation($this->stringTranslation)
      ->setModuleHandler($this->moduleHandler)
      ->setEntityTypeManager($this)
      ->setOperation($operation);
  }
  
  public function getRouteProviders($entity_type_id) {
    if (!isset($this->handlers['route_provider'][$entity_type_id])) {
      $route_provider_classes = $this
        ->getDefinition($entity_type_id, TRUE)
        ->getRouteProviderClasses();
      foreach ($route_provider_classes as $type => $class) {
        $this->handlers['route_provider'][$entity_type_id][$type] = $this
          ->createHandlerInstance($class, $this
          ->getDefinition($entity_type_id));
      }
    }
    return isset($this->handlers['route_provider'][$entity_type_id]) ? $this->handlers['route_provider'][$entity_type_id] : [];
  }
  
  public function getViewBuilder($entity_type_id) {
    return $this
      ->getHandler($entity_type_id, 'view_builder');
  }
  
  public function getAccessControlHandler($entity_type_id) {
    return $this
      ->getHandler($entity_type_id, 'access');
  }
  
  public function getHandler($entity_type_id, $handler_type) {
    if (!isset($this->handlers[$handler_type][$entity_type_id])) {
      $definition = $this
        ->getDefinition($entity_type_id);
      $class = $definition
        ->getHandlerClass($handler_type);
      if (!$class) {
        throw new InvalidPluginDefinitionException($entity_type_id, sprintf('The "%s" entity type did not specify a %s handler.', $entity_type_id, $handler_type));
      }
      $this->handlers[$handler_type][$entity_type_id] = $this
        ->createHandlerInstance($class, $definition);
    }
    return $this->handlers[$handler_type][$entity_type_id];
  }
  
  public function createHandlerInstance($class, EntityTypeInterface $definition = NULL) {
    if (is_subclass_of($class, 'Drupal\\Core\\Entity\\EntityHandlerInterface')) {
      $handler = $class::createInstance($this->container, $definition);
    }
    else {
      $handler = new $class($definition);
    }
    if (method_exists($handler, 'setModuleHandler')) {
      $handler
        ->setModuleHandler($this->moduleHandler);
    }
    if (method_exists($handler, 'setStringTranslation')) {
      $handler
        ->setStringTranslation($this->stringTranslation);
    }
    return $handler;
  }
}