BUR. Модуль для массовой регистрации пользователей

<?php
// $Id: bur.module,v 1.0 2013/05/01 00:30:25 servekon Exp $
/**
 * Implementation of hook_permission
 * */
function bur_permission(){
  return array(
    'access bur' => array(
      'title' => t('Access to Bulk user registration module'),
      'restrict access' => TRUE,
    ),
	'administer bur' => array(
      'title' => t('Administer of the Bulk user registration module'),
      'restrict access' => TRUE,
    ),
  );
}
/**
 * Implementation of hook_menu
 * */
function bur_menu() {
	/*
	 * Administration interface
	 * */
	$menu['admin/bur'] = array(
		'title' => 'Bulk user registration',
		'page callback' => 'drupal_get_form',
		'page arguments' => array('bur_import_form'),
		'access arguments' => array('administer bur'),
	);
	
	$menu['admin/bur/import'] = array(
		'title' => 'Import', 
		'type' => MENU_DEFAULT_LOCAL_TASK, 
		'weight' => -1,
	);
	
	$menu['admin/bur/fields'] = array(
		'title' => 'Customize Fields',
		'page callback' => 'drupal_get_form',
		'page arguments' => array('bur_sort_form'),
		'access arguments' => array('administer bur'),
		'type' => MENU_LOCAL_TASK,
		'weight' => 1,
	);

	$menu['admin/bur/fields/add'] = array(
		'title' => 'Add field',
		'page callback' => 'drupal_get_form',
		'page arguments' => array('bur_add_field_form'),
		'access arguments' => array('administer bur'),
		'type' => MENU_LOCAL_ACTION,
		'weight' => 1,
	);
	
	$menu['admin/bur/fields/del/%'] = array(
		'title' => 'Delete field',
		'page callback' => 'bur_del_field',
		'page arguments' => array(4),
		'access arguments' => array('administer bur'),
		'type' => MENU_CALLBACK,
		'weight' => 2,
	);
	
	$menu['admin/bur/file_format'] = array(
		'title' => 'The setting for the file format',
		'page callback' => 'drupal_get_form',
		'page arguments' => array('bur_file_format_form'),
		'access arguments' => array('administer bur'),
		'type' => MENU_LOCAL_TASK,
		'weight' => 2,
	);
	return $menu;
}
/**
 * Implements hook_theme().
 */
function bur_theme() {
  return array(
    // Theme function for the 'simple' example.
    'bur_sort_form' => array(
      'render element' => 'form',
    ),
  );
}

/**
 * Import form
 * */
