header(); $step = empty( $_GET['step'] ) ? 0 : (int) $_GET['step']; switch ( $step ) { case 0: $this->greet(); break; case 1: check_admin_referer( 'import-attachment-files' ); set_time_limit(0); $this->import(); break; } $this->footer(); } /** * The main controller for the actual import stage. * * @param string $file Path to the WXR file for importing */ function import() { add_filter( 'import_post_meta_key', array( $this, 'is_valid_meta_key' ) ); add_filter( 'http_request_timeout', array( &$this, 'bump_request_timeout' ) ); $this->import_start(); wp_suspend_cache_invalidation( true ); $this->process_posts(); wp_suspend_cache_invalidation( false ); $this->import_end(); } /** * Parses the WXR file and prepares us for the task of processing parsed data * * @param string $file Path to the WXR file for importing */ function import_start() { if (empty($_POST['server_url'])) { echo '

' . __( 'Sorry, there has been an error.', 'attachment-files-importer' ) . '
'; echo __( 'The server URL is not valid, please try again.', 'attachment-files-importer' ) . '

'; $this->footer(); die(); } $import_data = $this->get_import_data(); $this->posts = $import_data['posts']; $this->base_url = esc_url($_POST['server_url']); do_action( 'import_start' ); } /** * Performs post-import cleanup of files and the cache */ function import_end() { wp_cache_flush(); echo '

' . __( 'All done.', 'attachment-files-importer' ) . ' ' . __( 'Have fun!', 'attachment-files-importer' ) . '' . '

'; do_action( 'import_end' ); } /** * Create new posts based on import information * * Posts marked as having a parent which doesn't exist will become top level items. * Doesn't create a new post if: the post type doesn't exist, the given post ID * is already noted as imported or a post with the same title and date already exists. * Note that new/updated terms, comments and meta are imported for the last of the above. */ function process_posts() { $this->posts = apply_filters( 'attachment_files_importer_posts', $this->posts ); foreach ( $this->posts as $post ) { $post = apply_filters( 'attachment_files_importer_post_data_raw', $post ); if ( $post['status'] == 'auto-draft' ) continue; $post_type_object = get_post_type_object( $post['post_type'] ); $postdata = array( 'import_id' => $post['post_id'], 'post_author' => $author, 'post_date' => $post['post_date'], 'post_date_gmt' => $post['post_date_gmt'], 'post_content' => $post['post_content'], 'post_excerpt' => $post['post_excerpt'], 'post_title' => $post['post_title'], 'post_status' => $post['status'], 'post_name' => $post['post_name'], 'comment_status' => $post['comment_status'], 'ping_status' => $post['ping_status'], 'guid' => $post['guid'], 'post_parent' => $post_parent, 'menu_order' => $post['menu_order'], 'post_type' => $post['post_type'], 'post_password' => $post['post_password'] ); $postdata = apply_filters( 'attachment_files_importer_post_data_processed', $postdata, $post ); if ( 'attachment' == $postdata['post_type'] ) { $local_url = ! empty($post['attachment_url']) ? $post['attachment_url'] : $post['guid']; $remote_url = $this->base_url . parse_url($local_url, PHP_URL_PATH); // try to use _wp_attached file for upload folder placement to ensure the same location as the export site // e.g. location is 2003/05/image.jpg but the attachment post_date is 2010/09, see media_handle_upload() $postdata['upload_date'] = $post['post_date']; if ( isset( $post['postmeta'] ) ) { foreach( $post['postmeta'] as $meta ) { if ( $meta['key'] == '_wp_attached_file' ) { if ( preg_match( '%^[0-9]{4}/[0-9]{2}%', $meta['value'], $matches ) ) $postdata['upload_date'] = $matches[0]; break; } } } $this->process_attachment( $postdata, $remote_url ); } else { continue; } } unset( $this->posts ); } /** * If fetching attachments is enabled then attempt to create a new attachment * * @param array $post Attachment post details from WXR * @param string $url URL to fetch attachment from * @return int|WP_Error Post ID on success, WP_Error otherwise */ function process_attachment( $post, $url ) { // if the URL is absolute, but does not contain address, then upload it assuming base_site_url if ( preg_match( '|^/[\w\W]+$|', $url ) ) $url = rtrim( $this->base_url, '/' ) . $url; $upload = $this->fetch_remote_file( $url, $post ); if ( is_wp_error( $upload ) ) return $upload; if ( $info = wp_check_filetype( $upload['file'] ) ) $post['post_mime_type'] = $info['type']; else return new WP_Error( 'attachment_processing_error', __('Invalid file type', 'attachment-files-importer') ); } /** * Check if the file to download already exists * * @param string $filename * @param string $time Optional. Time formatted in 'yyyy/mm'. * @return boolean */ function is_file_exists( $filename, $time = null ) { // get the upload dir $upload = wp_upload_dir($time); if ( $upload['error'] ) return new WP_Error( 'upload_dir_error', $upload['error'] ); $dir = $upload['path']; // sanitize the file name before we begin processing $filename = sanitize_file_name($filename); return file_exists( $dir . "/$filename" ); } /** * Attempt to download a remote file attachment * * @param string $url URL of item to fetch * @param array $post Attachment details * @return array|WP_Error Local file location details on success, WP_Error otherwise */ function fetch_remote_file( $url, $post ) { // extract the file name and extension from the url $file_name = basename( $url ); // get placeholder file in the upload dir if it doesn't already exists if ( $this->is_file_exists( $file_name, $post['upload_date'] ) ) { $upload['error'] = __('File already exists', 'attachment-files-importer'); } else { $upload = wp_upload_bits( $file_name, 0, '', $post['upload_date'] ); } if ( $upload['error'] ) return new WP_Error( 'upload_dir_error', $upload['error'] ); // fetch the remote url and write it to the placeholder file $headers = wp_get_http( $url, $upload['file'] ); // request failed if ( ! $headers ) { @unlink( $upload['file'] ); return new WP_Error( 'import_file_error', __('Remote server did not respond', 'attachment-files-importer') ); } // make sure the fetch was successful if ( $headers['response'] != '200' ) { @unlink( $upload['file'] ); return new WP_Error( 'import_file_error', sprintf( __('Remote server returned error response %1$d %2$s', 'attachment-files-importer'), esc_html($headers['response']), get_status_header_desc($headers['response']) ) ); } $filesize = filesize( $upload['file'] ); if ( isset( $headers['content-length'] ) && $filesize != $headers['content-length'] ) { @unlink( $upload['file'] ); return new WP_Error( 'import_file_error', __('Remote file is incorrect size', 'attachment-files-importer') ); } if ( 0 == $filesize ) { @unlink( $upload['file'] ); return new WP_Error( 'import_file_error', __('Zero size file downloaded', 'attachment-files-importer') ); } $max_size = (int) $this->max_attachment_size(); if ( ! empty( $max_size ) && $filesize > $max_size ) { @unlink( $upload['file'] ); return new WP_Error( 'import_file_error', sprintf(__('Remote file is too large, limit is %s', 'attachment-files-importer'), size_format($max_size) ) ); } return $upload; } /** * Get the data of the attachment files to import, same format as a WXR file. * * @return array Information gathered from the DB */ function get_import_data() { $post_attachments = get_posts(array( 'posts_per_page' => -1, 'post_type' => 'attachment', )); $import_data = array( 'posts' => array(), ); for ($k=0;$kpost_author); $import_data['posts'][] = array( 'post_title' => $post_attachments[$k]->post_title, 'guid' => $post_attachments[$k]->guid, 'post_author' => $userdata->user_login, 'post_content' => $post_attachments[$k]->post_content, 'post_id' => $post_attachments[$k]->ID, 'post_date' => $post_attachments[$k]->post_date, 'post_date_gmt' => $post_attachments[$k]->post_date_gmt, 'comment_status' => $post_attachments[$k]->comment_status, 'ping_status' => $post_attachments[$k]->ping_status, 'post_name' => $post_attachments[$k]->post_name, 'status' => $post_attachments[$k]->post_status, 'post_parent' => $post_attachments[$k]->post_parent, 'menu_order' => $post_attachments[$k]->menu_order, 'post_type' => $post_attachments[$k]->post_type, 'post_password' => $post_attachments[$k]->post_password, 'is_sticky' => is_sticky($post_attachments[$k]->ID)?1:0, 'attachment_url' => wp_get_attachment_url($post_attachments[$k]->ID), 'postmeta' => get_post_meta($post_attachments[$k]->ID), ); } return $import_data; } // Display import page title function header() { echo '
'; screen_icon(); echo '

' . __( 'Import Attachment Files', 'attachment-files-importer' ) . '

'; $updates = get_plugin_updates(); $basename = plugin_basename(__FILE__); if ( isset( $updates[$basename] ) ) { $update = $updates[$basename]; echo '

'; printf( __( 'A new version of this importer is available. Please update to version %s to ensure compatibility with newer export files.', 'attachment-files-importer' ), $update->update->new_version ); echo '

'; } } // Close div.wrap function footer() { echo '
'; } /** * Display introductory text and file upload form */ function greet() { echo '
'; echo '

'.__( 'Indicate the URL of the server that contain the attachment files corresponding to the attachments of this site and we"ll download them into this site.', 'attachment-files-importer' ).'

'; echo '

'.__( 'Enter the URL of the server, then click Import.', 'attachment-files-importer' ).'

'; $bytes = apply_filters( 'import_upload_size_limit', wp_max_upload_size() ); $size = wp_convert_bytes_to_hr( $bytes ); $upload_dir = wp_upload_dir(); if ( ! empty( $upload_dir['error'] ) ) : ?>

'; } /** * Decide if the given meta key maps to information we will want to import * * @param string $key The meta key to check * @return string|bool The key if we do want to import, false if not */ function is_valid_meta_key( $key ) { // skip attachment metadata since we'll regenerate it from scratch // skip _edit_lock as not relevant for import if ( in_array( $key, array( '_wp_attached_file', '_wp_attachment_metadata', '_edit_lock' ) ) ) return false; return $key; } /** * Decide what the maximum file size for downloaded attachments is. * Default is 0 (unlimited), can be filtered via import_attachment_size_limit * * @return int Maximum attachment file size to import */ function max_attachment_size() { return apply_filters( 'import_attachment_size_limit', 0 ); } /** * Added to http_request_timeout filter to force timeout at 60 seconds during import * @return int 60 */ function bump_request_timeout() { return 60; } // return the difference in length between two strings function cmpr_strlen( $a, $b ) { return strlen($b) - strlen($a); } } } // class_exists( 'WP_Importer' ) function attachment_files_importer_init() { // load_plugin_textdomain( 'attachment-files-importer', false, dirname( plugin_basename( __FILE__ ) ) . '/languages' ); /** * Attachment Files Importer object for registering the import callback * @global AF_Import $af_import */ $GLOBALS['af_import'] = new AF_Import(); register_importer( 'attachment-files', 'Attachment files', __('Import attachment files from an external server based on the attachment entries already presents in your Wordpress installation.', 'attachment-files-importer'), array( $GLOBALS['af_import'], 'dispatch' ) ); } add_action( 'admin_init', 'attachment_files_importer_init' );