function download_count_file_access in Download Count 8

Implements hook_file_access().

File

./download_count.module, line 29

Tracks file downloads for files stored in the drupal files table.

Code

function download_count_file_access($entity, $operation, $account) {
  if ($operation == 'download') {
    $flood = Drupal::flood();
    $config = Drupal::config('download_count.settings');
    $flood_window = $config
      ->get('download_count_flood_window');
    $time = REQUEST_TIME;

    // If ($account->hasPermission('skip download counts')){
    // return;
    // }
    // check flood control.
    $flood_limit = $config
      ->get('download_count_flood_limit');
    if ($flood_limit > 0) {
      if (!$flood
        ->isAllowed('download_count-fid_' . $entity
        ->id(), $flood_limit, $flood_window)) {
        return;
      }
    }

    // Validate file has extension that should be counted, if not return.
    $extensions = explode(' ', Unicode::strtolower(trim($config
      ->get('download_count_excluded_file_extensions'))));
    if (count($extensions)) {
      $extension = Unicode::strtolower(pathinfo($entity
        ->getFilename(), PATHINFO_EXTENSION));
      if (in_array($extension, $extensions)) {
        return;
      }
    }

    // Count the download.
    $ip = Drupal::request()
      ->getClientIp();
    $referrer = isset($_SERVER['HTTP_REFERER']) ? $_SERVER['HTTP_REFERER'] : t('Direct download');

    // Echo "<pre>";print_R($entity);exit;
    $references = file_get_file_references($entity, NULL, EntityStorageInterface::FIELD_LOAD_REVISION, NULL);
    foreach ($references as $entity_map) {
      foreach ($entity_map as $referencing_entities) {
        foreach ($referencing_entities as $referencing_entity) {
          $entity_type = $referencing_entity
            ->getEntityTypeId();
          $entity_id = $referencing_entity
            ->id();
        }
      }
    }
    $connection = Database::getConnection();
    $connection
      ->insert('download_count')
      ->fields([
      'fid' => $entity
        ->id(),
      'uid' => $account
        ->id(),
      'type' => $entity_type,
      'id' => $entity_id,
      'ip_address' => $ip,
      'referrer' => $referrer,
      'timestamp' => $time,
    ])
      ->execute();
    $flood
      ->register('download_count-fid_' . $entity
      ->id(), $flood_window);
    Drupal::logger('download_count')
      ->notice('%file was downloaded by user #%uid from %ip', [
      '%file' => $entity
        ->getFilename(),
      '%uid' => $account
        ->id(),
      '%ip' => $ip,
    ]);
    if (Drupal::moduleHandler()
      ->moduleExists('rules')) {
      $host_entity = Drupal::entityTypeManager()
        ->getStorage($entity_type)
        ->load($entity_id);
      if (function_exists('rules_invoke_event')) {
        rules_invoke_event('download_count_file_download', $entity, $account, $host_entity);
      }
    }
  }
}