root/tags/1.2.4/wp-includes/wpmu-functions.php

<?php
/*
    Helper functions for WPMU
*/
function load_muplugin_textdomain($domain, $path = false) {
    $locale = get_locale();
    if ( empty($locale) )
        $locale = 'en_US';

    if ( false === $path )
        $path = MUPLUGINDIR;

    $mofile = ABSPATH . "$path/$domain-$locale.mo";
    load_textdomain($domain, $mofile);
}

function wpmu_update_blogs_date() {
    global $wpdb;

    $wpdb->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 (<i>%2.2fMB</i>)'), $spaceAvailable);


    if (($spaceAllowed-$size)>0) {
        return $action;
    } else {
        // No space left
        return 'not-writable';
    }
}
add_filter('fileupload_init','wpmu_checkAvailableSpace');

if( defined( "WP_INSTALLING" ) == false ) {
    header( "X-totalblogs: " . get_blog_count() );
    header( "X-rootblog: http://" . $current_site->domain . $current_site->path );
    header( "X-created-on: " . $current_blog->registered );

    if( empty( $WPMU_date ) == false )
        header( "X-wpmu-date: $WPMU_date" );
}


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 == true ) {
        $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();

        wp_cache_add( $id, serialize( $details ), 'blog-details' );

        $key = md5( $details->domain . $details->path );
        wp_cache_add( $key, serialize( $details ), 'blog-lookup' );
    }

    return $details;
}

function refresh_blog_details( $id ) {
    global $wpdb, $wpmuBaseTablePrefix;

    $details = get_blog_details( $id );
    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 ( null !== $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;

    if ( $value == get_site_option( $key ) )
         return;

    if ( get_site_option( $key, false, false ) === false )
        add_site_option( $key, $value );

    if ( is_array($value) || is_object($value) )
        $value = serialize($value);

    $safe_key = $wpdb->escape( $key );

    $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[ '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';
    $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->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);