<?php

/**

 * Multisite WordPress API

 *

 * @package WordPress

 * @subpackage Multisite

 * @since 3.0.0

 */

/**

 * Gets the network's site and user counts.

 *

 * @since MU 1.0

 * @uses get_blog_count()

 * @uses get_user_count()

 *

 * @return array Site and user count for the network.

 */

function get_sitestats() {

	$stats = array(

		'blogs' => get_blog_count(),

		'users' => get_user_count(),

	);

	return $stats;

}

/**

 * Get the admin for a domain/path combination.

 *

 * @since MU 1.0

 *

 * @param string $sitedomain Optional. Site domain.

 * @param string $path Optional. Site path.

 * @return array The network admins

 */

function get_admin_users_for_domain( $sitedomain = '', $path = '' ) {

	global $wpdb;

	if ( ! $sitedomain )

		$site_id = $wpdb->siteid;

	else

		$site_id = $wpdb->get_var( $wpdb->prepare( "SELECT id FROM $wpdb->site WHERE domain = %s AND path = %s", $sitedomain, $path ) );

	if ( $site_id )

		return $wpdb->get_results( $wpdb->prepare( "SELECT u.ID, u.user_login, u.user_pass FROM $wpdb->users AS u, $wpdb->sitemeta AS sm WHERE sm.meta_key = 'admin_user_id' AND u.ID = sm.meta_value AND sm.site_id = %d", $site_id ), ARRAY_A );

	return false;

}

/**

 * Get one of a user's active blogs

 *

 * Returns the user's primary blog, if she has one and

 * it is active. If it's inactive, function returns another

 * active blog of the user. If none are found, the user

 * is added as a Subscriber to the Dashboard Blog and that blog

 * is returned.

 *

 * @since MU 1.0

 * @uses get_blogs_of_user()

 * @uses add_user_to_blog()

 * @uses get_blog_details()

 *

 * @param int $user_id The unique ID of the user

 * @return object The blog object

 */

function get_active_blog_for_user( $user_id ) {

	global $wpdb;

	$blogs = get_blogs_of_user( $user_id );

	if ( empty( $blogs ) )

		return null;

	if ( !is_multisite() )

		return $blogs[$wpdb->blogid];

	$primary_blog = get_user_meta( $user_id, 'primary_blog', true );

	$first_blog = current($blogs);

	if ( false !== $primary_blog ) {

		if ( ! isset( $blogs[ $primary_blog ] ) ) {

			update_user_meta( $user_id, 'primary_blog', $first_blog->userblog_id );

			$primary = get_blog_details( $first_blog->userblog_id );

		} else {

			$primary = get_blog_details( $primary_blog );

		}

	} else {

		//TODO Review this call to add_user_to_blog too - to get here the user must have a role on this blog?

		add_user_to_blog( $first_blog->userblog_id, $user_id, 'subscriber' );

		update_user_meta( $user_id, 'primary_blog', $first_blog->userblog_id );

		$primary = $first_blog;

	}

	if ( ( ! is_object( $primary ) ) || ( $primary->archived == 1 || $primary->spam == 1 || $primary->deleted == 1 ) ) {

		$blogs = get_blogs_of_user( $user_id, true ); // if a user's primary blog is shut down, check their other blogs.

		$ret = false;

		if ( is_array( $blogs ) && count( $blogs ) > 0 ) {

			foreach ( (array) $blogs as $blog_id => $blog ) {

				if ( $blog->site_id != $wpdb->siteid )

					continue;

				$details = get_blog_details( $blog_id );

				if ( is_object( $details ) && $details->archived == 0 && $details->spam == 0 && $details->deleted == 0 ) {

					$ret = $blog;

					if ( get_user_meta( $user_id , 'primary_blog', true ) != $blog_id )

						update_user_meta( $user_id, 'primary_blog', $blog_id );

					if ( !get_user_meta($user_id , 'source_domain', true) )

						update_user_meta( $user_id, 'source_domain', $blog->domain );

					break;

				}

			}

		} else {

			return null;

		}

		return $ret;

	} else {

		return $primary;

	}

}

/**

 * The number of active users in your installation.

 *

 * The count is cached and updated twice daily. This is not a live count.

 *

 * @since MU 2.7

 *

 * @return int

 */

function get_user_count() {

	return get_site_option( 'user_count' );

}

/**

 * The number of active sites on your installation.

 *

 * The count is cached and updated twice daily. This is not a live count.

 *

 * @since MU 1.0

 *

 * @param int $network_id Deprecated, not supported.

 * @return int

 */

function get_blog_count( $network_id = 0 ) {

	if ( func_num_args() )

		_deprecated_argument( __FUNCTION__, '3.1' );

	return get_site_option( 'blog_count' );

}

