228 lines
6.0 KiB
PHP
228 lines
6.0 KiB
PHP
#!/usr/bin/env php
|
|
<?php
|
|
/*
|
|
* StatusNet - a distributed open-source microblogging tool
|
|
* Copyright (C) 2008, 2009, StatusNet, Inc.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* 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
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
|
|
|
|
$helptext = <<<END_OF_CHECKSCHEMA_HELP
|
|
Attempt to pull a schema definition for a given table.
|
|
|
|
--all run over all defined core tables
|
|
--diff show differences between the expected and live table defs
|
|
--raw skip compatibility filtering for diffs
|
|
--update dump SQL that would be run to update or create this table
|
|
--build dump SQL that would be run to create this table fresh
|
|
|
|
|
|
END_OF_CHECKSCHEMA_HELP;
|
|
|
|
$longoptions = array('diff', 'all', 'build', 'update', 'raw');
|
|
require_once INSTALLDIR.'/scripts/commandline.inc';
|
|
|
|
function indentOptions($indent)
|
|
{
|
|
$cutoff = 3;
|
|
if ($indent < $cutoff) {
|
|
$space = $indent ? str_repeat(' ', $indent * 4) : '';
|
|
$sep = ",";
|
|
$lf = "\n";
|
|
$endspace = "$lf" . ($indent ? str_repeat(' ', ($indent - 1) * 4) : '');
|
|
} else {
|
|
$space = '';
|
|
$sep = ", ";
|
|
$lf = '';
|
|
$endspace = '';
|
|
}
|
|
if ($indent - 1 < $cutoff) {
|
|
}
|
|
return array($space, $sep, $lf, $endspace);
|
|
}
|
|
|
|
function prettyDumpArray($arr, $key=null, $indent=0)
|
|
{
|
|
// hack
|
|
if ($key == 'primary key') {
|
|
$subIndent = $indent + 2;
|
|
} else {
|
|
$subIndent = $indent + 1;
|
|
}
|
|
|
|
list($space, $sep, $lf, $endspace) = indentOptions($indent);
|
|
list($inspace, $insep, $inlf, $inendspace) = indentOptions($subIndent);
|
|
|
|
print "{$space}";
|
|
if (!is_numeric($key)) {
|
|
print "'$key' => ";
|
|
}
|
|
if (is_array($arr)) {
|
|
print "array({$inlf}";
|
|
$n = 0;
|
|
foreach ($arr as $key => $row) {
|
|
$n++;
|
|
prettyDumpArray($row, $key, $subIndent);
|
|
if ($n < count($arr)) {
|
|
print "$insep$inlf";
|
|
}
|
|
}
|
|
// hack!
|
|
print "{$inendspace})";
|
|
} else {
|
|
print var_export($arr, true);
|
|
}
|
|
}
|
|
|
|
function getCoreSchema($tableName)
|
|
{
|
|
$schema = array();
|
|
include INSTALLDIR . '/db/core.php';
|
|
return $schema[$tableName];
|
|
}
|
|
|
|
function getCoreTables()
|
|
{
|
|
$schema = array();
|
|
include INSTALLDIR . '/db/core.php';
|
|
return array_keys($schema);
|
|
}
|
|
|
|
function dumpTable($tableName, $live)
|
|
{
|
|
if ($live) {
|
|
$schema = Schema::get();
|
|
$def = $schema->getTableDef($tableName);
|
|
} else {
|
|
// hack
|
|
$def = getCoreSchema($tableName);
|
|
}
|
|
prettyDumpArray($def, $tableName);
|
|
print "\n";
|
|
}
|
|
|
|
function dumpBuildTable($tableName)
|
|
{
|
|
echo "-- \n";
|
|
echo "-- $tableName\n";
|
|
echo "-- \n";
|
|
|
|
$schema = Schema::get();
|
|
$def = getCoreSchema($tableName);
|
|
$sql = $schema->buildCreateTable($tableName, $def);
|
|
$sql[] = '';
|
|
|
|
echo implode(";\n", $sql);
|
|
echo "\n";
|
|
}
|
|
|
|
function dumpEnsureTable($tableName)
|
|
{
|
|
$schema = Schema::get();
|
|
$def = getCoreSchema($tableName);
|
|
$sql = $schema->buildEnsureTable($tableName, $def);
|
|
|
|
if ($sql) {
|
|
echo "-- \n";
|
|
echo "-- $tableName\n";
|
|
echo "-- \n";
|
|
|
|
$sql[] = '';
|
|
echo implode(";\n", $sql);
|
|
echo "\n";
|
|
}
|
|
}
|
|
|
|
function dumpDiff($tableName, $filter)
|
|
{
|
|
$schema = Schema::get();
|
|
$def = getCoreSchema($tableName);
|
|
try {
|
|
$old = $schema->getTableDef($tableName);
|
|
} catch (Exception $e) {
|
|
// @fixme this is a terrible check :D
|
|
if (preg_match('/no such table/', $e->getMessage())) {
|
|
return dumpTable($tableName, false);
|
|
} else {
|
|
throw $e;
|
|
}
|
|
}
|
|
|
|
if ($filter) {
|
|
$old = $schema->filterDef($old);
|
|
$def = $schema->filterDef($def);
|
|
}
|
|
|
|
// @hack
|
|
$old = tweakPrimaryKey($old);
|
|
$def = tweakPrimaryKey($def);
|
|
|
|
$sections = array_unique(array_merge(array_keys($old), array_keys($def)));
|
|
$final = array();
|
|
foreach ($sections as $section) {
|
|
if ($section == 'fields') {
|
|
// this shouldn't be needed maybe... wait what?
|
|
}
|
|
$diff = $schema->diffArrays($old, $def, $section, $compare);
|
|
$chunks = array('del', 'mod', 'add');
|
|
foreach ($chunks as $chunk) {
|
|
if ($diff[$chunk]) {
|
|
foreach ($diff[$chunk] as $key) {
|
|
if ($chunk == 'del') {
|
|
$final[$section]["DEL $key"] = $old[$section][$key];
|
|
} else if ($chunk == 'add') {
|
|
$final[$section]["ADD $key"] = $def[$section][$key];
|
|
} else if ($chunk == 'mod') {
|
|
$final[$section]["OLD $key"] = $old[$section][$key];
|
|
$final[$section]["NEW $key"] = $def[$section][$key];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
prettyDumpArray($final, $tableName);
|
|
print "\n";
|
|
}
|
|
|
|
function tweakPrimaryKey($def)
|
|
{
|
|
if (isset($def['primary key'])) {
|
|
$def['primary keys'] = array('primary key' => $def['primary key']);
|
|
unset($def['primary key']);
|
|
}
|
|
return $def;
|
|
}
|
|
|
|
if (have_option('all')) {
|
|
$args = getCoreTables();
|
|
}
|
|
|
|
if (count($args)) {
|
|
foreach ($args as $tableName) {
|
|
if (have_option('diff')) {
|
|
dumpDiff($tableName, !have_option('raw'));
|
|
} else if (have_option('build')) {
|
|
dumpBuildTable($tableName);
|
|
} else if (have_option('update')) {
|
|
dumpEnsureTable($tableName);
|
|
} else {
|
|
dumpTable($tableName, true);
|
|
}
|
|
}
|
|
} else {
|
|
show_help($helptext);
|
|
} |