root/tags/2.0/wp-includes/functions-post.php

<?php

/**** DB Functions ****/

/*
 * generic function for inserting data into the posts table.
 */
function wp_insert_post($postarr = array()) {
    global $wpdb, $allowedtags, $user_ID;

    if ( is_object($postarr) )
        $postarr = get_object_vars($postarr);

    // export array as variables
    extract($postarr);

    // Are we updating or creating?
    $update = false;
    if ( !empty($ID) ) {
        $update = true;
        $post = & get_post($ID);
        $previous_status = $post->post_status;
    }

    // Get the basics.
    $post_content    = apply_filters('content_save_pre',   $post_content);
    $post_excerpt    = apply_filters('excerpt_save_pre',   $post_excerpt);
    $post_title      = apply_filters('title_save_pre',     $post_title);
    $post_category   = apply_filters('category_save_pre',  $post_category);
    $post_status     = apply_filters('status_save_pre',    $post_status);
    $post_name       = apply_filters('name_save_pre',      $post_name);
    $comment_status  = apply_filters('comment_status_pre', $comment_status);
    $ping_status     = apply_filters('ping_status_pre',    $ping_status);
    
    // Make sure we set a valid category
    if (0 == count($post_category) || !is_array($post_category)) {
        $post_category = array(get_option('default_category'));
    }
    $post_cat = $post_category[0];

    if ( empty($post_author) )
        $post_author = $user_ID;

    if ( empty($post_status) )
        $post_status = 'draft';
    
    // Get the post ID.
    if ( $update )
        $post_ID = $ID;

    // Create a valid post name.  Drafts are allowed to have an empty
    // post name.
    if ( empty($post_name) ) {
        if ( 'draft' != $post_status )
            $post_name = sanitize_title($post_title);
    } else {
        $post_name = sanitize_title($post_name);
    }
    

    // If the post date is empty (due to having been new or a draft) and status is not 'draft', set date to now
    if (empty($post_date)) {
        if ( 'draft' != $post_status )
            $post_date = current_time('mysql');
    }

    if (empty($post_date_gmt)) {
        if ( 'draft' != $post_status )
            $post_date_gmt = get_gmt_from_date($post_date);
    }

    if ( empty($comment_status) ) {
        if ( $update )
            $comment_status = 'closed';
        else
            $comment_status = get_settings('default_comment_status');
    }
    if ( empty($ping_status) )
        $ping_status = get_settings('default_ping_status');
    if ( empty($post_pingback) )
        $post_pingback = get_option('default_pingback_flag');

    if ( isset($to_ping) )
        $to_ping = preg_replace('|\s+|', "\n", $to_ping);
    else
        $to_ping = '';

    if ( ! isset($pinged) )
        $pinged = '';

    if ( isset($post_parent) )
        $post_parent = (int) $post_parent;
    else
        $post_parent = 0;

    if ( isset($menu_order) )
        $menu_order = (int) $menu_order;
    else
        $menu_order = 0;

    if ( !isset($post_password) )
        $post_password = '';

    if ( ('publish' == $post_status) || ('static' == $post_status) ) {
        $post_name_check = ('publish' == $post_status)
            ? $wpdb->get_var("SELECT post_name FROM $wpdb->posts WHERE post_name = '$post_name' AND post_status = 'publish' AND ID != '$post_ID' LIMIT 1")
            : $wpdb->get_var("SELECT post_name FROM $wpdb->posts WHERE post_name = '$post_name' AND post_status = 'static' AND ID != '$post_ID' AND post_parent = '$post_parent' LIMIT 1");

        if ($post_name_check) {
            $suffix = 2;
            while ($post_name_check) {
                $alt_post_name = $post_name . "-$suffix";
                $post_name_check = ('publish' == $post_status)
                    ? $wpdb->get_var("SELECT post_name FROM $wpdb->posts WHERE post_name = '$alt_post_name' AND post_status = 'publish' AND ID != '$post_ID' LIMIT 1")
                    : $wpdb->get_var("SELECT post_name FROM $wpdb->posts WHERE post_name = '$alt_post_name' AND post_status = 'static' AND ID != '$post_ID' AND post_parent = '$post_parent' LIMIT 1");
                $suffix++;
            }
            $post_name = $alt_post_name;
        }
    }

    if ($update) {
        $wpdb->query(
            "UPDATE $wpdb->posts SET
            post_author = '$post_author',
            post_date = '$post_date',
            post_date_gmt = '$post_date_gmt',
            post_content = '$post_content',
            post_content_filtered = '$post_content_filtered',
            post_title = '$post_title',
            post_excerpt = '$post_excerpt',
            post_status = '$post_status',
            comment_status = '$comment_status',
            ping_status = '$ping_status',
            post_password = '$post_password',
            post_name = '$post_name',
            to_ping = '$to_ping',
            pinged = '$pinged',
            post_modified = '".current_time('mysql')."',
            post_modified_gmt = '".current_time('mysql',1)."',
            post_parent = '$post_parent',
            menu_order = '$menu_order'
            WHERE ID = $post_ID");
    } else {
        $wpdb->query(
            "INSERT INTO $wpdb->posts
            (post_author, post_date, post_date_gmt, post_content, post_content_filtered, post_title, post_excerpt,  post_status, comment_status, ping_status, post_password, post_name, to_ping, pinged, post_modified, post_modified_gmt, post_parent, menu_order, post_mime_type)
            VALUES
            ('$post_author', '$post_date', '$post_date_gmt', '$post_content', '$post_content_filtered', '$post_title', '$post_excerpt', '$post_status', '$comment_status', '$ping_status', '$post_password', '$post_name', '$to_ping', '$pinged', '$post_date', '$post_date_gmt', '$post_parent', '$menu_order', '$post_mime_type')");
            $post_ID = $wpdb->insert_id;            
    }

    if ( empty($post_name) && 'draft' != $post_status ) {
        $post_name = sanitize_title($post_title, $post_ID);
        $wpdb->query( "UPDATE $wpdb->posts SET post_name = '$post_name' WHERE ID = '$post_ID'" );
    }

    wp_set_post_cats('', $post_ID, $post_category);

    if ( 'static' == $post_status ) {
        clean_page_cache($post_ID);
        wp_cache_delete($post_ID, 'pages');
    } else {
        clean_post_cache($post_ID);
    }

    // Set GUID
    if ( ! $update )
        $wpdb->query("UPDATE $wpdb->posts SET guid = '" . get_permalink($post_ID) . "' WHERE ID = '$post_ID'");

    if ( $update) {
        if ($previous_status != 'publish' && $post_status == 'publish') {
            // Reset GUID if transitioning to publish.
            $wpdb->query("UPDATE $wpdb->posts SET guid = '" . get_permalink($post_ID) . "' WHERE ID = '$post_ID'");
            do_action('private_to_published', $post_ID);
        }
        
        do_action('edit_post', $post_ID);
    }

    if ($post_status == 'publish') {
        do_action('publish_post', $post_ID);

        if ( !defined('WP_IMPORTING') ) {
            if ( $post_pingback )
                $result = $wpdb->query("
                    INSERT INTO $wpdb->postmeta 
                    (post_id,meta_key,meta_value) 
                    VALUES ('$post_ID','_pingme','1')
                ");
            $result = $wpdb->query("
                INSERT INTO $wpdb->postmeta 
                (post_id,meta_key,meta_value) 
                VALUES ('$post_ID','_encloseme','1')
            ");
            spawn_pinger();
        }
    } else if ($post_status == 'static') {
        generate_page_rewrite_rules();

        if ( !empty($page_template) )
            if ( ! update_post_meta($post_ID, '_wp_page_template',  $page_template))
                add_post_meta($post_ID, '_wp_page_template',  $page_template, true);
    }

    do_action('save_post', $post_ID);
    do_action('wp_insert_post', $post_ID);

    return $post_ID;
}

function wp_insert_attachment($object, $file = false, $post_parent = 0) {
    global $wpdb, $user_ID;

    if ( is_object($object) )
        $object = get_object_vars($object);

    // Export array as variables
    extract($object);

    // Get the basics.
    $post_content    = apply_filters('content_save_pre',   $post_content);
    $post_excerpt    = apply_filters('excerpt_save_pre',   $post_excerpt);
    $post_title      = apply_filters('title_save_pre',     $post_title);
    $post_category   = apply_filters('category_save_pre',  $post_category);
    $post_name       = apply_filters('name_save_pre',      $post_name);
    $comment_status  = apply_filters('comment_status_pre', $comment_status);
    $ping_status     = apply_filters('ping_status_pre',    $ping_status);
    $post_mime_type  = apply_filters('post_mime_type_pre', $post_mime_type);

    // Make sure we set a valid category
    if (0 == count($post_category) || !is_array($post_category)) {
        $post_category = array(get_option('default_category'));
    }
    $post_cat = $post_category[0];

    if ( empty($post_author) )
        $post_author = $user_ID;

    $post_status = 'attachment';

    // Are we updating or creating?
    $update = false;
    if ( !empty($ID) ) {
        $update = true;
        $post_ID = $ID;    
    }

    // Create a valid post name.
    if ( empty($post_name) )
        $post_name = sanitize_title($post_title);
    else
        $post_name = sanitize_title($post_name);
    
    if (empty($post_date))
        $post_date = current_time('mysql');
    if (empty($post_date_gmt)) 
        $post_date_gmt = current_time('mysql', 1);

    if ( empty($comment_status) ) {
        if ( $update )
            $comment_status = 'closed';
        else
            $comment_status = get_settings('default_comment_status');
    }
    if ( empty($ping_status) )
        $ping_status = get_settings('default_ping_status');
    if ( empty($post_pingback) )
        $post_pingback = get_option('default_pingback_flag');

    if ( isset($to_ping) )
        $to_ping = preg_replace('|\s+|', "\n", $to_ping);
    else
        $to_ping = '';

    if ( isset($post_parent) )
        $post_parent = (int) $post_parent;
    else
        $post_parent = 0;

    if ( isset($menu_order) )
        $menu_order = (int) $menu_order;
    else
        $menu_order = 0;

    if ( !isset($post_password) )
        $post_password = '';

    if ( isset($to_ping) )
        $to_ping = preg_replace('|\s+|', "\n", $to_ping);
    else
        $to_ping = '';

    if ( ! isset($pinged) )
        $pinged = '';

    if ($update) {
        $wpdb->query(
            "UPDATE $wpdb->posts SET
            post_author = '$post_author',
            post_date = '$post_date',
            post_date_gmt = '$post_date_gmt',
            post_content = '$post_content',
            post_title = '$post_title',
            post_excerpt = '$post_excerpt',
            post_status = '$post_status',
            comment_status = '$comment_status',
            ping_status = '$ping_status',
            post_password = '$post_password',
            post_name = '$post_name',
            to_ping = '$to_ping',
            pinged = '$pinged',
            post_modified = '".current_time('mysql')."',
            post_modified_gmt = '".current_time('mysql',1)."',
            post_parent = '$post_parent',
            menu_order = '$menu_order',
            post_mime_type = '$post_mime_type',
            guid = '$guid'
            WHERE ID = $post_ID");
    } else {
        $wpdb->query(
            "INSERT INTO $wpdb->posts
            (post_author, post_date, post_date_gmt, post_content, post_title, post_excerpt,  post_status, comment_status, ping_status, post_password, post_name, to_ping, pinged, post_modified, post_modified_gmt, post_parent, menu_order, post_mime_type, guid)
            VALUES
            ('$post_author', '$post_date', '$post_date_gmt', '$post_content', '$post_title', '$post_excerpt', '$post_status', '$comment_status', '$ping_status', '$post_password', '$post_name', '$to_ping', '$pinged', '$post_date', '$post_date_gmt', '$post_parent', '$menu_order', '$post_mime_type', '$guid')");
            $post_ID = $wpdb->insert_id;            
    }
    
    if ( empty($post_name) ) {
        $post_name = sanitize_title($post_title, $post_ID);
        $wpdb->query( "UPDATE $wpdb->posts SET post_name = '$post_name' WHERE ID = '$post_ID'" );
    }

    wp_set_post_cats('', $post_ID, $post_category);

    if ( $file )
        add_post_meta($post_ID, '_wp_attached_file', $file);

    clean_post_cache($post_ID);

    if ( $update) {
        do_action('edit_attachment', $post_ID);
    } else {
        do_action('add_attachment', $post_ID);
    }
    
    return $post_ID;
}

function wp_delete_attachment($postid) {
    global $wpdb;
    $postid = (int) $postid;

    if ( !$post = $wpdb->get_row("SELECT * FROM $wpdb->posts WHERE ID = $postid") )
        return $post;

    if ( 'attachment' != $post->post_status )
        return false;

    $meta = get_post_meta($postid, '_wp_attachment_metadata', true);
    $file = get_post_meta($postid, '_wp_attached_file', true);

    $wpdb->query("DELETE FROM $wpdb->posts WHERE ID = $postid");

    $wpdb->query("DELETE FROM $wpdb->comments WHERE comment_post_ID = $postid");

    $wpdb->query("DELETE FROM $wpdb->post2cat WHERE post_id = $postid");

    $wpdb->query("DELETE FROM $wpdb->postmeta WHERE post_id = $postid");

    if ( ! empty($meta['thumb']) ) {
        // Don't delete the thumb if another attachment uses it
        if (! $foo = $wpdb->get_row("SELECT meta_id FROM $wpdb->postmeta WHERE meta_key = '_wp_attachment_metadata' AND meta_value LIKE '%".$wpdb->escape($meta['thumb'])."%' AND post_id <> $postid"))
            @ unlink(str_replace(basename($file), $meta['thumb'], $file));
    }

    if ( ! empty($file) )
        @ unlink($file);

    do_action('delete_attachment', $postid);

    return $post;
}

function wp_get_single_post($postid = 0, $mode = OBJECT) {
    global $wpdb;

    $post = get_post($postid, $mode);
    
    // Set categories
    if($mode == OBJECT) {
        $post->post_category = wp_get_post_cats('',$postid);
    } 
    else {
        $post['post_category'] = wp_get_post_cats('',$postid);
    }

    return $post;
}

function wp_get_recent_posts($num = 10) {
    global $wpdb;

    // Set the limit clause, if we got a limit
    if ($num) {
        $limit = "LIMIT $num";
    }

    $sql = "SELECT * FROM $wpdb->posts WHERE post_status IN ('publish', 'draft', 'private') ORDER BY post_date DESC $limit";
    $result = $wpdb->get_results($sql,ARRAY_A);

    return $result?$result:array();
}

function wp_update_post($postarr = array()) {
    global $wpdb;

    if ( is_object($postarr) )
        $postarr = get_object_vars($postarr);

    // First, get all of the original fields
    $post = wp_get_single_post($postarr['ID'], ARRAY_A);    

    // Escape data pulled from DB.
    $post = add_magic_quotes($post);

    // Passed post category list overwrites existing category list if not empty.
     if ( isset($postarr['post_category']) && is_array($postarr['post_category'])
             && 0 != count($postarr['post_category']) )
         $post_cats = $postarr['post_category'];
     else 
         $post_cats = $post['post_category'];

    // Drafts shouldn't be assigned a date unless explicitly done so by the user
    if ( 'draft' == $post['post_status'] && empty($postarr['edit_date']) && empty($postarr['post_date']) && 
         ('0000-00-00 00:00:00' == $post['post_date']) )
        $clear_date = true;
    else
        $clear_date = false;

     // Merge old and new fields with new fields overwriting old ones.
     $postarr = array_merge($post, $postarr);
     $postarr['post_category'] = $post_cats;    
    if ( $clear_date ) {
        $postarr['post_date'] = '';
        $postarr['post_date_gmt'] = '';
    }

    if ($postarr['post_status'] == 'attachment')
        return wp_insert_attachment($postarr);

    return wp_insert_post($postarr);
}

function wp_get_post_cats($blogid = '1', $post_ID = 0) {
    global $wpdb;
    
    $sql = "SELECT category_id 
        FROM $wpdb->post2cat 
        WHERE post_id = $post_ID 
        ORDER BY category_id";

    $result = $wpdb->get_col($sql);

    if ( !$result )
        $result = array();

    return array_unique($result);
}

function wp_set_post_cats($blogid = '1', $post_ID = 0, $post_categories = array()) {
    global $wpdb;
    // If $post_categories isn't already an array, make it one:
    if (!is_array($post_categories) || 0 == count($post_categories))
        $post_categories = array(get_option('default_category'));
    
    $post_categories = array_unique($post_categories);

    // First the old categories
    $old_categories = $wpdb->get_col("
        SELECT category_id 
        FROM $wpdb->post2cat 
        WHERE post_id = $post_ID");
    
    if (!$old_categories) {
        $old_categories = array();
    } else {
        $old_categories = array_unique($old_categories);
    }


    $oldies = printr($old_categories,1);
    $newbies = printr($post_categories,1);

    // Delete any?
    $delete_cats = array_diff($old_categories,$post_categories);

    if ($delete_cats) {
        foreach ($delete_cats as $del) {
            $wpdb->query("
                DELETE FROM $wpdb->post2cat 
                WHERE category_id = $del 
                    AND post_id = $post_ID 
                ");
        }
    }

    // Add any?
    $add_cats = array_diff($post_categories, $old_categories);

    if ($add_cats) {
        foreach ($add_cats as $new_cat) {
            $wpdb->query("
                INSERT INTO $wpdb->post2cat (post_id, category_id) 
                VALUES ($post_ID, $new_cat)");
        }
    }
    
    // Update category counts.
    $all_affected_cats = array_unique(array_merge($post_categories, $old_categories));
    foreach ( $all_affected_cats as $cat_id ) {
        $count = $wpdb->get_var("SELECT COUNT(*) FROM $wpdb->post2cat, $wpdb->posts WHERE $wpdb->posts.ID=$wpdb->post2cat.post_id AND post_status='publish' AND category_id = '$cat_id'");
        $wpdb->query("UPDATE $wpdb->categories SET category_count = '$count' WHERE cat_ID = '$cat_id'");
        wp_cache_delete($cat_id, 'category');        
    }
}    // wp_set_post_cats()

function wp_delete_post($postid = 0) {
    global $wpdb;
    $postid = (int) $postid;

    if ( !$post = $wpdb->get_row("SELECT * FROM $wpdb->posts WHERE ID = $postid") )