View source
<?php
define('FEEDS_BLOCK_NON_EXISTENT', 'block');
class FeedsUserProcessor extends FeedsProcessor {
const ROLE_SEARCH_NAME = 'name';
const ROLE_SEARCH_RID = 'rid';
const PASS_UNENCRYPTED = 'none';
const PASS_MD5 = 'md5';
const PASS_SHA512 = 'sha512';
public function entityType() {
return 'user';
}
protected function entityInfo() {
$info = parent::entityInfo();
$info['label plural'] = t('Users');
return $info;
}
protected function newEntity(FeedsSource $source) {
$account = parent::newEntity($source);
$account->uid = 0;
$account->roles = array_filter($this->config['roles']);
$account->status = $this->config['status'];
$account->is_new = TRUE;
return $account;
}
protected function entityLoad(FeedsSource $source, $uid) {
$user = parent::entityLoad($source, $uid);
$user->feeds_original_pass = $user->pass;
if ($this->config['update_existing'] == FEEDS_REPLACE_EXISTING) {
$user->roles = array_filter($this->config['roles']);
$user->status = $this->config['status'];
if (!empty($user->data) && @unserialize($user->data)) {
$user->data = unserialize($user->data);
}
else {
$user->data = array();
}
}
return $user;
}
protected function entityValidate($account, FeedsSource $source = NULL) {
parent::entityValidate($account);
if (empty($account->name) || empty($account->mail) || !valid_email_address($account->mail)) {
throw new FeedsValidationException(t('User name missing or email not valid.'));
}
$uid_is_set = FALSE;
if (!empty($account->uid)) {
$is_new = !empty($account->feeds_item->is_new);
$different = !empty($account->feeds_item->entity_id) && $account->feeds_item->entity_id != $account->uid;
if ($is_new || $different) {
$uid_is_set = TRUE;
$exists = entity_load_unchanged('user', $account->uid);
if ($exists) {
throw new FeedsValidationException(t('Could not update user ID to @uid since that ID is already in use.', array(
'@uid' => $account->uid,
)));
}
}
}
$errors = array();
if ($error = user_validate_name($account->name)) {
$errors[] = $error;
}
if (!$uid_is_set && (bool) db_select('users')
->fields('users', array(
'uid',
))
->condition('uid', $account->uid, '<>')
->condition('name', db_like($account->name), 'LIKE')
->range(0, 1)
->execute()
->fetchField()) {
$errors[] = t("The name '@name' is already taken.", array(
'@name' => $account->name,
));
}
if (!$uid_is_set && (bool) db_select('users')
->fields('users', array(
'uid',
))
->condition('uid', $account->uid, '<>')
->condition('mail', db_like($account->mail), 'LIKE')
->range(0, 1)
->execute()
->fetchField()) {
$errors[] = t("The e-mail address '@email' is already taken.", array(
'@email' => $account->mail,
));
}
if (!empty($account->timezone) && !array_key_exists($account->timezone, system_time_zones())) {
$errors[] = t("Failed importing '@name'. User's timezone is not valid.", array(
'@name' => $account->name,
));
}
if (!empty($errors)) {
throw new FeedsValidationException(implode("\n", $errors));
}
}
protected function entitySave($account) {
if ($this->config['defuse_mail']) {
$account->mail = $account->mail . '_test';
}
$edit = (array) $account;
if (isset($account->feeds_original_pass) && $account->pass == $account->feeds_original_pass) {
unset($edit['pass']);
}
if (!empty($account->feeds_item->entity_id) && $account->feeds_item->entity_id != $account->uid) {
db_update('users')
->fields(array(
'uid' => $account->uid,
))
->condition('uid', $account->feeds_item->entity_id)
->execute();
}
user_save($account, $edit);
if ($account->uid && !empty($account->pass_crypted)) {
db_update('users')
->fields(array(
'pass' => $account->pass_crypted,
))
->condition('uid', $account->uid)
->execute();
}
if ($account->uid && !empty($account->openid)) {
$authmap = array(
'uid' => $account->uid,
'module' => 'openid',
'authname' => $account->openid,
);
if (SAVED_UPDATED != drupal_write_record('authmap', $authmap, array(
'uid',
'module',
))) {
drupal_write_record('authmap', $authmap);
}
}
}
protected function entityDeleteMultiple($uids) {
if (in_array(1, $uids)) {
$uids = array_diff($uids, array(
1,
));
db_delete('feeds_item')
->condition('entity_type', $this
->entityType())
->condition('entity_id', 1)
->execute();
}
user_delete_multiple($uids);
}
public function configDefaults() {
return array(
'roles' => array(),
'status' => 1,
'defuse_mail' => FALSE,
) + parent::configDefaults();
}
public function configForm(&$form_state) {
$form = parent::configForm($form_state);
$form['status'] = array(
'#type' => 'radios',
'#title' => t('Status'),
'#description' => t('Select whether users should be imported active or blocked.'),
'#options' => array(
0 => t('Blocked'),
1 => t('Active'),
),
'#default_value' => $this->config['status'],
);
$roles = user_roles(TRUE);
unset($roles[2]);
if (count($roles)) {
$form['roles'] = array(
'#type' => 'checkboxes',
'#title' => t('Additional roles'),
'#description' => t('Every user is assigned the "authenticated user" role. Select additional roles here.'),
'#default_value' => $this->config['roles'],
'#options' => $roles,
);
}
$form['defuse_mail'] = array(
'#type' => 'checkbox',
'#title' => t('Defuse e-mail addresses'),
'#description' => t('This appends _test to all imported e-mail addresses to ensure they cannot be used as recipients.'),
'#default_value' => $this->config['defuse_mail'],
);
$form['update_non_existent']['#options'][FEEDS_BLOCK_NON_EXISTENT] = t('Block non-existent users');
return $form;
}
protected function map(FeedsSource $source, FeedsParserResult $result, $target_item = NULL) {
$target_item = parent::map($source, $result, $target_item);
$roles = array_filter($this->config['roles']);
foreach ($roles as $rid) {
$target_item->roles[$rid] = $rid;
}
return $target_item;
}
public function setTargetElement(FeedsSource $source, $target_user, $target_element, $value, array $mapping = array()) {
switch ($target_element) {
case 'pass':
$this
->setPassTarget($source, $target_user, $target_element, $value, $mapping);
break;
case 'created':
$target_user->created = feeds_to_unixtime($value, REQUEST_TIME);
break;
case 'language':
$target_user->language = strtolower($value);
break;
case 'roles_list':
$value = (array) $value;
$this
->rolesListSetTarget($source, $target_user, $target_element, $value, $mapping);
break;
case 'timezone':
$target_user->timezone = $value;
break;
default:
parent::setTargetElement($source, $target_user, $target_element, $value);
break;
}
}
public function getMappingTargets() {
$targets = parent::getMappingTargets();
$targets += array(
'uid' => array(
'name' => t('User ID'),
'description' => t('The uid of the user. NOTE: use this feature with care, user ids are usually assigned by Drupal.'),
'optional_unique' => TRUE,
),
'name' => array(
'name' => t('User name'),
'description' => t('Name of the user.'),
'optional_unique' => TRUE,
),
'mail' => array(
'name' => t('Email address'),
'description' => t('Email address of the user.'),
'optional_unique' => TRUE,
),
'created' => array(
'name' => t('Created date'),
'description' => t('The created (e. g. joined) data of the user.'),
),
'pass' => array(
'name' => t('Password'),
'description' => t('The user password.'),
'summary_callbacks' => array(
array(
$this,
'passSummaryCallback',
),
),
'form_callbacks' => array(
array(
$this,
'passFormCallback',
),
),
),
'status' => array(
'name' => t('Account status'),
'description' => t('Whether a user is active or not. 1 stands for active, 0 for blocked.'),
),
'language' => array(
'name' => t('User language'),
'description' => t('Default language for the user.'),
),
'timezone' => array(
'name' => t('Timezone'),
'description' => t('The timezone identifier, like UTC or Europe/Lisbon.'),
),
'roles_list' => array(
'name' => t('User roles'),
'description' => t('User roles provided as role names in comma separated list.'),
'summary_callbacks' => array(
array(
$this,
'rolesListSummaryCallback',
),
),
'form_callbacks' => array(
array(
$this,
'rolesListFormCallback',
),
),
),
);
if (module_exists('openid')) {
$targets['openid'] = array(
'name' => t('OpenID identifier'),
'description' => t('The OpenID identifier of the user. <strong>CAUTION:</strong> Use only for migration purposes, misconfiguration of the OpenID identifier can lead to severe security breaches like users gaining access to accounts other than their own.'),
'optional_unique' => TRUE,
);
}
$this
->getHookTargets($targets);
return $targets;
}
protected function existingEntityId(FeedsSource $source, FeedsParserResult $result) {
if ($uid = parent::existingEntityId($source, $result)) {
return $uid;
}
foreach ($this
->uniqueTargets($source, $result) as $target => $value) {
switch ($target) {
case 'uid':
$uid = db_query("SELECT uid FROM {users} WHERE uid = :uid", array(
':uid' => $value,
))
->fetchField();
break;
case 'name':
$uid = db_query("SELECT uid FROM {users} WHERE name = :name", array(
':name' => $value,
))
->fetchField();
break;
case 'mail':
$uid = db_query("SELECT uid FROM {users} WHERE mail = :mail", array(
':mail' => $value,
))
->fetchField();
break;
case 'openid':
$uid = db_query("SELECT uid FROM {authmap} WHERE authname = :authname AND module = 'openid'", array(
':authname' => $value,
))
->fetchField();
break;
}
if ($uid) {
return $uid;
}
}
return 0;
}
protected function initEntitiesToBeRemoved(FeedsSource $source, FeedsState $state) {
parent::initEntitiesToBeRemoved($source, $state);
unset($state->removeList[1]);
}
protected function clean(FeedsState $state) {
if (!isset($this->config['update_non_existent']) || $this->config['update_non_existent'] !== FEEDS_BLOCK_NON_EXISTENT) {
return parent::clean($state);
}
if (!empty($state->removeList)) {
foreach (user_load_multiple($state->removeList) as $account) {
$this
->loadItemInfo($account);
$account->feeds_item->hash = $this->config['update_non_existent'];
$account->original = clone $account;
user_save($account, array(
'status' => 0,
));
$state->blocked++;
}
}
}
public function rolesListDefaults() {
$roles = user_roles(TRUE);
unset($roles[DRUPAL_AUTHENTICATED_RID]);
$rids = array_keys($roles);
$rids = array_combine($rids, $rids);
return array(
'role_search' => self::ROLE_SEARCH_NAME,
'allowed_roles' => $rids,
'autocreate' => 0,
'revoke_roles' => 1,
);
}
public function rolesListSummaryCallback(array $mapping, $target, array $form, array $form_state) {
$options = array();
$defaults = $this
->rolesListDefaults();
$mapping += $defaults;
$mapping['allowed_roles'] += $defaults['allowed_roles'];
$role_search_options = $this
->rolesListRoleSearchOptions();
$options[] = t('Search roles by: <strong>@search</strong>', array(
'@search' => $role_search_options[$mapping['role_search']],
));
$role_names = array();
$roles = user_roles(TRUE);
foreach (array_filter($mapping['allowed_roles']) as $rid => $enabled) {
$role_names[] = $roles[$rid];
}
if (empty($role_names)) {
$role_names = array(
'<' . t('none') . '>',
);
}
$options[] = t('Allowed roles: %roles', array(
'%roles' => implode(', ', $role_names),
));
if ($mapping['autocreate']) {
$options[] = t('Automatically create roles');
}
else {
$options[] = t('Only assign existing roles');
}
if ($mapping['revoke_roles']) {
$options[] = t('Revoke roles: yes');
}
else {
$options[] = t('Revoke roles: no');
}
return implode('<br />', $options);
}
public function rolesListFormCallback(array $mapping, $target, array $form, array $form_state) {
$defaults = $this
->rolesListDefaults();
$mapping += $defaults;
$mapping['allowed_roles'] += $defaults['allowed_roles'];
$allowed_roles_options = user_roles(TRUE);
unset($allowed_roles_options[DRUPAL_AUTHENTICATED_RID]);
return array(
'role_search' => array(
'#type' => 'select',
'#title' => t('Search roles by'),
'#options' => $this
->rolesListRoleSearchOptions(),
'#default_value' => $mapping['role_search'],
),
'allowed_roles' => array(
'#type' => 'checkboxes',
'#title' => t('Allowed roles'),
'#options' => $allowed_roles_options,
'#default_value' => $mapping['allowed_roles'],
'#description' => t('Select the roles to accept from the feed.<br />Any other roles will be ignored.'),
),
'autocreate' => array(
'#type' => 'checkbox',
'#title' => t('Auto create'),
'#description' => t("Create the role if it doesn't exist."),
'#default_value' => $mapping['autocreate'],
),
'revoke_roles' => array(
'#type' => 'checkbox',
'#title' => t('Revoke roles'),
'#description' => t('If enabled, roles that are not provided by the feed will be revoked for the user. This affects only the "Allowed roles" as configured above.'),
'#default_value' => $mapping['revoke_roles'],
),
);
}
public function rolesListRoleSearchOptions() {
return array(
self::ROLE_SEARCH_NAME => t('Role name'),
self::ROLE_SEARCH_RID => t('Role ID'),
);
}
public function rolesListSetTarget(FeedsSource $source, $entity, $target, array $values, array $mapping) {
$defaults = $this
->rolesListDefaults();
$mapping += $defaults;
$mapping['allowed_roles'] += $defaults['allowed_roles'];
if ($mapping['revoke_roles']) {
foreach ($mapping['allowed_roles'] as $rid) {
unset($entity->roles[$rid]);
}
}
foreach ($values as $value) {
$role = NULL;
$value = trim($value);
if (strlen($value) < 1) {
continue;
}
switch ($mapping['role_search']) {
case self::ROLE_SEARCH_NAME:
$role = user_role_load_by_name($value);
if (!$role && !empty($mapping['autocreate'])) {
$role = new stdClass();
$role->name = $value;
user_role_save($role);
$role = user_role_load_by_name($role->name);
}
break;
case self::ROLE_SEARCH_RID:
$role = user_role_load($value);
break;
}
if ($role) {
if (isset($mapping['allowed_roles'][$role->rid]) && !$mapping['allowed_roles'][$role->rid]) {
continue;
}
$entity->roles[$role->rid] = $role->name;
}
}
}
public function passSummaryCallback($mapping, $target, $form, $form_state) {
$options = $this
->passSummaryCallbackOptions();
if (!isset($mapping['pass_encryption'])) {
$mapping['pass_encryption'] = self::PASS_UNENCRYPTED;
}
return t('Password encryption: <strong>@encryption</strong>', array(
'@encryption' => $options[$mapping['pass_encryption']],
));
}
protected function passSummaryCallbackOptions() {
return array(
self::PASS_UNENCRYPTED => t('None'),
self::PASS_MD5 => t('MD5'),
self::PASS_SHA512 => t('SHA512'),
);
}
public function passFormCallback($mapping, $target, $form, $form_state) {
return array(
'pass_encryption' => array(
'#type' => 'select',
'#title' => t('Password encryption'),
'#options' => $this
->passFormCallbackOptions(),
'#default_value' => !empty($mapping['pass_encryption']) ? $mapping['pass_encryption'] : self::PASS_UNENCRYPTED,
),
);
}
protected function passFormCallbackOptions() {
return array(
self::PASS_UNENCRYPTED => t('Unencrypted'),
self::PASS_MD5 => t('MD5 (used in older versions of Drupal)'),
self::PASS_SHA512 => t('SHA512 (default in Drupal 7)'),
);
}
protected function setPassTarget($source, $target_user, $target_element, $value, $mapping) {
if (empty($value)) {
return;
}
if (!isset($mapping['pass_encryption'])) {
$mapping['pass_encryption'] = self::PASS_UNENCRYPTED;
}
switch ($mapping['pass_encryption']) {
case self::PASS_MD5:
require_once DRUPAL_ROOT . '/' . variable_get('password_inc', 'includes/password.inc');
$new_hash = user_hash_password($value);
if ($new_hash) {
$new_hash = 'U' . $new_hash;
$target_user->pass = $target_user->pass_crypted = $new_hash;
}
break;
case self::PASS_SHA512:
$target_user->pass = $target_user->pass_crypted = $value;
break;
default:
$target_user->pass = $value;
break;
}
}
}