root/tags/1.5-rc1/wp-app.php

<?php
/*
 * wp-app.php - Atom Publishing Protocol support for WordPress
 * Original code by: Elias Torres, http://torrez.us/archives/2006/08/31/491/
 * Modified by: Dougal Campbell, http://dougal.gunters.org/
 *
 * Version: 1.0.5-dc
 */

define('APP_REQUEST', true);

require_once('./wp-config.php');
require_once(ABSPATH . WPINC . '/post-template.php');
require_once(ABSPATH . WPINC . '/atomlib.php');
require_once(ABSPATH . WPINC . '/feed.php');

$_SERVER['PATH_INFO'] = preg_replace( '/.*\/wp-app\.php/', '', $_SERVER['REQUEST_URI'] );

$app_logging = 0;

// TODO: Should be an option somewhere
$always_authenticate = 1;

function log_app($label,$msg) {
    global $app_logging;
    if ($app_logging) {
        $fp = fopen( 'wp-app.log', 'a+');
        $date = gmdate( 'Y-m-d H:i:s' );
        fwrite($fp, "\n\n$date - $label\n$msg\n");
        fclose($fp);
    }
}

if ( !function_exists('wp_set_current_user') ) :
function wp_set_current_user($id, $name = '') {
    global $current_user;

    if ( isset($current_user) && ($id == $current_user->ID) )
        return $current_user;

    $current_user = new WP_User($id, $name);

    return $current_user;
}
endif;

function wa_posts_where_include_drafts_filter($where) {
        $where = str_replace("post_status = 'publish'","post_status = 'publish' OR post_status = 'future' OR post_status = 'draft' OR post_status = 'inherit'", $where);
        return $where;

}
add_filter('posts_where', 'wa_posts_where_include_drafts_filter');

class AtomServer {

    var $ATOM_CONTENT_TYPE = 'application/atom+xml';
    var $CATEGORIES_CONTENT_TYPE = 'application/atomcat+xml';
    var $SERVICE_CONTENT_TYPE = 'application/atomsvc+xml';

    var $ATOM_NS = 'http://www.w3.org/2005/Atom';
    var $ATOMPUB_NS = 'http://www.w3.org/2007/app';

    var $ENTRIES_PATH = "posts";
    var $CATEGORIES_PATH = "categories";
    var $MEDIA_PATH = "attachments";
    var $ENTRY_PATH = "post";
    var $SERVICE_PATH = "service";
    var $MEDIA_SINGLE_PATH = "attachment";

    var $params = array();
    var $media_content_types = array('image/*','audio/*','video/*');
    var $atom_content_types = array('application/atom+xml');

    var $selectors = array();

    // support for head
    var $do_output = true;

    function AtomServer() {

        $this->script_name = array_pop(explode('/',$_SERVER['SCRIPT_NAME']));
        $this->app_base = get_bloginfo('url') . '/' . $this->script_name . '/';
        if ( isset($_SERVER['HTTPS']) && strtolower($_SERVER['HTTPS']) == 'on' ) {
            $this->app_base = preg_replace( '/^http:\/\//', 'https://', $this->app_base );
        }

        $this->selectors = array(
            '@/service$@' =>
                array('GET' => 'get_service'),
            '@/categories$@' =>
                array('GET' => 'get_categories_xml'),
            '@/post/(\d+)$@' =>
                array('GET' => 'get_post',
                        'PUT' => 'put_post',
                        'DELETE' => 'delete_post'),
            '@/posts/?(\d+)?$@' =>
                array('GET' => 'get_posts',
                        'POST' => 'create_post'),
            '@/attachments/?(\d+)?$@' =>
                array('GET' => 'get_attachment',
                        'POST' => 'create_attachment'),
            '@/attachment/file/(\d+)$@' =>
                array('GET' => 'get_file',
                        'PUT' => 'put_file',
                        'DELETE' => 'delete_file'),
            '@/attachment/(\d+)$@' =>
                array('GET' => 'get_attachment',
                        'PUT' => 'put_attachment',
                        'DELETE' => 'delete_attachment'),
        );
    }