function bur_import_form($form=array(), &$form_state){
	
	$count  = 0;
	
	//~ Get all language
	$lanArr = array();
	$langList = language_list();
	$langArr = array(t('No'));
	foreach($langList as $lang){
		$langArr[$lang->language] = $lang->native;
	}
	
	//~ Get timezones
	$timeZoneList = system_time_zones();
	array_unshift($timeZoneList, t('No'));
	
	//~ Get user roles
	$userRoleList = $tblUser = array();
	$userRoleList = user_roles();
	array_unshift($userRoleList, t('No'));
	
	if(db_table_exists('bur_bulk_user_reg_tmp')){
		$tblUser = db_select('bur_bulk_user_reg_tmp', 't')
		->fields('t')
		->range(0, 10)
		->orderRandom()
		->execute();
		
		foreach($tblUser as $item){
			unset($item->random_field);
			$rangeTmp[] = (array)$item;
		}
		
		$count = db_select('bur_bulk_user_reg_tmp')
		->countQuery()
		->execute()
		->fetchField();
	}
	
	if($count > 0){
		drupal_set_message(t('%time left', array('%time'=>$count)).'.', 'warning', false);
	}
	
	$form['bur_bulk_user_reg_file'] = array(
		'#type' => 'file', 
		'#title' => t('Choose a file'), 
		'#size' => 40,
		'#weight' => 1,
	);
	
	$form['bur_bulk_user_reg_rebuild'] = array(
		'#type' => 'checkbox',
		'#title' => t('Re-create a temporary table'),
		'#return_value' => 1,
		'#weight' => 2,
    );
	
	$form['bur_bulk_user_reg_count'] = array(
		'#type' => 'select',
		'#title' => t('Number of entries to be processed per pass'),
		'#options' => array(
			0 => t('No'),
			10 => '10',
			100 => '100',
			750 => '750',
			1500 => '1500',
			2000 => '2000',
			2500 => '2500',
			3000 => '3000',
			3500 => '3500',
			4000 => '4000',
			4500 => '4500',
			5000 => '5000',
			7500 => '7500',
			10000 => '10000',
			15000 => '15000',
			20000 => '20000',
		),
		'#default_value' => (int)(isset($_COOKIE['bur_quantity']) ? $_COOKIE['bur_quantity'] : '10000'),
		'#description' => t('Select "No", if you just need to create a temporary table and fill it with data.'),
		'#weight' => 3,
	);
	
	$form['bur_extra'] = array(
      '#type' => 'fieldset',
      '#title' => t('Additional settings for all users'),
      '#weight' => 5,
      '#collapsible' => TRUE,
      '#collapsed' =>FALSE,
    );
	
	$form['bur_extra']['bur_extra_set_all_active'] = array(
		'#type' => 'checkbox',
		'#title' => t('Account activation'),
		'#return_value' => 1,
		'#default_value' => (int)(isset($_COOKIE['bur_isactive']) ? $_COOKIE['bur_isactive'] : 0),
		'#weight' => 2,
    );
	
	$form['bur_extra']['bur_extra_user_role'] = array(
		'#type' => 'select',
		'#title' => t('User roles and permissions'),
		'#options' => $userRoleList,
		'#default_value' => (int)(isset($_COOKIE['bur_user_role']) ? $_COOKIE['bur_user_role'] : 0),
		'#weight' => 3,
    );
	
	$form['bur_extra']['bur_extra_lang'] = array(
		'#type' => 'select',
		'#title' => t('Default language'),
		'#options' => $langArr,
		'#default_value' => check_plain((isset($_COOKIE['bur_lang']) ? $_COOKIE['bur_lang'] : 0)),
		'#weight' => 4,
    );
	
	$form['bur_extra']['bur_extra_timezone'] = array(
		'#type' => 'select',
		'#title' => t('Timezone'),
		'#options' => $timeZoneList,
		'#default_value' => check_plain((isset($_COOKIE['bur_timezone']) ? $_COOKIE['bur_timezone'] : 'Europe/Moscow')),
		'#weight' => 5,
    );
	
	if($count > 0){
		$form['bur_ten_random'] = array(
		  '#type' => 'fieldset',
		  '#title' => t('Ten random records'),
		  '#weight' => 7,
		  '#collapsible' => TRUE,
		  '#collapsed' =>TRUE,
		);
		
		$form['bur_ten_random']['bur_ten_random_table'] = array(
		  '#type' => 'item',
		  '#title' => '',
		  '#prefix' => '<div>'.theme('table', array('header'=>array_keys($rangeTmp[0]),'rows'=>$rangeTmp)).'</div>',
		);
	}
	
	$form['submit'] = array(
			'#type' => 'submit',
			'#value' => t('Save and continue'),
			'#weight' => 10,
	);
	
	$form['bur_cancel'] = array(
		'#type' => 'link',
		'#title' => t('Cancel'),
		'#href' => 'admin/bur',
		'#weight' => 11,
	);
	
	return $form;
}
function bur_import_form_submit($form, &$form_state){
	module_load_install('user');
	$usFieldStuct = user_schema();
	$usFieldStuct = $usFieldStuct['users'];
	
	$tblBulk 	= 'bur_bulk_user_reg_tmp';
	$file 		= $_FILES['files']['tmp_name']['bur_bulk_user_reg_file'];
	$tblUser 	= array();
	$quantity	= (int)$form_state['values']['bur_bulk_user_reg_count'];
	setcookie('bur_quantity', $quantity);
	$recreate	= (int)$form_state['values']['bur_bulk_user_reg_rebuild'];
	$isActive	= (int)$form_state['values']['bur_extra_set_all_active'];
	setcookie('bur_isactive', $isActive);
	$userRole	= (int)$form_state['values']['bur_extra_user_role'];
	setcookie('bur_user_role', $userRole);
	$lang		= $form_state['values']['bur_extra_lang'];
	setcookie('bur_lang', $lang);
	$timezone	= $form_state['values']['bur_extra_timezone'];
	setcookie('bur_timezone', $timezone);
	$pref		= '';
	$termArr	= array(
		0 => '\\t',
		1 => ';',
		2 => ',',
		3 => '|',
		4 => '^',
	);

	$enclArr	= array(
		0 => '\\\'',
		1 => '\\\'\\\'',
		2 => '"',
		3 => '%%',
	);

	$lineTerArr	= array(
		0 => '\\r\\n',
		1 => '\\n',
		2 => '\\r',
	);
	
	$escdArr 	= array('\\\\');
	
	$schema['fields']['id'] = array('type' => 'serial', 'size' => 'big', 'not null' => TRUE, 'description'=> "Bur fields ID");
	$burquery	= db_query('SELECT `bur_user_field`, `bur_is_data` FROM {bur_fields_table} ORDER BY `bur_weight` ASC');
	
	foreach($burquery as $field){
		if($field->bur_is_data > 0){
			$pref = 'is_data_';
		}
		else{
			$pref = '';
		}
		$burFields[$field->bur_user_field] = $pref.$field->bur_user_field;
		$burArrField[] = '`'.$field->bur_user_field.'`';
		if($field->bur_is_data == 0){
			$schema['fields'][$field->bur_user_field] = $usFieldStuct['fields'][$field->bur_user_field];
			
			if(isset($usFieldStuct['indexes'][$field->bur_user_field])){
				$schema['indexes'][$field->bur_user_field] = $usFieldStuct['indexes'][$field->bur_user_field];
			}
		}
		else{
			$schema['fields'][$field->bur_user_field] = array('type' => 'varchar', 'size' => 'normal', 'not null' => TRUE, 'length' => 255);
		}
	}
	
	$schema['unique keys']['name'] = array ('name');
	$schema['primary key'] = array('id');
	
	//Create table from fields config
	if(!db_table_exists($tblBulk) or ($recreate >0 and db_table_exists($tblBulk))){
		try{
			db_drop_table($tblBulk);
			db_create_table($tblBulk, $schema);
		}
		catch(Exception $e){
			watchdog_exception('error', $e);
			drupal_set_message(t('Error'), 'warning'); return false;
		}
	}
	elseif(!empty($file)){
		db_truncate($tblBulk)->execute();
	}
	
	//Import data from CSV file
	if(!empty($file)){
		try{
			$terminated = (int)variable_get('bur_fields_terminated', '\t');
			$enclosed	= (int)variable_get('bur_fields_enclosed', '"');
			$escaped	= (int)variable_get('bur_fields_escaped', '\\');
			$line_term	= (int)variable_get('bur_lines_terminated', '\r\n');
			
			$q = 'LOAD DATA LOCAL INFILE :file INTO TABLE {'.check_plain($tblBulk).'} CHARACTER SET utf8 FIELDS TERMINATED BY  \''.$termArr[$terminated].'\' ENCLOSED BY  \''.$enclArr[$enclosed].'\' ESCAPED BY  \''.$escdArr[$escaped].'\' LINES TERMINATED BY  \''.$lineTerArr[$line_term].'\' ('.implode(',', $burArrField).')';
			db_query($q, array(':file'=>$file), array(PDO::MYSQL_ATTR_LOCAL_INFILE => 1));
		}
		catch(Exception $e){
			watchdog_exception('error', $e);
			drupal_set_message('When importing file the error occurred. More information in '.l('syslog','admin/reports/dblog').'.'.'<br />'.$e->getMessage(), 'error', FALSE); return false;
		}
	}
	
	//~ debug($q);
	//Clear importing data
	db_query('DELETE a.* FROM {'.$tblBulk.'} as a INNER JOIN {users} as u on a.name=u.name');
	
	$tblUser = db_select($tblBulk, 't')
	->fields('t')
	->range(0, $quantity)
	->execute();
	
	$count  = $tblUser->rowCount();
	
	$newAcc = (object)array('is_new'=>true);
	
	if($quantity == 0){
		drupal_set_message(t('Registration of new users is missed.'), 'warning', FALSE); return false;
	}
	
	if($count > 0){
		foreach($tblUser as $val){
			$newUser = array();
			foreach($burFields as $bKey=>$bval){
				if(substr($bval, 0, 7) == 'is_data'){
					$newUser['data'][$bKey] = $val->$bKey;
				}
				else{
					$newUser[$bKey] = $val->$bKey;
				}
			}
			unset($bKey,$bVal);
			
			if(!empty($isActive)){
				$newUser['status'] 						= $isActive;
			}
			if(!empty($lang)){
				$newUser['language'] 					= check_plain($lang);
			}
			if(!empty($timezone)){
				$newUser['timezone'] 					= check_plain($timezone);
			}
			if(!empty($userRole)){
				$newUser['roles']						= array($userRole => $userRole);
			}
			$operations[] = array('user_save', array($newAcc, $newUser));
		}
		unset($val,$newUser,$tblUser);
		
		$batch = array(
			'operations' => $operations,
		);
		batch_set($batch);
	}
	else{
		drupal_set_message(t('Table is empty.'), 'error', FALSE); return false;
	}
}
/**
 * Manage fields form
 * */