/**

 * Get a blog post from any site on the network.

 *

 * @since MU 1.0

 *

 * @param int $blog_id ID of the blog.

 * @param int $post_id ID of the post you're looking for.

 * @return WP_Post|null WP_Post on success or null on failure

 */

function get_blog_post( $blog_id, $post_id ) {

	switch_to_blog( $blog_id );

	$post = get_post( $post_id );

	restore_current_blog();

	return $post;

}

/**

 * Add a user to a blog.

 *

 * Use the 'add_user_to_blog' action to fire an event when

 * users are added to a blog.

 *

 * @since MU 1.0

 *

 * @param int $blog_id ID of the blog you're adding the user to.

 * @param int $user_id ID of the user you're adding.

 * @param string $role The role you want the user to have

 * @return bool

 */

function add_user_to_blog( $blog_id, $user_id, $role ) {

	switch_to_blog($blog_id);

	$user = get_userdata( $user_id );

	if ( ! $user ) {

		restore_current_blog();

		return new WP_Error( 'user_does_not_exist', __( 'The requested user does not exist.' ) );

	}

	if ( !get_user_meta($user_id, 'primary_blog', true) ) {

		update_user_meta($user_id, 'primary_blog', $blog_id);

		$details = get_blog_details($blog_id);

		update_user_meta($user_id, 'source_domain', $details->domain);

	}

	$user->set_role($role);

	do_action('add_user_to_blog', $user_id, $role, $blog_id);

	wp_cache_delete( $user_id, 'users' );

	restore_current_blog();

	return true;

}

/**

 * Remove a user from a blog.

 *

 * Use the 'remove_user_from_blog' action to fire an event when

 * users are removed from a blog.

 *

 * Accepts an optional $reassign parameter, if you want to

 * reassign the user's blog posts to another user upon removal.

 *

 * @since MU 1.0

 *

 * @param int $user_id ID of the user you're removing.

 * @param int $blog_id ID of the blog you're removing the user from.

 * @param string $reassign Optional. A user to whom to reassign posts.

 * @return bool

 */

function remove_user_from_blog($user_id, $blog_id = '', $reassign = '') {

	global $wpdb;

	switch_to_blog($blog_id);

	$user_id = (int) $user_id;

	do_action('remove_user_from_blog', $user_id, $blog_id);

	// If being removed from the primary blog, set a new primary if the user is assigned

	// to multiple blogs.

	$primary_blog = get_user_meta($user_id, 'primary_blog', true);

	if ( $primary_blog == $blog_id ) {

		$new_id = '';

		$new_domain = '';

		$blogs = get_blogs_of_user($user_id);

		foreach ( (array) $blogs as $blog ) {

			if ( $blog->userblog_id == $blog_id )

				continue;

			$new_id = $blog->userblog_id;

			$new_domain = $blog->domain;

			break;

		}

		update_user_meta($user_id, 'primary_blog', $new_id);

		update_user_meta($user_id, 'source_domain', $new_domain);

	}

	// wp_revoke_user($user_id);

	$user = get_userdata( $user_id );

	if ( ! $user ) {

		restore_current_blog();

		return new WP_Error('user_does_not_exist', __('That user does not exist.'));

	}

	$user->remove_all_caps();

	$blogs = get_blogs_of_user($user_id);

	if ( count($blogs) == 0 ) {

		update_user_meta($user_id, 'primary_blog', '');

		update_user_meta($user_id, 'source_domain', '');

	}

	if ( $reassign != '' ) {

		$reassign = (int) $reassign;

		$wpdb->query( $wpdb->prepare("UPDATE $wpdb->posts SET post_author = %d WHERE post_author = %d", $reassign, $user_id) );

		$wpdb->query( $wpdb->prepare("UPDATE $wpdb->links SET link_owner = %d WHERE link_owner = %d", $reassign, $user_id) );

	}

	restore_current_blog();

	return true;

}

/**

 * Create an empty blog.

 *

 * @since MU 1.0

 * @uses install_blog()

 *

 * @param string $domain The new blog's domain.

 * @param string $path The new blog's path.

 * @param string $weblog_title The new blog's title.

 * @param int $site_id Optional. Defaults to 1.

 * @return int The ID of the newly created blog

 */

function create_empty_blog( $domain, $path, $weblog_title, $site_id = 1 ) {

	if ( empty($path) )

		$path = '/';

	// Check if the domain has been used already. We should return an error message.

	if ( domain_exists($domain, $path, $site_id) )

		return __( '<strong>ERROR</strong>: Site URL already taken.' );

	// Need to back up wpdb table names, and create a new wp_blogs entry for new blog.

	// Need to get blog_id from wp_blogs, and create new table names.

	// Must restore table names at the end of function.

	if ( ! $blog_id = insert_blog($domain, $path, $site_id) )

		return __( '<strong>ERROR</strong>: problem creating site entry.' );

	switch_to_blog($blog_id);

	install_blog($blog_id);

	restore_current_blog();

	return $blog_id;

}

