add-user-autocomplete.php

<?php
/*
Plugin Name: Add User Autocomplete
Plugin URI: http://github.com/boonebgorges/add-user-autocomplete
Description: Enables autocomplete for the Add Existing User field at Dashboard > Users > Add New
Author: Boone Gorges
Version: 1.1
Author URI: http://boonebgorges.com
*/

if ( !defined( 'ABSPATH' ) )
	return;


class Add_User_Autocomplete {
	var $base_url;

	function __construct() {
		add_action( 'admin_print_styles-user-new.php', array( $this, 'add_admin_styles' ) );
		add_action( 'admin_print_scripts-user-new.php', array( $this, 'add_admin_scripts' ) );
		add_action( 'admin_print_scripts-user-new.php', array( $this, 'load_string_filter' ) );
		add_action( 'user_new_form_tag', array( $this, 'unload_string_filter' ) );
		add_action( 'wp_ajax_add_to_blog_find_user', array( &$this, 'autocomplete_results' ) );
		add_action( 'admin_init', array( &$this, 'catch_submit' ) );
		add_action( 'admin_notices', array( &$this, 'admin_notices' ) );

		if ( $this->allow_subsite_admin_to_noconfirm() ) {
			add_action( 'user_new_form', array( $this, 'subsite_noconfirm' ) );
		}
	}

	/**
	 * Helper method to determine if sub-site admins can skip confirmation emails.
	 *
	 * @since 1.2.0
	 *
	 * @return bool
	 */
	protected function allow_subsite_admin_to_noconfirm() {
		/**
		 * Filter to determine if sub-site admins can skip confirmation emails?
		 *
		 * @since 1.2.0
		 *
		 * @param bool $retval Defaults to false.
		 */
		$retval = apply_filters( 'aua_allow_subsite_admin_to_noconfirm', false );
		return $retval;
	}

	function add_admin_styles() {
		wp_enqueue_style( 'add-user-autocomplete-css', plugins_url() . '/add-user-autocomplete/css/add-user-autocomplete.css' );
	}

	function add_admin_scripts() {
		// Dequeue WP's autocomplete
		wp_dequeue_script( 'user-suggest' );

		wp_enqueue_script( 'jquery.autocomplete', plugins_url() . '/add-user-autocomplete/js/jquery.autocomplete/jquery.autocomplete.js', array( 'jquery' ) );
		wp_enqueue_script( 'add-user-autocomplete-js', plugins_url() . '/add-user-autocomplete/js/add-user-autocomplete.js', array( 'jquery', 'jquery.autocomplete' ) );

	}

	/**
	 * Load our custom textdomain filter.
	 *
	 * @since 1.2.0
	 */
	function load_string_filter() {
		// No need to do this for super admins.
		if ( current_user_can( 'manage_network_users' ) ) {
			return;
		}

		add_filter( 'gettext', array( $this, 'modify_strings' ), 10, 2 );
	}

	/**
	 * Unload our custom textdomain filter.
	 *
	 * @since 1.2.0
	 */
	function unload_string_filter() {
		remove_filter( 'gettext', array( $this, 'modify_strings' ), 10 );
	}

	/**
	 * String changer method.
	 *
	 * @since 1.2.0
	 *
	 * @param  string $translation Translated text.
	 * @param  string $text        Untranslated text.
	 * @return string
	 */
	function modify_strings( $translation, $text ) {
		switch ( $text ) {
			case 'Email' :
				// Use built-in WordPress string.
				return __( 'Email or Username' );
				break;

			case 'Enter the email address of an existing user on this network to invite them to this site. That person will be sent an email asking them to confirm the invite.' :
				// Use built-in WordPress string.
				return __( 'Enter the email address or username of an existing user on this network to invite them to this site. That person will be sent an email asking them to confirm the invite.' );
				break;

			default :
				return $translation;
				break;
		}
	}

	function autocomplete_results() {

		$return = array(
			'query' 	=> $_REQUEST['query'],
			'data' 		=> array(),
			'suggestions' 	=> array()
		);

		// Exclude current users of this blog
		$this_blog_users = new WP_User_Query( array(
			'blog_id' => get_current_blog_id()
		) );

		$tbu_ids = array();
		if ( !empty( $this_blog_users->results ) ) {
			foreach( $this_blog_users->results as $this_blog_user ) {
				$tbu_ids[] = $this_blog_user->ID;
			}
		}

		$users = new A2B_User_Query( array(
			'blog_id' => false,
			'search'  => '*' . $_REQUEST['query'] . '*',
			'exclude' => $tbu_ids
		) );

		if ( !empty( $users->results ) ) {
			$suggestions = array();
			$data 	     = array();

			foreach ( $users->results as $user ) {
				$suggestions[] 	= $user->display_name . ' (' . $user->user_login . ')';
				$data[] 	= $user->ID;
			}

			$return['suggestions'] = $suggestions;
			$return['data']	       = $data;
		}

		echo json_encode( $return );
		die();
	}

