query( "UPDATE {$wpdb->blogs} SET last_updated = NOW() WHERE blog_id = '{$wpdb->blogid}'" );
refresh_blog_details( $wpdb->blogid );
}
add_action('delete_post', 'wpmu_update_blogs_date');
add_action('private_to_published', 'wpmu_update_blogs_date');
add_action('publish_phone', 'wpmu_update_blogs_date');
add_action('publish_post', 'wpmu_update_blogs_date');
/*
Determines if the available space defined by the admin has been exceeded by the user
*/
/**
* Returns how much space is available (also shows a picture) for the current client blog, retrieving the value from the master blog 'main' option table
*
* @param string $action
* @return string
*/
function wpmu_checkAvailableSpace($action) {
// Using the action.
// Set the action to 'not-writable' to block the upload
// Default space allowed is 10 MB
$spaceAllowed = get_site_option( "blog_upload_space" );
if( $spaceAllowed == false )
$spaceAllowed = 10;
$dirName = constant( "ABSPATH" ) . constant( "UPLOADS" );
$dir = dir($dirName);
$size = 0;
while($file = $dir->read()) {
if ($file != '.' && $file != '..') {
if (is_dir($file)) {
$size += dirsize($dirName . '/' . $file);
} else {
$size += filesize($dirName . '/' . $file);
}
}
}
$dir->close();
$size = $size / 1024 / 1024;
$spaceAvailable = sprintf( "%2.2f", ( ($spaceAllowed-$size) ) );
echo sprintf(__('Space Available (%2.2fMB)'), $spaceAvailable);
if (($spaceAllowed-$size)>0) {
return $action;
} else {
// No space left
return 'not-writable';
}
}
add_filter('fileupload_init','wpmu_checkAvailableSpace');
function get_blogaddress_by_id( $blog_id ) {
global $hostname, $domain, $base, $wpdb;
$bloginfo = get_blog_details( $blog_id, false ); // only get bare details!
return "http://" . $bloginfo->domain . $bloginfo->path;
}
function get_blogaddress_by_name( $blogname ) {
global $hostname, $domain, $base, $wpdb;
if( defined( "VHOST" ) && constant( "VHOST" ) == 'yes' ) {
if( $blogname == 'main' )
$blogname = 'www';
return "http://".$blogname.".".$domain.$base;
} else {
return "http://".$hostname.$base.$blogname;
}
}
function get_blogaddress_by_domain( $domain, $path ){
if( defined( "VHOST" ) && constant( "VHOST" ) == 'yes' ) {
$url = "http://".$domain.$path;
} else {
if( $domain != $_SERVER[ 'HTTP_HOST' ] ) {
$blogname = substr( $domain, 0, strpos( $domain, '.' ) );
if( $blogname != 'www.' ) {
$url = 'http://' . substr( $domain, strpos( $domain, '.' ) + 1 ) . $path . $blogname . '/';
} else { // we're installing the main blog
$url = 'http://' . substr( $domain, strpos( $domain, '.' ) + 1 ) . $path;
}
} else { // main blog
$url = 'http://' . $domain . $path;
}
}
return $url;
}
function get_sitestats() {
global $wpdb, $basedomain, $base;
$stats[ 'blogs' ] = get_blog_count();
$count_ts = get_site_option( "get_user_count_ts" );
if( time() - $count_ts > 3600 ) {
$count = $wpdb->get_var( "SELECT count(*) as c FROM {$wpdb->users}" );
update_site_option( "user_count", $count );
update_site_option( "user_count_ts", time() );
} else {
$count = get_site_option( "user_count" );
}
$stats[ 'users' ] = $count;
return $stats;
}
function get_admin_users_for_domain( $sitedomain = '', $path = '' ) {
global $domain, $base, $basedomain, $wpdb, $wpmuBaseTablePrefix;
if( $sitedomain == '' ) {
$sitedomain = $basedomain;
$path = $base;
$site_id = $wpdb->siteid;
} else {
$query = "SELECT id FROM ".$wpdb->site." WHERE domain = '".$sitedomain."' AND path = '".$path."'";
$site_id = $wpdb->get_var( $query );
}
if( $site_id != false ) {
$query = "SELECT ID, user_login, user_pass FROM ".$wpdb->users.", ".$wpdb->sitemeta." WHERE meta_key = 'admin_user_id' AND ".$wpdb->users.".ID = ".$wpdb->sitemeta.".meta_value AND ".$wpdb->sitemeta.".site_id = '".$site_id."'";
$details = $wpdb->get_results( $query, ARRAY_A );
} else {
$details = false;
}
return $details;
}
function get_user_details( $username ) {
global $wpdb;
return $wpdb->get_row( "SELECT * FROM $wpdb->users WHERE user_login = '$username'" );
}
function get_blog_details( $id, $all = true ) {
global $wpdb;
$details = wp_cache_get( $id, 'blog-details' );
if ( $details ) {
if ( $details == -1 )
return false;
elseif ( !is_object($details) ) // Clear old pre-serialized objects. Cache clients do better with that.
wp_cache_delete( $id, 'blog-details' );
else
return $details;
}
$details = $wpdb->get_row( "SELECT * FROM $wpdb->blogs WHERE blog_id = '$id' /* get_blog_details */" );
if ( !$details ) {
wp_cache_set( $id, -1, 'blog-details' );
return false;
}
if ( !$all ) {
wp_cache_add( $id, $details, 'blog-details' );
return $details;
}
$wpdb->hide_errors();
$details->blogname = get_blog_option($id, 'blogname');
$details->siteurl = get_blog_option($id, 'siteurl');
$details->post_count = get_blog_option($id, 'post_count');
$wpdb->show_errors();
$details = apply_filters('blog_details', $details);
wp_cache_set( $id, $details, 'blog-details' );
$key = md5( $details->domain . $details->path );
wp_cache_set( $key, $details, 'blog-lookup' );
return $details;
}
function refresh_blog_details( $id ) {
global $wpdb, $wpmuBaseTablePrefix;
$details = get_blog_details( $id, false );
wp_cache_delete( $id , 'blog-details' );
$key = md5( $details->domain . $details->path );
wp_cache_delete( $key , 'blog-lookup' );
//return $details;
}
function get_current_user_id() {
global $current_user;
return $current_user->ID;
}
function is_site_admin( $user_login = false ) {
global $wpdb, $current_user;
if ( !$current_user && !$user_login )
return false;
if ( $user_login )
$user_login = sanitize_user( $user_login );
else
$user_login = $current_user->user_login;
$site_admins = get_site_option( 'site_admins', array('admin') );
if( is_array( $site_admins ) && in_array( $user_login, $site_admins ) )
return true;
return false;
}
// expects key not to be SQL escaped
function get_site_option( $key, $default = false, $use_cache = true ) {
global $wpdb;
$safe_key = $wpdb->escape( $key );
if( $use_cache == true ) {
$value = wp_cache_get($wpdb->siteid . $key, 'site-options');
} else {
$value = false;
}
if ( false === $value ) {
$value = $wpdb->get_var("SELECT meta_value FROM $wpdb->sitemeta WHERE meta_key = '$safe_key' AND site_id = '{$wpdb->siteid}'");
if ( ! is_null($value) ) {
wp_cache_add($wpdb->siteid . $key, $value, 'site-options');
} elseif ( $default ) {
wp_cache_add($wpdb->siteid . $key, addslashes( $default ), 'site-options');
return $default;
} else {
wp_cache_add($wpdb->siteid . $key, false, 'site-options');
return false;
}
}
$value = stripslashes( $value );
@ $kellogs = unserialize($value);
if ( $kellogs !== FALSE )
return $kellogs;
else
return $value;
}
// expects $key, $value not to be SQL escaped
function add_site_option( $key, $value ) {
global $wpdb;
$safe_key = $wpdb->escape( $key );
$exists = $wpdb->get_row("SELECT meta_value FROM $wpdb->sitemeta WHERE meta_key = '$safe_key' AND site_id = '{$wpdb->siteid}'");
if ( is_object( $exists ) ) {// If we already have it
update_site_option( $key, $value );
return false;
}
if ( is_array($value) || is_object($value) )
$value = serialize($value);
wp_cache_delete($wpdb->siteid . $key, 'site-options');
$wpdb->query( "INSERT INTO $wpdb->sitemeta ( site_id , meta_key , meta_value ) VALUES ( '{$wpdb->siteid}', '$safe_key', '" . $wpdb->escape( $value ) . "')" );
return $wpdb->insert_id;
}
// expects $key, $value not to be SQL escaped
function update_site_option( $key, $value ) {
global $wpdb;
$safe_key = $wpdb->escape( $key );
if ( $value == get_site_option( $key ) )
return;
$exists = $wpdb->get_row("SELECT meta_value FROM $wpdb->sitemeta WHERE meta_key = '$safe_key' AND site_id = '{$wpdb->siteid}'");
if ( false == is_object( $exists ) ) // It's a new record
return add_site_option( $key, $value );
if ( is_array($value) || is_object($value) )
$value = serialize($value);
$value = $wpdb->escape( $value );
$wpdb->query( "UPDATE $wpdb->sitemeta SET meta_value = '" . $wpdb->escape( $value ) . "' WHERE site_id='{$wpdb->siteid}' AND meta_key = '$safe_key'" );
wp_cache_delete( $wpdb->siteid . $key, 'site-options' );
}
/*
function get_blog_option( $id, $key, $default='na' ) {
switch_to_blog($id);
$opt = get_option( $key );
restore_current_blog();
return $opt;
}
*/
function get_blog_option( $blog_id, $setting, $default='na' ) {
global $wpdb, $wpmuBaseTablePrefix;
$key = $blog_id."-".$setting."-blog_option";
$value = wp_cache_get( $key, "site-options" );
if( $value == null ) {
$row = $wpdb->get_row( "SELECT * FROM {$wpmuBaseTablePrefix}{$blog_id}_options WHERE option_name = '{$setting}'" );
if( is_object( $row) ) { // Has to be get_row instead of get_var because of funkiness with 0, false, null values
$value = $row->option_value;
if( $value == false )
$value = 'falsevalue';
wp_cache_set($key, $value, 'site-options');
} else { // option does not exist, so we must cache its non-existence
wp_cache_set($key, 'noop', 'site-options');
}
} elseif( $value == 'noop' ) {
return false;
} elseif( $value == 'falsevalue' ) {
return false;
}
// If home is not set use siteurl.
if ( 'home' == $setting && '' == $value )
return get_blog_option($blog_id, 'siteurl');
if ( 'siteurl' == $setting || 'home' == $setting || 'category_base' == $setting )
$value = preg_replace('|/+$|', '', $value);
if (! unserialize($value) )
$value = stripslashes( $value );
return apply_filters( 'option_' . $setting, maybe_unserialize($value) );
}
function add_blog_option( $id, $key, $value ) {
switch_to_blog($id);
add_option( $key, $value );
restore_current_blog();
$opt = $id."-".$key."-blog_option";
wp_cache_set($opt, $value, 'site-options');
}
function delete_blog_option( $id, $key ) {
switch_to_blog($id);
delete_option( $key );
restore_current_blog();
$opt = $id."-".$key."-blog_option";
wp_cache_set($opt, '', 'site-options');
}
function update_blog_option( $id, $key, $value, $refresh = true ) {
switch_to_blog($id);
$opt = update_option( $key, $value );
restore_current_blog();
if( $refresh == true )
refresh_blog_details( $id );
$opt = $id."-".$key."-blog_option";
wp_cache_set($opt, $value, 'site-options');
}
function switch_to_blog( $new_blog ) {
global $tmpoldblogdetails, $wpdb, $wpmuBaseTablePrefix, $table_prefix, $blog_id, $switched, $switched_stack, $wp_roles, $current_user;
if ( empty($new_blog) )
$new_blog = $blog_id;
if ( empty($switched_stack) )
$switched_stack = array();
$switched_stack[] = $blog_id;
// backup
$tmpoldblogdetails[ 'blogid' ] = $wpdb->blogid;
$tmpoldblogdetails[ 'posts' ] = $wpdb->posts;
$tmpoldblogdetails[ 'categories' ] = $wpdb->categories;
$tmpoldblogdetails[ 'post2cat' ] = $wpdb->post2cat;
$tmpoldblogdetails[ 'comments' ] = $wpdb->comments;
$tmpoldblogdetails[ 'links' ] = $wpdb->links;
$tmpoldblogdetails[ 'link2cat' ] = $wpdb->link2cat;
$tmpoldblogdetails[ 'linkcategories' ] = $wpdb->linkcategories;
$tmpoldblogdetails[ 'options' ] = $wpdb->options;
$tmpoldblogdetails[ 'postmeta' ] = $wpdb->postmeta;
$tmpoldblogdetails[ 'terms' ] = $wpdb->terms;
$tmpoldblogdetails[ 'term_taxonomy' ] = $wpdb->term_taxonomy;
$tmpoldblogdetails[ 'term_relationships' ] = $wpdb->term_relationships;
$tmpoldblogdetails[ 'prefix' ] = $wpdb->prefix;
$tmpoldblogdetails[ 'table_prefix' ] = $table_prefix;
$tmpoldblogdetails[ 'blog_id' ] = $blog_id;
// fix the new prefix.
$table_prefix = $wpmuBaseTablePrefix . $new_blog . "_";
$wpdb->prefix = $table_prefix;
$wpdb->blogid = $new_blog;
$wpdb->posts = $table_prefix . 'posts';
$wpdb->categories = $table_prefix . 'categories';
$wpdb->post2cat = $table_prefix . 'post2cat';
$wpdb->comments = $table_prefix . 'comments';
$wpdb->links = $table_prefix . 'links';
$wpdb->link2cat = $table_prefix . 'link2cat';
$wpdb->linkcategories = $table_prefix . 'linkcategories';
$wpdb->options = $table_prefix . 'options';
$wpdb->postmeta = $table_prefix . 'postmeta';
$wpdb->terms = $table_prefix . 'terms';
$wpdb->term_taxonomy = $table_prefix . 'term_taxonomy';
$wpdb->term_relationships = $table_prefix . 'term_relationships';
$blog_id = $new_blog;
if( is_object( $wp_roles ) ) {
$wpdb->hide_errors();
$wp_roles->_init();
$wpdb->show_errors();
}
if ( is_object( $current_user ) ) {
$current_user->_init_caps();
}
do_action('switch_blog', $blog_id, $tmpoldblogdetails[ 'blog_id' ]);
$switched = true;
}
function restore_current_blog() {
global $table_prefix, $tmpoldblogdetails, $wpdb, $wpmuBaseTablePrefix, $blog_id, $switched, $switched_stack, $wp_roles, $current_user;
if ( !$switched )
return;
$blog = array_pop($switched_stack);
if ( $blog_id == $blog )
return;
// backup
$wpdb->blogid = $tmpoldblogdetails[ 'blogid' ];
$wpdb->posts = $tmpoldblogdetails[ 'posts' ];
$wpdb->categories = $tmpoldblogdetails[ 'categories' ];
$wpdb->post2cat = $tmpoldblogdetails[ 'post2cat' ];
$wpdb->comments = $tmpoldblogdetails[ 'comments' ];
$wpdb->links = $tmpoldblogdetails[ 'links' ];
$wpdb->link2cat = $tmpoldblogdetails[ 'link2cat' ];
$wpdb->linkcategories = $tmpoldblogdetails[ 'linkcategories' ];
$wpdb->options = $tmpoldblogdetails[ 'options' ];
$wpdb->postmeta = $tmpoldblogdetails[ 'postmeta' ];
$wpdb->terms = $tmpoldblogdetails[ 'terms' ];
$wpdb->term_taxonomy = $tmpoldblogdetails[ 'term_taxonomy' ];
$wpdb->term_relationships = $tmpoldblogdetails[ 'term_relationships' ];
$wpdb->prefix = $tmpoldblogdetails[ 'prefix' ];
$table_prefix = $tmpoldblogdetails[ 'table_prefix' ];
$prev_blog_id = $blog_id;
$blog_id = $tmpoldblogdetails[ 'blog_id' ];
unset( $tmpoldblogdetails );
if( is_object( $wp_roles ) ) {
$wpdb->hide_errors();
$wp_roles->_init();
$wpdb->show_errors();
}
if ( is_object( $current_user ) ) {
$current_user->_init_caps();
}
do_action('switch_blog', $blog_id, $prev_blog_id);
$switched = false;
}
function get_blogs_of_user( $id, $all = false ) {
global $wpdb, $wpmuBaseTablePrefix;
$user = get_userdata( $id );
if ( !$user )
return false;
$blogs = array();
$i = 0;
foreach ( $user as $key => $value ) {
if ( strstr( $key, '_capabilities') && strstr( $key, 'wp_') ) {
preg_match('/' . $wpmuBaseTablePrefix . '(\d+)_capabilities/', $key, $match);
$blog = get_blog_details( $match[1] );
if ( $blog && isset( $blog->domain ) && ( $all == true || $all == false && ( $blog->archived == 0 && $blog->spam == 0 && $blog->deleted == 0 ) ) ) {
$blogs[$match[1]]->userblog_id = $match[1];
$blogs[$match[1]]->blogname = $blog->blogname;
$blogs[$match[1]]->domain = $blog->domain;
$blogs[$match[1]]->path = $blog->path;
$blogs[$match[1]]->site_id = $blog->site_id;
$blogs[$match[1]]->siteurl = $blog->siteurl;
}
}
}
return $blogs;
}
function get_active_blog_for_user( $user_id ) { // get an active blog for user - either primary blog or from blogs list
$primary_blog = get_usermeta( $user_id, "primary_blog" );
if( $primary_blog == false ) {
$details = false;
} else {
$details = get_blog_details( $primary_blog );
}
if( ( is_object( $details ) == false ) || ( is_object( $details ) && $details->archived == 1 || $details->spam == 1 || $details->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( $blogs as $blog_id => $blog ) {
$details = get_blog_details( $blog_id );
if( is_object( $details ) && $details->archived == 0 && $details->spam == 0 && $details->deleted == 0 ) {
$ret = $blog;
break;
}
}
} else {
$ret = "username only"; // user has no blogs. We can add details for dashboard.wordpress.com here.
}
return $ret;
} else {
return $details;
}
}
function is_user_member_of_blog( $user_id, $blog_id = 0 ) {
global $wpdb;
if( $blog_id == 0 )
$blog_id = $wpdb->blogid;
$blogs = get_blogs_of_user( $user_id );
if( is_array( $blogs ) ) {
return array_key_exists( $blog_id, $blogs );
} else {
return false;
}
}
function is_archived( $id ) {
return get_blog_status($id, 'archived');
}
function update_archived( $id, $archived ) {
update_blog_status($id, 'archived', $archived);
return $archived;
}
function update_user_status( $id, $pref, $value, $refresh = 1 ) {
global $wpdb;
$wpdb->query( "UPDATE {$wpdb->users} SET {$pref} = '{$value}' WHERE ID = '$id'" );
if( $refresh == 1 )
refresh_user_details($id);
if( $pref == 'spam' ) {
if( $value == 1 )
do_action( "make_spam_user", $id );
else
do_action( "make_ham_user", $id );
}
return $value;
}
function refresh_user_details($id) {
global $wpdb, $wpmuBaseTablePrefix;
if ( !$user = get_userdata( $id ) )
return false;
wp_cache_delete($id, 'users');
wp_cache_delete($user->user_login, 'userlogins');
return $id;
}
function update_blog_status( $id, $pref, $value, $refresh = 1 ) {
global $wpdb;
$wpdb->query( "UPDATE {$wpdb->blogs} SET {$pref} = '{$value}', last_updated = NOW() WHERE blog_id = '$id'" );
if( $refresh == 1 )
refresh_blog_details($id);
if( $pref == 'spam' ) {
if( $value == 1 ) {
do_action( "make_spam_blog", $id );
} else {
do_action( "make_ham_blog", $id );
}
}
return $value;
}
function get_blog_status( $id, $pref ) {
global $wpdb;
$details = get_blog_details( $id, false );
if( $details ) {
return $details->$pref;
}
return $wpdb->get_var( "SELECT $pref FROM {$wpdb->blogs} WHERE blog_id = '$id'" );
}
function get_last_updated( $display = false ) {
global $wpdb;
$blogs = $wpdb->get_results( "SELECT blog_id, domain, path FROM $wpdb->blogs WHERE site_id = '$wpdb->siteid' AND public = '1' AND archived = '0' AND mature = '0' AND spam = '0' AND deleted = '0' AND last_updated != '0000-00-00 00:00:00' ORDER BY last_updated DESC limit 0,40", ARRAY_A );
return $blogs;
}
function get_most_active_blogs( $num = 10, $display = true ) {
global $wpdb;
$most_active = get_site_option( "most_active" );
$update = false;
if( is_array( $most_active ) ) {
if( ( $most_active[ 'time' ] + 60 ) < time() ) { // cache for 60 seconds.
$update = true;
}
} else {
$update = true;
}
if( $update == true ) {
unset( $most_active );
$blogs = get_blog_list( 0, 'all', false ); // $blog_id -> $details
if( is_array( $blogs ) ) {
reset( $blogs );
while( list( $key, $details ) = each( $blogs ) ) {
$most_active[ $details[ 'blog_id' ] ] = $details[ 'postcount' ];
$blog_list[ $details[ 'blog_id' ] ] = $details; // array_slice() removes keys!!
}
arsort( $most_active );
reset( $most_active );
while( list( $key, $details ) = each( $most_active ) ) {
$t[ $key ] = $blog_list[ $key ];
}
unset( $most_active );
$most_active = $t;
}
update_site_option( "most_active", $most_active );
}
if( $display == true ) {
if( is_array( $most_active ) ) {
reset( $most_active );
while( list( $key, $details ) = each( $most_active ) ) {
$url = "http://" . $details[ 'domain' ] . $details[ 'path' ];
print "
" . $details[ 'postcount' ] . " $url";
}
}
}
return array_slice( $most_active, 0, $num );
}
function get_blog_list( $start = 0, $num = 10, $display = true ) {
global $wpdb, $wpmuBaseTablePrefix;
$blogs = get_site_option( "blog_list" );
$update = false;
if( is_array( $blogs ) ) {
if( ( $blogs[ 'time' ] + 60 ) < time() ) { // cache for 60 seconds.
$update = true;
}
} else {
$update = true;
}
if( $update == true ) {
unset( $blogs );
$blogs = $wpdb->get_results( "SELECT blog_id, domain, path FROM $wpdb->blogs WHERE site_id = '$wpdb->siteid' AND public = '1' AND archived = '0' AND mature = '0' AND spam = '0' AND deleted = '0' ORDER BY registered DESC", ARRAY_A );
if( is_array( $blogs ) ) {
while( list( $key, $details ) = each( $blogs ) ) {
$blog_list[ $details[ 'blog_id' ] ] = $details;
$blog_list[ $details[ 'blog_id' ] ][ 'postcount' ] = $wpdb->get_var( "SELECT count(*) FROM " . $wpmuBaseTablePrefix . $details[ 'blog_id' ] . "_posts WHERE post_status='publish' AND post_type='post'" );
}
unset( $blogs );
$blogs = $blog_list;
}
update_site_option( "blog_list", $blogs );
}
if( $num == 'all' ) {
return array_slice( $blogs, $start, count( $blogs ) );
} else {
return array_slice( $blogs, $start, $num );
}
}
function get_blog_count( $id = 0 ) {
global $wpdb;
if( $id == 0 )
$id = $wpdb->siteid;
$count_ts = get_site_option( "blog_count_ts" );
if( time() - $count_ts > 3600 ) {
$count = $wpdb->get_var( "SELECT count(*) as c FROM $wpdb->blogs WHERE site_id = '$id' AND spam='0' AND deleted='0' and archived='0'" );
update_site_option( "blog_count", $count );
update_site_option( "blog_count_ts", time() );
}
$count = get_site_option( "blog_count" );
return $count;
}
function get_blog_post( $blog_id, $post_id ) {
global $wpdb, $wpmuBaseTablePrefix;
$key = $blog_id."-".$post_id."-blog_post";
$post = wp_cache_get( $key, "site-options" );
if( $post == false ) {
$post = $wpdb->get_row( "SELECT * FROM {$wpmuBaseTablePrefix}{$blog_id}_posts WHERE ID = '{$post_id}'" );
wp_cache_add( $key, $post, "site-options", 120 );
}
return $post;
}
function add_user_to_blog( $blog_id, $user_id, $role ) {
switch_to_blog($blog_id);
$user = new WP_User($user_id);
if ( empty($user) )
return new WP_Error('user_does_not_exist', __('That user does not exist.'));
if ( !get_usermeta($user_id, 'primary_blog') ) {
update_usermeta($user_id, 'primary_blog', $blog_id);
$details = get_blog_details($blog_id);
update_usermeta($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();
}
function remove_user_from_blog($user_id, $blog_id = '') {
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_usermeta($user_id, 'primary_blog');
if ( $primary_blog == $blog_id ) {
$new_id = '';
$new_domain = '';
$blogs = get_blogs_of_user($user_id);
if ( count($blogs) > 1 ) {
foreach ( $blogs as $blog ) {
if ( $blog->userblog_id == $blog_id )
continue;
$new_id = $blog->userblog_id;
$new_domain = $blog->domain;
break;
}
}
update_usermeta($user_id, 'primary_blog', $new_id);
update_usermeta($user_id, 'source_domain', $new_domain);
}
wp_revoke_user($user_id);
$blogs = get_blogs_of_user($user_id);
if ( count($blogs) == 0 ) {
update_usermeta($user_id, 'primary_blog', '');
update_usermeta($user_id, 'source_domain', '');
}
restore_current_blog();
}
function create_empty_blog( $domain, $path, $weblog_title, $site_id = 1 ) {
global $wpdb, $table_prefix, $wp_queries, $wpmuBaseTablePrefix, $current_site;
$domain = addslashes( $domain );
$weblog_title = addslashes( $weblog_title );
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 'error: Blog URL already taken.';
// Need to backup 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 "error: problem creating blog entry";
switch_to_blog($blog_id);
install_blog($blog_id);
restore_current_blog();
return true;
}
function get_blog_permalink( $blog_id, $post_id ) {
$key = "{$blog_id}-{$post_id}-blog_permalink";
$link = wp_cache_get( $key, 'site-options' );
if( $link == false ) {
switch_to_blog( $blog_id );
$link = get_permalink( $post_id );
restore_current_blog();
wp_cache_add( $key, $link, "site-options", 30 );
}
return $link;
}
// wpmu admin functions
function wpmu_admin_do_redirect( $url = '' ) {
$ref = '';
if ( isset( $_GET['ref'] ) )
$ref = $_GET['ref'];
if ( isset( $_POST['ref'] ) )
$ref = $_POST['ref'];
if( $ref ) {
$ref = wpmu_admin_redirect_add_updated_param( $ref );
wp_redirect( $ref );
die();
}
if( empty( $_SERVER[ 'HTTP_REFERER' ] ) == false ) {
wp_redirect( $_SERVER[ 'HTTP_REFERER' ] );
die();
}
$url = wpmu_admin_redirect_add_updated_param( $url );
if( isset( $_GET[ 'redirect' ] ) ) {
if( substr( $_GET[ 'redirect' ], 0, 2 ) == 's_' ) {
$url .= "&action=blogs&s=". wp_specialchars( substr( $_GET[ 'redirect' ], 2 ) );
}
} elseif( isset( $_POST[ 'redirect' ] ) ) {
$url = wpmu_admin_redirect_add_updated_param( $_POST[ 'redirect' ] );
}
wp_redirect( $url );
die();
}
function wpmu_admin_redirect_add_updated_param( $url = '' ) {
if( strpos( $url, 'updated=true' ) === false ) {
if( strpos( $url, '?' ) === false ) {
return $url . '?updated=true';
} else {
return $url . '&updated=true';
}
}
return $url;
}
function wpmu_admin_redirect_url() {
if( isset( $_GET[ 's' ] ) ) {
return "s_".$_GET[ 's' ];
}
}
function is_blog_user( $blog_id = 0 ) {
global $current_user, $wpdb, $wpmuBaseTablePrefix;
if ( !$blog_id )
$blog_id = $wpdb->blogid;
$cap_key = $wpmuBaseTablePrefix . $blog_id . '_capabilities';
if ( is_array($current_user->$cap_key) && in_array(1, $current_user->$cap_key) )
return true;
return false;
}
function validate_email( $email, $check_domain = true) {
if (ereg('^[-!#$%&\'*+\\./0-9=?A-Z^_`a-z{|}~]+'.'@'.
'[-!#$%&\'*+\\/0-9=?A-Z^_`a-z{|}~]+\.'.
'[-!#$%&\'*+\\./0-9=?A-Z^_`a-z{|}~]+$', $email))
{
if ($check_domain && function_exists('checkdnsrr')) {
list (, $domain) = explode('@', $email);
if (checkdnsrr($domain.'.', 'MX') || checkdnsrr($domain.'.', 'A')) {
return true;
}
return false;
}
return true;
}
return false;
}
function is_email_address_unsafe( $user_email ) {
$banned_names = get_site_option( "banned_email_domains" );
if ( is_array( $banned_names ) && empty( $banned_names ) == false ) {
$email_domain = strtolower( substr( $user_email, 1 + strpos( $user_email, '@' ) ) );
foreach( $banned_names as $banned_domain ) {
if( $banned_domain == '' )
continue;
if (
strstr( $email_domain, $banned_domain ) ||
(
strstr( $banned_domain, '/' ) &&
preg_match( $banned_domain, $email_domain )
)
)
return true;
}
}
return false;
}
function wpmu_validate_user_signup($user_name, $user_email) {
global $wpdb, $current_site;
$errors = new WP_Error();
$user_name = sanitize_user($user_name);
$user_email = sanitize_email( $user_email );
if ( empty( $user_name ) )
$errors->add('user_name', __("Please enter a username"));
preg_match( "/[a-z0-9]+/", $user_name, $maybe );
if( $user_name != $maybe[0] ) {
$errors->add('user_name', __("Only lowercase letters and numbers allowed"));
}
$illegal_names = get_site_option( "illegal_names" );
if( is_array( $illegal_names ) == false ) {
$illegal_names = array( "www", "web", "root", "admin", "main", "invite", "administrator" );
add_site_option( "illegal_names", $illegal_names );
}
if( in_array( $user_name, $illegal_names ) == true ) {
$errors->add('user_name', __("That username is not allowed"));
}
if( is_email_address_unsafe( $user_email ) )
$errors->add('user_email', __("You cannot use that email address to signup. We are having problems with them blocking some of our email. Please use another email provider."));
if( strlen( $user_name ) < 4 ) {
$errors->add('user_name', __("Username must be at least 4 characters"));
}
if ( strpos( " " . $user_name, "_" ) != false )
$errors->add('user_name', __("Sorry, usernames may not contain the character '_'!"));
// all numeric?
preg_match( '/[0-9]*/', $user_name, $match );
if ( $match[0] == $user_name )
$errors->add('user_name', __("Sorry, usernames must have letters too!"));
if ( !is_email( $user_email ) )
$errors->add('user_email', __("Please enter a correct email address"));
if ( !validate_email( $user_email ) )
$errors->add('user_email', __("Please check your email address."));
$limited_email_domains = get_site_option( 'limited_email_domains' );
if ( is_array( $limited_email_domains ) && empty( $limited_email_domains ) == false ) {
$emaildomain = substr( $user_email, 1 + strpos( $user_email, '@' ) );
if( in_array( $emaildomain, $limited_email_domains ) == false ) {
$errors->add('user_email', __("Sorry, that email address is not allowed!"));
}
}
// Check if the username has been used already.
if ( username_exists($user_name) )
$errors->add('user_name', __("Sorry, that username already exists!"));
// Check if the email address has been used already.
if ( email_exists($user_email) )
$errors->add('user_email', __("Sorry, that email address is already used!"));
// Has someone already signed up for this username?
$signup = $wpdb->get_row("SELECT * FROM $wpdb->signups WHERE user_login = '$user_name'");
if ( $signup != null ) {
$registered_at = mysql2date('U', $signup->registered);
$now = current_time( 'timestamp', true );
$diff = $now - $registered_at;
// If registered more than two days ago, cancel registration and let this signup go through.
if ( $diff > 172800 ) {
$wpdb->query("DELETE FROM $wpdb->signups WHERE user_login = '$user_name'");
} else {
$errors->add('user_name', __("That username is currently reserved but may be available in a couple of days."));
}
if( $signup->active == 0 && $signup->user_email == $user_email )
$errors->add('user_email_used', __("username and email used"));
}
$signup = $wpdb->get_row("SELECT * FROM $wpdb->signups WHERE user_email = '$user_email'");
if ( $signup != null ) {
$registered_at = mysql2date('U', $signup->registered);
$now = current_time( 'timestamp', true );
$diff = $now - $registered_at;
// If registered more than two days ago, cancel registration and let this signup go through.
if ( $diff > 172800 ) {
$wpdb->query("DELETE FROM $wpdb->signups WHERE user_email = '$user_email'");
} else {
$errors->add('user_email', __("That email address has already been used. Please check your inbox for an activation email. It will become available in a couple of days if you do nothing."));
}
}
$result = array('user_name' => $user_name, 'user_email' => $user_email, 'errors' => $errors);
return apply_filters('wpmu_validate_user_signup', $result);
}
function wpmu_validate_blog_signup($blog_id, $blog_title, $user = '') {
global $wpdb, $domain, $base;
$blog_id = sanitize_user( $blog_id );
$blog_title = strip_tags( $blog_title );
$errors = new WP_Error();
$illegal_names = get_site_option( "illegal_names" );
if( $illegal_names == false ) {
$illegal_names = array( "www", "web", "root", "admin", "main", "invite", "administrator" );
add_site_option( "illegal_names", $illegal_names );
}
if ( empty( $blog_id ) )
$errors->add('blog_id', __("Please enter a blog name"));
preg_match( "/[a-z0-9]+/", $blog_id, $maybe );
if( $blog_id != $maybe[0] ) {
$errors->add('blog_id', __("Only lowercase letters and numbers allowed"));
}
if( in_array( $blog_id, $illegal_names ) == true ) {
$errors->add('blog_id', __("That name is not allowed"));
}
if( strlen( $blog_id ) < 4 && !is_site_admin() ) {
$errors->add('blog_id', __("Blog name must be at least 4 characters"));
}
if ( strpos( " " . $blog_id, "_" ) != false )
$errors->add('blog_id', __("Sorry, blog names may not contain the character '_'!"));
// all numeric?
preg_match( '/[0-9]*/', $blog_id, $match );
if ( $match[0] == $blog_id )
$errors->add('blog_id', __("Sorry, blog names must have letters too!"));
$blog_id = apply_filters( "newblog_id", $blog_id );
$blog_title = stripslashes( $blog_title );
if ( empty( $blog_title ) )
$errors->add('blog_title', __("Please enter a blog title"));
// Check if the domain/path has been used already.
if( constant( "VHOST" ) == 'yes' ) {
$mydomain = "$blog_id.$domain";
$path = $base;
} else {
$mydomain = "$domain";
$path = $base.$blog_id.'/';
}
if ( domain_exists($mydomain, $path) )
$errors->add('blog_id', __("Sorry, that blog already exists!"));
if ( username_exists($blog_id) ) {
if ( !is_object($user) && ( $user->user_login != $blog_id ) )
$errors->add('blog_id', __("Sorry, that blog is reserved!"));
}
// Has someone already signed up for this domain?
// TODO: Check email too?
$signup = $wpdb->get_row("SELECT * FROM $wpdb->signups WHERE domain = '$mydomain' AND path = '$path'");
if ( ! empty($signup) ) {
$registered_at = mysql2date('U', $signup->registered);
$now = current_time( 'timestamp', true );
$diff = $now - $registered_at;
// If registered more than two days ago, cancel registration and let this signup go through.
if ( $diff > 172800 ) {
$wpdb->query("DELETE FROM $wpdb->signups WHERE domain = '$mydomain' AND path = '$path'");
} else {
$errors->add('blog_id', __("That blog is currently reserved but may be available in a couple days."));
}
}
$result = array('domain' => $mydomain, 'path' => $path, 'blog_id' => $blog_id, 'blog_title' => $blog_title,
'errors' => $errors);
return apply_filters('wpmu_validate_blog_signup', $result);
}
// Record signup information for future activation. wpmu_validate_signup() should be run
// on the inputs before calling wpmu_signup().
function wpmu_signup_blog($domain, $path, $title, $user, $user_email, $meta = '') {
global $wpdb;
$key = substr( md5( time() . rand() . $domain ), 0, 16 );
$registered = current_time('mysql', true);
$meta = serialize($meta);
$domain = $wpdb->escape($domain);
$path = $wpdb->escape($path);
$title = $wpdb->escape($title);
$wpdb->query( "INSERT INTO $wpdb->signups ( domain, path, title, user_login, user_email, registered, activation_key, meta )
VALUES ( '$domain', '$path', '$title', '$user', '$user_email', '$registered', '$key', '$meta' )" );
wpmu_signup_blog_notification($domain, $path, $title, $user, $user_email, $key, $meta);
}
function wpmu_signup_user($user, $user_email, $meta = '') {
global $wpdb;
$user = sanitize_user( $user );
$user_email = sanitize_email( $user_email );
$key = substr( md5( time() . rand() . $user_email ), 0, 16 );
$registered = current_time('mysql', true);
$meta = serialize($meta);
$wpdb->query( "INSERT INTO $wpdb->signups ( domain, path, title, user_login, user_email, registered, activation_key, meta )
VALUES ( '', '', '', '$user', '$user_email', '$registered', '$key', '$meta' )" );
wpmu_signup_user_notification($user, $user_email, $key, $meta);
}
// Notify user of signup success.
function wpmu_signup_blog_notification($domain, $path, $title, $user, $user_email, $key, $meta = '') {
global $current_site;
// Send email with activation link.
if( constant( "VHOST" ) == 'no' ) {
$activate_url = "http://" . $current_site->domain . $current_site->path . "wp-activate.php?key=$key";
} else {
$activate_url = "http://{$domain}{$path}wp-activate.php?key=$key";
}
$admin_email = get_site_option( "admin_email" );
if( $admin_email == '' )
$admin_email = 'support@' . $_SERVER[ 'SERVER_NAME' ];
$from_name = get_site_option( "site_name" ) == '' ? 'WordPress' : wp_specialchars( get_site_option( "site_name" ) );
$message_headers = "MIME-Version: 1.0\n" . "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
$message = sprintf(__("To activate your blog, please click the following link:\n\n%s\n\nAfter you activate, you will receive *another email* with your login.\n\nAfter you activate, you can visit your blog here:\n\n%s"), $activate_url, "http://{$domain}{$path}");
// TODO: Don't hard code activation link.
$subject = sprintf(__('Activate %s'), $domain.$path);
wp_mail($user_email, $subject, $message, $message_headers);
}
function wpmu_signup_user_notification($user, $user_email, $key, $meta = '') {
global $current_site;
// Send email with activation link.
$admin_email = get_site_option( "admin_email" );
if( $admin_email == '' )
$admin_email = 'support@' . $_SERVER[ 'SERVER_NAME' ];
$from_name = get_site_option( "site_name" ) == '' ? 'WordPress' : wp_specialchars( get_site_option( "site_name" ) );
$message_headers = "MIME-Version: 1.0\n" . "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
$message = sprintf(__("To activate your user, please click the following link:\n\n%s\n\nAfter you activate, you will receive *another email* with your login.\n\n"), "http://{$current_site->domain}{$current_site->path}wp-activate.php?key=$key" );
// TODO: Don't hard code activation link.
$subject = sprintf(__('Activate %s'), $user);
wp_mail($user_email, $subject, $message, $message_headers);
}
function wpmu_activate_signup($key) {
global $wpdb;
$result = array();
$signup = $wpdb->get_row("SELECT * FROM $wpdb->signups WHERE activation_key = '$key'");
if ( empty($signup) )
return new WP_Error('invalid_key', __('Invalid activation key.'));
if ( $signup->active )
return new WP_Error('already_active', __('The blog is already active.'), $signup);
$meta = unserialize($signup->meta);
$user_login = $wpdb->escape($signup->user_login);
$user_email = $wpdb->escape($signup->user_email);
wpmu_validate_user_signup($user_login, $user_email);
$password = generate_random_password();
$user_id = username_exists($user_login);
if ( ! $user_id )
$user_id = wpmu_create_user($user_login, $password, $user_email);
else
$user_already_exists = true;
if ( ! $user_id )
return new WP_Error('create_user', __('Could not create user'), $signup);
$now = current_time('mysql', true);
if ( empty($signup->domain) ) {
$wpdb->query("UPDATE $wpdb->signups SET active = '1', activated = '$now' WHERE activation_key = '$key'");
if ( isset($user_already_exists) )
return new WP_Error('user_already_exists', __('That username is already activated.'), $signup);
wpmu_welcome_user_notification($user_id, $password, $meta);
add_user_to_blog('1', $user_id, 'subscriber');
do_action('wpmu_activate_user', $user_id, $password, $meta);
return array('user_id' => $user_id, 'password' => $password, 'meta' => $meta);
}
wpmu_validate_blog_signup($signup->domain, $signup->title);
$blog_id = wpmu_create_blog($signup->domain, $signup->path, $signup->title, $user_id, $meta);
// TODO: What to do if we create a user but cannot create a blog?
if ( is_wp_error($blog_id) ) {
// If blog is taken, that means a previous attempt to activate this blog failed in between creating the blog and
// setting the activation flag. Let's just set the active flag and instruct the user to reset their password.
if ( 'blog_taken' == $blog_id->get_error_code() ) {
$blog_id->add_data($signup);
$wpdb->query("UPDATE $wpdb->signups SET active = '1', activated = '$now' WHERE activation_key = '$key'");
error_log("Blog $blog_id failed to complete activation.", 0);
}
return $blog_id;
}
$wpdb->query("UPDATE $wpdb->signups SET active = '1', activated = '$now' WHERE activation_key = '$key'");
wpmu_welcome_notification($blog_id, $user_id, $password, $signup->title, $meta);
do_action('wpmu_activate_blog', $blog_id, $user_id, $password, $signup->title, $meta);
return array('blog_id' => $blog_id, 'user_id' => $user_id, 'password' => $password, 'title' => $signup->title, 'meta' => $meta);
}
function generate_random_password() {
$random_password = substr(md5(uniqid(microtime())), 0, 6);
$random_password = apply_filters('random_password', $random_password);
return $random_password;
}
function wpmu_create_user( $user_name, $password, $email) {
if ( username_exists($user_name) )
return false;
// Check if the email address has been used already.
if ( email_exists($email) )
return false;
$user_id = wp_create_user( $user_name, $password, $email );
$user = new WP_User($user_id);
// Newly created users have no roles or caps until they are added to a blog.
update_usermeta($user_id, 'capabilities', '');
update_usermeta($user_id, 'user_level', '');
do_action( 'wpmu_new_user', $user_id );
return $user_id;
}
function wpmu_create_blog($domain, $path, $title, $user_id, $meta = '', $site_id = 1) {
$domain = sanitize_user( $domain );
$title = strip_tags( $title );
$user_id = (int) $user_id;
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 new WP_Error('blog_taken', __('Blog already exists.'));
if ( !defined("WP_INSTALLING") )
define( "WP_INSTALLING", true );
if ( ! $blog_id = insert_blog($domain, $path, $site_id) )
return new WP_Error('insert_blog', __('Could not create blog.'));
switch_to_blog($blog_id);
install_blog($blog_id, $title);
install_blog_defaults($blog_id, $user_id);
add_user_to_blog($blog_id, $user_id, 'administrator');
restore_current_blog();
if ( is_array($meta) ) foreach ($meta as $key => $value) {
update_blog_status( $blog_id, $key, $value );
update_blog_option( $blog_id, $key, $value );
}
if(get_usermeta( $user_id, 'primary_blog' ) == 1 )
update_usermeta( $user_id, 'primary_blog', $blog_id );
do_action( 'wpmu_new_blog', $blog_id, $user_id );
return $blog_id;
}
function newblog_notify_siteadmin( $blog_id, $user_id ) {
global $current_site;
if( get_site_option( 'registrationnotification' ) != 'yes' )
return;
$email = get_site_option( 'admin_email' );
if( is_email( $email ) == false )
return false;
$msg = "New Blog: " . get_blog_option( $blog_id, "blogname" ) . "\nURL: " . get_blog_option( $blog_id, "siteurl" ) . "\nRemote IP: {$_SERVER[ 'REMOTE_ADDR' ]}\n\nDisable these notifications: http://{$current_site->domain}{$current_site->path}wp-admin/wpmu-options.php";
$msg = apply_filters( 'newblog_notify_siteadmin', $msg );
wp_mail( $email, "New Blog Registration: " . get_blog_option( $blog_id, "siteurl" ), $msg );
}
add_action( "wpmu_new_blog", "newblog_notify_siteadmin", 10, 2 );
function newuser_notify_siteadmin( $user_id ) {
global $current_site;
if( get_site_option( 'registrationnotification' ) != 'yes' )
return;
$email = get_site_option( 'admin_email' );
if( is_email( $email ) == false )
return false;
$user = new WP_User($user_id);
$msg = "New User: " . $user->user_login . "\nRemote IP: {$_SERVER[ 'REMOTE_ADDR' ]}\n\nDisable these notifications: http://{$current_site->domain}{$current_site->path}wp-admin/wpmu-options.php";
$msg = apply_filters( 'newuser_notify_siteadmin', $msg );
wp_mail( $email, "New User Registration: " . $user->user_login, $msg );
}
add_action( "wpmu_new_user", "newuser_notify_siteadmin" );
function domain_exists($domain, $path, $site_id = 1) {
global $wpdb;
return $wpdb->get_var("SELECT blog_id FROM $wpdb->blogs WHERE domain = '$domain' AND path = '$path' AND site_id = '$site_id'" );
}
function insert_blog($domain, $path, $site_id) {
global $wpdb;
$path = trailingslashit( $path );
$query = "INSERT INTO $wpdb->blogs ( blog_id, site_id, domain, path, registered ) VALUES ( NULL, '$site_id', '$domain', '$path', NOW( ))";
$result = $wpdb->query( $query );
if ( ! $result )
return false;
$id = $wpdb->insert_id;
refresh_blog_details($id);
return $id;
}
// Install an empty blog. wpdb should already be switched.
function install_blog($blog_id, $blog_title = '') {
global $wpdb, $table_prefix, $wp_roles;
$wpdb->hide_errors();
require_once( ABSPATH . 'wp-admin/includes/upgrade.php');
$installed = $wpdb->get_results("SELECT * FROM $wpdb->posts");
if ($installed) die(__('Already Installed
You appear to have already installed WordPress. To reinstall please clear your old database tables first.
') . '