laipower/wp-content/plugins/easy-digital-downloads/includes/database/engine/class-query.php

3018 lines
76 KiB
PHP
Raw Normal View History

<?php
/**
* Base Custom Database Table Query Class.
*
* @package Database
* @subpackage Query
* @copyright Copyright (c) 2020
* @license https://opensource.org/licenses/gpl-2.0.php GNU Public License
* @since 1.0.0
*/
namespace EDD\Database;
// Exit if accessed directly
defined( 'ABSPATH' ) || exit;
/**
* Base class used for querying custom database tables.
*
* This class is intended to be extended for each unique database table,
* including global tables for multisite, and users tables.
*
* @since 1.0.0
*
* @see Query::__construct() for accepted arguments.
*
* @property string $prefix
* @property string $table_name
* @property string $table_alias
* @property string $table_schema
* @property string $item_name
* @property string $item_name_plural
* @property string $item_shape
* @property string $cache_group
* @property string $columns
* @property string $query_clauses
* @property string $request_clauses
* @property Queries\Meta $meta_query
* @property Queries\Date $date_query
* @property Queries\Compare $compare
* @property array $query_vars
* @property array $query_var_originals
* @property array $query_var_defaults
* @property string $query_var_default_value
* @property array $items
* @property int $found_items
* @property int $max_num_pages
* @property string $request
* @property int $last_changed
*/
class Query extends Base {
/** Table Properties ******************************************************/
/**
* Name of the database table to query.
*
* @since 1.0.0
* @var string
*/
protected $table_name = '';
/**
* String used to alias the database table in MySQL statement.
*
* Keep this short, but descriptive. I.E. "tr" for term relationships.
*
* This is used to avoid collisions with JOINs.
*
* @since 1.0.0
* @var string
*/
protected $table_alias = '';
/**
* Name of class used to setup the database schema.
*
* @since 1.0.0
* @var string
*/
protected $table_schema = '\\EDD\\Database\\Schema';
/** Item ******************************************************************/
/**
* Name for a single item.
*
* Use underscores between words. I.E. "term_relationship"
*
* This is used to automatically generate action hooks.
*
* @since 1.0.0
* @var string
*/
protected $item_name = '';
/**
* Plural version for a group of items.
*
* Use underscores between words. I.E. "term_relationships"
*
* This is used to automatically generate action hooks.
*
* @since 1.0.0
* @var string
*/
protected $item_name_plural = '';
/**
* Name of class used to turn IDs into first-class objects.
*
* This is used when looping through return values to guarantee their shape.
*
* @since 1.0.0
* @var mixed
*/
protected $item_shape = '\\EDD\\Database\\Row';
/** Cache *****************************************************************/
/**
* Group to cache queries and queried items in.
*
* Use underscores between words. I.E. "some_items"
*
* Do not use colons: ":". These are reserved for internal use only.
*
* @since 1.0.0
* @var string
*/
protected $cache_group = '';
/**
* The last updated time.
*
* @since 1.0.0
* @var int
*/
protected $last_changed = 0;
/** Columns ***************************************************************/
/**
* Array of all database column objects.
*
* @since 1.0.0
* @var array
*/
protected $columns = array();
/** Clauses ***************************************************************/
/**
* SQL query clauses.
*
* @since 1.0.0
* @var array
*/
protected $query_clauses = array(
'select' => '',
'from' => '',
'where' => array(),
'groupby' => '',
'orderby' => '',
'limits' => ''
);
/**
* Request clauses.
*
* @since 1.0.0
* @var array
*/
protected $request_clauses = array(
'select' => '',
'from' => '',
'where' => '',
'groupby' => '',
'orderby' => '',
'limits' => ''
);
/**
* Meta query container.
*
* @since 1.0.0
* @var object|Queries\Meta
*/
protected $meta_query = false;
/**
* Date query container.
*
* @since 1.0.0
* @var object|Queries\Date
*/
protected $date_query = false;
/**
* Compare query container.
*
* @since 1.0.0
* @var object|Queries\Compare
*/
protected $compare_query = false;
/** Query Variables *******************************************************/
/**
* Parsed query vars set by the application, possibly filtered and changed.
*
* This is specifically marked as public, to allow byref actions to change
* them from outside the class methods proper and inside filter functions.
*
* @since 1.0.0
* @var array
*/
public $query_vars = array();
/**
* Original query vars set by the application.
*
* These are the original query variables before any filters are applied,
* and are the results of merging $query_var_defaults with $query_vars.
*
* @since 1.0.0
* @var array
*/
protected $query_var_originals = array();
/**
* Default values for query vars.
*
* These are computed at runtime based on the registered columns for the
* database table this query relates to.
*
* @since 1.0.0
* @var array
*/
protected $query_var_defaults = array();
/**
* This private variable temporarily holds onto a random string used as the
* default query var value. This is used internally when performing
* comparisons, and allows for querying by falsy values.
*
* @since 1.1.0
* @var string
*/
protected $query_var_default_value = '';
/** Results ***************************************************************/
/**
* List of items located by the query.
*
* @since 1.0.0
* @var array
*/
public $items = array();
/**
* The amount of found items for the current query.
*
* @since 1.0.0
* @var int
*/
protected $found_items = 0;
/**
* The number of pages.
*
* @since 1.0.0
* @var int
*/
protected $max_num_pages = 0;
/**
* SQL for database query.
*
* @since 1.0.0
* @var string
*/
protected $request = '';
/** Methods ***************************************************************/
/**
* Sets up the item query, based on the query vars passed.
*
* @since 1.0.0
*
* @param string|array $query {
* Optional. Array or query string of item query parameters.
* Default empty.
*
* @type string $fields Site fields to return. Accepts 'ids' (returns an array of item IDs)
* or empty (returns an array of complete item objects). Default empty.
* To do a date query against a field, append the field name with _query
* @type bool $count Whether to return a item count (true) or array of item objects.
* Default false.
* @type int $number Limit number of items to retrieve. Use 0 for no limit.
* Default 100.
* @type int $offset Number of items to offset the query. Used to build LIMIT clause.
* Default 0.
* @type bool $no_found_rows Whether to disable the `SQL_CALC_FOUND_ROWS` query.
* Default true.
* @type string|array $orderby Accepts false, an empty array, or 'none' to disable `ORDER BY` clause.
* Default 'id'.
* @type string $item How to item retrieved items. Accepts 'ASC', 'DESC'.
* Default 'DESC'.
* @type string $search Search term(s) to retrieve matching items for.
* Default empty.
* @type array $search_columns Array of column names to be searched.
* Default empty array.
* @type bool $update_item_cache Whether to prime the cache for found items.
* Default false.
* @type bool $update_meta_cache Whether to prime the meta cache for found items.
* Default false.
* }
*/
public function __construct( $query = array() ) {
// Setup
$this->set_alias();
$this->set_prefix();
$this->set_columns();
$this->set_item_shape();
$this->set_query_var_defaults();
// Maybe execute a query if arguments were passed
if ( ! empty( $query ) ) {
$this->query( $query );
}
}
/**
* Queries the database and retrieves items or counts.
*
* This method is public to allow subclasses to perform JIT manipulation
* of the parameters passed into it.
*
* @since 1.0.0
*
* @param string|array $query Array or URL query string of parameters.
* @return array|int List of items, or number of items when 'count' is passed as a query var.
*/
public function query( $query = array() ) {
$this->parse_query( $query );
return $this->get_items();
}
/** Private Setters *******************************************************/
/**
* Set the time when items were last changed.
*
* We set this locally to avoid inconsistencies between method calls.
*
* @since 1.0.0
*/
protected function set_last_changed() {
$this->last_changed = microtime();
}
/**
* Set up the table alias if not already set in the class.
*
* This happens before prefixes are applied.
*
* @since 1.0.0
*/
protected function set_alias() {
if ( empty( $this->table_alias ) ) {
$this->table_alias = $this->first_letters( $this->table_name );
}
}
/**
* Prefix table names, cache groups, and other things.
*
* This is to avoid conflicts with other plugins or themes that might be
* doing their own things.
*
* @since 1.0.0
*/
protected function set_prefix() {
$this->table_name = $this->apply_prefix( $this->table_name );
$this->table_alias = $this->apply_prefix( $this->table_alias );
$this->cache_group = $this->apply_prefix( $this->cache_group, '-' );
}
/**
* Set columns objects
*
* @since 1.0.0
*/
protected function set_columns() {
// Bail if no table schema
if ( ! class_exists( $this->table_schema ) ) {
return;
}
// Invoke a new table schema class
$schema = new $this->table_schema;
// Maybe get the column objects
if ( ! empty( $schema->columns ) ) {
$this->columns = $schema->columns;
}
}
/**
* Set the default item shape if none exists
*
* @since 1.0.0
*/
protected function set_item_shape() {
if ( empty( $this->item_shape ) || ! class_exists( $this->item_shape ) ) {
$this->item_shape = __NAMESPACE__ . '\\Row';
}
}
/**
* Set default query vars based on columns
*
* @since 1.0.0
*/
protected function set_query_var_defaults() {
// Default query variable value
$this->query_var_default_value = function_exists( 'random_bytes' )
? $this->apply_prefix( bin2hex( random_bytes( 18 ) ) )
: $this->apply_prefix( uniqid( '_', true ) );
// Default query variables
$this->query_var_defaults = array(
'fields' => '',
'number' => 100,
'offset' => '',
'orderby' => 'id',
'order' => 'DESC',
'groupby' => '',
'search' => '',
'search_columns' => array(),
'count' => false,
'meta_query' => null, // See Queries\Meta
'date_query' => null, // See Queries\Date
'compare_query' => null, // See Queries\Compare
'no_found_rows' => true,
// Caching
'update_item_cache' => true,
'update_meta_cache' => true
);
// Bail if no columns
if ( empty( $this->columns ) ) {
return;
}
// Direct column names
$names = wp_list_pluck( $this->columns, 'name' );
foreach ( $names as $name ) {
$this->query_var_defaults[ $name ] = $this->query_var_default_value;
}
// Possible ins
$possible_ins = $this->get_columns( array( 'in' => true ), 'and', 'name' );
foreach ( $possible_ins as $in ) {
$key = "{$in}__in";
$this->query_var_defaults[ $key ] = false;
}
// Possible not ins
$possible_not_ins = $this->get_columns( array( 'not_in' => true ), 'and', 'name' );
foreach ( $possible_not_ins as $in ) {
$key = "{$in}__not_in";
$this->query_var_defaults[ $key ] = false;
}
// Possible dates
$possible_dates = $this->get_columns( array( 'date_query' => true ), 'and', 'name' );
foreach ( $possible_dates as $date ) {
$key = "{$date}_query";
$this->query_var_defaults[ $key ] = false;
}
}
/**
* Set the request clauses
*
* @since 1.0.0
*
* @param array $clauses
*/
protected function set_request_clauses( $clauses = array() ) {
// Found rows
$found_rows = empty( $this->query_vars['no_found_rows'] )
? 'SQL_CALC_FOUND_ROWS'
: '';
// Fields
$fields = ! empty( $clauses['fields'] )
? $clauses['fields']
: '';
// Join
$join = ! empty( $clauses['join'] )
? $clauses['join']
: '';
// Where
$where = ! empty( $clauses['where'] )
? "WHERE {$clauses['where']}"
: '';
// Group by
$groupby = ! empty( $clauses['groupby'] )
? "GROUP BY {$clauses['groupby']}"
: '';
// Order by
$orderby = ! empty( $clauses['orderby'] )
? "ORDER BY {$clauses['orderby']}"
: '';
// Limits
$limits = ! empty( $clauses['limits'] )
? $clauses['limits']
: '';
// Select & From
$table = $this->get_table_name();
$select = "SELECT {$found_rows} {$fields}";
$from = "FROM {$table} {$this->table_alias} {$join}";
// Put query into clauses array
$this->request_clauses['select'] = $select;
$this->request_clauses['from'] = $from;
$this->request_clauses['where'] = $where;
$this->request_clauses['groupby'] = $groupby;
$this->request_clauses['orderby'] = $orderby;
$this->request_clauses['limits'] = $limits;
}
/**
* Set the request
*
* @since 1.0.0
*/
protected function set_request() {
$filtered = array_filter( $this->request_clauses );
$clauses = array_map( 'trim', $filtered );
$this->request = implode( ' ', $clauses );
}
/**
* Set items by mapping them through the single item callback.
*
* @since 1.0.0
* @param array $item_ids
*/
protected function set_items( $item_ids = array() ) {
// Bail if counting, to avoid shaping items
if ( ! empty( $this->query_vars['count'] ) ) {
$this->items = $item_ids;
return;
}
// Cast to integers
$item_ids = array_map( 'intval', $item_ids );
// Prime item caches
$this->prime_item_caches( $item_ids );
// Shape the items
$this->items = $this->shape_items( $item_ids );
}
/**
* Populates found_items and max_num_pages properties for the current query
* if the limit clause was used.
*
* @since 1.0.0
*
* @param array $item_ids Optional array of item IDs
*/
protected function set_found_items( $item_ids = array() ) {
// Items were not found
if ( empty( $item_ids ) ) {
return;
}
// Default to number of item IDs
$this->found_items = count( (array) $item_ids );
// Count query
if ( ! empty( $this->query_vars['count'] ) ) {
// Not grouped
if ( is_numeric( $item_ids ) && empty( $this->query_vars['groupby'] ) ) {
$this->found_items = intval( $item_ids );
}
// Not a count query
} elseif ( is_array( $item_ids ) && ( ! empty( $this->query_vars['number'] ) && empty( $this->query_vars['no_found_rows'] ) ) ) {
/**
* Filters the query used to retrieve found item count.
*
* @since 1.0.0
*
* @param string $found_items_query SQL query. Default 'SELECT FOUND_ROWS()'.
* @param object $item_query The object instance.
*/
$found_items_query = (string) apply_filters_ref_array( $this->apply_prefix( "found_{$this->item_name_plural}_query" ), array( 'SELECT FOUND_ROWS()', &$this ) );
// Maybe query for found items
if ( ! empty( $found_items_query ) ) {
$this->found_items = (int) $this->get_db()->get_var( $found_items_query );
}
}
}
/** Public Setters ********************************************************/
/**
* Set a query var, to both defaults and request arrays.
*
* This method is used to expose the private query_vars array to hooks,
* allowing them to manipulate query vars just-in-time.
*
* @since 1.0.0
*
* @param string $key
* @param string $value
*/
public function set_query_var( $key = '', $value = '' ) {
$this->query_var_defaults[ $key ] = $value;
$this->query_vars[ $key ] = $value;
}
/**
* Check whether a query variable strictly equals the unique default
* starting value.
*
* @since 1.1.0
* @param string $key
* @return bool
*/
public function is_query_var_default( $key = '' ) {
return (bool) ( $this->query_vars[ $key ] === $this->query_var_default_value );
}
/** Private Getters *******************************************************/
/**
* Pass-through method to return a new Meta object.
*
* @since 1.0.0
*
* @param array $args See Queries\Meta
*
* @return Queries\Meta
*/
private function get_meta_query( $args = array() ) {
return new Queries\Meta( $args );
}
/**
* Pass-through method to return a new Compare object.
*
* @since 1.0.0
*
* @param array $args See Queries\Compare
*
* @return Queries\Compare
*/
private function get_compare_query( $args = array() ) {
return new Queries\Compare( $args );
}
/**
* Pass-through method to return a new Queries\Date object.
*
* @since 1.0.0
*
* @param array $args See Queries\Date
*
* @return Queries\Date
*/
private function get_date_query( $args = array() ) {
return new Queries\Date( $args );
}
/**
* Return the current time as a UTC timestamp
*
* This is used by add_item() and update_item()
*
* @since 1.0.0
*
* @return string
*/
protected function get_current_time() {
return gmdate( "Y-m-d\TH:i:s\Z" );
}
/**
* Return the literal table name (with prefix) from the database interface.
*
* @since 1.0.0
*
* @return string
*/
protected function get_table_name() {
return $this->get_db()->{$this->table_name};
}
/**
* Return array of column names
*
* @since 1.0.0
*
* @return array
*/
protected function get_column_names() {
return array_flip( $this->get_columns( array(), 'and', 'name' ) );
}
/**
* Return the primary database column name
*
* @since 1.0.0
*
* @return string Default "id", Primary column name if not empty
*/
protected function get_primary_column_name() {
return $this->get_column_field( array( 'primary' => true ), 'name', 'id' );
}
/**
* Get a column from an array of arguments
*
* @since 1.0.0
*
* @return mixed Column object, or false
*/
protected function get_column_field( $args = array(), $field = '', $default = false ) {
// Get column
$column = $this->get_column_by( $args );
// Return field, or default
return isset( $column->{$field} )
? $column->{$field}
: $default;
}
/**
* Get a column from an array of arguments
*
* @since 1.0.0
*
* @return mixed Column object, or false
*/
protected function get_column_by( $args = array() ) {
// Filter columns
$filter = $this->get_columns( $args );
// Return column or false
return ! empty( $filter )
? reset( $filter )
: false;
}
/**
* Get columns from an array of arguments
*
* @since 1.0.0
*/
protected function get_columns( $args = array(), $operator = 'and', $field = false ) {
// Filter columns
$filter = wp_filter_object_list( $this->columns, $args, $operator, $field );
// Return column or false
return ! empty( $filter )
? array_values( $filter )
: array();
}
/**
* Get a single database row by any column and value, skipping cache.
*
* @since 1.0.0
*
* @param string $column_name Name of database column
* @param string $column_value Value to query for
* @return object|false False if empty/error, Object if successful
*/
protected function get_item_raw( $column_name = '', $column_value = '' ) {
// Bail if no name or value
if ( empty( $column_name ) || empty( $column_value ) ) {
return false;
}
// Bail if values aren't query'able
if ( ! is_string( $column_name ) || ! is_scalar( $column_value ) ) {
return false;
}
// Query database for row
$pattern = $this->get_column_field( array( 'name' => $column_name ), 'pattern', '%s' );
$table = $this->get_table_name();
$select = $this->get_db()->prepare( "SELECT * FROM {$table} WHERE {$column_name} = {$pattern}", $column_value );
$result = $this->get_db()->get_row( $select );
// Bail on failure
if ( ! $this->is_success( $result ) ) {
return false;
}
// Return row
return $result;
}
/**
* Retrieves a list of items matching the query vars.
*
* @since 1.0.0
*
* @return array|int List of items, or number of items when 'count' is passed as a query var.
*/
protected function get_items() {
/**
* Fires before object items are retrieved.
*
* @since 1.0.0
*
* @param Query &$this Current instance of Query, passed by reference.
*/
do_action_ref_array( $this->apply_prefix( "pre_get_{$this->item_name_plural}" ), array( &$this ) );
// Never limit, never update item/meta caches when counting
if ( ! empty( $this->query_vars['count'] ) ) {
$this->query_vars['number'] = false;
$this->query_vars['no_found_rows'] = true;
$this->query_vars['update_item_cache'] = false;
$this->query_vars['update_meta_cache'] = false;
}
// Check the cache
$cache_key = $this->get_cache_key();
$cache_value = $this->cache_get( $cache_key, $this->cache_group );
// No cache value
if ( false === $cache_value ) {
$item_ids = $this->get_item_ids();
// Set the number of found items
$this->set_found_items( $item_ids );
// Format the cached value
$cache_value = array(
'item_ids' => $item_ids,
'found_items' => intval( $this->found_items ),
);
// Add value to the cache
$this->cache_add( $cache_key, $cache_value, $this->cache_group );
// Value exists in cache
} else {
$item_ids = $cache_value['item_ids'];
$this->found_items = intval( $cache_value['found_items'] );
}
// Pagination
if ( ! empty( $this->found_items ) && ! empty( $this->query_vars['number'] ) ) {
$this->max_num_pages = ceil( $this->found_items / $this->query_vars['number'] );
}
// Cast to int if not grouping counts
if ( ! empty( $this->query_vars['count'] ) && empty( $this->query_vars['groupby'] ) ) {
$item_ids = intval( $item_ids );
}
// Set items from IDs
$this->set_items( $item_ids );
// Return array of items
return $this->items;
}
/**
* Used internally to get a list of item IDs matching the query vars.
*
* @since 1.0.0
*
* @return int|array A single count of item IDs if a count query. An array of item IDs if a full query.
*/
protected function get_item_ids() {
// Setup primary column, and parse the where clause
$this->parse_where();
// Order & Order By
$order = $this->parse_order( $this->query_vars['order'] );
$orderby = $this->get_order_by( $order );
// Limit & Offset
$limit = absint( $this->query_vars['number'] );
$offset = absint( $this->query_vars['offset'] );
// Limits
if ( ! empty( $limit ) ) {
$limits = ! empty( $offset )
? "LIMIT {$offset}, {$limit}"
: "LIMIT {$limit}";
} else {
$limits = '';
}
// Where & Join
$where = implode( ' AND ', $this->query_clauses['where'] );
$join = implode( ', ', $this->query_clauses['join'] );
// Group by
$groupby = $this->parse_groupby( $this->query_vars['groupby'] );
// Fields
$fields = $this->parse_fields( $this->query_vars['fields'] );
// Setup the query array (compact() is too opaque here)
$query = array(
'fields' => $fields,
'join' => $join,
'where' => $where,
'orderby' => $orderby,
'limits' => $limits,
'groupby' => $groupby
);
/**
* Filters the item query clauses.
*
* @since 1.0.0
*
* @param array $pieces A compacted array of item query clauses.
* @param Query &$this Current instance passed by reference.
*/
$clauses = (array) apply_filters_ref_array( $this->apply_prefix( "{$this->item_name_plural}_query_clauses" ), array( $query, &$this ) );
// Setup request
$this->set_request_clauses( $clauses );
$this->set_request();
// Return count
if ( ! empty( $this->query_vars['count'] ) ) {
// Get vars or results
$retval = empty( $this->query_vars['groupby'] )
? $this->get_db()->get_var( $this->request )
: $this->get_db()->get_results( $this->request, ARRAY_A );
// Return vars or results
return $retval;
}
// Get IDs
$item_ids = $this->get_db()->get_col( $this->request );
// Return parsed IDs
return wp_parse_id_list( $item_ids );
}
/**
* Get the ORDERBY clause.
*
* @since 1.0.0
*
* @param string $order
* @return string
*/
protected function get_order_by( $order = '' ) {
// Default orderby primary column
$orderby = "{$this->parse_orderby()} {$order}";
// Disable ORDER BY if counting, or: 'none', an empty array, or false.
if ( ! empty( $this->query_vars['count'] ) || in_array( $this->query_vars['orderby'], array( 'none', array(), false ), true ) ) {
$orderby = '';
// Ordering by something, so figure it out
} elseif ( ! empty( $this->query_vars['orderby'] ) ) {
// Array of keys, or comma separated
$ordersby = is_array( $this->query_vars['orderby'] )
? $this->query_vars['orderby']
: preg_split( '/[,\s]/', $this->query_vars['orderby'] );
$orderby_array = array();
$possible_ins = $this->get_columns( array( 'in' => true ), 'and', 'name' );
$sortables = $this->get_columns( array( 'sortable' => true ), 'and', 'name' );
// Loop through possible order by's
foreach ( $ordersby as $_key => $_value ) {
// Skip if empty
if ( empty( $_value ) ) {
continue;
}
// Key is numeric
if ( is_int( $_key ) ) {
$_orderby = $_value;
$_item = $order;
// Key is string
} else {
$_orderby = $_key;
$_item = $_value;
}
// Skip if not sortable
if ( ! in_array( $_value, $sortables, true ) ) {
continue;
}
// Parse orderby
$parsed = $this->parse_orderby( $_orderby );
// Skip if empty
if ( empty( $parsed ) ) {
continue;
}
// Set if __in
if ( in_array( $_orderby, $possible_ins, true ) ) {
$orderby_array[] = "{$parsed} {$order}";
continue;
}
// Append parsed orderby to array
$orderby_array[] = $parsed . ' ' . $this->parse_order( $_item );
}
// Only set if valid orderby
if ( ! empty( $orderby_array ) ) {
$orderby = implode( ', ', $orderby_array );
}
}
// Return parsed orderby
return $orderby;
}
/**
* Used internally to generate an SQL string for searching across multiple columns.
*
* @since 1.0.0
*
* @param string $string Search string.
* @param array $columns Columns to search.
* @return string Search SQL.
*/
protected function get_search_sql( $string = '', $columns = array() ) {
// Array or String
$like = ( false !== strpos( $string, '*' ) )
? '%' . implode( '%', array_map( array( $this->get_db(), 'esc_like' ), explode( '*', $string ) ) ) . '%'
: '%' . $this->get_db()->esc_like( $string ) . '%';
// Default array
$searches = array();
// Build search SQL
foreach ( $columns as $column ) {
$searches[] = $this->get_db()->prepare( "{$column} LIKE %s", $like );
}
// Return the clause
return '(' . implode( ' OR ', $searches ) . ')';
}
/** Private Parsers *******************************************************/
/**
* Parses arguments passed to the item query with default query parameters.
*
* @since 1.0.0
*
* @see Query::__construct()
*
* @param string|array $query Array or string of Query arguments.
*/
private function parse_query( $query = array() ) {
// Setup the query_vars_original var
$this->query_var_originals = wp_parse_args( $query );
// Setup the query_vars parsed var
$this->query_vars = wp_parse_args(
$this->query_var_originals,
$this->query_var_defaults
);
/**
* Fires after the item query vars have been parsed.
*
* @since 1.0.0
*
* @param Query &$this The Query instance (passed by reference).
*/
do_action_ref_array( $this->apply_prefix( "parse_{$this->item_name_plural}_query" ), array( &$this ) );
}
/**
* Parse the where clauses for all known columns
*
* @since 1.0.0
*/
private function parse_where() {
// Defaults
$where = $join = $searchable = $date_query = array();
// Loop through columns
foreach ( $this->columns as $column ) {
// Maybe add name to searchable array
if ( true === $column->searchable ) {
$searchable[] = $column->name;
}
// Literal column comparison
if ( ! $this->is_query_var_default( $column->name ) ) {
// Array (unprepared)
if ( is_array( $this->query_vars[ $column->name ] ) ) {
$where_id = "'" . implode( "', '", $this->get_db()->_escape( $this->query_vars[ $column->name ] ) ) . "'";
$statement = "{$this->table_alias}.{$column->name} IN ({$where_id})";
// Add to where array
$where[ $column->name ] = $statement;
// Numeric/String/Float (prepared)
} else {
$pattern = $this->get_column_field( array( 'name' => $column->name ), 'pattern', '%s' );
$where_id = $this->query_vars[ $column->name ];
$statement = "{$this->table_alias}.{$column->name} = {$pattern}";
// Add to where array
$where[ $column->name ] = $this->get_db()->prepare( $statement, $where_id );
}
}
// __in
if ( true === $column->in ) {
$where_id = "{$column->name}__in";
// Parse item for an IN clause.
if ( isset( $this->query_vars[ $where_id ] ) && is_array( $this->query_vars[ $where_id ] ) ) {
// Convert single item arrays to literal column comparisons
if ( 1 === count( $this->query_vars[ $where_id ] ) ) {
$column_value = reset( $this->query_vars[ $where_id ] );
$statement = "{$this->table_alias}.{$column->name} = %s";
$where[ $column->name ] = $this->get_db()->prepare( $statement, $column_value );
// Implode
} else {
$where[ $where_id ] = "{$this->table_alias}.{$column->name} IN ( '" . implode( "', '", $this->get_db()->_escape( $this->query_vars[ $where_id ] ) ) . "' )";
}
}
}
// __not_in
if ( true === $column->not_in ) {
$where_id = "{$column->name}__not_in";
// Parse item for a NOT IN clause.
if ( isset( $this->query_vars[ $where_id ] ) && is_array( $this->query_vars[ $where_id ] ) ) {
// Convert single item arrays to literal column comparisons
if ( 1 === count( $this->query_vars[ $where_id ] ) ) {
$column_value = reset( $this->query_vars[ $where_id ] );
$statement = "{$this->table_alias}.{$column->name} != %s";
$where[ $column->name ] = $this->get_db()->prepare( $statement, $column_value );
// Implode
} else {
$where[ $where_id ] = "{$this->table_alias}.{$column->name} NOT IN ( '" . implode( "', '", $this->get_db()->_escape( $this->query_vars[ $where_id ] ) ) . "' )";
}
}
}
// date_query
if ( true === $column->date_query ) {
$where_id = "{$column->name}_query";
$column_date = $this->query_vars[ $where_id ];
// Parse item
if ( ! empty( $column_date ) ) {
// Default arguments
$defaults = array(
'column' => "{$this->table_alias}.{$column->name}",
'before' => $column_date,
'inclusive' => true
);
// Default date query
if ( is_string( $column_date ) ) {
$date_query[] = $defaults;
// Array query var
} elseif ( is_array( $column_date ) ) {
// Auto-fill column if empty
if ( empty( $column_date['column'] ) ) {
$column_date['column'] = $defaults['column'];
}
// Add clause to date query
$date_query[] = $column_date;
}
}
}
}
// Maybe search if columns are searchable.
if ( ! empty( $searchable ) && strlen( $this->query_vars['search'] ) ) {
$search_columns = array();
// Intersect against known searchable columns
if ( ! empty( $this->query_vars['search_columns'] ) ) {
$search_columns = array_intersect(
$this->query_vars['search_columns'],
$searchable
);
}
// Default to all searchable columns
if ( empty( $search_columns ) ) {
$search_columns = $searchable;
}
/**
* Filters the columns to search in a Query search.
*
* @since 1.0.0
*
* @param array $search_columns Array of column names to be searched.
* @param string $search Text being searched.
* @param object $this The current Query instance.
*/
$search_columns = (array) apply_filters( $this->apply_prefix( "{$this->item_name_plural}_search_columns" ), $search_columns, $this->query_vars['search'], $this );
// Add search query clause
$where['search'] = $this->get_search_sql( $this->query_vars['search'], $search_columns );
}
// Get the primary column & table
$primary = $this->get_primary_column_name();
$table = $this->get_meta_type();
$and = '/^\s*AND\s*/';
// Maybe perform a meta query.
$meta_query = $this->query_vars['meta_query'];
if ( ! empty( $meta_query ) && is_array( $meta_query ) ) {
$this->meta_query = $this->get_meta_query( $meta_query );
$clauses = $this->meta_query->get_sql( $table, $this->table_alias, $primary, $this );
// Not all objects have meta, so make sure this one exists
if ( false !== $clauses ) {
// Set join
if ( ! empty( $clauses['join'] ) ) {
$join['meta_query'] = $clauses['join'];
}
// Remove " AND " from meta_query query where clause
if ( ! empty( $clauses['where'] ) ) {
$where['meta_query'] = preg_replace( $and, '', $clauses['where'] );
}
}
}
// Maybe perform a compare query.
$compare_query = $this->query_vars['compare_query'];
if ( ! empty( $compare_query ) && is_array( $compare_query ) ) {
$this->compare_query = $this->get_compare_query( $compare_query );
$clauses = $this->compare_query->get_sql( $table, $this->table_alias, $primary, $this );
// Not all objects can compare, so make sure this one exists
if ( false !== $clauses ) {
// Set join
if ( ! empty( $clauses['join'] ) ) {
$join['compare_query'] = $clauses['join'];
}
// Remove " AND " from query where clause.
$where['compare_query'] = preg_replace( $and, '', $clauses['where'] );
}
}
// Only do a date query with an array
$date_query = ! empty( $date_query )
? $date_query
: $this->query_vars['date_query'];
// Maybe perform a date query
if ( ! empty( $date_query ) && is_array( $date_query ) ) {
$this->date_query = $this->get_date_query( $date_query );
$clauses = $this->date_query->get_sql( $this->table_name, $this->table_alias, $primary, $this );
// Not all objects are dates, so make sure this one exists
if ( false !== $clauses ) {
// Set join
if ( ! empty( $clauses['join'] ) ) {
$join['date_query'] = $clauses['join'];
}
// Remove " AND " from query where clause.
$where['date_query'] = preg_replace( $and, '', $clauses['where'] );
}
}
// Set where and join clauses
$this->query_clauses['where'] = $where;
$this->query_clauses['join'] = $join;
}
/**
* Parse which fields to query for
*
* @since 1.0.0
*
* @param string $fields
* @param bool $alias
* @return string
*/
private function parse_fields( $fields = '', $alias = true ) {
// Default return value
$primary = $this->get_primary_column_name();
$retval = ( true === $alias )
? "{$this->table_alias}.{$primary}"
: $primary;
// No fields
if ( empty( $fields ) && ! empty( $this->query_vars['count'] ) ) {
// Possible fields to group by
$groupby_names = $this->parse_groupby( $this->query_vars['groupby'], $alias );
$groupby_names = ! empty( $groupby_names )
? "{$groupby_names}"
: '';
// Group by or total count
$retval = ! empty( $groupby_names )
? "{$groupby_names}, COUNT(*) as count"
: 'COUNT(*)';
}
// Return fields (or COUNT)
return $retval;
}
/**
* Parses and sanitizes the 'groupby' keys passed into the item query
*
* @since 1.0.0
*
* @param string $groupby
* @param bool $alias
* @return string
*/
private function parse_groupby( $groupby = '', $alias = true ) {
// Bail if empty
if ( empty( $groupby ) ) {
return '';
}
// Sanitize groupby columns
$groupby = (array) array_map( 'sanitize_key', (array) $groupby );
// Re'flip column names back around
$columns = array_flip( $this->get_column_names() );
// Get the intersection of allowed column names to groupby columns
$intersect = array_intersect( $columns, $groupby );
// Bail if invalid column
if ( empty( $intersect ) ) {
return '';
}
// Default return value
$retval = array();
// Maybe prepend table alias to key
foreach ( $intersect as $key ) {
$retval[] = ( true === $alias )
? "{$this->table_alias}.{$key}"
: $key;
}
// Separate sanitized columns
return implode( ',', array_values( $retval ) );
}
/**
* Parses and sanitizes 'orderby' keys passed to the item query.
*
* @since 1.0.0
*
* @param string $orderby Field for the items to be ordered by.
* @return string|false Value to used in the ORDER clause. False otherwise.
*/
private function parse_orderby( $orderby = 'id' ) {
// Default value
$parsed = "{$this->table_alias}.{$this->get_primary_column_name()}";
// __in
if ( false !== strstr( $orderby, '__in' ) ) {
$column_name = str_replace( '__in', '', $orderby );
$column = $this->get_column_by( array( 'name' => $column_name ) );
$item_in = $column->is_numeric()
? implode( ',', array_map( 'absint', $this->query_vars[ $orderby ] ) )
: implode( ',', $this->query_vars[ $orderby ] );
$parsed = "FIELD( {$this->table_alias}.{$column->name}, {$item_in} )";
// Specific column
} else {
// Orderby is a literal, sortable column name
$sortables = $this->get_columns( array( 'sortable' => true ), 'and', 'name' );
if ( in_array( $orderby, $sortables, true ) ) {
$parsed = "{$this->table_alias}.{$orderby}";
}
}
// Return parsed value
return $parsed;
}
/**
* Parses an 'order' query variable and cast it to 'ASC' or 'DESC' as necessary.
*
* @since 1.0.0
*
* @param string $order The 'order' query variable.
* @return string The sanitized 'order' query variable.
*/
private function parse_order( $order = '' ) {
// Bail if malformed
if ( empty( $order ) || ! is_string( $order ) ) {
return 'DESC';
}
// Ascending or Descending
return ( 'ASC' === strtoupper( $order ) )
? 'ASC'
: 'DESC';
}
/** Private Shapers *******************************************************/
/**
* Shape items into their most relevant objects.
*
* This will try to use item_shape, but will fallback to a private
* method for querying and caching items.
*
* If using the `fields` parameter, results will have unique shapes based on
* exactly what was requested.
*
* @since 1.0.0
*
* @param array $items
* @return array
*/
private function shape_items( $items = array() ) {
// Force to stdClass if querying for fields
if ( ! empty( $this->query_vars['fields'] ) ) {
$this->item_shape = 'stdClass';
}
// Default return value
$retval = array();
// Use foreach because it's faster than array_map()
if ( ! empty( $items ) ) {
foreach ( $items as $item ) {
$retval[] = $this->get_item( $item );
}
}
/**
* Filters the object query results.
*
* Looks like `edd_get_customers`
*
* @since 1.0.0
*
* @param array $retval An array of items.
* @param object &$this Current instance of Query, passed by reference.
*/
$retval = (array) apply_filters_ref_array( $this->apply_prefix( "the_{$this->item_name_plural}" ), array( $retval, &$this ) );
// Return filtered results
return ! empty( $this->query_vars['fields'] )
? $this->get_item_fields( $retval )
: $retval;
}
/**
* Get specific item fields based on query_vars['fields'].
*
* @since 1.0.0
*
* @param array $items
* @return array
*/
private function get_item_fields( $items = array() ) {
// Get the primary column
$primary = $this->get_primary_column_name();
$fields = $this->query_vars['fields'];
// Strings need to be single columns
if ( is_string( $fields ) ) {
$field = sanitize_key( $fields );
$items = ( 'ids' === $fields )
? wp_list_pluck( $items, $primary )
: wp_list_pluck( $items, $field, $primary );
// Arrays could be anything
} elseif ( is_array( $fields ) ) {
$new_items = array();
$fields = array_flip( $fields );
// Loop through items and pluck out the fields
foreach ( $items as $item_id => $item ) {
$new_items[ $item_id ] = (object) array_intersect_key( (array) $item, $fields );
}
// Set the items and unset the new items
$items = $new_items;
unset( $new_items );
}
// Return the item, possibly reduced
return $items;
}
/**
* Shape an item ID from an object, array, or numeric value
*
* @since 1.0.0
*
* @param mixed $item
* @return int
*/
private function shape_item_id( $item = 0 ) {
// Default return value
$retval = 0;
// Get the primary column name
$primary = $this->get_primary_column_name();
// Numeric item ID
if ( is_numeric( $item ) ) {
$retval = $item;
// Object item
} elseif ( is_object( $item ) && isset( $item->{$primary} ) ) {
$retval = $item->{$primary};
// Array item
} elseif ( is_array( $item ) && isset( $item[ $primary ] ) ) {
$retval = $item[ $primary ];
}
// Return the item ID
return absint( $retval );
}
/** Queries ***************************************************************/
/**
* Get a single database row by the primary column ID, possibly from cache.
*
* Accepts an integer, object, or array, and attempts to get the ID from it,
* then attempts to retrieve that item fresh from the database or cache.
*
* @since 1.0.0
*
* @param int|array|object $item_id The ID of the item
* @return object|false False if empty/error, Object if successful
*/
public function get_item( $item_id = 0 ) {
// Shape the item ID
$item_id = $this->shape_item_id( $item_id );
// Bail if no item to get by
if ( empty( $item_id ) ) {
return false;
}
// Get the primary column name
$column_name = $this->get_primary_column_name();
// Get item by ID
return $this->get_item_by( $column_name, $item_id );
}
/**
* Get a single database row by any column and value, possibly from cache.
*
* Take care to only use this method on columns with unique values,
* preferably with a cache group for that column. See: get_item().
*
* @since 1.0.0
*
* @param string $column_name Name of database column
* @param int|string $column_value Value to query for
* @return object|false False if empty/error, Object if successful
*/
public function get_item_by( $column_name = '', $column_value = '' ) {
// Default return value
$retval = false;
// Bail if no key or value
if ( empty( $column_name ) || empty( $column_value ) ) {
return $retval;
}
// Bail if name is not a string
if ( ! is_string( $column_name ) ) {
return $retval;
}
// Bail if value is not scalar (null values also not allowed)
if ( ! is_scalar( $column_value ) ) {
return $retval;
}
// Get column names
$columns = $this->get_column_names();
// Bail if column does not exist
if ( ! isset( $columns[ $column_name ] ) ) {
return $retval;
}
// Cache groups
$groups = $this->get_cache_groups();
// Check cache
if ( ! empty( $groups[ $column_name ] ) ) {
$retval = $this->cache_get( $column_value, $groups[ $column_name ] );
}
// Item not cached
if ( false === $retval ) {
// Try to get item directly from DB
$retval = $this->get_item_raw( $column_name, $column_value );
// Bail on failure
if ( ! $this->is_success( $retval ) ) {
return false;
}
// Cache
$this->update_item_cache( $retval );
}
// Reduce the item
$retval = $this->reduce_item( 'select', $retval );
// Return result
return $this->shape_item( $retval );
}
/**
* Add an item to the database
*
* @since 1.0.0
*
* @param array $data
* @return false|int Returns the item ID on success; false on failure.
*/
public function add_item( $data = array() ) {
// Get primary column
$primary = $this->get_primary_column_name();
// If data includes primary column, check if item already exists
if ( ! empty( $data[ $primary ] ) ) {
// Shape the primary item ID
$item_id = $this->shape_item_id( $data[ $primary ] );
// Get item by ID (from database, not cache)
$item = $this->get_item_raw( $primary, $item_id );
// Bail if item already exists
if ( ! empty( $item ) ) {
return false;
}
// Set data primary ID to newly shaped ID
$data[ $primary ] = $item_id;
}
// Get default values for item (from columns)
$item = $this->default_item();
// Unset the primary key if not part of data array (auto-incremented)
if ( empty( $data[ $primary ] ) ) {
unset( $item[ $primary ] );
}
// Cut out non-keys for meta
$columns = $this->get_column_names();
$data = array_merge( $item, $data );
$meta = array_diff_key( $data, $columns );
$save = array_intersect_key( $data, $columns );
// Get the current time (maybe used by created/modified)
$time = $this->get_current_time();
// If date-created exists, but is empty or default, use the current time
$created = $this->get_column_by( array( 'created' => true ) );
if ( ! empty( $created ) && ( empty( $save[ $created->name ] ) || ( $save[ $created->name ] === $created->default ) ) ) {
$save[ $created->name ] = $time;
}
// If date-modified exists, but is empty or default, use the current time
$modified = $this->get_column_by( array( 'modified' => true ) );
if ( ! empty( $modified ) && ( empty( $save[ $modified->name ] ) || ( $save[ $modified->name ] === $modified->default ) ) ) {
$save[ $modified->name ] = $time;
}
// Try to add
$table = $this->get_table_name();
$reduce = $this->reduce_item( 'insert', $save );
$save = $this->validate_item( $reduce );
$result = ! empty( $save )
? $this->get_db()->insert( $table, $save )
: false;
// Bail on failure
if ( ! $this->is_success( $result ) ) {
return false;
}
// Get the new item ID
$item_id = $this->get_db()->insert_id;
// Maybe save meta keys
if ( ! empty( $meta ) ) {
$this->save_extra_item_meta( $item_id, $meta );
}
// Use get item to prime caches
$this->update_item_cache( $item_id );
// Transition item data
$this->transition_item( $save, $item_id );
// Return result
return $item_id;
}
/**
* Update an item in the database
*
* @since 1.0.0
*
* @param int $item_id
* @param array $data
* @return bool
*/
public function update_item( $item_id = 0, $data = array() ) {
// Bail if no item ID
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) ) {
return false;
}
// Get primary column
$primary = $this->get_primary_column_name();
// Get item to update (from database, not cache)
$item = $this->get_item_raw( $primary, $item_id );
// Bail if item does not exist to update
if ( empty( $item ) ) {
return false;
}
// Cast as an array for easier manipulation
$item = (array) $item;
// Unset the primary key from data to parse
unset( $data[ $primary ] );
// Splice new data into item, and cut out non-keys for meta
$columns = $this->get_column_names();
$data = array_merge( $item, $data );
$meta = array_diff_key( $data, $columns );
$save = array_intersect_key( $data, $columns );
// Maybe save meta keys
if ( ! empty( $meta ) ) {
$this->save_extra_item_meta( $item_id, $meta );
}
// Bail if no change
if ( (array) $save === (array) $item ) {
return true;
}
// Unset the primary key from data to save
unset( $save[ $primary ] );
// If date-modified is empty, use the current time
$modified = $this->get_column_by( array( 'modified' => true ) );
if ( ! empty( $modified ) ) {
$save[ $modified->name ] = $this->get_current_time();
}
// Try to update
$where = array( $primary => $item_id );
$table = $this->get_table_name();
$reduce = $this->reduce_item( 'update', $save );
$save = $this->validate_item( $reduce );
$result = ! empty( $save )
? $this->get_db()->update( $table, $save, $where )
: false;
// Bail on failure
if ( ! $this->is_success( $result ) ) {
return false;
}
// Use get item to prime caches
$this->update_item_cache( $item_id );
// Transition item data
$this->transition_item( $save, $item );
// Return result
return $result;
}
/**
* Delete an item from the database
*
* @since 1.0.0
*
* @param int $item_id
* @return bool
*/
public function delete_item( $item_id = 0 ) {
// Bail if no item ID
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) ) {
return false;
}
// Get vars
$primary = $this->get_primary_column_name();
// Get item (before it's deleted)
$item = $this->get_item_raw( $primary, $item_id );
// Bail if item does not exist to delete
if ( empty( $item ) ) {
return false;
}
// Attempt to reduce this item
$item = $this->reduce_item( 'delete', $item );
// Bail if item was reduced to nothing
if ( empty( $item ) ) {
return false;
}
// Try to delete
$table = $this->get_table_name();
$where = array( $primary => $item_id );
$result = $this->get_db()->delete( $table, $where );
// Bail on failure
if ( ! $this->is_success( $result ) ) {
return false;
}
// Clean caches on successful delete
$this->delete_all_item_meta( $item_id );
$this->clean_item_cache( $item );
// Return result
return $result;
}
/**
* Filter an item before it is inserted of updated in the database.
*
* This method is public to allow subclasses to perform JIT manipulation
* of the parameters passed into it.
*
* @since 1.0.0
*
* @param array $item
* @return array
*/
public function filter_item( $item = array() ) {
return (array) apply_filters_ref_array( $this->apply_prefix( "filter_{$this->item_name}_item" ), array( $item, &$this ) );
}
/**
* Shape an item from the database into the type of object it always wanted
* to be when it grew up.
*
* @since 1.0.0
*
* @param mixed ID of item, or row from database
* @return mixed False on error, Object of single-object class type on success
*/
private function shape_item( $item = 0 ) {
// Get the item from an ID
if ( is_numeric( $item ) ) {
$item = $this->get_item( $item );
}
// Return the item if it's already shaped
if ( $item instanceof $this->item_shape ) {
return $item;
}
// Shape the item as needed
$item = ! empty( $this->item_shape )
? new $this->item_shape( $item )
: (object) $item;
// Return the item object
return $item;
}
/**
* Validate an item before it is updated in or added to the database.
*
* @since 1.0.0
*
* @param array $item
* @return array|false False on error, Array of validated values on success
*/
private function validate_item( $item = array() ) {
// Bail if item is empty or not an array
if ( empty( $item ) || ! is_array( $item ) ) {
return $item;
}
// Loop through item attributes
foreach ( $item as $key => $value ) {
// Strip slashes from all strings
if ( is_string( $value ) ) {
$value = stripslashes( $value );
}
// Get column
$column = $this->get_column_by( array( 'name' => $key ) );
// Null value is special for all item keys
if ( is_null( $value ) ) {
// Bail if null is not allowed
if ( false === $column->allow_null ) {
return false;
}
// Attempt to validate
} elseif ( ! empty( $column->validate ) && is_callable( $column->validate ) ) {
$validated = call_user_func( $column->validate, $value );
// Bail if error
if ( is_wp_error( $validated ) ) {
return false;
}
// Update the value
$item[ $key ] = $validated;
/**
* Fallback to using the raw value.
*
* Note: This may change at a later date, so do not rely on this.
* Please always validate all data.
*/
} else {
$item[ $key ] = $value;
}
}
// Return the validated item
return $this->filter_item( $item );
}
/**
* Reduce an item down to the keys and values the current user has the
* appropriate capabilities to select|insert|update|delete.
*
* Note that internally, this method works with both arrays and objects of
* any type, and also resets the key values. It looks weird, but is
* currently by design to protect the integrity of the return value.
*
* @since 1.0.0
*
* @param string $method select|insert|update|delete
* @param array|object $item Object|Array of keys/values to reduce
*
* @return mixed Object|Array without keys the current user does not have caps for
*/
private function reduce_item( $method = 'update', $item = array() ) {
// Bail if item is empty
if ( empty( $item ) ) {
return $item;
}
// Loop through item attributes
foreach ( $item as $key => $value ) {
// Get callback for column
$caps = $this->get_column_field( array( 'name' => $key ), 'caps' );
// Unset if not explicitly allowed
if ( empty( $caps[ $method ] ) || ! current_user_can( $caps[ $method ] ) ) {
if ( is_array( $item ) ) {
unset( $item[ $key ] );
} elseif ( is_object( $item ) ) {
$item->{$key} = null;
}
// Set if explicitly allowed
} elseif ( is_array( $item ) ) {
$item[ $key ] = $value;
} elseif ( is_object( $item ) ) {
$item->{$key} = $value;
}
}
// Return the reduced item
return $item;
}
/**
* Return an item comprised of all default values
*
* This is used by `add_item()` to populate known default values, to ensure
* new item data is always what we expect it to be.
*
* @since 1.0.0
*
* @return array
*/
private function default_item() {
// Default return value
$retval = array();
// Get column names and defaults
$names = $this->get_columns( array(), 'and', 'name' );
$defaults = $this->get_columns( array(), 'and', 'default' );
// Put together an item using default values
foreach ( $names as $key => $name ) {
$retval[ $name ] = $defaults[ $key ];
}
// Return
return $retval;
}
/**
* Transition an item when adding or updating.
*
* This method takes the data being saved, looks for any columns that are
* known to transition between values, and fires actions on them.
*
* @since 1.0.0
*
* @param array $item
* @return array
*/
private function transition_item( $new_data = array(), $old_data = array() ) {
// Look for transition columns
$columns = $this->get_columns( array( 'transition' => true ), 'and', 'name' );
// Bail if no columns to transition
if ( empty( $columns ) ) {
return;
}
// Get the item ID
$item_id = $this->shape_item_id( $old_data );
// Bail if item ID cannot be retrieved
if ( empty( $item_id ) ) {
return;
}
// If no old value(s), it's new
if ( ! is_array( $old_data ) ) {
$old_data = $new_data;
// Set all old values to "new"
foreach ( $old_data as $key => $value ) {
$value = 'new';
$old_data[ $key ] = $value;
}
}
// Compare
$keys = array_flip( $columns );
$new = array_intersect_key( $new_data, $keys );
$old = array_intersect_key( $old_data, $keys );
// Get the difference
$diff = array_diff( $new, $old );
// Bail if nothing is changing
if ( empty( $diff ) ) {
return;
}
// Do the actions
foreach ( $diff as $key => $value ) {
$old_value = $old_data[ $key ];
$new_value = $new_data[ $key ];
$key_action = $this->apply_prefix( "transition_{$this->item_name}_{$key}" );
/**
* Fires after an object value has transitioned.
*
* @since 1.0.0
*
* @param mixed $old_value The value being transitioned FROM.
* @param mixed $new_value The value being transitioned TO.
* @param int $item_id The ID of the item that is transitioning.
*/
do_action( $key_action, $old_value, $new_value, $item_id );
}
}
/** Meta ******************************************************************/
/**
* Add meta data to an item
*
* @since 1.0.0
*
* @param int $item_id
* @param string $meta_key
* @param string $meta_value
* @param string $unique
* @return int|false The meta ID on success, false on failure.
*/
protected function add_item_meta( $item_id = 0, $meta_key = '', $meta_value = '', $unique = false ) {
// Bail if no meta was returned
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) || empty( $meta_key ) ) {
return false;
}
// Bail if no meta table exists
if ( false === $this->get_meta_table_name() ) {
return false;
}
// Return results of get meta data
return add_metadata( $table, $item_id, $meta_key, $meta_value, $unique );
}
/**
* Get meta data for an item
*
* @since 1.0.0
*
* @param int $item_id
* @param string $meta_key
* @param bool $single
* @return mixed Single metadata value, or array of values
*/
protected function get_item_meta( $item_id = 0, $meta_key = '', $single = false ) {
// Bail if no meta was returned
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) || empty( $meta_key ) ) {
return false;
}
// Bail if no meta table exists
if ( false === $this->get_meta_table_name() ) {
return false;
}
// Get meta type
$meta_type = $this->get_meta_type();
// Return results of getting meta data
return get_metadata( $meta_type, $item_id, $meta_key, $single );
}
/**
* Update meta data for an item
*
* @since 1.0.0
*
* @param int $item_id
* @param string $meta_key
* @param string $meta_value
* @param string $prev_value
* @return bool True on successful update, false on failure.
*/
protected function update_item_meta( $item_id = 0, $meta_key = '', $meta_value = '', $prev_value = '' ) {
// Bail if no meta was returned
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) || empty( $meta_key ) ) {
return false;
}
// Bail if no meta table exists
if ( false === $this->get_meta_table_name() ) {
return false;
}
// Get meta type
$meta_type = $this->get_meta_type();
// Return results of updating meta data
return update_metadata( $meta_type, $item_id, $meta_key, $meta_value, $prev_value );
}
/**
* Delete meta data for an item
*
* @since 1.0.0
*
* @param int $item_id
* @param string $meta_key
* @param string $meta_value
* @param string $delete_all
* @return bool True on successful delete, false on failure.
*/
protected function delete_item_meta( $item_id = 0, $meta_key = '', $meta_value = '', $delete_all = false ) {
// Bail if no meta was returned
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) || empty( $meta_key ) ) {
return false;
}
// Bail if no meta table exists
if ( false === $this->get_meta_table_name() ) {
return false;
}
// Get meta type
$meta_type = $this->get_meta_type();
// Return results of deleting meta data
return delete_metadata( $meta_type, $item_id, $meta_key, $meta_value, $delete_all );
}
/**
* Get registered meta data keys
*
* @since 1.0.0
*
* @param string $object_subtype The sub-type of meta keys
*
* @return array
*/
private function get_registered_meta_keys( $object_subtype = '' ) {
// Get the object type
$object_type = $this->get_meta_type();
// Return the keys
return get_registered_meta_keys( $object_type, $object_subtype );
}
/**
* Maybe update meta values on item update/save
*
* @since 1.0.0
*
* @param array $meta
*/
private function save_extra_item_meta( $item_id = 0, $meta = array() ) {
// Bail if there is no bulk meta to save
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) || empty( $meta ) ) {
return;
}
// Bail if no meta table exists
if ( false === $this->get_meta_table_name() ) {
return;
}
// Only save registered keys
$keys = $this->get_registered_meta_keys();
$meta = array_intersect_key( $meta, $keys );
// Bail if no registered meta keys
if ( empty( $meta ) ) {
return;
}
// Save or delete meta data
foreach ( $meta as $key => $value ) {
! empty( $value )
? $this->update_item_meta( $item_id, $key, $value )
: $this->delete_item_meta( $item_id, $key );
}
}
/**
* Delete all meta data for an item
*
* @since 1.0.0
*
* @param int $item_id
*/
private function delete_all_item_meta( $item_id = 0 ) {
// Bail if no meta was returned
$item_id = $this->shape_item_id( $item_id );
if ( empty( $item_id ) ) {
return;
}
// Get the meta table name
$table = $this->get_meta_table_name();
// Bail if no meta table exists
if ( empty( $table ) ) {
return;
}
// Guess the item ID column for the meta table
$primary_id = $this->get_primary_column_name();
$item_id_column = $this->apply_prefix( "{$this->item_name}_{$primary_id}" );
// Get meta IDs
$query = "SELECT meta_id FROM {$table} WHERE {$item_id_column} = %d";
$prepared = $this->get_db()->prepare( $query, $item_id );
$meta_ids = $this->get_db()->get_col( $prepared );
// Bail if no meta IDs to delete
if ( empty( $meta_ids ) ) {
return;
}
// Get the meta type
$meta_type = $this->get_meta_type();
// Delete all meta data for this item ID
foreach ( $meta_ids as $mid ) {
delete_metadata_by_mid( $meta_type, $mid );
}
}
/**
* Get the meta table for this query
*
* Forked from WordPress\_get_meta_table() so it can be more accurately
* predicted in a future iteration and default to returning false.
*
* @since 1.0.0
*
* @return mixed Table name if exists, False if not
*/
private function get_meta_table_name() {
// Get the meta-type
$type = $this->get_meta_type();
// Append "meta" to end of meta-type
$table_name = "{$type}meta";
// Variable'ize the database interface, to use inside empty()
$db = $this->get_db();
// If not empty, return table name
if ( ! empty( $db->{$table_name} ) ) {
return $db->{$table_name};
}
// Default return false
return false;
}
/**
* Get the meta type for this query
*
* This method exists to reduce some duplication for now. Future iterations
* will likely use Column::relationships to
*
* @since 1.1.0
*
* @return string
*/
private function get_meta_type() {
return $this->apply_prefix( $this->item_name );
}
/** Cache *****************************************************************/
/**
* Get cache key from query_vars and query_var_defaults.
*
* @since 1.0.0
*
* @return string
*/
private function get_cache_key( $group = '' ) {
// Slice query vars
$slice = wp_array_slice_assoc( $this->query_vars, array_keys( $this->query_var_defaults ) );
// Unset `fields` so it does not effect the cache key
unset( $slice['fields'] );
// Setup key & last_changed
$key = md5( serialize( $slice ) );
$last_changed = $this->get_last_changed_cache( $group );
// Concatenate and return cache key
return "get_{$this->item_name_plural}:{$key}:{$last_changed}";
}
/**
* Get the cache group, or fallback to the primary one.
*
* @since 1.0.0
*
* @param string $group
* @return string
*/
private function get_cache_group( $group = '' ) {
// Get the primary column
$primary = $this->get_primary_column_name();
// Default return value
$retval = $this->cache_group;
// Only allow non-primary groups
if ( ! empty( $group ) && ( $group !== $primary ) ) {
$retval = $group;
}
// Return the group
return $retval;
}
/**
* Get array of which database columns have uniquely cached groups
*
* @since 1.0.0
*
* @return array
*/
private function get_cache_groups() {
// Return value
$cache_groups = array();
// Get cache groups
$groups = $this->get_columns( array( 'cache_key' => true ), 'and', 'name' );
if ( ! empty( $groups ) ) {
// Get the primary column
$primary = $this->get_primary_column_name();
// Setup return values
foreach ( $groups as $name ) {
if ( $primary !== $name ) {
$cache_groups[ $name ] = "{$this->cache_group}-by-{$name}";
} else {
$cache_groups[ $name ] = $this->cache_group;
}
}
}
// Return cache groups array
return $cache_groups;
}
/**
* Maybe prime item & item-meta caches by querying 1 time for all un-cached
* items.
*
* Accepts a single ID, or an array of IDs.
*
* The reason this accepts only IDs is because it gets called immediately
* after an item is inserted in the database, but before items have been
* "shaped" into proper objects, so object properties may not be set yet.
*
* @since 1.0.0
*
* @param array $item_ids
* @param bool $force
*
* @return bool False if empty
*/
private function prime_item_caches( $item_ids = array(), $force = false ) {
// Bail if no items to cache
if ( empty( $item_ids ) ) {
return false;
}
// Accepts single values, so cast to array
$item_ids = (array) $item_ids;
// Update item caches
if ( ! empty( $force ) || ! empty( $this->query_vars['update_item_cache'] ) ) {
// Look for non-cached IDs
$ids = $this->get_non_cached_ids( $item_ids, $this->cache_group );
// Bail if IDs are cached
if ( empty( $ids ) ) {
return false;
}
// Query
$table = $this->get_table_name();
$primary = $this->get_primary_column_name();
$query = "SELECT * FROM {$table} WHERE {$primary} IN (%s)";
$ids = join( ',', array_map( 'absint', $ids ) );
$prepare = sprintf( $query, $ids );
$results = $this->get_db()->get_results( $prepare );
// Update item caches
$this->update_item_cache( $results );
}
// Update meta data caches
if ( ! empty( $this->query_vars['update_meta_cache'] ) ) {
$singular = rtrim( $this->table_name, 's' ); // sic
update_meta_cache( $singular, $item_ids );
}
}
/**
* Update the cache for an item. Does not update item-meta cache.
*
* Accepts a single object, or an array of objects.
*
* The reason this does not accept ID's is because this gets called
* after an item is already updated in the database, so we want to avoid
* querying for it again. It's just safer this way.
*
* @since 1.0.0
*
* @param array $items
*/
private function update_item_cache( $items = array() ) {
// Maybe query for single item
if ( is_numeric( $items ) ) {
$primary = $this->get_primary_column_name();
$items = $this->get_item_raw( $primary, $items );
}
// Bail if no items to cache
if ( empty( $items ) ) {
return false;
}
// Make sure items are an array (without casting objects to arrays)
if ( ! is_array( $items ) ) {
$items = array( $items );
}
// Get cache groups
$groups = $this->get_cache_groups();
// Loop through all items and cache them
foreach ( $items as $item ) {
// Skip if item is not an object
if ( ! is_object( $item ) ) {
continue;
}
// Loop through groups and set cache
if ( ! empty( $groups ) ) {
foreach ( $groups as $key => $group ) {
$this->cache_set( $item->{$key}, $item, $group );
}
}
}
// Update last changed
$this->update_last_changed_cache();
}
/**
* Clean the cache for an item. Does not clean item-meta.
*
* Accepts a single object, or an array of objects.
*
* The reason this does not accept ID's is because this gets called
* after an item is already deleted from the database, so it cannot be
* queried and may not exist in the cache. It's just safer this way.
*
* @since 1.0.0
*
* @param mixed $items Single object item, or Array of object items
*
* @return bool
*/
private function clean_item_cache( $items = array() ) {
// Bail if no items to clean
if ( empty( $items ) ) {
return false;
}
// Make sure items are an array
if ( ! is_array( $items ) ) {
$items = array( $items );
}
// Get all cache groups
$groups = $this->get_cache_groups();
// Loop through all items and clean them
foreach ( $items as $item ) {
// Skip if item is not an object
if ( ! is_object( $item ) ) {
continue;
}
// Loop through groups and delete cache
if ( ! empty( $groups ) ) {
foreach ( $groups as $key => $group ) {
$this->cache_delete( $item->{$key}, $group );
}
}
}
// Update last changed
$this->update_last_changed_cache();
}
/**
* Update the last_changed key for the cache group
*
* @since 1.0.0
*/
private function update_last_changed_cache( $group = '' ) {
// Fallback to microtime
if ( empty( $this->last_changed ) ) {
$this->set_last_changed();
}
// Set the last changed time for this cache group
$this->cache_set( 'last_changed', $this->last_changed, $group );
// Return the last changed time
return $this->last_changed;
}
/**
* Get the last_changed key for a cache group
*
* @since 1.0.0
*
* @param string $group Cache group. Defaults to $this->cache_group
*
* @return int The last time a cache group was changed
*/
private function get_last_changed_cache( $group = '' ) {
// Get the last changed cache value
$last_changed = $this->cache_get( 'last_changed', $group );
// Maybe update the last changed value
if ( false === $last_changed ) {
$last_changed = $this->update_last_changed_cache( $group );
}
// Return the last changed value for the cache group
return $last_changed;
}
/**
* Get array of non-cached item IDs.
*
* @since 1.0.0
*
* @param array $item_ids Array of item IDs
* @param string $group Cache group. Defaults to $this->cache_group
*
* @return array
*/
private function get_non_cached_ids( $item_ids = array(), $group = '' ) {
$retval = array();
// Bail if no item IDs
if ( empty( $item_ids ) ) {
return $retval;
}
// Loop through item IDs
foreach ( $item_ids as $id ) {
$id = $this->shape_item_id( $id );
if ( false === $this->cache_get( $id, $group ) ) {
$retval[] = $id;
}
}
// Return array of IDs
return $retval;
}
/**
* Add a cache value for a key and group.
*
* @since 1.0.0
*
* @param string $key Cache key.
* @param mixed $value Cache value.
* @param string $group Cache group. Defaults to $this->cache_group
* @param int $expire Expiration.
*/
private function cache_add( $key = '', $value = '', $group = '', $expire = 0 ) {
// Bail if cache invalidation is suspended
if ( wp_suspend_cache_addition() ) {
return;
}
// Bail if no cache key
if ( empty( $key ) ) {
return;
}
// Get the cache group
$group = $this->get_cache_group( $group );
// Add to the cache
wp_cache_add( $key, $value, $group, $expire );
}
/**
* Get a cache value for a key and group.
*
* @since 1.0.0
*
* @param string $key Cache key.
* @param string $group Cache group. Defaults to $this->cache_group
* @param bool $force
*/
private function cache_get( $key = '', $group = '', $force = false ) {
// Bail if no cache key
if ( empty( $key ) ) {
return;
}
// Get the cache group
$group = $this->get_cache_group( $group );
// Get from the cache
return wp_cache_get( $key, $group, $force );
}
/**
* Set a cache value for a key and group.
*
* @since 1.0.0
*
* @param string $key Cache key.
* @param mixed $value Cache value.
* @param string $group Cache group. Defaults to $this->cache_group
* @param int $expire Expiration.
*/
private function cache_set( $key = '', $value = '', $group = '', $expire = 0 ) {
// Bail if cache invalidation is suspended
if ( wp_suspend_cache_addition() ) {
return;
}
// Bail if no cache key
if ( empty( $key ) ) {
return;
}
// Get the cache group
$group = $this->get_cache_group( $group );
// Update the cache
wp_cache_set( $key, $value, $group, $expire );
}
/**
* Delete a cache key for a group.
*
* @since 1.0.0
*
* @global bool $_wp_suspend_cache_invalidation
*
* @param string $key Cache key.
* @param string $group Cache group. Defaults to $this->cache_group
*/
private function cache_delete( $key = '', $group = '' ) {
global $_wp_suspend_cache_invalidation;
// Bail if cache invalidation is suspended
if ( ! empty( $_wp_suspend_cache_invalidation ) ) {
return;
}
// Bail if no cache key
if ( empty( $key ) ) {
return;
}
// Get the cache group
$group = $this->get_cache_group( $group );
// Delete the cache
wp_cache_delete( $key, $group );
}
/**
* Fetch raw results directly from the database.
*
* @since 1.0.0
*
* @param array $cols Columns for `SELECT`.
* @param array $where_cols Where clauses. Each key-value pair in the array
* represents a column and a comparison.
* @param int $limit Optional. LIMIT value. Default 25.
* @param null $offset Optional. OFFSET value. Default null.
* @param string $output Optional. Any of ARRAY_A | ARRAY_N | OBJECT | OBJECT_K constants.
* Default OBJECT.
* With one of the first three, return an array of
* rows indexed from 0 by SQL result row number.
* Each row is an associative array (column => value, ...),
* a numerically indexed array (0 => value, ...),
* or an object. ( ->column = value ), respectively.
* With OBJECT_K, return an associative array of
* row objects keyed by the value of each row's
* first column's value.
*
* @return array|object|null Database query results.
*/
public function get_results( $cols = array(), $where_cols = array(), $limit = 25, $offset = null, $output = OBJECT ) {
// Bail if no columns have been passed.
if ( empty( $cols ) ) {
return null;
}
// Fetch all the columns for the table being queried.
$column_names = $this->get_column_names();
// Ensure valid column names have been passed for the `SELECT` clause.
foreach ( $cols as $index => $column ) {
if ( ! array_key_exists( $column, $column_names ) ) {
unset( $cols[ $index ] );
}
}
// Setup base SQL query.
$query = "SELECT ";
$query .= implode( ',', $cols );
$query .= " FROM {$this->get_table_name()} {$this->table_alias} ";
$query .= " WHERE 1=1 ";
// Ensure valid columns have been passed for the `WHERE` clause.
if ( ! empty( $where_cols ) ) {
// Get keys from where columns
$columns = array_keys( $where_cols );
// Loop through columns and unset any invalid names
foreach ( $columns as $index => $column ) {
if ( ! array_key_exists( $column, $column_names ) ) {
unset( $where_cols[ $index ] );
}
}
// Parse WHERE clauses.
foreach ( $where_cols as $column => $compare ) {
// Basic WHERE clause.
if ( ! is_array( $compare ) ) {
$pattern = $this->get_column_field( array( 'name' => $column ), 'pattern', '%s' );
$statement = " AND {$this->table_alias}.{$column} = {$pattern} ";
$query .= $this->get_db()->prepare( $statement, $compare );
// More complex WHERE clause.
} else {
$value = isset( $compare['value'] )
? $compare['value']
: false;
// Skip if a value was not provided.
if ( false === $value ) {
continue;
}
// Default compare clause to equals.
$compare_clause = isset( $compare['compare_query'] )
? trim( strtoupper( $compare['compare_query'] ) )
: '=';
// Array (unprepared)
if ( is_array( $compare['value'] ) ) {
// Default to IN if clause not specified.
if ( ! in_array( $compare_clause, array( 'IN', 'NOT IN', 'BETWEEN' ), true ) ) {
$compare_clause = 'IN';
}
// Parse & escape for IN and NOT IN.
if ( 'IN' === $compare_clause || 'NOT IN' === $compare_clause ) {
$value = "('" . implode( "','", $this->get_db()->_escape( $compare['value'] ) ) . "')";
// Parse & escape for BETWEEN.
} elseif ( is_array( $value ) && 2 === count( $value ) && 'BETWEEN' === $compare_clause ) {
$_this = $this->get_db()->_escape( $value[0] );
$_that = $this->get_db()->_escape( $value[1] );
$value = " {$_this} AND {$_that} ";
}
}
// Add WHERE clause.
$query .= " AND {$this->table_alias}.{$column} {$compare_clause} {$value} ";
}
}
}
// Maybe set an offset.
if ( ! empty( $offset ) ) {
$values = explode( ',', $offset );
$values = array_filter( $values, 'intval' );
$offset = implode( ',', $values );
$query .= " OFFSET {$offset} ";
}
// Maybe set a limit.
if ( ! empty( $limit ) && ( $limit > 0 ) ) {
$limit = intval( $limit );
$query .= " LIMIT {$limit} ";
}
// Execute query.
$results = $this->get_db()->get_results( $query, $output );
// Return results.
return $results;
}
}