123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390 |
- <?php
- /*
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * This software consists of voluntary contributions made by many individuals
- * and is licensed under the MIT license. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\DBAL\Schema;
- use Doctrine\DBAL\DBALException;
- /**
- * Sqlite SchemaManager.
- *
- * @author Konsta Vesterinen <kvesteri@cc.hut.fi>
- * @author Lukas Smith <smith@pooteeweet.org> (PEAR MDB2 library)
- * @author Jonathan H. Wage <jonwage@gmail.com>
- * @author Martin Hasoň <martin.hason@gmail.com>
- * @since 2.0
- */
- class SqliteSchemaManager extends AbstractSchemaManager
- {
- /**
- * {@inheritdoc}
- */
- public function dropDatabase($database)
- {
- if (file_exists($database)) {
- unlink($database);
- }
- }
- /**
- * {@inheritdoc}
- */
- public function createDatabase($database)
- {
- $params = $this->_conn->getParams();
- $driver = $params['driver'];
- $options = array(
- 'driver' => $driver,
- 'path' => $database
- );
- $conn = \Doctrine\DBAL\DriverManager::getConnection($options);
- $conn->connect();
- $conn->close();
- }
- /**
- * {@inheritdoc}
- */
- public function renameTable($name, $newName)
- {
- $tableDiff = new TableDiff($name);
- $tableDiff->fromTable = $this->listTableDetails($name);
- $tableDiff->newName = $newName;
- $this->alterTable($tableDiff);
- }
- /**
- * {@inheritdoc}
- */
- public function createForeignKey(ForeignKeyConstraint $foreignKey, $table)
- {
- $tableDiff = $this->getTableDiffForAlterForeignKey($foreignKey, $table);
- $tableDiff->addedForeignKeys[] = $foreignKey;
- $this->alterTable($tableDiff);
- }
- /**
- * {@inheritdoc}
- */
- public function dropAndCreateForeignKey(ForeignKeyConstraint $foreignKey, $table)
- {
- $tableDiff = $this->getTableDiffForAlterForeignKey($foreignKey, $table);
- $tableDiff->changedForeignKeys[] = $foreignKey;
- $this->alterTable($tableDiff);
- }
- /**
- * {@inheritdoc}
- */
- public function dropForeignKey($foreignKey, $table)
- {
- $tableDiff = $this->getTableDiffForAlterForeignKey($foreignKey, $table);
- $tableDiff->removedForeignKeys[] = $foreignKey;
- $this->alterTable($tableDiff);
- }
- /**
- * {@inheritdoc}
- */
- public function listTableForeignKeys($table, $database = null)
- {
- if (null === $database) {
- $database = $this->_conn->getDatabase();
- }
- $sql = $this->_platform->getListTableForeignKeysSQL($table, $database);
- $tableForeignKeys = $this->_conn->fetchAll($sql);
- if ( ! empty($tableForeignKeys)) {
- $createSql = $this->_conn->fetchAll("SELECT sql FROM (SELECT * FROM sqlite_master UNION ALL SELECT * FROM sqlite_temp_master) WHERE type = 'table' AND name = '$table'");
- $createSql = isset($createSql[0]['sql']) ? $createSql[0]['sql'] : '';
- if (preg_match_all('#
- (?:CONSTRAINT\s+([^\s]+)\s+)?
- (?:FOREIGN\s+KEY[^\)]+\)\s*)?
- REFERENCES\s+[^\s]+\s+(?:\([^\)]+\))?
- (?:
- [^,]*?
- (NOT\s+DEFERRABLE|DEFERRABLE)
- (?:\s+INITIALLY\s+(DEFERRED|IMMEDIATE))?
- )?#isx',
- $createSql, $match)) {
- $names = array_reverse($match[1]);
- $deferrable = array_reverse($match[2]);
- $deferred = array_reverse($match[3]);
- } else {
- $names = $deferrable = $deferred = array();
- }
- foreach ($tableForeignKeys as $key => $value) {
- $id = $value['id'];
- $tableForeignKeys[$key]['constraint_name'] = isset($names[$id]) && '' != $names[$id] ? $names[$id] : $id;
- $tableForeignKeys[$key]['deferrable'] = isset($deferrable[$id]) && 'deferrable' == strtolower($deferrable[$id]) ? true : false;
- $tableForeignKeys[$key]['deferred'] = isset($deferred[$id]) && 'deferred' == strtolower($deferred[$id]) ? true : false;
- }
- }
- return $this->_getPortableTableForeignKeysList($tableForeignKeys);
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableTableDefinition($table)
- {
- return $table['name'];
- }
- /**
- * {@inheritdoc}
- *
- * @license New BSD License
- * @link http://ezcomponents.org/docs/api/trunk/DatabaseSchema/ezcDbSchemaPgsqlReader.html
- */
- protected function _getPortableTableIndexesList($tableIndexes, $tableName=null)
- {
- $indexBuffer = array();
- // fetch primary
- $stmt = $this->_conn->executeQuery( "PRAGMA TABLE_INFO ('$tableName')" );
- $indexArray = $stmt->fetchAll(\PDO::FETCH_ASSOC);
- foreach($indexArray as $indexColumnRow) {
- if($indexColumnRow['pk'] == "1") {
- $indexBuffer[] = array(
- 'key_name' => 'primary',
- 'primary' => true,
- 'non_unique' => false,
- 'column_name' => $indexColumnRow['name']
- );
- }
- }
- // fetch regular indexes
- foreach($tableIndexes as $tableIndex) {
- // Ignore indexes with reserved names, e.g. autoindexes
- if (strpos($tableIndex['name'], 'sqlite_') !== 0) {
- $keyName = $tableIndex['name'];
- $idx = array();
- $idx['key_name'] = $keyName;
- $idx['primary'] = false;
- $idx['non_unique'] = $tableIndex['unique']?false:true;
- $stmt = $this->_conn->executeQuery( "PRAGMA INDEX_INFO ( '{$keyName}' )" );
- $indexArray = $stmt->fetchAll(\PDO::FETCH_ASSOC);
- foreach ( $indexArray as $indexColumnRow ) {
- $idx['column_name'] = $indexColumnRow['name'];
- $indexBuffer[] = $idx;
- }
- }
- }
- return parent::_getPortableTableIndexesList($indexBuffer, $tableName);
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableTableIndexDefinition($tableIndex)
- {
- return array(
- 'name' => $tableIndex['name'],
- 'unique' => (bool) $tableIndex['unique']
- );
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableTableColumnList($table, $database, $tableColumns)
- {
- $list = parent::_getPortableTableColumnList($table, $database, $tableColumns);
- $autoincrementColumn = null;
- $autoincrementCount = 0;
- foreach ($tableColumns as $tableColumn) {
- if ('1' == $tableColumn['pk']) {
- $autoincrementCount++;
- if (null === $autoincrementColumn && 'integer' == strtolower($tableColumn['type'])) {
- $autoincrementColumn = $tableColumn['name'];
- }
- }
- }
- if (1 == $autoincrementCount && null !== $autoincrementColumn) {
- foreach ($list as $column) {
- if ($autoincrementColumn == $column->getName()) {
- $column->setAutoincrement(true);
- }
- }
- }
- return $list;
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableTableColumnDefinition($tableColumn)
- {
- $e = explode('(', $tableColumn['type']);
- $tableColumn['type'] = $e[0];
- if (isset($e[1])) {
- $length = trim($e[1], ')');
- $tableColumn['length'] = $length;
- }
- $dbType = strtolower($tableColumn['type']);
- $length = isset($tableColumn['length']) ? $tableColumn['length'] : null;
- $unsigned = (boolean) isset($tableColumn['unsigned']) ? $tableColumn['unsigned'] : false;
- $fixed = false;
- $type = $this->_platform->getDoctrineTypeMapping($dbType);
- $default = $tableColumn['dflt_value'];
- if ($default == 'NULL') {
- $default = null;
- }
- if ($default !== null) {
- // SQLite returns strings wrapped in single quotes, so we need to strip them
- $default = preg_replace("/^'(.*)'$/", '\1', $default);
- }
- $notnull = (bool) $tableColumn['notnull'];
- if ( ! isset($tableColumn['name'])) {
- $tableColumn['name'] = '';
- }
- $precision = null;
- $scale = null;
- switch ($dbType) {
- case 'char':
- $fixed = true;
- break;
- case 'float':
- case 'double':
- case 'real':
- case 'decimal':
- case 'numeric':
- if (isset($tableColumn['length'])) {
- if (strpos($tableColumn['length'], ',') === false) {
- $tableColumn['length'] .= ",0";
- }
- list($precision, $scale) = array_map('trim', explode(',', $tableColumn['length']));
- }
- $length = null;
- break;
- }
- $options = array(
- 'length' => $length,
- 'unsigned' => (bool) $unsigned,
- 'fixed' => $fixed,
- 'notnull' => $notnull,
- 'default' => $default,
- 'precision' => $precision,
- 'scale' => $scale,
- 'autoincrement' => false,
- );
- return new Column($tableColumn['name'], \Doctrine\DBAL\Types\Type::getType($type), $options);
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableViewDefinition($view)
- {
- return new View($view['name'], $view['sql']);
- }
- /**
- * {@inheritdoc}
- */
- protected function _getPortableTableForeignKeysList($tableForeignKeys)
- {
- $list = array();
- foreach ($tableForeignKeys as $value) {
- $value = array_change_key_case($value, CASE_LOWER);
- $name = $value['constraint_name'];
- if ( ! isset($list[$name])) {
- if ( ! isset($value['on_delete']) || $value['on_delete'] == "RESTRICT") {
- $value['on_delete'] = null;
- }
- if ( ! isset($value['on_update']) || $value['on_update'] == "RESTRICT") {
- $value['on_update'] = null;
- }
- $list[$name] = array(
- 'name' => $name,
- 'local' => array(),
- 'foreign' => array(),
- 'foreignTable' => $value['table'],
- 'onDelete' => $value['on_delete'],
- 'onUpdate' => $value['on_update'],
- 'deferrable' => $value['deferrable'],
- 'deferred'=> $value['deferred'],
- );
- }
- $list[$name]['local'][] = $value['from'];
- $list[$name]['foreign'][] = $value['to'];
- }
- $result = array();
- foreach($list as $constraint) {
- $result[] = new ForeignKeyConstraint(
- array_values($constraint['local']), $constraint['foreignTable'],
- array_values($constraint['foreign']), $constraint['name'],
- array(
- 'onDelete' => $constraint['onDelete'],
- 'onUpdate' => $constraint['onUpdate'],
- 'deferrable' => $constraint['deferrable'],
- 'deferred'=> $constraint['deferred'],
- )
- );
- }
- return $result;
- }
- /**
- * @param \Doctrine\DBAL\Schema\ForeignKeyConstraint $foreignKey
- * @param \Doctrine\DBAL\Schema\Table|string $table
- *
- * @return \Doctrine\DBAL\Schema\TableDiff
- *
- * @throws \Doctrine\DBAL\DBALException
- */
- private function getTableDiffForAlterForeignKey(ForeignKeyConstraint $foreignKey, $table)
- {
- if ( ! $table instanceof Table) {
- $tableDetails = $this->tryMethod('listTableDetails', $table);
- if (false === $table) {
- throw new DBALException(sprintf('Sqlite schema manager requires to modify foreign keys table definition "%s".', $table));
- }
- $table = $tableDetails;
- }
- $tableDiff = new TableDiff($table->getName());
- $tableDiff->fromTable = $table;
- return $tableDiff;
- }
- }
|