171 lines
3.2 KiB
PHP
171 lines
3.2 KiB
PHP
|
<?php
|
||
|
/**
|
||
|
* WP Async Request
|
||
|
*
|
||
|
* @package WP-Background-Processing
|
||
|
*/
|
||
|
/*
|
||
|
Library URI: https://github.com/deliciousbrains/wp-background-processing/blob/fbbc56f2480910d7959972ec9ec0819a13c6150a/classes/wp-async-request.php
|
||
|
Author: Delicious Brains Inc.
|
||
|
Author URI: https://deliciousbrains.com/
|
||
|
License: GNU General Public License v2.0
|
||
|
License URI: https://github.com/deliciousbrains/wp-background-processing/commit/126d7945dd3d39f39cb6488ca08fe1fb66cb351a
|
||
|
*/
|
||
|
|
||
|
if ( ! class_exists( 'WP_Async_Request' ) ) {
|
||
|
|
||
|
/**
|
||
|
* Abstract WP_Async_Request class.
|
||
|
*
|
||
|
* @abstract
|
||
|
*/
|
||
|
abstract class WP_Async_Request {
|
||
|
|
||
|
/**
|
||
|
* Prefix
|
||
|
*
|
||
|
* (default value: 'wp')
|
||
|
*
|
||
|
* @var string
|
||
|
* @access protected
|
||
|
*/
|
||
|
protected $prefix = 'wp';
|
||
|
|
||
|
/**
|
||
|
* Action
|
||
|
*
|
||
|
* (default value: 'async_request')
|
||
|
*
|
||
|
* @var string
|
||
|
* @access protected
|
||
|
*/
|
||
|
protected $action = 'async_request';
|
||
|
|
||
|
/**
|
||
|
* Identifier
|
||
|
*
|
||
|
* @var mixed
|
||
|
* @access protected
|
||
|
*/
|
||
|
protected $identifier;
|
||
|
|
||
|
/**
|
||
|
* Data
|
||
|
*
|
||
|
* (default value: array())
|
||
|
*
|
||
|
* @var array
|
||
|
* @access protected
|
||
|
*/
|
||
|
protected $data = array();
|
||
|
|
||
|
/**
|
||
|
* Initiate new async request
|
||
|
*/
|
||
|
public function __construct() {
|
||
|
$this->identifier = $this->prefix . '_' . $this->action;
|
||
|
|
||
|
add_action( 'wp_ajax_' . $this->identifier, array( $this, 'maybe_handle' ) );
|
||
|
add_action( 'wp_ajax_nopriv_' . $this->identifier, array( $this, 'maybe_handle' ) );
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Set data used during the request
|
||
|
*
|
||
|
* @param array $data Data.
|
||
|
*
|
||
|
* @return $this
|
||
|
*/
|
||
|
public function data( $data ) {
|
||
|
$this->data = $data;
|
||
|
|
||
|
return $this;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Dispatch the async request
|
||
|
*
|
||
|
* @return array|WP_Error
|
||
|
*/
|
||
|
public function dispatch() {
|
||
|
$url = add_query_arg( $this->get_query_args(), $this->get_query_url() );
|
||
|
$args = $this->get_post_args();
|
||
|
|
||
|
return wp_remote_post( esc_url_raw( $url ), $args );
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get query args
|
||
|
*
|
||
|
* @return array
|
||
|
*/
|
||
|
protected function get_query_args() {
|
||
|
if ( property_exists( $this, 'query_args' ) ) {
|
||
|
return $this->query_args;
|
||
|
}
|
||
|
|
||
|
return array(
|
||
|
'action' => $this->identifier,
|
||
|
'nonce' => wp_create_nonce( $this->identifier ),
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get query URL
|
||
|
*
|
||
|
* @return string
|
||
|
*/
|
||
|
protected function get_query_url() {
|
||
|
if ( property_exists( $this, 'query_url' ) ) {
|
||
|
return $this->query_url;
|
||
|
}
|
||
|
|
||
|
return admin_url( 'admin-ajax.php' );
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get post args
|
||
|
*
|
||
|
* @return array
|
||
|
*/
|
||
|
protected function get_post_args() {
|
||
|
if ( property_exists( $this, 'post_args' ) ) {
|
||
|
return $this->post_args;
|
||
|
}
|
||
|
|
||
|
return array(
|
||
|
'timeout' => 0.01,
|
||
|
'blocking' => false,
|
||
|
'body' => $this->data,
|
||
|
'cookies' => $_COOKIE,
|
||
|
'sslverify' => apply_filters( 'https_local_ssl_verify', false ),
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Maybe handle
|
||
|
*
|
||
|
* Check for correct nonce and pass to handler.
|
||
|
*/
|
||
|
public function maybe_handle() {
|
||
|
// Don't lock up other requests while processing
|
||
|
session_write_close();
|
||
|
|
||
|
check_ajax_referer( $this->identifier, 'nonce' );
|
||
|
|
||
|
$this->handle();
|
||
|
|
||
|
wp_die();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Handle
|
||
|
*
|
||
|
* Override this method to perform any actions required
|
||
|
* during the async request.
|
||
|
*/
|
||
|
abstract protected function handle();
|
||
|
|
||
|
}
|
||
|
}
|