]> git.mxchange.org Git - quix0rs-gnu-social.git/blob - scripts/dumpschema.php
818a4d5eae4cbe4ee567ca8554e3da8b9f3cc5b4
[quix0rs-gnu-social.git] / 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 = str_repeat(' ', $indent * 4);
36         $sep = ",";
37         $lf = "\n";
38         $endspace = "$lf" . 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 }