function bur_sort_form($form_state){
	$form['bur_fields']['#tree'] = true;
	
	$yesno = array(0=>t('No'), 1=>t('Yes'));
	
	$result = db_query('SELECT bur_id, bur_title, bur_user_field, bur_is_data,bur_weight FROM {bur_fields_table} ORDER BY bur_weight ASC');
	
	foreach($result as $field){
		$form['bur_fields'][$field->bur_id]=array(
			'user_field' => array(
				'#markup' => check_plain($field->bur_user_field)
			),
			'title' => array(
				'#type' => 'textfield',
				'#default_value' => check_plain($field->bur_title),
				'#size' => 20,
				'#maxlength' => 255,
			),
			'is_data' =>array(
				'#markup' => $yesno[$field->bur_is_data]
			),
			'actions' =>array(
				'#markup' => l(t('Delete'), 'admin/bur/fields/del/'.$field->bur_id)
			),
			'weight' => array(
				'#type' => 'weight',
				'#title' => t('Weight'),
				'#default_value' => $field->bur_weight,
				'#delta' => 10,
				'#title-display' => 'invisible',
			),
		);
	}
	
	$form['actions'] = array('#type' => 'actions');
	$form['actions']['submit'] = array('#type' => 'submit', '#value' => t('Save settings'));
	return $form;
}

