Viewing file: Forge.php (30.01 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
<?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\Database;
use CodeIgniter\Database\Exceptions\DatabaseException; use InvalidArgumentException; use RuntimeException; use Throwable;
/** * The Forge class transforms migrations to executable * SQL statements. */ class Forge { /** * The active database connection. * * @var BaseConnection */ protected $db;
/** * List of fields. * * @var array */ protected $fields = [];
/** * List of keys. * * @var array */ protected $keys = [];
/** * List of unique keys. * * @var array */ protected $uniqueKeys = [];
/** * List of primary keys. * * @var array */ protected $primaryKeys = [];
/** * List of foreign keys. * * @var array */ protected $foreignKeys = [];
/** * Character set used. * * @var string */ protected $charset = '';
/** * CREATE DATABASE statement * * @var false|string */ protected $createDatabaseStr = 'CREATE DATABASE %s';
/** * CREATE DATABASE IF statement * * @var string */ protected $createDatabaseIfStr;
/** * CHECK DATABASE EXIST statement * * @var string */ protected $checkDatabaseExistStr;
/** * DROP DATABASE statement * * @var false|string */ protected $dropDatabaseStr = 'DROP DATABASE %s';
/** * CREATE TABLE statement * * @var string */ protected $createTableStr = "%s %s (%s\n)";
/** * CREATE TABLE IF statement * * @var bool|string */ protected $createTableIfStr = 'CREATE TABLE IF NOT EXISTS';
/** * CREATE TABLE keys flag * * Whether table keys are created from within the * CREATE TABLE statement. * * @var bool */ protected $createTableKeys = false;
/** * DROP TABLE IF EXISTS statement * * @var bool|string */ protected $dropTableIfStr = 'DROP TABLE IF EXISTS';
/** * RENAME TABLE statement * * @var false|string */ protected $renameTableStr = 'ALTER TABLE %s RENAME TO %s';
/** * UNSIGNED support * * @var array|bool */ protected $unsigned = true;
/** * NULL value representation in CREATE/ALTER TABLE statements * * @var string * * @internal Used for marking nullable fields. Not covered by BC promise. */ protected $null = '';
/** * DEFAULT value representation in CREATE/ALTER TABLE statements * * @var false|string */ protected $default = ' DEFAULT ';
/** * DROP CONSTRAINT statement * * @var string */ protected $dropConstraintStr;
/** * DROP INDEX statement * * @var string */ protected $dropIndexStr = 'DROP INDEX %s ON %s';
/** * Constructor. */ public function __construct(BaseConnection $db) { $this->db = $db; }
/** * Provides access to the forge's current database connection. * * @return ConnectionInterface */ public function getConnection() { return $this->db; }
/** * Create database * * @param bool $ifNotExists Whether to add IF NOT EXISTS condition * * @throws DatabaseException */ public function createDatabase(string $dbName, bool $ifNotExists = false): bool { if ($ifNotExists && $this->createDatabaseIfStr === null) { if ($this->databaseExists($dbName)) { return true; }
$ifNotExists = false; }
if ($this->createDatabaseStr === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; // @codeCoverageIgnore }
try { if (! $this->db->query(sprintf($ifNotExists ? $this->createDatabaseIfStr : $this->createDatabaseStr, $dbName, $this->db->charset, $this->db->DBCollat))) { // @codeCoverageIgnoreStart if ($this->db->DBDebug) { throw new DatabaseException('Unable to create the specified database.'); }
return false; // @codeCoverageIgnoreEnd }
if (! empty($this->db->dataCache['db_names'])) { $this->db->dataCache['db_names'][] = $dbName; }
return true; } catch (Throwable $e) { if ($this->db->DBDebug) { throw new DatabaseException('Unable to create the specified database.', 0, $e); }
return false; // @codeCoverageIgnore } }
/** * Determine if a database exists * * @throws DatabaseException */ private function databaseExists(string $dbName): bool { if ($this->checkDatabaseExistStr === null) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
return $this->db->query($this->checkDatabaseExistStr, $dbName)->getRow() !== null; }
/** * Drop database * * @throws DatabaseException */ public function dropDatabase(string $dbName): bool { if ($this->dropDatabaseStr === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
if (! $this->db->query(sprintf($this->dropDatabaseStr, $dbName))) { if ($this->db->DBDebug) { throw new DatabaseException('Unable to drop the specified database.'); }
return false; }
if (! empty($this->db->dataCache['db_names'])) { $key = array_search(strtolower($dbName), array_map('strtolower', $this->db->dataCache['db_names']), true); if ($key !== false) { unset($this->db->dataCache['db_names'][$key]); } }
return true; }
/** * Add Key * * @param array|string $key * * @return Forge */ public function addKey($key, bool $primary = false, bool $unique = false) { if ($primary) { foreach ((array) $key as $one) { $this->primaryKeys[] = $one; } } else { $this->keys[] = $key;
if ($unique) { $this->uniqueKeys[] = count($this->keys) - 1; } }
return $this; }
/** * Add Primary Key * * @param array|string $key * * @return Forge */ public function addPrimaryKey($key) { return $this->addKey($key, true); }
/** * Add Unique Key * * @param array|string $key * * @return Forge */ public function addUniqueKey($key) { return $this->addKey($key, false, true); }
/** * Add Field * * @param array|string $field * * @return Forge */ public function addField($field) { if (is_string($field)) { if ($field === 'id') { $this->addField([ 'id' => [ 'type' => 'INT', 'constraint' => 9, 'auto_increment' => true, ], ]); $this->addKey('id', true); } else { if (strpos($field, ' ') === false) { throw new InvalidArgumentException('Field information is required for that operation.'); }
$fieldName = explode(' ', $field, 2)[0]; $fieldName = trim($fieldName, '`\'"');
$this->fields[$fieldName] = $field; } }
if (is_array($field)) { foreach ($field as $idx => $f) { if (is_string($f)) { $this->addField($f);
continue; }
if (is_array($f)) { $this->fields = array_merge($this->fields, [$idx => $f]); } } }
return $this; }
/** * Add Foreign Key * * @param string|string[] $fieldName * @param string|string[] $tableField * * @throws DatabaseException * * @return Forge */ public function addForeignKey($fieldName = '', string $tableName = '', $tableField = '', string $onUpdate = '', string $onDelete = '') { $fieldName = (array) $fieldName; $tableField = (array) $tableField; $errorNames = [];
foreach ($fieldName as $name) { if (! isset($this->fields[$name])) { $errorNames[] = $name; } }
if ($errorNames !== []) { $errorNames[0] = implode(', ', $errorNames);
throw new DatabaseException(lang('Database.fieldNotExists', $errorNames)); }
$this->foreignKeys[] = [ 'field' => $fieldName, 'referenceTable' => $tableName, 'referenceField' => $tableField, 'onDelete' => strtoupper($onDelete), 'onUpdate' => strtoupper($onUpdate), ];
return $this; }
/** * Drop Key * * @throws DatabaseException * * @return bool */ public function dropKey(string $table, string $keyName) { $sql = sprintf( $this->dropIndexStr, $this->db->escapeIdentifiers($this->db->DBPrefix . $keyName), $this->db->escapeIdentifiers($this->db->DBPrefix . $table), );
if ($sql === '') { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
return $this->db->query($sql); }
/** * @throws DatabaseException * * @return BaseResult|bool|false|mixed|Query */ public function dropForeignKey(string $table, string $foreignName) { $sql = sprintf( (string) $this->dropConstraintStr, $this->db->escapeIdentifiers($this->db->DBPrefix . $table), $this->db->escapeIdentifiers($this->db->DBPrefix . $foreignName) );
if ($sql === '') { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
return $this->db->query($sql); }
/** * @throws DatabaseException * * @return mixed */ public function createTable(string $table, bool $ifNotExists = false, array $attributes = []) { if ($table === '') { throw new InvalidArgumentException('A table name is required for that operation.'); }
$table = $this->db->DBPrefix . $table;
if ($this->fields === []) { throw new RuntimeException('Field information is required.'); }
$sql = $this->_createTable($table, $ifNotExists, $attributes);
if (is_bool($sql)) { $this->reset(); if ($sql === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
return true; }
if (($result = $this->db->query($sql)) !== false) { if (isset($this->db->dataCache['table_names']) && ! in_array($table, $this->db->dataCache['table_names'], true)) { $this->db->dataCache['table_names'][] = $table; }
// Most databases don't support creating indexes from within the CREATE TABLE statement if (! empty($this->keys)) { for ($i = 0, $sqls = $this->_processIndexes($table), $c = count($sqls); $i < $c; $i++) { $this->db->query($sqls[$i]); } } }
$this->reset();
return $result; }
/** * @return bool|string */ protected function _createTable(string $table, bool $ifNotExists, array $attributes) { // For any platforms that don't support Create If Not Exists... if ($ifNotExists === true && $this->createTableIfStr === false) { if ($this->db->tableExists($table)) { return true; }
$ifNotExists = false; }
$sql = ($ifNotExists) ? sprintf($this->createTableIfStr, $this->db->escapeIdentifiers($table)) : 'CREATE TABLE';
$columns = $this->_processFields(true);
for ($i = 0, $c = count($columns); $i < $c; $i++) { $columns[$i] = ($columns[$i]['_literal'] !== false) ? "\n\t" . $columns[$i]['_literal'] : "\n\t" . $this->_processColumn($columns[$i]); }
$columns = implode(',', $columns);
$columns .= $this->_processPrimaryKeys($table); $columns .= $this->_processForeignKeys($table);
if ($this->createTableKeys === true) { $indexes = $this->_processIndexes($table); if (is_string($indexes)) { $columns .= $indexes; } }
return sprintf( $this->createTableStr . '%s', $sql, $this->db->escapeIdentifiers($table), $columns, $this->_createTableAttributes($attributes) ); }
protected function _createTableAttributes(array $attributes): string { $sql = '';
foreach (array_keys($attributes) as $key) { if (is_string($key)) { $sql .= ' ' . strtoupper($key) . ' ' . $this->db->escape($attributes[$key]); } }
return $sql; }
/** * @throws DatabaseException * * @return mixed */ public function dropTable(string $tableName, bool $ifExists = false, bool $cascade = false) { if ($tableName === '') { if ($this->db->DBDebug) { throw new DatabaseException('A table name is required for that operation.'); }
return false; }
if ($this->db->DBPrefix && strpos($tableName, $this->db->DBPrefix) === 0) { $tableName = substr($tableName, strlen($this->db->DBPrefix)); }
if (($query = $this->_dropTable($this->db->DBPrefix . $tableName, $ifExists, $cascade)) === true) { return true; }
$this->db->disableForeignKeyChecks();
$query = $this->db->query($query);
$this->db->enableForeignKeyChecks();
if ($query && ! empty($this->db->dataCache['table_names'])) { $key = array_search( strtolower($this->db->DBPrefix . $tableName), array_map('strtolower', $this->db->dataCache['table_names']), true );
if ($key !== false) { unset($this->db->dataCache['table_names'][$key]); } }
return $query; }
/** * Generates a platform-specific DROP TABLE string * * @return bool|string */ protected function _dropTable(string $table, bool $ifExists, bool $cascade) { $sql = 'DROP TABLE';
if ($ifExists) { if ($this->dropTableIfStr === false) { if (! $this->db->tableExists($table)) { return true; } } else { $sql = sprintf($this->dropTableIfStr, $this->db->escapeIdentifiers($table)); } }
return $sql . ' ' . $this->db->escapeIdentifiers($table); }
/** * @throws DatabaseException * * @return mixed */ public function renameTable(string $tableName, string $newTableName) { if ($tableName === '' || $newTableName === '') { throw new InvalidArgumentException('A table name is required for that operation.'); }
if ($this->renameTableStr === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
$result = $this->db->query(sprintf( $this->renameTableStr, $this->db->escapeIdentifiers($this->db->DBPrefix . $tableName), $this->db->escapeIdentifiers($this->db->DBPrefix . $newTableName) ));
if ($result && ! empty($this->db->dataCache['table_names'])) { $key = array_search( strtolower($this->db->DBPrefix . $tableName), array_map('strtolower', $this->db->dataCache['table_names']), true );
if ($key !== false) { $this->db->dataCache['table_names'][$key] = $this->db->DBPrefix . $newTableName; } }
return $result; }
/** * @param array|string $field * * @throws DatabaseException */ public function addColumn(string $table, $field): bool { // Work-around for literal column definitions if (! is_array($field)) { $field = [$field]; }
foreach (array_keys($field) as $k) { $this->addField([$k => $field[$k]]); }
$sqls = $this->_alterTable('ADD', $this->db->DBPrefix . $table, $this->_processFields()); $this->reset(); if ($sqls === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
foreach ($sqls as $sql) { if ($this->db->query($sql) === false) { return false; } }
return true; }
/** * @param array|string $columnName * * @throws DatabaseException * * @return mixed */ public function dropColumn(string $table, $columnName) { $sql = $this->_alterTable('DROP', $this->db->DBPrefix . $table, $columnName); if ($sql === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
return $this->db->query($sql); }
/** * @param array|string $field * * @throws DatabaseException */ public function modifyColumn(string $table, $field): bool { // Work-around for literal column definitions if (! is_array($field)) { $field = [$field]; }
foreach (array_keys($field) as $k) { $this->addField([$k => $field[$k]]); }
if ($this->fields === []) { throw new RuntimeException('Field information is required'); }
$sqls = $this->_alterTable('CHANGE', $this->db->DBPrefix . $table, $this->_processFields()); $this->reset(); if ($sqls === false) { if ($this->db->DBDebug) { throw new DatabaseException('This feature is not available for the database you are using.'); }
return false; }
if ($sqls !== null) { foreach ($sqls as $sql) { if ($this->db->query($sql) === false) { return false; } } }
return true; }
/** * @param mixed $fields * * @return false|string|string[] */ protected function _alterTable(string $alterType, string $table, $fields) { $sql = 'ALTER TABLE ' . $this->db->escapeIdentifiers($table) . ' ';
// DROP has everything it needs now. if ($alterType === 'DROP') { if (is_string($fields)) { $fields = explode(',', $fields); }
$fields = array_map(fn ($field) => 'DROP COLUMN ' . $this->db->escapeIdentifiers(trim($field)), $fields);
return $sql . implode(', ', $fields); }
$sql .= ($alterType === 'ADD') ? 'ADD ' : $alterType . ' COLUMN ';
$sqls = [];
foreach ($fields as $data) { $sqls[] = $sql . ($data['_literal'] !== false ? $data['_literal'] : $this->_processColumn($data)); }
return $sqls; }
/** * Process fields */ protected function _processFields(bool $createTable = false): array { $fields = [];
foreach ($this->fields as $key => $attributes) { if (! is_array($attributes)) { $fields[] = ['_literal' => $attributes];
continue; }
$attributes = array_change_key_case($attributes, CASE_UPPER);
if ($createTable === true && empty($attributes['TYPE'])) { continue; }
if (isset($attributes['TYPE'])) { $this->_attributeType($attributes); }
$field = [ 'name' => $key, 'new_name' => $attributes['NAME'] ?? null, 'type' => $attributes['TYPE'] ?? null, 'length' => '', 'unsigned' => '', 'null' => '', 'unique' => '', 'default' => '', 'auto_increment' => '', '_literal' => false, ];
if (isset($attributes['TYPE'])) { $this->_attributeUnsigned($attributes, $field); }
if ($createTable === false) { if (isset($attributes['AFTER'])) { $field['after'] = $attributes['AFTER']; } elseif (isset($attributes['FIRST'])) { $field['first'] = (bool) $attributes['FIRST']; } }
$this->_attributeDefault($attributes, $field);
if (isset($attributes['NULL'])) { if ($attributes['NULL'] === true) { $field['null'] = empty($this->null) ? '' : ' ' . $this->null; } else { $field['null'] = ' NOT NULL'; } } elseif ($createTable === true) { $field['null'] = ' NOT NULL'; }
$this->_attributeAutoIncrement($attributes, $field); $this->_attributeUnique($attributes, $field);
if (isset($attributes['COMMENT'])) { $field['comment'] = $this->db->escape($attributes['COMMENT']); }
if (isset($attributes['TYPE']) && ! empty($attributes['CONSTRAINT'])) { if (is_array($attributes['CONSTRAINT'])) { $attributes['CONSTRAINT'] = $this->db->escape($attributes['CONSTRAINT']); $attributes['CONSTRAINT'] = implode(',', $attributes['CONSTRAINT']); }
$field['length'] = '(' . $attributes['CONSTRAINT'] . ')'; }
$fields[] = $field; }
return $fields; }
/** * Process column */ protected function _processColumn(array $field): string { return $this->db->escapeIdentifiers($field['name']) . ' ' . $field['type'] . $field['length'] . $field['unsigned'] . $field['default'] . $field['null'] . $field['auto_increment'] . $field['unique']; }
/** * Performs a data type mapping between different databases. */ protected function _attributeType(array &$attributes) { // Usually overridden by drivers }
/** * Depending on the unsigned property value: * * - TRUE will always set $field['unsigned'] to 'UNSIGNED' * - FALSE will always set $field['unsigned'] to '' * - array(TYPE) will set $field['unsigned'] to 'UNSIGNED', * if $attributes['TYPE'] is found in the array * - array(TYPE => UTYPE) will change $field['type'], * from TYPE to UTYPE in case of a match */ protected function _attributeUnsigned(array &$attributes, array &$field) { if (empty($attributes['UNSIGNED']) || $attributes['UNSIGNED'] !== true) { return; }
// Reset the attribute in order to avoid issues if we do type conversion $attributes['UNSIGNED'] = false;
if (is_array($this->unsigned)) { foreach (array_keys($this->unsigned) as $key) { if (is_int($key) && strcasecmp($attributes['TYPE'], $this->unsigned[$key]) === 0) { $field['unsigned'] = ' UNSIGNED';
return; }
if (is_string($key) && strcasecmp($attributes['TYPE'], $key) === 0) { $field['type'] = $key;
return; } }
return; }
$field['unsigned'] = ($this->unsigned === true) ? ' UNSIGNED' : ''; }
protected function _attributeDefault(array &$attributes, array &$field) { if ($this->default === false) { return; }
if (array_key_exists('DEFAULT', $attributes)) { if ($attributes['DEFAULT'] === null) { $field['default'] = empty($this->null) ? '' : $this->default . $this->null;
// Override the NULL attribute if that's our default $attributes['NULL'] = true; $field['null'] = empty($this->null) ? '' : ' ' . $this->null; } elseif ($attributes['DEFAULT'] instanceof RawSql) { $field['default'] = $this->default . $attributes['DEFAULT']; } else { $field['default'] = $this->default . $this->db->escape($attributes['DEFAULT']); } } }
protected function _attributeUnique(array &$attributes, array &$field) { if (! empty($attributes['UNIQUE']) && $attributes['UNIQUE'] === true) { $field['unique'] = ' UNIQUE'; } }
protected function _attributeAutoIncrement(array &$attributes, array &$field) { if (! empty($attributes['AUTO_INCREMENT']) && $attributes['AUTO_INCREMENT'] === true && stripos($field['type'], 'int') !== false ) { $field['auto_increment'] = ' AUTO_INCREMENT'; } }
protected function _processPrimaryKeys(string $table): string { $sql = '';
for ($i = 0, $c = count($this->primaryKeys); $i < $c; $i++) { if (! isset($this->fields[$this->primaryKeys[$i]])) { unset($this->primaryKeys[$i]); } }
if ($this->primaryKeys !== []) { $sql .= ",\n\tCONSTRAINT " . $this->db->escapeIdentifiers('pk_' . $table) . ' PRIMARY KEY(' . implode(', ', $this->db->escapeIdentifiers($this->primaryKeys)) . ')'; }
return $sql; }
protected function _processIndexes(string $table) { $sqls = [];
for ($i = 0, $c = count($this->keys); $i < $c; $i++) { $this->keys[$i] = (array) $this->keys[$i];
for ($i2 = 0, $c2 = count($this->keys[$i]); $i2 < $c2; $i2++) { if (! isset($this->fields[$this->keys[$i][$i2]])) { unset($this->keys[$i][$i2]); } }
if (count($this->keys[$i]) <= 0) { continue; }
if (in_array($i, $this->uniqueKeys, true)) { $sqls[] = 'ALTER TABLE ' . $this->db->escapeIdentifiers($table) . ' ADD CONSTRAINT ' . $this->db->escapeIdentifiers($table . '_' . implode('_', $this->keys[$i])) . ' UNIQUE (' . implode(', ', $this->db->escapeIdentifiers($this->keys[$i])) . ')';
continue; }
$sqls[] = 'CREATE INDEX ' . $this->db->escapeIdentifiers($table . '_' . implode('_', $this->keys[$i])) . ' ON ' . $this->db->escapeIdentifiers($table) . ' (' . implode(', ', $this->db->escapeIdentifiers($this->keys[$i])) . ')'; }
return $sqls; }
protected function _processForeignKeys(string $table): string { $sql = '';
$allowActions = [ 'CASCADE', 'SET NULL', 'NO ACTION', 'RESTRICT', 'SET DEFAULT', ];
foreach ($this->foreignKeys as $fkey) { $nameIndex = $table . '_' . implode('_', $fkey['field']) . '_foreign'; $nameIndexFilled = $this->db->escapeIdentifiers($nameIndex); $foreignKeyFilled = implode(', ', $this->db->escapeIdentifiers($fkey['field'])); $referenceTableFilled = $this->db->escapeIdentifiers($this->db->DBPrefix . $fkey['referenceTable']); $referenceFieldFilled = implode(', ', $this->db->escapeIdentifiers($fkey['referenceField']));
$formatSql = ",\n\tCONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s(%s)"; $sql .= sprintf($formatSql, $nameIndexFilled, $foreignKeyFilled, $referenceTableFilled, $referenceFieldFilled);
if ($fkey['onDelete'] !== false && in_array($fkey['onDelete'], $allowActions, true)) { $sql .= ' ON DELETE ' . $fkey['onDelete']; }
if ($fkey['onUpdate'] !== false && in_array($fkey['onUpdate'], $allowActions, true)) { $sql .= ' ON UPDATE ' . $fkey['onUpdate']; } }
return $sql; }
/** * Resets table creation vars */ public function reset() { $this->fields = $this->keys = $this->uniqueKeys = $this->primaryKeys = $this->foreignKeys = []; } }
|