    function handle_request() {
        global $always_authenticate;

        $path = $_SERVER['PATH_INFO'];
        $method = $_SERVER['REQUEST_METHOD'];

        log_app('REQUEST',"$method $path\n================");

        $this->process_conditionals();
        //$this->process_conditionals();

        // exception case for HEAD (treat exactly as GET, but don't output)
        if($method == 'HEAD') {
            $this->do_output = false;
            $method = 'GET';
        }

        // redirect to /service in case no path is found.
        if(strlen($path) == 0 || $path == '/') {
            $this->redirect($this->get_service_url());
        }

        // dispatch
        foreach($this->selectors as $regex => $funcs) {
            if(preg_match($regex, $path, $matches)) {
            if(isset($funcs[$method])) {

                // authenticate regardless of the operation and set the current
                // user. each handler will decide if auth is required or not.
                $this->authenticate();
                $u = wp_get_current_user();
                if(!isset($u) || $u->ID == 0) {
                    if ($always_authenticate) {
                        $this->auth_required('Credentials required.');
                    }
                }

                array_shift($matches);
                call_user_func_array(array(&$this,$funcs[$method]), $matches);
                exit();
            } else {
                // only allow what we have handlers for...
                $this->not_allowed(array_keys($funcs));
            }
            }
        }

        // oops, nothing found
        $this->not_found();
    }

    function get_service() {
        log_app('function','get_service()');

        if( !current_user_can( 'edit_posts' ) )
            $this->auth_required( __( 'Sorry, you do not have the right to access this blog.' ) );

        $entries_url = attribute_escape($this->get_entries_url());
        $categories_url = attribute_escape($this->get_categories_url());
        $media_url = attribute_escape($this->get_attachments_url());
                foreach ($this->media_content_types as $med) {
                  $accepted_media_types = $accepted_media_types . "<accept>" . $med . "</accept>";
                }
        $atom_prefix="atom";
        $atom_blogname=get_bloginfo('name');
        $service_doc = <<<EOD
<service xmlns="$this->ATOMPUB_NS" xmlns:$atom_prefix="$this->ATOM_NS">
  <workspace>
    <$atom_prefix:title>$atom_blogname Workspace</$atom_prefix:title>
    <collection href="$entries_url">
      <$atom_prefix:title>$atom_blogname Posts</$atom_prefix:title>
      <accept>$this->ATOM_CONTENT_TYPE;type=entry</accept>
      <categories href="$categories_url" />
    </collection>
    <collection href="$media_url">
      <$atom_prefix:title>$atom_blogname Media</$atom_prefix:title>
      $accepted_media_types
    </collection>
  </workspace>
</service>

EOD;

        $this->output($service_doc, $this->SERVICE_CONTENT_TYPE);
    }

    function get_categories_xml() {
        log_app('function','get_categories_xml()');

        if( !current_user_can( 'edit_posts' ) )
            $this->auth_required( __( 'Sorry, you do not have the right to access this blog.' ) );

        $home = attribute_escape(get_bloginfo_rss('home'));

        $categories = "";
        $cats = get_categories("hierarchical=0&hide_empty=0");
        foreach ((array) $cats as $cat) {
            $categories .= "    <category term=\"" . attribute_escape($cat->name) .  "\" />\n";
}
        $output = <<<EOD
<app:categories xmlns:app="$this->ATOMPUB_NS"
    xmlns="$this->ATOM_NS"
    fixed="yes" scheme="$home">
    $categories
</app:categories>
EOD;
    $this->output($output, $this->CATEGORIES_CONTENT_TYPE);
}

