Wordpress 2.3.2
[autoinstalls/wordpress.git] / wp-admin / import / rss.php
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 ?>