EntityUsage.php in Entity Usage 8.3

Namespace

Drupal\entity_usage

File

src/EntityUsage.php

<?php

namespace Drupal\entity_usage;

use Drupal\Core\Config\ConfigFactoryInterface;
use Drupal\Core\Database\Connection;
use Drupal\Core\Database\IntegrityConstraintViolationException;
use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\entity_usage\Events\Events;
use Drupal\entity_usage\Events\EntityUsageEvent;
use Symfony\Component\EventDispatcher\EventDispatcherInterface;

/**
 * Defines the entity usage base class.
 */
class EntityUsage implements EntityUsageInterface {

  /**
   * The database connection used to store entity usage information.
   *
   * @var \Drupal\Core\Database\Connection
   */
  protected $connection;

  /**
   * The name of the SQL table used to store entity usage information.
   *
   * @var string
   */
  protected $tableName;

  /**
   * An event dispatcher instance.
   *
   * @var \Symfony\Component\EventDispatcher\EventDispatcherInterface
   */
  protected $eventDispatcher;

  /**
   * The config factory.
   *
   * @var \Drupal\Core\Config\ImmutableConfig
   */
  protected $config;

  /**
   * The ModuleHandler service.
   *
   * @var \Drupal\Core\Extension\ModuleHandlerInterface
   */
  protected $moduleHandler;

  /**
   * Construct the EntityUsage object.
   *
   * @param \Drupal\Core\Database\Connection $connection
   *   The database connection which will be used to store the entity usage
   *   information.
   * @param \Symfony\Component\EventDispatcher\EventDispatcherInterface $event_dispatcher
   *   An event dispatcher instance to use for events.
   * @param \Drupal\Core\Config\ConfigFactoryInterface $config_factory
   *   The config factory.
   * @param \Drupal\Core\Extension\ModuleHandlerInterface $module_handler
   *   The ModuleHandler service.
   * @param string $table
   *   (optional) The table to store the entity usage info. Defaults to
   *   'entity_usage'.
   */
  public function __construct(Connection $connection, EventDispatcherInterface $event_dispatcher, ConfigFactoryInterface $config_factory, ModuleHandlerInterface $module_handler, $table = 'entity_usage') {
    $this->connection = $connection;
    $this->tableName = $table;
    $this->eventDispatcher = $event_dispatcher;
    $this->moduleHandler = $module_handler;
    $this->config = $config_factory
      ->get('entity_usage.settings');
  }

