1 | <?php |
||
2 | |||
3 | namespace Lagdo\DbAdmin\Driver\Sqlite\Db; |
||
4 | |||
5 | use Lagdo\DbAdmin\Driver\Entity\TableEntity; |
||
6 | use Lagdo\DbAdmin\Driver\Db\Database as AbstractDatabase; |
||
7 | |||
8 | class Database extends AbstractDatabase |
||
9 | { |
||
10 | /** |
||
11 | * Recreate a table |
||
12 | * |
||
13 | * @param TableEntity $tableAttrs |
||
14 | * @param string $table |
||
15 | * |
||
16 | * @return bool |
||
17 | */ |
||
18 | private function recreateTable(TableEntity $tableAttrs, string $table = '') |
||
19 | { |
||
20 | $alter = []; |
||
21 | $originals = []; |
||
22 | $indexes = []; |
||
23 | foreach ($tableAttrs->fields as $field) { |
||
24 | if ($field[1]) { |
||
25 | $alter[] = (\is_string($field[1]) ? $field[1] : 'ADD ' . implode($field[1])); |
||
26 | if ($field[0] != '') { |
||
27 | $originals[$field[0]] = $field[1][0]; |
||
28 | } |
||
29 | } |
||
30 | } |
||
31 | |||
32 | if ($table != '') { |
||
33 | if (empty($fields)) { |
||
0 ignored issues
–
show
Comprehensibility
Best Practice
introduced
by
Loading history...
|
|||
34 | foreach ($this->driver->fields($table) as $key => $field) { |
||
35 | if (!empty($indexes)) { |
||
36 | $field->autoIncrement = 0; |
||
37 | } |
||
38 | $fields[] = $this->util->processField($field, $field); |
||
39 | $originals[$key] = $this->driver->escapeId($key); |
||
40 | } |
||
41 | } |
||
42 | $primary_key = false; |
||
43 | foreach ($fields as $field) { |
||
0 ignored issues
–
show
Comprehensibility
Best Practice
introduced
by
|
|||
44 | if ($field[6]) { |
||
45 | $primary_key = true; |
||
46 | } |
||
47 | } |
||
48 | $drop_indexes = []; |
||
49 | foreach ($indexes as $key => $val) { |
||
50 | if ($val[2] == 'DROP') { |
||
51 | $drop_indexes[$val[1]] = true; |
||
52 | unset($indexes[$key]); |
||
53 | } |
||
54 | } |
||
55 | foreach ($this->driver->indexes($table) as $key_name => $index) { |
||
56 | $columns = []; |
||
57 | foreach ($index->columns as $key => $column) { |
||
58 | if (!$originals[$column]) { |
||
59 | continue 2; |
||
60 | } |
||
61 | $columns[] = $originals[$column] . ($index->descs[$key] ? ' DESC' : ''); |
||
62 | } |
||
63 | if (!$drop_indexes[$key_name]) { |
||
64 | if ($index->type != 'PRIMARY' || !$primary_key) { |
||
65 | $indexes[] = [$index->type, $key_name, $columns]; |
||
66 | } |
||
67 | } |
||
68 | } |
||
69 | foreach ($indexes as $key => $val) { |
||
70 | if ($val[0] == 'PRIMARY') { |
||
71 | unset($indexes[$key]); |
||
72 | $foreign[] = ' PRIMARY KEY (' . implode(', ', $val[2]) . ')'; |
||
73 | } |
||
74 | } |
||
75 | foreach ($this->driver->foreignKeys($table) as $key_name => $foreignKey) { |
||
76 | foreach ($foreignKey->source as $key => $column) { |
||
77 | if (!$originals[$column]) { |
||
78 | continue 2; |
||
79 | } |
||
80 | $foreignKey->source[$key] = $this->driver->unescapeId($originals[$column]); |
||
81 | } |
||
82 | if (!isset($foreign[" $key_name"])) { |
||
83 | $foreign[] = ' ' . $this->driver->formatForeignKey($foreignKey); |
||
84 | } |
||
85 | } |
||
86 | $this->driver->execute('BEGIN'); |
||
87 | } |
||
88 | foreach ($fields as $key => $field) { |
||
89 | $fields[$key] = ' ' . implode($field); |
||
90 | } |
||
91 | $fields = array_merge($fields, array_filter($foreign)); |
||
0 ignored issues
–
show
Comprehensibility
Best Practice
introduced
by
|
|||
92 | $tempName = ($table == $name ? "adminer_$name" : $name); |
||
0 ignored issues
–
show
Comprehensibility
Best Practice
introduced
by
|
|||
93 | if (!$this->driver->execute('CREATE TABLE ' . $this->driver->table($tempName) . |
||
94 | " (\n" . implode(",\n", $fields) . "\n)")) { |
||
95 | // implicit ROLLBACK to not overwrite $this->driver->error() |
||
96 | return false; |
||
97 | } |
||
98 | if ($table != '') { |
||
99 | if ($originals && !$this->driver->execute('INSERT INTO ' . $this->driver->table($tempName) . |
||
100 | ' (' . implode(', ', $originals) . ') SELECT ' . implode( |
||
101 | ', ', |
||
102 | array_map(function ($key) { |
||
103 | return $this->driver->escapeId($key); |
||
104 | }, array_keys($originals)) |
||
105 | ) . ' FROM ' . $this->driver->table($table))) { |
||
106 | return false; |
||
107 | } |
||
108 | $triggers = []; |
||
109 | foreach ($this->driver->triggers($table) as $trigger_name => $timing_event) { |
||
110 | $trigger = $this->driver->trigger($trigger_name); |
||
111 | $triggers[] = 'CREATE TRIGGER ' . $this->driver->escapeId($trigger_name) . ' ' . |
||
112 | implode(' ', $timing_event) . ' ON ' . $this->driver->table($name) . "\n$trigger[Statement]"; |
||
113 | } |
||
114 | $autoIncrement = $autoIncrement ? 0 : |
||
0 ignored issues
–
show
Comprehensibility
Best Practice
introduced
by
|
|||
115 | $this->connection->result('SELECT seq FROM sqlite_sequence WHERE name = ' . |
||
116 | $this->driver->quote($table)); // if $autoIncrement is set then it will be updated later |
||
117 | // drop before creating indexes and triggers to allow using old names |
||
118 | if (!$this->driver->execute('DROP TABLE ' . $this->driver->table($table)) || |
||
119 | ($table == $name && !$this->driver->execute('ALTER TABLE ' . $this->driver->table($tempName) . |
||
120 | ' RENAME TO ' . $this->driver->table($name))) || !$this->alterIndexes($name, $indexes) |
||
121 | ) { |
||
122 | return false; |
||
123 | } |
||
124 | if ($autoIncrement) { |
||
125 | $this->driver->execute('UPDATE sqlite_sequence SET seq = $autoIncrement WHERE name = ' . $this->driver->quote($name)); // ignores error |
||
126 | } |
||
127 | foreach ($triggers as $trigger) { |
||
128 | if (!$this->driver->execute($trigger)) { |
||
129 | return false; |
||
130 | } |
||
131 | } |
||
132 | $this->driver->execute('COMMIT'); |
||
133 | } |
||
134 | return true; |
||
135 | } |
||
136 | |||
137 | /** |
||
138 | * @param string $table |
||
139 | * @param int $autoIncrement |
||
140 | * |
||
141 | * @return void |
||
142 | */ |
||
143 | private function setAutoIncrement(string $table, int $autoIncrement) |
||
144 | { |
||
145 | if ($autoIncrement) { |
||
146 | $this->driver->execute('BEGIN'); |
||
147 | $this->driver->execute("UPDATE sqlite_sequence SET seq = $autoIncrement WHERE name = " . |
||
148 | $this->driver->quote($table)); // ignores error |
||
149 | if (!$this->driver->affectedRows()) { |
||
150 | $this->driver->execute('INSERT INTO sqlite_sequence (name, seq) VALUES (' . |
||
151 | $this->driver->quote($table) . ", $autoIncrement)"); |
||
152 | } |
||
153 | $this->driver->execute('COMMIT'); |
||
154 | } |
||
155 | } |
||
156 | |||
157 | /** |
||
158 | * @inheritDoc |
||
159 | */ |
||
160 | public function createTable(TableEntity $tableAttrs) |
||
161 | { |
||
162 | if (!$this->recreateTable($tableAttrs)) { |
||
163 | return false; |
||
164 | } |
||
165 | $this->setAutoIncrement($tableAttrs->name, $tableAttrs->autoIncrement); |
||
166 | return true; |
||
167 | } |
||
168 | |||
169 | /** |
||
170 | * @inheritDoc |
||
171 | */ |
||
172 | public function alterTable(string $table, TableEntity $tableAttrs) |
||
173 | { |
||
174 | $use_all_fields = !empty($tableAttrs->foreign); |
||
175 | foreach ($tableAttrs->fields as $field) { |
||
176 | if ($field[0] != '' || !$field[1] || $field[2]) { |
||
177 | $use_all_fields = true; |
||
178 | break; |
||
179 | } |
||
180 | } |
||
181 | if (!$use_all_fields) { |
||
182 | $alter = []; |
||
183 | foreach ($tableAttrs->fields as $field) { |
||
184 | if ($field[1]) { |
||
185 | $alter[] = ($use_all_fields ? $field[1] : 'ADD ' . implode($field[1])); |
||
186 | } |
||
187 | } |
||
188 | foreach ($alter as $val) { |
||
189 | if (!$this->driver->execute('ALTER TABLE ' . $this->driver->table($table) . " $val")) { |
||
190 | return false; |
||
191 | } |
||
192 | } |
||
193 | if ($table != $tableAttrs->name && !$this->driver->execute('ALTER TABLE ' . |
||
194 | $this->driver->table($table) . ' RENAME TO ' . $this->driver->table($tableAttrs->name))) { |
||
195 | return false; |
||
196 | } |
||
197 | } elseif (!$this->recreateTable($tableAttrs, $table)) { |
||
198 | return false; |
||
199 | } |
||
200 | $this->setAutoIncrement($tableAttrs->name, $tableAttrs->autoIncrement); |
||
201 | return true; |
||
202 | } |
||
203 | |||
204 | /** |
||
205 | * @inheritDoc |
||
206 | */ |
||
207 | public function alterIndexes(string $table, array $alter, array $drop) |
||
208 | { |
||
209 | foreach ($alter as $index) { |
||
210 | if ($index->type == 'PRIMARY') { |
||
211 | // return $this->recreateTable($table, $table, [], [], [], 0, $alter); |
||
212 | // Do not alter primary keys, since it requires to recreate the table. |
||
213 | return false; |
||
214 | } |
||
215 | } |
||
216 | foreach (array_reverse($drop) as $index) { |
||
217 | $this->driver->execute('DROP INDEX ' . $this->driver->escapeId($index->name)); |
||
218 | } |
||
219 | foreach (array_reverse($alter) as $index) { |
||
220 | $this->driver->execute($this->driver->sqlForCreateIndex($table, |
||
221 | $index->type, $index->name, '(' . implode(', ', $index->columns) . ')')); |
||
222 | } |
||
223 | return true; |
||
224 | } |
||
225 | |||
226 | /** |
||
227 | * @inheritDoc |
||
228 | */ |
||
229 | public function tables() |
||
230 | { |
||
231 | return $this->driver->keyValues('SELECT name, type FROM sqlite_master ' . |
||
232 | "WHERE type IN ('table', 'view') ORDER BY (name = 'sqlite_sequence'), name"); |
||
233 | } |
||
234 | |||
235 | /** |
||
236 | * @inheritDoc |
||
237 | */ |
||
238 | public function countTables(array $databases) |
||
239 | { |
||
240 | $connection = $this->driver->createConnection(); // New connection |
||
241 | $counts = []; |
||
242 | $query = "SELECT count(*) FROM sqlite_master WHERE type IN ('table', 'view')"; |
||
243 | foreach ($databases as $database) { |
||
244 | $counts[$database] = 0; |
||
245 | $connection->open($database); |
||
246 | $statement = $connection->query($query); |
||
247 | if (is_object($statement) && ($row = $statement->fetchRow())) { |
||
248 | $counts[$database] = intval($row[0]); |
||
249 | } |
||
250 | } |
||
251 | return $counts; |
||
252 | } |
||
253 | |||
254 | /** |
||
255 | * @inheritDoc |
||
256 | */ |
||
257 | public function dropViews(array $views) |
||
258 | { |
||
259 | return $this->driver->applyQueries('DROP VIEW', $views); |
||
260 | } |
||
261 | |||
262 | /** |
||
263 | * @inheritDoc |
||
264 | */ |
||
265 | public function dropTables(array $tables) |
||
266 | { |
||
267 | return $this->driver->applyQueries('DROP TABLE', $tables); |
||
268 | } |
||
269 | |||
270 | /** |
||
271 | * @inheritDoc |
||
272 | */ |
||
273 | public function moveTables(array $tables, array $views, string $target) |
||
274 | { |
||
275 | return false; |
||
276 | } |
||
277 | |||
278 | /** |
||
279 | * @inheritDoc |
||
280 | */ |
||
281 | public function truncateTables(array $tables) |
||
282 | { |
||
283 | return $this->driver->applyQueries('DELETE FROM', $tables); |
||
284 | } |
||
285 | } |
||
286 |