245 lines
7.3 KiB
PHP
245 lines
7.3 KiB
PHP
<?php
|
|
/**
|
|
* Import Functions
|
|
*
|
|
* These are functions are used for import data into Easy Digital Downloads.
|
|
*
|
|
* @package EDD
|
|
* @subpackage Admin/Import
|
|
* @copyright Copyright (c) 2018, Easy Digital Downloads, LLC
|
|
* @license http://opensource.org/licenses/gpl-2.0.php GNU Public License
|
|
*/
|
|
|
|
// Exit if accessed directly
|
|
defined( 'ABSPATH' ) || exit;
|
|
|
|
/**
|
|
* Upload an import file with ajax
|
|
*
|
|
* @since 2.6
|
|
* @return void
|
|
*/
|
|
function edd_do_ajax_import_file_upload() {
|
|
|
|
if ( ! wp_verify_nonce( $_REQUEST['edd_ajax_import'], 'edd_ajax_import' ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Nonce verification failed', 'easy-digital-downloads' ) ) );
|
|
}
|
|
|
|
if ( empty( $_POST['edd-import-class'] ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Missing import parameters. Import class must be specified.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
if ( ! function_exists( 'wp_handle_upload' ) ) {
|
|
require_once ABSPATH . 'wp-admin/includes/file.php';
|
|
}
|
|
|
|
require_once EDD_PLUGIN_DIR . 'includes/admin/import/class-batch-import.php';
|
|
|
|
$importer_class = sanitize_text_field( $_POST['edd-import-class'] );
|
|
$is_class_allowed = edd_importer_is_class_allowed( $importer_class );
|
|
if ( false === $is_class_allowed ) {
|
|
wp_send_json_error( array( 'error' => __( 'Invalid importer class supplied', 'easy-digital-downloads' ) ) );
|
|
}
|
|
|
|
do_action( 'edd_batch_import_class_include', $importer_class );
|
|
|
|
$import = new $importer_class();
|
|
|
|
// The import class checks for the user's capability.
|
|
if ( ! $import->can_import() ) {
|
|
wp_send_json_error( array( 'error' => __( 'You do not have permission to import data', 'easy-digital-downloads' ) ) );
|
|
}
|
|
|
|
if ( empty( $_FILES['edd-import-file'] ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Missing import file. Please provide an import file.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
if ( empty( $_FILES['edd-import-file']['type'] ) || ! in_array( strtolower( $_FILES['edd-import-file']['type'] ), edd_importer_accepted_mime_types(), true ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'The file you uploaded does not appear to be a CSV file.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
if ( ! file_exists( $_FILES['edd-import-file']['tmp_name'] ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Something went wrong during the upload process, please try again.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
// Let WordPress import the file. We will remove it after import is complete
|
|
$import_file = wp_handle_upload( $_FILES['edd-import-file'], array( 'test_form' => false ) );
|
|
|
|
if ( $import_file && empty( $import_file['error'] ) ) {
|
|
|
|
$import->set_up_csv( $import_file['file'] );
|
|
wp_send_json_success( array(
|
|
'form' => $_POST,
|
|
'class' => $importer_class,
|
|
'upload' => $import_file,
|
|
'first_row' => $import->get_first_row(),
|
|
'columns' => $import->get_columns(),
|
|
'nonce' => wp_create_nonce( 'edd_ajax_import', 'edd_ajax_import' )
|
|
) );
|
|
|
|
} else {
|
|
|
|
/**
|
|
* Error generated by _wp_handle_upload()
|
|
* @see _wp_handle_upload() in wp-admin/includes/file.php
|
|
*/
|
|
|
|
wp_send_json_error( array( 'error' => $import_file['error'] ) );
|
|
}
|
|
|
|
exit;
|
|
|
|
}
|
|
add_action( 'edd_upload_import_file', 'edd_do_ajax_import_file_upload' );
|
|
|
|
/**
|
|
* Process batch imports via ajax
|
|
*
|
|
* @since 2.6
|
|
* @return void
|
|
*/
|
|
function edd_do_ajax_import() {
|
|
|
|
require_once EDD_PLUGIN_DIR . 'includes/admin/import/class-batch-import.php';
|
|
|
|
if( ! wp_verify_nonce( $_REQUEST['nonce'], 'edd_ajax_import' ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Nonce verification failed', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
if ( empty( $_REQUEST['class'] ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Missing import parameters. Import class must be specified.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
if ( ! file_exists( $_REQUEST['upload']['file'] ) ) {
|
|
wp_send_json_error( array( 'error' => __( 'Something went wrong during the upload process, please try again.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
$file = sanitize_text_field( $_REQUEST['upload']['file'] );
|
|
|
|
$mime_type_allowed = false;
|
|
if ( is_callable( 'mime_content_type' ) ) {
|
|
if ( in_array( mime_content_type( $file ), edd_importer_accepted_mime_types(), true ) ) {
|
|
$mime_type_allowed = true;
|
|
}
|
|
} else {
|
|
if ( wp_check_filetype( $file, edd_importer_accepted_mime_types() ) ) {
|
|
$mime_type_allowed = true;
|
|
}
|
|
}
|
|
|
|
if ( false === $mime_type_allowed ) {
|
|
wp_send_json_error( array( 'error' => __( 'The file you uploaded does not appear to be a CSV file.', 'easy-digital-downloads' ), 'request' => $_REQUEST ) );
|
|
}
|
|
|
|
$importer_class = sanitize_text_field( $_REQUEST['class'] );
|
|
$is_class_allowed = edd_importer_is_class_allowed( $importer_class );
|
|
|
|
if ( ! $is_class_allowed ) {
|
|
wp_send_json_error( array( 'error' => __( 'Invalid importer class supplied', 'easy-digital-downloads' ) ) );
|
|
}
|
|
|
|
do_action( 'edd_batch_import_class_include', $importer_class );
|
|
|
|
$step = absint( $_REQUEST['step'] );
|
|
$class = $importer_class;
|
|
$import = new $class( '', $step );
|
|
|
|
if ( ! $import->can_import() ) {
|
|
wp_send_json_error( array( 'error' => __( 'You do not have permission to import data', 'easy-digital-downloads' ) ) );
|
|
}
|
|
$import->set_up_csv( $file );
|
|
|
|
parse_str( $_REQUEST['mapping'], $map );
|
|
|
|
$import->map_fields( $map['edd-import-field'] );
|
|
|
|
$ret = $import->process_step( $step );
|
|
|
|
$percentage = $import->get_percentage_complete();
|
|
|
|
if( $ret ) {
|
|
|
|
$step += 1;
|
|
wp_send_json_success( array(
|
|
'step' => $step,
|
|
'percentage' => $percentage,
|
|
'columns' => $import->get_columns(),
|
|
'mapping' => $import->field_mapping,
|
|
'total' => $import->total
|
|
) );
|
|
|
|
} elseif ( true === $import->is_empty ) {
|
|
|
|
wp_send_json_error( array(
|
|
'error' => __( 'No data found for import parameters', 'easy-digital-downloads' )
|
|
) );
|
|
|
|
} else {
|
|
|
|
wp_send_json_success( array(
|
|
'step' => 'done',
|
|
'message' => sprintf(
|
|
__( 'Import complete! <a href="%s">View imported %s</a>.', 'easy-digital-downloads' ),
|
|
esc_url( $import->get_list_table_url() ),
|
|
esc_html( $import->get_import_type_label() )
|
|
)
|
|
) );
|
|
|
|
}
|
|
}
|
|
add_action( 'wp_ajax_edd_do_ajax_import', 'edd_do_ajax_import' );
|
|
|
|
/**
|
|
* Returns the array of accepted mime types for the importer.
|
|
*
|
|
* @since 3.0
|
|
* @return array
|
|
*/
|
|
function edd_importer_accepted_mime_types() {
|
|
return array(
|
|
'text/csv',
|
|
'text/comma-separated-values',
|
|
'text/plain',
|
|
'text/anytext',
|
|
'text/*',
|
|
'text/plain',
|
|
'text/anytext',
|
|
'text/*',
|
|
'application/csv',
|
|
'application/excel',
|
|
'application/vnd.ms-excel',
|
|
'application/vnd.msexcel',
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Given an importer class name, is it allowed to process as an importer.
|
|
*
|
|
* @since 3.0.2
|
|
*
|
|
* @param string $class The class name to check.
|
|
*
|
|
* @return bool If the class is allowed to be used as an importer.
|
|
*/
|
|
function edd_importer_is_class_allowed( $class = '' ) {
|
|
$allowed_importer_classes = edd_get_importer_accepted_classes();
|
|
return in_array( $class, $allowed_importer_classes, true );
|
|
}
|
|
|
|
/**
|
|
* Returns a list of allowed importer classes.
|
|
*
|
|
* @since 3.0.2
|
|
*
|
|
* @return array An array of class names to allow during imports.
|
|
*/
|
|
function edd_get_importer_accepted_classes() {
|
|
return apply_filters(
|
|
'edd_accepted_importer_classes',
|
|
array(
|
|
'EDD_Batch_Downloads_Import',
|
|
'EDD_Batch_Payments_Import',
|
|
)
|
|
);
|
|
}
|