root/tags/1.3/wp-admin/import/rss.php

Revision 1069, 4.9 kB (checked in by donncha, 1 year ago)

Merge with WP 2.3 - testing use only!
Move pluggable functions out of wpmu-functions and into pluggable.php, fixes #439

Line 
1 <?php
2
3 class RSS_Import {
4
5     var $posts = array ();
6     var $file;
7
8     function header() {
9         echo '<div class="wrap">';
10         echo '<h2>'.__('Import RSS').'</h2>';
11     }
12
13     function footer() {
14         echo '</div>';
15     }
16
17     function unhtmlentities($string) { // From php.net for < 4.3 compat
18         $trans_tbl = get_html_translation_table(HTML_ENTITIES);
19         $trans_tbl = array_flip($trans_tbl);
20         return strtr($string, $trans_tbl);
21     }
22
23     function greet() {
24         echo '<div class="narrow">';
25         echo '<p>'.__('Howdy! This importer allows you to extract posts from an RSS 2.0 file into your blog. This is useful if you want to import your posts from a system that is not handled by a custom import tool. Pick an RSS file to upload and click Import.').'</p>';
26         wp_import_upload_form("admin.php?import=rss&amp;step=1");
27         echo '</div>';
28     }
29
30     function get_posts() {
31         global $wpdb;
32
33         set_magic_quotes_runtime(0);
34         $datalines = file($this->file); // Read the file into an array
35         $importdata = implode('', $datalines); // squish it
36         $importdata = str_replace(array ("\r\n", "\r"), "\n", $importdata);
37
38         preg_match_all('|<item>(.*?)</item>|is', $importdata, $this->posts);
39         $this->posts = $this->posts[1];
40         $index = 0;
41         foreach ($this->posts as $post) {
42             preg_match('|<title>(.*?)</title>|is', $post, $post_title);
43             $post_title = str_replace(array('<![CDATA[', ']]>'), '', $wpdb->escape( trim($post_title[1]) ));
44
45             preg_match('|<pubdate>(.*?)</pubdate>|is', $post, $post_date_gmt);
46
47             if ($post_date_gmt) {
48                 $post_date_gmt = strtotime($post_date_gmt[1]);
49             } else {
50                 // if we don't already have something from pubDate
51                 preg_match('|<dc:date>(.*?)</dc:date>|is', $post, $post_date_gmt);
52                 $post_date_gmt = preg_replace('|([-+])([0-9]+):([0-9]+)$|', '\1\2\3', $post_date_gmt[1]);
53                 $post_date_gmt = str_replace('T', ' ', $post_date_gmt);
54                 $post_date_gmt = strtotime($post_date_gmt);
55             }
56
57             $post_date_gmt = gmdate('Y-m-d H:i:s', $post_date_gmt);
58             $post_date = get_date_from_gmt( $post_date_gmt );
59
60             preg_match_all('|<category>(.*?)</category>|is', $post, $categories);
61             $categories = $categories[1];
62
63             if (!$categories) {
64                 preg_match_all('|<dc:subject>(.*?)</dc:subject>|is', $post, $categories);
65                 $categories = $categories[1];
66             }
67
68             $cat_index = 0;
69             foreach ($categories as $category) {
70                 $categories[$cat_index] = $wpdb->escape($this->unhtmlentities($category));
71                 $cat_index++;
72             }
73
74             preg_match('|<guid.+?>(.*?)</guid>|is', $post, $guid);
75             if ($guid)
76                 $guid = $wpdb->escape(trim($guid[1]));
77             else
78                 $guid = '';
79
80             preg_match('|<content:encoded>(.*?)</content:encoded>|is', $post, $post_content);
81             $post_content = str_replace(array ('<![CDATA[', ']]>'), '', $wpdb->escape(trim($post_content[1])));
82
83             if (!$post_content) {
84                 // This is for feeds that put content in description
85                 preg_match('|<description>(.*?)</description>|is', $post, $post_content);
86                 $post_content = $wpdb->escape($this->unhtmlentities(trim($post_content[1])));
87             }
88
89             // Clean up content
90             $post_content = preg_replace('|<(/?[A-Z]+)|e', "'<' . strtolower('$1')", $post_content);
91             $post_content = str_replace('<br>', '<br />', $post_content);
92             $post_content = str_replace('<hr>', '<hr />', $post_content);
93
94             $post_author = 1;
95             $post_status = 'publish';
96             $this->posts[$index] = compact('post_author', 'post_date', 'post_date_gmt', 'post_content', 'post_title', 'post_status', 'guid', 'categories');
97             $index++;
98         }
99     }
100
101     function import_posts() {
102         echo '<ol>';
103
104         foreach ($this->posts as $post) {
105             echo "<li>".__('Importing post...');
106
107             extract($post);
108
109             if ($post_id = post_exists($post_title, $post_content, $post_date)) {
110                 _e('Post already imported');
111             } else {
112                 $post_id = wp_insert_post($post);
113                 if ( is_wp_error( $post_id ) )
114                     return $post_id;
115                 if (!$post_id) {
116                     _e("Couldn't get post ID");
117                     return;
118                 }
119
120                 if (0 != count($categories))
121                     wp_create_categories($categories, $post_id);
122                 _e('Done !');
123             }
124             echo '</li>';
125         }
126
127         echo '</ol>';
128
129     }
130
131     function import() {
132         $file = wp_import_handle_upload();
133         if ( isset($file['error']) ) {
134             echo $file['error'];
135             return;
136         }
137
138         $this->file = $file['file'];
139         $this->get_posts();
140         $result = $this->import_posts();
141         if ( is_wp_error( $result ) )
142             return $result;
143         wp_import_cleanup($file['id']);
144
145         echo '<h3>';
146         printf(__('All done. <a href="%s">Have fun!</a>'), get_option('home'));
147         echo '</h3>';
148     }
149
150     function dispatch() {
151         if (empty ($_GET['step']))
152             $step = 0;
153         else
154             $step = (int) $_GET['step'];
155
156         $this->header();
157
158         switch ($step) {
159             case 0 :
160                 $this->greet();
161                 break;
162             case 1 :
163                 check_admin_referer('import-upload');
164                 $result = $this->import();
165                 if ( is_wp_error( $result ) )
166                     echo $result->get_error_message();
167                 break;
168         }
169
170         $this->footer();
171     }
172
173     function RSS_Import() {
174         // Nothing.
175     }
176 }
177
178 $rss_import = new RSS_Import();
179
180 register_importer('rss', __('RSS'), __('Import posts from an RSS feed'), array ($rss_import, 'dispatch'));
181 ?>
182
Note: See TracBrowser for help on using the browser.