pecintaseni
/
engineN
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
180 Commits
1 Branch
0 Tags
508 MiB
 
 
 
 
 
 
Tree: 67fcac0a76
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '67fcac0a76'
${ noResults }
engineN/system/BaseModel.php

1919 lines
56 KiB
Raw Blame History

<?php
/**
* This file is part of CodeIgniter 4 framework.
*
* (c) CodeIgniter Foundation <admin@codeigniter.com>
*
* For the full copyright and license information, please view
* the LICENSE file that was distributed with this source code.
*/
namespace CodeIgniter;
use Closure;
use CodeIgniter\Database\BaseConnection;
use CodeIgniter\Database\BaseResult;
use CodeIgniter\Database\Exceptions\DatabaseException;
use CodeIgniter\Database\Exceptions\DataException;
use CodeIgniter\Database\Query;
use CodeIgniter\Exceptions\ModelException;
use CodeIgniter\I18n\Time;
use CodeIgniter\Pager\Pager;
use CodeIgniter\Validation\ValidationInterface;
use Config\Services;
use InvalidArgumentException;
use ReflectionClass;
use ReflectionException;
use ReflectionProperty;
use stdClass;
/**
* The BaseModel class provides a number of convenient features that
* makes working with a databases less painful. Extending this class
* provide means of implementing various database systems
*
* It will:
* - simplifies pagination
* - allow specifying the return type (array, object, etc) with each call
* - automatically set and update timestamps
* - handle soft deletes
* - ensure validation is run against objects when saving items
* - process various callbacks
* - allow intermingling calls to the db connection
*
* @phpstan-type row_array array<int|string, float|int|null|object|string>
* @phpstan-type event_data_beforeinsert array{data: row_array}
* @phpstan-type event_data_afterinsert array{id: int|string, data: row_array, result: bool}
* @phpstan-type event_data_beforefind array{id?: int|string, method: string, singleton: bool, limit?: int, offset?: int}
* @phpstan-type event_data_afterfind array{id: int|string|null|list<int|string>, data: row_array|list<row_array>|object|null, method: string, singleton: bool}
* @phpstan-type event_data_beforeupdate array{id: null|list<int|string>, data: row_array}
* @phpstan-type event_data_afterupdate array{id: null|list<int|string>, data: row_array|object, result: bool}
* @phpstan-type event_data_beforedelete array{id: null|list<int|string>, purge: bool}
* @phpstan-type event_data_afterdelete array{id: null|list<int|string>, data: null, purge: bool, result: bool}
*/
abstract class BaseModel
{
/**
* Pager instance.
* Populated after calling $this->paginate()
*
* @var Pager
*/
public $pager;
/**
* Last insert ID
*
* @var int|string
*/
protected $insertID = 0;
/**
* The Database connection group that
* should be instantiated.
*
* @var non-empty-string|null
*/
protected $DBGroup;
/**
* The format that the results should be returned as.
* Will be overridden if the as* methods are used.
*
* @var string
*/
protected $returnType = 'array';
/**
* If this model should use "softDeletes" and
* simply set a date when rows are deleted, or
* do hard deletes.
*
* @var bool
*/
protected $useSoftDeletes = false;
/**
* An array of field names that are allowed
* to be set by the user in inserts/updates.
*
* @var list<string>
*/
protected $allowedFields = [];
/**
* If true, will set created_at, and updated_at
* values during insert and update routines.
*
* @var bool
*/
protected $useTimestamps = false;
/**
* The type of column that created_at and updated_at
* are expected to.
*
* Allowed: 'datetime', 'date', 'int'
*
* @var string
*/
protected $dateFormat = 'datetime';
/**
* The column used for insert timestamps
*
* @var string
*/
protected $createdField = 'created_at';
/**
* The column used for update timestamps
*
* @var string
*/
protected $updatedField = 'updated_at';
/**
* Used by withDeleted to override the
* model's softDelete setting.
*
* @var bool
*/
protected $tempUseSoftDeletes;
/**
* The column used to save soft delete state
*
* @var string
*/
protected $deletedField = 'deleted_at';
/**
* Used by asArray and asObject to provide
* temporary overrides of model default.
*
* @var string
*/
protected $tempReturnType;
/**
* Whether we should limit fields in inserts
* and updates to those available in $allowedFields or not.
*
* @var bool
*/
protected $protectFields = true;
/**
* Database Connection
*
* @var BaseConnection
*/
protected $db;
/**
* Rules used to validate data in insert(), update(), and save() methods.
*
* The array must match the format of data passed to the Validation
* library.
*
* @see https://codeigniter4.github.io/userguide/models/model.html#setting-validation-rules
*
* @var array<string, array<string, array<string, string>|string>|string>|string
*/
protected $validationRules = [];
/**
* Contains any custom error messages to be
* used during data validation.
*
* @var array<string, array<string, string>>
*/
protected $validationMessages = [];
/**
* Skip the model's validation. Used in conjunction with skipValidation()
* to skip data validation for any future calls.
*
* @var bool
*/
protected $skipValidation = false;
/**
* Whether rules should be removed that do not exist
* in the passed data. Used in updates.
*
* @var bool
*/
protected $cleanValidationRules = true;
/**
* Our validator instance.
*
* @var ValidationInterface
*/
protected $validation;
/*
* Callbacks.
*
* Each array should contain the method names (within the model)
* that should be called when those events are triggered.
*
* "Update" and "delete" methods are passed the same items that
* are given to their respective method.
*
* "Find" methods receive the ID searched for (if present), and
* 'afterFind' additionally receives the results that were found.
*/
/**
* Whether to trigger the defined callbacks
*
* @var bool
*/
protected $allowCallbacks = true;
/**
* Used by allowCallbacks() to override the
* model's allowCallbacks setting.
*
* @var bool
*/
protected $tempAllowCallbacks;
/**
* Callbacks for beforeInsert
*
* @var list<string>
*/
protected $beforeInsert = [];
/**
* Callbacks for afterInsert
*
* @var list<string>
*/
protected $afterInsert = [];
/**
* Callbacks for beforeUpdate
*
* @var list<string>
*/
protected $beforeUpdate = [];
/**
* Callbacks for afterUpdate
*
* @var list<string>
*/
protected $afterUpdate = [];
/**
* Callbacks for beforeInsertBatch
*
* @var list<string>
*/
protected $beforeInsertBatch = [];
/**
* Callbacks for afterInsertBatch
*
* @var list<string>
*/
protected $afterInsertBatch = [];
/**
* Callbacks for beforeUpdateBatch
*
* @var list<string>
*/
protected $beforeUpdateBatch = [];
/**
* Callbacks for afterUpdateBatch
*
* @var list<string>
*/
protected $afterUpdateBatch = [];
/**
* Callbacks for beforeFind
*
* @var list<string>
*/
protected $beforeFind = [];
/**
* Callbacks for afterFind
*
* @var list<string>
*/
protected $afterFind = [];
/**
* Callbacks for beforeDelete
*
* @var list<string>
*/
protected $beforeDelete = [];
/**
* Callbacks for afterDelete
*
* @var list<string>
*/
protected $afterDelete = [];
/**
* Whether to allow inserting empty data.
*/
protected bool $allowEmptyInserts = false;
public function __construct(?ValidationInterface $validation = null)
{
$this->tempReturnType = $this->returnType;
$this->tempUseSoftDeletes = $this->useSoftDeletes;
$this->tempAllowCallbacks = $this->allowCallbacks;
/**
* @var ValidationInterface|null $validation
*/
$validation ??= Services::validation(null, false);
$this->validation = $validation;
$this->initialize();
}
/**
* Initializes the instance with any additional steps.
* Optionally implemented by child classes.
*
* @return void
*/
protected function initialize()
{
}
/**
* Fetches the row of database.
* This method works only with dbCalls.
*
* @param bool $singleton Single or multiple results
* @param array|int|string|null $id One primary key or an array of primary keys
*
* @return array|object|null The resulting row of data, or null.
*/
abstract protected function doFind(bool $singleton, $id = null);
/**
* Fetches the column of database.
* This method works only with dbCalls.
*
* @param string $columnName Column Name
*
* @return array|null The resulting row of data, or null if no data found.
*
* @throws DataException
*/
abstract protected function doFindColumn(string $columnName);
/**
* Fetches all results, while optionally limiting them.
* This method works only with dbCalls.
*
* @param int $limit Limit
* @param int $offset Offset
*
* @return array
*/
abstract protected function doFindAll(int $limit = 0, int $offset = 0);
/**
* Returns the first row of the result set.
* This method works only with dbCalls.
*
* @return array|object|null
*/
abstract protected function doFirst();
/**
* Inserts data into the current database.
* This method works only with dbCalls.
*
* @param array $row Row data
* @phpstan-param row_array $row
*
* @return bool
*/
abstract protected function doInsert(array $row);
/**
* Compiles batch insert and runs the queries, validating each row prior.
* This method works only with dbCalls.
*
* @param array|null $set An associative array of insert values
* @param bool|null $escape Whether to escape values
* @param int $batchSize The size of the batch to run
* @param bool $testing True means only number of records is returned, false will execute the query
*
* @return bool|int Number of rows inserted or FALSE on failure
*/
abstract protected function doInsertBatch(?array $set = null, ?bool $escape = null, int $batchSize = 100, bool $testing = false);
/**
* Updates a single record in the database.
* This method works only with dbCalls.
*
* @param array|int|string|null $id ID
* @param array|null $row Row data
* @phpstan-param row_array|null $row
*/
abstract protected function doUpdate($id = null, $row = null): bool;
/**
* Compiles an update and runs the query.
* This method works only with dbCalls.
*
* @param array|null $set An associative array of update values
* @param string|null $index The where key
* @param int $batchSize The size of the batch to run
* @param bool $returnSQL True means SQL is returned, false will execute the query
*
* @return false|int|list<string> Number of rows affected or FALSE on failure, SQL array when testMode
*
* @throws DatabaseException
*/
abstract protected function doUpdateBatch(?array $set = null, ?string $index = null, int $batchSize = 100, bool $returnSQL = false);
/**
* Deletes a single record from the database where $id matches.
* This method works only with dbCalls.
*
* @param array|int|string|null $id The rows primary key(s)
* @param bool $purge Allows overriding the soft deletes setting.
*
* @return bool|string
*
* @throws DatabaseException
*/
abstract protected function doDelete($id = null, bool $purge = false);
/**
* Permanently deletes all rows that have been marked as deleted.
* through soft deletes (deleted = 1).
* This method works only with dbCalls.
*
* @return bool|string Returns a string if in test mode.
*/
abstract protected function doPurgeDeleted();
/**
* Works with the find* methods to return only the rows that
* have been deleted.
* This method works only with dbCalls.
*
* @return void
*/
abstract protected function doOnlyDeleted();
/**
* Compiles a replace and runs the query.
* This method works only with dbCalls.
*
* @param array|null $row Row data
* @phpstan-param row_array|null $row
* @param bool $returnSQL Set to true to return Query String
*
* @return BaseResult|false|Query|string
*/
abstract protected function doReplace(?array $row = null, bool $returnSQL = false);
/**
* Grabs the last error(s) that occurred from the Database connection.
* This method works only with dbCalls.
*
* @return array|null
*/
abstract protected function doErrors();
/**
* Returns the id value for the data array or object.
*
* @param array|object $data Data
*
* @return array|int|string|null
*
* @deprecated Add an override on getIdValue() instead. Will be removed in version 5.0.
*/
abstract protected function idValue($data);
/**
* Public getter to return the id value using the idValue() method.
* For example with SQL this will return $data->$this->primaryKey.
*
* @param array|object $row Row data
* @phpstan-param row_array|object $row
*
* @return array|int|string|null
*
* @todo: Make abstract in version 5.0
*/
public function getIdValue($row)
{
return $this->idValue($row);
}
/**
* Override countAllResults to account for soft deleted accounts.
* This method works only with dbCalls.
*
* @param bool $reset Reset
* @param bool $test Test
*
* @return int|string
*/
abstract public function countAllResults(bool $reset = true, bool $test = false);
/**
* Loops over records in batches, allowing you to operate on them.
* This method works only with dbCalls.
*
* @param int $size Size
* @param Closure $userFunc Callback Function
*
* @return void
*
* @throws DataException
*/
abstract public function chunk(int $size, Closure $userFunc);
/**
* Fetches the row of database.
*
* @param array|int|string|null $id One primary key or an array of primary keys
*
* @return array|object|null The resulting row of data, or null.
* @phpstan-return ($id is int|string ? row_array|object|null : list<row_array|object>)
*/
public function find($id = null)
{
$singleton = is_numeric($id) || is_string($id);
if ($this->tempAllowCallbacks) {
// Call the before event and check for a return
$eventData = $this->trigger('beforeFind', [
'id' => $id,
'method' => 'find',
'singleton' => $singleton,
]);
if (isset($eventData['returnData']) && $eventData['returnData'] === true) {
return $eventData['data'];
}
}
$eventData = [
'id' => $id,
'data' => $this->doFind($singleton, $id),
'method' => 'find',
'singleton' => $singleton,
];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('afterFind', $eventData);
}
$this->tempReturnType = $this->returnType;
$this->tempUseSoftDeletes = $this->useSoftDeletes;
$this->tempAllowCallbacks = $this->allowCallbacks;
return $eventData['data'];
}
/**
* Fetches the column of database.
*
* @param string $columnName Column Name
*
* @return array|null The resulting row of data, or null if no data found.
*
* @throws DataException
*/
public function findColumn(string $columnName)
{
if (strpos($columnName, ',') !== false) {
throw DataException::forFindColumnHaveMultipleColumns();
}
$resultSet = $this->doFindColumn($columnName);
return $resultSet ? array_column($resultSet, $columnName) : null;
}
/**
* Fetches all results, while optionally limiting them.
*
* @param int $limit Limit
* @param int $offset Offset
*
* @return array
*/
public function findAll(int $limit = 0, int $offset = 0)
{
if ($this->tempAllowCallbacks) {
// Call the before event and check for a return
$eventData = $this->trigger('beforeFind', [
'method' => 'findAll',
'limit' => $limit,
'offset' => $offset,
'singleton' => false,
]);
if (isset($eventData['returnData']) && $eventData['returnData'] === true) {
return $eventData['data'];
}
}
$eventData = [
'data' => $this->doFindAll($limit, $offset),
'limit' => $limit,
'offset' => $offset,
'method' => 'findAll',
'singleton' => false,
];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('afterFind', $eventData);
}
$this->tempReturnType = $this->returnType;
$this->tempUseSoftDeletes = $this->useSoftDeletes;
$this->tempAllowCallbacks = $this->allowCallbacks;
return $eventData['data'];
}
/**
* Returns the first row of the result set.
*
* @return array|object|null
*/
public function first()
{
if ($this->tempAllowCallbacks) {
// Call the before event and check for a return
$eventData = $this->trigger('beforeFind', [
'method' => 'first',
'singleton' => true,
]);
if (isset($eventData['returnData']) && $eventData['returnData'] === true) {
return $eventData['data'];
}
}
$eventData = [
'data' => $this->doFirst(),
'method' => 'first',
'singleton' => true,
];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('afterFind', $eventData);
}
$this->tempReturnType = $this->returnType;
$this->tempUseSoftDeletes = $this->useSoftDeletes;
$this->tempAllowCallbacks = $this->allowCallbacks;
return $eventData['data'];
}
/**
* A convenience method that will attempt to determine whether the
* data should be inserted or updated. Will work with either
* an array or object. When using with custom class objects,
* you must ensure that the class will provide access to the class
* variables, even if through a magic method.
*
* @param array|object $row Row data
* @phpstan-param row_array|object $row
*
* @throws ReflectionException
*/
public function save($row): bool
{
if ((array) $row === []) {
return true;
}
if ($this->shouldUpdate($row)) {
$response = $this->update($this->getIdValue($row), $row);
} else {
$response = $this->insert($row, false);
if ($response !== false) {
$response = true;
}
}
return $response;
}
/**
* This method is called on save to determine if entry have to be updated.
* If this method returns false insert operation will be executed
*
* @param array|object $row Row data
* @phpstan-param row_array|object $row
*/
protected function shouldUpdate($row): bool
{
$id = $this->getIdValue($row);
return ! ($id === null || $id === [] || $id === '');
}
/**
* Returns last insert ID or 0.
*
* @return int|string
*/
public function getInsertID()
{
return is_numeric($this->insertID) ? (int) $this->insertID : $this->insertID;
}
/**
* Inserts data into the database. If an object is provided,
* it will attempt to convert it to an array.
*
* @param array|object|null $row Row data
* @phpstan-param row_array|object|null $row
* @param bool $returnID Whether insert ID should be returned or not.
*
* @return bool|int|string insert ID or true on success. false on failure.
* @phpstan-return ($returnID is true ? int|string|false : bool)
*
* @throws ReflectionException
*/
public function insert($row = null, bool $returnID = true)
{
$this->insertID = 0;
// Set $cleanValidationRules to false temporary.
$cleanValidationRules = $this->cleanValidationRules;
$this->cleanValidationRules = false;
$row = $this->transformDataToArray($row, 'insert');
// Validate data before saving.
if (! $this->skipValidation && ! $this->validate($row)) {
// Restore $cleanValidationRules
$this->cleanValidationRules = $cleanValidationRules;
return false;
}
// Restore $cleanValidationRules
$this->cleanValidationRules = $cleanValidationRules;
// Must be called first, so we don't
// strip out created_at values.
$row = $this->doProtectFieldsForInsert($row);
// doProtectFields() can further remove elements from
// $row, so we need to check for empty dataset again
if (! $this->allowEmptyInserts && $row === []) {
throw DataException::forEmptyDataset('insert');
}
// Set created_at and updated_at with same time
$date = $this->setDate();
$row = $this->setCreatedField($row, $date);
$row = $this->setUpdatedField($row, $date);
$eventData = ['data' => $row];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('beforeInsert', $eventData);
}
$result = $this->doInsert($eventData['data']);
$eventData = [
'id' => $this->insertID,
'data' => $eventData['data'],
'result' => $result,
];
if ($this->tempAllowCallbacks) {
// Trigger afterInsert events with the inserted data and new ID
$this->trigger('afterInsert', $eventData);
}
$this->tempAllowCallbacks = $this->allowCallbacks;
// If insertion failed, get out of here
if (! $result) {
return $result;
}
// otherwise return the insertID, if requested.
return $returnID ? $this->insertID : $result;
}
/**
* Set datetime to created field.
*
* @phpstan-param row_array $row
* @param int|string $date timestamp or datetime string
*/
protected function setCreatedField(array $row, $date): array
{
if ($this->useTimestamps && $this->createdField !== '' && ! array_key_exists($this->createdField, $row)) {
$row[$this->createdField] = $date;
}
return $row;
}
/**
* Set datetime to updated field.
*
* @phpstan-param row_array $row
* @param int|string $date timestamp or datetime string
*/
protected function setUpdatedField(array $row, $date): array
{
if ($this->useTimestamps && $this->updatedField !== '' && ! array_key_exists($this->updatedField, $row)) {
$row[$this->updatedField] = $date;
}
return $row;
}
/**
* Compiles batch insert runs the queries, validating each row prior.
*
* @param list<array|object>|null $set an associative array of insert values
* @phpstan-param list<row_array|object>|null $set
* @param bool|null $escape Whether to escape values
* @param int $batchSize The size of the batch to run
* @param bool $testing True means only number of records is returned, false will execute the query
*
* @return bool|int Number of rows inserted or FALSE on failure
*
* @throws ReflectionException
*/
public function insertBatch(?array $set = null, ?bool $escape = null, int $batchSize = 100, bool $testing = false)
{
// Set $cleanValidationRules to false temporary.
$cleanValidationRules = $this->cleanValidationRules;
$this->cleanValidationRules = false;
if (is_array($set)) {
foreach ($set as &$row) {
// If $row is using a custom class with public or protected
// properties representing the collection elements, we need to grab
// them as an array.
if (is_object($row) && ! $row instanceof stdClass) {
$row = $this->objectToArray($row, false, true);
}
// If it's still a stdClass, go ahead and convert to
// an array so doProtectFields and other model methods
// don't have to do special checks.
if (is_object($row)) {
$row = (array) $row;
}
// Validate every row.
if (! $this->skipValidation && ! $this->validate($row)) {
// Restore $cleanValidationRules
$this->cleanValidationRules = $cleanValidationRules;
return false;
}
// Must be called first so we don't
// strip out created_at values.
$row = $this->doProtectFieldsForInsert($row);
// Set created_at and updated_at with same time
$date = $this->setDate();
$row = $this->setCreatedField($row, $date);
$row = $this->setUpdatedField($row, $date);
}
}
// Restore $cleanValidationRules
$this->cleanValidationRules = $cleanValidationRules;
$eventData = ['data' => $set];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('beforeInsertBatch', $eventData);
}
$result = $this->doInsertBatch($eventData['data'], $escape, $batchSize, $testing);
$eventData = [
'data' => $eventData['data'],
'result' => $result,
];
if ($this->tempAllowCallbacks) {
// Trigger afterInsert events with the inserted data and new ID
$this->trigger('afterInsertBatch', $eventData);
}
$this->tempAllowCallbacks = $this->allowCallbacks;
return $result;
}
/**
* Updates a single record in the database. If an object is provided,
* it will attempt to convert it into an array.
*
* @param array|int|string|null $id
* @param array|object|null $row Row data
* @phpstan-param row_array|object|null $row
*
* @throws ReflectionException
*/
public function update($id = null, $row = null): bool
{
if (is_bool($id)) {
throw new InvalidArgumentException('update(): argument #1 ($id) should not be boolean.');
}
if (is_numeric($id) || is_string($id)) {
$id = [$id];
}
$row = $this->transformDataToArray($row, 'update');
// Validate data before saving.
if (! $this->skipValidation && ! $this->validate($row)) {
return false;
}
// Must be called first, so we don't
// strip out updated_at values.
$row = $this->doProtectFields($row);
// doProtectFields() can further remove elements from
// $row, so we need to check for empty dataset again
if ($row === []) {
throw DataException::forEmptyDataset('update');
}
$row = $this->setUpdatedField($row, $this->setDate());
$eventData = [
'id' => $id,
'data' => $row,
];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('beforeUpdate', $eventData);
}
$eventData = [
'id' => $id,
'data' => $eventData['data'],
'result' => $this->doUpdate($id, $eventData['data']),
];
if ($this->tempAllowCallbacks) {
$this->trigger('afterUpdate', $eventData);
}
$this->tempAllowCallbacks = $this->allowCallbacks;
return $eventData['result'];
}
/**
* Compiles an update and runs the query.
*
* @param list<array|object>|null $set an associative array of insert values
* @phpstan-param list<row_array|object>|null $set
* @param string|null $index The where key
* @param int $batchSize The size of the batch to run
* @param bool $returnSQL True means SQL is returned, false will execute the query
*
* @return false|int|list<string> Number of rows affected or FALSE on failure, SQL array when testMode
*
* @throws DatabaseException
* @throws ReflectionException
*/
public function updateBatch(?array $set = null, ?string $index = null, int $batchSize = 100, bool $returnSQL = false)
{
if (is_array($set)) {
foreach ($set as &$row) {
// If $row is using a custom class with public or protected
// properties representing the collection elements, we need to grab
// them as an array.
if (is_object($row) && ! $row instanceof stdClass) {
// For updates the index field is needed even if it is not changed.
// So set $onlyChanged to false.
$row = $this->objectToArray($row, false, true);
}
// If it's still a stdClass, go ahead and convert to
// an array so doProtectFields and other model methods
// don't have to do special checks.
if (is_object($row)) {
$row = (array) $row;
}
// Validate data before saving.
if (! $this->skipValidation && ! $this->validate($row)) {
return false;
}
// Save updateIndex for later
$updateIndex = $row[$index] ?? null;
if ($updateIndex === null) {
throw new InvalidArgumentException(
'The index ("' . $index . '") for updateBatch() is missing in the data: '
. json_encode($row)
);
}
// Must be called first so we don't
// strip out updated_at values.
$row = $this->doProtectFields($row);
// Restore updateIndex value in case it was wiped out
if ($updateIndex !== null) {
$row[$index] = $updateIndex;
}
$row = $this->setUpdatedField($row, $this->setDate());
}
}
$eventData = ['data' => $set];
if ($this->tempAllowCallbacks) {
$eventData = $this->trigger('beforeUpdateBatch', $eventData);
}
$result = $this->doUpdateBatch($eventData['data'], $index, $batchSize, $returnSQL);
$eventData = [
'data' => $eventData['data'],
'result' => $result,
];
if ($this->tempAllowCallbacks) {
// Trigger afterInsert events with the inserted data and new ID
$this->trigger('afterUpdateBatch', $eventData);
}
$this->tempAllowCallbacks = $this->allowCallbacks;
return $result;
}
/**
* Deletes a single record from the database where $id matches.
*
* @param array|int|string|null $id The rows primary key(s)
* @param bool $purge Allows overriding the soft deletes setting.
*
* @return BaseResult|bool
*
* @throws DatabaseException
*/
public function delete($id = null, bool $purge = false)
{
if (is_bool($id)) {
throw new InvalidArgumentException('delete(): argument #1 ($id) should not be boolean.');
}
if ($id && (is_numeric($id) || is_string($id))) {
$id = [$id];
}
$eventData = [
'id' => $id,
'purge' => $purge,
];
if ($this->tempAllowCallbacks) {
$this->trigger('beforeDelete', $eventData);
}
$eventData = [
'id' => $id,
'data' => null,
'purge' => $purge,
'result' => $this->doDelete($id, $purge),
];
if ($this->tempAllowCallbacks) {
$this->trigger('afterDelete', $eventData);
}
$this->tempAllowCallbacks = $this->allowCallbacks;
return $eventData['result'];
}
/**
* Permanently deletes all rows that have been marked as deleted
* through soft deletes (deleted = 1).
*
* @return bool|string Returns a string if in test mode.
*/
public function purgeDeleted()
{
if (! $this->useSoftDeletes) {
return true;
}
return $this->doPurgeDeleted();
}
/**
* Sets $useSoftDeletes value so that we can temporarily override
* the soft deletes settings. Can be used for all find* methods.
*
* @param bool $val Value
*
* @return $this
*/
public function withDeleted(bool $val = true)
{
$this->tempUseSoftDeletes = ! $val;
return $this;
}
/**
* Works with the find* methods to return only the rows that
* have been deleted.
*
* @return $this
*/
public function onlyDeleted()
{
$this->tempUseSoftDeletes = false;
$this->doOnlyDeleted();
return $this;
}
/**
* Compiles a replace and runs the query.
*
* @param array|null $row Row data
* @phpstan-param row_array|null $row
* @param bool $returnSQL Set to true to return Query String
*
* @return BaseResult|false|Query|string
*/
public function replace(?array $row = null, bool $returnSQL = false)
{
// Validate data before saving.
if (($row !== null) && ! $this->skipValidation && ! $this->validate($row)) {
return false;
}
$row = $this->setUpdatedField((array) $row, $this->setDate());
return $this->doReplace($row, $returnSQL);
}
/**
* Grabs the last error(s) that occurred. If data was validated,
* it will first check for errors there, otherwise will try to
* grab the last error from the Database connection.
*
* The return array should be in the following format:
* ['source' => 'message']
*
* @param bool $forceDB Always grab the db error, not validation
*
* @return array<string,string>
*/
public function errors(bool $forceDB = false)
{
// Do we have validation errors?
if (! $forceDB && ! $this->skipValidation && ($errors = $this->validation->getErrors())) {
return $errors;
}
return $this->doErrors();
}
/**
* Works with Pager to get the size and offset parameters.
* Expects a GET variable (?page=2) that specifies the page of results
* to display.
*
* @param int|null $perPage Items per page
* @param string $group Will be used by the pagination library to identify a unique pagination set.
* @param int|null $page Optional page number (useful when the page number is provided in different way)
* @param int $segment Optional URI segment number (if page number is provided by URI segment)
*
* @return array|null
*/
public function paginate(?int $perPage = null, string $group = 'default', ?int $page = null, int $segment = 0)
{
// Since multiple models may use the Pager, the Pager must be shared.
$pager = Services::pager();
if ($segment !== 0) {
$pager->setSegment($segment, $group);
}
$page = $page >= 1 ? $page : $pager->getCurrentPage($group);
// Store it in the Pager library, so it can be paginated in the views.
$this->pager = $pager->store($group, $page, $perPage, $this->countAllResults(false), $segment);
$perPage = $this->pager->getPerPage($group);
$offset = ($pager->getCurrentPage($group) - 1) * $perPage;
return $this->findAll($perPage, $offset);
}
/**
* It could be used when you have to change default or override current allowed fields.
*
* @param array $allowedFields Array with names of fields
*
* @return $this
*/
public function setAllowedFields(array $allowedFields)
{
$this->allowedFields = $allowedFields;
return $this;
}
/**
* Sets whether or not we should whitelist data set during
* updates or inserts against $this->availableFields.
*
* @param bool $protect Value
*
* @return $this
*/
public function protect(bool $protect = true)
{
$this->protectFields = $protect;
return $this;
}
/**
* Ensures that only the fields that are allowed to be updated are
* in the data array.
*
* @used-by update() to protect against mass assignment vulnerabilities.
* @used-by updateBatch() to protect against mass assignment vulnerabilities.
*
* @param array $row Row data
* @phpstan-param row_array $row
*
* @throws DataException
*/
protected function doProtectFields(array $row): array
{
if (! $this->protectFields) {
return $row;
}
if ($this->allowedFields === []) {
throw DataException::forInvalidAllowedFields(static::class);
}
foreach (array_keys($row) as $key) {
if (! in_array($key, $this->allowedFields, true)) {
unset($row[$key]);
}
}
return $row;
}
/**
* Ensures that only the fields that are allowed to be inserted are in
* the data array.
*
* @used-by insert() to protect against mass assignment vulnerabilities.
* @used-by insertBatch() to protect against mass assignment vulnerabilities.
*
* @param array $row Row data
* @phpstan-param row_array $row
*
* @throws DataException
*/
protected function doProtectFieldsForInsert(array $row): array
{
return $this->doProtectFields($row);
}
/**
* Sets the date or current date if null value is passed.
*
* @param int|null $userData An optional PHP timestamp to be converted.
*
* @return int|string
*
* @throws ModelException
*/
protected function setDate(?int $userData = null)
{
$currentDate = $userData ?? Time::now()->getTimestamp();
return $this->intToDate($currentDate);
}
/**
* A utility function to allow child models to use the type of
* date/time format that they prefer. This is primarily used for
* setting created_at, updated_at and deleted_at values, but can be
* used by inheriting classes.
*
* The available time formats are:
* - 'int' - Stores the date as an integer timestamp
* - 'datetime' - Stores the data in the SQL datetime format
* - 'date' - Stores the date (only) in the SQL date format.
*
* @param int $value value
*
* @return int|string
*
* @throws ModelException
*/
protected function intToDate(int $value)
{
switch ($this->dateFormat) {
case 'int':
return $value;
case 'datetime':
return date('Y-m-d H:i:s', $value);
case 'date':
return date('Y-m-d', $value);
default:
throw ModelException::forNoDateFormat(static::class);
}
}
/**
* Converts Time value to string using $this->dateFormat.
*
* The available time formats are:
* - 'int' - Stores the date as an integer timestamp
* - 'datetime' - Stores the data in the SQL datetime format
* - 'date' - Stores the date (only) in the SQL date format.
*
* @param Time $value value
*
* @return int|string
*/
protected function timeToDate(Time $value)
{
switch ($this->dateFormat) {
case 'datetime':
return $value->format('Y-m-d H:i:s');
case 'date':
return $value->format('Y-m-d');
case 'int':
return $value->getTimestamp();
default:
return (string) $value;
}
}
/**
* Set the value of the skipValidation flag.
*
* @param bool $skip Value
*
* @return $this
*/
public function skipValidation(bool $skip = true)
{
$this->skipValidation = $skip;
return $this;
}
/**
* Allows to set (and reset) validation messages.
* It could be used when you have to change default or override current validate messages.
*
* @param array $validationMessages Value
*
* @return $this
*/
public function setValidationMessages(array $validationMessages)
{
$this->validationMessages = $validationMessages;
return $this;
}
/**
* Allows to set field wise validation message.
* It could be used when you have to change default or override current validate messages.
*
* @param string $field Field Name
* @param array $fieldMessages Validation messages
*
* @return $this
*/
public function setValidationMessage(string $field, array $fieldMessages)
{
$this->validationMessages[$field] = $fieldMessages;
return $this;
}
/**
* Allows to set (and reset) validation rules.
* It could be used when you have to change default or override current validate rules.
*
* @param array<string, array<string, array<string, string>|string>|string> $validationRules Value
*
* @return $this
*/
public function setValidationRules(array $validationRules)
{
$this->validationRules = $validationRules;
return $this;
}
/**
* Allows to set field wise validation rules.
* It could be used when you have to change default or override current validate rules.
*
* @param string $field Field Name
* @param array|string $fieldRules Validation rules
*
* @return $this
*/
public function setValidationRule(string $field, $fieldRules)
{
$rules = $this->validationRules;
// ValidationRules can be either a string, which is the group name,
// or an array of rules.
if (is_string($rules)) {
[$rules, $customErrors] = $this->validation->loadRuleGroup($rules);
$this->validationRules = $rules;
$this->validationMessages += $customErrors;
}
$this->validationRules[$field] = $fieldRules;
return $this;
}
/**
* Should validation rules be removed before saving?
* Most handy when doing updates.
*
* @param bool $choice Value
*
* @return $this
*/
public function cleanRules(bool $choice = false)
{
$this->cleanValidationRules = $choice;
return $this;
}
/**
* Validate the row data against the validation rules (or the validation group)
* specified in the class property, $validationRules.
*
* @param array|object $row Row data
* @phpstan-param row_array|object $row
*/
public function validate($row): bool
{
$rules = $this->getValidationRules();
// Validation requires array, so cast away.
if (is_object($row)) {
$row = (array) $row;
}
if ($this->skipValidation || $rules === [] || $row === []) {
return true;
}
$rules = $this->cleanValidationRules ? $this->cleanValidationRules($rules, $row) : $rules;
// If no data existed that needs validation
// our job is done here.
if ($rules === []) {
return true;
}
$this->validation->reset()->setRules($rules, $this->validationMessages);
return $this->validation->run($row, null, $this->DBGroup);
}
/**
* Returns the model's defined validation rules so that they
* can be used elsewhere, if needed.
*
* @param array $options Options
*/
public function getValidationRules(array $options = []): array
{
$rules = $this->validationRules;
// ValidationRules can be either a string, which is the group name,
// or an array of rules.
if (is_string($rules)) {
[$rules, $customErrors] = $this->validation->loadRuleGroup($rules);
$this->validationMessages += $customErrors;
}
if (isset($options['except'])) {
$rules = array_diff_key($rules, array_flip($options['except']));
} elseif (isset($options['only'])) {
$rules = array_intersect_key($rules, array_flip($options['only']));
}
return $rules;
}
/**
* Returns the model's validation messages, so they
* can be used elsewhere, if needed.
*/
public function getValidationMessages(): array
{
return $this->validationMessages;
}
/**
* Removes any rules that apply to fields that have not been set
* currently so that rules don't block updating when only updating
* a partial row.
*
* @param array $rules Array containing field name and rule
* @param array $row Row data (@TODO Remove null in param type)
* @phpstan-param row_array $row
*/
protected function cleanValidationRules(array $rules, ?array $row = null): array
{
if ($row === null || $row === []) {
return [];
}
foreach (array_keys($rules) as $field) {
if (! array_key_exists($field, $row)) {
unset($rules[$field]);
}
}
return $rules;
}
/**
* Sets $tempAllowCallbacks value so that we can temporarily override
* the setting. Resets after the next method that uses triggers.
*
* @param bool $val value
*
* @return $this
*/
public function allowCallbacks(bool $val = true)
{
$this->tempAllowCallbacks = $val;
return $this;
}
/**
* A simple event trigger for Model Events that allows additional
* data manipulation within the model. Specifically intended for
* usage by child models this can be used to format data,
* save/load related classes, etc.
*
* It is the responsibility of the callback methods to return
* the data itself.
*
* Each $eventData array MUST have a 'data' key with the relevant
* data for callback methods (like an array of key/value pairs to insert
* or update, an array of results, etc.)
*
* If callbacks are not allowed then returns $eventData immediately.
*
* @param string $event Event
* @param array $eventData Event Data
*
* @return array
*
* @throws DataException
*/
protected function trigger(string $event, array $eventData)
{
// Ensure it's a valid event
if (! isset($this->{$event}) || $this->{$event} === []) {
return $eventData;
}
foreach ($this->{$event} as $callback) {
if (! method_exists($this, $callback)) {
throw DataException::forInvalidMethodTriggered($callback);
}
$eventData = $this->{$callback}($eventData);
}
return $eventData;
}
/**
* Sets the return type of the results to be as an associative array.
*
* @return $this
*/
public function asArray()
{
$this->tempReturnType = 'array';
return $this;
}
/**
* Sets the return type to be of the specified type of object.
* Defaults to a simple object, but can be any class that has
* class vars with the same name as the collection columns,
* or at least allows them to be created.
*
* @param string $class Class Name
*
* @return $this
*/
public function asObject(string $class = 'object')
{
$this->tempReturnType = $class;
return $this;
}
/**
* Takes a class and returns an array of its public and protected
* properties as an array suitable for use in creates and updates.
* This method uses objectToRawArray() internally and does conversion
* to string on all Time instances
*
* @param object $object Object
* @param bool $onlyChanged Only Changed Property
* @param bool $recursive If true, inner entities will be cast as array as well
*
* @return array<string, mixed>
*
* @throws ReflectionException
*/
protected function objectToArray($object, bool $onlyChanged = true, bool $recursive = false): array
{
$properties = $this->objectToRawArray($object, $onlyChanged, $recursive);
// Convert any Time instances to appropriate $dateFormat
return $this->timeToString($properties);
}
/**
* Convert any Time instances to appropriate $dateFormat.
*
* @param array<string, mixed> $properties
*
* @return array<string, mixed>
*/
protected function timeToString(array $properties): array
{
if ($properties === []) {
return [];
}
return array_map(function ($value) {
if ($value instanceof Time) {
return $this->timeToDate($value);
}
return $value;
}, $properties);
}
/**
* Takes a class and returns an array of its public and protected
* properties as an array with raw values.
*
* @param object $object Object
* @param bool $onlyChanged Only Changed Property
* @param bool $recursive If true, inner entities will be casted as array as well
*
* @return array<string, mixed>
*
* @throws ReflectionException
*/
protected function objectToRawArray($object, bool $onlyChanged = true, bool $recursive = false): array
{
// Entity::toRawArray() returns array.
if (method_exists($object, 'toRawArray')) {
$properties = $object->toRawArray($onlyChanged, $recursive);
} else {
$mirror = new ReflectionClass($object);
$props = $mirror->getProperties(ReflectionProperty::IS_PUBLIC | ReflectionProperty::IS_PROTECTED);
$properties = [];
// Loop over each property,
// saving the name/value in a new array we can return.
foreach ($props as $prop) {
// Must make protected values accessible.
$prop->setAccessible(true);
$properties[$prop->getName()] = $prop->getValue($object);
}
}
return $properties;
}
/**
* Transform data to array.
*
* @param array|object|null $row Row data
* @phpstan-param row_array|object|null $row
* @param string $type Type of data (insert|update)
*
* @throws DataException
* @throws InvalidArgumentException
* @throws ReflectionException
*/
protected function transformDataToArray($row, string $type): array
{
if (! in_array($type, ['insert', 'update'], true)) {
throw new InvalidArgumentException(sprintf('Invalid type "%s" used upon transforming data to array.', $type));
}
if (! $this->allowEmptyInserts && ($row === null || (array) $row === [])) {
throw DataException::forEmptyDataset($type);
}
// If $row is using a custom class with public or protected
// properties representing the collection elements, we need to grab
// them as an array.
if (is_object($row) && ! $row instanceof stdClass) {
// If it validates with entire rules, all fields are needed.
$onlyChanged = ($this->skipValidation === false && $this->cleanValidationRules === false)
? false : ($type === 'update');
$row = $this->objectToArray($row, $onlyChanged, true);
}
// If it's still a stdClass, go ahead and convert to
// an array so doProtectFields and other model methods
// don't have to do special checks.
if (is_object($row)) {
$row = (array) $row;
}
// If it's still empty here, means $row is no change or is empty object
if (! $this->allowEmptyInserts && ($row === null || $row === [])) {
throw DataException::forEmptyDataset($type);
}
return $row;
}
/**
* Provides the db connection and model's properties.
*
* @param string $name Name
*
* @return array|bool|float|int|object|string|null
*/
public function __get(string $name)
{
if (property_exists($this, $name)) {
return $this->{$name};
}
return $this->db->{$name} ?? null;
}
/**
* Checks for the existence of properties across this model, and db connection.
*
* @param string $name Name
*/
public function __isset(string $name): bool
{
if (property_exists($this, $name)) {
return true;
}
return isset($this->db->{$name});
}
/**
* Provides direct access to method in the database connection.
*
* @param string $name Name
* @param array $params Params
*
* @return $this|null
*/
public function __call(string $name, array $params)
{
if (method_exists($this->db, $name)) {
return $this->db->{$name}(...$params);
}
return null;
}
/**
* Replace any placeholders within the rules with the values that
* match the 'key' of any properties being set. For example, if
* we had the following $data array:
*
* [ 'id' => 13 ]
*
* and the following rule:
*
* 'required|is_unique[users,email,id,{id}]'
*
* The value of {id} would be replaced with the actual id in the form data:
*
* 'required|is_unique[users,email,id,13]'
*
* @param array $rules Validation rules
* @param array $data Data
*
* @codeCoverageIgnore
*
* @deprecated use fillPlaceholders($rules, $data) from Validation instead
*/
protected function fillPlaceholders(array $rules, array $data): array
{
$replacements = [];
foreach ($data as $key => $value) {
$replacements['{' . $key . '}'] = $value;
}
if ($replacements !== []) {
foreach ($rules as &$rule) {
if (is_array($rule)) {
foreach ($rule as &$row) {
// Should only be an `errors` array
// which doesn't take placeholders.
if (is_array($row)) {
continue;
}
$row = strtr($row, $replacements);
}
continue;
}
$rule = strtr($rule, $replacements);
}
}
return $rules;
}
/**
* Sets $allowEmptyInserts.
*/
public function allowEmptyInserts(bool $value = true): self
{
$this->allowEmptyInserts = $value;
return $this;
}
}