Merged in feature/from-pantheon (pull request #16)
code from pantheon * code from pantheon
This commit is contained in:
Tony Volpe
@@ -0,0 +1,235 @@
|
||||
<?php
|
||||
/**
|
||||
* CustomMetaDataStore class file.
|
||||
*/
|
||||
|
||||
namespace Automattic\WooCommerce\Internal\DataStores;
|
||||
|
||||
/**
|
||||
* Implements functions similar to WP's add_metadata(), get_metadata(), and friends using a custom table.
|
||||
*
|
||||
* @see WC_Data_Store_WP For an implementation using WP's metadata functions and tables.
|
||||
*/
|
||||
abstract class CustomMetaDataStore {
|
||||
|
||||
/**
|
||||
* Returns the name of the table used for storage.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
abstract protected function get_table_name();
|
||||
|
||||
/**
|
||||
* Returns the name of the field/column used for identifiying metadata entries.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_meta_id_field() {
|
||||
return 'id';
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the name of the field/column used for associating meta with objects.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_object_id_field() {
|
||||
return 'object_id';
|
||||
}
|
||||
|
||||
/**
|
||||
* Describes the structure of the metadata table.
|
||||
*
|
||||
* @return array Array elements: table, object_id_field, meta_id_field.
|
||||
*/
|
||||
protected function get_db_info() {
|
||||
return array(
|
||||
'table' => $this->get_table_name(),
|
||||
'meta_id_field' => $this->get_meta_id_field(),
|
||||
'object_id_field' => $this->get_object_id_field(),
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns an array of meta for an object.
|
||||
*
|
||||
* @param WC_Data $object WC_Data object.
|
||||
* @return array
|
||||
*/
|
||||
public function read_meta( &$object ) {
|
||||
global $wpdb;
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
|
||||
// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
$raw_meta_data = $wpdb->get_results(
|
||||
$wpdb->prepare(
|
||||
"SELECT {$db_info['meta_id_field']} AS meta_id, meta_key, meta_value FROM {$db_info['table']} WHERE {$db_info['object_id_field']} = %d ORDER BY meta_id",
|
||||
$object->get_id()
|
||||
)
|
||||
);
|
||||
// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
|
||||
return $raw_meta_data;
|
||||
}
|
||||
|
||||
/**
|
||||
* Deletes meta based on meta ID.
|
||||
*
|
||||
* @param WC_Data $object WC_Data object.
|
||||
* @param stdClass $meta (containing at least ->id).
|
||||
*
|
||||
* @return bool
|
||||
*/
|
||||
public function delete_meta( &$object, $meta ) : bool {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! isset( $meta->id ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
$meta_id = absint( $meta->id );
|
||||
|
||||
return (bool) $wpdb->delete( $db_info['table'], array( $db_info['meta_id_field'] => $meta_id ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Add new piece of meta.
|
||||
*
|
||||
* @param WC_Data $object WC_Data object.
|
||||
* @param stdClass $meta (containing ->key and ->value).
|
||||
*
|
||||
* @return int|false meta ID
|
||||
*/
|
||||
public function add_meta( &$object, $meta ) {
|
||||
global $wpdb;
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
|
||||
$object_id = $object->get_id();
|
||||
$meta_key = wp_unslash( wp_slash( $meta->key ) );
|
||||
$meta_value = maybe_serialize( is_string( $meta->value ) ? wp_unslash( wp_slash( $meta->value ) ) : $meta->value );
|
||||
|
||||
// phpcs:disable WordPress.DB.SlowDBQuery.slow_db_query_meta_value,WordPress.DB.SlowDBQuery.slow_db_query_meta_key
|
||||
$result = $wpdb->insert(
|
||||
$db_info['table'],
|
||||
array(
|
||||
$db_info['object_id_field'] => $object_id,
|
||||
'meta_key' => $meta_key,
|
||||
'meta_value' => $meta_value,
|
||||
)
|
||||
);
|
||||
// phpcs:enable WordPress.DB.SlowDBQuery.slow_db_query_meta_value,WordPress.DB.SlowDBQuery.slow_db_query_meta_key
|
||||
|
||||
return $result ? (int) $wpdb->insert_id : false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Update meta.
|
||||
*
|
||||
* @param WC_Data $object WC_Data object.
|
||||
* @param stdClass $meta (containing ->id, ->key and ->value).
|
||||
*
|
||||
* @return bool
|
||||
*/
|
||||
public function update_meta( &$object, $meta ) : bool {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! isset( $meta->id ) || empty( $meta->key ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// phpcs:disable WordPress.DB.SlowDBQuery.slow_db_query_meta_value,WordPress.DB.SlowDBQuery.slow_db_query_meta_key
|
||||
$data = array(
|
||||
'meta_key' => $meta->key,
|
||||
'meta_value' => maybe_serialize( $meta->value ),
|
||||
);
|
||||
// phpcs:enable WordPress.DB.SlowDBQuery.slow_db_query_meta_value,WordPress.DB.SlowDBQuery.slow_db_query_meta_key
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
|
||||
$result = $wpdb->update(
|
||||
$db_info['table'],
|
||||
$data,
|
||||
array( $db_info['meta_id_field'] => $meta->id ),
|
||||
'%s',
|
||||
'%d'
|
||||
);
|
||||
|
||||
return 1 === $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieves metadata by meta ID.
|
||||
*
|
||||
* @param int $meta_id Meta ID.
|
||||
* @return object|bool Metadata object or FALSE if not found.
|
||||
*/
|
||||
public function get_metadata_by_id( $meta_id ) {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! is_numeric( $meta_id ) || floor( $meta_id ) != $meta_id ) { // phpcs:ignore WordPress.PHP.StrictComparisons.LooseComparison
|
||||
return false;
|
||||
}
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
|
||||
$meta_id = absint( $meta_id );
|
||||
// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
$meta = $wpdb->get_row(
|
||||
$wpdb->prepare(
|
||||
"SELECT {$db_info['meta_id_field']}, meta_key, meta_value, {$db_info['object_id_field']} FROM {$db_info['table']} WHERE {$db_info['meta_id_field']} = %d",
|
||||
$meta_id
|
||||
)
|
||||
);
|
||||
// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
|
||||
if ( empty( $meta ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if ( isset( $meta->meta_value ) ) {
|
||||
$meta->meta_value = maybe_unserialize( $meta->meta_value );
|
||||
}
|
||||
|
||||
return $meta;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieves metadata by meta key.
|
||||
*
|
||||
* @param \WC_Abstract_Order $object Object ID.
|
||||
* @param string $meta_key Meta key.
|
||||
*
|
||||
* @return \stdClass|bool Metadata object or FALSE if not found.
|
||||
*/
|
||||
public function get_metadata_by_key( &$object, string $meta_key ) {
|
||||
global $wpdb;
|
||||
|
||||
$db_info = $this->get_db_info();
|
||||
|
||||
// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
$meta = $wpdb->get_results(
|
||||
$wpdb->prepare(
|
||||
"SELECT {$db_info['meta_id_field']}, meta_key, meta_value, {$db_info['object_id_field']} FROM {$db_info['table']} WHERE meta_key = %s AND {$db_info['object_id_field']} = %d",
|
||||
$meta_key,
|
||||
$object->get_id(),
|
||||
)
|
||||
);
|
||||
// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
|
||||
if ( empty( $meta ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
foreach ( $meta as $row ) {
|
||||
if ( isset( $row->meta_value ) ) {
|
||||
$row->meta_value = maybe_unserialize( $row->meta_value );
|
||||
}
|
||||
}
|
||||
|
||||
return $meta;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,596 @@
|
||||
<?php
|
||||
/**
|
||||
* CustomOrdersTableController class file.
|
||||
*/
|
||||
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
use Automattic\WooCommerce\Caches\OrderCache;
|
||||
use Automattic\WooCommerce\Caches\OrderCacheController;
|
||||
use Automattic\WooCommerce\Internal\BatchProcessing\BatchProcessingController;
|
||||
use Automattic\WooCommerce\Internal\Features\FeaturesController;
|
||||
use Automattic\WooCommerce\Internal\Traits\AccessiblePrivateMethods;
|
||||
use Automattic\WooCommerce\Utilities\PluginUtil;
|
||||
use ActionScheduler;
|
||||
|
||||
defined( 'ABSPATH' ) || exit;
|
||||
|
||||
/**
|
||||
* This is the main class that controls the custom orders tables feature. Its responsibilities are:
|
||||
*
|
||||
* - Displaying UI components (entries in the tools page and in settings)
|
||||
* - Providing the proper data store for orders via 'woocommerce_order_data_store' hook
|
||||
*
|
||||
* ...and in general, any functionality that doesn't imply database access.
|
||||
*/
|
||||
class CustomOrdersTableController {
|
||||
|
||||
use AccessiblePrivateMethods;
|
||||
|
||||
private const SYNC_QUERY_ARG = 'wc_hpos_sync_now';
|
||||
|
||||
/**
|
||||
* The name of the option for enabling the usage of the custom orders tables
|
||||
*/
|
||||
public const CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION = 'woocommerce_custom_orders_table_enabled';
|
||||
|
||||
/**
|
||||
* The name of the option that tells whether database transactions are to be used or not for data synchronization.
|
||||
*/
|
||||
public const USE_DB_TRANSACTIONS_OPTION = 'woocommerce_use_db_transactions_for_custom_orders_table_data_sync';
|
||||
|
||||
/**
|
||||
* The name of the option to store the transaction isolation level to use when database transactions are enabled.
|
||||
*/
|
||||
public const DB_TRANSACTIONS_ISOLATION_LEVEL_OPTION = 'woocommerce_db_transactions_isolation_level_for_custom_orders_table_data_sync';
|
||||
|
||||
public const DEFAULT_DB_TRANSACTIONS_ISOLATION_LEVEL = 'READ UNCOMMITTED';
|
||||
|
||||
/**
|
||||
* The data store object to use.
|
||||
*
|
||||
* @var OrdersTableDataStore
|
||||
*/
|
||||
private $data_store;
|
||||
|
||||
/**
|
||||
* Refunds data store object to use.
|
||||
*
|
||||
* @var OrdersTableRefundDataStore
|
||||
*/
|
||||
private $refund_data_store;
|
||||
|
||||
/**
|
||||
* The data synchronizer object to use.
|
||||
*
|
||||
* @var DataSynchronizer
|
||||
*/
|
||||
private $data_synchronizer;
|
||||
|
||||
/**
|
||||
* The batch processing controller to use.
|
||||
*
|
||||
* @var BatchProcessingController
|
||||
*/
|
||||
private $batch_processing_controller;
|
||||
|
||||
/**
|
||||
* The features controller to use.
|
||||
*
|
||||
* @var FeaturesController
|
||||
*/
|
||||
private $features_controller;
|
||||
|
||||
/**
|
||||
* The orders cache object to use.
|
||||
*
|
||||
* @var OrderCache
|
||||
*/
|
||||
private $order_cache;
|
||||
|
||||
/**
|
||||
* The orders cache controller object to use.
|
||||
*
|
||||
* @var OrderCacheController
|
||||
*/
|
||||
private $order_cache_controller;
|
||||
|
||||
/**
|
||||
* The plugin util object to use.
|
||||
*
|
||||
* @var PluginUtil
|
||||
*/
|
||||
private $plugin_util;
|
||||
|
||||
/**
|
||||
* Class constructor.
|
||||
*/
|
||||
public function __construct() {
|
||||
$this->init_hooks();
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the hooks used by the class.
|
||||
*/
|
||||
private function init_hooks() {
|
||||
self::add_filter( 'woocommerce_order_data_store', array( $this, 'get_orders_data_store' ), 999, 1 );
|
||||
self::add_filter( 'woocommerce_order-refund_data_store', array( $this, 'get_refunds_data_store' ), 999, 1 );
|
||||
self::add_filter( 'woocommerce_debug_tools', array( $this, 'add_initiate_regeneration_entry_to_tools_array' ), 999, 1 );
|
||||
self::add_filter( 'updated_option', array( $this, 'process_updated_option' ), 999, 3 );
|
||||
self::add_filter( 'pre_update_option', array( $this, 'process_pre_update_option' ), 999, 3 );
|
||||
self::add_action( 'woocommerce_after_register_post_type', array( $this, 'register_post_type_for_order_placeholders' ), 10, 0 );
|
||||
self::add_action( 'woocommerce_sections_advanced', array( $this, 'sync_now' ) );
|
||||
self::add_filter( 'removable_query_args', array( $this, 'register_removable_query_arg' ) );
|
||||
self::add_action( 'woocommerce_register_feature_definitions', array( $this, 'add_feature_definition' ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Class initialization, invoked by the DI container.
|
||||
*
|
||||
* @internal
|
||||
* @param OrdersTableDataStore $data_store The data store to use.
|
||||
* @param DataSynchronizer $data_synchronizer The data synchronizer to use.
|
||||
* @param OrdersTableRefundDataStore $refund_data_store The refund data store to use.
|
||||
* @param BatchProcessingController $batch_processing_controller The batch processing controller to use.
|
||||
* @param FeaturesController $features_controller The features controller instance to use.
|
||||
* @param OrderCache $order_cache The order cache engine to use.
|
||||
* @param OrderCacheController $order_cache_controller The order cache controller to use.
|
||||
* @param PluginUtil $plugin_util The plugin util to use.
|
||||
*/
|
||||
final public function init(
|
||||
OrdersTableDataStore $data_store,
|
||||
DataSynchronizer $data_synchronizer,
|
||||
OrdersTableRefundDataStore $refund_data_store,
|
||||
BatchProcessingController $batch_processing_controller,
|
||||
FeaturesController $features_controller,
|
||||
OrderCache $order_cache,
|
||||
OrderCacheController $order_cache_controller,
|
||||
PluginUtil $plugin_util
|
||||
) {
|
||||
$this->data_store = $data_store;
|
||||
$this->data_synchronizer = $data_synchronizer;
|
||||
$this->batch_processing_controller = $batch_processing_controller;
|
||||
$this->refund_data_store = $refund_data_store;
|
||||
$this->features_controller = $features_controller;
|
||||
$this->order_cache = $order_cache;
|
||||
$this->order_cache_controller = $order_cache_controller;
|
||||
$this->plugin_util = $plugin_util;
|
||||
}
|
||||
|
||||
/**
|
||||
* Is the custom orders table usage enabled via settings?
|
||||
* This can be true only if the feature is enabled and a table regeneration has been completed.
|
||||
*
|
||||
* @return bool True if the custom orders table usage is enabled
|
||||
*/
|
||||
public function custom_orders_table_usage_is_enabled(): bool {
|
||||
return get_option( self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION ) === 'yes';
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the instance of the orders data store to use.
|
||||
*
|
||||
* @param \WC_Object_Data_Store_Interface|string $default_data_store The default data store (as received via the woocommerce_order_data_store hook).
|
||||
*
|
||||
* @return \WC_Object_Data_Store_Interface|string The actual data store to use.
|
||||
*/
|
||||
private function get_orders_data_store( $default_data_store ) {
|
||||
return $this->get_data_store_instance( $default_data_store, 'order' );
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the instance of the refunds data store to use.
|
||||
*
|
||||
* @param \WC_Object_Data_Store_Interface|string $default_data_store The default data store (as received via the woocommerce_order-refund_data_store hook).
|
||||
*
|
||||
* @return \WC_Object_Data_Store_Interface|string The actual data store to use.
|
||||
*/
|
||||
private function get_refunds_data_store( $default_data_store ) {
|
||||
return $this->get_data_store_instance( $default_data_store, 'order_refund' );
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the instance of a given data store.
|
||||
*
|
||||
* @param \WC_Object_Data_Store_Interface|string $default_data_store The default data store (as received via the appropriate hooks).
|
||||
* @param string $type The type of the data store to get.
|
||||
*
|
||||
* @return \WC_Object_Data_Store_Interface|string The actual data store to use.
|
||||
*/
|
||||
private function get_data_store_instance( $default_data_store, string $type ) {
|
||||
if ( $this->custom_orders_table_usage_is_enabled() ) {
|
||||
switch ( $type ) {
|
||||
case 'order_refund':
|
||||
return $this->refund_data_store;
|
||||
default:
|
||||
return $this->data_store;
|
||||
}
|
||||
} else {
|
||||
return $default_data_store;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Add an entry to Status - Tools to create or regenerate the custom orders table,
|
||||
* and also an entry to delete the table as appropriate.
|
||||
*
|
||||
* @param array $tools_array The array of tools to add the tool to.
|
||||
* @return array The updated array of tools-
|
||||
*/
|
||||
private function add_initiate_regeneration_entry_to_tools_array( array $tools_array ): array {
|
||||
if ( ! $this->data_synchronizer->check_orders_table_exists() ) {
|
||||
return $tools_array;
|
||||
}
|
||||
|
||||
if ( $this->custom_orders_table_usage_is_enabled() || $this->data_synchronizer->data_sync_is_enabled() ) {
|
||||
$disabled = true;
|
||||
$message = __( 'This will delete the custom orders tables. The tables can be deleted only if the "High-Performance order storage" is not authoritative and sync is disabled (via Settings > Advanced > Features).', 'woocommerce' );
|
||||
} else {
|
||||
$disabled = false;
|
||||
$message = __( 'This will delete the custom orders tables. To create them again enable the "High-Performance order storage" feature (via Settings > Advanced > Features).', 'woocommerce' );
|
||||
}
|
||||
|
||||
$tools_array['delete_custom_orders_table'] = array(
|
||||
'name' => __( 'Delete the custom orders tables', 'woocommerce' ),
|
||||
'desc' => sprintf(
|
||||
'<strong class="red">%1$s</strong> %2$s',
|
||||
__( 'Note:', 'woocommerce' ),
|
||||
$message
|
||||
),
|
||||
'requires_refresh' => true,
|
||||
'callback' => function () {
|
||||
$this->features_controller->change_feature_enable( self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION, false );
|
||||
$this->delete_custom_orders_tables();
|
||||
return __( 'Custom orders tables have been deleted.', 'woocommerce' );
|
||||
},
|
||||
'button' => __( 'Delete', 'woocommerce' ),
|
||||
'disabled' => $disabled,
|
||||
);
|
||||
|
||||
return $tools_array;
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete the custom orders tables and any related options and data in response to the user pressing the tool button.
|
||||
*
|
||||
* @throws \Exception Can't delete the tables.
|
||||
*/
|
||||
private function delete_custom_orders_tables() {
|
||||
if ( $this->custom_orders_table_usage_is_enabled() ) {
|
||||
throw new \Exception( "Can't delete the custom orders tables: they are currently in use (via Settings > Advanced > Features)." );
|
||||
}
|
||||
|
||||
delete_option( self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION );
|
||||
$this->data_synchronizer->delete_database_tables();
|
||||
}
|
||||
|
||||
/**
|
||||
* Handler for the individual setting updated hook.
|
||||
*
|
||||
* @param string $option Setting name.
|
||||
* @param mixed $old_value Old value of the setting.
|
||||
* @param mixed $value New value of the setting.
|
||||
*/
|
||||
private function process_updated_option( $option, $old_value, $value ) {
|
||||
if ( DataSynchronizer::ORDERS_DATA_SYNC_ENABLED_OPTION === $option && 'no' === $value ) {
|
||||
$this->data_synchronizer->cleanup_synchronization_state();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Handler for the setting pre-update hook.
|
||||
* We use it to verify that authoritative orders table switch doesn't happen while sync is pending.
|
||||
*
|
||||
* @param mixed $value New value of the setting.
|
||||
* @param string $option Setting name.
|
||||
* @param mixed $old_value Old value of the setting.
|
||||
*
|
||||
* @throws \Exception Attempt to change the authoritative orders table while orders sync is pending.
|
||||
*/
|
||||
private function process_pre_update_option( $value, $option, $old_value ) {
|
||||
if ( DataSynchronizer::ORDERS_DATA_SYNC_ENABLED_OPTION === $option && $value !== $old_value ) {
|
||||
$this->order_cache->flush();
|
||||
return $value;
|
||||
}
|
||||
|
||||
if ( self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION !== $option ) {
|
||||
return $value;
|
||||
}
|
||||
|
||||
$this->order_cache->flush();
|
||||
if ( ! $this->data_synchronizer->check_orders_table_exists() ) {
|
||||
$this->data_synchronizer->create_database_tables();
|
||||
}
|
||||
|
||||
$tables_created = get_option( DataSynchronizer::ORDERS_TABLE_CREATED ) === 'yes';
|
||||
if ( ! $tables_created ) {
|
||||
return 'no';
|
||||
}
|
||||
|
||||
$sync_is_pending = 0 !== $this->data_synchronizer->get_current_orders_pending_sync_count();
|
||||
if ( $sync_is_pending && ! $this->changing_data_source_with_sync_pending_is_allowed() ) {
|
||||
throw new \Exception( "The authoritative table for orders storage can't be changed while there are orders out of sync" );
|
||||
}
|
||||
|
||||
return $value;
|
||||
}
|
||||
|
||||
/**
|
||||
* Callback to trigger a sync immediately by clicking a button on the Features screen.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
private function sync_now() {
|
||||
$section = filter_input( INPUT_GET, 'section' );
|
||||
if ( 'features' !== $section ) {
|
||||
return;
|
||||
}
|
||||
|
||||
if ( filter_input( INPUT_GET, self::SYNC_QUERY_ARG, FILTER_VALIDATE_BOOLEAN ) ) {
|
||||
$this->batch_processing_controller->enqueue_processor( DataSynchronizer::class );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Tell WP Admin to remove the sync query arg from the URL.
|
||||
*
|
||||
* @param array $query_args The query args that are removable.
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
private function register_removable_query_arg( $query_args ) {
|
||||
$query_args[] = self::SYNC_QUERY_ARG;
|
||||
|
||||
return $query_args;
|
||||
}
|
||||
|
||||
/**
|
||||
* Handler for the woocommerce_after_register_post_type post,
|
||||
* registers the post type for placeholder orders.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
private function register_post_type_for_order_placeholders(): void {
|
||||
wc_register_order_type(
|
||||
DataSynchronizer::PLACEHOLDER_ORDER_POST_TYPE,
|
||||
array(
|
||||
'public' => false,
|
||||
'exclude_from_search' => true,
|
||||
'publicly_queryable' => false,
|
||||
'show_ui' => false,
|
||||
'show_in_menu' => false,
|
||||
'show_in_nav_menus' => false,
|
||||
'show_in_admin_bar' => false,
|
||||
'show_in_rest' => false,
|
||||
'rewrite' => false,
|
||||
'query_var' => false,
|
||||
'can_export' => false,
|
||||
'supports' => array(),
|
||||
'capabilities' => array(),
|
||||
'exclude_from_order_count' => true,
|
||||
'exclude_from_order_views' => true,
|
||||
'exclude_from_order_reports' => true,
|
||||
'exclude_from_order_sales_reports' => true,
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Add the definition for the HPOS feature.
|
||||
*
|
||||
* @param FeaturesController $features_controller The instance of FeaturesController.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
private function add_feature_definition( $features_controller ) {
|
||||
$definition = array(
|
||||
'option_key' => self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION,
|
||||
'is_experimental' => false,
|
||||
'enabled_by_default' => false,
|
||||
'order' => 50,
|
||||
'setting' => $this->get_hpos_setting_for_feature(),
|
||||
'additional_settings' => array(
|
||||
$this->get_hpos_setting_for_sync(),
|
||||
),
|
||||
);
|
||||
|
||||
$features_controller->add_feature_definition(
|
||||
'custom_order_tables',
|
||||
__( 'High-Performance order storage', 'woocommerce' ),
|
||||
$definition
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the HPOS setting for rendering HPOS vs Post setting block in Features section of the settings page.
|
||||
*
|
||||
* @return array Feature setting object.
|
||||
*/
|
||||
private function get_hpos_setting_for_feature() {
|
||||
if ( 'yes' === get_transient( 'wc_installing' ) ) {
|
||||
return array();
|
||||
}
|
||||
|
||||
$get_value = function() {
|
||||
return $this->custom_orders_table_usage_is_enabled() ? 'yes' : 'no';
|
||||
};
|
||||
|
||||
/**
|
||||
* ⚠️The FeaturesController instance must only be accessed from within the callback functions. Otherwise it
|
||||
* gets called while it's still being instantiated and creates and endless loop.
|
||||
*/
|
||||
|
||||
$get_desc = function() {
|
||||
$plugin_compatibility = $this->features_controller->get_compatible_plugins_for_feature( 'custom_order_tables', true );
|
||||
|
||||
return $this->plugin_util->generate_incompatible_plugin_feature_warning( 'custom_order_tables', $plugin_compatibility );
|
||||
};
|
||||
|
||||
$get_disabled = function() {
|
||||
$plugin_compatibility = $this->features_controller->get_compatible_plugins_for_feature( 'custom_order_tables', true );
|
||||
$sync_complete = 0 === $this->get_orders_pending_sync_count();
|
||||
$disabled = array();
|
||||
// Changing something here? might also want to look at `enable|disable` functions in CLIRunner.
|
||||
if ( count( array_merge( $plugin_compatibility['uncertain'], $plugin_compatibility['incompatible'] ) ) > 0 ) {
|
||||
$disabled = array( 'yes' );
|
||||
}
|
||||
if ( ! $sync_complete && ! $this->changing_data_source_with_sync_pending_is_allowed() ) {
|
||||
$disabled = array( 'yes', 'no' );
|
||||
}
|
||||
|
||||
return $disabled;
|
||||
};
|
||||
|
||||
return array(
|
||||
'id' => self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION,
|
||||
'title' => __( 'Order data storage', 'woocommerce' ),
|
||||
'type' => 'radio',
|
||||
'options' => array(
|
||||
'no' => __( 'WordPress posts storage (legacy)', 'woocommerce' ),
|
||||
'yes' => __( 'High-performance order storage (recommended)', 'woocommerce' ),
|
||||
),
|
||||
'value' => $get_value,
|
||||
'disabled' => $get_disabled,
|
||||
'desc' => $get_desc,
|
||||
'desc_at_end' => true,
|
||||
'row_class' => self::CUSTOM_ORDERS_TABLE_USAGE_ENABLED_OPTION,
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the setting for rendering sync enabling setting block in Features section of the settings page.
|
||||
*
|
||||
* @return array Feature setting object.
|
||||
*/
|
||||
private function get_hpos_setting_for_sync() {
|
||||
if ( 'yes' === get_transient( 'wc_installing' ) ) {
|
||||
return array();
|
||||
}
|
||||
|
||||
$get_value = function() {
|
||||
return get_option( DataSynchronizer::ORDERS_DATA_SYNC_ENABLED_OPTION );
|
||||
};
|
||||
|
||||
$get_sync_message = function() {
|
||||
$orders_pending_sync_count = $this->get_orders_pending_sync_count();
|
||||
$sync_in_progress = $this->batch_processing_controller->is_enqueued( get_class( $this->data_synchronizer ) );
|
||||
$sync_enabled = $this->data_synchronizer->data_sync_is_enabled();
|
||||
$sync_is_pending = $orders_pending_sync_count > 0;
|
||||
$sync_message = array();
|
||||
|
||||
$is_dangerous = $sync_is_pending && $this->changing_data_source_with_sync_pending_is_allowed();
|
||||
|
||||
if ( $is_dangerous ) {
|
||||
$sync_message[] = wp_kses_data(
|
||||
sprintf(
|
||||
// translators: %d: number of pending orders.
|
||||
_n(
|
||||
"There's %d order pending sync. <b>Switching data storage while sync is incomplete is dangerous and can lead to order data corruption or loss!</b>",
|
||||
'There are %d orders pending sync. <b>Switching data storage while sync is incomplete is dangerous and can lead to order data corruption or loss!</b>',
|
||||
$orders_pending_sync_count,
|
||||
'woocommerce'
|
||||
),
|
||||
$orders_pending_sync_count,
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
if ( ! $sync_enabled && $this->data_synchronizer->background_sync_is_enabled() ) {
|
||||
$sync_message[] = __( 'Background sync is enabled.', 'woocommerce' );
|
||||
}
|
||||
|
||||
if ( $sync_in_progress && $sync_is_pending ) {
|
||||
$sync_message[] = sprintf(
|
||||
// translators: %d: number of pending orders.
|
||||
__( 'Currently syncing orders... %d pending', 'woocommerce' ),
|
||||
$orders_pending_sync_count
|
||||
);
|
||||
} elseif ( $sync_is_pending ) {
|
||||
$sync_now_url = add_query_arg(
|
||||
array(
|
||||
self::SYNC_QUERY_ARG => true,
|
||||
),
|
||||
wc_get_container()->get( FeaturesController::class )->get_features_page_url()
|
||||
);
|
||||
|
||||
if ( ! $is_dangerous ) {
|
||||
$sync_message[] = wp_kses_data(
|
||||
sprintf(
|
||||
// translators: %d: number of pending orders.
|
||||
_n(
|
||||
"There's %d order pending sync. You can switch order data storage <strong>only when the posts and orders tables are in sync</strong>.",
|
||||
'There are %d orders pending sync. You can switch order data storage <strong>only when the posts and orders tables are in sync</strong>.',
|
||||
$orders_pending_sync_count,
|
||||
'woocommerce'
|
||||
),
|
||||
$orders_pending_sync_count
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
$sync_message[] = sprintf(
|
||||
'<a href="%1$s" class="button button-link">%2$s</a>',
|
||||
esc_url( $sync_now_url ),
|
||||
sprintf(
|
||||
// translators: %d: number of pending orders.
|
||||
_n(
|
||||
'Sync %s pending order',
|
||||
'Sync %s pending orders',
|
||||
$orders_pending_sync_count,
|
||||
'woocommerce'
|
||||
),
|
||||
number_format_i18n( $orders_pending_sync_count )
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
return implode( '<br />', $sync_message );
|
||||
};
|
||||
|
||||
$get_description_is_error = function() {
|
||||
$sync_is_pending = $this->get_orders_pending_sync_count() > 0;
|
||||
|
||||
return $sync_is_pending && $this->changing_data_source_with_sync_pending_is_allowed();
|
||||
};
|
||||
|
||||
return array(
|
||||
'id' => DataSynchronizer::ORDERS_DATA_SYNC_ENABLED_OPTION,
|
||||
'title' => '',
|
||||
'type' => 'checkbox',
|
||||
'desc' => __( 'Enable compatibility mode (synchronizes orders to the posts table).', 'woocommerce' ),
|
||||
'value' => $get_value,
|
||||
'desc_tip' => $get_sync_message,
|
||||
'description_is_error' => $get_description_is_error,
|
||||
'row_class' => DataSynchronizer::ORDERS_DATA_SYNC_ENABLED_OPTION,
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a value indicating if changing the authoritative data source for orders while there are orders pending synchronization is allowed.
|
||||
*
|
||||
* @return bool
|
||||
*/
|
||||
private function changing_data_source_with_sync_pending_is_allowed(): bool {
|
||||
/**
|
||||
* Filter to allow changing where order data is stored, even when there are orders pending synchronization.
|
||||
*
|
||||
* DANGER! This filter is intended for usage when doing manual and automated testing in development environments only,
|
||||
* it should NEVER be used in production environments. Order data corruption or loss can happen!
|
||||
*
|
||||
* @param bool $allow True to allow changing order storage when there are orders pending synchronization, false to disallow.
|
||||
* @returns bool
|
||||
*
|
||||
* @since 8.3.0
|
||||
*/
|
||||
return apply_filters( 'wc_allow_changing_orders_storage_while_sync_is_pending', false );
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the count of orders pending synchronization.
|
||||
*
|
||||
* @return int
|
||||
*/
|
||||
private function get_orders_pending_sync_count(): int {
|
||||
return $this->data_synchronizer->get_sync_status()['current_pending_count'];
|
||||
}
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,33 @@
|
||||
<?php
|
||||
/**
|
||||
* OrdersTableDataStoreMeta class file.
|
||||
*/
|
||||
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
use Automattic\WooCommerce\Internal\DataStores\CustomMetaDataStore;
|
||||
|
||||
/**
|
||||
* Mimics a WP metadata (i.e. add_metadata(), get_metadata() and friends) implementation using a custom table.
|
||||
*/
|
||||
class OrdersTableDataStoreMeta extends CustomMetaDataStore {
|
||||
|
||||
/**
|
||||
* Returns the name of the table used for storage.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_table_name() {
|
||||
return OrdersTableDataStore::get_meta_table_name();
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the name of the field/column used for associating meta with objects.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_object_id_field() {
|
||||
return 'order_id';
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,356 @@
|
||||
<?php
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
defined( 'ABSPATH' ) || exit;
|
||||
|
||||
/**
|
||||
* Provides the implementation for `field_query` in {@see OrdersTableQuery} used to build
|
||||
* complex queries against order fields in the database.
|
||||
*
|
||||
* @internal
|
||||
*/
|
||||
class OrdersTableFieldQuery {
|
||||
|
||||
/**
|
||||
* List of valid SQL operators to use as field_query 'compare' values.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private const VALID_COMPARISON_OPERATORS = array(
|
||||
'=',
|
||||
'!=',
|
||||
'LIKE',
|
||||
'NOT LIKE',
|
||||
'IN',
|
||||
'NOT IN',
|
||||
'EXISTS',
|
||||
'NOT EXISTS',
|
||||
'RLIKE',
|
||||
'REGEXP',
|
||||
'NOT REGEXP',
|
||||
'>',
|
||||
'>=',
|
||||
'<',
|
||||
'<=',
|
||||
'BETWEEN',
|
||||
'NOT BETWEEN',
|
||||
);
|
||||
|
||||
/**
|
||||
* The original query object.
|
||||
*
|
||||
* @var OrdersTableQuery
|
||||
*/
|
||||
private $query = null;
|
||||
|
||||
/**
|
||||
* Determines whether the field query should produce no results due to an invalid argument.
|
||||
*
|
||||
* @var boolean
|
||||
*/
|
||||
private $force_no_results = false;
|
||||
|
||||
/**
|
||||
* Holds a sanitized version of the `field_query`.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $queries = array();
|
||||
|
||||
/**
|
||||
* JOIN clauses to add to the main SQL query.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $join = array();
|
||||
|
||||
/**
|
||||
* WHERE clauses to add to the main SQL query.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $where = array();
|
||||
|
||||
/**
|
||||
* Table aliases in use by the field query. Used to keep track of JOINs and optimize when possible.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $table_aliases = array();
|
||||
|
||||
|
||||
/**
|
||||
* Constructor.
|
||||
*
|
||||
* @param OrdersTableQuery $q The main query being performed.
|
||||
*/
|
||||
public function __construct( OrdersTableQuery $q ) {
|
||||
$field_query = $q->get( 'field_query' );
|
||||
|
||||
if ( ! $field_query || ! is_array( $field_query ) ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$this->query = $q;
|
||||
$this->queries = $this->sanitize_query( $field_query );
|
||||
$this->where = ( ! $this->force_no_results ) ? $this->process( $this->queries ) : '1=0';
|
||||
}
|
||||
|
||||
/**
|
||||
* Sanitizes the field_query argument.
|
||||
*
|
||||
* @param array $q A field_query array.
|
||||
* @return array A sanitized field query array.
|
||||
* @throws \Exception When field table info is missing.
|
||||
*/
|
||||
private function sanitize_query( array $q ) {
|
||||
$sanitized = array();
|
||||
|
||||
foreach ( $q as $key => $arg ) {
|
||||
if ( 'relation' === $key ) {
|
||||
$relation = $arg;
|
||||
} elseif ( ! is_array( $arg ) ) {
|
||||
continue;
|
||||
} elseif ( $this->is_atomic( $arg ) ) {
|
||||
if ( isset( $arg['value'] ) && array() === $arg['value'] ) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Sanitize 'compare'.
|
||||
$arg['compare'] = strtoupper( $arg['compare'] ?? '=' );
|
||||
$arg['compare'] = in_array( $arg['compare'], self::VALID_COMPARISON_OPERATORS, true ) ? $arg['compare'] : '=';
|
||||
|
||||
if ( '=' === $arg['compare'] && isset( $arg['value'] ) && is_array( $arg['value'] ) ) {
|
||||
$arg['compare'] = 'IN';
|
||||
}
|
||||
|
||||
// Sanitize 'cast'.
|
||||
$arg['cast'] = $this->sanitize_cast_type( $arg['type'] ?? '' );
|
||||
|
||||
$field_info = $this->query->get_field_mapping_info( $arg['field'] );
|
||||
if ( ! $field_info ) {
|
||||
$this->force_no_results = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
$arg = array_merge( $arg, $field_info );
|
||||
|
||||
$sanitized[ $key ] = $arg;
|
||||
} else {
|
||||
$sanitized_arg = $this->sanitize_query( $arg );
|
||||
|
||||
if ( $sanitized_arg ) {
|
||||
$sanitized[ $key ] = $sanitized_arg;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if ( $sanitized ) {
|
||||
$sanitized['relation'] = 1 === count( $sanitized ) ? 'OR' : $this->sanitize_relation( $relation ?? 'AND' );
|
||||
}
|
||||
|
||||
return $sanitized;
|
||||
}
|
||||
|
||||
/**
|
||||
* Makes sure we use an AND or OR relation. Defaults to AND.
|
||||
*
|
||||
* @param string $relation An unsanitized relation prop.
|
||||
* @return string
|
||||
*/
|
||||
private function sanitize_relation( string $relation ): string {
|
||||
if ( ! empty( $relation ) && 'OR' === strtoupper( $relation ) ) {
|
||||
return 'OR';
|
||||
}
|
||||
|
||||
return 'AND';
|
||||
}
|
||||
|
||||
/**
|
||||
* Processes field_query entries and generates the necessary table aliases, JOIN statements and WHERE conditions.
|
||||
*
|
||||
* @param array $q A field query.
|
||||
* @return string An SQL WHERE statement.
|
||||
*/
|
||||
private function process( array $q ) {
|
||||
$where = '';
|
||||
|
||||
if ( empty( $q ) ) {
|
||||
return $where;
|
||||
}
|
||||
|
||||
if ( $this->is_atomic( $q ) ) {
|
||||
$q['alias'] = $this->find_or_create_table_alias_for_clause( $q );
|
||||
$where = $this->generate_where_for_clause( $q );
|
||||
} else {
|
||||
$relation = $q['relation'];
|
||||
unset( $q['relation'] );
|
||||
|
||||
foreach ( $q as $query ) {
|
||||
$chunks[] = $this->process( $query );
|
||||
}
|
||||
|
||||
if ( 1 === count( $chunks ) ) {
|
||||
$where = $chunks[0];
|
||||
} else {
|
||||
$where = '(' . implode( " {$relation} ", $chunks ) . ')';
|
||||
}
|
||||
}
|
||||
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks whether a given field_query clause is atomic or not (i.e. not nested).
|
||||
*
|
||||
* @param array $q The field_query clause.
|
||||
* @return boolean TRUE if atomic, FALSE otherwise.
|
||||
*/
|
||||
private function is_atomic( $q ) {
|
||||
return isset( $q['field'] );
|
||||
}
|
||||
|
||||
/**
|
||||
* Finds a common table alias that the field_query clause can use, or creates one.
|
||||
*
|
||||
* @param array $q An atomic field_query clause.
|
||||
* @return string A table alias for use in an SQL JOIN clause.
|
||||
* @throws \Exception When table info for clause is missing.
|
||||
*/
|
||||
private function find_or_create_table_alias_for_clause( $q ) {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! empty( $q['alias'] ) ) {
|
||||
return $q['alias'];
|
||||
}
|
||||
|
||||
if ( empty( $q['table'] ) || empty( $q['column'] ) ) {
|
||||
throw new \Exception( __( 'Missing table info for query arg.', 'woocommerce' ) );
|
||||
}
|
||||
|
||||
$join = '';
|
||||
|
||||
if ( isset( $q['mapping_id'] ) ) {
|
||||
// Re-use JOINs and aliases from OrdersTableQuery for core tables.
|
||||
$alias = $this->query->get_core_mapping_alias( $q['mapping_id'] );
|
||||
$join = $this->query->get_core_mapping_join( $q['mapping_id'] );
|
||||
} else {
|
||||
$alias = $q['table'];
|
||||
$join = '';
|
||||
}
|
||||
|
||||
if ( in_array( $alias, $this->table_aliases, true ) ) {
|
||||
return $alias;
|
||||
}
|
||||
|
||||
$this->table_aliases[] = $alias;
|
||||
|
||||
if ( $join ) {
|
||||
$this->join[ $alias ] = $join;
|
||||
}
|
||||
|
||||
return $alias;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the correct type for a given clause 'type'.
|
||||
*
|
||||
* @param string $type MySQL type.
|
||||
* @return string MySQL type.
|
||||
*/
|
||||
private function sanitize_cast_type( $type ) {
|
||||
$clause_type = strtoupper( $type );
|
||||
|
||||
if ( ! $clause_type || ! preg_match( '/^(?:BINARY|CHAR|DATE|DATETIME|SIGNED|UNSIGNED|TIME|NUMERIC(?:\(\d+(?:,\s?\d+)?\))?|DECIMAL(?:\(\d+(?:,\s?\d+)?\))?)$/', $clause_type ) ) {
|
||||
return 'CHAR';
|
||||
}
|
||||
|
||||
if ( 'NUMERIC' === $clause_type ) {
|
||||
$clause_type = 'SIGNED';
|
||||
}
|
||||
|
||||
return $clause_type;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates an SQL WHERE clause for a given field_query atomic clause.
|
||||
*
|
||||
* @param array $clause An atomic field_query clause.
|
||||
* @return string An SQL WHERE clause or an empty string if $clause is invalid.
|
||||
*/
|
||||
private function generate_where_for_clause( $clause ): string {
|
||||
global $wpdb;
|
||||
|
||||
$clause_value = $clause['value'] ?? '';
|
||||
|
||||
if ( in_array( $clause['compare'], array( 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN' ), true ) ) {
|
||||
if ( ! is_array( $clause_value ) ) {
|
||||
$clause_value = preg_split( '/[,\s]+/', $clause_value );
|
||||
}
|
||||
} elseif ( is_string( $clause_value ) ) {
|
||||
$clause_value = trim( $clause_value );
|
||||
}
|
||||
|
||||
$clause_compare = $clause['compare'];
|
||||
|
||||
switch ( $clause_compare ) {
|
||||
case 'IN':
|
||||
case 'NOT IN':
|
||||
$where = $wpdb->prepare( '(' . substr( str_repeat( ',%s', count( $clause_value ) ), 1 ) . ')', $clause_value ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
case 'BETWEEN':
|
||||
case 'NOT BETWEEN':
|
||||
$where = $wpdb->prepare( '%s AND %s', $clause_value[0], $clause_value[1] ?? $clause_value[0] );
|
||||
break;
|
||||
case 'LIKE':
|
||||
case 'NOT LIKE':
|
||||
$where = $wpdb->prepare( '%s', '%' . $wpdb->esc_like( $clause_value ) . '%' );
|
||||
break;
|
||||
case 'EXISTS':
|
||||
// EXISTS with a value is interpreted as '='.
|
||||
if ( $clause_value ) {
|
||||
$clause_compare = '=';
|
||||
$where = $wpdb->prepare( '%s', $clause_value );
|
||||
} else {
|
||||
$clause_compare = 'IS NOT';
|
||||
$where = 'NULL';
|
||||
}
|
||||
|
||||
break;
|
||||
case 'NOT EXISTS':
|
||||
// 'value' is ignored for NOT EXISTS.
|
||||
$clause_compare = 'IS';
|
||||
$where = 'NULL';
|
||||
break;
|
||||
default:
|
||||
$where = $wpdb->prepare( '%s', $clause_value );
|
||||
break;
|
||||
}
|
||||
|
||||
if ( $where ) {
|
||||
if ( 'CHAR' === $clause['cast'] ) {
|
||||
return "`{$clause['alias']}`.`{$clause['column']}` {$clause_compare} {$where}";
|
||||
} else {
|
||||
return "CAST(`{$clause['alias']}`.`{$clause['column']}` AS {$clause['cast']}) {$clause_compare} {$where}";
|
||||
}
|
||||
}
|
||||
|
||||
return '';
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns JOIN and WHERE clauses to be appended to the main SQL query.
|
||||
*
|
||||
* @return array {
|
||||
* @type string $join JOIN clause.
|
||||
* @type string $where WHERE clause.
|
||||
* }
|
||||
*/
|
||||
public function get_sql_clauses() {
|
||||
return array(
|
||||
'join' => $this->join,
|
||||
'where' => $this->where ? array( $this->where ) : array(),
|
||||
);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,659 @@
|
||||
<?php
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
defined( 'ABSPATH' ) || exit;
|
||||
|
||||
/**
|
||||
* Class used to implement meta queries for the orders table datastore via {@see OrdersTableQuery}.
|
||||
* Heavily inspired by WordPress' own `WP_Meta_Query` for backwards compatibility reasons.
|
||||
*
|
||||
* Parts of the implementation have been adapted from {@link https://core.trac.wordpress.org/browser/tags/6.0.1/src/wp-includes/class-wp-meta-query.php}.
|
||||
*/
|
||||
class OrdersTableMetaQuery {
|
||||
|
||||
/**
|
||||
* List of non-numeric SQL operators used for comparisons in meta queries.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private const NON_NUMERIC_OPERATORS = array(
|
||||
'=',
|
||||
'!=',
|
||||
'LIKE',
|
||||
'NOT LIKE',
|
||||
'IN',
|
||||
'NOT IN',
|
||||
'EXISTS',
|
||||
'NOT EXISTS',
|
||||
'RLIKE',
|
||||
'REGEXP',
|
||||
'NOT REGEXP',
|
||||
);
|
||||
|
||||
/**
|
||||
* List of numeric SQL operators used for comparisons in meta queries.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private const NUMERIC_OPERATORS = array(
|
||||
'>',
|
||||
'>=',
|
||||
'<',
|
||||
'<=',
|
||||
'BETWEEN',
|
||||
'NOT BETWEEN',
|
||||
|
||||
);
|
||||
|
||||
/**
|
||||
* Prefix used when generating aliases for the metadata table.
|
||||
*
|
||||
* @var string
|
||||
*/
|
||||
private const ALIAS_PREFIX = 'meta';
|
||||
|
||||
/**
|
||||
* Name of the main orders table.
|
||||
*
|
||||
* @var string
|
||||
*/
|
||||
private $meta_table = '';
|
||||
|
||||
/**
|
||||
* Name of the metadata table.
|
||||
*
|
||||
* @var string
|
||||
*/
|
||||
private $orders_table = '';
|
||||
|
||||
/**
|
||||
* Sanitized `meta_query`.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $queries = array();
|
||||
|
||||
/**
|
||||
* Flat list of clauses by name.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $flattened_clauses = array();
|
||||
|
||||
/**
|
||||
* JOIN clauses to add to the main SQL query.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $join = array();
|
||||
|
||||
/**
|
||||
* WHERE clauses to add to the main SQL query.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $where = array();
|
||||
|
||||
/**
|
||||
* Table aliases in use by the meta query. Used to optimize JOINs when possible.
|
||||
*
|
||||
* @var array
|
||||
*/
|
||||
private $table_aliases = array();
|
||||
|
||||
/**
|
||||
* Constructor.
|
||||
*
|
||||
* @param OrdersTableQuery $q The main query being performed.
|
||||
*/
|
||||
public function __construct( OrdersTableQuery $q ) {
|
||||
$meta_query = $q->get( 'meta_query' );
|
||||
|
||||
if ( ! $meta_query ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$this->queries = $this->sanitize_meta_query( $meta_query );
|
||||
|
||||
$this->meta_table = $q->get_table_name( 'meta' );
|
||||
$this->orders_table = $q->get_table_name( 'orders' );
|
||||
|
||||
$this->build_query();
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns JOIN and WHERE clauses to be appended to the main SQL query.
|
||||
*
|
||||
* @return array {
|
||||
* @type string $join JOIN clause.
|
||||
* @type string $where WHERE clause.
|
||||
* }
|
||||
*/
|
||||
public function get_sql_clauses(): array {
|
||||
return array(
|
||||
'join' => $this->sanitize_join( $this->join ),
|
||||
'where' => $this->flatten_where_clauses( $this->where ),
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a list of names (corresponding to meta_query clauses) that can be used as an 'orderby' arg.
|
||||
*
|
||||
* @since 7.4
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function get_orderby_keys(): array {
|
||||
if ( ! $this->flattened_clauses ) {
|
||||
return array();
|
||||
}
|
||||
|
||||
$keys = array();
|
||||
$keys[] = 'meta_value';
|
||||
$keys[] = 'meta_value_num';
|
||||
|
||||
$first_clause = reset( $this->flattened_clauses );
|
||||
if ( $first_clause && ! empty( $first_clause['key'] ) ) {
|
||||
$keys[] = $first_clause['key'];
|
||||
}
|
||||
|
||||
$keys = array_merge(
|
||||
$keys,
|
||||
array_keys( $this->flattened_clauses )
|
||||
);
|
||||
|
||||
return $keys;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns an SQL fragment for the given meta_query key that can be used in an ORDER BY clause.
|
||||
* Call {@see 'get_orderby_keys'} to obtain a list of valid keys.
|
||||
*
|
||||
* @since 7.4
|
||||
*
|
||||
* @param string $key The key name.
|
||||
* @return string
|
||||
*
|
||||
* @throws \Exception When an invalid key is passed.
|
||||
*/
|
||||
public function get_orderby_clause_for_key( string $key ): string {
|
||||
$clause = false;
|
||||
|
||||
if ( isset( $this->flattened_clauses[ $key ] ) ) {
|
||||
$clause = $this->flattened_clauses[ $key ];
|
||||
} else {
|
||||
$first_clause = reset( $this->flattened_clauses );
|
||||
|
||||
if ( $first_clause && ! empty( $first_clause['key'] ) ) {
|
||||
if ( 'meta_value_num' === $key ) {
|
||||
return "{$first_clause['alias']}.meta_value+0";
|
||||
}
|
||||
|
||||
if ( 'meta_value' === $key || $first_clause['key'] === $key ) {
|
||||
$clause = $first_clause;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if ( ! $clause ) {
|
||||
// translators: %s is a meta_query key.
|
||||
throw new \Exception( sprintf( __( 'Invalid meta_query clause key: %s.', 'woocommerce' ), $key ) );
|
||||
}
|
||||
|
||||
return "CAST({$clause['alias']}.meta_value AS {$clause['cast']})";
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks whether a given meta_query clause is atomic or not (i.e. not nested).
|
||||
*
|
||||
* @param array $arg The meta_query clause.
|
||||
* @return boolean TRUE if atomic, FALSE otherwise.
|
||||
*/
|
||||
private function is_atomic( array $arg ): bool {
|
||||
return isset( $arg['key'] ) || isset( $arg['value'] );
|
||||
}
|
||||
|
||||
/**
|
||||
* Sanitizes the meta_query argument.
|
||||
*
|
||||
* @param array $q A meta_query array.
|
||||
* @return array A sanitized meta query array.
|
||||
*/
|
||||
private function sanitize_meta_query( array $q ): array {
|
||||
$sanitized = array();
|
||||
|
||||
foreach ( $q as $key => $arg ) {
|
||||
if ( 'relation' === $key ) {
|
||||
$relation = $arg;
|
||||
} elseif ( ! is_array( $arg ) ) {
|
||||
continue;
|
||||
} elseif ( $this->is_atomic( $arg ) ) {
|
||||
if ( isset( $arg['value'] ) && array() === $arg['value'] ) {
|
||||
unset( $arg['value'] );
|
||||
}
|
||||
|
||||
$arg['compare'] = isset( $arg['compare'] ) ? strtoupper( $arg['compare'] ) : ( isset( $arg['value'] ) && is_array( $arg['value'] ) ? 'IN' : '=' );
|
||||
$arg['compare_key'] = isset( $arg['compare_key'] ) ? strtoupper( $arg['compare_key'] ) : ( isset( $arg['key'] ) && is_array( $arg['key'] ) ? 'IN' : '=' );
|
||||
|
||||
if ( ! in_array( $arg['compare'], self::NON_NUMERIC_OPERATORS, true ) && ! in_array( $arg['compare'], self::NUMERIC_OPERATORS, true ) ) {
|
||||
$arg['compare'] = '=';
|
||||
}
|
||||
|
||||
if ( ! in_array( $arg['compare_key'], self::NON_NUMERIC_OPERATORS, true ) ) {
|
||||
$arg['compare_key'] = '=';
|
||||
}
|
||||
|
||||
$sanitized[ $key ] = $arg;
|
||||
$sanitized[ $key ]['index'] = $key;
|
||||
} else {
|
||||
$sanitized_arg = $this->sanitize_meta_query( $arg );
|
||||
|
||||
if ( $sanitized_arg ) {
|
||||
$sanitized[ $key ] = $sanitized_arg;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if ( $sanitized ) {
|
||||
$sanitized['relation'] = 1 === count( $sanitized ) ? 'OR' : $this->sanitize_relation( $relation ?? 'AND' );
|
||||
}
|
||||
|
||||
return $sanitized;
|
||||
}
|
||||
|
||||
/**
|
||||
* Makes sure we use an AND or OR relation. Defaults to AND.
|
||||
*
|
||||
* @param string $relation An unsanitized relation prop.
|
||||
* @return string
|
||||
*/
|
||||
private function sanitize_relation( string $relation ): string {
|
||||
if ( ! empty( $relation ) && 'OR' === strtoupper( $relation ) ) {
|
||||
return 'OR';
|
||||
}
|
||||
|
||||
return 'AND';
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the correct type for a given meta type.
|
||||
*
|
||||
* @param string $type MySQL type.
|
||||
* @return string MySQL type.
|
||||
*/
|
||||
private function sanitize_cast_type( string $type = '' ): string {
|
||||
$meta_type = strtoupper( $type );
|
||||
|
||||
if ( ! $meta_type || ! preg_match( '/^(?:BINARY|CHAR|DATE|DATETIME|SIGNED|UNSIGNED|TIME|NUMERIC(?:\(\d+(?:,\s?\d+)?\))?|DECIMAL(?:\(\d+(?:,\s?\d+)?\))?)$/', $meta_type ) ) {
|
||||
return 'CHAR';
|
||||
}
|
||||
|
||||
if ( 'NUMERIC' === $meta_type ) {
|
||||
$meta_type = 'SIGNED';
|
||||
}
|
||||
|
||||
return $meta_type;
|
||||
}
|
||||
|
||||
/**
|
||||
* Makes sure a JOIN array does not have duplicates.
|
||||
*
|
||||
* @param array $join A JOIN array.
|
||||
* @return array A sanitized JOIN array.
|
||||
*/
|
||||
private function sanitize_join( array $join ): array {
|
||||
return array_filter( array_unique( array_map( 'trim', $join ) ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Flattens a nested WHERE array.
|
||||
*
|
||||
* @param array $where A possibly nested WHERE array with AND/OR operators.
|
||||
* @return string An SQL WHERE clause.
|
||||
*/
|
||||
private function flatten_where_clauses( $where ): string {
|
||||
if ( is_string( $where ) ) {
|
||||
return trim( $where );
|
||||
}
|
||||
|
||||
$chunks = array();
|
||||
$operator = $this->sanitize_relation( $where['operator'] ?? '' );
|
||||
|
||||
foreach ( $where as $key => $w ) {
|
||||
if ( 'operator' === $key ) {
|
||||
continue;
|
||||
}
|
||||
|
||||
$flattened = $this->flatten_where_clauses( $w );
|
||||
if ( $flattened ) {
|
||||
$chunks[] = $flattened;
|
||||
}
|
||||
}
|
||||
|
||||
if ( $chunks ) {
|
||||
return '(' . implode( " {$operator} ", $chunks ) . ')';
|
||||
} else {
|
||||
return '';
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Builds all the required internal bits for this meta query.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
private function build_query(): void {
|
||||
if ( ! $this->queries ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$queries = $this->queries;
|
||||
$sql_where = $this->process( $queries );
|
||||
$this->where = $sql_where;
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Processes meta_query entries and generates the necessary table aliases, JOIN statements and WHERE conditions.
|
||||
*
|
||||
* @param array $arg A meta query.
|
||||
* @param null|array $parent The parent of the element being processed.
|
||||
* @return array A nested array of WHERE conditions.
|
||||
*/
|
||||
private function process( array &$arg, &$parent = null ): array {
|
||||
$where = array();
|
||||
|
||||
if ( $this->is_atomic( $arg ) ) {
|
||||
$arg['alias'] = $this->find_or_create_table_alias_for_clause( $arg, $parent );
|
||||
$arg['cast'] = $this->sanitize_cast_type( $arg['type'] ?? '' );
|
||||
|
||||
$where = array_filter(
|
||||
array(
|
||||
$this->generate_where_for_clause_key( $arg ),
|
||||
$this->generate_where_for_clause_value( $arg ),
|
||||
)
|
||||
);
|
||||
|
||||
// Store clauses by their key for ORDER BY purposes.
|
||||
$flat_clause_key = is_int( $arg['index'] ) ? $arg['alias'] : $arg['index'];
|
||||
|
||||
$unique_flat_key = $flat_clause_key;
|
||||
$i = 1;
|
||||
while ( isset( $this->flattened_clauses[ $unique_flat_key ] ) ) {
|
||||
$unique_flat_key = $flat_clause_key . '-' . $i;
|
||||
$i++;
|
||||
}
|
||||
|
||||
$this->flattened_clauses[ $unique_flat_key ] =& $arg;
|
||||
} else {
|
||||
// Nested.
|
||||
$relation = $arg['relation'];
|
||||
unset( $arg['relation'] );
|
||||
|
||||
foreach ( $arg as $index => &$clause ) {
|
||||
$chunks[] = $this->process( $clause, $arg );
|
||||
}
|
||||
|
||||
// Merge chunks of the form OR(m) with the surrounding clause.
|
||||
if ( 1 === count( $chunks ) ) {
|
||||
$where = $chunks[0];
|
||||
} else {
|
||||
$where = array_merge(
|
||||
array(
|
||||
'operator' => $relation,
|
||||
),
|
||||
$chunks
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates a JOIN clause to handle an atomic meta_query clause.
|
||||
*
|
||||
* @param array $clause An atomic meta_query clause.
|
||||
* @param string $alias Metadata table alias to use.
|
||||
* @return string An SQL JOIN clause.
|
||||
*/
|
||||
private function generate_join_for_clause( array $clause, string $alias ): string {
|
||||
global $wpdb;
|
||||
|
||||
if ( 'NOT EXISTS' === $clause['compare'] ) {
|
||||
if ( 'LIKE' === $clause['compare_key'] ) {
|
||||
return $wpdb->prepare(
|
||||
"LEFT JOIN {$this->meta_table} AS {$alias} ON ( {$this->orders_table}.id = {$alias}.order_id AND {$alias}.meta_key LIKE %s )", // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
'%' . $wpdb->esc_like( $clause['key'] ) . '%'
|
||||
);
|
||||
} else {
|
||||
return $wpdb->prepare(
|
||||
"LEFT JOIN {$this->meta_table} AS {$alias} ON ( {$this->orders_table}.id = {$alias}.order_id AND {$alias}.meta_key = %s )", // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
$clause['key']
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
return "INNER JOIN {$this->meta_table} AS {$alias} ON ( {$this->orders_table}.id = {$alias}.order_id )";
|
||||
}
|
||||
|
||||
/**
|
||||
* Finds a common table alias that the meta_query clause can use, or creates one.
|
||||
*
|
||||
* @param array $clause An atomic meta_query clause.
|
||||
* @param array $parent_query The parent query this clause is in.
|
||||
* @return string A table alias for use in an SQL JOIN clause.
|
||||
*/
|
||||
private function find_or_create_table_alias_for_clause( array $clause, array $parent_query ): string {
|
||||
if ( ! empty( $clause['alias'] ) ) {
|
||||
return $clause['alias'];
|
||||
}
|
||||
|
||||
$alias = false;
|
||||
$siblings = array_filter(
|
||||
$parent_query,
|
||||
array( __CLASS__, 'is_atomic' )
|
||||
);
|
||||
|
||||
foreach ( $siblings as $sibling ) {
|
||||
if ( empty( $sibling['alias'] ) ) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( $this->is_operator_compatible_with_shared_join( $clause, $sibling, $parent_query['relation'] ?? 'AND' ) ) {
|
||||
$alias = $sibling['alias'];
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if ( ! $alias ) {
|
||||
$alias = self::ALIAS_PREFIX . count( $this->table_aliases );
|
||||
$this->join[] = $this->generate_join_for_clause( $clause, $alias );
|
||||
$this->table_aliases[] = $alias;
|
||||
}
|
||||
|
||||
return $alias;
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks whether two meta_query clauses can share a JOIN.
|
||||
*
|
||||
* @param array $clause An atomic meta_query clause.
|
||||
* @param array $sibling An atomic meta_query clause.
|
||||
* @param string $relation The relation involving both clauses.
|
||||
* @return boolean TRUE if the clauses can share a table alias, FALSE otherwise.
|
||||
*/
|
||||
private function is_operator_compatible_with_shared_join( array $clause, array $sibling, string $relation = 'AND' ): bool {
|
||||
if ( ! $this->is_atomic( $clause ) || ! $this->is_atomic( $sibling ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$valid_operators = array();
|
||||
|
||||
if ( 'OR' === $relation ) {
|
||||
$valid_operators = array( '=', 'IN', 'BETWEEN', 'LIKE', 'REGEXP', 'RLIKE', '>', '>=', '<', '<=' );
|
||||
} elseif ( isset( $sibling['key'] ) && isset( $clause['key'] ) && $sibling['key'] === $clause['key'] ) {
|
||||
$valid_operators = array( '!=', 'NOT IN', 'NOT LIKE' );
|
||||
}
|
||||
|
||||
return in_array( strtoupper( $clause['compare'] ), $valid_operators, true ) && in_array( strtoupper( $sibling['compare'] ), $valid_operators, true );
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates an SQL WHERE clause for a given meta_query atomic clause based on its meta key.
|
||||
* Adapted from WordPress' `WP_Meta_Query::get_sql_for_clause()` method.
|
||||
*
|
||||
* @param array $clause An atomic meta_query clause.
|
||||
* @return string An SQL WHERE clause or an empty string if $clause is invalid.
|
||||
*/
|
||||
private function generate_where_for_clause_key( array $clause ): string {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! array_key_exists( 'key', $clause ) ) {
|
||||
return '';
|
||||
}
|
||||
|
||||
if ( 'NOT EXISTS' === $clause['compare'] ) {
|
||||
return "{$clause['alias']}.order_id IS NULL";
|
||||
}
|
||||
|
||||
$alias = $clause['alias'];
|
||||
|
||||
if ( in_array( $clause['compare_key'], array( '!=', 'NOT IN', 'NOT LIKE', 'NOT EXISTS', 'NOT REGEXP' ), true ) ) {
|
||||
$i = count( $this->table_aliases );
|
||||
$subquery_alias = self::ALIAS_PREFIX . $i;
|
||||
$this->table_aliases[] = $subquery_alias;
|
||||
|
||||
$meta_compare_string_start = 'NOT EXISTS (';
|
||||
$meta_compare_string_start .= "SELECT 1 FROM {$this->meta_table} {$subquery_alias} ";
|
||||
$meta_compare_string_start .= "WHERE {$subquery_alias}.order_id = {$alias}.order_id ";
|
||||
$meta_compare_string_end = 'LIMIT 1';
|
||||
$meta_compare_string_end .= ')';
|
||||
}
|
||||
|
||||
switch ( $clause['compare_key'] ) {
|
||||
case '=':
|
||||
case 'EXISTS':
|
||||
$where = $wpdb->prepare( "$alias.meta_key = %s", trim( $clause['key'] ) ); // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
break;
|
||||
case 'LIKE':
|
||||
$meta_compare_value = '%' . $wpdb->esc_like( trim( $clause['key'] ) ) . '%';
|
||||
$where = $wpdb->prepare( "$alias.meta_key LIKE %s", $meta_compare_value ); // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
break;
|
||||
case 'IN':
|
||||
$meta_compare_string = "$alias.meta_key IN (" . substr( str_repeat( ',%s', count( $clause['key'] ) ), 1 ) . ')';
|
||||
$where = $wpdb->prepare( $meta_compare_string, $clause['key'] ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
case 'RLIKE':
|
||||
case 'REGEXP':
|
||||
$operator = $clause['compare_key'];
|
||||
if ( isset( $clause['type_key'] ) && 'BINARY' === strtoupper( $clause['type_key'] ) ) {
|
||||
$cast = 'BINARY';
|
||||
} else {
|
||||
$cast = '';
|
||||
}
|
||||
$where = $wpdb->prepare( "$alias.meta_key $operator $cast %s", trim( $clause['key'] ) ); // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
break;
|
||||
case '!=':
|
||||
case 'NOT EXISTS':
|
||||
$meta_compare_string = $meta_compare_string_start . "AND $subquery_alias.meta_key = %s " . $meta_compare_string_end;
|
||||
$where = $wpdb->prepare( $meta_compare_string, $clause['key'] ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
case 'NOT LIKE':
|
||||
$meta_compare_string = $meta_compare_string_start . "AND $subquery_alias.meta_key LIKE %s " . $meta_compare_string_end;
|
||||
|
||||
$meta_compare_value = '%' . $wpdb->esc_like( trim( $clause['key'] ) ) . '%';
|
||||
$where = $wpdb->prepare( $meta_compare_string, $meta_compare_value ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
case 'NOT IN':
|
||||
$array_subclause = '(' . substr( str_repeat( ',%s', count( $clause['key'] ) ), 1 ) . ') ';
|
||||
$meta_compare_string = $meta_compare_string_start . "AND $subquery_alias.meta_key IN " . $array_subclause . $meta_compare_string_end;
|
||||
$where = $wpdb->prepare( $meta_compare_string, $clause['key'] ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
case 'NOT REGEXP':
|
||||
$operator = $clause['compare_key'];
|
||||
if ( isset( $clause['type_key'] ) && 'BINARY' === strtoupper( $clause['type_key'] ) ) {
|
||||
$cast = 'BINARY';
|
||||
} else {
|
||||
$cast = '';
|
||||
}
|
||||
|
||||
$meta_compare_string = $meta_compare_string_start . "AND $subquery_alias.meta_key REGEXP $cast %s " . $meta_compare_string_end;
|
||||
$where = $wpdb->prepare( $meta_compare_string, $clause['key'] ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
default:
|
||||
$where = '';
|
||||
break;
|
||||
}
|
||||
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates an SQL WHERE clause for a given meta_query atomic clause based on its meta value.
|
||||
* Adapted from WordPress' `WP_Meta_Query::get_sql_for_clause()` method.
|
||||
*
|
||||
* @param array $clause An atomic meta_query clause.
|
||||
* @return string An SQL WHERE clause or an empty string if $clause is invalid.
|
||||
*/
|
||||
private function generate_where_for_clause_value( $clause ): string {
|
||||
global $wpdb;
|
||||
|
||||
if ( ! array_key_exists( 'value', $clause ) ) {
|
||||
return '';
|
||||
}
|
||||
|
||||
$meta_value = $clause['value'];
|
||||
|
||||
if ( in_array( $clause['compare'], array( 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN' ), true ) ) {
|
||||
if ( ! is_array( $meta_value ) ) {
|
||||
$meta_value = preg_split( '/[,\s]+/', $meta_value );
|
||||
}
|
||||
} elseif ( is_string( $meta_value ) ) {
|
||||
$meta_value = trim( $meta_value );
|
||||
}
|
||||
|
||||
$meta_compare = $clause['compare'];
|
||||
|
||||
switch ( $meta_compare ) {
|
||||
case 'IN':
|
||||
case 'NOT IN':
|
||||
$where = $wpdb->prepare( '(' . substr( str_repeat( ',%s', count( $meta_value ) ), 1 ) . ')', $meta_value ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
break;
|
||||
|
||||
case 'BETWEEN':
|
||||
case 'NOT BETWEEN':
|
||||
$where = $wpdb->prepare( '%s AND %s', $meta_value[0], $meta_value[1] );
|
||||
break;
|
||||
|
||||
case 'LIKE':
|
||||
case 'NOT LIKE':
|
||||
$where = $wpdb->prepare( '%s', '%' . $wpdb->esc_like( $meta_value ) . '%' );
|
||||
break;
|
||||
|
||||
// EXISTS with a value is interpreted as '='.
|
||||
case 'EXISTS':
|
||||
$meta_compare = '=';
|
||||
$where = $wpdb->prepare( '%s', $meta_value );
|
||||
break;
|
||||
|
||||
// 'value' is ignored for NOT EXISTS.
|
||||
case 'NOT EXISTS':
|
||||
$where = '';
|
||||
break;
|
||||
|
||||
default:
|
||||
$where = $wpdb->prepare( '%s', $meta_value );
|
||||
break;
|
||||
}
|
||||
|
||||
if ( $where ) {
|
||||
if ( 'CHAR' === $clause['cast'] ) {
|
||||
return "{$clause['alias']}.meta_value {$meta_compare} {$where}";
|
||||
} else {
|
||||
return "CAST({$clause['alias']}.meta_value AS {$clause['cast']}) {$meta_compare} {$where}";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,213 @@
|
||||
<?php
|
||||
/**
|
||||
* Order refund data store. Refunds are based on orders (essentially negative orders) but there is slight difference in how we save them.
|
||||
* For example, order save hooks etc can't be fired when saving refund, so we need to do it a separate datastore.
|
||||
*/
|
||||
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
use \WC_Cache_Helper;
|
||||
use \WC_Meta_Data;
|
||||
|
||||
/**
|
||||
* Class OrdersTableRefundDataStore.
|
||||
*/
|
||||
class OrdersTableRefundDataStore extends OrdersTableDataStore {
|
||||
|
||||
/**
|
||||
* Data stored in meta keys, but not considered "meta" for refund.
|
||||
*
|
||||
* @var string[]
|
||||
*/
|
||||
protected $internal_meta_keys = array(
|
||||
'_refund_amount',
|
||||
'_refund_reason',
|
||||
'_refunded_by',
|
||||
'_refunded_payment',
|
||||
);
|
||||
|
||||
/**
|
||||
* We do not have and use all the getters and setters from OrderTableDataStore, so we only select the props we actually need.
|
||||
*
|
||||
* @var \string[][]
|
||||
*/
|
||||
protected $operational_data_column_mapping = array(
|
||||
'id' => array( 'type' => 'int' ),
|
||||
'order_id' => array( 'type' => 'int' ),
|
||||
'woocommerce_version' => array(
|
||||
'type' => 'string',
|
||||
'name' => 'version',
|
||||
),
|
||||
'prices_include_tax' => array(
|
||||
'type' => 'bool',
|
||||
'name' => 'prices_include_tax',
|
||||
),
|
||||
'coupon_usages_are_counted' => array(
|
||||
'type' => 'bool',
|
||||
'name' => 'recorded_coupon_usage_counts',
|
||||
),
|
||||
'shipping_tax_amount' => array(
|
||||
'type' => 'decimal',
|
||||
'name' => 'shipping_tax',
|
||||
),
|
||||
'shipping_total_amount' => array(
|
||||
'type' => 'decimal',
|
||||
'name' => 'shipping_total',
|
||||
),
|
||||
'discount_tax_amount' => array(
|
||||
'type' => 'decimal',
|
||||
'name' => 'discount_tax',
|
||||
),
|
||||
'discount_total_amount' => array(
|
||||
'type' => 'decimal',
|
||||
'name' => 'discount_total',
|
||||
),
|
||||
);
|
||||
|
||||
/**
|
||||
* Delete a refund order from database.
|
||||
*
|
||||
* @param \WC_Order $refund Refund object to delete.
|
||||
* @param array $args Array of args to pass to the delete method.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function delete( &$refund, $args = array() ) {
|
||||
$refund_id = $refund->get_id();
|
||||
if ( ! $refund_id ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$refund_cache_key = WC_Cache_Helper::get_cache_prefix( 'orders' ) . 'refunds' . $refund->get_parent_id();
|
||||
wp_cache_delete( $refund_cache_key, 'orders' );
|
||||
|
||||
$this->delete_order_data_from_custom_order_tables( $refund_id );
|
||||
$refund->set_id( 0 );
|
||||
|
||||
$orders_table_is_authoritative = $refund->get_data_store()->get_current_class_name() === self::class;
|
||||
|
||||
if ( $orders_table_is_authoritative ) {
|
||||
$data_synchronizer = wc_get_container()->get( DataSynchronizer::class );
|
||||
if ( $data_synchronizer->data_sync_is_enabled() ) {
|
||||
// Delete the associated post, which in turn deletes order items, etc. through {@see WC_Post_Data}.
|
||||
// Once we stop creating posts for orders, we should do the cleanup here instead.
|
||||
wp_delete_post( $refund_id );
|
||||
} else {
|
||||
$this->handle_order_deletion_with_sync_disabled( $refund_id );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper method to set refund props.
|
||||
*
|
||||
* @param \WC_Order_Refund $refund Refund object.
|
||||
* @param object $data DB data object.
|
||||
*
|
||||
* @since 8.0.0
|
||||
*/
|
||||
protected function set_order_props_from_data( &$refund, $data ) {
|
||||
parent::set_order_props_from_data( $refund, $data );
|
||||
foreach ( $data->meta_data as $meta ) {
|
||||
switch ( $meta->meta_key ) {
|
||||
case '_refund_amount':
|
||||
$refund->set_amount( $meta->meta_value );
|
||||
break;
|
||||
case '_refunded_by':
|
||||
$refund->set_refunded_by( $meta->meta_value );
|
||||
break;
|
||||
case '_refunded_payment':
|
||||
$refund->set_refunded_payment( wc_string_to_bool( $meta->meta_value ) );
|
||||
break;
|
||||
case '_refund_reason':
|
||||
$refund->set_reason( $meta->meta_value );
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Method to create a refund in the database.
|
||||
*
|
||||
* @param \WC_Abstract_Order $refund Refund object.
|
||||
*/
|
||||
public function create( &$refund ) {
|
||||
$refund->set_status( 'completed' ); // Refund are always marked completed.
|
||||
$this->persist_save( $refund );
|
||||
}
|
||||
|
||||
/**
|
||||
* Update refund in database.
|
||||
*
|
||||
* @param \WC_Order $refund Refund object.
|
||||
*/
|
||||
public function update( &$refund ) {
|
||||
$this->persist_updates( $refund );
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper method that updates post meta based on an refund object.
|
||||
* Mostly used for backwards compatibility purposes in this datastore.
|
||||
*
|
||||
* @param \WC_Order $refund Refund object.
|
||||
*/
|
||||
public function update_order_meta( &$refund ) {
|
||||
parent::update_order_meta( $refund );
|
||||
|
||||
// Update additional props.
|
||||
$updated_props = array();
|
||||
$meta_key_to_props = array(
|
||||
'_refund_amount' => 'amount',
|
||||
'_refunded_by' => 'refunded_by',
|
||||
'_refunded_payment' => 'refunded_payment',
|
||||
'_refund_reason' => 'reason',
|
||||
);
|
||||
|
||||
$props_to_update = $this->get_props_to_update( $refund, $meta_key_to_props );
|
||||
foreach ( $props_to_update as $meta_key => $prop ) {
|
||||
$meta_object = new WC_Meta_Data();
|
||||
$meta_object->key = $meta_key;
|
||||
$meta_object->value = $refund->{"get_$prop"}( 'edit' );
|
||||
$existing_meta = $this->data_store_meta->get_metadata_by_key( $refund, $meta_key );
|
||||
if ( $existing_meta ) {
|
||||
$existing_meta = $existing_meta[0];
|
||||
$meta_object->id = $existing_meta->id;
|
||||
$this->update_meta( $refund, $meta_object );
|
||||
} else {
|
||||
$this->add_meta( $refund, $meta_object );
|
||||
}
|
||||
$updated_props[] = $prop;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fires after updating meta for a order refund.
|
||||
*
|
||||
* @since 2.7.0
|
||||
*/
|
||||
do_action( 'woocommerce_order_refund_object_updated_props', $refund, $updated_props );
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a title for the new post type.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_post_title() {
|
||||
return sprintf(
|
||||
/* translators: %s: Order date */
|
||||
__( 'Refund – %s', 'woocommerce' ),
|
||||
( new \DateTime( 'now' ) )->format( _x( 'M d, Y @ h:i A', 'Order date parsed by DateTime::format', 'woocommerce' ) ) // phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment, WordPress.WP.I18n.UnorderedPlaceholdersText
|
||||
);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Returns data store object to use backfilling.
|
||||
*
|
||||
* @return \WC_Order_Refund_Data_Store_CPT
|
||||
*/
|
||||
protected function get_post_data_store_for_backfill() {
|
||||
return new \WC_Order_Refund_Data_Store_CPT();
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,165 @@
|
||||
<?php
|
||||
|
||||
namespace Automattic\WooCommerce\Internal\DataStores\Orders;
|
||||
|
||||
use Exception;
|
||||
|
||||
/**
|
||||
* Creates the join and where clauses needed to perform an order search using Custom Order Tables.
|
||||
*
|
||||
* @internal
|
||||
*/
|
||||
class OrdersTableSearchQuery {
|
||||
/**
|
||||
* Holds the Orders Table Query object.
|
||||
*
|
||||
* @var OrdersTableQuery
|
||||
*/
|
||||
private $query;
|
||||
|
||||
/**
|
||||
* Holds the search term to be used in the WHERE clauses.
|
||||
*
|
||||
* @var string
|
||||
*/
|
||||
private $search_term;
|
||||
|
||||
/**
|
||||
* Creates the JOIN and WHERE clauses needed to execute a search of orders.
|
||||
*
|
||||
* @internal
|
||||
*
|
||||
* @param OrdersTableQuery $query The order query object.
|
||||
*/
|
||||
public function __construct( OrdersTableQuery $query ) {
|
||||
$this->query = $query;
|
||||
$this->search_term = urldecode( $query->get( 's' ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Supplies an array of clauses to be used in an order query.
|
||||
*
|
||||
* @internal
|
||||
* @throws Exception If unable to generate either the JOIN or WHERE SQL fragments.
|
||||
*
|
||||
* @return array {
|
||||
* @type string $join JOIN clause.
|
||||
* @type string $where WHERE clause.
|
||||
* }
|
||||
*/
|
||||
public function get_sql_clauses(): array {
|
||||
return array(
|
||||
'join' => array( $this->generate_join() ),
|
||||
'where' => array( $this->generate_where() ),
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates the necessary JOIN clauses for the order search to be performed.
|
||||
*
|
||||
* @throws Exception May be triggered if a table name cannot be determined.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
private function generate_join(): string {
|
||||
$orders_table = $this->query->get_table_name( 'orders' );
|
||||
$items_table = $this->query->get_table_name( 'items' );
|
||||
|
||||
return "
|
||||
LEFT JOIN $items_table AS search_query_items ON search_query_items.order_id = $orders_table.id
|
||||
";
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates the necessary WHERE clauses for the order search to be performed.
|
||||
*
|
||||
* @throws Exception May be triggered if a table name cannot be determined.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
private function generate_where(): string {
|
||||
global $wpdb;
|
||||
$where = '';
|
||||
$possible_order_id = (string) absint( $this->search_term );
|
||||
$order_table = $this->query->get_table_name( 'orders' );
|
||||
|
||||
// Support the passing of an order ID as the search term.
|
||||
if ( (string) $this->query->get( 's' ) === $possible_order_id ) {
|
||||
$where = "`$order_table`.id = $possible_order_id OR ";
|
||||
}
|
||||
|
||||
$meta_sub_query = $this->generate_where_for_meta_table();
|
||||
|
||||
$where .= $wpdb->prepare(
|
||||
"
|
||||
search_query_items.order_item_name LIKE %s
|
||||
OR `$order_table`.id IN ( $meta_sub_query )
|
||||
",
|
||||
'%' . $wpdb->esc_like( $this->search_term ) . '%'
|
||||
);
|
||||
|
||||
return " ( $where ) ";
|
||||
}
|
||||
|
||||
/**
|
||||
* Generates where clause for meta table.
|
||||
*
|
||||
* Note we generate the where clause as a subquery to be used by calling function inside the IN clause. This is against the general wisdom for performance, but in this particular case, a subquery is able to use the order_id-meta_key-meta_value index, which is not possible with a join.
|
||||
*
|
||||
* Since it can use the index, which otherwise would not be possible, it is much faster than both LEFT JOIN or SQL_CALC approach that could have been used.
|
||||
*
|
||||
* @return string The where clause for meta table.
|
||||
*/
|
||||
private function generate_where_for_meta_table(): string {
|
||||
global $wpdb;
|
||||
$meta_table = $this->query->get_table_name( 'meta' );
|
||||
$meta_fields = $this->get_meta_fields_to_be_searched();
|
||||
return $wpdb->prepare(
|
||||
"
|
||||
SELECT search_query_meta.order_id
|
||||
FROM $meta_table as search_query_meta
|
||||
WHERE search_query_meta.meta_key IN ( $meta_fields )
|
||||
AND search_query_meta.meta_value LIKE %s
|
||||
GROUP BY search_query_meta.order_id
|
||||
",
|
||||
'%' . $wpdb->esc_like( $this->search_term ) . '%'
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the order meta field keys to be searched.
|
||||
*
|
||||
* These will be returned as a single string, where the meta keys have been escaped, quoted and are
|
||||
* comma-separated (ie, "'abc', 'foo'" - ready for inclusion in a SQL IN() clause).
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
private function get_meta_fields_to_be_searched(): string {
|
||||
/**
|
||||
* Controls the order meta keys to be included in search queries.
|
||||
*
|
||||
* This hook is used when Custom Order Tables are in use: the corresponding hook when CPT-orders are in use
|
||||
* is 'woocommerce_shop_order_search_fields'.
|
||||
*
|
||||
* @since 7.0.0
|
||||
*
|
||||
* @param array
|
||||
*/
|
||||
$meta_keys = apply_filters(
|
||||
'woocommerce_order_table_search_query_meta_keys',
|
||||
array(
|
||||
'_billing_address_index',
|
||||
'_shipping_address_index',
|
||||
)
|
||||
);
|
||||
|
||||
$meta_keys = (array) array_map(
|
||||
function ( string $meta_key ): string {
|
||||
return "'" . esc_sql( wc_clean( $meta_key ) ) . "'";
|
||||
},
|
||||
$meta_keys
|
||||
);
|
||||
|
||||
return implode( ',', $meta_keys );
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user