<?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\Exceptions\ModelException; use CodeIgniter\I18n\Time; use CodeIgniter\Pager\Pager; use CodeIgniter\Validation\Validation; 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 */ 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 string */ 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 array */ 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. * * @var array|string */ protected $validationRules = []; /** * Contains any custom error messages to be * used during data validation. * * @var array */ 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 in data. Used between inserts/updates. * * @var bool */ protected $cleanValidationRules = true; /** * Our validator instance. * * @var Validation */ 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 array */ protected $beforeInsert = []; /** * Callbacks for afterInsert * * @var array */ protected $afterInsert = []; /** * Callbacks for beforeUpdate * * @var array */ protected $beforeUpdate = []; /** * Callbacks for afterUpdate * * @var array */ protected $afterUpdate = []; /** * Callbacks for beforeFind * * @var array */ protected $beforeFind = []; /** * Callbacks for afterFind * * @var array */ protected $afterFind = []; /** * Callbacks for beforeDelete * * @var array */ protected $beforeDelete = []; /** * Callbacks for afterDelete * * @var array */ protected $afterDelete = []; public function __construct(?ValidationInterface $validation = null) { $this->tempReturnType = $this->returnType; $this->tempUseSoftDeletes = $this->useSoftDeletes; $this->tempAllowCallbacks = $this->allowCallbacks; /** * @var Validation $validation */ $validation = $validation ?? Services::validation(null, false); $this->validation = $validation; $this->initialize(); } /** * Initializes the instance with any additional steps. * Optionally implemented by child classes. */ protected function initialize() { } /** * Fetches the row of database * This methods 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 methods works only with dbCalls * * @param string $columnName Column Name * * @throws DataException * * @return array|null The resulting row of data, or null if no data found. */ abstract protected function doFindColumn(string $columnName); /** * Fetches all results, while optionally limiting them. * This methods 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 methods works only with dbCalls * * @return array|object|null */ abstract protected function doFirst(); /** * Inserts data into the current database * This methods works only with dbCalls * * @param array $data Data * * @return bool|int|string */ abstract protected function doInsert(array $data); /** * Compiles batch insert and runs the queries, validating each row prior. * This methods works only with dbCalls * * @param array|null $set An associative array of insert values * @param bool|null $escape Whether to escape values and identifiers * @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 methods works only with dbCalls * * @param array|int|string|null $id ID * @param array|null $data Data */ abstract protected function doUpdate($id = null, $data = null): bool; /** * Compiles an update and runs the query * This methods 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 * * @throws DatabaseException * * @return mixed Number of rows affected or FALSE on failure */ 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 methods works only with dbCalls * * @param array|int|string|null $id The rows primary key(s) * @param bool $purge Allows overriding the soft deletes setting. * * @throws DatabaseException * * @return bool|string */ 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 methods works only with dbCalls * * @return bool|mixed */ abstract protected function doPurgeDeleted(); /** * Works with the find* methods to return only the rows that * have been deleted. * This methods works only with dbCalls */ abstract protected function doOnlyDeleted(); /** * Compiles a replace and runs the query * This methods works only with dbCalls * * @param array|null $data Data * @param bool $returnSQL Set to true to return Query String * * @return mixed */ abstract protected function doReplace(?array $data = null, bool $returnSQL = false); /** * Grabs the last error(s) that occurred from the Database connection. * This methods 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 $data * * @return array|int|string|null * * @todo: Make abstract in version 5.0 */ public function getIdValue($data) { return $this->idValue($data); } /** * Override countAllResults to account for soft deleted accounts. * This methods works only with dbCalls * * @param bool $reset Reset * @param bool $test Test * * @return mixed */ abstract public function countAllResults(bool $reset = true, bool $test = false); /** * Loops over records in batches, allowing you to operate on them. * This methods works only with dbCalls * * @param int $size Size * @param Closure $userFunc Callback Function * * @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. */ 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 (! empty($eventData['returnData'])) { 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 * * @throws DataException * * @return array|null The resulting row of data, or null if no data found. */ 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 (! empty($eventData['returnData'])) { 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 (! empty($eventData['returnData'])) { 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 $data Data * * @throws ReflectionException */ public function save($data): bool { if (empty($data)) { return true; } if ($this->shouldUpdate($data)) { $response = $this->update($this->getIdValue($data), $data); } else { $response = $this->insert($data, 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 return false insert operation will be executed * * @param array|object $data Data */ protected function shouldUpdate($data): bool { return ! empty($this->getIdValue($data)); } /** * 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 $data Data * @param bool $returnID Whether insert ID should be returned or not. * * @throws ReflectionException * * @return bool|int|string */ public function insert($data = null, bool $returnID = true) { $this->insertID = 0; $data = $this->transformDataToArray($data, 'insert'); // Validate data before saving. if (! $this->skipValidation && ! $this->cleanRules()->validate($data)) { return false; } // Must be called first so we don't // strip out created_at values. $data = $this->doProtectFields($data); // doProtectFields() can further remove elements from // $data so we need to check for empty dataset again if (empty($data)) { throw DataException::forEmptyDataset('insert'); } // Set created_at and updated_at with same time $date = $this->setDate(); if ($this->useTimestamps && $this->createdField && ! array_key_exists($this->createdField, $data)) { $data[$this->createdField] = $date; } if ($this->useTimestamps && $this->updatedField && ! array_key_exists($this->updatedField, $data)) { $data[$this->updatedField] = $date; } $eventData = ['data' => $data]; 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; } /** * Compiles batch insert runs the queries, validating each row prior. * * @param array|null $set an associative array of insert values * @param bool|null $escape Whether to escape values and identifiers * @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 * * @throws ReflectionException * * @return bool|int Number of rows inserted or FALSE on failure */ public function insertBatch(?array $set = null, ?bool $escape = null, int $batchSize = 100, bool $testing = false) { if (is_array($set)) { foreach ($set as &$row) { // If $data 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->cleanRules()->validate($row)) { return false; } // Must be called first so we don't // strip out created_at values. $row = $this->doProtectFields($row); // Set created_at and updated_at with same time $date = $this->setDate(); if ($this->useTimestamps && $this->createdField && ! array_key_exists($this->createdField, $row)) { $row[$this->createdField] = $date; } if ($this->useTimestamps && $this->updatedField && ! array_key_exists($this->updatedField, $row)) { $row[$this->updatedField] = $date; } } } return $this->doInsertBatch($set, $escape, $batchSize, $testing); } /** * 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 $data * * @throws ReflectionException */ public function update($id = null, $data = null): bool { if (is_numeric($id) || is_string($id)) { $id = [$id]; } $data = $this->transformDataToArray($data, 'update'); // Validate data before saving. if (! $this->skipValidation && ! $this->cleanRules(true)->validate($data)) { return false; } // Must be called first so we don't // strip out updated_at values. $data = $this->doProtectFields($data); // doProtectFields() can further remove elements from // $data so we need to check for empty dataset again if (empty($data)) { throw DataException::forEmptyDataset('update'); } if ($this->useTimestamps && $this->updatedField && ! array_key_exists($this->updatedField, $data)) { $data[$this->updatedField] = $this->setDate(); } $eventData = [ 'id' => $id, 'data' => $data, ]; 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 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 * * @throws DatabaseException * @throws ReflectionException * * @return mixed Number of rows affected or FALSE on failure */ public function updateBatch(?array $set = null, ?string $index = null, int $batchSize = 100, bool $returnSQL = false) { if (is_array($set)) { foreach ($set as &$row) { // If $data 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, true, 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->cleanRules(true)->validate($row)) { return false; } // Save updateIndex for later $updateIndex = $row[$index] ?? null; // 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; } if ($this->useTimestamps && $this->updatedField && ! array_key_exists($this->updatedField, $row)) { $row[$this->updatedField] = $this->setDate(); } } } return $this->doUpdateBatch($set, $index, $batchSize, $returnSQL); } /** * 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. * * @throws DatabaseException * * @return BaseResult|bool */ public function delete($id = null, bool $purge = false) { 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 mixed */ 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 $data Data * @param bool $returnSQL Set to true to return Query String * * @return mixed */ public function replace(?array $data = null, bool $returnSQL = false) { // Validate data before saving. if ($data && ! $this->skipValidation && ! $this->cleanRules(true)->validate($data)) { return false; } return $this->doReplace($data, $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) { $pager = Services::pager(null, null, false); if ($segment) { $pager->setSegment($segment); } $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 = ($page - 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 insert() and update() to protect against mass assignment * vulnerabilities. * * @param array $data Data * * @throws DataException */ protected function doProtectFields(array $data): array { if (! $this->protectFields) { return $data; } if (empty($this->allowedFields)) { throw DataException::forInvalidAllowedFields(static::class); } foreach (array_keys($data) as $key) { if (! in_array($key, $this->allowedFields, true)) { unset($data[$key]); } } return $data; } /** * Sets the date or current date if null value is passed * * @param int|null $userData An optional PHP timestamp to be converted. * * @throws ModelException * * @return mixed */ protected function setDate(?int $userData = null) { $currentDate = $userData ?? time(); 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 * * @throws ModelException * * @return int|string */ 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 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 validation rules. * It could be used when you have to change default or override current validate rules. * * @param array $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) { $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 data against the validation rules (or the validation group) * specified in the class property, $validationRules. * * @param array|object $data Data */ public function validate($data): bool { $rules = $this->getValidationRules(); if ($this->skipValidation || empty($rules) || empty($data)) { return true; } // Validation requires array, so cast away. if (is_object($data)) { $data = (array) $data; } $rules = $this->cleanValidationRules ? $this->cleanValidationRules($rules, $data) : $rules; // If no data existed that needs validation // our job is done here. if (empty($rules)) { return true; } return $this->validation->setRules($rules, $this->validationMessages)->run($data, 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 = $this->validation->loadRuleGroup($rules); } 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 define 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|null $data Data */ protected function cleanValidationRules(array $rules, ?array $data = null): array { if (empty($data)) { return []; } foreach (array_keys($rules) as $field) { if (! array_key_exists($field, $data)) { 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 * * @throws DataException * * @return mixed */ protected function trigger(string $event, array $eventData) { // Ensure it's a valid event if (! isset($this->{$event}) || empty($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 an returns an array of it's public and protected * properties as an array suitable for use in creates and updates. * This method use objectToRawArray internally and does conversion * to string on all Time instances * * @param object|string $data Data * @param bool $onlyChanged Only Changed Property * @param bool $recursive If true, inner entities will be casted as array as well * * @throws ReflectionException * * @return array Array */ protected function objectToArray($data, bool $onlyChanged = true, bool $recursive = false): array { $properties = $this->objectToRawArray($data, $onlyChanged, $recursive); // Convert any Time instances to appropriate $dateFormat if ($properties) { $properties = array_map(function ($value) { if ($value instanceof Time) { return $this->timeToDate($value); } return $value; }, $properties); } return $properties; } /** * Takes a class an returns an array of it's public and protected * properties as an array with raw values. * * @param object|string $data Data * @param bool $onlyChanged Only Changed Property * @param bool $recursive If true, inner entities will be casted as array as well * * @throws ReflectionException * * @return array|null Array */ protected function objectToRawArray($data, bool $onlyChanged = true, bool $recursive = false): ?array { if (method_exists($data, 'toRawArray')) { $properties = $data->toRawArray($onlyChanged, $recursive); } else { $mirror = new ReflectionClass($data); $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($data); } } return $properties; } /** * Transform data to array * * @param array|object|null $data Data * @param string $type Type of data (insert|update) * * @throws DataException * @throws InvalidArgumentException * @throws ReflectionException */ protected function transformDataToArray($data, 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 (empty($data)) { throw DataException::forEmptyDataset($type); } // If $data 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($data) && ! $data instanceof stdClass) { $data = $this->objectToArray($data, true, 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($data)) { $data = (array) $data; } // If it's still empty here, means $data is no change or is empty object if (empty($data)) { throw DataException::forEmptyDataset($type); } return $data; } /** * Provides the db connection and model's properties. * * @param string $name Name * * @return mixed */ 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 (! empty($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; } }