/**

 * Get the permalink for a post on another blog.

 *

 * @since MU 1.0

 *

 * @param int $blog_id ID of the source blog.

 * @param int $post_id ID of the desired post.

 * @return string The post's permalink

 */

function get_blog_permalink( $blog_id, $post_id ) {

	switch_to_blog( $blog_id );

	$link = get_permalink( $post_id );

	restore_current_blog();

	return $link;

}

/**

 * Get a blog's numeric ID from its URL.

 *

 * On a subdirectory installation like example.com/blog1/,

 * $domain will be the root 'example.com' and $path the

 * subdirectory '/blog1/'. With subdomains like blog1.example.com,

 * $domain is 'blog1.example.com' and $path is '/'.

 *

 * @since MU 2.6.5

 *

 * @param string $domain

 * @param string $path Optional. Not required for subdomain installations.

 * @return int 0 if no blog found, otherwise the ID of the matching blog

 */

function get_blog_id_from_url( $domain, $path = '/' ) {

	global $wpdb;

	$domain = strtolower( $domain );

	$path = strtolower( $path );

	$id = wp_cache_get( md5( $domain . $path ), 'blog-id-cache' );

	if ( $id == -1 ) // blog does not exist

		return 0;

	elseif ( $id )

		return (int) $id;

	$id = $wpdb->get_var( $wpdb->prepare( "SELECT blog_id FROM $wpdb->blogs WHERE domain = %s and path = %s /* get_blog_id_from_url */", $domain, $path ) );

	if ( ! $id ) {

		wp_cache_set( md5( $domain . $path ), -1, 'blog-id-cache' );

		return 0;

	}

	wp_cache_set( md5( $domain . $path ), $id, 'blog-id-cache' );

	return $id;

}

// Admin functions

/**

 * Checks an email address against a list of banned domains.

 *

 * This function checks against the Banned Email Domains list

 * at wp-admin/network/settings.php. The check is only run on

 * self-registrations; user creation at wp-admin/network/users.php

 * bypasses this check.

 *

 * @since MU

 *

 * @param string $user_email The email provided by the user at registration.

 * @return bool Returns true when the email address is banned.

 */

function is_email_address_unsafe( $user_email ) {

	$banned_names = get_site_option( 'banned_email_domains' );

	if ( $banned_names && ! is_array( $banned_names ) )

		$banned_names = explode( "\n", $banned_names );

	$is_email_address_unsafe = false;

	if ( $banned_names && is_array( $banned_names ) ) {

		$banned_names = array_map( 'strtolower', $banned_names );

		$normalized_email = strtolower( $user_email );

		list( $email_local_part, $email_domain ) = explode( '@', $normalized_email );

		foreach ( $banned_names as $banned_domain ) {

			if ( ! $banned_domain )

				continue;

			if ( $email_domain == $banned_domain ) {

				$is_email_address_unsafe = true;

				break;

			}

			$dotted_domain = ".$banned_domain";

			if ( $dotted_domain === substr( $normalized_email, -strlen( $dotted_domain ) ) ) {

				$is_email_address_unsafe = true;

				break;

			}

		}

	}

	return apply_filters( 'is_email_address_unsafe', $is_email_address_unsafe, $user_email );

}

/**

 * Processes new user registrations.

 *

 * Checks the data provided by the user during signup. Verifies

 * the validity and uniqueness of user names and user email addresses,

 * and checks email addresses against admin-provided domain

 * whitelists and blacklists.

 *

 * The hook 'wpmu_validate_user_signup' provides an easy way

 * to modify the signup process. The value $result, which is passed

 * to the hook, contains both the user-provided info and the error

 * messages created by the function. 'wpmu_validate_user_signup' allows

 * you to process the data in any way you'd like, and unset the

 * relevant errors if necessary.

 *

 * @since MU

 * @uses is_email_address_unsafe()

 * @uses username_exists()

 * @uses email_exists()

 *

 * @param string $user_name The login name provided by the user.

 * @param string $user_email The email provided by the user.

 * @return array Contains username, email, and error messages.

 */

function wpmu_validate_user_signup($user_name, $user_email) {

	global $wpdb;

	$errors = new WP_Error();

	$orig_username = $user_name;

	$user_name = preg_replace( '/\s+/', '', sanitize_user( $user_name, true ) );