function theme_bur_sort_form($variables) {
  $form = $variables['form'];

  // Initialize the variable which will store our table rows.
  $rows = array();

  // Iterate over each element in our $form['example_items'] array.
  foreach (element_children($form['bur_fields']) as $id) {

    
    $form['bur_fields'][$id]['weight']['#attributes']['class'] = array('bur-field-weight');

    $rows[] = array(
      'data' => array(
        // Add our 'user_field' column.
        drupal_render($form['bur_fields'][$id]['user_field']),
        // Add our 'title' column.
        drupal_render($form['bur_fields'][$id]['title']),
        drupal_render($form['bur_fields'][$id]['is_data']),
        drupal_render($form['bur_fields'][$id]['actions']),
        // Add our 'weight' column.
        drupal_render($form['bur_fields'][$id]['weight']),
      ),
      'class' => array('draggable'),
    );
  }

  $header = array(t('Field of user table'), t('Title'), t('Is optional field'), t('Actions'), t('Weight'));

  $table_id = 'bur-fields-table';

  // We can render our tabledrag table for output.
  $output = theme('table', array(
    'header' => $header,
    'rows' => $rows,
    'attributes' => array('id' => $table_id),
  ));

  // And then render any remaining form elements (such as our submit button).
  $output .= drupal_render_children($form);

  drupal_add_tabledrag($table_id, 'order', 'sibling', 'bur-field-weight');

  return $output;
}
/**
 * Bulk user registration sort form submit
 * */
