WordPress.org

Make WordPress Core

Ticket #6269: rss.php

File rss.php, 5.0 KB (added by sweetdeal, 6 years ago)

tweaked rss.php (import) file

Line 
1<?php
2
3class 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 = str_replace(array ('<![CDATA[', ']]>'), '', $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                do_action('import_done', 'rss');
145
146                echo '<h3>';
147                printf(__('All done. <a href="%s">Have fun!</a>'), get_option('home'));
148                echo '</h3>';
149        }
150
151        function dispatch() {
152                if (empty ($_GET['step']))
153                        $step = 0;
154                else
155                        $step = (int) $_GET['step'];
156
157                $this->header();
158
159                switch ($step) {
160                        case 0 :
161                                $this->greet();
162                                break;
163                        case 1 :
164                                check_admin_referer('import-upload');
165                                $result = $this->import();
166                                if ( is_wp_error( $result ) )
167                                        echo $result->get_error_message();
168                                break;
169                }
170
171                $this->footer();
172        }
173
174        function RSS_Import() {
175                // Nothing.
176        }
177}
178
179$rss_import = new RSS_Import();
180
181register_importer('rss', __('RSS'), __('Import posts from an RSS feed.'), array ($rss_import, 'dispatch'));
182?>