pm-4.1.x-dev/modules/pm_invoice/pm_invoice.module

<?php

/**
 * @file
 * Provides a pm invoice entity type.
 */

use Drupal\Core\Render\Element;
use Drupal\user\UserInterface;

/**
 * Implements hook_theme().
 */
function pm_invoice_theme() {
  return [
    'pm_invoice' => [
      'render element' => 'elements',
    ],
  ];
}

/**
 * Prepares variables for pm invoice templates.
 *
 * Default template: pm-invoice.html.twig.
 *
 * @param array $variables
 *   An associative array containing:
 *   - elements: An associative array containing
 *     the pm invoice information and any
 *     fields attached to the entity.
 *   - attributes: HTML attributes for the containing element.
 */
function template_preprocess_pm_invoice(array &$variables) {
  $variables['view_mode'] = $variables['elements']['#view_mode'];
  foreach (Element::children($variables['elements']) as $key) {
    $variables['content'][$key] = $variables['elements'][$key];
  }
}

/**
 * Implements hook_user_cancel().
 */
function pm_invoice_user_cancel($edit, UserInterface $account, $method) {
  switch ($method) {
    case 'user_cancel_reassign':
      // Anonymize pm invoices.
      $storage = \Drupal::entityTypeManager()->getStorage('pm_invoice');
      $pm_invoice_ids = $storage->getQuery()
        ->condition('uid', $account->id())
        ->accessCheck(FALSE)
        ->execute();
      foreach ($storage->loadMultiple($pm_invoice_ids) as $pm_invoice) {
        $pm_invoice->setOwnerId(0);
        $pm_invoice->save();
      }
      break;
  }
}

/**
 * Implements hook_ENTITY_TYPE_predelete() for user entities.
 */
function pm_invoice_user_predelete(UserInterface $account) {
  // Delete pm invoices.
  $storage = \Drupal::entityTypeManager()->getStorage('pm_invoice');
  $pm_invoice_ids = $storage->getQuery()
    ->condition('uid', $account->id())
    ->accessCheck(FALSE)
    ->execute();
  $pm_invoices = $storage->loadMultiple($pm_invoice_ids);
  $storage->delete($pm_invoices);
}