function bur_sort_form_submit($form, &$form_state){
	foreach ($form_state['values']['bur_fields'] as $id => $field) {
		db_update('bur_fields_table')
		->fields(array(
				'bur_title' => check_plain($field['title']),
				'bur_weight'=> $field['weight']
		))
		->condition('bur_id', (int)$id)
		->execute();
	}
}
/**
 * Form for add new field
 * */
function bur_add_field_form($form=array(), &$form_state){
	
	$fields 	= $weightArr = array();
	
	for($i=0; $i<20; $i++){
		$weightArr[$i] = $i;
	}
	 	
	try{
		$getFields 	= db_query('SHOW COLUMNS FROM {users}');
		$extFields	= db_query('SELECT `bur_user_field` as field FROM {bur_fields_table}');
		$maxWeight	= db_query('SELECT MAX(`bur_weight`) as m FROM {bur_fields_table}')->fetchObject();
		
		foreach ($getFields as $field){
			$fields[$field->Field] = $field->Field;
		}
		foreach($extFields as $extField){
			$noFields[$extField->field] = $extField->field;
		}
		//Allow add only previously unmounted fields
		$diffFields	= array_diff($fields, $noFields);
	}
	catch(Exception $e){
		watchdog_exception('error', $e);
		drupal_set_message(t('Error'), 'warning'); return false;
	}
	
	$form['user_field']=array(
		'#type' => 'select',
		'#title' => t('Field of user table'),
		'#required' => TRUE,
		'#options' => $diffFields,
		'#ajax' => array(
			'callback' => 'bur_is_data_callback',
			'wrapper' => 'user-data-wrapper',
			'event' => 'change',
			'method' => 'replace',
		),
	);
	
	$form['data_field']=array(
		'#prefix' => '<div id="user-data-wrapper">',
		'#suffix' => '</div>',
	);
	
	if(isset($form_state['input']['user_field']) and $form_state['input']['user_field'] == 'data'){
		$form['data_field'] = array(
			'#type' => 'textfield',
			'#title' => t('Name of user variable'),
			'#required' => TRUE,
			'#size' => 50,
			'#prefix' => '<div id="user-data-wrapper">',
			'#suffix' => '</div>',
		);
	}

	$form['title']=array(
		'#type' => 'textfield',
		'#title' => t('Title'),
		'#required' => TRUE,
		'#size' => 50,
	);
	
	$form['weight']=array(
		'#type' => 'select',
		'#title' => t('Weight'),
		'#required' => TRUE,
		'#default_value' => ($maxWeight->m+1),
		'#options' => $weightArr,
	);
	
	$form['submit']=array(
		'#type' => 'submit',
		'#value' => t('Add field'),
	);
	$form['bur_cancel'] = array(
		'#type' => 'link',
		'#title' => t('Cancel'),
		'#href' => 'admin/bur',
	);
	return $form;
}
/**
 * Ajax function for choose type field
 * */
