wp-includes/ms-functions.php

   1 <?php
   2 /**
   3  * Multi-site WordPress API
   4  *
   5  * @package WordPress
   6  * @subpackage Multisite
   7  * @since 3.0.0
   8  */
   9
  10 function get_sitestats() {
  11         global $wpdb;
  12
  13         $stats['blogs'] = get_blog_count();
  14
  15         $count_ts = get_site_option( 'user_count_ts' );
  16         if ( time() - $count_ts > 3600 ) {
  17                 $count = $wpdb->get_var( "SELECT COUNT(ID) FROM $wpdb->users" );
  18                 update_site_option( 'user_count', $count );
  19                 update_site_option( 'user_count_ts', time() );
  20         } else {
  21                 $count = get_site_option( 'user_count' );
  22         }
  23         $stats['users'] = $count;
  24         return $stats;
  25 }
  26
  27 function get_admin_users_for_domain( $sitedomain = '', $path = '' ) {
  28         global $wpdb;
  29
  30         if ( ! $sitedomain )
  31                 $site_id = $wpdb->siteid;
  32         else
  33                 $site_id = $wpdb->get_var( $wpdb->prepare( "SELECT id FROM $wpdb->site WHERE domain = %s AND path = %s", $sitedomain, $path ) );
  34
  35         if ( $site_id )
  36                 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 );
  37
  38         return false;
  39 }
  40
  41 function get_blogs_of_user( $id, $all = false ) {
  42         global $wpdb;
  43
  44         $cache_suffix = $all ? '_all' : '_short';
  45         $return = wp_cache_get( 'blogs_of_user_' . $id . $cache_suffix, 'users' );
  46         if ( $return )
  47                 return apply_filters( 'get_blogs_of_user', $return, $id, $all );
  48
  49         $user = get_userdata( (int) $id );
  50         if ( !$user )
  51                 return false;
  52
  53         $blogs = $match = array();
  54         $prefix_length = strlen($wpdb->base_prefix);
  55         foreach ( (array) $user as $key => $value ) {
  56                 if ( $prefix_length && substr($key, 0, $prefix_length) != $wpdb->base_prefix )
  57                         continue;
  58                 if ( substr($key, -12, 12) != 'capabilities' )
  59                         continue;
  60                 if ( preg_match( '/^' . $wpdb->base_prefix . '((\d+)_)?capabilities$/', $key, $match ) ) {
  61                         if ( count( $match ) > 2 )
  62                                 $blog_id = $match[ 2 ];
  63                         else
  64                                 $blog_id = 1;
  65                         $blog = get_blog_details( $blog_id );
  66                         if ( $blog && isset( $blog->domain ) && ( $all == true || $all == false && ( $blog->archived == 0 && $blog->spam == 0 && $blog->deleted == 0 ) ) ) {
  67                                 $blogs[ $blog_id ]->userblog_id = $blog_id;
  68                                 $blogs[ $blog_id ]->blogname            = $blog->blogname;
  69                                 $blogs[ $blog_id ]->domain              = $blog->domain;
  70                                 $blogs[ $blog_id ]->path                        = $blog->path;
  71                                 $blogs[ $blog_id ]->site_id             = $blog->site_id;
  72                                 $blogs[ $blog_id ]->siteurl             = $blog->siteurl;
  73                         }
  74                 }
  75         }
  76
  77         wp_cache_add( 'blogs_of_user_' . $id . $cache_suffix, $blogs, 'users', 5 );
  78         return apply_filters( 'get_blogs_of_user', $blogs, $id, $all );
  79 }
  80
  81 function get_active_blog_for_user( $user_id ) { // get an active blog for user - either primary blog or from blogs list
  82         global $wpdb;
  83         $blogs = get_blogs_of_user( $user_id );
  84         if ( empty( $blogs ) ) {
  85                 $details = get_dashboard_blog();
  86                 add_user_to_blog( $details->blog_id, $user_id, 'subscriber' );
  87                 update_user_meta( $user_id, 'primary_blog', $details->blog_id );
  88                 wp_cache_delete( $user_id, 'users' );
  89                 return $details;
  90         }
  91
  92         $primary_blog = get_user_meta( $user_id, 'primary_blog', true );
  93         $details = get_dashboard_blog();
  94         if ( $primary_blog ) {
  95                 $blogs = get_blogs_of_user( $user_id );
  96                 if ( isset( $blogs[ $primary_blog ] ) == false ) {
  97                         add_user_to_blog( $details->blog_id, $user_id, 'subscriber' );
  98                         update_user_meta( $user_id, 'primary_blog', $details->blog_id );
  99                         wp_cache_delete( $user_id, 'users' );
 100                 } else {
 101                         $details = get_blog_details( $primary_blog );
 102                 }
 103         } else {
 104                 add_user_to_blog( $details->blog_id, $user_id, 'subscriber' ); // Add subscriber permission for dashboard blog
 105                 update_user_meta( $user_id, 'primary_blog', $details->blog_id );
 106         }
 107
 108         if ( ( is_object( $details ) == false ) || ( is_object( $details ) && $details->archived == 1 || $details->spam == 1 || $details->deleted == 1 ) ) {
 109                 $blogs = get_blogs_of_user( $user_id, true ); // if a user's primary blog is shut down, check their other blogs.
 110                 $ret = false;
 111                 if ( is_array( $blogs ) && count( $blogs ) > 0 ) {
 112                         foreach ( (array) $blogs as $blog_id => $blog ) {
 113                                 if ( $blog->site_id != $wpdb->siteid )
 114                                         continue;
 115                                 $details = get_blog_details( $blog_id );
 116                                 if ( is_object( $details ) && $details->archived == 0 && $details->spam == 0 && $details->deleted == 0 ) {
 117                                         $ret = $blog;
 118                                         $changed = false;
 119                                         if ( get_user_meta( $user_id , 'primary_blog', true ) != $blog_id ) {
 120                                                 update_user_meta( $user_id, 'primary_blog', $blog_id );
 121                                                 $changed = true;
 122                                         }
 123                                         if ( !get_user_meta($user_id , 'source_domain', true) ) {
 124                                                 update_user_meta( $user_id, 'source_domain', $blog->domain );
 125                                                 $changed = true;
 126                                         }
 127                                         if ( $changed )
 128                                                 wp_cache_delete( $user_id, 'users' );
 129                                         break;
 130                                 }
 131                         }
 132                 } else {
 133                         // Should never get here
 134                         $dashboard_blog = get_dashboard_blog();
 135                         add_user_to_blog( $dashboard_blog->blog_id, $user_id, 'subscriber' ); // Add subscriber permission for dashboard blog
 136                         update_user_meta( $user_id, 'primary_blog', $dashboard_blog->blog_id );
 137                         return $dashboard_blog;
 138                 }
 139                 return $ret;
 140         } else {
 141                 return $details;
 142         }
 143 }
 144
 145 function is_user_member_of_blog( $user_id, $blog_id = 0 ) {
 146         $user_id = (int) $user_id;
 147         $blog_id = (int) $blog_id;
 148
 149         if ( $blog_id == 0 ) {
 150                 global $wpdb;
 151                 $blog_id = $wpdb->blogid;
 152         }
 153
 154         $blogs = get_blogs_of_user( $user_id );
 155         if ( is_array( $blogs ) )
 156                 return array_key_exists( $blog_id, $blogs );
 157         else
 158                 return false;
 159 }
 160
 161 function get_user_count() {
 162         global $wpdb;
 163
 164         $count_ts = get_site_option( 'user_count_ts' );
 165         if ( time() - $count_ts > 3600 ) {
 166                 $count = $wpdb->get_var( $wpdb->prepare("SELECT COUNT(ID) as c FROM $wpdb->users WHERE spam = '0' AND deleted = '0'") );
 167                 update_site_option( 'user_count', $count );
 168                 update_site_option( 'user_count_ts', time() );
 169         }
 170
 171         $count = get_site_option( 'user_count' );
 172
 173         return $count;
 174 }
 175
 176 function get_blog_count( $id = 0 ) {
 177         global $wpdb;
 178
 179         if ( $id == 0 )
 180                 $id = $wpdb->siteid;
 181
 182         $count_ts = get_site_option( 'blog_count_ts' );
 183         if ( time() - $count_ts > 3600 ) {
 184                 $count = $wpdb->get_var( $wpdb->prepare("SELECT COUNT(blog_id) as c FROM $wpdb->blogs WHERE site_id = %d AND spam = '0' AND deleted = '0' and archived = '0'", $id) );
 185                 update_site_option( 'blog_count', $count );
 186                 update_site_option( 'blog_count_ts', time() );
 187         }
 188
 189         $count = get_site_option( 'blog_count' );
 190
 191         return $count;
 192 }
 193
 194 function get_blog_post( $blog_id, $post_id ) {
 195         global $wpdb;
 196
 197         $key = $blog_id . '-' . $post_id;
 198         $post = wp_cache_get( $key, 'global-posts' );
 199         if ( $post == false ) {
 200                 $post = $wpdb->get_row( $wpdb->prepare( 'SELECT * FROM ' . $wpdb->get_blog_prefix( $blog_id ) . 'posts WHERE ID = %d', $post_id ) );
 201                 wp_cache_add( $key, $post, 'global-posts' );
 202         }
 203
 204         return $post;
 205 }
 206
 207 function add_user_to_blog( $blog_id, $user_id, $role ) {
 208         switch_to_blog($blog_id);
 209
 210         $user = new WP_User($user_id);
 211
 212         if ( empty( $user->ID ) )
 213                 return new WP_Error('user_does_not_exist', __('That user does not exist.'));
 214
 215         if ( !get_user_meta($user_id, 'primary_blog', true) ) {
 216                 update_user_meta($user_id, 'primary_blog', $blog_id);
 217                 $details = get_blog_details($blog_id);
 218                 update_user_meta($user_id, 'source_domain', $details->domain);
 219         }
 220
 221         $user->set_role($role);
 222
 223         do_action('add_user_to_blog', $user_id, $role, $blog_id);
 224         wp_cache_delete( $user_id, 'users' );
 225         restore_current_blog();
 226         return true;
 227 }
 228
 229 function remove_user_from_blog($user_id, $blog_id = '', $reassign = '') {
 230         global $wpdb;
 231         switch_to_blog($blog_id);
 232         $user_id = (int) $user_id;
 233         do_action('remove_user_from_blog', $user_id, $blog_id);
 234
 235         // If being removed from the primary blog, set a new primary if the user is assigned
 236         // to multiple blogs.
 237         $primary_blog = get_user_meta($user_id, 'primary_blog', true);
 238         if ( $primary_blog == $blog_id ) {
 239                 $new_id = '';
 240                 $new_domain = '';
 241                 $blogs = get_blogs_of_user($user_id);
 242                 foreach ( (array) $blogs as $blog ) {
 243                         if ( $blog->userblog_id == $blog_id )
 244                                 continue;
 245                         $new_id = $blog->userblog_id;
 246                         $new_domain = $blog->domain;
 247                         break;
 248                 }
 249
 250                 update_user_meta($user_id, 'primary_blog', $new_id);
 251                 update_user_meta($user_id, 'source_domain', $new_domain);
 252         }
 253
 254         // wp_revoke_user($user_id);
 255         $user = new WP_User($user_id);
 256         if ( empty( $user->ID ) )
 257                 return new WP_Error('user_does_not_exist', __('That user does not exist.'));
 258
 259         $user->remove_all_caps();
 260
 261         $blogs = get_blogs_of_user($user_id);
 262         if ( count($blogs) == 0 ) {
 263                 update_user_meta($user_id, 'primary_blog', '');
 264                 update_user_meta($user_id, 'source_domain', '');
 265         }
 266
 267         if ( $reassign != '' ) {
 268                 $reassign = (int) $reassign;
 269                 $wpdb->query( $wpdb->prepare("UPDATE $wpdb->posts SET post_author = %d WHERE post_author = %d", $reassign, $user_id) );
 270                 $wpdb->query( $wpdb->prepare("UPDATE $wpdb->links SET link_owner = %d WHERE link_owner = %d", $reassign, $user_id) );
 271         }
 272
 273         restore_current_blog();
 274 }
 275
 276 function create_empty_blog( $domain, $path, $weblog_title, $site_id = 1 ) {
 277         $domain                 = addslashes( $domain );
 278         $weblog_title   = addslashes( $weblog_title );
 279
 280         if ( empty($path) )
 281                 $path = '/';
 282
 283         // Check if the domain has been used already. We should return an error message.
 284         if ( domain_exists($domain, $path, $site_id) )
 285                 return __( 'Error: Site URL already taken.' );
 286
 287         // Need to backup wpdb table names, and create a new wp_blogs entry for new blog.
 288         // Need to get blog_id from wp_blogs, and create new table names.
 289         // Must restore table names at the end of function.
 290
 291         if ( ! $blog_id = insert_blog($domain, $path, $site_id) )
 292                 return __( 'Error: problem creating site entry.' );
 293
 294         switch_to_blog($blog_id);
 295         install_blog($blog_id);
 296         restore_current_blog();
 297
 298         return $blog_id;
 299 }
 300
 301 function get_blog_permalink( $_blog_id, $post_id ) {
 302         $key = "{$_blog_id}-{$post_id}-blog_permalink";
 303         $link = wp_cache_get( $key, 'site-options' );
 304         if ( $link == false ) {
 305                 switch_to_blog( $_blog_id );
 306                 $link = get_permalink( $post_id );
 307                 restore_current_blog();
 308                 wp_cache_add( $key, $link, 'site-options', 360 );
 309         }
 310         return $link;
 311 }
 312
 313 function get_blog_id_from_url( $domain, $path = '/' ) {
 314         global $wpdb;
 315
 316         $domain = strtolower( $wpdb->escape( $domain ) );
 317         $path = strtolower( $wpdb->escape( $path ) );
 318         $id = wp_cache_get( md5( $domain . $path ), 'blog-id-cache' );
 319
 320         if ( $id == -1 ) { // blog does not exist
 321                 return 0;
 322         } elseif ( $id ) {
 323                 return (int)$id;
 324         }
 325
 326         $id = $wpdb->get_var( "SELECT blog_id FROM $wpdb->blogs WHERE domain = '$domain' and path = '$path' /* get_blog_id_from_url */" );
 327
 328         if ( !$id ) {
 329                 wp_cache_set( md5( $domain . $path ), -1, 'blog-id-cache' );
 330                 return false;
 331         }
 332         wp_cache_set( md5( $domain . $path ), $id, 'blog-id-cache' );
 333
 334         return $id;
 335 }
 336
 337 // wpmu admin functions
 338
 339 function wpmu_admin_do_redirect( $url = '' ) {
 340         $ref = '';
 341         if ( isset( $_GET['ref'] ) )
 342                 $ref = $_GET['ref'];
 343         if ( isset( $_POST['ref'] ) )
 344                 $ref = $_POST['ref'];
 345
 346         if ( $ref ) {
 347                 $ref = wpmu_admin_redirect_add_updated_param( $ref );
 348                 wp_redirect( $ref );
 349                 exit();
 350         }
 351         if ( empty( $_SERVER['HTTP_REFERER'] ) == false ) {
 352                 wp_redirect( $_SERVER['HTTP_REFERER'] );
 353                 exit();
 354         }
 355
 356         $url = wpmu_admin_redirect_add_updated_param( $url );
 357         if ( isset( $_GET['redirect'] ) ) {
 358                 if ( substr( $_GET['redirect'], 0, 2 ) == 's_' )
 359                         $url .= '&action=blogs&s='. esc_html( substr( $_GET['redirect'], 2 ) );
 360         } elseif ( isset( $_POST['redirect'] ) ) {
 361                 $url = wpmu_admin_redirect_add_updated_param( $_POST['redirect'] );
 362         }
 363         wp_redirect( $url );
 364         exit();
 365 }
 366
 367 function wpmu_admin_redirect_add_updated_param( $url = '' ) {
 368         if ( strpos( $url, 'updated=true' ) === false ) {
 369                 if ( strpos( $url, '?' ) === false )
 370                         return $url . '?updated=true';
 371                 else
 372                         return $url . '&updated=true';
 373         }
 374         return $url;
 375 }
 376
 377 function is_blog_user( $blog_id = 0 ) {
 378         global $wpdb;
 379
 380         $current_user = wp_get_current_user();
 381         if ( !$blog_id )
 382                 $blog_id = $wpdb->blogid;
 383
 384         $cap_key = $wpdb->base_prefix . $blog_id . '_capabilities';
 385
 386         if ( is_array($current_user->$cap_key) && in_array(1, $current_user->$cap_key) )
 387                 return true;
 388
 389         return false;
 390 }
 391
 392 function is_email_address_unsafe( $user_email ) {
 393         $banned_names = get_site_option( 'banned_email_domains' );
 394         if ($banned_names && !is_array( $banned_names ))
 395                 $banned_names = explode( "\n", $banned_names);
 396
 397         if ( is_array( $banned_names ) && empty( $banned_names ) == false ) {
 398                 $email_domain = strtolower( substr( $user_email, 1 + strpos( $user_email, '@' ) ) );
 399                 foreach ( (array) $banned_names as $banned_domain ) {
 400                         if ( $banned_domain == '' )
 401                                 continue;
 402                         if (
 403                                 strstr( $email_domain, $banned_domain ) ||
 404                                 (
 405                                         strstr( $banned_domain, '/' ) &&
 406                                         preg_match( $banned_domain, $email_domain )
 407                                 )
 408                         )
 409                         return true;
 410                 }
 411         }
 412         return false;
 413 }
 414
 415 function wpmu_validate_user_signup($user_name, $user_email) {
 416         global $wpdb;
 417
 418         $errors = new WP_Error();
 419
 420         $orig_username = $user_name;
 421         $user_name = preg_replace( '/\s+/', '', sanitize_user( $user_name, true ) );
 422         $maybe = array();
 423         preg_match( '/[a-z0-9]+/', $user_name, $maybe );
 424
 425         if ( $user_name != $orig_username || $user_name != $maybe[0] ) {
 426                 $errors->add( 'user_name', __( "Only the lowercase letters a-z and numbers allowed" ) );
 427                 $user_name = $orig_username;
 428         }
 429
 430         $user_email = sanitize_email( $user_email );
 431
 432         if ( empty( $user_name ) )
 433                 $errors->add('user_name', __('Please enter a username'));
 434
 435         $illegal_names = get_site_option( 'illegal_names' );
 436         if ( is_array( $illegal_names ) == false ) {
 437                 $illegal_names = array(  'www', 'web', 'root', 'admin', 'main', 'invite', 'administrator' );
 438                 add_site_option( 'illegal_names', $illegal_names );
 439         }
 440         if ( in_array( $user_name, $illegal_names ) == true )
 441                 $errors->add('user_name',  __('That username is not allowed'));
 442
 443         if ( is_email_address_unsafe( $user_email ) )
 444                 $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.'));
 445
 446         if ( strlen( $user_name ) < 4 )
 447                 $errors->add('user_name',  __('Username must be at least 4 characters'));
 448
 449         if ( strpos( ' ' . $user_name, '_' ) != false )
 450                 $errors->add( 'user_name', __( 'Sorry, usernames may not contain the character &#8220;_&#8221;!' ) );
 451
 452         // all numeric?
 453         $match = array();
 454         preg_match( '/[0-9]*/', $user_name, $match );
 455         if ( $match[0] == $user_name )
 456                 $errors->add('user_name', __('Sorry, usernames must have letters too!'));
 457
 458         if ( !is_email( $user_email ) )
 459                 $errors->add('user_email', __('Please enter a correct email address'));
 460
 461         $limited_email_domains = get_site_option( 'limited_email_domains' );
 462         if ( is_array( $limited_email_domains ) && empty( $limited_email_domains ) == false ) {
 463                 $emaildomain = substr( $user_email, 1 + strpos( $user_email, '@' ) );
 464                 if ( in_array( $emaildomain, $limited_email_domains ) == false )
 465                         $errors->add('user_email', __('Sorry, that email address is not allowed!'));
 466         }
 467
 468         // Check if the username has been used already.
 469         if ( username_exists($user_name) )
 470                 $errors->add('user_name', __('Sorry, that username already exists!'));
 471
 472         // Check if the email address has been used already.
 473         if ( email_exists($user_email) )
 474                 $errors->add('user_email', __('Sorry, that email address is already used!'));
 475
 476         // Has someone already signed up for this username?
 477         $signup = $wpdb->get_row( $wpdb->prepare("SELECT * FROM $wpdb->signups WHERE user_login = %s", $user_name) );
 478         if ( $signup != null ) {
 479                 $registered_at =  mysql2date('U', $signup->registered);
 480                 $now = current_time( 'timestamp', true );
 481                 $diff = $now - $registered_at;
 482                 // If registered more than two days ago, cancel registration and let this signup go through.
 483                 if ( $diff > 172800 )
 484                         $wpdb->query( $wpdb->prepare("DELETE FROM $wpdb->signups WHERE user_login = %s", $user_name) );
 485                 else
 486                         $errors->add('user_name', __('That username is currently reserved but may be available in a couple of days.'));
 487
 488                 if ( $signup->active == 0 && $signup->user_email == $user_email )
 489                         $errors->add('user_email_used', __('username and email used'));
 490         }
 491
 492         $signup = $wpdb->get_row( $wpdb->prepare("SELECT * FROM $wpdb->signups WHERE user_email = %s", $user_email) );
 493         if ( $signup != null ) {
 494                 $diff = current_time( 'timestamp', true ) - mysql2date('U', $signup->registered);
 495                 // If registered more than two days ago, cancel registration and let this signup go through.
 496                 if ( $diff > 172800 )
 497                         $wpdb->query( $wpdb->prepare("DELETE FROM $wpdb->signups WHERE user_email = %s", $user_email) );
 498                 else
 499                         $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.'));
 500         }
 501
 502         $result = array('user_name' => $user_name, 'orig_username' => $orig_username, 'user_email' => $user_email, 'errors' => $errors);
 503
 504         return apply_filters('wpmu_validate_user_signup', $result);
 505 }
 506
 507 function wpmu_validate_blog_signup($blogname, $blog_title, $user = '') {
 508         global $wpdb, $domain, $base, $current_site;
 509
 510         $blog_title = strip_tags( $blog_title );
 511         $blog_title = substr( $blog_title, 0, 50 );
 512
 513         $errors = new WP_Error();
 514         $illegal_names = get_site_option( 'illegal_names' );
 515         if ( $illegal_names == false ) {
 516                 $illegal_names = array( 'www', 'web', 'root', 'admin', 'main', 'invite', 'administrator' );
 517                 add_site_option( 'illegal_names', $illegal_names );
 518         }
 519
 520         // On sub dir installs, Some names are so illegal, only a filter can spring them from jail
 521         if (! is_subdomain_install() )
 522                 $illegal_names = array_merge($illegal_names, apply_filters( 'subdirectory_reserved_names', array( 'page', 'comments', 'blog', 'files', 'feed' ) ) );
 523
 524
 525         if ( empty( $blogname ) )
 526                 $errors->add('blogname', __('Please enter a site name'));
 527
 528         $maybe = array();
 529         preg_match( '/[a-z0-9]+/', $blogname, $maybe );
 530         if ( $blogname != $maybe[0] )
 531                 $errors->add('blogname', __('Only lowercase letters and numbers allowed'));
 532
 533         if ( in_array( $blogname, $illegal_names ) == true )
 534                 $errors->add('blogname',  __('That name is not allowed'));
 535
 536         if ( strlen( $blogname ) < 4 && !is_super_admin() )
 537                 $errors->add('blogname',  __('Site name must be at least 4 characters'));
 538
 539         if ( strpos( ' ' . $blogname, '_' ) != false )
 540                 $errors->add( 'blogname', __( 'Sorry, site names may not contain the character &#8220;_&#8221;!' ) );
 541
 542         // do not allow users to create a blog that conflicts with a page on the main blog.
 543         if ( !is_subdomain_install() && $wpdb->get_var( $wpdb->prepare( "SELECT post_name FROM " . $wpdb->get_blog_prefix( $current_site->blog_id ) . "posts WHERE post_type = 'page' AND post_name = %s", $blogname ) ) )
 544                 $errors->add( 'blogname', __( 'Sorry, you may not use that site name.' ) );
 545
 546         // all numeric?
 547         $match = array();
 548         preg_match( '/[0-9]*/', $blogname, $match );
 549         if ( $match[0] == $blogname )
 550                 $errors->add('blogname', __('Sorry, site names must have letters too!'));
 551
 552         $blogname = apply_filters( 'newblogname', $blogname );
 553
 554         $blog_title = stripslashes(  $blog_title );
 555
 556         if ( empty( $blog_title ) )
 557                 $errors->add('blog_title', __('Please enter a site title'));
 558
 559         // Check if the domain/path has been used already.
 560         if ( is_subdomain_install() ) {
 561                 $mydomain = $blogname . '.' . preg_replace( '|^www\.|', '', $domain );
 562                 $path = $base;
 563         } else {
 564                 $mydomain = "$domain";
 565                 $path = $base.$blogname.'/';
 566         }
 567         if ( domain_exists($mydomain, $path) )
 568                 $errors->add('blogname', __('Sorry, that site already exists!'));
 569
 570         if ( username_exists( $blogname ) ) {
 571                 if ( is_object( $user ) == false || ( is_object($user) && ( $user->user_login != $blogname ) ) )
 572                         $errors->add( 'blogname', __( 'Sorry, that site is reserved!' ) );
 573         }
 574
 575         // Has someone already signed up for this domain?
 576         $signup = $wpdb->get_row( $wpdb->prepare("SELECT * FROM $wpdb->signups WHERE domain = %s AND path = %s", $mydomain, $path) ); // TODO: Check email too?
 577         if ( ! empty($signup) ) {
 578                 $diff = current_time( 'timestamp', true ) - mysql2date('U', $signup->registered);
 579                 // If registered more than two days ago, cancel registration and let this signup go through.
 580                 if ( $diff > 172800 )
 581                         $wpdb->query( $wpdb->prepare("DELETE FROM $wpdb->signups WHERE domain = %s AND path = %s", $mydomain, $path) );
 582                 else
 583                         $errors->add('blogname', __('That site is currently reserved but may be available in a couple days.'));
 584         }
 585
 586         $result = array('domain' => $mydomain, 'path' => $path, 'blogname' => $blogname, 'blog_title' => $blog_title, 'errors' => $errors);
 587         return apply_filters('wpmu_validate_blog_signup', $result);
 588 }
 589
 590 // Record signup information for future activation. wpmu_validate_signup() should be run
 591 // on the inputs before calling wpmu_signup().
 592 function wpmu_signup_blog($domain, $path, $title, $user, $user_email, $meta = '') {
 593         global $wpdb;
 594
 595         $key = substr( md5( time() . rand() . $domain ), 0, 16 );
 596         $meta = serialize($meta);
 597         $domain = $wpdb->escape($domain);
 598         $path = $wpdb->escape($path);
 599         $title = $wpdb->escape($title);
 600
 601         $wpdb->insert( $wpdb->signups, array(
 602                 'domain' => $domain,
 603                 'path' => $path,
 604                 'title' => $title,
 605                 'user_login' => $user,
 606                 'user_email' => $user_email,
 607                 'registered' => current_time('mysql', true),
 608                 'activation_key' => $key,
 609                 'meta' => $meta
 610         ) );
 611
 612         wpmu_signup_blog_notification($domain, $path, $title, $user, $user_email, $key, $meta);
 613 }
 614
 615 function wpmu_signup_user($user, $user_email, $meta = '') {
 616         global $wpdb;
 617
 618         // Format data
 619         $user = preg_replace( '/\s+/', '', sanitize_user( $user, true ) );
 620         $user_email = sanitize_email( $user_email );
 621         $key = substr( md5( time() . rand() . $user_email ), 0, 16 );
 622         $meta = serialize($meta);
 623
 624         $wpdb->insert( $wpdb->signups, array(
 625                 'domain' => '',
 626                 'path' => '',
 627                 'title' => '',
 628                 'user_login' => $user,
 629                 'user_email' => $user_email,
 630                 'registered' => current_time('mysql', true),
 631                 'activation_key' => $key,
 632                 'meta' => $meta
 633         ) );
 634
 635         wpmu_signup_user_notification($user, $user_email, $key, $meta);
 636 }
 637
 638 // Notify user of signup success.
 639 function wpmu_signup_blog_notification($domain, $path, $title, $user, $user_email, $key, $meta = '') {
 640         global $current_site;
 641
 642         if ( !apply_filters('wpmu_signup_blog_notification', $domain, $path, $title, $user, $user_email, $key, $meta) )
 643                 return false;
 644
 645         // Send email with activation link.
 646         if ( !is_subdomain_install() || $current_site->id != 1 )
 647                 $activate_url = network_site_url("wp-activate.php?key=$key");
 648         else
 649                 $activate_url = "http://{$domain}{$path}wp-activate.php?key=$key"; // @todo use *_url() API
 650
 651         $activate_url = esc_url($activate_url);
 652         $admin_email = get_site_option( 'admin_email' );
 653         if ( $admin_email == '' )
 654                 $admin_email = 'support@' . $_SERVER['SERVER_NAME'];
 655         $from_name = get_site_option( 'site_name' ) == '' ? 'WordPress' : esc_html( get_site_option( 'site_name' ) );
 656         $message_headers = "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
 657         $message = sprintf( apply_filters( 'wpmu_signup_blog_notification_email', __( "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 site here:\n\n%s" ) ), $activate_url, esc_url( "http://{$domain}{$path}" ), $key );
 658         // TODO: Don't hard code activation link.
 659         $subject = sprintf( apply_filters( 'wpmu_signup_blog_notification_subject', __( '[%1$s] Activate %2$s' ) ), $from_name, esc_url( 'http://' . $domain . $path ) );
 660         wp_mail($user_email, $subject, $message, $message_headers);
 661         return true;
 662 }
 663
 664 function wpmu_signup_user_notification($user, $user_email, $key, $meta = '') {
 665         if ( !apply_filters('wpmu_signup_user_notification', $user, $user_email, $key, $meta) )
 666                 return false;
 667
 668         // Send email with activation link.
 669         $admin_email = get_site_option( 'admin_email' );
 670         if ( $admin_email == '' )
 671                 $admin_email = 'support@' . $_SERVER['SERVER_NAME'];
 672         $from_name = get_site_option( 'site_name' ) == '' ? 'WordPress' : esc_html( get_site_option( 'site_name' ) );
 673         $message_headers = "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
 674         $message = sprintf( apply_filters( 'wpmu_signup_user_notification_email', __( "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" ) ), site_url( "wp-activate.php?key=$key" ), $key );
 675         // TODO: Don't hard code activation link.
 676         $subject = sprintf( __( apply_filters( 'wpmu_signup_user_notification_subject', '[%1$s] Activate %2$s' ) ), $from_name, $user);
 677         wp_mail($user_email, $subject, $message, $message_headers);
 678         return true;
 679 }
 680
 681 function wpmu_activate_signup($key) {
 682         global $wpdb, $current_site;
 683
 684         $signup = $wpdb->get_row( $wpdb->prepare("SELECT * FROM $wpdb->signups WHERE activation_key = %s", $key) );
 685
 686         if ( empty($signup) )
 687                 return new WP_Error('invalid_key', __('Invalid activation key.'));
 688
 689         if ( $signup->active )
 690                 return new WP_Error('already_active', __('The site is already active.'), $signup);
 691
 692         $meta = unserialize($signup->meta);
 693         $user_login = $wpdb->escape($signup->user_login);
 694         $user_email = $wpdb->escape($signup->user_email);
 695         $password = wp_generate_password();
 696
 697         $user_id = username_exists($user_login);
 698
 699         if ( ! $user_id )
 700                 $user_id = wpmu_create_user($user_login, $password, $user_email);
 701         else
 702                 $user_already_exists = true;
 703
 704         if ( ! $user_id )
 705                 return new WP_Error('create_user', __('Could not create user'), $signup);
 706
 707         $now = current_time('mysql', true);
 708
 709         if ( empty($signup->domain) ) {
 710                 $wpdb->update( $wpdb->signups, array('active' => 1, 'activated' => $now), array('activation_key' => $key) );
 711
 712                 if ( isset( $user_already_exists ) )
 713                         return new WP_Error( 'user_already_exists', __( 'That username is already activated.' ), $signup);
 714
 715                 wpmu_welcome_user_notification($user_id, $password, $meta);
 716                 $user_site = get_site_option( 'dashboard_blog', $current_site->blog_id );
 717
 718                 if ( $user_site == false )
 719                         add_user_to_blog( '1', $user_id, get_site_option( 'default_user_role', 'subscriber' ) );
 720                 else
 721                         add_user_to_blog( $user_site, $user_id, get_site_option( 'default_user_role', 'subscriber' ) );
 722
 723                 add_new_user_to_blog( $user_id, $user_email, $meta );
 724                 do_action('wpmu_activate_user', $user_id, $password, $meta);
 725                 return array('user_id' => $user_id, 'password' => $password, 'meta' => $meta);
 726         }
 727
 728         $blog_id = wpmu_create_blog( $signup->domain, $signup->path, $signup->title, $user_id, $meta, $wpdb->siteid );
 729
 730         // TODO: What to do if we create a user but cannot create a blog?
 731         if ( is_wp_error($blog_id) ) {
 732                 // If blog is taken, that means a previous attempt to activate this blog failed in between creating the blog and
 733                 // setting the activation flag.  Let's just set the active flag and instruct the user to reset their password.
 734                 if ( 'blog_taken' == $blog_id->get_error_code() ) {
 735                         $blog_id->add_data( $signup );
 736                         $wpdb->update( $wpdb->signups, array( 'active' => 1, 'activated' => $now ), array( 'activation_key' => $key ) );
 737                 }
 738                 return $blog_id;
 739         }
 740
 741         $wpdb->update( $wpdb->signups, array('active' => 1, 'activated' => $now), array('activation_key' => $key) );
 742         wpmu_welcome_notification($blog_id, $user_id, $password, $signup->title, $meta);
 743         do_action('wpmu_activate_blog', $blog_id, $user_id, $password, $signup->title, $meta);
 744
 745         return array('blog_id' => $blog_id, 'user_id' => $user_id, 'password' => $password, 'title' => $signup->title, 'meta' => $meta);
 746 }
 747
 748 function wpmu_create_user( $user_name, $password, $email) {
 749         $user_name = preg_replace( '/\s+/', '', sanitize_user( $user_name, true ) );
 750
 751         $user_id = wp_create_user( $user_name, $password, $email );
 752         if ( is_wp_error($user_id) )
 753                 return false;
 754
 755         // Newly created users have no roles or caps until they are added to a blog.
 756         delete_user_option( $user_id, 'capabilities' );
 757         delete_user_option( $user_id, 'user_level' );
 758
 759         do_action( 'wpmu_new_user', $user_id );
 760
 761         return $user_id;
 762 }
 763
 764 function wpmu_create_blog($domain, $path, $title, $user_id, $meta = '', $site_id = 1) {
 765         $domain = preg_replace( '/\s+/', '', sanitize_user( $domain, true ) );
 766
 767         if ( is_subdomain_install() )
 768                 $domain = str_replace( '@', '', $domain );
 769
 770         $title = strip_tags( $title );
 771         $user_id = (int) $user_id;
 772
 773         if ( empty($path) )
 774                 $path = '/';
 775
 776         // Check if the domain has been used already. We should return an error message.
 777         if ( domain_exists($domain, $path, $site_id) )
 778                 return new WP_Error('blog_taken', __('Site already exists.'));
 779
 780         if ( !defined('WP_INSTALLING') )
 781                 define( 'WP_INSTALLING', true );
 782
 783         if ( ! $blog_id = insert_blog($domain, $path, $site_id) )
 784                 return new WP_Error('insert_blog', __('Could not create site.'));
 785
 786         switch_to_blog($blog_id);
 787         install_blog($blog_id, $title);
 788         wp_install_defaults($user_id);
 789
 790         add_user_to_blog($blog_id, $user_id, 'administrator');
 791
 792         if ( is_array($meta) ) foreach ($meta as $key => $value) {
 793                 if ( $key == 'public' || $key == 'archived' || $key == 'mature' || $key == 'spam' || $key == 'deleted' || $key == 'lang_id' )
 794                         update_blog_status( $blog_id, $key, $value );
 795                 else
 796                         update_option( $key, $value );
 797         }
 798
 799         add_option( 'WPLANG', get_site_option( 'WPLANG' ) );
 800         update_option( 'blog_public', (int)$meta['public'] );
 801
 802         if ( !is_super_admin() && get_user_meta( $user_id, 'primary_blog', true ) == get_site_option( 'dashboard_blog', 1 ) )
 803                 update_user_meta( $user_id, 'primary_blog', $blog_id );
 804
 805         restore_current_blog();
 806         do_action( 'wpmu_new_blog', $blog_id, $user_id, $domain, $path, $site_id, $meta );
 807
 808         return $blog_id;
 809 }
 810
 811 function newblog_notify_siteadmin( $blog_id, $deprecated = '' ) {
 812         if ( get_site_option( 'registrationnotification' ) != 'yes' )
 813                 return false;
 814
 815         $email = get_site_option( 'admin_email' );
 816         if ( is_email($email) == false )
 817                 return false;
 818
 819         $options_site_url = esc_url(network_admin_url('ms-options.php'));
 820
 821         switch_to_blog( $blog_id );
 822         $blogname = get_option( 'blogname' );
 823         $siteurl = site_url();
 824         restore_current_blog();
 825
 826         $msg = sprintf( __( 'New Site: %1s
 827 URL: %2s
 828 Remote IP: %3s
 829
 830 Disable these notifications: %4s' ), $blogname, $siteurl, $_SERVER['REMOTE_ADDR'], $options_site_url);
 831         $msg = apply_filters( 'newblog_notify_siteadmin', $msg );
 832
 833         wp_mail( $email, sprintf( __( 'New Site Registration: %s' ), $siteurl ), $msg );
 834         return true;
 835 }
 836
 837 function newuser_notify_siteadmin( $user_id ) {
 838         if ( get_site_option( 'registrationnotification' ) != 'yes' )
 839                 return false;
 840
 841         $email = get_site_option( 'admin_email' );
 842
 843         if ( is_email($email) == false )
 844                 return false;
 845
 846         $user = new WP_User($user_id);
 847
 848         $options_site_url = esc_url(network_admin_url('ms-options.php'));
 849         $msg = sprintf(__('New User: %1s
 850 Remote IP: %2s
 851
 852 Disable these notifications: %3s'), $user->user_login, $_SERVER['REMOTE_ADDR'], $options_site_url);
 853
 854         $msg = apply_filters( 'newuser_notify_siteadmin', $msg );
 855         wp_mail( $email, sprintf(__('New User Registration: %s'), $user->user_login), $msg );
 856         return true;
 857 }
 858
 859 function domain_exists($domain, $path, $site_id = 1) {
 860         global $wpdb;
 861         return $wpdb->get_var( $wpdb->prepare("SELECT blog_id FROM $wpdb->blogs WHERE domain = %s AND path = %s AND site_id = %d", $domain, $path, $site_id) );
 862 }
 863
 864 function insert_blog($domain, $path, $site_id) {
 865         global $wpdb;
 866
 867         $path = trailingslashit($path);
 868         $site_id = (int) $site_id;
 869
 870         $result = $wpdb->insert( $wpdb->blogs, array('site_id' => $site_id, 'domain' => $domain, 'path' => $path, 'registered' => current_time('mysql')) );
 871         if ( ! $result )
 872                 return false;
 873
 874         refresh_blog_details($wpdb->insert_id);
 875         return $wpdb->insert_id;
 876 }
 877
 878 // Install an empty blog.  wpdb should already be switched.
 879 function install_blog($blog_id, $blog_title = '') {
 880         global $wpdb, $table_prefix, $wp_roles;
 881         $wpdb->suppress_errors();
 882
 883         // Cast for security
 884         $blog_id = (int) $blog_id;
 885
 886         require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );
 887
 888         if ( $wpdb->get_results("SELECT ID FROM $wpdb->posts") )
 889                 die(__('<h1>Already Installed</h1><p>You appear to have already installed WordPress. To reinstall please clear your old database tables first.</p>') . '</body></html>');
 890
 891         $wpdb->suppress_errors(false);
 892
 893         $url = get_blogaddress_by_id($blog_id);
 894
 895         // Set everything up
 896         make_db_current_silent();
 897         populate_options();
 898         populate_roles();
 899         $wp_roles->_init();
 900
 901         // fix url.
 902         update_option('siteurl', $url);
 903         update_option('home', $url);
 904         update_option('fileupload_url', $url . "files" );
 905         update_option('upload_path', "wp-content/blogs.dir/" . $blog_id . "/files");
 906         update_option('blogname', stripslashes( $blog_title ) );
 907         update_option('admin_email', '');
 908         $wpdb->update( $wpdb->options, array('option_value' => ''), array('option_name' => 'admin_email') );
 909
 910         // remove all perms
 911         $wpdb->query( $wpdb->prepare("DELETE FROM $wpdb->usermeta WHERE meta_key = %s", $table_prefix.'user_level') );
 912         $wpdb->query( $wpdb->prepare("DELETE FROM $wpdb->usermeta WHERE meta_key = %s", $table_prefix.'capabilities') );
 913
 914         $wpdb->suppress_errors( false );
 915 }
 916
 917 // Deprecated, use wp_install_defaults()
 918 // should be switched already as $blog_id is ignored.
 919 function install_blog_defaults($blog_id, $user_id) {
 920         global $wpdb;
 921
 922         require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );
 923
 924         $wpdb->suppress_errors();
 925
 926         wp_install_defaults($user_id);
 927
 928         $wpdb->suppress_errors( false );
 929 }
 930
 931 function wpmu_welcome_notification($blog_id, $user_id, $password, $title, $meta = '') {
 932         global $current_site;
 933
 934         if ( !apply_filters('wpmu_welcome_notification', $blog_id, $user_id, $password, $title, $meta) )
 935                 return false;
 936
 937         $welcome_email = stripslashes( get_site_option( 'welcome_email' ) );
 938         if ( $welcome_email == false )
 939                 $welcome_email = stripslashes( __( 'Dear User,
 940
 941 Your new SITE_NAME site has been successfully set up at:
 942 BLOG_URL
 943
 944 You can log in to the administrator account with the following information:
 945 Username: USERNAME
 946 Password: PASSWORD
 947 Login Here: BLOG_URLwp-login.php
 948
 949 We hope you enjoy your new site.
 950 Thanks!
 951
 952 --The Team @ SITE_NAME' ) );
 953
 954         $url = get_blogaddress_by_id($blog_id);
 955         $user = new WP_User($user_id);
 956
 957         $welcome_email = str_replace( 'SITE_NAME', $current_site->site_name, $welcome_email );
 958         $welcome_email = str_replace( 'BLOG_TITLE', $title, $welcome_email );
 959         $welcome_email = str_replace( 'BLOG_URL', $url, $welcome_email );
 960         $welcome_email = str_replace( 'USERNAME', $user->user_login, $welcome_email );
 961         $welcome_email = str_replace( 'PASSWORD', $password, $welcome_email );
 962
 963         $welcome_email = apply_filters( 'update_welcome_email', $welcome_email, $blog_id, $user_id, $password, $title, $meta);
 964         $admin_email = get_site_option( 'admin_email' );
 965
 966         if ( $admin_email == '' )
 967                 $admin_email = 'support@' . $_SERVER['SERVER_NAME'];
 968
 969         $from_name = get_site_option( 'site_name' ) == '' ? 'WordPress' : esc_html( get_site_option( 'site_name' ) );
 970         $message_headers = "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
 971         $message = $welcome_email;
 972
 973         if ( empty( $current_site->site_name ) )
 974                 $current_site->site_name = 'WordPress MU';
 975
 976         $subject = apply_filters( 'update_welcome_subject', sprintf(__('New %1$s Site: %2$s'), $current_site->site_name, stripslashes( $title ) ) );
 977         wp_mail($user->user_email, $subject, $message, $message_headers);
 978         return true;
 979 }
 980
 981 function wpmu_welcome_user_notification($user_id, $password, $meta = '') {
 982         global $current_site;
 983
 984         if ( !apply_filters('wpmu_welcome_user_notification', $user_id, $password, $meta) )
 985                 return false;
 986
 987         $welcome_email = get_site_option( 'welcome_user_email' );
 988
 989         $user = new WP_User($user_id);
 990
 991         $welcome_email = apply_filters( 'update_welcome_user_email', $welcome_email, $user_id, $password, $meta);
 992         $welcome_email = str_replace( 'SITE_NAME', $current_site->site_name, $welcome_email );
 993         $welcome_email = str_replace( 'USERNAME', $user->user_login, $welcome_email );
 994         $welcome_email = str_replace( 'PASSWORD', $password, $welcome_email );
 995         $welcome_email = str_replace( 'LOGINLINK', wp_login_url(), $welcome_email );
 996
 997         $admin_email = get_site_option( 'admin_email' );
 998
 999         if ( $admin_email == '' )
1000                 $admin_email = 'support@' . $_SERVER['SERVER_NAME'];
1001
1002         $from_name = get_site_option( 'site_name' ) == '' ? 'WordPress' : esc_html( get_site_option( 'site_name' ) );
1003         $message_headers = "From: \"{$from_name}\" <{$admin_email}>\n" . "Content-Type: text/plain; charset=\"" . get_option('blog_charset') . "\"\n";
1004         $message = $welcome_email;
1005
1006         if ( empty( $current_site->site_name ) )
1007                 $current_site->site_name = 'WordPress MU';
1008
1009         $subject = apply_filters( 'update_welcome_user_subject', sprintf(__('New %1$s User: %2$s'), $current_site->site_name, $user->user_login) );
1010         wp_mail($user->user_email, $subject, $message, $message_headers);
1011         return true;
1012 }
1013
1014 function get_current_site() {
1015         global $current_site;
1016         return $current_site;
1017 }
1018
1019 function get_user_id_from_string( $string ) {
1020         $user_id = 0;
1021         if ( is_email( $string ) ) {
1022                 $user = get_user_by('email', $string);
1023                 if ( $user )
1024                         $user_id = $user->ID;
1025         } elseif ( is_numeric( $string ) ) {
1026                 $user_id = $string;
1027         } else {
1028                 $user = get_user_by('login', $string);
1029                 if ( $user )
1030                         $user_id = $user->ID;
1031         }
1032
1033         return $user_id;
1034 }
1035
1036 function get_most_recent_post_of_user( $user_id ) {
1037         global $wpdb;
1038
1039         $user_blogs = get_blogs_of_user( (int) $user_id );
1040         $most_recent_post = array();
1041
1042         // Walk through each blog and get the most recent post
1043         // published by $user_id
1044         foreach ( (array) $user_blogs as $blog ) {
1045                 $recent_post = $wpdb->get_row( $wpdb->prepare("SELECT ID, post_date_gmt FROM {$wpdb->base_prefix}{$blog->userblog_id}_posts WHERE post_author = %d AND post_type = 'post' AND post_status = 'publish' ORDER BY post_date_gmt DESC LIMIT 1", $user_id ), ARRAY_A);
1046
1047                 // Make sure we found a post
1048                 if ( isset($recent_post['ID']) ) {
1049                         $post_gmt_ts = strtotime($recent_post['post_date_gmt']);
1050
1051                         // If this is the first post checked or if this post is
1052                         // newer than the current recent post, make it the new
1053                         // most recent post.
1054                         if ( !isset($most_recent_post['post_gmt_ts']) || ( $post_gmt_ts > $most_recent_post['post_gmt_ts'] ) ) {
1055                                 $most_recent_post = array(
1056                                         'blog_id'               => $blog->userblog_id,
1057                                         'post_id'               => $recent_post['ID'],
1058                                         'post_date_gmt' => $recent_post['post_date_gmt'],
1059                                         'post_gmt_ts'   => $post_gmt_ts
1060                                 );
1061                         }
1062                 }
1063         }
1064
1065         return $most_recent_post;
1066 }
1067
1068 /* Misc functions */
1069 function get_dirsize( $directory ) {
1070         $dirsize = get_transient( 'dirsize_cache' );
1071         if ( is_array( $dirsize ) && isset( $dirsize[ $directory ][ 'size' ] ) )
1072                 return $dirsize[ $directory ][ 'size' ];
1073
1074         if ( false == is_array( $dirsize ) )
1075                 $dirsize = array();
1076
1077         $dirsize[ $directory ][ 'size' ] = recurse_dirsize( $directory );
1078
1079         set_transient( 'dirsize_cache', $dirsize, 3600 );
1080         return $dirsize[ $directory ][ 'size' ];
1081 }
1082
1083 function recurse_dirsize( $directory ) {
1084         $size = 0;
1085
1086         if ( substr( $directory, -1 ) == '/' )
1087                 $directory = substr($directory,0,-1);
1088
1089         if ( !file_exists($directory) || !is_dir( $directory ) || !is_readable( $directory ) )
1090                 return false;
1091
1092         if ($handle = opendir($directory)) {
1093                 while(($file = readdir($handle)) !== false) {
1094                         $path = $directory.'/'.$file;
1095                         if ($file != '.' && $file != '..') {
1096                                 if (is_file($path)) {
1097                                         $size += filesize($path);
1098                                 } elseif (is_dir($path)) {
1099                                         $handlesize = recurse_dirsize($path);
1100                                         if ($handlesize > 0)
1101                                                 $size += $handlesize;
1102                                 }
1103                         }
1104                 }
1105                 closedir($handle);
1106         }
1107         return $size;
1108 }
1109
1110 function upload_is_user_over_quota( $echo = true ) {
1111         if ( get_site_option( 'upload_space_check_disabled' ) )
1112                 return true;
1113
1114         $spaceAllowed = get_space_allowed();
1115         if ( empty( $spaceAllowed ) || !is_numeric( $spaceAllowed ) )
1116                 $spaceAllowed = 10;     // Default space allowed is 10 MB
1117
1118         $dirName = BLOGUPLOADDIR;
1119         $size = get_dirsize($dirName) / 1024 / 1024;
1120
1121         if ( ($spaceAllowed-$size) < 0 ) {
1122                 if ( $echo )
1123                         _e( 'Sorry, you have used your space allocation. Please delete some files to upload more files.' ); // No space left
1124                 return true;
1125         } else {
1126                 return false;
1127         }
1128 }
1129
1130 function check_upload_mimes( $mimes ) {
1131         $site_exts = explode( ' ', get_site_option( 'upload_filetypes' ) );
1132         foreach ( $site_exts as $ext ) {
1133                 foreach ( $mimes as $ext_pattern => $mime ) {
1134                         if ( $ext != '' && strpos( $ext_pattern, $ext ) !== false )
1135                                 $site_mimes[$ext_pattern] = $mime;
1136                 }
1137         }
1138         return $site_mimes;
1139 }
1140
1141 function update_posts_count( $deprecated = '' ) {
1142         global $wpdb;
1143         update_option( 'post_count', (int) $wpdb->get_var( "SELECT COUNT(ID) FROM {$wpdb->posts} WHERE post_status = 'publish' and post_type = 'post'" ) );
1144 }
1145
1146 function wpmu_log_new_registrations( $blog_id, $user_id ) {
1147         global $wpdb;
1148         $user = new WP_User( (int) $user_id );
1149         $wpdb->insert( $wpdb->registration_log, array('email' => $user->user_email, 'IP' => preg_replace( '/[^0-9., ]/', '',$_SERVER['REMOTE_ADDR'] ), 'blog_id' => $blog_id, 'date_registered' => current_time('mysql')) );
1150 }
1151
1152 function fix_import_form_size( $size ) {
1153         if ( upload_is_user_over_quota( false ) == true )
1154                 return 0;
1155
1156         $spaceAllowed = 1024 * 1024 * get_space_allowed();
1157         $dirName = BLOGUPLOADDIR;
1158         $dirsize = get_dirsize($dirName) ;
1159         if ( $size > $spaceAllowed - $dirsize )
1160                 return $spaceAllowed - $dirsize; // remaining space
1161         else
1162                 return $size; // default
1163 }
1164
1165 /**
1166  * Maintains a canonical list of terms by syncing terms created for each blog with the global terms table.
1167  *
1168  * @since 3.0.0
1169  *
1170  * @see term_id_filter
1171  *
1172  * @param int $term_id An ID for a term on the current blog.
1173  * @return int An ID from the global terms table mapped from $term_id.
1174  */
1175 function global_terms( $term_id, $deprecated = '' ) {
1176         global $wpdb;
1177         static $global_terms_recurse = null;
1178
1179         if ( !global_terms_enabled() )
1180                 return $term_id;
1181
1182         // prevent a race condition
1183         $recurse_start = false;
1184         if ( $global_terms_recurse === null ) {
1185                 $recurse_start = true;
1186                 $global_terms_recurse = 1;
1187         } elseif ( 10 < $global_terms_recurse++ ) {
1188                 return $term_id;
1189         }
1190
1191         $term_id = intval( $term_id );
1192         $c = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM $wpdb->terms WHERE term_id = %d", $term_id ) );
1193
1194         $global_id = $wpdb->get_var( $wpdb->prepare( "SELECT cat_ID FROM $wpdb->sitecategories WHERE category_nicename = %s", $c->slug ) );
1195         if ( $global_id == null ) {
1196                 $used_global_id = $wpdb->get_var( $wpdb->prepare( "SELECT cat_ID FROM $wpdb->sitecategories WHERE cat_ID = %d", $c->term_id ) );
1197                 if ( null == $used_global_id ) {
1198                         $wpdb->insert( $wpdb->sitecategories, array( 'cat_ID' => $term_id, 'cat_name' => $c->name, 'category_nicename' => $c->slug ) );
1199                         $global_id = $wpdb->insert_id;
1200                         if ( empty( $global_id ) )
1201                                 return $term_id;
1202                 } else {
1203                         $max_global_id = $wpdb->get_var( "SELECT MAX(cat_ID) FROM $wpdb->sitecategories" );
1204                         $max_local_id = $wpdb->get_var( "SELECT MAX(term_id) FROM $wpdb->terms" );
1205                         $new_global_id = max( $max_global_id, $max_local_id ) + mt_rand( 100, 400 );
1206                         $wpdb->insert( $wpdb->sitecategories, array( 'cat_ID' => $new_global_id, 'cat_name' => $c->name, 'category_nicename' => $c->slug ) );
1207                         $global_id = $wpdb->insert_id;
1208                 }
1209         } elseif ( $global_id != $term_id ) {
1210                 $local_id = $wpdb->get_row( $wpdb->prepare( "SELECT term_id FROM $wpdb->terms WHERE term_id = %d", $global_id ) );
1211                 if ( null != $local_id )
1212                         $local_id = global_terms( $local_id );
1213                         if ( 10 < $global_terms_recurse )
1214                                 $global_id = $term_id;
1215         }
1216
1217         if ( $global_id != $term_id ) {
1218                 if ( get_option( 'default_category' ) == $term_id )
1219                         update_option( 'default_category', $global_id );
1220
1221                 $wpdb->update( $wpdb->terms, array('term_id' => $global_id), array('term_id' => $term_id) );
1222                 $wpdb->update( $wpdb->term_taxonomy, array('term_id' => $global_id), array('term_id' => $term_id) );
1223                 $wpdb->update( $wpdb->term_taxonomy, array('parent' => $global_id), array('parent' => $term_id) );
1224
1225                 clean_term_cache($term_id);
1226         }
1227         if( $recurse_start )
1228                 $global_terms_recurse = null;
1229
1230         return $global_id;
1231 }
1232
1233 function redirect_this_site( $deprecated = '' ) {
1234         global $current_site;
1235         return array( $current_site->domain );
1236 }
1237
1238 function upload_is_file_too_big( $upload ) {
1239         if ( is_array( $upload ) == false || defined( 'WP_IMPORTING' ) )
1240                 return $upload;
1241
1242         if ( strlen( $upload['bits'] )  > ( 1024 * get_site_option( 'fileupload_maxk', 1500 ) ) )
1243                 return sprintf( __( 'This file is too big. Files must be less than %d KB in size.' ) . '<br />', get_site_option( 'fileupload_maxk', 1500 ));
1244
1245         return $upload;
1246 }
1247
1248 function wordpressmu_wp_mail_from( $email ) {
1249         if ( strpos( $email, 'wordpress@' ) !== false )
1250                 $email = get_option( 'admin_email' );
1251         return $email;
1252 }
1253
1254 function signup_nonce_fields() {
1255         $id = mt_rand();
1256         echo "<input type='hidden' name='signup_form_id' value='{$id}' />";
1257         wp_nonce_field('signup_form_' . $id, '_signup_form', false);
1258 }
1259
1260 function signup_nonce_check( $result ) {
1261         if ( !strpos( $_SERVER[ 'PHP_SELF' ], 'wp-signup.php' ) )
1262                 return $result;
1263
1264         if ( wp_create_nonce('signup_form_' . $_POST[ 'signup_form_id' ]) != $_POST['_signup_form'] )
1265                 wp_die( __('Please try again!') );
1266
1267         return $result;
1268 }
1269
1270 function maybe_redirect_404() {
1271         global $current_site;
1272         if ( is_main_site() && is_404() && defined( 'NOBLOGREDIRECT' ) && ( $destination = apply_filters( 'blog_redirect_404', NOBLOGREDIRECT ) ) ) {
1273                 if ( $destination == '%siteurl%' )
1274                         $destination = network_home_url();
1275                 wp_redirect( $destination );
1276                 exit();
1277         }
1278 }
1279
1280 function maybe_add_existing_user_to_blog() {
1281         if ( false === strpos( $_SERVER[ 'REQUEST_URI' ], '/newbloguser/' ) )
1282                 return false;
1283
1284         $parts = explode( '/', $_SERVER[ 'REQUEST_URI' ] );
1285         $key = array_pop( $parts );
1286
1287         if ( $key == '' )
1288                 $key = array_pop( $parts );
1289
1290         $details = get_option( 'new_user_' . $key );
1291         if ( !empty( $details ) )
1292                 delete_option( 'new_user_' . $key );
1293
1294         if ( empty( $details ) || is_wp_error( add_existing_user_to_blog( $details ) ) )
1295                 wp_die( sprintf(__('An error occurred adding you to this site. Back to the <a href="%s">homepage</a>.'), site_url() ) );
1296
1297         wp_die( sprintf(__('You have been added to this site. Please visit the <a href="%s">homepage</a> or <a href="%s">login</a> using your username and password.'), site_url(), admin_url() ), __('Success') );
1298 }
1299
1300 function add_existing_user_to_blog( $details = false ) {
1301         if ( is_array( $details ) ) {
1302                 $result = add_user_to_blog( '', $details[ 'user_id' ], $details[ 'role' ] );
1303                 do_action( 'added_existing_user', $details[ 'user_id' ], $result );
1304         }
1305         return $result;
1306 }
1307
1308 function add_new_user_to_blog( $user_id, $email, $meta ) {
1309         global $current_site;
1310         if ( $meta[ 'add_to_blog' ] ) {
1311                 $blog_id = $meta[ 'add_to_blog' ];
1312                 $role = $meta[ 'new_role' ];
1313                 remove_user_from_blog($user_id, $current_site->blogid); // remove user from main blog.
1314                 add_user_to_blog( $blog_id, $user_id, $role );
1315                 update_user_meta( $user_id, 'primary_blog', $blog_id );
1316         }
1317 }
1318
1319 function fix_phpmailer_messageid( $phpmailer ) {
1320         global $current_site;
1321         $phpmailer->Hostname = $current_site->domain;
1322 }
1323
1324 function is_user_spammy( $username = 0 ) {
1325         if ( $username == 0 ) {
1326                 $user_id = get_current_user_id();
1327         } else {
1328                 $user_id = get_user_id_from_string( $username );
1329         }
1330         $u = new WP_User( $user_id );
1331
1332         return ( isset( $u->spam ) && $u->spam == 1 );
1333 }
1334
1335 function update_blog_public( $old_value, $value ) {
1336         global $wpdb;
1337         do_action('update_blog_public');
1338         update_blog_status( $wpdb->blogid, 'public', (int) $value );
1339 }
1340 add_action('update_option_blog_public', 'update_blog_public', 10, 2);
1341
1342 /* Redirect all hits to "dashboard" blog to wp-admin/ Dashboard. */
1343 function redirect_mu_dashboard() {
1344         global $current_site, $current_blog;
1345
1346         $dashboard_blog = get_dashboard_blog();
1347         if ( $current_blog->blog_id == $dashboard_blog->blog_id && $dashboard_blog->blog_id != $current_site->blog_id ) {
1348                 $protocol = ( is_ssl() ? 'https://' : 'http://' );
1349                 wp_redirect( $protocol . $dashboard_blog->domain . trailingslashit( $dashboard_blog->path ) . 'wp-admin/' );
1350                 die();
1351         }
1352 }
1353 add_action( 'template_redirect', 'redirect_mu_dashboard' );
1354
1355 function get_dashboard_blog() {
1356         if ( $blog = get_site_option( 'dashboard_blog' ) )
1357                 return get_blog_details( $blog );
1358
1359         return get_blog_details( $GLOBALS['current_site']->blog_id );
1360 }
1361
1362 function is_user_option_local( $key, $user_id = 0, $blog_id = 0 ) {
1363         global $wpdb;
1364
1365         $current_user = wp_get_current_user();
1366         if ( $user_id == 0 )
1367                 $user_id = $current_user->ID;
1368         if ( $blog_id == 0 )
1369                 $blog_id = $wpdb->blogid;
1370
1371         $local_key = $wpdb->base_prefix . $blog_id . '_' . $key;
1372
1373         if ( isset( $current_user->$local_key ) )
1374                 return true;
1375
1376         return false;
1377 }
1378
1379 function users_can_register_signup_filter() {
1380         $registration = get_site_option('registration');
1381         if ( $registration == 'all' || $registration == 'user' )
1382                 return true;
1383
1384         return false;
1385 }
1386 add_filter('option_users_can_register', 'users_can_register_signup_filter');
1387
1388 function welcome_user_msg_filter( $text ) {
1389         if ( !$text ) {
1390                 return __( 'Dear User,
1391
1392 Your new account is set up.
1393
1394 You can log in with the following information:
1395 Username: USERNAME
1396 Password: PASSWORD
1397 LOGINLINK
1398
1399 Thanks!
1400
1401 --The Team @ SITE_NAME' );
1402         }
1403         return $text;
1404 }
1405 add_filter( 'site_option_welcome_user_email', 'welcome_user_msg_filter' );
1406
1407 /**
1408  * Whether to force SSL on content.
1409  *
1410  * @since 2.8.5
1411  *
1412  * @param string|bool $force
1413  * @return bool True if forced, false if not forced.
1414  */
1415 function force_ssl_content( $force = '' ) {
1416         static $forced_content;
1417
1418         if ( '' != $force ) {
1419                 $old_forced = $forced_content;
1420                 $forced_content = $force;
1421                 return $old_forced;
1422         }
1423
1424         return $forced_content;
1425 }
1426
1427 /**
1428  * Formats an String URL to use HTTPS if HTTP is found.
1429  * Useful as a filter.
1430  *
1431  * @since 2.8.5
1432  **/
1433 function filter_SSL( $url ) {
1434         if ( !is_string( $url ) )
1435                 return get_bloginfo( 'url' ); //return home blog url with proper scheme
1436
1437         $arrURL = parse_url( $url );
1438
1439         if ( force_ssl_content() && is_ssl() ) {
1440                 if ( 'http' === $arrURL['scheme'] && 'https' !== $arrURL['scheme'] )
1441                         $url = str_replace( $arrURL['scheme'], 'https', $url );
1442         }
1443
1444         return $url;
1445 }
1446
1447 ?>