/home/arranoyd/public_html/wp-content/plugins/wordpress-seo/lib/orm.php
<?php
/**
* Yoast ORM class.
*
* @package Yoast\WP\Lib
*/
namespace Yoast\WP\Lib;
use wpdb;
use Yoast\WP\SEO\Config\Migration_Status;
/**
*
* Based on Idiorm
*
* http://github.com/j4mie/idiorm/
*
* A single-class super-simple database abstraction layer for PHP.
* Provides (nearly) zero-configuration object-relational mapping
* and a fluent interface for building basic, commonly-used queries.
*
* BSD Licensed.
*
* Copyright (c) 2010, Jamie Matthews
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*
* The methods documented below are magic methods that conform to PSR-1.
* This documentation exposes these methods to doc generators and IDEs.
*
* @see http://www.php-fig.org/psr/psr-1/
*/
class ORM implements \ArrayAccess {
// ----------------------- //
// --- CLASS CONSTANTS --- //
// ----------------------- //
// WHERE and HAVING condition array keys
const CONDITION_FRAGMENT = 0;
const CONDITION_VALUES = 1;
// --------------------------- //
// --- INSTANCE PROPERTIES --- //
// --------------------------- //
/**
* The wrapped find_one and find_many classes will return an instance or
* instances of this class.
*
* @var string
*/
protected $class_name;
/**
* The name of the table the current ORM instance is associated with.
*
* @var string
*/
protected $_table_name;
/**
* Alias for the table to be used in SELECT queries
*
* @var string
*/
protected $_table_alias = null;
/**
* Values to be bound to the query
*
* @var array
*/
protected $_values = [];
/**
* Columns to select in the result
*
* @var array
*/
protected $_result_columns = [ '*' ];
/**
* Are we using the default result column or have these been manually changed?
*
* @var boolean
*/
protected $_using_default_result_columns = true;
/**
* Join sources
*
* @var array
*/
protected $_join_sources = [];
/**
* Should the query include a DISTINCT keyword?
*
* @var boolean
*/
protected $_distinct = false;
/**
* Is this a raw query?
*
* @var boolean
*/
protected $_is_raw_query = false;
/**
* The raw query
*
* @var string
*/
protected $_raw_query = '';
/**
* The raw query parameters
*
* @var array
*/
protected $_raw_parameters = [];
/**
* Array of WHERE clauses
*
* @var array
*/
protected $_where_conditions = [];
/**
* LIMIT
*
* @var int
*/
protected $_limit = null;
/**
* OFFSET
*
* @var int
*/
protected $_offset = null;
/**
* ORDER BY
*
* @var array
*/
protected $_order_by = [];
/**
* GROUP BY
*
* @var array
*/
protected $_group_by = [];
/**
* HAVING
*
* @var array
*/
protected $_having_conditions = [];
/**
* The data for a hydrated instance of the class
*
* @var array
*/
protected $_data = [];
/**
* lifetime of the object
*
* @var array
*/
protected $_dirty_fields = [];
/**
* Fields that are to be inserted in the DB raw
*
* @var array
*/
protected $_expr_fields = [];
/**
* Is this a new object (has create() been called)?
*
* @var boolean
*/
protected $_is_new = false;
/**
* Name of the column to use as the primary key for
* this instance only. Overrides the config settings.
*
* @var string
*/
protected $_instance_id_column = null;
// ---------------------- //
// --- STATIC METHODS --- //
// ---------------------- //
/**
* Factory method, return an instance of this class bound to the supplied
* table name.
*
* A repeat of content in parent::for_table, so that created class is ORM.
*
* @param string $table_name The table to create instance for.
*
* @return ORM Instance of the ORM.
*/
public static function for_table( $table_name ) {
return new static( $table_name, [] );
}
/**
* Executes a raw query as a wrapper for wpdb::query.
* Useful for queries that can't be accomplished through Idiorm,
* particularly those using engine-specific features.
*
* @param string $query The raw SQL query
* @param array $parameters Optional bound parameters
*
* @return bool Success
* @example raw_execute('SELECT `name`, AVG(`order`) FROM `customer` GROUP BY `name` HAVING AVG(`order`) > 10')
*
* @example raw_execute('INSERT OR REPLACE INTO `widget` (`id`, `name`) SELECT `id`, `name` FROM `other_table`')
*/
public static function raw_execute( $query, $parameters = [] ) {
return self::_execute( $query, $parameters );
}
/**
* Internal helper method for executing statements.
*
* @param string $query The query.
* @param array $parameters An array of parameters to be bound in to the query.
*
* @return bool|int Response of wpdb::query
*/
protected static function _execute( $query, $parameters = [] ) {
/**
* @var wpdb wpdb
*/
global $wpdb;
$show_errors = $wpdb->show_errors;
if ( YoastSEO()->classes->get( Migration_Status::class )->get_error( 'free' ) ) {
$wpdb->show_errors = false;
}
$parameters = \array_filter( $parameters, function( $parameter ) {
return $parameter !== null;
} );
if ( ! empty( $parameters ) ) {
$query = $wpdb->prepare( $query, $parameters );
}
$result = $wpdb->query( $query );
$wpdb->show_errors = $show_errors;
return $result;
}
// ------------------------ //
// --- INSTANCE METHODS --- //
// ------------------------ //
/**
* "Private" constructor; shouldn't be called directly.
* Use the ORM::for_table factory method instead.
*
* @param string $table_name Table name.
* @param array $data Data to populate table.
*/
protected function __construct( $table_name, $data = [] ) {
$this->_table_name = $table_name;
$this->_data = $data;
}
/**
* Set the name of the class which the wrapped methods should return
* instances of.
*
* @param string $class_name The classname to set.
*
* @return void
*/
public function set_class_name( $class_name ) {
$this->class_name = $class_name;
}
/**
* Create a new, empty instance of the class. Used
* to add a new row to your database. May optionally
* be passed an associative array of data to populate
* the instance. If so, all fields will be flagged as
* dirty so all will be saved to the database when
* save() is called.
*
* @param array|null $data Data to populate table.
*
* @return bool|Model|ORM
*/
public function create( $data = null ) {
$this->_is_new = true;
if ( ! \is_null( $data ) ) {
$this->hydrate( $data )->force_all_dirty();
}
return $this->create_model_instance( $this );
}
/**
* Specify the ID column to use for this instance or array of instances only.
* This overrides the id_column and id_column_overrides settings.
*
* This is mostly useful for libraries built on top of Idiorm, and will
* not normally be used in manually built queries. If you don't know why
* you would want to use this, you should probably just ignore it.
*
* @param string $id_column The ID column.
*
* @return ORM
*/
public function use_id_column( $id_column ) {
$this->_instance_id_column = $id_column;
return $this;
}
/**
* Create an ORM instance from the given row (an associative
* array of data fetched from the database)
*
* @param array $row A row from the database.
*
* @return bool|Model
*/
protected function _create_instance_from_row( $row ) {
$instance = self::for_table( $this->_table_name );
$instance->use_id_column( $this->_instance_id_column );
$instance->hydrate( $row );
return $this->create_model_instance( $instance );
}
/**
* Tell the ORM that you are expecting a single result
* back from your query, and execute it. Will return
* a single instance of the ORM class, or false if no
* rows were returned.
* As a shortcut, you may supply an ID as a parameter
* to this method. This will perform a primary key
* lookup on the table.
*
* @param null|int $id An (optional) ID.
*
* @return bool|Model
*/
public function find_one( $id = null ) {
if ( ! \is_null( $id ) ) {
$this->where_id_is( $id );
}
$this->limit( 1 );
$rows = $this->_run();
if ( empty( $rows ) ) {
return false;
}
return $this->_create_instance_from_row( $rows[0] );
}
/**
* Tell the ORM that you are expecting multiple results
* from your query, and execute it. Will return an array
* of instances of the ORM class, or an empty array if
* no rows were returned.
*
* @return array
*/
public function find_many() {
$rows = $this->_run();
if ( $rows === false ) {
return [];
}
return \array_map( [ $this, '_create_instance_from_row' ], $rows );
}
/**
* Method to create an instance of the model class associated with this
* wrapper and populate it with the supplied Idiorm instance.
*
* @param ORM $orm The ORM used by model.
*
* @return bool|Model Instance of the model class.
*/
protected function create_model_instance( $orm ) {
if ( $orm === false ) {
return false;
}
/**
* An instance of Model is being made.
*
* @var Model $model
*/
$model = new $this->class_name();
$model->set_orm( $orm );
return $model;
}
/**
* Tell the ORM that you are expecting multiple results
* from your query, and execute it. Will return an array,
* or an empty array if no rows were returned.
*
* @return array The query results.
*/
public function find_array() {
return $this->_run();
}
/**
* Tell the ORM that you wish to execute a COUNT query.
*
* @param string $column The table column.
*
* @return float|int An integer representing the number of rows returned.
*/
public function count( $column = '*' ) {
return $this->_call_aggregate_db_function( __FUNCTION__, $column );
}
/**
* Tell the ORM that you wish to execute a MAX query.
*
* @param string $column The table column.
*
* @return float|int The max value of the chosen column.
*/
public function max( $column ) {
return $this->_call_aggregate_db_function( __FUNCTION__, $column );
}
/**
* Tell the ORM that you wish to execute a MIN query.
*
* @param string $column The table column.
*
* @return float|int The min value of the chosen column.
*/
public function min( $column ) {
return $this->_call_aggregate_db_function( __FUNCTION__, $column );
}
/**
* Tell the ORM that you wish to execute a AVG query.
*
* @param string $column The table column.
*
* @return float|int The average value of the chosen column.
*/
public function avg( $column ) {
return $this->_call_aggregate_db_function( __FUNCTION__, $column );
}
/**
* Tell the ORM that you wish to execute a SUM query.
*
* @param string $column The table column.
*
* @return float|int The sum of the chosen column.
*/
public function sum( $column ) {
return $this->_call_aggregate_db_function( __FUNCTION__, $column );
}
/**
* Returns the select query as SQL.
*
* @return string The select query in SQL.
*/
public function get_sql() {
return $this->_build_select();
}
/**
* Returns the update query as SQL.
*
* @return string The update query in SQL.
*/
public function get_update_sql() {
return $this->_build_update();
}
/**
* Execute an aggregate query on the current connection.
*
* @param string $sql_function The aggregate function to call eg. MIN, COUNT, etc
* @param string $column The column to execute the aggregate query against
*
* @return int
*/
protected function _call_aggregate_db_function( $sql_function, $column ) {
$alias = \strtolower( $sql_function );
$sql_function = \strtoupper( $sql_function );
if ( '*' != $column ) {
$column = $this->_quote_identifier( $column );
}
$result_columns = $this->_result_columns;
$this->_result_columns = [];
$this->select_expr( "{$sql_function}({$column})", $alias );
$result = $this->find_one();
$this->_result_columns = $result_columns;
$return_value = 0;
if ( $result !== false && isset( $result->{$alias} ) ) {
if ( ! \is_numeric( $result->{$alias} ) ) {
$return_value = $result->{$alias};
} elseif ( (int) $result->{$alias} == (float) $result->{$alias} ) {
$return_value = (int) $result->{$alias};
} else {
$return_value = (float) $result->{$alias};
}
}
return $return_value;
}
/**
* This method can be called to hydrate (populate) this
* instance of the class from an associative array of data.
* This will usually be called only from inside the class,
* but it's public in case you need to call it directly.
*
* @param array $data Data to populate table.
*
* @return ORM
*/
public function hydrate( $data = [] ) {
$this->_data = $data;
return $this;
}
/**
* Force the ORM to flag all the fields in the $data array
* as "dirty" and therefore update them when save() is called.
*
* @return ORM
*/
public function force_all_dirty() {
$this->_dirty_fields = $this->_data;
return $this;
}
/**
* Perform a raw query. The query can contain placeholders in
* either named or question mark style. If placeholders are
* used, the parameters should be an array of values which will
* be bound to the placeholders in the query. If this method
* is called, all other query building methods will be ignored.
*
* @param array $query The query.
* @param array $parameters
*
* @return ORM
*/
public function raw_query( $query, $parameters = [] ) {
$this->_is_raw_query = true;
$this->_raw_query = $query;
$this->_raw_parameters = $parameters;
return $this;
}
/**
* Add an alias for the main table to be used in SELECT queries
*
* @param string $alias
*
* @return ORM
*/
public function table_alias( $alias ) {
$this->_table_alias = $alias;
return $this;
}
/**
* Internal method to add an unquoted expression to the set
* of columns returned by the SELECT query. The second optional
* argument is the alias to return the expression as.
*
* @param string $expr
* @param null|string $alias
*
* @return ORM
*/
protected function _add_result_column( $expr, $alias = null ) {
if ( ! \is_null( $alias ) ) {
$expr .= ' AS ' . $this->_quote_identifier( $alias );
}
if ( $this->_using_default_result_columns ) {
$this->_result_columns = [ $expr ];
$this->_using_default_result_columns = false;
} else {
$this->_result_columns[] = $expr;
}
return $this;
}
/**
* Counts the number of columns that belong to the primary
* key and their value is null.
*
* @return int
*
* @throws \Exception
*/
public function count_null_id_columns() {
if ( \is_array( $this->_get_id_column_name() ) ) {
return \count( \array_filter( $this->id(), 'is_null' ) );
} else {
return \is_null( $this->id() ) ? 1 : 0;
}
}
/**
* Add a column to the list of columns returned by the SELECT
* query. This defaults to '*'. The second optional argument is
* the alias to return the column as.
*
* @param string $column
* @param null|string $alias
*
* @return ORM
*/
public function select( $column, $alias = null ) {
$column = $this->_quote_identifier( $column );
return $this->_add_result_column( $column, $alias );
}
/**
* Add an unquoted expression to the list of columns returned
* by the SELECT query. The second optional argument is
* the alias to return the column as.
*
* @param string $expr
* @param null|string $alias
*
* @return ORM
*/
public function select_expr( $expr, $alias = null ) {
return $this->_add_result_column( $expr, $alias );
}
/**
* Add columns to the list of columns returned by the SELECT
* query. This defaults to '*'. Many columns can be supplied
* as either an array or as a list of parameters to the method.
*
* Note that the alias must not be numeric - if you want a
* numeric alias then prepend it with some alpha chars. eg. a1
*
* @example select_many(array('column', 'column2', 'column3'), 'column4', 'column5');
* @example select_many(array('alias' => 'column', 'column2', 'alias2' => 'column3'), 'column4', 'column5');
* @example select_many('column', 'column2', 'column3');
*
* @return ORM
*/
public function select_many() {
$columns = \func_get_args();
if ( ! empty( $columns ) ) {
$columns = $this->_normalise_select_many_columns( $columns );
foreach ( $columns as $alias => $column ) {
if ( \is_numeric( $alias ) ) {
$alias = null;
}
$this->select( $column, $alias );
}
}
return $this;
}
/**
* Add an unquoted expression to the list of columns returned
* by the SELECT query. Many columns can be supplied as either
* an array or as a list of parameters to the method.
*
* Note that the alias must not be numeric - if you want a
* numeric alias then prepend it with some alpha chars. eg. a1
*
* @return ORM
* @example select_many_expr(array('column', 'column2', 'column3'), 'column4', 'column5')
* @example select_many_expr(array('alias' => 'column', 'column2', 'alias2' => 'column3'), 'column4', 'column5')
*
* @example select_many_expr('column', 'column2', 'column3')
*/
public function select_many_expr() {
$columns = \func_get_args();
if ( ! empty( $columns ) ) {
$columns = $this->_normalise_select_many_columns( $columns );
foreach ( $columns as $alias => $column ) {
if ( \is_numeric( $alias ) ) {
$alias = null;
}
$this->select_expr( $column, $alias );
}
}
return $this;
}
/**
* Take a column specification for the select many methods and convert it
* into a normalised array of columns and aliases.
*
* It is designed to turn the following styles into a normalised array:
*
* array(array('alias' => 'column', 'column2', 'alias2' => 'column3'), 'column4', 'column5'))
*
* @param array $columns
*
* @return array
*/
protected function _normalise_select_many_columns( $columns ) {
$return = [];
foreach ( $columns as $column ) {
if ( \is_array( $column ) ) {
foreach ( $column as $key => $value ) {
if ( ! \is_numeric( $key ) ) {
$return[ $key ] = $value;
} else {
$return[] = $value;
}
}
} else {
$return[] = $column;
}
}
return $return;
}
/**
* Add a DISTINCT keyword before the list of columns in the SELECT query
*
* @return ORM
*/
public function distinct() {
$this->_distinct = true;
return $this;
}
/**
* Internal method to add a JOIN source to the query.
*
* The join_operator should be one of INNER, LEFT OUTER, CROSS etc - this
* will be prepended to JOIN.
*
* The table should be the name of the table to join to.
*
* The constraint may be either a string or an array with three elements. If it
* is a string, it will be compiled into the query as-is, with no escaping. The
* recommended way to supply the constraint is as an array with three elements:
*
* first_column, operator, second_column
*
* Example: array('user.id', '=', 'profile.user_id')
*
* will compile to
*
* ON `user`.`id` = `profile`.`user_id`
*
* The final (optional) argument specifies an alias for the joined table.
*
* @param string $join_operator
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
protected function _add_join_source( $join_operator, $table, $constraint, $table_alias = null ) {
$join_operator = \trim( "{$join_operator} JOIN" );
$table = $this->_quote_identifier( $table );
// Add table alias if present
if ( ! \is_null( $table_alias ) ) {
$table_alias = $this->_quote_identifier( $table_alias );
$table .= " {$table_alias}";
}
// Build the constraint
if ( \is_array( $constraint ) ) {
list( $first_column, $operator, $second_column ) = $constraint;
$first_column = $this->_quote_identifier( $first_column );
$second_column = $this->_quote_identifier( $second_column );
$constraint = "{$first_column} {$operator} {$second_column}";
}
$this->_join_sources[] = "{$join_operator} {$table} ON {$constraint}";
return $this;
}
/**
* Add a RAW JOIN source to the query
*
* @param string $table
* @param string $constraint
* @param string $table_alias
* @param array $parameters
*
* @return ORM
*/
public function raw_join( $table, $constraint, $table_alias, $parameters = [] ) {
// Add table alias if present
if ( ! \is_null( $table_alias ) ) {
$table_alias = $this->_quote_identifier( $table_alias );
$table .= " {$table_alias}";
}
$this->_values = \array_merge( $this->_values, $parameters );
// Build the constraint
if ( \is_array( $constraint ) ) {
list( $first_column, $operator, $second_column ) = $constraint;
$first_column = $this->_quote_identifier( $first_column );
$second_column = $this->_quote_identifier( $second_column );
$constraint = "{$first_column} {$operator} {$second_column}";
}
$this->_join_sources[] = "{$table} ON {$constraint}";
return $this;
}
/**
* Add a simple JOIN source to the query
*
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
public function join( $table, $constraint, $table_alias = null ) {
return $this->_add_join_source( '', $table, $constraint, $table_alias );
}
/**
* Add an INNER JOIN souce to the query
*
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
public function inner_join( $table, $constraint, $table_alias = null ) {
return $this->_add_join_source( 'INNER', $table, $constraint, $table_alias );
}
/**
* Add a LEFT OUTER JOIN souce to the query
*
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
public function left_outer_join( $table, $constraint, $table_alias = null ) {
return $this->_add_join_source( 'LEFT OUTER', $table, $constraint, $table_alias );
}
/**
* Add an RIGHT OUTER JOIN souce to the query
*
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
public function right_outer_join( $table, $constraint, $table_alias = null ) {
return $this->_add_join_source( 'RIGHT OUTER', $table, $constraint, $table_alias );
}
/**
* Add an FULL OUTER JOIN souce to the query
*
* @param string $table
* @param string $constraint
* @param null $table_alias
*
* @return ORM
*/
public function full_outer_join( $table, $constraint, $table_alias = null ) {
return $this->_add_join_source( 'FULL OUTER', $table, $constraint, $table_alias );
}
/**
* Internal method to add a HAVING condition to the query
*
* @param string $fragment
* @param array $values
*
* @return ORM
*/
protected function _add_having( $fragment, $values = [] ) {
return $this->_add_condition( 'having', $fragment, $values );
}
/**
* Internal method to add a HAVING condition to the query
*
* @param string $column_name The table column.
* @param string $separator
* @param $value
*
* @return ORM
*/
protected function _add_simple_having( $column_name, $separator, $value ) {
return $this->_add_simple_condition( 'having', $column_name, $separator, $value );
}
/**
* Internal method to add a HAVING clause with multiple values (like IN and NOT IN)
*
* @param string|array $column_name The table column.
* @param string $separator
* @param $values
*
* @return ORM
*/
public function _add_having_placeholder( $column_name, $separator, $values ) {
if ( ! \is_array( $column_name ) ) {
$data = [ $column_name => $values ];
} else {
$data = $column_name;
}
$result = $this;
foreach ( $data as $key => $val ) {
$column = $result->_quote_identifier( $key );
$placeholders = $result->_create_placeholders( $val );
$result = $result->_add_having( "{$column} {$separator} ({$placeholders})", $val );
}
return $result;
}
/**
* Internal method to add a HAVING clause with no parameters(like IS NULL and IS NOT NULL)
*
* @param string $column_name
* @param $operator
*
* @return ORM
*/
public function _add_having_no_value( $column_name, $operator ) {
$conditions = \is_array( $column_name ) ? $column_name : [ $column_name ];
$result = $this;
foreach ( $conditions as $column ) {
$column = $this->_quote_identifier( $column );
$result = $result->_add_having( "{$column} {$operator}" );
}
return $result;
}
/**
* Internal method to add a WHERE condition to the query
*
* @param string $fragment
* @param array $values
*
* @return ORM
*/
protected function _add_where( $fragment, $values = [] ) {
return $this->_add_condition( 'where', $fragment, $values );
}
/**
* Internal method to add a WHERE condition to the query
*
* @param string|array $column_name The table column
* @param string $separator
* @param $value
*
* @return ORM
*/
protected function _add_simple_where( $column_name, $separator, $value ) {
return $this->_add_simple_condition( 'where', $column_name, $separator, $value );
}
/**
* Add a WHERE clause with multiple values (like IN and NOT IN)
*
* @param string|array $column_name The table column
* @param string $separator
* @param $values
*
* @return ORM
*/
public function _add_where_placeholder( $column_name, $separator, $values ) {
if ( ! \is_array( $column_name ) ) {
$data = [ $column_name => $values ];
} else {
$data = $column_name;
}
$result = $this;
foreach ( $data as $key => $val ) {
$column = $result->_quote_identifier( $key );
$placeholders = $result->_create_placeholders( $val );
$result = $result->_add_where( "{$column} {$separator} ({$placeholders})", $val );
}
return $result;
}
/**
* Add a WHERE clause with no parameters(like IS NULL and IS NOT NULL)
*
* @param string $column_name
* @param $operator
*
* @return ORM
*/
public function _add_where_no_value( $column_name, $operator ) {
$conditions = \is_array( $column_name ) ? $column_name : [ $column_name ];
$result = $this;
foreach ( $conditions as $column ) {
$column = $this->_quote_identifier( $column );
$result = $result->_add_where( "{$column} {$operator}" );
}
return $result;
}
/**
* Internal method to add a HAVING or WHERE condition to the query
*
* @param $type
* @param string $fragment
* @param array $values
*
* @return ORM
*/
protected function _add_condition( $type, $fragment, $values = [] ) {
$conditions_class_property_name = "_{$type}_conditions";
if ( ! \is_array( $values ) ) {
$values = [ $values ];
}
\array_push( $this->{$conditions_class_property_name}, [
self::CONDITION_FRAGMENT => $fragment,
self::CONDITION_VALUES => $values,
] );
return $this;
}
/**
* Helper method to compile a simple COLUMN SEPARATOR VALUE
* style HAVING or WHERE condition into a string and value ready to
* be passed to the _add_condition method. Avoids duplication
* of the call to _quote_identifier
*
* If column_name is an associative array, it will add a condition for each column
*
* @param $type
* @param string|array $column_name The table column
* @param string $separator
* @param $value
*
* @return ORM
*/
protected function _add_simple_condition( $type, $column_name, $separator, $value ) {
$multiple = \is_array( $column_name ) ? $column_name : [ $column_name => $value ];
$result = $this;
foreach ( $multiple as $key => $val ) {
// Add the table name in case of ambiguous columns
if ( \count( $result->_join_sources ) > 0 && \strpos( $key, '.' ) === false ) {
$table = $result->_table_name;
if ( ! \is_null( $result->_table_alias ) ) {
$table = $result->_table_alias;
}
$key = "{$table}.{$key}";
}
$key = $result->_quote_identifier( $key );
$placeholder = ( $val === null ) ? 'NULL' : '%s';
$result = $result->_add_condition( $type, "{$key} {$separator} {$placeholder}", $val );
}
return $result;
}
/**
* Return a string containing the given number of question marks,
* separated by commas. Eg "?, ?, ?"
*
* @param array $fields Fields to create placeholder for.
*
* @return string
*/
protected function _create_placeholders( $fields ) {
if ( ! empty( $fields ) ) {
$db_fields = [];
foreach ( $fields as $key => $value ) {
// Process expression fields directly into the query
if ( \array_key_exists( $key, $this->_expr_fields ) ) {
$db_fields[] = $value;
} else {
$db_fields[] = ( $value === null ) ? 'NULL' : '%s';
}
}
return \implode( ', ', $db_fields );
}
return '';
}
/**
* Helper method that filters a column/value array returning only those
* columns that belong to a compound primary key.
*
* If the key contains a column that does not exist in the given array,
* a null value will be returned for it.
*
* @param $value
*
* @return array
*/
protected function _get_compound_id_column_values( $value ) {
$filtered = [];
foreach ( $this->_get_id_column_name() as $key ) {
$filtered[ $key ] = isset( $value[ $key ] ) ? $value[ $key ] : null;
}
return $filtered;
}
/**
* Helper method that filters an array containing compound column/value
* arrays.
*
* @param $values
*
* @return array
*/
protected function _get_compound_id_column_values_array( $values ) {
$filtered = [];
foreach ( $values as $value ) {
$filtered[] = $this->_get_compound_id_column_values( $value );
}
return $filtered;
}
/**
* Add a WHERE column = value clause to your query. Each time
* this is called in the chain, an additional WHERE will be
* added, and these will be ANDed together when the final query
* is built.
*
* If you use an array in $column_name, a new clause will be
* added for each element. In this case, $value is ignored.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where( $column_name, $value = null ) {
return $this->where_equal( $column_name, $value );
}
/**
* More explicitly named version of for the where() method.
* Can be used if preferred.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_equal( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '=', $value );
}
/**
* Add a WHERE column != value clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_not_equal( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '!=', $value );
}
/**
* Special method to query the table by its primary key
*
* If primary key is compound, only the columns that
* belong to they key will be used for the query
*
* @param $id
*
* @return ORM
*/
public function where_id_is( $id ) {
return \is_array( $this->_get_id_column_name() ) ? $this->where( $this->_get_compound_id_column_values( $id ), null ) : $this->where( $this->_get_id_column_name(), $id );
}
/**
* Allows adding a WHERE clause that matches any of the conditions
* specified in the array. Each element in the associative array will
* be a different condition, where the key will be the column name.
*
* By default, an equal operator will be used against all columns, but
* it can be overriden for any or every column using the second parameter.
*
* Each condition will be ORed together when added to the final query.
*
* @param $values
* @param string $operator
*
* @return ORM
*/
public function where_any_is( $values, $operator = '=' ) {
$data = [];
$query = [ '((' ];
$first = true;
foreach ( $values as $value ) {
if ( $first ) {
$first = false;
} else {
$query[] = ') OR (';
}
$firstsub = true;
foreach ( $value as $key => $item ) {
$op = \is_string( $operator ) ? $operator : ( isset( $operator[ $key ] ) ? $operator[ $key ] : '=' );
if ( $firstsub ) {
$firstsub = false;
} else {
$query[] = 'AND';
}
$query[] = $this->_quote_identifier( $key );
$data[] = $item;
$query[] = $op . ( ( $item === null ) ? 'NULL' : '%s' );
}
}
$query[] = '))';
return $this->where_raw( \join( $query, ' ' ), $data );
}
/**
* Similar to where_id_is() but allowing multiple primary keys.
*
* If primary key is compound, only the columns that
* belong to they key will be used for the query
*
* @param $ids
*
* @return ORM
*/
public function where_id_in( $ids ) {
return \is_array( $this->_get_id_column_name() ) ? $this->where_any_is( $this->_get_compound_id_column_values_array( $ids ) ) : $this->where_in( $this->_get_id_column_name(), $ids );
}
/**
* Add a WHERE ... LIKE clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_like( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, 'LIKE', $value );
}
/**
* Add where WHERE ... NOT LIKE clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_not_like( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, 'NOT LIKE', $value );
}
/**
* Add a WHERE ... > clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_gt( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '>', $value );
}
/**
* Add a WHERE ... < clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_lt( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '<', $value );
}
/**
* Add a WHERE ... >= clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_gte( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '>=', $value );
}
/**
* Add a WHERE ... <= clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function where_lte( $column_name, $value = null ) {
return $this->_add_simple_where( $column_name, '<=', $value );
}
/**
* Add a WHERE ... IN clause to your query
*
* @param string|array $column_name The table column
* @param $values
*
* @return ORM
*/
public function where_in( $column_name, $values ) {
return $this->_add_where_placeholder( $column_name, 'IN', $values );
}
/**
* Add a WHERE ... NOT IN clause to your query
*
* @param string|array $column_name The table column
* @param $values
*
* @return ORM
*/
public function where_not_in( $column_name, $values ) {
return $this->_add_where_placeholder( $column_name, 'NOT IN', $values );
}
/**
* Add a WHERE column IS NULL clause to your query
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function where_null( $column_name ) {
return $this->_add_where_no_value( $column_name, 'IS NULL' );
}
/**
* Add a WHERE column IS NOT NULL clause to your query
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function where_not_null( $column_name ) {
return $this->_add_where_no_value( $column_name, 'IS NOT NULL' );
}
/**
* Add a raw WHERE clause to the query. The clause should
* contain question mark placeholders, which will be bound
* to the parameters supplied in the second argument.
*
* @param $clause
* @param array $parameters
*
* @return ORM
*/
public function where_raw( $clause, $parameters = [] ) {
return $this->_add_where( $clause, $parameters );
}
/**
* Add a LIMIT to the query
*
* @param $limit
*
* @return ORM
*/
public function limit( $limit ) {
$this->_limit = $limit;
return $this;
}
/**
* Add an OFFSET to the query
*
* @param $offset
*
* @return ORM
*/
public function offset( $offset ) {
$this->_offset = $offset;
return $this;
}
/**
* Add an ORDER BY clause to the query
*
* @param string $column_name
* @param $ordering
*
* @return ORM
*/
protected function _add_order_by( $column_name, $ordering ) {
$column_name = $this->_quote_identifier( $column_name );
$this->_order_by[] = "{$column_name} {$ordering}";
return $this;
}
/**
* Add an ORDER BY column DESC clause
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function order_by_desc( $column_name ) {
return $this->_add_order_by( $column_name, 'DESC' );
}
/**
* Add an ORDER BY column ASC clause
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function order_by_asc( $column_name ) {
return $this->_add_order_by( $column_name, 'ASC' );
}
/**
* Add an unquoted expression as an ORDER BY clause
*
* @param $clause
*
* @return ORM
*/
public function order_by_expr( $clause ) {
$this->_order_by[] = $clause;
return $this;
}
/**
* Add a column to the list of columns to GROUP BY
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function group_by( $column_name ) {
$column_name = $this->_quote_identifier( $column_name );
$this->_group_by[] = $column_name;
return $this;
}
/**
* Add an unquoted expression to the list of columns to GROUP BY
*
* @param string $expr
*
* @return ORM
*/
public function group_by_expr( $expr ) {
$this->_group_by[] = $expr;
return $this;
}
/**
* Add a HAVING column = value clause to your query. Each time
* this is called in the chain, an additional HAVING will be
* added, and these will be ANDed together when the final query
* is built.
*
* If you use an array in $column_name, a new clause will be
* added for each element. In this case, $value is ignored.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having( $column_name, $value = null ) {
return $this->having_equal( $column_name, $value );
}
/**
* More explicitly named version of for the having() method.
* Can be used if preferred.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_equal( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '=', $value );
}
/**
* Add a HAVING column != value clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_not_equal( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '!=', $value );
}
/**
* Special method to query the table by its primary key.
*
* If primary key is compound, only the columns that
* belong to they key will be used for the query
*
* @param $id
*
* @return ORM
*/
public function having_id_is( $id ) {
return \is_array( $this->_get_id_column_name() ) ? $this->having( $this->_get_compound_id_column_values( $id ), null ) : $this->having( $this->_get_id_column_name(), $id );
}
/**
* Add a HAVING ... LIKE clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_like( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, 'LIKE', $value );
}
/**
* Add where HAVING ... NOT LIKE clause to your query.
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_not_like( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, 'NOT LIKE', $value );
}
/**
* Add a HAVING ... > clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_gt( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '>', $value );
}
/**
* Add a HAVING ... < clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_lt( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '<', $value );
}
/**
* Add a HAVING ... >= clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_gte( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '>=', $value );
}
/**
* Add a HAVING ... <= clause to your query
*
* @param string|array $column_name The table column
* @param null $value
*
* @return ORM
*/
public function having_lte( $column_name, $value = null ) {
return $this->_add_simple_having( $column_name, '<=', $value );
}
/**
* Add a HAVING ... IN clause to your query
*
* @param string|array $column_name The table column
* @param null $values
*
* @return ORM
*/
public function having_in( $column_name, $values = null ) {
return $this->_add_having_placeholder( $column_name, 'IN', $values );
}
/**
* Add a HAVING ... NOT IN clause to your query
*
* @param string|array $column_name The table column
* @param null $values
*
* @return ORM
*/
public function having_not_in( $column_name, $values = null ) {
return $this->_add_having_placeholder( $column_name, 'NOT IN', $values );
}
/**
* Add a HAVING column IS NULL clause to your query
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function having_null( $column_name ) {
return $this->_add_having_no_value( $column_name, 'IS NULL' );
}
/**
* Add a HAVING column IS NOT NULL clause to your query
*
* @param string|array $column_name The table column
*
* @return ORM
*/
public function having_not_null( $column_name ) {
return $this->_add_having_no_value( $column_name, 'IS NOT NULL' );
}
/**
* Add a raw HAVING clause to the query. The clause should
* contain question mark placeholders, which will be bound
* to the parameters supplied in the second argument.
*
* @param $clause
* @param array $parameters
*
* @return ORM
*/
public function having_raw( $clause, $parameters = [] ) {
return $this->_add_having( $clause, $parameters );
}
/**
* Build a SELECT statement based on the clauses that have
* been passed to this instance by chaining method calls.
*
* @return string
*/
protected function _build_select() {
// If the query is raw, just set the $this->_values to be
// the raw query parameters and return the raw query
if ( $this->_is_raw_query ) {
$this->_values = $this->_raw_parameters;
return $this->_raw_query;
}
// Build and return the full SELECT statement by concatenating
// the results of calling each separate builder method.
return $this->_join_if_not_empty( ' ', [
$this->_build_select_start(),
$this->_build_join(),
$this->_build_where(),
$this->_build_group_by(),
$this->_build_having(),
$this->_build_order_by(),
$this->_build_limit(),
$this->_build_offset(),
] );
}
/**
* Build the start of the SELECT statement
*
* @return string
*/
protected function _build_select_start() {
$fragment = 'SELECT ';
$result_columns = \join( ', ', $this->_result_columns );
if ( $this->_distinct ) {
$result_columns = 'DISTINCT ' . $result_columns;
}
$fragment .= "{$result_columns} FROM " . $this->_quote_identifier( $this->_table_name );
if ( ! \is_null( $this->_table_alias ) ) {
$fragment .= ' ' . $this->_quote_identifier( $this->_table_alias );
}
return $fragment;
}
/**
* Build the JOIN sources
*
* @return string
*/
protected function _build_join() {
if ( \count( $this->_join_sources ) === 0 ) {
return '';
}
return \join( ' ', $this->_join_sources );
}
/**
* Build the WHERE clause(s)
*
* @return string
*/
protected function _build_where() {
return $this->_build_conditions( 'where' );
}
/**
* Build the HAVING clause(s)
*
* @return string
*/
protected function _build_having() {
return $this->_build_conditions( 'having' );
}
/**
* Build GROUP BY
*
* @return string
*/
protected function _build_group_by() {
if ( \count( $this->_group_by ) === 0 ) {
return '';
}
return 'GROUP BY ' . \join( ', ', $this->_group_by );
}
/**
* Build a WHERE or HAVING clause
*
* @param string $type
*
* @return string
*/
protected function _build_conditions( $type ) {
$conditions_class_property_name = "_{$type}_conditions";
// If there are no clauses, return empty string
if ( \count( $this->{$conditions_class_property_name} ) === 0 ) {
return '';
}
$conditions = [];
foreach ( $this->{$conditions_class_property_name} as $condition ) {
$conditions[] = $condition[ self::CONDITION_FRAGMENT ];
$this->_values = \array_merge( $this->_values, $condition[ self::CONDITION_VALUES ] );
}
return \strtoupper( $type ) . ' ' . \join( ' AND ', $conditions );
}
/**
* Build ORDER BY
*/
protected function _build_order_by() {
if ( \count( $this->_order_by ) === 0 ) {
return '';
}
return 'ORDER BY ' . \join( ', ', $this->_order_by );
}
/**
* Build LIMIT
*/
protected function _build_limit() {
if ( ! \is_null( $this->_limit ) ) {
return "LIMIT {$this->_limit}";
}
return '';
}
/**
* Build OFFSET
*/
protected function _build_offset() {
if ( ! \is_null( $this->_offset ) ) {
return 'OFFSET ' . $this->_offset;
}
return '';
}
/**
* Wrapper around PHP's join function which
* only adds the pieces if they are not empty.
*
* @param $glue
* @param $pieces
*
* @return string
*/
protected function _join_if_not_empty( $glue, $pieces ) {
$filtered_pieces = [];
foreach ( $pieces as $piece ) {
if ( \is_string( $piece ) ) {
$piece = \trim( $piece );
}
if ( ! empty( $piece ) ) {
$filtered_pieces[] = $piece;
}
}
return \join( $glue, $filtered_pieces );
}
/**
* Quote a string that is used as an identifier
* (table names, column names etc). This method can
* also deal with dot-separated identifiers eg table.column
*
* @param $identifier
*
* @return string
*/
protected function _quote_one_identifier( $identifier ) {
$parts = \explode( '.', $identifier );
$parts = \array_map( [ $this, '_quote_identifier_part' ], $parts );
return \join( '.', $parts );
}
/**
* Quote a string that is used as an identifier
* (table names, column names etc) or an array containing
* multiple identifiers. This method can also deal with
* dot-separated identifiers eg table.column
*
* @param $identifier
*
* @return string
*/
protected function _quote_identifier( $identifier ) {
if ( \is_array( $identifier ) ) {
$result = \array_map( [ $this, '_quote_one_identifier' ], $identifier );
return \join( ', ', $result );
} else {
return $this->_quote_one_identifier( $identifier );
}
}
/**
* This method performs the actual quoting of a single
* part of an identifier, using the identifier quote
* character specified in the config (or autodetected).
*
* @param $part
*
* @return string
*/
protected function _quote_identifier_part( $part ) {
if ( $part === '*' ) {
return $part;
}
$quote_character = '`';
// double up any identifier quotes to escape them
return $quote_character . \str_replace( $quote_character, $quote_character . $quote_character, $part ) . $quote_character;
}
/**
* Execute the SELECT query that has been built up by chaining methods
* on this class. Return an array of rows as associative arrays.
*
* @return array|false The result rows. False if the query failed.
*/
protected function _run() {
global $wpdb;
$query = $this->_build_select();
$success = self::_execute( $query, $this->_values );
if ( $success === false ) {
// If the query fails run the migrations and try again.
// Action is intentionally undocumented and should not be used by third-parties.
\do_action( '_yoast_run_migrations' );
$success = self::_execute( $query, $this->_values );
}
$this->_reset_idiorm_state();
if ( $success === false ) {
return false;
}
$rows = [];
foreach ( $wpdb->last_result as $row ) {
$rows[] = \get_object_vars( $row );
}
return $rows;
}
/**
* Reset the Idiorm instance state
*/
private function _reset_idiorm_state() {
$this->_values = [];
$this->_result_columns = [ '*' ];
$this->_using_default_result_columns = true;
}
/**
* Return the raw data wrapped by this ORM
* instance as an associative array. Column
* names may optionally be supplied as arguments,
* if so, only those keys will be returned.
*/
public function as_array() {
if ( \func_num_args() === 0 ) {
return $this->_data;
}
$args = \func_get_args();
return \array_intersect_key( $this->_data, \array_flip( $args ) );
}
/**
* Return the value of a property of this object (database row)
* or null if not present.
*
* If a column-names array is passed, it will return a associative array
* with the value of each column or null if it is not present.
*
* @param $key
*
* @return array|mixed|null
*/
public function get( $key ) {
if ( \is_array( $key ) ) {
$result = [];
foreach ( $key as $column ) {
$result[ $column ] = isset( $this->_data[ $column ] ) ? $this->_data[ $column ] : null;
}
return $result;
} else {
return isset( $this->_data[ $key ] ) ? $this->_data[ $key ] : null;
}
}
/**
* Return the name of the column in the database table which contains
* the primary key ID of the row.
*/
protected function _get_id_column_name() {
if ( ! \is_null( $this->_instance_id_column ) ) {
return $this->_instance_id_column;
}
return 'id';
}
/**
* Get the primary key ID of this object.
*
* @param bool $disallow_null
*
* @return array|mixed|null
* @throws \Exception
*/
public function id( $disallow_null = false ) {
$id = $this->get( $this->_get_id_column_name() );
if ( $disallow_null ) {
if ( \is_array( $id ) ) {
foreach ( $id as $id_part ) {
if ( $id_part === null ) {
throw new \Exception( 'Primary key ID contains null value(s)' );
}
}
} else {
if ( $id === null ) {
throw new \Exception( 'Primary key ID missing from row or is null' );
}
}
}
return $id;
}
/**
* Set a property to a particular value on this object.
* To set multiple properties at once, pass an associative array
* as the first parameter and leave out the second parameter.
* Flags the properties as 'dirty' so they will be saved to the
* database when save() is called.
*
* @param $key
* @param null $value
*
* @return ORM
*/
public function set( $key, $value = null ) {
return $this->_set_orm_property( $key, $value );
}
/**
* Set a property to a particular value on this object.
* To set multiple properties at once, pass an associative array
* as the first parameter and leave out the second parameter.
* Flags the properties as 'dirty' so they will be saved to the
* database when save() is called.
*
* @param string|array $key
* @param string|null $value
*
* @return ORM
*/
public function set_expr( $key, $value = null ) {
return $this->_set_orm_property( $key, $value, true );
}
/**
* Set a property on the ORM object.
*
* @param string|array $key
* @param string|null $value
* @param bool $expr
*
* @return ORM
*/
protected function _set_orm_property( $key, $value = null, $expr = false ) {
if ( ! \is_array( $key ) ) {
$key = [ $key => $value ];
}
foreach ( $key as $field => $value ) {
$this->_data[ $field ] = $value;
$this->_dirty_fields[ $field ] = $value;
if ( false === $expr and isset( $this->_expr_fields[ $field ] ) ) {
unset( $this->_expr_fields[ $field ] );
} else {
if ( true === $expr ) {
$this->_expr_fields[ $field ] = true;
}
}
}
return $this;
}
/**
* Check whether the given field has been changed since this
* object was saved.
*
* @param $key
*
* @return bool
*/
public function is_dirty( $key ) {
return \array_key_exists( $key, $this->_dirty_fields );
}
/**
* Check whether the model was the result of a call to create() or not
*
* @return bool
*/
public function is_new() {
return $this->_is_new;
}
/**
* Save any fields which have been modified on this object
* to the database.
*/
public function save() {
global $wpdb;
// remove any expression fields as they are already baked into the query
$values = \array_values( \array_diff_key( $this->_dirty_fields, $this->_expr_fields ) );
if ( ! $this->_is_new ) {
// UPDATE
// If there are no dirty values, do nothing
if ( empty( $values ) && empty( $this->_expr_fields ) ) {
return true;
}
$query = $this->_build_update();
$id = $this->id( true );
if ( \is_array( $id ) ) {
$values = \array_merge( $values, \array_values( $id ) );
} else {
$values[] = $id;
}
} else {
// INSERT
$query = $this->_build_insert();
}
$success = self::_execute( $query, $values );
// If we've just inserted a new record, set the ID of this object
if ( $this->_is_new ) {
$this->_is_new = false;
if ( $this->count_null_id_columns() != 0 ) {
$column = $this->_get_id_column_name();
// if the primary key is compound, assign the last inserted id
// to the first column
if ( \is_array( $column ) ) {
$column = \reset( $column );
}
$this->_data[ $column ] = $wpdb->insert_id;
}
}
$this->_dirty_fields = $this->_expr_fields = [];
return $success;
}
/**
* Add a WHERE clause for every column that belongs to the primary key
*
* @param array $query The query.
*
* @return void
*/
public function _add_id_column_conditions( &$query ) {
$query[] = 'WHERE';
$keys = \is_array( $this->_get_id_column_name() ) ? $this->_get_id_column_name() : [ $this->_get_id_column_name() ];
$first = true;
foreach ( $keys as $key ) {
if ( $first ) {
$first = false;
} else {
$query[] = 'AND';
}
$query[] = $this->_quote_identifier( $key );
$query[] = '= %s';
}
}
/**
* Build an UPDATE query
*
* @return string The update query.
*/
protected function _build_update() {
$query = [];
$query[] = "UPDATE {$this->_quote_identifier($this->_table_name)} SET";
$field_list = [];
foreach ( $this->_dirty_fields as $key => $value ) {
if ( ! \array_key_exists( $key, $this->_expr_fields ) ) {
$value = ( $value === null ) ? 'NULL' : '%s';
}
$field_list[] = "{$this->_quote_identifier($key)} = {$value}";
}
$query[] = \join( ', ', $field_list );
$this->_add_id_column_conditions( $query );
return \join( ' ', $query );
}
/**
* Build an INSERT query
*
* @return string The insert query.
*/
protected function _build_insert() {
$query[] = 'INSERT INTO';
$query[] = $this->_quote_identifier( $this->_table_name );
$field_list = \array_map( [ $this, '_quote_identifier' ], \array_keys( $this->_dirty_fields ) );
$query[] = '(' . \join( ', ', $field_list ) . ')';
$query[] = 'VALUES';
$placeholders = $this->_create_placeholders( $this->_dirty_fields );
$query[] = "({$placeholders})";
return \join( ' ', $query );
}
/**
* Delete this record from the database
*
* @return string The delete query.
*
* @throws \Exception
*/
public function delete() {
$query = [ 'DELETE FROM', $this->_quote_identifier( $this->_table_name ) ];
$this->_add_id_column_conditions( $query );
return self::_execute( \join( ' ', $query ), \is_array( $this->id( true ) ) ? \array_values( $this->id( true ) ) : [ $this->id( true ) ] );
}
/**
* Delete many records from the database
*/
public function delete_many() {
// Build and return the full DELETE statement by concatenating
// the results of calling each separate builder method.
$query = $this->_join_if_not_empty( ' ', [
'DELETE FROM',
$this->_quote_identifier( $this->_table_name ),
$this->_build_where(),
] );
return self::_execute( $query, $this->_values );
}
// --------------------- //
// --- ArrayAccess --- //
// --------------------- //
/**
* @param mixed $key
*
* @return bool
*/
public function offsetExists( $key ) {
return \array_key_exists( $key, $this->_data );
}
/**
* @param mixed $key
*
* @return array|mixed|null
*/
public function offsetGet( $key ) {
return $this->get( $key );
}
/**
* @param mixed $key
* @param mixed $value
*/
public function offsetSet( $key, $value ) {
if ( \is_null( $key ) ) {
return;
}
$this->set( $key, $value );
}
/**
* @param mixed $key
*/
public function offsetUnset( $key ) {
unset( $this->_data[ $key ] );
unset( $this->_dirty_fields[ $key ] );
}
// --------------------- //
// --- MAGIC METHODS --- //
// --------------------- //
/**
* @param $key
*
* @return array|mixed|null
*/
public function __get( $key ) {
return $this->offsetGet( $key );
}
/**
* @param $key
* @param $value
*/
public function __set( $key, $value ) {
$this->offsetSet( $key, $value );
}
/**
* @param $key
*/
public function __unset( $key ) {
$this->offsetUnset( $key );
}
/**
* @param $key
*
* @return bool
*/
public function __isset( $key ) {
return $this->offsetExists( $key );
}
}