function bur_is_data_callback($form, $form_state){
	return $form['data_field'];
}
function bur_add_field_form_submit($form, &$form_state){
	form_state_values_clean($form_state);
	
	$q 		= $form_state['values'];
	$isData = 0;
	$field	= check_plain($q['user_field']);
	
	if(!empty($q['data_field'])){
		$isData = 1;
		$field	= check_plain($q['data_field']);
	}
	
	try{
		$ins = db_insert('bur_fields_table')->fields(array(
			'bur_title' => $q['title'],
			'bur_user_field' => $field,
			'bur_weight' => (int)$q['weight'],
			'bur_is_data' => $isData,
		))->execute();
		drupal_set_message(t('Field added.'), 'status'); return true;
	}
	catch(Exception $e){
		watchdog_exception('error', $e);
		drupal_set_message(t('Error'), 'warning'); return false;
	}
}

/**
 * Delete field
 * */
function bur_del_field($id){
	return drupal_render(drupal_get_form('bur_del_field_form', $id));
}
/**
 * Delete field form
 * */
function bur_del_field_form($form=array(), &$form_state, $id){
	
	$field = db_query('SELECT `bur_title` FROM {bur_fields_table} WHERE bur_id = :id LIMIT 0,1', array(':id'=>$id))->fetchObject();
	
	$form['del_id']=array(
		'#type' => 'hidden',
		'#value' => (int)$id,
	);
	
	$form['confirm']=array(
		'#type' => 'item',
		'#markup' => t('Do you want to remove a field "%field"?', array('%field'=>(isset($field->bur_title) ? $field->bur_title : ''))),
	);
	
	$form['submit'] = array(
			'#type' => 'submit',
			'#value' => t('Yes'),
	);
	
	$form['bur_cancel'] = array(
		'#type' => 'link',
		'#title' => t('Cancel'),
		'#href' => 'admin/bur',
	);
	
	return $form;
}
/**
 * Delete field form submit
 * */
function bur_del_field_form_submit($form, &$form_state){
	$id = $form_state['values']['del_id'];
	try{
		db_delete('bur_fields_table')->condition('bur_id', $id)->execute();
		drupal_set_message(t('Field deleted.'));
	}
	catch(Exception $e){
		watchdog_exception('error', $e);
		drupal_set_message(t('Error'), 'warning'); return false;
	}
	drupal_goto('admin/bur/fields');
}
/**
 * Seettings of file format
 * */
function bur_file_format_form($form=array()){
	$form['bur_fields_terminated']=array(
		'#type' => 'select',
		'#title' => t('The field separator'),
		'#default_value' => (int)variable_get('bur_fields_terminated', 0),
		'#size' => 1,
		'#options' => array(
			0 => 'TAB',
			1 => ';',
			2 => ',',
			3 => '|',
			4 => '^',
		),
		//'#description' => t('Possible values: ').l('[TAB]', '#', array('attributes'=>array('onclick'=>'javascript:jQuery("#edit-bur-fields-terminated").val("\\\\t"); return false;'))),
		'#required' => TRUE,
	);
	//ESCAPED
	$form['bur_fields_enclosed']=array(
		'#type' => 'select',
		'#title' => t('The fields are enclosed in'),
		'#default_value' => (int)variable_get('bur_fields_enclosed', 0),
		'#size' => 1,
		'#options' => array(
			0 => '\' ('.t('Single quote').')',
			1 => '\'\' ('.t('Two single quotes').')',
			2 => '" ('.t('Double quote').')',
			3 => '%%',
		),
		//'#description' => t('Set zero if empty'),
		'#required' => TRUE,
	);
	
	$form['bur_fields_escaped']=array(
		'#type' => 'select',
		'#title' => t('The escape character'),
		'#default_value' => (int)variable_get('bur_fields_escaped', 0),
		'#size' => 1,
		'#options' => array(
			0 => '\\',
		),
		'#required' => TRUE,
	);
	
	$form['bur_lines_terminated']=array(
		'#type' => 'select',
		'#title' => t('Line separator'),
		'#default_value' => (int)variable_get('bur_lines_terminated', 0),
		'#size' => 1,
		'#options' => array(
			0 => '\\r\\n',
			1 => '\\n',
			2 => '\\r',
		),
		'#required' => TRUE,
	);
	
	return system_settings_form($form);
}