' . esc_html( $error ) . '
'; + } + } + } + + /** + * Dispatch current step and show correct view. */ public function dispatch() { - include_once( WC_ABSPATH . 'includes/import/class-wc-product-csv-importer.php' ); - - $this->header(); - - $step = empty( $_GET['step'] ) ? 0 : (int) $_GET['step']; - - switch ( $step ) { - - case 0: - $this->greet(); - break; - - case 1 : - check_admin_referer( 'import-upload' ); - - if ( $this->handle_upload() ) { - if ( $this->id ) { - $file = get_attached_file( $this->id ); - } else { - $file = ABSPATH . $this->file_url; - } - - $this->importer_mapping( $file ); - } - break; + if ( ! empty( $_POST['save_step'] ) && isset( $this->steps[ $this->step ]['handler'] ) ) { + call_user_func( $this->steps[ $this->step ]['handler'], $this ); + } + $this->output_header(); + $this->output_steps(); + $this->output_errors(); + call_user_func( $this->steps[ $this->step ]['view'], $this ); + $this->output_footer(); + /*switch ( $this->step ) { case 2 : check_admin_referer( 'woocommerce-csv-importer' ); $file = null; @@ -100,43 +196,124 @@ class WC_Product_WP_Importer extends WP_Importer { $this->import( $file ); break; - } - - $this->footer(); + }*/ } /** - * Import is starting. + * Output information about the uploading process. */ - private function import_start() { - if ( function_exists( 'gc_enable' ) ) { - gc_enable(); + protected function upload_form() { + $bytes = apply_filters( 'import_upload_size_limit', wp_max_upload_size() ); + $size = size_format( $bytes ); + $upload_dir = wp_upload_dir(); + + include( dirname( __FILE__ ) . '/views/html-product-csv-import-form.php' ); + } + + /** + * Handle the upload form and store options. + */ + public function upload_form_handler() { + check_admin_referer( 'woocommerce-csv-importer' ); + + $file = $this->handle_upload(); + + if ( is_wp_error( $file ) ) { + $this->add_error( $file->get_message() ); + return; + } else { + $this->file = $file; } - wc_set_time_limit( 0 ); - @ob_flush(); - @flush(); - @ini_set( 'auto_detect_line_endings', '1' ); + + wp_redirect( esc_url_raw( $this->get_next_step_link() ) ); + exit; + } + + /** + * Handles the CSV upload and initial parsing of the file to prepare for. + * displaying author import options. + * + * @return string|WP_Error + */ + public function handle_upload() { + if ( empty( $_POST['file_url'] ) ) { + if ( ! isset( $_FILES['import'] ) ) { + return new WP_Error( __( 'File is empty. Please upload something more substantial. This error could also be caused by uploads being disabled in your php.ini or by post_max_size being defined as smaller than upload_max_filesize in php.ini.', 'woocommerce' ) ); + } + + $overrides = array( 'test_form' => false, 'test_type' => false ); + $_FILES['import']['name'] .= '.txt'; + $upload = wp_handle_upload( $_FILES['import'], $overrides ); + + if ( isset( $upload['error'] ) ) { + return new WP_Error( $upload['error'] ); + } + + // Construct the object array + $object = array( + 'post_title' => basename( $upload['file'] ), + 'post_content' => $upload['url'], + 'post_mime_type' => $upload['type'], + 'guid' => $upload['url'], + 'context' => 'import', + 'post_status' => 'private', + ); + + // Save the data + $id = wp_insert_attachment( $object, $upload['file'] ); + + /* + * Schedule a cleanup for one day from now in case of failed + * import or missing wp_import_cleanup() call. + */ + wp_schedule_single_event( time() + DAY_IN_SECONDS, 'importer_scheduled_cleanup', array( $id ) ); + + return $upload['file']; + } elseif ( file_exists( ABSPATH . $_POST['file_url'] ) ) { + return ABSPATH . $_POST['file_url']; + } + + return new WP_Error( __( 'Please upload or provide the link to a valid CSV file.', 'woocommerce' ) ); + } + + /** + * Mapping step @todo + */ + protected function mapping_form() { + $importer = $this->get_importer( $this->file, array( 'lines' => 1 ) ); + $headers = $importer->get_raw_keys(); + $sample = current( $importer->get_raw_data() ); + + if ( empty( $sample ) ) { + $this->add_error( __( 'The file is empty, please try again with a new file.', 'woocommerce' ) ); + return; + } + + // Check if all fields matches. + if ( 0 === count( array_diff( $headers, $this->get_default_fields() ) ) ) { + wp_redirect( esc_url_raw( $this->get_next_step_link() ) ); + exit; + } + + include_once( dirname( __FILE__ ) . '/views/html-csv-import-mapping.php' ); } /** * Import the file if it exists and is valid. - * - * @param mixed $file */ - public function import( $file ) { - if ( ! is_file( $file ) ) { - $this->import_error( __( 'The file does not exist, please try again.', 'woocommerce' ) ); + public function import() { + if ( ! is_file( $this->file ) ) { + $this->add_error( __( 'The file does not exist, please try again.', 'woocommerce' ) ); + return; } - $this->import_start(); - $args = array( 'parse' => true ); if ( ! empty( $_POST['map_to'] ) ) { $args['mapping'] = wp_unslash( $_POST['map_to'] ); } - $importer = $this->get_importer( $file, $args ); + $importer = $this->get_importer( $this->file, $args ); $data = $importer->import(); $imported = count( $data['imported'] ); $failed = count( $data['failed'] ); @@ -161,134 +338,13 @@ class WC_Product_WP_Importer extends WP_Importer { /* translators: %d: import results */ printf( __( 'Import complete: %s', 'woocommerce' ), $results ); echo ''; - - $this->import_end(); } /** - * Performs post-import cleanup of files and the cache. + * @todo */ - public function import_end() { - echo '' . __( 'All done!', 'woocommerce' ) . ' ' . __( 'View products', 'woocommerce' ) . '' . '
'; + protected function done() { - do_action( 'import_end' ); - } - - /** - * Handles the CSV upload and initial parsing of the file to prepare for. - * displaying author import options. - * - * @return bool False if error uploading or invalid file, true otherwise - */ - public function handle_upload() { - if ( empty( $_POST['file_url'] ) ) { - - $file = wp_import_handle_upload(); - - if ( isset( $file['error'] ) ) { - $this->import_error( $file['error'] ); - } - - $this->id = absint( $file['id'] ); - - } elseif ( file_exists( ABSPATH . $_POST['file_url'] ) ) { - $this->file_url = esc_attr( $_POST['file_url'] ); - } else { - $this->import_error(); - } - - return true; - } - - /** - * Output header html. - */ - public function header() { - echo '' . __( 'Hi there! Upload a CSV file containing products to import them into your shop. Choose a .csv file to upload, then click "Upload file and import".', 'woocommerce' ) . '
'; - - $action = 'admin.php?import=woocommerce_product_csv&step=1'; - - $bytes = apply_filters( 'import_upload_size_limit', wp_max_upload_size() ); - $size = size_format( $bytes ); - $upload_dir = wp_upload_dir(); - if ( ! empty( $upload_dir['error'] ) ) : - ?>' . __( 'Sorry, there has been an error.', 'woocommerce' ) . '
';
- if ( $message ) {
- echo esc_html( $message );
- }
- echo '