protected function Entity::getTypeSchema in Schema 8

1 call to Entity::getTypeSchema()

Entity::get in src/Plugin/Schema/Entity.php

File

src/Plugin/Schema/Entity.php, line 43

Contains Drupal\schema\Plugin\Schema\Entity.

Class

Entity

Provides schema information related to entities and fields.

Namespace

Drupal\schema\Plugin\Schema

Code

protected function getTypeSchema(EntityTypeInterface $entity_type, $rebuild) {
  $storage = $this->entityManager
    ->getStorage($entity_type
    ->id());

  // We need to build the entity schema from entity base tables as well as
  // shared and dedicated field tables. Too bad core doesn't this for us. :-/
  $entity_schema = array();
  if ($entity_type instanceof ContentEntityTypeInterface && $storage instanceof SqlContentEntityStorage) {
    $class = $entity_type
      ->getHandlerClass('storage_schema') ?: 'Drupal\\Core\\Entity\\Sql\\SqlContentEntityStorageSchema';
    $storage_schema = new $class($this->entityManager, $entity_type, $storage, \Drupal::database());
    $table_mapping = $storage
      ->getTableMapping();

    // Use reflection to access protected methods.
    $reflection = new ReflectionClass($class);
    $entity_schema_method = $reflection
      ->getMethod('getEntitySchema');
    $entity_schema_method
      ->setAccessible(TRUE);
    $entity_schema += $entity_schema_method
      ->invoke($storage_schema, $entity_type, $rebuild);

    // Add schema for shared field tables
    $field_types = $this->entityManager
      ->getFieldStorageDefinitions($entity_type
      ->id());
    foreach ($field_types as $field_storage) {
      if ($table_mapping
        ->requiresDedicatedTableStorage($field_storage)) {
        $entity_schema_method = $reflection
          ->getMethod('getDedicatedTableSchema');
        $entity_schema_method
          ->setAccessible(TRUE);
        $entity_schema += $entity_schema_method
          ->invoke($storage_schema, $field_storage);
      }
      elseif ($table_mapping
        ->allowsSharedTableStorage($field_storage)) {
        $field_name = $field_storage
          ->getName();
        foreach (array_diff($table_mapping
          ->getTableNames(), $table_mapping
          ->getDedicatedTableNames()) as $table_name) {
          if (in_array($field_name, $table_mapping
            ->getFieldNames($table_name))) {
            $column_names = $table_mapping
              ->getColumnNames($field_name);
            $entity_schema_method = $reflection
              ->getMethod('getSharedTableFieldSchema');
            $entity_schema_method
              ->setAccessible(TRUE);
            $entity_schema[$table_name] += $entity_schema_method
              ->invoke($storage_schema, $field_storage, $table_name, $column_names);
          }
        }
      }
      else {

        // Field uses custom storage which we cannot determine automatically.
        // @todo Is this correct?
      }
    }
  }

  // Add module information, if it doesn't exist.
  foreach ($entity_schema as $name => &$table) {
    if (!isset($table['module'])) {
      $table['module'] = $entity_type
        ->getProvider();
    }
  }
  return $entity_schema;
}