| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378 |
- <?php
- /*************************************************
- * Copyright (c) 2020, Code Atlantic LLC
- *************************************************/
- if ( ! defined( 'ABSPATH' ) ) {
- exit;
- }
- /**
- * Handles some of our AJAX requests including post/taxonomy search from conditions
- */
- class PUM_Admin_Ajax {
- /**
- * Hooks our methods into AJAX actions.
- * Hooks our methods into AJAX actions.
- */
- public static function init() {
- add_action( 'wp_ajax_pum_object_search', array( __CLASS__, 'object_search' ) );
- add_action( 'wp_ajax_pum_process_batch_request', array( __CLASS__, 'process_batch_request' ) );
- add_action( 'wp_ajax_pum_save_enabled_state', array( __CLASS__, 'save_popup_enabled_state' ) );
- }
- /**
- * Sets the enabled meta field to on or off
- *
- * @since 1.12.0
- */
- public static function save_popup_enabled_state() {
- $args = wp_parse_args(
- $_REQUEST,
- array(
- 'popupID' => 0,
- 'active' => 1,
- )
- );
- // Ensures Popup ID is an int and not 0.
- $popup_id = intval( $args['popupID'] );
- if ( 0 === $popup_id ) {
- wp_send_json_error( 'Invalid popup ID provided.' );
- }
- // Ensures active state is 0 or 1.
- $enabled = intval( $args['enabled'] );
- if ( ! in_array( $enabled, array( 0, 1 ), true ) ) {
- wp_send_json_error( 'Invalid enabled state provided.' );
- }
- // Verify the nonce.
- if ( ! wp_verify_nonce( $_REQUEST['nonce'], "pum_save_enabled_state_$popup_id" ) ) {
- wp_send_json_error();
- }
- // Get our popup and previous value.
- $popup = pum_get_popup( $popup_id );
- $previous = $popup->get_meta( 'enabled' );
- // If value is the same, bail now.
- if ( $previous === $enabled ) {
- wp_send_json_success();
- }
- // Update our value.
- $results = $popup->update_meta( 'enabled', $enabled );
- if ( false === $results ) {
- wp_send_json_error( 'Error updating enabled state.' );
- PUM_Utils_Logging::instance()->log( "Error updating enabled state on $popup_id. Previous value: $previous. New value: $enabled" );
- } else {
- wp_send_json_success();
- }
- }
- /**
- * Searches posts, taxonomies, and users
- *
- * Uses passed array with keys of object_type, object_key, include, exclude. Echos our results as JSON.
- */
- public static function object_search() {
- $results = array(
- 'items' => array(),
- 'total_count' => 0,
- );
- $object_type = sanitize_text_field( $_REQUEST['object_type'] );
- $include = ! empty( $_REQUEST['include'] ) ? wp_parse_id_list( $_REQUEST['include'] ) : [];
- $exclude = ! empty( $_REQUEST['exclude'] ) ? wp_parse_id_list( $_REQUEST['exclude'] ) : [];
- if ( ! empty( $include ) ) {
- $exclude = array_merge( $include, $exclude );
- }
- switch ( $object_type ) {
- case 'post_type':
- $post_type = ! empty( $_REQUEST['object_key'] ) ? sanitize_text_field( $_REQUEST['object_key'] ) : 'post';
- if ( ! empty( $include ) ) {
- $include_query = PUM_Helpers::post_type_selectlist_query(
- $post_type,
- array(
- 'post__in' => $include,
- 'posts_per_page' => - 1,
- ),
- true
- );
- foreach ( $include_query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $include_query['total_count'];
- }
- $query = PUM_Helpers::post_type_selectlist_query(
- $post_type,
- array(
- 's' => ! empty( $_REQUEST['s'] ) ? sanitize_text_field( $_REQUEST['s'] ) : null,
- 'paged' => ! empty( $_REQUEST['paged'] ) ? absint( $_REQUEST['paged'] ) : null,
- 'post__not_in' => $exclude,
- 'posts_per_page' => 10,
- ),
- true
- );
- foreach ( $query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $query['total_count'];
- break;
- case 'taxonomy':
- $taxonomy = ! empty( $_REQUEST['object_key'] ) ? sanitize_text_field( $_REQUEST['object_key'] ) : 'category';
- if ( ! empty( $include ) ) {
- $include_query = PUM_Helpers::taxonomy_selectlist_query(
- $taxonomy,
- array(
- 'include' => $include,
- 'number' => 0,
- ),
- true
- );
- foreach ( $include_query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $include_query['total_count'];
- }
- $query = PUM_Helpers::taxonomy_selectlist_query(
- $taxonomy,
- array(
- 'search' => ! empty( $_REQUEST['s'] ) ? sanitize_text_field( $_REQUEST['s'] ) : null,
- 'paged' => ! empty( $_REQUEST['paged'] ) ? absint( $_REQUEST['paged'] ) : null,
- 'exclude' => $exclude,
- 'number' => 10,
- ),
- true
- );
- foreach ( $query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $query['total_count'];
- break;
- case 'user':
- $user_role = ! empty( $_REQUEST['object_key'] ) ? $_REQUEST['object_key'] : null;
- if ( ! empty( $include ) ) {
- $include_query = PUM_Helpers::user_selectlist_query(
- array(
- 'role' => $user_role,
- 'include' => $include,
- 'number' => - 1,
- ),
- true
- );
- foreach ( $include_query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $include_query['total_count'];
- }
- $query = PUM_Helpers::user_selectlist_query(
- array(
- 'role' => $user_role,
- 'search' => ! empty( $_REQUEST['s'] ) ? '*' . $_REQUEST['s'] . '*' : null,
- 'paged' => ! empty( $_REQUEST['paged'] ) ? absint( $_REQUEST['paged'] ) : null,
- 'exclude' => $exclude,
- 'number' => 10,
- ),
- true
- );
- foreach ( $query['items'] as $id => $name ) {
- $results['items'][] = array(
- 'id' => $id,
- 'text' => "$name (ID: $id)",
- );
- }
- $results['total_count'] += (int) $query['total_count'];
- break;
- }
- // Take out keys which were only used to deduplicate.
- $results['items'] = array_values( $results['items'] );
- echo PUM_Utils_Array::safe_json_encode( $results );
- die();
- }
- /**
- * Handles Ajax for processing a single batch request.
- */
- public static function process_batch_request() {
- // Batch ID.
- $batch_id = isset( $_REQUEST['batch_id'] ) ? sanitize_key( $_REQUEST['batch_id'] ) : false;
- if ( ! $batch_id ) {
- wp_send_json_error(
- array(
- 'error' => __( 'A batch process ID must be present to continue.', 'popup-maker' ),
- )
- );
- }
- // Nonce.
- if ( ! isset( $_REQUEST['nonce'] ) || ( isset( $_REQUEST['nonce'] ) && false === wp_verify_nonce( $_REQUEST['nonce'], "{$batch_id}_step_nonce" ) ) ) {
- wp_send_json_error(
- array(
- 'error' => __( 'You do not have permission to initiate this request. Contact an administrator for more information.', 'popup-maker' ),
- )
- );
- }
- // Attempt to retrieve the batch attributes from memory.
- $batch = PUM_Batch_Process_Registry::instance()->get( $batch_id );
- if ( false === $batch ) {
- wp_send_json_error(
- array(
- 'error' => sprintf( __( '%s is an invalid batch process ID.', 'popup-maker' ), esc_html( $_REQUEST['batch_id'] ) ),
- )
- );
- }
- $class = isset( $batch['class'] ) ? sanitize_text_field( $batch['class'] ) : '';
- $class_file = isset( $batch['file'] ) ? $batch['file'] : '';
- if ( empty( $class_file ) || ! file_exists( $class_file ) ) {
- wp_send_json_error(
- array(
- 'error' => sprintf( __( 'An invalid file path is registered for the %1$s batch process handler.', 'popup-maker' ), "<code>{$batch_id}</code>" ),
- )
- );
- } else {
- require_once $class_file;
- }
- if ( empty( $class ) || ! class_exists( $class ) ) {
- wp_send_json_error(
- array(
- 'error' => sprintf( __( '%1$s is an invalid handler for the %2$s batch process. Please try again.', 'popup-maker' ), "<code>{$class}</code>", "<code>{$batch_id}</code>" ),
- )
- );
- }
- $step = sanitize_text_field( $_REQUEST['step'] );
- /**
- * Instantiate the batch class.
- *
- * @var PUM_Interface_Batch_Exporter|PUM_Interface_Batch_Process|PUM_Interface_Batch_PrefetchProcess $process
- */
- if ( isset( $_REQUEST['data']['upload']['file'] ) ) {
- // If this is an import, instantiate with the file and step.
- $file = sanitize_text_field( $_REQUEST['data']['upload']['file'] );
- $process = new $class( $file, $step );
- } else {
- // Otherwise just the step.
- $process = new $class( $step );
- }
- // Garbage collect any old temporary data.
- // TODO Should this be here? Likely here to prevent case ajax passes step 1 without resetting process counts?
- if ( $step < 2 ) {
- $process->finish();
- }
- $using_prefetch = ( $process instanceof PUM_Interface_Batch_PrefetchProcess );
- // Handle pre-fetching data.
- if ( $using_prefetch ) {
- // Initialize any data needed to process a step.
- $data = isset( $_REQUEST['form'] ) ? $_REQUEST['form'] : array();
- $process->init( $data );
- $process->pre_fetch();
- }
- /** @var int|string|WP_Error $step */
- $step = $process->process_step();
- if ( is_wp_error( $step ) ) {
- wp_send_json_error( $step );
- } else {
- $response_data = array( 'step' => $step );
- // Map fields if this is an import.
- if ( isset( $process->field_mapping ) && ( $process instanceof PUM_Interface_CSV_Importer ) ) {
- $response_data['columns'] = $process->get_columns();
- $response_data['mapping'] = $process->field_mapping;
- }
- // Finish and set the status flag if done.
- if ( 'done' === $step ) {
- $response_data['done'] = true;
- $response_data['message'] = $process->get_message( 'done' );
- // If this is an export class and not an empty export, send the download URL.
- if ( method_exists( $process, 'can_export' ) ) {
- if ( ! $process->is_empty ) {
- $response_data['url'] = pum_admin_url(
- 'tools',
- array(
- 'step' => $step,
- 'nonce' => wp_create_nonce( 'pum-batch-export' ),
- 'batch_id' => $batch_id,
- 'pum_action' => 'download_batch_export',
- )
- );
- }
- }
- // Once all calculations have finished, run cleanup.
- $process->finish();
- } else {
- $response_data['done'] = false;
- $response_data['percentage'] = $process->get_percentage_complete();
- }
- wp_send_json_success( $response_data );
- }
- }
- }
|