	function catch_submit() {
		if ( isset( $_POST['add_ids'] ) ) {
			$redirect    = 'user-new.php';
			foreach( (array)$_POST['add_ids'] as $user_id ) {
				if ( isset( $_POST[ 'noconfirmation' ] ) && ( ( ! current_user_can( 'manage_network_users' ) && $this->allow_subsite_admin_to_noconfirm() ) || is_super_admin() ) ) {
					add_existing_user_to_blog( array( 'user_id' => $user_id, 'role' => $_REQUEST[ 'role' ] ) );
					$users_added[] = (int)$user_id;
				} else {
					$user_details = get_userdata( $user_id );
					$newuser_key = substr( md5( $user_id ), 0, 5 );
					add_option( 'new_user_' . $newuser_key, array( 'user_id' => $user_id, 'email' => $user_details->user_email, 'role' => $_REQUEST[ 'role' ] ) );
					$message = __("Hi,\n\nYou have been invited to join '%s' at\n%s as a %s.\nPlease click the following link to confirm the invite:\n%s\n");
					wp_mail( $user_details->user_email, sprintf( __( '[%s] Joining confirmation' ), get_option( 'blogname' ) ),  sprintf($message, get_option('blogname'), site_url(), $_REQUEST[ 'role' ], site_url("/newbloguser/$newuser_key/")));
					$users_invited[] = (int)$user_id;
				}
			}

			if ( !empty( $users_added ) ) {
				$redirect = add_query_arg( 'users_added', implode( ',', $users_added ), $redirect );
			}

			if ( !empty( $users_invited ) ) {
				$redirect = add_query_arg( 'users_invited', implode( ',', $users_invited ), $redirect );
			}

			unset( $_POST );
			wp_redirect( $redirect );
		}
	}

	function admin_notices() {
		$type = '';
		if ( isset( $_GET['users_added'] ) ) {
			$user_ids = explode( ',', $_GET['users_added'] );
			$type = 'added';
		} else if ( isset( $_GET['users_invited'] ) ) {
			$user_ids = explode( ',', $_GET['users_invited'] );
			$type = 'invited';
		}

		if ( $type ) {
			$ustring = '<ul style="list-style:disc; padding-left: 20px;">';
			foreach( $user_ids as $user_id ) {
				$userdata = get_userdata( $user_id );
				$ustring .= '<li>' . sprintf( __( '%1$s (%2$s)', 'a2b' ), $userdata->display_name, $userdata->user_login ) . '</li>';
			}
			$ustring .= '</ul>';

			switch ( $type ) {
				case 'added' :
					$message = sprintf( __( 'The following users were successfully added to your site: %s', 'a2b' ), $ustring );
					break;

				case 'invited' :
					$message = sprintf( __( 'Invitation emails sent to the following users: %s A confirmation link must be clicked for them to be added to your site.', 'a2b' ), $ustring );
					break;
			}

			echo '<div id="message" class="updated"><p>' . $message . '</p></div>';
		}
	}

	/**
	 * Add back the 'Skip Confirmation Email' option for sub-site admins.
	 *
	 * @since 1.2.0
	 */
	public function subsite_noconfirm( $context ) {
		// No need to do this for super admins!
		if ( current_user_can( 'manage_network_users' ) ) {
			return;
		}

		// Use built-in WordPress strings.
		$heading = __( 'Skip Confirmation Email' );
		$desc    = __( 'Add the user without sending an email that requires their confirmation.' );

		echo '

<table class="form-table">
	<tr>
		<th scope="row">' . $heading . '</th>
		<td>
			<input type="checkbox" name="noconfirmation" id="adduser-noconfirmation" value="1" />
			<label for="adduser-noconfirmation">' . $desc . '</label>
		</td>
	</tr>
</table>

		';
	}
}

function aua_init() {
	if ( is_admin() ) {
		$add_user_autocomplete = new Add_User_Autocomplete;
	}
}
add_action( 'init', 'aua_init' );

/**
 * Extends WP_User_Query to make search better
 */
class A2B_User_Query extends WP_User_Query {
	/**
	 * @see WP_User_Query::get_search_sql()
	 */
	function get_search_sql( $string, $cols, $wild = false ) {
		$string = esc_sql( $string );

		// Always search all columns
		$cols = array(
			'user_email',
			'user_login',
			'user_nicename',
			'user_url',
			'display_name'
		);

		// Always do 'both' for trailing_wild
		$wild = 'both';

		$searches = array();
		$leading_wild = ( 'leading' == $wild || 'both' == $wild ) ? '%' : '';
		$trailing_wild = ( 'trailing' == $wild || 'both' == $wild ) ? '%' : '';
		foreach ( $cols as $col ) {
			if ( 'ID' == $col )
				$searches[] = "$col = '$string'";
			else
				$searches[] = "$col LIKE '$leading_wild" . like_escape($string) . "$trailing_wild'";
		}

		return ' AND (' . implode(' OR ', $searches) . ')';
	}
}




?>