  /**
   * {@inheritdoc}
   */
  public function registerUsage($target_id, $target_type, $source_id, $source_type, $source_langcode, $source_vid) {

    // Check if target entity type is enabled, all entity types are enabled by
    // default.
    $enabled_target_entity_types = $this->config
      ->get('track_enabled_target_entity_types');
    if (is_array($enabled_target_entity_types) && !in_array($target_type, $enabled_target_entity_types, TRUE)) {
      return;
    }

    // Allow modules to block this operation.
    $context = [
      'target_id' => $target_id,
      'target_type' => $target_type,
      'source_id' => $source_id,
      'source_type' => $source_type,
      'source_langcode' => $source_langcode,
      'source_vid' => $source_vid,
    ];
    $abort = $this->moduleHandler
      ->invokeAll('entity_usage_block_tracking', $context);

    // If at least one module wants to block the tracking, bail out.
    if (in_array(TRUE, $abort, TRUE)) {
      return;
    }

    // Entities can have string IDs. We support that by using different columns
    // on each case.
    $target_id_column = $this
      ->isInt($target_id) ? 'target_id' : 'target_id_string';
    $source_id_column = $this
      ->isInt($source_id) ? 'source_id' : 'source_id_string';

    // Since this tries to blindly write a new record, do nothing if the record
    // already exists. We could probably avoid the try/catch if #2922755 lands.
    try {
      $this->connection
        ->insert($this->tableName)
        ->fields([
        $target_id_column => $target_id,
        'target_type' => $target_type,
        $source_id_column => $source_id,
        'source_type' => $source_type,
        'source_langcode' => $source_langcode,
        'source_vid' => $source_vid ?: 0,
      ])
        ->execute();
    } catch (IntegrityConstraintViolationException $e) {
      if (strpos($e
        ->getMessage(), 'Duplicate entry') !== FALSE) {
        return;
      }
      throw $e;
    }
    $event = new EntityUsageEvent($target_id, $target_type, $source_id, $source_type, $source_langcode, $source_vid);
    $this->eventDispatcher
      ->dispatch(Events::USAGE_REGISTER, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function deleteUsage($target_id, $target_type, $source_id, $source_type, $source_langcode, $source_vid) {
    $target_id_column = $this
      ->isInt($target_id) ? 'target_id' : 'target_id_string';
    $source_id_column = $this
      ->isInt($source_id) ? 'source_id' : 'source_id_string';
    $query = $this->connection
      ->delete($this->tableName)
      ->condition('target_type', $target_type)
      ->condition($target_id_column, $target_id)
      ->condition('source_type', $source_type)
      ->condition($source_id_column, $source_id)
      ->condition('source_langcode', $source_langcode)
      ->condition('source_vid', $source_vid);
    $query
      ->execute();
    $event = new EntityUsageEvent($target_id, $target_type, $source_id, $source_type, $source_langcode, $source_vid);
    $this->eventDispatcher
      ->dispatch(Events::BULK_DELETE_SOURCES, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function bulkDeleteTargets($target_type) {
    $query = $this->connection
      ->delete($this->tableName)
      ->condition('target_type', $target_type);
    $query
      ->execute();
    $event = new EntityUsageEvent(NULL, $target_type, NULL, NULL, NULL, NULL);
    $this->eventDispatcher
      ->dispatch(Events::BULK_DELETE_DESTINATIONS, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function bulkDeleteSources($source_type) {
    $query = $this->connection
      ->delete($this->tableName)
      ->condition('source_type', $source_type);
    $query
      ->execute();
    $event = new EntityUsageEvent(NULL, NULL, NULL, $source_type, NULL, NULL);
    $this->eventDispatcher
      ->dispatch(Events::BULK_DELETE_SOURCES, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function deleteBySourceEntity($source_id, $source_type, $source_langcode = NULL, $source_vid = NULL) {

    // Entities can have string IDs. We support that by using different columns
    // on each case.
    $source_id_column = $this
      ->isInt($source_id) ? 'source_id' : 'source_id_string';
    $query = $this->connection
      ->delete($this->tableName)
      ->condition($source_id_column, $source_id)
      ->condition('source_type', $source_type);
    if ($source_langcode) {
      $query
        ->condition('source_langcode', $source_langcode);
    }
    if ($source_vid) {
      $query
        ->condition('source_vid', $source_vid);
    }
    $query
      ->execute();
    $event = new EntityUsageEvent(NULL, NULL, $source_id, $source_type, $source_langcode, $source_vid);
    $this->eventDispatcher
      ->dispatch(Events::DELETE_BY_SOURCE_ENTITY, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function deleteByTargetEntity($target_id, $target_type) {

    // Entities can have string IDs. We support that by using different columns
    // on each case.
    $target_id_column = $this
      ->isInt($target_id) ? 'target_id' : 'target_id_string';
    $query = $this->connection
      ->delete($this->tableName)
      ->condition($target_id_column, $target_id)
      ->condition('target_type', $target_type);
    $query
      ->execute();
    $event = new EntityUsageEvent($target_id, $target_type, NULL, NULL, NULL, NULL);
    $this->eventDispatcher
      ->dispatch(Events::DELETE_BY_TARGET_ENTITY, $event);
  }

  /**
   * {@inheritdoc}
   */
  public function listSources(EntityInterface $target_entity, $nest_results = TRUE) {

    // Entities can have string IDs. We support that by using different columns
    // on each case.
    $target_id_column = $this
      ->isInt($target_entity
      ->id()) ? 'target_id' : 'target_id_string';
    $result = $this->connection
      ->select($this->tableName, 'e')
      ->fields('e', [
      'source_id',
      'source_id_string',
      'source_type',
      'source_langcode',
      'source_vid',
    ])
      ->condition($target_id_column, $target_entity
      ->id())
      ->condition('target_type', $target_entity
      ->getEntityTypeId())
      ->orderBy('source_type')
      ->orderBy('source_id', 'DESC')
      ->orderBy('source_vid', 'DESC')
      ->orderBy('source_langcode')
      ->execute();
    $references = [];
    foreach ($result as $usage) {
      $source_id_value = !empty($usage->source_id) ? (string) $usage->source_id : (string) $usage->source_id_string;
      if ($nest_results) {
        $references[$usage->source_type][$source_id_value][] = [
          'source_langcode' => $usage->source_langcode,
          'source_vid' => $usage->source_vid,
        ];
      }
      else {
        $references[] = [
          'source_type' => $usage->source_type,
          'source_id' => $source_id_value,
          'source_langcode' => $usage->source_langcode,
          'source_vid' => $usage->source_vid,
        ];
      }
    }
    return $references;
  }

  /**
   * {@inheritdoc}
   */
  public function listTargets(EntityInterface $source_entity) {

    // Entities can have string IDs. We support that by using different columns
    // on each case.
    $source_id_column = $this
      ->isInt($source_entity
      ->id()) ? 'source_id' : 'source_id_string';
    $result = $this->connection
      ->select($this->tableName, 'e')
      ->fields('e', [
      'target_id',
      'target_id_string',
      'target_type',
    ])
      ->condition($source_id_column, $source_entity
      ->id())
      ->condition('source_type', $source_entity
      ->getEntityTypeId())
      ->orderBy('target_id', 'DESC')
      ->execute();
    $references = [];
    foreach ($result as $usage) {
      $target_id_value = !empty($usage->target_id) ? $usage->target_id : $usage->target_id_string;
      $references[$usage->target_type][] = $target_id_value;
    }
    return $references;
  }

  /**
   * Check if a value is an integer, or an integer string.
   *
   * Core doesn't support big integers (bigint) for entity reference fields.
   * Therefore we consider integers with more than 10 digits (big integer) to be
   * strings.
   * @todo: Fix bigint support once fixed in core. More info on #2680571 and
   * #2989033.
   *
   * @param int|string $value
   *   The value to check.
   *
   * @return bool
   *   TRUE if the value is a numeric integer or a string containing an integer,
   *   FALSE otherwise.
   */
  protected function isInt($value) {
    return (string) (int) $value === (string) $value && strlen($value) < 11;
  }

}

Classes

Name	Description
EntityUsage	Defines the entity usage base class.