scripts/dumpschema.php

   1 #!/usr/bin/env php
   2 <?php
   3 /*
   4  * StatusNet - a distributed open-source microblogging tool
   5  * Copyright (C) 2008, 2009, StatusNet, Inc.
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
  22
  23 $helptext = <<<END_OF_CHECKSCHEMA_HELP
  24 Attempt to pull a schema definition for a given table.
  25
  26 END_OF_CHECKSCHEMA_HELP;
  27
  28 $longoptions = array('diff');
  29 require_once INSTALLDIR.'/scripts/commandline.inc';
  30
  31 function indentOptions($indent)
  32 {
  33     $cutoff = 3;
  34     if ($indent < $cutoff) {
  35         $space = $indent ? str_repeat(' ', $indent * 4) : '';
  36         $sep = ",";
  37         $lf = "\n";
  38         $endspace = "$lf" . ($indent ? str_repeat(' ', ($indent - 1) * 4) : '');
  39     } else {
  40         $space = '';
  41         $sep = ", ";
  42         $lf = '';
  43         $endspace = '';
  44     }
  45     if ($indent - 1 < $cutoff) {
  46     }
  47     return array($space, $sep, $lf, $endspace);
  48 }
  49
  50 function prettyDumpArray($arr, $key=null, $indent=0)
  51 {
  52     // hack
  53     if ($key == 'primary key') {
  54         $subIndent = $indent + 2;
  55     } else {
  56         $subIndent = $indent + 1;
  57     }
  58
  59     list($space, $sep, $lf, $endspace) = indentOptions($indent);
  60     list($inspace, $insep, $inlf, $inendspace) = indentOptions($subIndent);
  61
  62     print "{$space}";
  63     if (!is_numeric($key)) {
  64         print "'$key' => ";
  65     }
  66     if (is_array($arr)) {
  67         print "array({$inlf}";
  68         $n = 0;
  69         foreach ($arr as $key => $row) {
  70             $n++;
  71             prettyDumpArray($row, $key, $subIndent);
  72             if ($n < count($arr)) {
  73                 print "$insep$inlf";
  74             }
  75         }
  76         // hack!
  77         print "{$inendspace})";
  78     } else {
  79         print var_export($arr, true);
  80     }
  81 }
  82
  83 function getCoreSchema($tableName)
  84 {
  85     $schema = array();
  86     include INSTALLDIR . '/db/core.php';
  87     return $schema[$tableName];
  88 }
  89
  90 function dumpTable($tableName, $live)
  91 {
  92     if ($live) {
  93         $schema = Schema::get();
  94         $def = $schema->getTableDef($tableName);
  95     } else {
  96         // hack
  97         $def = getCoreSchema($tableName);
  98     }
  99     prettyDumpArray($def, $tableName);
 100     echo "\n";
 101 }
 102
 103 function showDiff($a, $b)
 104 {
 105     $fnameA = tempnam(sys_get_temp_dir(), 'defined-diff-a');
 106     file_put_contents($fnameA, $a);
 107
 108     $fnameB = tempnam(sys_get_temp_dir(), 'detected-diff-b');
 109     file_put_contents($fnameB, $b);
 110
 111     $cmd = sprintf('diff -U 100 %s %s',
 112             escapeshellarg($fnameA),
 113             escapeshellarg($fnameB));
 114     passthru($cmd);
 115
 116     unlink($fnameA);
 117     unlink($fnameB);
 118 }
 119
 120 if (count($args)) {
 121     foreach ($args as $tableName) {
 122         if (have_option('diff')) {
 123             ob_start();
 124             dumpTable($tableName, false);
 125             $defined = ob_get_clean();
 126
 127             ob_start();
 128             dumpTable($tableName, true);
 129             $detected = ob_get_clean();
 130
 131             showDiff($defined, $detected);
 132         } else {
 133             dumpTable($tableName, true);
 134         }
 135     }
 136 } else {
 137     show_help($helptext);
 138 }