<?php
namespace CodeIgniter\Database;
use CodeIgniter\Database\Exceptions\DatabaseException;
class Forge
{
protected $db;
protected $fields = [];
protected $keys = [];
protected $uniqueKeys = [];
protected $primaryKeys = [];
protected $foreignKeys = [];
protected $charset = '';
protected $createDatabaseStr = 'CREATE DATABASE %s';
protected $createDatabaseIfStr = null;
protected $checkDatabaseExistStr = null;
protected $dropDatabaseStr = 'DROP DATABASE %s';
protected $createTableStr = "%s %s (%s\n)";
protected $createTableIfStr = 'CREATE TABLE IF NOT EXISTS';
protected $createTableKeys = false;
protected $dropTableIfStr = 'DROP TABLE IF EXISTS';
protected $renameTableStr = 'ALTER TABLE %s RENAME TO %s;';
protected $unsigned = true;
protected $null = '';
protected $default = ' DEFAULT ';
public function __construct(ConnectionInterface $db)
{
$this->db = &$db;
}
public function getConnection()
{
return $this->db;
}
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;
}
elseif (! $this->db->query(sprintf($ifNotExists ? $this->createDatabaseIfStr : $this->createDatabaseStr, $dbName, $this->db->charset, $this->db->DBCollat))
)
{
if ($this->db->DBDebug)
{
throw new DatabaseException('Unable to create the specified database.');
}
return false;
}
if (! empty($this->db->dataCache['db_names']))
{
$this->db->dataCache['db_names'][] = $dbName;
}
return true;
}
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;
}
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;
}
elseif (! $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;
}
public function addKey($key, bool $primary = false, bool $unique = false)
{
if ($primary === true)
{
foreach ((array)$key as $one)
{
$this->primaryKeys[] = $one;
}
}
else
{
$this->keys[] = $key;
if ($unique === true)
{
$this->uniqueKeys[] = ($c = count($this->keys)) ? $c - 1 : 0;
}
}
return $this;
}
public function addPrimaryKey($key)
{
return $this->addKey($key, true);
}
public function addUniqueKey($key)
{
return $this->addKey($key, false, true);
}
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.');
}
$this->fields[] = $field;
}
}
if (is_array($field))
{
$this->fields = array_merge($this->fields, $field);
}
return $this;
}
public function addForeignKey(string $fieldName = '', string $tableName = '', string $tableField = '', string $onUpdate = '', string $onDelete = '')
{
if (! isset($this->fields[$fieldName]))
{
throw new DatabaseException(lang('Database.fieldNotExists', [$fieldName]));
}
$this->foreignKeys[$fieldName] = [
'table' => $tableName,
'field' => $tableField,
'onDelete' => strtoupper($onDelete),
'onUpdate' => strtoupper($onUpdate),
];
return $this;
}
public function dropForeignKey(string $table, string $foreignName)
{
$sql = sprintf($this->dropConstraintStr, $this->db->escapeIdentifiers($this->db->DBPrefix . $table),
$this->db->escapeIdentifiers($this->db->DBPrefix . $foreignName));
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);
}
public function createTable(string $table, bool $if_not_exists = false, array $attributes = [])
{
if ($table === '')
{
throw new \InvalidArgumentException('A table name is required for that operation.');
}
$table = $this->db->DBPrefix . $table;
if (count($this->fields) === 0)
{
throw new \RuntimeException('Field information is required.');
}
$sql = $this->_createTable($table, $if_not_exists, $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;
}
}
if (($result = $this->db->query($sql)) !== false)
{
if (! isset($this->db->dataCache['table_names'][$table]))
{
$this->db->dataCache['table_names'][] = $table;
}
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;
}
protected function _createTable(string $table, bool $if_not_exists, array $attributes)
{
if ($if_not_exists === true && $this->createTableIfStr === false)
{
if ($this->db->tableExists($table))
{
return true;
}
$if_not_exists = false;
}
$sql = ($if_not_exists) ? 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)
{
$columns .= $this->_processIndexes($table);
}
$sql = sprintf($this->createTableStr . '%s', $sql, $this->db->escapeIdentifiers($table), $columns,
$this->_createTableAttributes($attributes));
return $sql;
}
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;
}
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;
}
protected function _dropTable(string $table, bool $if_exists, bool $cascade): string
{
$sql = 'DROP TABLE';
if ($if_exists)
{
if ($this->dropTableIfStr === false)
{
if (! $this->db->tableExists($table))
{
return true;
}
}
else
{
$sql = sprintf($this->dropTableIfStr, $this->db->escapeIdentifiers($table));
}
}
$sql = $sql . ' ' . $this->db->escapeIdentifiers($table);
return $sql;
}
public function renameTable(string $table_name, string $new_table_name)
{
if ($table_name === '' || $new_table_name === '')
{
throw new \InvalidArgumentException('A table name is required for that operation.');
}
elseif ($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 . $table_name),
$this->db->escapeIdentifiers($this->db->DBPrefix . $new_table_name))
);
if ($result && ! empty($this->db->dataCache['table_names']))
{
$key = array_search(strtolower($this->db->DBPrefix . $table_name),
array_map('strtolower', $this->db->dataCache['table_names']), true);
if ($key !== false)
{
$this->db->dataCache['table_names'][$key] = $this->db->DBPrefix . $new_table_name;
}
}
return $result;
}
public function addColumn(string $table, $field): bool
{
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;
}
for ($i = 0, $c = count($sqls); $i < $c; $i++)
{
if ($this->db->query($sqls[$i]) === false)
{
return false;
}
}
return true;
}
public function dropColumn(string $table, string $column_name)
{
$sql = $this->_alterTable('DROP', $this->db->DBPrefix . $table, $column_name);
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);
}
public function modifyColumn(string $table, $field): bool
{
is_array($field) || $field = [$field];
foreach (array_keys($field) as $k)
{
$this->addField([$k => $field[$k]]);
}
if (count($this->fields) === 0)
{
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)
{
for ($i = 0, $c = count($sqls); $i < $c; $i++)
{
if ($this->db->query($sqls[$i]) === false)
{
return false;
}
}
}
return true;
}
protected function _alterTable(string $alter_type, string $table, $field)
{
$sql = 'ALTER TABLE ' . $this->db->escapeIdentifiers($table) . ' ';
if ($alter_type === 'DROP')
{
return $sql . 'DROP COLUMN ' . $this->db->escapeIdentifiers($field);
}
$sql .= ($alter_type === 'ADD') ? 'ADD ' : $alter_type . ' COLUMN ';
$sqls = [];
foreach ($field as $data)
{
$sqls[] = $sql
. ($data['_literal'] !== false ? $data['_literal'] : $this->_processColumn($data));
}
return $sqls;
}
protected function _processFields(bool $create_table = false): array
{
$fields = [];
foreach ($this->fields as $key => $attributes)
{
if (is_int($key) && ! is_array($attributes))
{
$fields[] = ['_literal' => $attributes];
continue;
}
$attributes = array_change_key_case($attributes, CASE_UPPER);
if ($create_table === true && empty($attributes['TYPE']))
{
continue;
}
isset($attributes['TYPE']) && $this->_attributeType($attributes);
$field = [
'name' => $key,
'new_name' => isset($attributes['NAME']) ? $attributes['NAME'] : null,
'type' => isset($attributes['TYPE']) ? $attributes['TYPE'] : null,
'length' => '',
'unsigned' => '',
'null' => '',
'unique' => '',
'default' => '',
'auto_increment' => '',
'_literal' => false,
];
isset($attributes['TYPE']) && $this->_attributeUnsigned($attributes, $field);
if ($create_table === 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 ($create_table === 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;
}
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'];
}
protected function _attributeType(array &$attributes)
{
}
protected function _attributeUnsigned(array &$attributes, array &$field)
{
if (empty($attributes['UNSIGNED']) || $attributes['UNSIGNED'] !== true)
{
return;
}
$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;
}
elseif (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;
$attributes['NULL'] = true;
$field['null'] = empty($this->null) ? '' : ' ' . $this->null;
}
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 (count($this->primaryKeys) > 0)
{
$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))
{
$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',
];
if (count($this->foreignKeys) > 0)
{
foreach ($this->foreignKeys as $field => $fkey)
{
$name_index = $table . '_' . $field . '_foreign';
$sql .= ",\n\tCONSTRAINT " . $this->db->escapeIdentifiers($name_index)
. ' FOREIGN KEY(' . $this->db->escapeIdentifiers($field) . ') REFERENCES ' . $this->db->escapeIdentifiers($this->db->DBPrefix . $fkey['table']) . ' (' . $this->db->escapeIdentifiers($fkey['field']) . ')';
if ($fkey['onDelete'] !== false && in_array($fkey['onDelete'], $allowActions))
{
$sql .= ' ON DELETE ' . $fkey['onDelete'];
}
if ($fkey['onUpdate'] !== false && in_array($fkey['onUpdate'], $allowActions))
{
$sql .= ' ON UPDATE ' . $fkey['onUpdate'];
}
}
}
return $sql;
}
public function reset()
{
$this->fields = $this->keys = $this->uniqueKeys = $this->primaryKeys = $this->foreignKeys = [];
}
}