[Summary view] [Print] [Text view]

   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  ?>