Wordpress 2.5.1-scripts
[autoinstalls/wordpress.git] / wp-app.php
1 <?php
2 /*
3  * wp-app.php - Atom Publishing Protocol support for WordPress
4  * Original code by: Elias Torres, http://torrez.us/archives/2006/08/31/491/
5  * Modified by: Dougal Campbell, http://dougal.gunters.org/
6  *
7  * Version: 1.0.5-dc
8  */
9
10 define('APP_REQUEST', true);
11
12 require_once('./wp-config.php');
13 require_once(ABSPATH . WPINC . '/post-template.php');
14 require_once(ABSPATH . WPINC . '/atomlib.php');
15 require_once(ABSPATH . WPINC . '/feed.php');
16
17 $_SERVER['PATH_INFO'] = preg_replace( '/.*\/wp-app\.php/', '', $_SERVER['REQUEST_URI'] );
18
19 $app_logging = 0;
20
21 // TODO: Should be an option somewhere
22 $always_authenticate = 1;
23
24 function log_app($label,$msg) {
25         global $app_logging;
26         if ($app_logging) {
27                 $fp = fopen( 'wp-app.log', 'a+');
28                 $date = gmdate( 'Y-m-d H:i:s' );
29                 fwrite($fp, "\n\n$date - $label\n$msg\n");
30                 fclose($fp);
31         }
32 }
33
34 if ( !function_exists('wp_set_current_user') ) :
35 function wp_set_current_user($id, $name = '') {
36         global $current_user;
37
38         if ( isset($current_user) && ($id == $current_user->ID) )
39                 return $current_user;
40
41         $current_user = new WP_User($id, $name);
42
43         return $current_user;
44 }
45 endif;
46
47 function wa_posts_where_include_drafts_filter($where) {
48         $where = str_replace("post_status = 'publish'","post_status = 'publish' OR post_status = 'future' OR post_status = 'draft' OR post_status = 'inherit'", $where);
49         return $where;
50
51 }
52 add_filter('posts_where', 'wa_posts_where_include_drafts_filter');
53
54 class AtomServer {
55
56         var $ATOM_CONTENT_TYPE = 'application/atom+xml';
57         var $CATEGORIES_CONTENT_TYPE = 'application/atomcat+xml';
58         var $SERVICE_CONTENT_TYPE = 'application/atomsvc+xml';
59
60         var $ATOM_NS = 'http://www.w3.org/2005/Atom';
61         var $ATOMPUB_NS = 'http://www.w3.org/2007/app';
62
63         var $ENTRIES_PATH = "posts";
64         var $CATEGORIES_PATH = "categories";
65         var $MEDIA_PATH = "attachments";
66         var $ENTRY_PATH = "post";
67         var $SERVICE_PATH = "service";
68         var $MEDIA_SINGLE_PATH = "attachment";
69
70         var $params = array();
71         var $media_content_types = array('image/*','audio/*','video/*');
72         var $atom_content_types = array('application/atom+xml');
73
74         var $selectors = array();
75
76         // support for head
77         var $do_output = true;
78
79         function AtomServer() {
80
81                 $this->script_name = array_pop(explode('/',$_SERVER['SCRIPT_NAME']));
82                 $this->app_base = get_bloginfo('url') . '/' . $this->script_name . '/';
83                 if ( isset($_SERVER['HTTPS']) && strtolower($_SERVER['HTTPS']) == 'on' ) {
84                         $this->app_base = preg_replace( '/^http:\/\//', 'https://', $this->app_base );
85                 }
86
87                 $this->selectors = array(
88                         '@/service$@' =>
89                                 array('GET' => 'get_service'),
90                         '@/categories$@' =>
91                                 array('GET' => 'get_categories_xml'),
92                         '@/post/(\d+)$@' =>
93                                 array('GET' => 'get_post',
94                                                 'PUT' => 'put_post',
95                                                 'DELETE' => 'delete_post'),
96                         '@/posts/?(\d+)?$@' =>
97                                 array('GET' => 'get_posts',
98                                                 'POST' => 'create_post'),
99                         '@/attachments/?(\d+)?$@' =>
100                                 array('GET' => 'get_attachment',
101                                                 'POST' => 'create_attachment'),
102                         '@/attachment/file/(\d+)$@' =>
103                                 array('GET' => 'get_file',
104                                                 'PUT' => 'put_file',
105                                                 'DELETE' => 'delete_file'),
106                         '@/attachment/(\d+)$@' =>
107                                 array('GET' => 'get_attachment',
108                                                 'PUT' => 'put_attachment',
109                                                 'DELETE' => 'delete_attachment'),
110                 );
111         }
112
113         function handle_request() {
114                 global $always_authenticate;
115
116                 $path = $_SERVER['PATH_INFO'];
117                 $method = $_SERVER['REQUEST_METHOD'];
118
119                 log_app('REQUEST',"$method $path\n================");
120
121                 $this->process_conditionals();
122                 //$this->process_conditionals();
123
124                 // exception case for HEAD (treat exactly as GET, but don't output)
125                 if($method == 'HEAD') {
126                         $this->do_output = false;
127                         $method = 'GET';
128                 }
129
130                 // redirect to /service in case no path is found.
131                 if(strlen($path) == 0 || $path == '/') {
132                         $this->redirect($this->get_service_url());
133                 }
134
135                 // dispatch
136                 foreach($this->selectors as $regex => $funcs) {
137                         if(preg_match($regex, $path, $matches)) {
138                         if(isset($funcs[$method])) {
139
140                                 // authenticate regardless of the operation and set the current
141                                 // user. each handler will decide if auth is required or not.
142                                 $this->authenticate();
143                                 $u = wp_get_current_user();
144                                 if(!isset($u) || $u->ID == 0) {
145                                         if ($always_authenticate) {
146                                                 $this->auth_required('Credentials required.');
147                                         }
148                                 }
149
150                                 array_shift($matches);
151                                 call_user_func_array(array(&$this,$funcs[$method]), $matches);
152                                 exit();
153                         } else {
154                                 // only allow what we have handlers for...
155                                 $this->not_allowed(array_keys($funcs));
156                         }
157                         }
158                 }
159
160                 // oops, nothing found
161                 $this->not_found();
162         }
163
164         function get_service() {
165                 log_app('function','get_service()');
166
167                 if( !current_user_can( 'edit_posts' ) )
168                         $this->auth_required( __( 'Sorry, you do not have the right to access this blog.' ) );
169
170                 $entries_url = attribute_escape($this->get_entries_url());
171                 $categories_url = attribute_escape($this->get_categories_url());
172                 $media_url = attribute_escape($this->get_attachments_url());
173                 foreach ($this->media_content_types as $med) {
174                   $accepted_media_types = $accepted_media_types . "<accept>" . $med . "</accept>";
175                 }
176                 $atom_prefix="atom";
177                 $atom_blogname=get_bloginfo('name');
178                 $service_doc = <<<EOD
179 <service xmlns="$this->ATOMPUB_NS" xmlns:$atom_prefix="$this->ATOM_NS">
180   <workspace>
181     <$atom_prefix:title>$atom_blogname Workspace</$atom_prefix:title>
182     <collection href="$entries_url">
183       <$atom_prefix:title>$atom_blogname Posts</$atom_prefix:title>
184       <accept>$this->ATOM_CONTENT_TYPE;type=entry</accept>
185       <categories href="$categories_url" />
186     </collection>
187     <collection href="$media_url">
188       <$atom_prefix:title>$atom_blogname Media</$atom_prefix:title>
189       $accepted_media_types
190     </collection>
191   </workspace>
192 </service>
193
194 EOD;
195
196                 $this->output($service_doc, $this->SERVICE_CONTENT_TYPE);
197         }
198
199         function get_categories_xml() {
200                 log_app('function','get_categories_xml()');
201
202                 if( !current_user_can( 'edit_posts' ) )
203                         $this->auth_required( __( 'Sorry, you do not have the right to access this blog.' ) );
204
205                 $home = attribute_escape(get_bloginfo_rss('home'));
206
207                 $categories = "";
208                 $cats = get_categories("hierarchical=0&hide_empty=0");
209                 foreach ((array) $cats as $cat) {
210                         $categories .= "    <category term=\"" . attribute_escape($cat->name) .  "\" />\n";
211 }
212                 $output = <<<EOD
213 <app:categories xmlns:app="$this->ATOMPUB_NS"
214         xmlns="$this->ATOM_NS"
215         fixed="yes" scheme="$home">
216         $categories
217 </app:categories>
218 EOD;
219         $this->output($output, $this->CATEGORIES_CONTENT_TYPE);
220 }
221
222         /*
223          * Create Post (No arguments)
224          */
225         function create_post() {
226                 global $blog_id, $user_ID;
227                 $this->get_accepted_content_type($this->atom_content_types);
228
229                 $parser = new AtomParser();
230                 if(!$parser->parse()) {
231                         $this->client_error();
232                 }
233
234                 $entry = array_pop($parser->feed->entries);
235
236                 log_app('Received entry:', print_r($entry,true));
237
238                 $catnames = array();
239                 foreach($entry->categories as $cat)
240                         array_push($catnames, $cat["term"]);
241
242                 $wp_cats = get_categories(array('hide_empty' => false));
243
244                 $post_category = array();
245
246                 foreach($wp_cats as $cat) {
247                         if(in_array($cat->name, $catnames))
248                                 array_push($post_category, $cat->term_id);
249                 }
250
251                 $publish = (isset($entry->draft) && trim($entry->draft) == 'yes') ? false : true;
252
253                 $cap = ($publish) ? 'publish_posts' : 'edit_posts';
254
255                 if(!current_user_can($cap))
256                         $this->auth_required(__('Sorry, you do not have the right to edit/publish new posts.'));
257
258                 $blog_ID = (int ) $blog_id;
259                 $post_status = ($publish) ? 'publish' : 'draft';
260                 $post_author = (int) $user_ID;
261                 $post_title = $entry->title[1];
262                 $post_content = $entry->content[1];
263                 $post_excerpt = $entry->summary[1];
264                 $pubtimes = $this->get_publish_time($entry->published);
265                 $post_date = $pubtimes[0];
266                 $post_date_gmt = $pubtimes[1];
267
268                 if ( isset( $_SERVER['HTTP_SLUG'] ) )
269                         $post_name = $_SERVER['HTTP_SLUG'];
270
271                 $post_data = compact('blog_ID', 'post_author', 'post_date', 'post_date_gmt', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_name');
272
273                 $this->escape($post_data);
274                 log_app('Inserting Post. Data:', print_r($post_data,true));
275
276                 $postID = wp_insert_post($post_data);
277                 if ( is_wp_error( $postID ) )
278                         $this->internal_error($postID->get_error_message());
279
280                 if (!$postID)
281                         $this->internal_error(__('Sorry, your entry could not be posted. Something wrong happened.'));
282
283                 // getting warning here about unable to set headers
284                 // because something in the cache is printing to the buffer
285                 // could we clean up wp_set_post_categories or cache to not print
286                 // this could affect our ability to send back the right headers
287                 @wp_set_post_categories($postID, $post_category);
288
289                 $output = $this->get_entry($postID);
290
291                 log_app('function',"create_post($postID)");
292                 $this->created($postID, $output);
293         }
294
295         function get_post($postID) {
296                 global $entry;
297
298                 if( !current_user_can( 'edit_post', $postID ) )
299                         $this->auth_required( __( 'Sorry, you do not have the right to access this post.' ) );
300
301                 $this->set_current_entry($postID);
302                 $output = $this->get_entry($postID);
303                 log_app('function',"get_post($postID)");
304                 $this->output($output);
305
306         }
307
308         function put_post($postID) {
309                 // checked for valid content-types (atom+xml)
310                 // quick check and exit
311                 $this->get_accepted_content_type($this->atom_content_types);
312
313                 $parser = new AtomParser();
314                 if(!$parser->parse()) {
315                         $this->bad_request();
316                 }
317
318                 $parsed = array_pop($parser->feed->entries);
319
320                 log_app('Received UPDATED entry:', print_r($parsed,true));
321
322                 // check for not found
323                 global $entry;
324                 $this->set_current_entry($postID);
325
326                 if(!current_user_can('edit_post', $entry['ID']))
327                         $this->auth_required(__('Sorry, you do not have the right to edit this post.'));
328
329                 $publish = (isset($parsed->draft) && trim($parsed->draft) == 'yes') ? false : true;
330
331                 extract($entry);
332
333                 $post_title = $parsed->title[1];
334                 $post_content = $parsed->content[1];
335                 $post_excerpt = $parsed->summary[1];
336                 $pubtimes = $this->get_publish_time($entry->published);
337                 $post_date = $pubtimes[0];
338                 $post_date_gmt = $pubtimes[1];
339                 $pubtimes = $this->get_publish_time($parsed->updated);
340                 $post_modified = $pubtimes[0];
341                 $post_modified_gmt = $pubtimes[1];
342
343                 // let's not go backwards and make something draft again.
344                 if(!$publish && $post_status == 'draft') {
345                         $post_status = ($publish) ? 'publish' : 'draft';
346                 } elseif($publish) {
347                         $post_status = 'publish';
348                 }
349
350                 $postdata = compact('ID', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_date', 'post_date_gmt', 'post_modified', 'post_modified_gmt');
351                 $this->escape($postdata);
352
353                 $result = wp_update_post($postdata);
354
355                 if (!$result) {
356                         $this->internal_error(__('For some strange yet very annoying reason, this post could not be edited.'));
357                 }
358
359                 log_app('function',"put_post($postID)");
360                 $this->ok();
361         }
362
363         function delete_post($postID) {
364
365                 // check for not found
366                 global $entry;
367                 $this->set_current_entry($postID);
368
369                 if(!current_user_can('edit_post', $postID)) {
370                         $this->auth_required(__('Sorry, you do not have the right to delete this post.'));
371                 }
372
373                 if ($entry['post_type'] == 'attachment') {
374                         $this->delete_attachment($postID);
375                 } else {
376                         $result = wp_delete_post($postID);
377
378                         if (!$result) {
379                                 $this->internal_error(__('For some strange yet very annoying reason, this post could not be deleted.'));
380                         }
381
382                         log_app('function',"delete_post($postID)");
383                         $this->ok();
384                 }
385
386         }
387
388         function get_attachment($postID = NULL) {
389                 if( !current_user_can( 'upload_files' ) )
390                         $this->auth_required( __( 'Sorry, you do not have permission to upload files.' ) );
391
392                 if (!isset($postID)) {
393                         $this->get_attachments();
394                 } else {
395                         $this->set_current_entry($postID);
396                         $output = $this->get_entry($postID, 'attachment');
397                         log_app('function',"get_attachment($postID)");
398                         $this->output($output);
399                 }
400         }
401
402         function create_attachment() {
403
404                 $type = $this->get_accepted_content_type();
405
406                 if(!current_user_can('upload_files'))
407                         $this->auth_required(__('You do not have permission to upload files.'));
408
409                 $fp = fopen("php://input", "rb");
410                 $bits = NULL;
411                 while(!feof($fp)) {
412                         $bits .= fread($fp, 4096);
413                 }
414                 fclose($fp);
415
416                 $slug = '';
417                 if ( isset( $_SERVER['HTTP_SLUG'] ) )
418                         $slug = sanitize_file_name( $_SERVER['HTTP_SLUG'] );
419                 elseif ( isset( $_SERVER['HTTP_TITLE'] ) )
420                         $slug = sanitize_file_name( $_SERVER['HTTP_TITLE'] );
421                 elseif ( empty( $slug ) ) // just make a random name
422                         $slug = substr( md5( uniqid( microtime() ) ), 0, 7);
423                 $ext = preg_replace( '|.*/([a-z0-9]+)|', '$1', $_SERVER['CONTENT_TYPE'] );
424                 $slug = "$slug.$ext";
425                 $file = wp_upload_bits( $slug, NULL, $bits);
426
427                 log_app('wp_upload_bits returns:',print_r($file,true));
428
429                 $url = $file['url'];
430                 $file = $file['file'];
431
432                 do_action('wp_create_file_in_uploads', $file); // replicate
433
434                 // Construct the attachment array
435                 $attachment = array(
436                         'post_title' => $slug,
437                         'post_content' => $slug,
438                         'post_status' => 'attachment',
439                         'post_parent' => 0,
440                         'post_mime_type' => $type,
441                         'guid' => $url
442                         );
443
444                 // Save the data
445                 $postID = wp_insert_attachment($attachment, $file);
446
447                 if (!$postID)
448                         $this->internal_error(__('Sorry, your entry could not be posted. Something wrong happened.'));
449
450                 $output = $this->get_entry($postID, 'attachment');
451
452                 $this->created($postID, $output, 'attachment');
453                 log_app('function',"create_attachment($postID)");
454         }
455
456         function put_attachment($postID) {
457                 // checked for valid content-types (atom+xml)
458                 // quick check and exit
459                 $this->get_accepted_content_type($this->atom_content_types);
460
461                 $parser = new AtomParser();
462                 if(!$parser->parse()) {
463                         $this->bad_request();
464                 }
465
466                 $parsed = array_pop($parser->feed->entries);
467
468                 // check for not found
469                 global $entry;
470                 $this->set_current_entry($postID);
471
472                 if(!current_user_can('edit_post', $entry['ID']))
473                         $this->auth_required(__('Sorry, you do not have the right to edit this post.'));
474
475                 extract($entry);
476
477                 $post_title = $parsed->title[1];
478                 $post_content = $parsed->content[1];
479                 $pubtimes = $this->get_publish_time($parsed->updated);
480                 $post_modified = $pubtimes[0];
481                 $post_modified_gmt = $pubtimes[1];
482
483                 $postdata = compact('ID', 'post_content', 'post_title', 'post_category', 'post_status', 'post_excerpt', 'post_modified', 'post_modified_gmt');
484                 $this->escape($postdata);
485
486                 $result = wp_update_post($postdata);
487
488                 if (!$result) {
489                         $this->internal_error(__('For some strange yet very annoying reason, this post could not be edited.'));
490                 }
491
492                 log_app('function',"put_attachment($postID)");
493                 $this->ok();
494         }
495
496         function delete_attachment($postID) {
497                 log_app('function',"delete_attachment($postID). File '$location' deleted.");
498
499                 // check for not found
500                 global $entry;
501                 $this->set_current_entry($postID);
502
503                 if(!current_user_can('edit_post', $postID)) {
504                         $this->auth_required(__('Sorry, you do not have the right to delete this post.'));
505                 }
506
507                 $location = get_post_meta($entry['ID'], '_wp_attached_file', true);
508                 $filetype = wp_check_filetype($location);
509
510                 if(!isset($location) || 'attachment' != $entry['post_type'] || empty($filetype['ext']))
511                         $this->internal_error(__('Error ocurred while accessing post metadata for file location.'));
512
513                 // delete file
514                 @unlink($location);
515
516                 // delete attachment
517                 $result = wp_delete_post($postID);
518
519                 if (!$result) {
520                         $this->internal_error(__('For some strange yet very annoying reason, this post could not be deleted.'));
521                 }
522
523                 log_app('function',"delete_attachment($postID). File '$location' deleted.");
524                 $this->ok();
525         }
526
527         function get_file($postID) {
528
529                 // check for not found
530                 global $entry;
531                 $this->set_current_entry($postID);
532
533                 // then whether user can edit the specific post
534                 if(!current_user_can('edit_post', $postID)) {
535                         $this->auth_required(__('Sorry, you do not have the right to edit this post.'));
536                 }
537
538                 $location = get_post_meta($entry['ID'], '_wp_attached_file', true);
539                 $filetype = wp_check_filetype($location);
540
541                 if(!isset($location) || 'attachment' != $entry['post_type'] || empty($filetype['ext']))
542                         $this->internal_error(__('Error ocurred while accessing post metadata for file location.'));
543
544                 status_header('200');
545                 header('Content-Type: ' . $entry['post_mime_type']);
546                 header('Connection: close');
547
548                 $fp = fopen($location, "rb");
549                 while(!feof($fp)) {
550                         echo fread($fp, 4096);
551                 }
552                 fclose($fp);
553
554                 log_app('function',"get_file($postID)");
555                 exit;
556         }
557
558         function put_file($postID) {
559
560                 // first check if user can upload
561                 if(!current_user_can('upload_files'))
562                         $this->auth_required(__('You do not have permission to upload files.'));
563
564                 // check for not found
565                 global $entry;
566                 $this->set_current_entry($postID);
567
568                 // then whether user can edit the specific post
569                 if(!current_user_can('edit_post', $postID)) {
570                         $this->auth_required(__('Sorry, you do not have the right to edit this post.'));
571                 }
572
573                 $location = get_post_meta($entry['ID'], '_wp_attached_file', true);
574                 $filetype = wp_check_filetype($location);
575
576                 if(!isset($location) || 'attachment' != $entry['post_type'] || empty($filetype['ext']))
577                         $this->internal_error(__('Error ocurred while accessing post metadata for file location.'));
578
579                 $fp = fopen("php://input", "rb");
580                 $localfp = fopen($location, "w+");
581                 while(!feof($fp)) {
582                         fwrite($localfp,fread($fp, 4096));
583                 }
584                 fclose($fp);
585                 fclose($localfp);
586
587                 $ID = $entry['ID'];
588                 $pubtimes = $this->get_publish_time($entry->published);
589                 $post_date = $pubtimes[0];
590                 $post_date_gmt = $pubtimes[1];
591                 $pubtimes = $this->get_publish_time($parsed->updated);
592                 $post_modified = $pubtimes[0];
593                 $post_modified_gmt = $pubtimes[1];
594
595                 $post_data = compact('ID', 'post_date', 'post_date_gmt', 'post_modified', 'post_modified_gmt');
596                 $result = wp_update_post($post_data);
597
598                 if (!$result) {
599                         $this->internal_error(__('Sorry, your entry could not be posted. Something wrong happened.'));
600                 }
601
602                 log_app('function',"put_file($postID)");
603                 $this->ok();
604         }
605
606         function get_entries_url($page = NULL) {
607                 if($GLOBALS['post_type'] == 'attachment') {
608                         $path = $this->MEDIA_PATH;
609                 } else {
610                         $path = $this->ENTRIES_PATH;
611                 }
612                 $url = $this->app_base . $path;
613                 if(isset($page) && is_int($page)) {
614                         $url .= "/$page";
615                 }
616                 return $url;
617         }
618
619         function the_entries_url($page = NULL) {
620                 echo $this->get_entries_url($page);
621         }
622
623         function get_categories_url($deprecated = '') {
624                 return $this->app_base . $this->CATEGORIES_PATH;
625         }
626
627         function the_categories_url() {
628                 echo $this->get_categories_url();
629         }
630
631         function get_attachments_url($page = NULL) {
632                 $url = $this->app_base . $this->MEDIA_PATH;
633                 if(isset($page) && is_int($page)) {
634                         $url .= "/$page";
635                 }
636                 return $url;
637         }
638
639         function the_attachments_url($page = NULL) {
640                 echo $this->get_attachments_url($page);
641         }
642
643         function get_service_url() {
644                 return $this->app_base . $this->SERVICE_PATH;
645         }
646
647         function get_entry_url($postID = NULL) {
648                 if(!isset($postID)) {
649                         global $post;
650                         $postID = (int) $post->ID;
651                 }
652
653                 $url = $this->app_base . $this->ENTRY_PATH . "/$postID";
654
655                 log_app('function',"get_entry_url() = $url");
656                 return $url;
657         }
658
659         function the_entry_url($postID = NULL) {
660                 echo $this->get_entry_url($postID);
661         }
662
663         function get_media_url($postID = NULL) {
664                 if(!isset($postID)) {
665                         global $post;
666                         $postID = (int) $post->ID;
667                 }
668
669                 $url = $this->app_base . $this->MEDIA_SINGLE_PATH ."/file/$postID";
670
671                 log_app('function',"get_media_url() = $url");
672                 return $url;
673         }
674
675         function the_media_url($postID = NULL) {
676                 echo $this->get_media_url($postID);
677         }
678
679         function set_current_entry($postID) {
680                 global $entry;
681                 log_app('function',"set_current_entry($postID)");
682
683                 if(!isset($postID)) {
684                         // $this->bad_request();
685                         $this->not_found();
686                 }
687
688                 $entry = wp_get_single_post($postID,ARRAY_A);
689
690                 if(!isset($entry) || !isset($entry['ID']))
691                         $this->not_found();
692
693                 return;
694         }
695
696         function get_posts($page = 1, $post_type = 'post') {
697                         log_app('function',"get_posts($page, '$post_type')");
698                         $feed = $this->get_feed($page, $post_type);
699                         $this->output($feed);
700         }
701
702         function get_attachments($page = 1, $post_type = 'attachment') {
703             log_app('function',"get_attachments($page, '$post_type')");
704             $GLOBALS['post_type'] = $post_type;
705             $feed = $this->get_feed($page, $post_type);
706             $this->output($feed);
707         }
708
709         function get_feed($page = 1, $post_type = 'post') {
710                 global $post, $wp, $wp_query, $posts, $wpdb, $blog_id;
711                 log_app('function',"get_feed($page, '$post_type')");
712                 ob_start();
713
714                 if(!isset($page)) {
715                         $page = 1;
716                 }
717                 $page = (int) $page;
718
719                 $count = get_option('posts_per_rss');
720
721                 wp('what_to_show=posts&posts_per_page=' . $count . '&offset=' . ($count * ($page-1) . '&orderby=modified'));
722
723                 $post = $GLOBALS['post'];
724                 $posts = $GLOBALS['posts'];
725                 $wp = $GLOBALS['wp'];
726                 $wp_query = $GLOBALS['wp_query'];
727                 $wpdb = $GLOBALS['wpdb'];
728                 $blog_id = (int) $GLOBALS['blog_id'];
729                 log_app('function',"query_posts(# " . print_r($wp_query, true) . "#)");
730
731                 log_app('function',"total_count(# $wp_query->max_num_pages #)");
732                 $last_page = $wp_query->max_num_pages;
733                 $next_page = (($page + 1) > $last_page) ? NULL : $page + 1;
734                 $prev_page = ($page - 1) < 1 ? NULL : $page - 1;
735                 $last_page = ((int)$last_page == 1 || (int)$last_page == 0) ? NULL : (int) $last_page;
736                 $self_page = $page > 1 ? $page : NULL;
737 ?><feed xmlns="<?php echo $this->ATOM_NS ?>" xmlns:app="<?php echo $this->ATOMPUB_NS ?>" xml:lang="<?php echo get_option('rss_language'); ?>">
738 <id><?php $this->the_entries_url() ?></id>
739 <updated><?php echo mysql2date('Y-m-d\TH:i:s\Z', get_lastpostmodified('GMT')); ?></updated>
740 <title type="text"><?php bloginfo_rss('name') ?></title>
741 <subtitle type="text"><?php bloginfo_rss("description") ?></subtitle>
742 <link rel="first" type="<?php echo $this->ATOM_CONTENT_TYPE ?>" href="<?php $this->the_entries_url() ?>" />
743 <?php if(isset($prev_page)): ?>
744 <link rel="previous" type="<?php echo $this->ATOM_CONTENT_TYPE ?>" href="<?php $this->the_entries_url($prev_page) ?>" />
745 <?php endif; ?>
746 <?php if(isset($next_page)): ?>
747 <link rel="next" type="<?php echo $this->ATOM_CONTENT_TYPE ?>" href="<?php $this->the_entries_url($next_page) ?>" />
748 <?php endif; ?>
749 <link rel="last" type="<?php echo $this->ATOM_CONTENT_TYPE ?>" href="<?php $this->the_entries_url($last_page) ?>" />
750 <link rel="self" type="<?php echo $this->ATOM_CONTENT_TYPE ?>" href="<?php $this->the_entries_url($self_page) ?>" />
751 <rights type="text">Copyright <?php echo mysql2date('Y', get_lastpostdate('blog')); ?></rights>
752 <?php the_generator( 'atom' ); ?>
753 <?php if ( have_posts() ) {
754                         while ( have_posts() ) {
755                                 the_post();
756                                 $this->echo_entry();
757                         }
758                 }
759 ?></feed>
760 <?php
761                 $feed = ob_get_contents();
762                 ob_end_clean();
763                 return $feed;
764         }
765
766         function get_entry($postID, $post_type = 'post') {
767                 log_app('function',"get_entry($postID, '$post_type')");
768                 ob_start();
769                 switch($post_type) {
770                         case 'post':
771                                 $varname = 'p';
772                                 break;
773                         case 'attachment':
774                                 $varname = 'attachment_id';
775                                 break;
776                 }
777                 query_posts($varname . '=' . $postID);
778                 if ( have_posts() ) {
779                         while ( have_posts() ) {
780                                 the_post();
781                                 $this->echo_entry();
782                                 log_app('$post',print_r($GLOBALS['post'],true));
783                                 $entry = ob_get_contents();
784                                 break;
785                         }
786                 }
787                 ob_end_clean();
788
789                 log_app('get_entry returning:',$entry);
790                 return $entry;
791         }
792
793         function echo_entry() { ?>
794 <entry xmlns="<?php echo $this->ATOM_NS ?>"
795        xmlns:app="<?php echo $this->ATOMPUB_NS ?>" xml:lang="<?php echo get_option('rss_language'); ?>">
796         <id><?php the_guid($GLOBALS['post']->ID); ?></id>
797 <?php list($content_type, $content) = prep_atom_text_construct(get_the_title()); ?>
798         <title type="<?php echo $content_type ?>"><?php echo $content ?></title>
799         <updated><?php echo get_post_modified_time('Y-m-d\TH:i:s\Z', true); ?></updated>
800         <published><?php echo get_post_time('Y-m-d\TH:i:s\Z', true); ?></published>
801         <app:edited><?php echo get_post_modified_time('Y-m-d\TH:i:s\Z', true); ?></app:edited>
802         <app:control>
803                 <app:draft><?php echo ($GLOBALS['post']->post_status == 'draft' ? 'yes' : 'no') ?></app:draft>
804         </app:control>
805         <author>
806                 <name><?php the_author()?></name>
807 <?php if (get_the_author_url() && get_the_author_url() != 'http://') { ?>
808                 <uri><?php the_author_url()?></uri>
809 <?php } ?>
810         </author>
811 <?php if($GLOBALS['post']->post_type == 'attachment') { ?>
812         <link rel="edit-media" href="<?php $this->the_media_url() ?>" />
813         <content type="<?php echo $GLOBALS['post']->post_mime_type ?>" src="<?php the_guid(); ?>"/>
814 <?php } else { ?>
815         <link href="<?php the_permalink_rss() ?>" />
816 <?php if ( strlen( $GLOBALS['post']->post_content ) ) :
817 list($content_type, $content) = prep_atom_text_construct(get_the_content()); ?>
818         <content type="<?php echo $content_type ?>"><?php echo $content ?></content>
819 <?php endif; ?>
820 <?php } ?>
821         <link rel="edit" href="<?php $this->the_entry_url() ?>" />
822 <?php foreach(get_the_category() as $category) { ?>
823         <category scheme="<?php bloginfo_rss('home') ?>" term="<?php echo $category->name?>" />
824 <?php } ?>
825 <?php list($content_type, $content) = prep_atom_text_construct(get_the_excerpt()); ?>
826         <summary type="<?php echo $content_type ?>"><?php echo $content ?></summary>
827 </entry>
828 <?php }
829
830         function ok() {
831                 log_app('Status','200: OK');
832                 header('Content-Type: text/plain');
833                 status_header('200');
834                 exit;
835         }
836
837         function no_content() {
838                 log_app('Status','204: No Content');
839                 header('Content-Type: text/plain');
840                 status_header('204');
841                 echo "Deleted.";
842                 exit;
843         }
844
845         function internal_error($msg = 'Internal Server Error') {
846                 log_app('Status','500: Server Error');
847                 header('Content-Type: text/plain');
848                 status_header('500');
849                 echo $msg;
850                 exit;
851         }
852
853         function bad_request() {
854                 log_app('Status','400: Bad Request');
855                 header('Content-Type: text/plain');
856                 status_header('400');
857                 exit;
858         }
859
860         function length_required() {
861                 log_app('Status','411: Length Required');
862                 header("HTTP/1.1 411 Length Required");
863                 header('Content-Type: text/plain');
864                 status_header('411');
865                 exit;
866         }
867
868         function invalid_media() {
869                 log_app('Status','415: Unsupported Media Type');
870                 header("HTTP/1.1 415 Unsupported Media Type");
871                 header('Content-Type: text/plain');
872                 exit;
873         }
874
875         function not_found() {
876                 log_app('Status','404: Not Found');
877                 header('Content-Type: text/plain');
878                 status_header('404');
879                 exit;
880         }
881
882         function not_allowed($allow) {
883                 log_app('Status','405: Not Allowed');
884                 header('Allow: ' . join(',', $allow));
885                 status_header('405');
886                 exit;
887         }
888
889         function redirect($url) {
890
891                 log_app('Status','302: Redirect');
892                 $escaped_url = attribute_escape($url);
893                 $content = <<<EOD
894 <!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN">
895 <html>
896   <head>
897     <title>302 Found</title>
898   </head>
899 <body>
900   <h1>Found</h1>
901   <p>The document has moved <a href="$escaped_url">here</a>.</p>
902   </body>
903 </html>
904
905 EOD;
906                 header('HTTP/1.1 302 Moved');
907                 header('Content-Type: text/html');
908                 header('Location: ' . $url);
909                 echo $content;
910                 exit;
911
912         }
913
914
915         function client_error($msg = 'Client Error') {
916                 log_app('Status','400: Client Error');
917                 header('Content-Type: text/plain');
918                 status_header('400');
919                 exit;
920         }
921
922         function created($post_ID, $content, $post_type = 'post') {
923                 log_app('created()::$post_ID',"$post_ID, $post_type");
924                 $edit = $this->get_entry_url($post_ID);
925                 switch($post_type) {
926                         case 'post':
927                                 $ctloc = $this->get_entry_url($post_ID);
928                                 break;
929                         case 'attachment':
930                                 $edit = $this->app_base . "attachments/$post_ID";
931                                 break;
932                 }
933                 header("Content-Type: $this->ATOM_CONTENT_TYPE");
934                 if(isset($ctloc))
935                         header('Content-Location: ' . $ctloc);
936                 header('Location: ' . $edit);
937                 status_header('201');
938                 echo $content;
939                 exit;
940         }
941
942         function auth_required($msg) {
943                 log_app('Status','401: Auth Required');
944                 nocache_headers();
945                 header('WWW-Authenticate: Basic realm="WordPress Atom Protocol"');
946                 header("HTTP/1.1 401 $msg");
947                 header('Status: ' . $msg);
948                 header('Content-Type: text/html');
949                 $content = <<<EOD
950 <!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN">
951 <html>
952   <head>
953     <title>401 Unauthorized</title>
954   </head>
955 <body>
956     <h1>401 Unauthorized</h1>
957     <p>$msg</p>
958   </body>
959 </html>
960
961 EOD;
962                 echo $content;
963                 exit;
964         }
965
966         function output($xml, $ctype = 'application/atom+xml') {
967                         status_header('200');
968                         $xml = '<?xml version="1.0" encoding="' . strtolower(get_option('blog_charset')) . '"?>'."\n".$xml;
969                         header('Connection: close');
970                         header('Content-Length: '. strlen($xml));
971                         header('Content-Type: ' . $ctype);
972                         header('Content-Disposition: attachment; filename=atom.xml');
973                         header('Date: '. date('r'));
974                         if($this->do_output)
975                                 echo $xml;
976                         log_app('function', "output:\n$xml");
977                         exit;
978         }
979
980         function escape(&$array) {
981                 global $wpdb;
982
983                 foreach ($array as $k => $v) {
984                                 if (is_array($v)) {
985                                                 $this->escape($array[$k]);
986                                 } else if (is_object($v)) {
987                                                 //skip
988                                 } else {
989                                                 $array[$k] = $wpdb->escape($v);
990                                 }
991                 }
992         }
993
994         /*
995          * Access credential through various methods and perform login
996          */
997         function authenticate() {
998                 $login_data = array();
999                 $already_md5 = false;
1000
1001                 log_app("authenticate()",print_r($_ENV, true));
1002
1003                 // if using mod_rewrite/ENV hack
1004                 // http://www.besthostratings.com/articles/http-auth-php-cgi.html
1005                 if(isset($_SERVER['HTTP_AUTHORIZATION'])) {
1006                         list($_SERVER['PHP_AUTH_USER'], $_SERVER['PHP_AUTH_PW']) =
1007                                 explode(':', base64_decode(substr($_SERVER['HTTP_AUTHORIZATION'], 6)));
1008                 }
1009
1010                 // If Basic Auth is working...
1011                 if(isset($_SERVER['PHP_AUTH_USER']) && isset($_SERVER['PHP_AUTH_PW'])) {
1012                         $login_data = array('login' => $_SERVER['PHP_AUTH_USER'],       'password' => $_SERVER['PHP_AUTH_PW']);
1013                         log_app("Basic Auth",$login_data['login']);
1014                 } else {
1015                         // else, do cookie-based authentication
1016                         if (function_exists('wp_get_cookie_login')) {
1017                                 $login_data = wp_get_cookie_login();
1018                                 $already_md5 = true;
1019                         }
1020                 }
1021
1022                 // call wp_login and set current user
1023                 if (!empty($login_data) && wp_login($login_data['login'], $login_data['password'], $already_md5)) {
1024                          $current_user = new WP_User(0, $login_data['login']);
1025                          wp_set_current_user($current_user->ID);
1026                         log_app("authenticate()",$login_data['login']);
1027                 }
1028         }
1029
1030         function get_accepted_content_type($types = NULL) {
1031
1032                 if(!isset($types)) {
1033                         $types = $this->media_content_types;
1034                 }
1035
1036                 if(!isset($_SERVER['CONTENT_LENGTH']) || !isset($_SERVER['CONTENT_TYPE'])) {
1037                         $this->length_required();
1038                 }
1039
1040                 $type = $_SERVER['CONTENT_TYPE'];
1041                 list($type,$subtype) = explode('/',$type);
1042                 list($subtype) = explode(";",$subtype); // strip MIME parameters
1043                 log_app("get_accepted_content_type", "type=$type, subtype=$subtype");
1044
1045                 foreach($types as $t) {
1046                         list($acceptedType,$acceptedSubtype) = explode('/',$t);
1047                         if($acceptedType == '*' || $acceptedType == $type) {
1048                                 if($acceptedSubtype == '*' || $acceptedSubtype == $subtype)
1049                                         return $type . "/" . $subtype;
1050                         }
1051                 }
1052
1053                 $this->invalid_media();
1054         }
1055
1056         function process_conditionals() {
1057
1058                 if(empty($this->params)) return;
1059                 if($_SERVER['REQUEST_METHOD'] == 'DELETE') return;
1060
1061                 switch($this->params[0]) {
1062                         case $this->ENTRY_PATH:
1063                                 global $post;
1064                                 $post = wp_get_single_post($this->params[1]);
1065                                 $wp_last_modified = get_post_modified_time('D, d M Y H:i:s', true);
1066                                 $post = NULL;
1067                                 break;
1068                         case $this->ENTRIES_PATH:
1069                                 $wp_last_modified = mysql2date('D, d M Y H:i:s', get_lastpostmodified('GMT'), 0).' GMT';
1070                                 break;
1071                         default:
1072                                 return;
1073                 }
1074                 $wp_etag = md5($wp_last_modified);
1075                 @header("Last-Modified: $wp_last_modified");
1076                 @header("ETag: $wp_etag");
1077
1078                 // Support for Conditional GET
1079                 if (isset($_SERVER['HTTP_IF_NONE_MATCH']))
1080                         $client_etag = stripslashes($_SERVER['HTTP_IF_NONE_MATCH']);
1081                 else
1082                         $client_etag = false;
1083
1084                 $client_last_modified = trim( $_SERVER['HTTP_IF_MODIFIED_SINCE']);
1085                 // If string is empty, return 0. If not, attempt to parse into a timestamp
1086                 $client_modified_timestamp = $client_last_modified ? strtotime($client_last_modified) : 0;
1087
1088                 // Make a timestamp for our most recent modification...
1089                 $wp_modified_timestamp = strtotime($wp_last_modified);
1090
1091                 if ( ($client_last_modified && $client_etag) ?
1092                 (($client_modified_timestamp >= $wp_modified_timestamp) && ($client_etag == $wp_etag)) :
1093                 (($client_modified_timestamp >= $wp_modified_timestamp) || ($client_etag == $wp_etag)) ) {
1094                         status_header( 304 );
1095                         exit;
1096                 }
1097         }
1098
1099         function rfc3339_str2time($str) {
1100
1101             $match = false;
1102             if(!preg_match("/(\d{4}-\d{2}-\d{2})T(\d{2}\:\d{2}\:\d{2})\.?\d{0,3}(Z|[+-]+\d{2}\:\d{2})/", $str, $match))
1103                         return false;
1104
1105             if($match[3] == 'Z')
1106                         $match[3] == '+0000';
1107
1108             return strtotime($match[1] . " " . $match[2] . " " . $match[3]);
1109         }
1110
1111         function get_publish_time($published) {
1112
1113             $pubtime = $this->rfc3339_str2time($published);
1114
1115             if(!$pubtime) {
1116                         return array(current_time('mysql'),current_time('mysql',1));
1117             } else {
1118                         return array(date("Y-m-d H:i:s", $pubtime), gmdate("Y-m-d H:i:s", $pubtime));
1119             }
1120         }
1121
1122 }
1123
1124 $server = new AtomServer();
1125 $server->handle_request();
1126
1127 ?>