mirror of
https://github.com/PostgreSQL-For-Wordpress/postgresql-for-wordpress.git
synced 2026-01-26 08:22:19 +01:00
83 lines
2.8 KiB
PHP
83 lines
2.8 KiB
PHP
<?php
|
|
|
|
class CreateTableSQLRewriter extends AbstractSQLRewriter
|
|
{
|
|
private $stringReplacements = [
|
|
'bigint(20)' => 'bigint',
|
|
'bigint(10)' => 'int',
|
|
'int(11)' => 'int',
|
|
'int(1)' => 'smallint',
|
|
'tinytext' => 'text',
|
|
'mediumtext' => 'text',
|
|
'longtext' => 'text',
|
|
'unsigned' => '',
|
|
'gmt datetime NOT NULL default \'0000-00-00 00:00:00\'' => 'gmt timestamp NOT NULL DEFAULT timezone(\'gmt\'::text, now())',
|
|
'default \'0000-00-00 00:00:00\'' => 'DEFAULT now()',
|
|
'\'0000-00-00 00:00:00\'' => 'now()',
|
|
'datetime' => 'timestamp',
|
|
'DEFAULT CHARACTER SET utf8mb4' => '',
|
|
'DEFAULT CHARACTER SET utf8' => '',
|
|
|
|
// WP 2.7.1 compatibility
|
|
'int(4)' => 'smallint',
|
|
|
|
// For WPMU (starting with WP 3.2)
|
|
'tinyint(2)' => 'smallint',
|
|
'tinyint(1)' => 'smallint',
|
|
"enum('0','1')" => 'smallint',
|
|
'COLLATE utf8mb4_unicode_520_ci' => '',
|
|
'COLLATE utf8_general_ci' => '',
|
|
|
|
// For flash-album-gallery plugin
|
|
'tinyint' => 'smallint'
|
|
];
|
|
|
|
public function rewrite(): string
|
|
{
|
|
$sql = $this->original();
|
|
|
|
|
|
$sql = str_replace('CREATE TABLE IF NOT EXISTS ', 'CREATE TABLE ', $sql);
|
|
$pattern = '/CREATE TABLE [`]?(\w+)[`]?/';
|
|
preg_match($pattern, $sql, $matches);
|
|
$table = $matches[1];
|
|
|
|
// Remove trailing spaces
|
|
$sql = trim($sql) . ';';
|
|
|
|
// Translate types and some other replacements
|
|
$sql = str_ireplace(
|
|
array_keys($this->stringReplacements),
|
|
array_values($this->stringReplacements),
|
|
$sql
|
|
);
|
|
|
|
// Fix auto_increment by adding a sequence
|
|
$pattern = '/int[ ]+NOT NULL auto_increment/i';
|
|
preg_match($pattern, $sql, $matches);
|
|
if($matches) {
|
|
$seq = $table . '_seq';
|
|
$sql = str_ireplace('NOT NULL auto_increment', "NOT NULL DEFAULT nextval('$seq'::text)", $sql);
|
|
$sql .= "\nCREATE SEQUENCE $seq;";
|
|
}
|
|
|
|
// Support for UNIQUE INDEX creation
|
|
$pattern = '/,\s+(UNIQUE |)KEY\s+([^\s]+)\s+\(((?:[\w]+(?:\([\d]+\))?[,]?)*)\)/';
|
|
if(preg_match_all($pattern, $sql, $matches, PREG_SET_ORDER)) {
|
|
foreach($matches as $match) {
|
|
$unique = $match[1];
|
|
$index = $match[2];
|
|
$columns = $match[3];
|
|
$columns = preg_replace('/\(\d+\)/', '', $columns);
|
|
// Workaround for index name duplicate
|
|
$index = $table . '_' . $index;
|
|
$sql .= "\nCREATE {$unique}INDEX $index ON $table ($columns);";
|
|
}
|
|
}
|
|
// Now remove handled indexes
|
|
$sql = preg_replace($pattern, '', $sql);
|
|
|
|
return $sql;
|
|
}
|
|
}
|