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.
348 lines
8.7 KiB
348 lines
8.7 KiB
<?php |
|
/** |
|
* @copyright Copyright (c) 2016, ownCloud, Inc. |
|
* |
|
* @author Bart Visscher <bartv@thisnet.nl> |
|
* @author Jörn Friedrich Dreyer <jfd@butonic.de> |
|
* @author Morris Jobke <hey@morrisjobke.de> |
|
* @author Oliver Gasser <oliver.gasser@gmail.com> |
|
* @author Robin Appelman <robin@icewind.nl> |
|
* @author Robin McCorkell <robin@mccorkell.me.uk> |
|
* @author Thomas Müller <thomas.mueller@tmit.eu> |
|
* @author Victor Dubiniuk <dubiniuk@owncloud.com> |
|
* @author Vincent Petry <pvince81@owncloud.com> |
|
* |
|
* @license AGPL-3.0 |
|
* |
|
* This code is free software: you can redistribute it and/or modify |
|
* it under the terms of the GNU Affero General Public License, version 3, |
|
* as published by the Free Software Foundation. |
|
* |
|
* This program is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
* GNU Affero General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Affero General Public License, version 3, |
|
* along with this program. If not, see <http://www.gnu.org/licenses/> |
|
* |
|
*/ |
|
|
|
namespace OC\DB; |
|
|
|
use Doctrine\DBAL\Platforms\AbstractPlatform; |
|
use Doctrine\DBAL\Schema\Schema; |
|
use OCP\IConfig; |
|
|
|
class MDB2SchemaReader { |
|
|
|
/** |
|
* @var string $DBTABLEPREFIX |
|
*/ |
|
protected $DBTABLEPREFIX; |
|
|
|
/** |
|
* @var \Doctrine\DBAL\Platforms\AbstractPlatform $platform |
|
*/ |
|
protected $platform; |
|
|
|
/** @var IConfig */ |
|
protected $config; |
|
|
|
/** |
|
* @param \OCP\IConfig $config |
|
* @param \Doctrine\DBAL\Platforms\AbstractPlatform $platform |
|
*/ |
|
public function __construct(IConfig $config, AbstractPlatform $platform) { |
|
$this->platform = $platform; |
|
$this->config = $config; |
|
$this->DBTABLEPREFIX = $config->getSystemValue('dbtableprefix', 'oc_'); |
|
} |
|
|
|
/** |
|
* @param string $file |
|
* @param Schema $schema |
|
* @return Schema |
|
* @throws \DomainException |
|
*/ |
|
public function loadSchemaFromFile($file, Schema $schema) { |
|
$loadEntities = libxml_disable_entity_loader(false); |
|
$xml = simplexml_load_file($file); |
|
libxml_disable_entity_loader($loadEntities); |
|
foreach ($xml->children() as $child) { |
|
/** |
|
* @var \SimpleXMLElement $child |
|
*/ |
|
switch ($child->getName()) { |
|
case 'name': |
|
case 'create': |
|
case 'overwrite': |
|
case 'charset': |
|
break; |
|
case 'table': |
|
$this->loadTable($schema, $child); |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $child->getName()); |
|
|
|
} |
|
} |
|
return $schema; |
|
} |
|
|
|
/** |
|
* @param \Doctrine\DBAL\Schema\Schema $schema |
|
* @param \SimpleXMLElement $xml |
|
* @throws \DomainException |
|
*/ |
|
private function loadTable($schema, $xml) { |
|
$table = null; |
|
foreach ($xml->children() as $child) { |
|
/** |
|
* @var \SimpleXMLElement $child |
|
*/ |
|
switch ($child->getName()) { |
|
case 'name': |
|
$name = (string)$child; |
|
$name = str_replace('*dbprefix*', $this->DBTABLEPREFIX, $name); |
|
$name = $this->platform->quoteIdentifier($name); |
|
$table = $schema->createTable($name); |
|
break; |
|
case 'create': |
|
case 'overwrite': |
|
case 'charset': |
|
break; |
|
case 'declaration': |
|
if (is_null($table)) { |
|
throw new \DomainException('Table declaration before table name'); |
|
} |
|
$this->loadDeclaration($table, $child); |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $child->getName()); |
|
|
|
} |
|
} |
|
} |
|
|
|
/** |
|
* @param \Doctrine\DBAL\Schema\Table $table |
|
* @param \SimpleXMLElement $xml |
|
* @throws \DomainException |
|
*/ |
|
private function loadDeclaration($table, $xml) { |
|
foreach ($xml->children() as $child) { |
|
/** |
|
* @var \SimpleXMLElement $child |
|
*/ |
|
switch ($child->getName()) { |
|
case 'field': |
|
$this->loadField($table, $child); |
|
break; |
|
case 'index': |
|
$this->loadIndex($table, $child); |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $child->getName()); |
|
|
|
} |
|
} |
|
} |
|
|
|
/** |
|
* @param \Doctrine\DBAL\Schema\Table $table |
|
* @param \SimpleXMLElement $xml |
|
* @throws \DomainException |
|
*/ |
|
private function loadField($table, $xml) { |
|
$options = array( 'notnull' => false ); |
|
foreach ($xml->children() as $child) { |
|
/** |
|
* @var \SimpleXMLElement $child |
|
*/ |
|
switch ($child->getName()) { |
|
case 'name': |
|
$name = (string)$child; |
|
$name = $this->platform->quoteIdentifier($name); |
|
break; |
|
case 'type': |
|
$type = (string)$child; |
|
switch ($type) { |
|
case 'text': |
|
$type = 'string'; |
|
break; |
|
case 'clob': |
|
$type = 'text'; |
|
break; |
|
case 'timestamp': |
|
$type = 'datetime'; |
|
break; |
|
case 'numeric': |
|
$type = 'decimal'; |
|
break; |
|
} |
|
break; |
|
case 'length': |
|
$length = (string)$child; |
|
$options['length'] = $length; |
|
break; |
|
case 'unsigned': |
|
$unsigned = $this->asBool($child); |
|
$options['unsigned'] = $unsigned; |
|
break; |
|
case 'notnull': |
|
$notnull = $this->asBool($child); |
|
$options['notnull'] = $notnull; |
|
break; |
|
case 'autoincrement': |
|
$autoincrement = $this->asBool($child); |
|
$options['autoincrement'] = $autoincrement; |
|
break; |
|
case 'default': |
|
$default = (string)$child; |
|
$options['default'] = $default; |
|
break; |
|
case 'comments': |
|
$comment = (string)$child; |
|
$options['comment'] = $comment; |
|
break; |
|
case 'primary': |
|
$primary = $this->asBool($child); |
|
$options['primary'] = $primary; |
|
break; |
|
case 'precision': |
|
$precision = (string)$child; |
|
$options['precision'] = $precision; |
|
break; |
|
case 'scale': |
|
$scale = (string)$child; |
|
$options['scale'] = $scale; |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $child->getName()); |
|
|
|
} |
|
} |
|
if (isset($name) && isset($type)) { |
|
if (isset($options['default']) && empty($options['default'])) { |
|
if (empty($options['notnull']) || !$options['notnull']) { |
|
unset($options['default']); |
|
$options['notnull'] = false; |
|
} else { |
|
$options['default'] = ''; |
|
} |
|
if ($type == 'integer' || $type == 'decimal') { |
|
$options['default'] = 0; |
|
} elseif ($type == 'boolean') { |
|
$options['default'] = false; |
|
} |
|
if (!empty($options['autoincrement']) && $options['autoincrement']) { |
|
unset($options['default']); |
|
} |
|
} |
|
if ($type === 'integer' && isset($options['default'])) { |
|
$options['default'] = (int)$options['default']; |
|
} |
|
if ($type === 'integer' && isset($options['length'])) { |
|
$length = $options['length']; |
|
if ($length < 4) { |
|
$type = 'smallint'; |
|
} else if ($length > 4) { |
|
$type = 'bigint'; |
|
} |
|
} |
|
if ($type === 'boolean' && isset($options['default'])) { |
|
$options['default'] = $this->asBool($options['default']); |
|
} |
|
if (!empty($options['autoincrement']) |
|
&& !empty($options['notnull']) |
|
) { |
|
$options['primary'] = true; |
|
} |
|
|
|
$table->addColumn($name, $type, $options); |
|
if (!empty($options['primary']) && $options['primary']) { |
|
$table->setPrimaryKey(array($name)); |
|
} |
|
} |
|
} |
|
|
|
/** |
|
* @param \Doctrine\DBAL\Schema\Table $table |
|
* @param \SimpleXMLElement $xml |
|
* @throws \DomainException |
|
*/ |
|
private function loadIndex($table, $xml) { |
|
$name = null; |
|
$fields = array(); |
|
foreach ($xml->children() as $child) { |
|
/** |
|
* @var \SimpleXMLElement $child |
|
*/ |
|
switch ($child->getName()) { |
|
case 'name': |
|
$name = (string)$child; |
|
break; |
|
case 'primary': |
|
$primary = $this->asBool($child); |
|
break; |
|
case 'unique': |
|
$unique = $this->asBool($child); |
|
break; |
|
case 'field': |
|
foreach ($child->children() as $field) { |
|
/** |
|
* @var \SimpleXMLElement $field |
|
*/ |
|
switch ($field->getName()) { |
|
case 'name': |
|
$field_name = (string)$field; |
|
$field_name = $this->platform->quoteIdentifier($field_name); |
|
$fields[] = $field_name; |
|
break; |
|
case 'sorting': |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $field->getName()); |
|
|
|
} |
|
} |
|
break; |
|
default: |
|
throw new \DomainException('Unknown element: ' . $child->getName()); |
|
|
|
} |
|
} |
|
if (!empty($fields)) { |
|
if (isset($primary) && $primary) { |
|
if ($table->hasPrimaryKey()) { |
|
return; |
|
} |
|
$table->setPrimaryKey($fields, $name); |
|
} else { |
|
if (isset($unique) && $unique) { |
|
$table->addUniqueIndex($fields, $name); |
|
} else { |
|
$table->addIndex($fields, $name); |
|
} |
|
} |
|
} else { |
|
throw new \DomainException('Empty index definition: ' . $name . ' options:' . print_r($fields, true)); |
|
} |
|
} |
|
|
|
/** |
|
* @param \SimpleXMLElement|string $xml |
|
* @return bool |
|
*/ |
|
private function asBool($xml) { |
|
$result = (string)$xml; |
|
if ($result == 'true') { |
|
$result = true; |
|
} elseif ($result == 'false') { |
|
$result = false; |
|
} |
|
return (bool)$result; |
|
} |
|
|
|
}
|
|
|