You are here

user_reference.usermerge.inc in User Merge 7.2

Adds support for User Reference (References). Supplemental include loaded via usermerge_load_includes().

File

includes/user_reference.usermerge.inc
View source
<?php

/**
 * @file
 * Adds support for User Reference (References).
 * Supplemental include loaded via usermerge_load_includes().
 */

/**
 * Implements hook_usermerge_build_review_form_elements().
 */
function user_reference_usermerge_build_review_form_elements($review, $account_properties, $user_to_delete, $user_to_keep) {
  $entities = _user_reference_usermerge_get_referencing_entity_types();
  $user_references_user_to_delete = array();
  $user_references_user_to_keep = array();
  foreach ($entities as $entity_type => $fields) {
    foreach ($fields as $field) {
      $query = new EntityFieldQuery();
      $query
        ->entityCondition('entity_type', $entity_type)
        ->fieldCondition($field['field_name'], 'uid', $user_to_delete->uid);
      $result = $query
        ->execute();
      if (count($result)) {
        if (isset($user_references_user_to_delete[$entity_type])) {
          $user_references_user_to_delete[$entity_type] = $user_references_user_to_delete[$entity_type] + $result[$entity_type];
        }
        else {
          $user_references_user_to_delete[$entity_type] = $result[$entity_type];
        }
      }
      $query = new EntityFieldQuery();
      $query
        ->entityCondition('entity_type', $entity_type)
        ->fieldCondition($field['field_name'], 'uid', $user_to_keep->uid);
      $result = $query
        ->execute();
      if (count($result)) {
        if (isset($user_references_user_to_keep[$entity_type])) {
          $user_references_user_to_keep[$entity_type] = $user_references_user_to_keep[$entity_type] + $result[$entity_type];
        }
        else {
          $user_references_user_to_keep[$entity_type] = $result[$entity_type];
        }
      }
    }
  }
  if (count($user_references_user_to_delete) || count($user_references_user_to_keep)) {
    $review['user_reference'] = array(
      '#tree' => TRUE,
      '#theme' => 'usermerge_data_review_form_table',
      '#title' => t('Referencing entities (!module)', array(
        '!module' => t('References'),
      )),
      '#attributes' => array(
        'no_merge',
        'property_label' => t('Entity'),
      ),
    );
    foreach ($user_references_user_to_delete as $entity_type => $entity_list) {
      $review['user_reference'][$entity_type]['property_name'] = array(
        '#type' => 'markup',
        '#markup' => $entity_type,
      );
      $review['user_reference'][$entity_type]['options']['user_to_delete'] = array(
        '#type' => 'markup',
        '#markup' => format_plural(count($entity_list), '1 %entity entity to be reassigned', '@count %entity entities to be reassigned', array(
          '%entity' => $entity_type,
        )),
      );
      $review['user_reference'][$entity_type]['fields'] = array(
        '#type' => 'hidden',
        '#value' => serialize($entities[$entity_type]),
      );
    }
    foreach ($user_references_user_to_keep as $entity_type => $entity_list) {
      $review['user_reference'][$entity_type]['property_name'] = array(
        '#type' => 'markup',
        '#markup' => $entity_type,
      );
      $review['user_reference'][$entity_type]['options']['user_to_keep'] = array(
        '#type' => 'markup',
        '#markup' => format_plural(count($entity_list), '1 %entity entity to be maintained', '@count %entity entities to be maintained', array(
          '%entity' => $entity_type,
        )),
      );
    }
    ksort($review['user_reference']);
    return $review;
  }
}

/**
 * Implements hook_usermerge_actions_supported().
 */
function user_reference_usermerge_actions_supported($items) {
  return array(
    'user_reference' => t('Assigning to the new user any entities (such as nodes) referencing the old user using the References (User Reference) module.'),
  );
}

/**
 * Implements hook_usermerge_merge_accounts().
 */
function user_reference_usermerge_merge_accounts($user_to_delete, $user_to_keep, $review) {
  if (isset($review['user_reference'])) {
    $fields = array();
    foreach ($review['user_reference'] as $entity_type => $entity_properties) {
      $fields = array_merge($fields, unserialize($entity_properties['fields']));
    }
    if (count($fields)) {
      foreach ($fields as $field_name => $field_properties) {
        $tables = array(
          'data' => 'field_data_' . $field_name,
          'revision' => 'field_revision_' . $field_name,
        );
        foreach ($tables as $table_name) {
          $query = db_update($table_name)
            ->fields(array(
            $field_name . '_uid' => $user_to_keep->uid,
          ))
            ->condition($field_name . '_uid', $user_to_delete->uid)
            ->execute();
        }
      }
    }
  }
}

/**
 * Selects entities that reference users via the References module.
 */
function _user_reference_usermerge_get_referencing_entity_types() {
  $fields = field_info_fields();
  $referencing_entity_types = array();
  foreach ($fields as $field_name => $field) {
    if ($field['type'] == 'user_reference') {
      foreach ($field['bundles'] as $entity_type => $bundles) {
        $referencing_entity_types[$entity_type][$field_name] = $field;
      }
    }
  }
  return $referencing_entity_types;
}