    /*
     * Create Post (No arguments)
     */
    function create_post() {
        global $blog_id, $user_ID;
        $this->get_accepted_content_type($this->atom_content_types);

        $parser = new AtomParser();
        if(!$parser->parse()) {
            $this->client_error();
        }

        $entry = array_pop($parser->feed->entries);

        log_app('Received entry:', print_r($entry,true));

        $catnames = array();
        foreach($entry->categories as $cat)
            array_push($catnames, $cat["term"]);

        $wp_cats = get_categories(array('hide_empty' => false));

        $post_category = array();

        foreach($wp_cats as $cat) {
            if(in_array($cat->name, $catnames))
                array_push($post_category, $cat->term_id);
        }

        $publish = (isset($entry->draft) && trim($entry->draft) == 'yes') ? false : true;

        $cap = ($publish) ? 'publish_posts' : 'edit_posts';

        if(!current_user_can($cap))
            $this->auth_required(__('Sorry, you do not have the right to edit/publish new posts.'));

        $blog_ID = (int ) $blog_id;
        $post_status = ($publish) ? 'publish' : 'draft';
        $post_author = (int) $user_ID;
        $post_title = $entry->title[1];
        $post_content = $entry->content[1];
        $post_excerpt = $entry->summary[1];
        $pubtimes = $this->get_publish_time($entry->published);
        $post_date = $pubtimes[0];
        $post_date_gmt = $pubtimes[1];

        if ( isset( $_SERVER['HTTP_SLUG'] ) )
            $post_name = $_SERVER['HTTP_SLUG'];

        $post_data = compact('blog_ID', 'post_author', 'post_date', 'post_date_gmt', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_name');

        $this->escape($post_data);
        log_app('Inserting Post. Data:', print_r($post_data,true));

        $postID = wp_insert_post($post_data);
        if ( is_wp_error( $postID ) )
            $this->internal_error($postID->get_error_message());

        if (!$postID)
            $this->internal_error(__('Sorry, your entry could not be posted. Something wrong happened.'));

        // getting warning here about unable to set headers
        // because something in the cache is printing to the buffer
        // could we clean up wp_set_post_categories or cache to not print
        // this could affect our ability to send back the right headers
        @wp_set_post_categories($postID, $post_category);

        $output = $this->get_entry($postID);

        log_app('function',"create_post($postID)");
        $this->created($postID, $output);
    }

    function get_post($postID) {
        global $entry;

        if( !current_user_can( 'edit_post', $postID ) )
            $this->auth_required( __( 'Sorry, you do not have the right to access this post.' ) );

        $this->set_current_entry($postID);
        $output = $this->get_entry($postID);
        log_app('function',"get_post($postID)");
        $this->output($output);

    }

    function put_post($postID) {
        // checked for valid content-types (atom+xml)
        // quick check and exit
        $this->get_accepted_content_type($this->atom_content_types);

        $parser = new AtomParser();
        if(!$parser->parse()) {
            $this->bad_request();
        }

        $parsed = array_pop($parser->feed->entries);

        log_app('Received UPDATED entry:', print_r($parsed,true));

        // check for not found
        global $entry;
        $this->set_current_entry($postID);

        if(!current_user_can('edit_post', $entry['ID']))
            $this->auth_required(__('Sorry, you do not have the right to edit this post.'));

        $publish = (isset($parsed->draft) && trim($parsed->draft) == 'yes') ? false : true;

        extract($entry);

        $post_title = $parsed->title[1];
        $post_content = $parsed->content[1];
        $post_excerpt = $parsed->summary[1];
        $pubtimes = $this->get_publish_time($entry->published);
        $post_date = $pubtimes[0];
        $post_date_gmt = $pubtimes[1];
        $pubtimes = $this->get_publish_time($parsed->updated);
        $post_modified = $pubtimes[0];
        $post_modified_gmt = $pubtimes[1];

        // let's not go backwards and make something draft again.
        if(!$publish && $post_status == 'draft') {
            $post_status = ($publish) ? 'publish' : 'draft';
        } elseif($publish) {
            $post_status = 'publish';
        }

        $postdata = compact('ID', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_date', 'post_date_gmt', 'post_modified', 'post_modified_gmt');
        $this->escape($postdata);

        $result = wp_update_post($postdata);

        if (!$result) {
            $this->internal_error(__('For some strange yet very annoying reason, this post could not be edited.'));
        }

        log_app('function',"put_post($postID)");
        $this->ok();
    }

    function delete_post($postID) {

        // check for not found
        global $entry;
        $this->set_current_entry($postID);

        if(!current_user_can('edit_post', $postID)) {
            $this->auth_required(__('Sorry, you do not have the right to delete this post.'));
        }

        if ($entry['post_type'] == 'attachment') {
            $this->delete_attachment($postID);
        } else {
            $result = wp_delete_post($postID);

            if (!$result) {
                $this->internal_error(__('For some strange yet very annoying reason, this post could not be deleted.'));
            }

            log_app('function',"delete_post($postID)");
            $this->ok();
        }

    }

    function get_attachment($postID = NULL) {
        if( !current_user_can( 'upload_files' ) )
            $this->auth_required( __( 'Sorry, you do not have permission to upload files.' ) );

        if (!isset($postID)) {
            $this->get_attachments();
        } else {
            $this->set_current_entry($postID);
            $output = $this->get_entry($postID, 'attachment');
            log_app('function',"get_attachment($postID)");
            $this->output($output);
        }
    }

    function create_attachment() {

        $type = $this->get_accepted_content_type();

        if(!current_user_can('upload_files'))
            $this->auth_required(__('You do not have permission to upload files.'));

        $fp = fopen("php://input", "rb");
        $bits = NULL;
        while(!feof($fp)) {
            $bits .= fread($fp, 4096);
        }
        fclose($fp);

        $slug = '';
        if ( isset( $_SERVER['HTTP_SLUG'] ) )
            $slug = sanitize_file_name( $_SERVER['HTTP_SLUG'] );
        elseif ( isset( $_SERVER['HTTP_TITLE'] ) )
            $slug = sanitize_file_name( $_SERVER['HTTP_TITLE'] );
        elseif ( empty( $slug ) ) // just make a random name
            $slug = substr( md5( uniqid( microtime() ) ), 0, 7);
        $ext = preg_replace( '|.*/([a-z0-9]+)|', '$1', $_SERVER['CONTENT_TYPE'] );
        $slug = "$slug.$ext";
        $file = wp_upload_bits( $slug, NULL, $bits);

        log_app('wp_upload_bits returns:',print_r($file,true));

        $url = $file['url'];
        $file = $file['file'];

        do_action('wp_create_file_in_uploads', $file); // replicate

        // Construct the attachment array
        $attachment = array(
            'post_title' => $slug,
            'post_content' => $slug,
            'post_status' => 'attachment',
            'post_parent' => 0,
            'post_mime_type' => $type,
            'guid' => $url
            );

        // Save the data
        $postID = wp_insert_attachment($attachment, $file);

        if (!$postID)
            $this->internal_error(__('Sorry, your entry could not be posted. Something wrong happened.'));

        $output = $this->get_entry($postID, 'attachment');

        $this->created($postID, $output, 'attachment');
        log_app('function',"create_attachment($postID)");
    }

    function put_attachment($postID) {
        // checked for valid content-types (atom+xml)
        // quick check and exit
        $this->get_accepted_content_type($this->atom_content_types);

        $parser = new AtomParser();
        if(!$parser->parse()) {
            $this->bad_request();
        }

        $parsed = array_pop($parser->feed->entries);

        // check for not found
        global $entry;
        $this->set_current_entry($postID);

        if(!current_user_can('edit_post', $entry['ID']))
            $this->auth_required(__('Sorry, you do not have the right to edit this post.'));

        extract($entry);

        $post_title = $parsed->title[1];
        $post_content = $parsed->content[1];
        $pubtimes = $this->get_publish_time($parsed->updated);
        $post_modified = $pubtimes[0];
        $post_modified_gmt = $pubtimes[1];

        $postdata = compact('ID', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_modified', 'post_modified_gmt');
        $this->escape($postdata);