Test Failed
Pull Request — master (#205)
by Wilmer
03:58
created

Schema::getJsonColumns()   A

Complexity

Conditions 3
Paths 2

Size

Total Lines 14
Code Lines 7

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 12

Importance

Changes 0
Metric Value
cc 3
eloc 7
c 0
b 0
f 0
nc 2
nop 1
dl 0
loc 14
ccs 0
cts 0
cp 0
crap 12
rs 10
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Yiisoft\Db\Mysql;
6
7
use JsonException;
8
use Throwable;
9
use Yiisoft\Arrays\ArrayHelper;
10
use Yiisoft\Db\Constraint\Constraint;
11
use Yiisoft\Db\Constraint\ForeignKeyConstraint;
12
use Yiisoft\Db\Constraint\IndexConstraint;
13
use Yiisoft\Db\Exception\Exception;
14
use Yiisoft\Db\Exception\InvalidConfigException;
15
use Yiisoft\Db\Exception\NotSupportedException;
16
use Yiisoft\Db\Expression\Expression;
17
use Yiisoft\Db\Schema\ColumnSchemaInterface;
18
use Yiisoft\Db\Schema\Schema as AbstractSchema;
19
use Yiisoft\Db\Schema\TableSchemaInterface;
20
21
use function array_map;
22
use function array_merge;
23
use function array_values;
24
use function bindec;
25
use function explode;
26
use function in_array;
27
use function is_string;
28
use function ksort;
29
use function md5;
30
use function preg_match_all;
31
use function preg_match;
32
use function serialize;
33
use function stripos;
34
use function strtolower;
35
use function trim;
36
37
/**
38
 * The class Schema is the class for retrieving metadata from a Mysql database (version 5.7 and above).
39
 *
40
 * @psalm-type ColumnArray = array{
41
 *   table_schema: string,
42
 *   table_name: string,
43
 *   column_name: string,
44
 *   data_type: string,
45
 *   type_type: string|null,
46
 *   character_maximum_length: int,
47
 *   column_comment: string|null,
48
 *   modifier: int,
49
 *   is_nullable: bool,
50
 *   column_default: mixed,
51
 *   is_autoinc: bool,
52
 *   sequence_name: string|null,
53
 *   enum_values: array<array-key, float|int|string>|string|null,
54
 *   numeric_precision: int|null,
55
 *   numeric_scale: int|null,
56
 *   size: string|null,
57
 *   is_pkey: bool|null,
58
 *   dimension: int
59
 * }
60
 *
61
 * @psalm-type ColumnInfoArray = array{
62
 *   field: string,
63
 *   type: string,
64
 *   collation: string|null,
65
 *   null: string,
66
 *   key: string,
67
 *   default: string|null,
68
 *   extra: string,
69
 *   privileges: string,
70
 *   comment: string
71
 * }
72
 *
73
 * @psalm-type RowConstraint = array{
74
 *   constraint_name: string,
75
 *   column_name: string,
76
 *   referenced_table_name: string,
77
 *   referenced_column_name: string
78
 * }
79
 *
80
 * @psalm-type ConstraintArray = array<
81
 *   array-key,
82
 *   array {
83
 *     name: string,
84
 *     column_name: string,
85
 *     type: string,
86
 *     foreign_table_schema: string|null,
87
 *     foreign_table_name: string|null,
88
 *     foreign_column_name: string|null,
89
 *     on_update: string,
90
 *     on_delete: string,
91
 *     check_expr: string
92
 *   }
93
 * >
94
 */
95
final class Schema extends AbstractSchema
96
{
97
    /** @var array<array-key, string> $typeMap */
0 ignored issues
show
Documentation Bug introduced by
The doc comment array<array-key, string> at position 2 could not be parsed: Unknown type name 'array-key' at position 2 in array<array-key, string>.
Loading history...
98
    private array $typeMap = [
99
        'tinyint' => self::TYPE_TINYINT,
100
        'bit' => self::TYPE_INTEGER,
101
        'smallint' => self::TYPE_SMALLINT,
102
        'mediumint' => self::TYPE_INTEGER,
103
        'int' => self::TYPE_INTEGER,
104
        'integer' => self::TYPE_INTEGER,
105
        'bigint' => self::TYPE_BIGINT,
106
        'float' => self::TYPE_FLOAT,
107
        'double' => self::TYPE_DOUBLE,
108
        'real' => self::TYPE_FLOAT,
109
        'decimal' => self::TYPE_DECIMAL,
110
        'numeric' => self::TYPE_DECIMAL,
111
        'tinytext' => self::TYPE_TEXT,
112
        'mediumtext' => self::TYPE_TEXT,
113
        'longtext' => self::TYPE_TEXT,
114
        'longblob' => self::TYPE_BINARY,
115
        'blob' => self::TYPE_BINARY,
116
        'text' => self::TYPE_TEXT,
117
        'varchar' => self::TYPE_STRING,
118
        'string' => self::TYPE_STRING,
119
        'char' => self::TYPE_CHAR,
120
        'datetime' => self::TYPE_DATETIME,
121
        'year' => self::TYPE_DATE,
122
        'date' => self::TYPE_DATE,
123
        'time' => self::TYPE_TIME,
124
        'timestamp' => self::TYPE_TIMESTAMP,
125
        'enum' => self::TYPE_STRING,
126
        'varbinary' => self::TYPE_BINARY,
127
        'json' => self::TYPE_JSON,
128
    ];
129
130
    private array $jsonColumns = [];
131
    private array $columnOverrides = [];
132
    private string|null $currentTable = null;
133
134
    /**
135
     * Create a column schema builder instance giving the type and value precision.
136
     *
137
     * This method may be overridden by child classes to create a DBMS-specific column schema builder.
138
     *
139
     * @param string $type type of the column. See {@see ColumnSchemaBuilder::$type}.
140 10
     * @param array|int|string|null $length length or precision of the column. See {@see ColumnSchemaBuilder::$length}.
141
     *
142 10
     * @return ColumnSchemaBuilder column schema builder instance
143
     *
144
     * @psalm-param string[]|int[]|int|string|null $length
145
     */
146
    public function createColumnSchemaBuilder(string $type, array|int|string $length = null): ColumnSchemaBuilder
147
    {
148
        return new ColumnSchemaBuilder($type, $length, $this->db->getQuoter());
149
    }
150
151
    /**
152
     * Returns all unique indexes for the given table.
153
     *
154
     * Each array element is of the following structure:
155
     *
156
     * ```php
157
     * [
158
     *     'IndexName1' => ['col1' [, ...]],
159
     *     'IndexName2' => ['col2' [, ...]],
160
     * ]
161
     * ```
162
     *
163
     * @param TableSchemaInterface $table the table metadata.
164
     *
165 1
     * @throws Exception
166
     * @throws InvalidConfigException
167 1
     * @throws Throwable
168
     *
169 1
     * @return array all unique indexes for the given table.
170
     */
171 1
    public function findUniqueIndexes(TableSchemaInterface $table): array
172
    {
173 1
        $sql = $this->getCreateTableSql($table);
174 1
175 1
        $uniqueIndexes = [];
176 1
177 1
        $regexp = '/UNIQUE KEY\s+`(.+)`\s*\((`.+`)+\)/mi';
178
179
        if (preg_match_all($regexp, $sql, $matches, PREG_SET_ORDER)) {
180
            foreach ($matches as $match) {
181 1
                $indexName = $match[1];
182
                $indexColumns = array_map('trim', explode('`,`', trim($match[2], '`')));
183 1
                $uniqueIndexes[$indexName] = $indexColumns;
184
            }
185
        }
186
187
        ksort($uniqueIndexes);
188
189
        return $uniqueIndexes;
190
    }
191
192
    /**
193
     * Collects the metadata of table columns.
194
     *
195
     * @param TableSchemaInterface $table the table metadata.
196 147
     *
197
     * @throws Exception
198 147
     * @throws Throwable if DB query fails.
199 147
     *
200
     * @return bool whether the table exists in the database.
201
     */
202 147
    protected function findColumns(TableSchemaInterface $table): bool
203 30
    {
204 30
        $tableName = $table->getFullName() ?? '';
205
        $sql = 'SHOW FULL COLUMNS FROM ' . $this->db->getQuoter()->quoteTableName($tableName);
206 30
207
        $this->currentTable = $tableName;
208
209
        try {
210
            $columns = $this->db->createCommand($sql)->queryAll();
211
212 30
            $this->jsonColumns[$this->currentTable] = array_merge(
213
                $this->columnOverrides[$tableName] ?? [],
214
                $this->getJsonColumns($table),
215
            );
216
        } catch (Exception $e) {
217
            $previous = $e->getPrevious();
218
219 130
            if ($previous && str_contains($previous->getMessage(), 'SQLSTATE[42S02')) {
220 130
                /**
221
                 * table does not exist.
222 130
                 *
223 130
                 * https://dev.mysql.com/doc/refman/5.5/en/error-messages-server.html#error_er_bad_table_error
224
                 */
225 130
                return false;
226 82
            }
227 82
228 68
            throw $e;
229
        }
230
231
        /** @psalm-var ColumnInfoArray $info */
232
        foreach ($columns as $info) {
233 130
            $info = $this->normalizeRowKeyCase($info, false);
234
235
            $column = $this->loadColumnSchema($info);
236
            $table->columns($column->getName(), $column);
237
238
            if ($column->isPrimaryKey()) {
239
                $table->primaryKey($column->getName());
240
                if ($column->isAutoIncrement()) {
241
                    $table->sequenceName('');
242
                }
243
            }
244
        }
245 130
246
        return true;
247 130
    }
248
249
    /**
250
     * Collects the foreign key column details for the given table.
251
     *
252
     * @param TableSchemaInterface $table the table metadata.
253
     *
254
     * @throws Exception
255
     * @throws InvalidConfigException
256
     * @throws Throwable
257
     */
258
    protected function findConstraints(TableSchemaInterface $table): void
259
    {
260
        $sql = <<<SQL
261
        SELECT
262
            `kcu`.`CONSTRAINT_NAME` AS `constraint_name`,
263
            `kcu`.`COLUMN_NAME` AS `column_name`,
264
            `kcu`.`REFERENCED_TABLE_NAME` AS `referenced_table_name`,
265
            `kcu`.`REFERENCED_COLUMN_NAME` AS `referenced_column_name`
266
        FROM `information_schema`.`REFERENTIAL_CONSTRAINTS` AS `rc`
267 130
        JOIN `information_schema`.`KEY_COLUMN_USAGE` AS `kcu` ON
268
            (
269 130
                `kcu`.`CONSTRAINT_CATALOG` = `rc`.`CONSTRAINT_CATALOG` OR
270 130
                (
271 130
                    `kcu`.`CONSTRAINT_CATALOG` IS NULL AND
272 130
                    `rc`.`CONSTRAINT_CATALOG` IS NULL
273 130
                )
274
            ) AND
275
            `kcu`.`CONSTRAINT_SCHEMA` = `rc`.`CONSTRAINT_SCHEMA` AND
276 130
            `kcu`.`CONSTRAINT_NAME` = `rc`.`CONSTRAINT_NAME` AND
277 34
            `kcu`.`TABLE_SCHEMA` = `rc`.`CONSTRAINT_SCHEMA` AND
278 34
            `kcu`.`TABLE_NAME` = `rc`.`TABLE_NAME`
279
        WHERE `rc`.`CONSTRAINT_SCHEMA` = COALESCE(:schemaName, DATABASE()) AND `rc`.`TABLE_NAME` = :tableName
280
        SQL;
281 130
282
        $constraints = [];
283
        $rows = $this->db->createCommand($sql, [
284
            ':schemaName' => $table->getSchemaName(),
285
            ':tableName' => $table->getName(),
286 130
        ])->queryAll();
287 34
288 34
        /**  @psalm-var RowConstraint $row */
289 34
        foreach ($rows as $row) {
290 34
            $constraints[$row['constraint_name']]['referenced_table_name'] = $row['referenced_table_name'];
291 34
            $constraints[$row['constraint_name']]['columns'][$row['column_name']] = $row['referenced_column_name'];
292 34
        }
293 34
294
        $table->foreignKeys([]);
295
296
        /**
297
         * @var array{referenced_table_name: string, columns: array} $constraint
298
         */
299
        foreach ($constraints as $name => $constraint) {
300
            $table->foreignKey(
301
                $name,
302 1
                array_merge(
303
                    [$constraint['referenced_table_name']],
304 1
                    $constraint['columns']
305
                ),
306 1
            );
307
        }
308 1
    }
309
310
    /**
311
     * @throws Exception
312
     * @throws InvalidConfigException
313
     * @throws Throwable
314
     */
315
    protected function findSchemaNames(): array
316 147
    {
317
        $sql = <<<SQL
318 147
        SELECT schema_name FROM information_schema.schemata WHERE schema_name NOT IN ('information_schema', 'mysql', 'performance_schema', 'sys')
319
        SQL;
320
321
        return $this->db->createCommand($sql)->queryColumn();
322
    }
323
324 147
    /**
325
     * @throws Exception
326 147
     * @throws InvalidConfigException
327 147
     * @throws Throwable
328 147
     */
329 147
    protected function findTableComment(TableSchemaInterface $tableSchema): void
330
    {
331 147
        $sql = <<<SQL
332
        SELECT `TABLE_COMMENT`
333
        FROM `INFORMATION_SCHEMA`.`TABLES`
334
        WHERE
335
              `TABLE_SCHEMA` = COALESCE(:schemaName, DATABASE()) AND
336
              `TABLE_NAME` = :tableName;
337
        SQL;
338
339
        $comment = $this->db->createCommand($sql, [
340
            ':schemaName' => $tableSchema->getSchemaName(),
341
            ':tableName' => $tableSchema->getName(),
342
        ])->queryScalar();
343
344
        $tableSchema->comment(is_string($comment) ? $comment : null);
345
    }
346
347
    /**
348 12
     * Returns all table names in the database.
349
     *
350 12
     * This method should be overridden by child classes in order to support this feature because the default
351
     * implementation simply throws an exception.
352 12
     *
353 1
     * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema.
354
     *
355
     * @throws Exception
356 12
     * @throws InvalidConfigException
357
     * @throws Throwable
358
     *
359
     * @return array All table names in the database. The names have NO schema name prefix.
360
     */
361
    protected function findTableNames(string $schema = ''): array
362
    {
363
        $sql = 'SHOW TABLES';
364 1
365
        if ($schema !== '') {
366 1
            $sql .= ' FROM ' . $this->db->getQuoter()->quoteSimpleTableName($schema);
367 1
        }
368
369 1
        return $this->db->createCommand($sql)->queryColumn();
370 1
    }
371 1
372 1
    /**
373 1
     * @throws Exception
374
     * @throws InvalidConfigException
375
     * @throws Throwable
376 1
     */
377
    protected function findViewNames(string $schema = ''): array
378 1
    {
379 1
        $sql = match ($schema) {
380
            '' => <<<SQL
381
            SELECT table_name as view FROM information_schema.tables WHERE table_type LIKE 'VIEW' AND table_schema != 'sys'
382 1
            SQL,
383
            default => <<<SQL
384
            SELECT table_name as view FROM information_schema.tables WHERE table_type LIKE 'VIEW' AND table_schema = '$schema'
385
            SQL,
386
        };
387
388
        /** @psalm-var string[][] $views */
389
        $views = $this->db->createCommand($sql)->queryAll();
390
391
        foreach ($views as $key => $view) {
392 243
            $views[$key] = $view['view'];
393
        }
394 243
395
        return $views;
396
    }
397
398
    /**
399
     * Returns the cache key for the specified table name.
400
     *
401
     * @param string $name the table name.
402
     *
403
     * @return array the cache key.
404 244
     */
405
    protected function getCacheKey(string $name): array
406 244
    {
407
        return array_merge([self::class], $this->db->getCacheKey(), [$this->getRawTableName($name)]);
408
    }
409
410
    /**
411
     * Returns the cache tag name.
412
     *
413
     * This allows {@see refresh()} to invalidate all cached table schemas.
414
     *
415
     * @return string the cache tag name.
416
     */
417
    protected function getCacheTag(): string
418
    {
419
        return md5(serialize(array_merge([self::class], $this->db->getCacheKey())));
420 147
    }
421
422 147
    /**
423
     * Gets the CREATE TABLE sql string.
424
     *
425
     * @param TableSchemaInterface $table the table metadata.
426 147
     *
427 147
     * @throws Exception
428 147
     * @throws InvalidConfigException
429
     * @throws Throwable
430 130
     *
431 128
     * @return string $sql the result of 'SHOW CREATE TABLE'.
432
     */
433 4
    protected function getCreateTableSql(TableSchemaInterface $table): string
434 130
    {
435
        $tableName = $table->getFullName() ?? '';
436 30
437 30
        try {
438
            /** @var array<array-key, string> $row */
439
            $row = $this->db->createCommand(
440 147
                'SHOW CREATE TABLE ' . $this->db->getQuoter()->quoteTableName($tableName)
441
            )->queryOne();
442
443
            if (isset($row['Create Table'])) {
444
                $sql = $row['Create Table'];
445
            } else {
446
                $row = array_values($row);
0 ignored issues
show
Bug introduced by
$row of type null is incompatible with the type array expected by parameter $array of array_values(). ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

446
                $row = array_values(/** @scrutinizer ignore-type */ $row);
Loading history...
447
                $sql = $row[1];
448
            }
449
        } catch (Exception) {
450
            $sql = '';
451
        }
452 131
453
        return $sql;
454 131
    }
455
456
    /**
457 131
     * Loads the column information into a {@see ColumnSchemaInterface} object.
458 131
     *
459 131
     * @param array $info column information.
460 131
     *
461 131
     * @throws JsonException
462 131
     *
463 131
     * @return ColumnSchemaInterface the column schema object.
464 131
     */
465
    protected function loadColumnSchema(array $info): ColumnSchemaInterface
466 131
    {
467 131
        $column = $this->createColumnSchema();
468 28
469
        /** @psalm-var ColumnInfoArray $info */
470 131
        if (
471
            isset($this->jsonColumns[$this->currentTable]) &&
472 131
            in_array($info['field'], $this->jsonColumns[$this->currentTable], true)
473 131
        ) {
474
            $info['type'] = self::TYPE_JSON;
475 131
476 131
            if (is_string($info['default']) && preg_match("/^'(.*)'$/", $info['default'], $matches)) {
477
                $info['default'] = $matches[1];
478
            }
479 131
        }
480 102
481 26
        $column->name($info['field']);
482
        $column->allowNull($info['null'] === 'YES');
483 26
        $column->primaryKey(str_contains($info['key'], 'PRI'));
484 26
        $column->autoIncrement(stripos($info['extra'], 'auto_increment') !== false);
485
        $column->comment($info['comment']);
486
        $column->dbType($info['type']);
487 26
        $column->unsigned(stripos($column->getDbType(), 'unsigned') !== false);
488
        $column->type(self::TYPE_STRING);
489 102
490 102
        $extra = $info['extra'];
491 102
492
        if (str_starts_with($extra, 'DEFAULT_GENERATED')) {
493 102
            $extra = strtoupper(substr($extra, 18));
494 40
        }
495
        $column->extra(trim($extra));
496
497 102
        if (preg_match('/^(\w+)(?:\(([^)]+)\))?/', $column->getDbType(), $matches)) {
498 27
            $type = strtolower($matches[1]);
499 102
500 27
            if (isset($this->typeMap[$type])) {
501 4
                $column->type($this->typeMap[$type]);
502 27
            }
503 4
504
            if (!empty($matches[2])) {
505
                if ($type === 'enum') {
506
                    preg_match_all("/'[^']*'/", $matches[2], $values);
507
508
                    foreach ($values[0] as $i => $value) {
509
                        $values[$i] = trim($value, "'");
510 131
                    }
511
512 131
                    $column->enumValues($values);
513
                } else {
514
                    $values = explode(',', $matches[2]);
515
                    $column->precision((int) $values[0]);
516
                    $column->size((int) $values[0]);
517
518
                    if (isset($values[1])) {
519
                        $column->scale((int) $values[1]);
520 129
                    }
521 129
522
                    if ($column->getSize() === 1 && $type === 'tinyint') {
523 29
                        $column->type(self::TYPE_BOOLEAN);
524 29
                    } elseif ($type === 'bit') {
525 126
                        if ($column->getSize() > 32) {
526 27
                            $column->type(self::TYPE_BIGINT);
527
                        } elseif ($column->getSize() === 32) {
528 129
                            $column->type(self::TYPE_INTEGER);
529
                        }
530 82
                    }
531 3
                }
532
            }
533
        }
534 131
535
        $column->phpType($this->getColumnPhpType($column));
536
537
        if (!$column->isPrimaryKey()) {
538
            /**
539
             * When displayed in the INFORMATION_SCHEMA.COLUMNS table, a default CURRENT TIMESTAMP is displayed
540
             * as CURRENT_TIMESTAMP up until MariaDB 10.2.2, and as current_timestamp() from MariaDB 10.2.3.
541
             *
542
             * See details here: https://mariadb.com/kb/en/library/now/#description
543
             */
544
            if (
545
                ($column->getType() === 'timestamp' || $column->getType() === 'datetime')
546 16
                && preg_match('/^current_timestamp(?:\((\d*)\))?$/i', (string) $info['default'], $matches)
547
            ) {
548 16
                $column->defaultValue(new Expression('CURRENT_TIMESTAMP' . (!empty($matches[1])
549
                    ? '(' . $matches[1] . ')' : '')));
550
            } elseif (isset($type) && $type === 'bit') {
551
                $column->defaultValue(bindec(trim((string) $info['default'], 'b\'')));
552
            } else {
553
                $column->defaultValue($column->phpTypecast($info['default']));
554
            }
555
        } elseif ($info['default'] !== null) {
556
            $column->defaultValue($column->phpTypecast($info['default']));
557
        }
558
559
        return $column;
560
    }
561
562
    /**
563
     * Loads all check constraints for the given table.
564
     *
565
     * @param string $tableName table name.
566 71
     *
567
     * @throws NotSupportedException
568 71
     *
569
     * @return array check constraints for the given table.
570
     */
571
    protected function loadTableChecks(string $tableName): array
572
    {
573
        throw new NotSupportedException(__METHOD__ . ' is not supported by MySQL.');
574
    }
575
576
    /**
577
     * Loads multiple types of constraints and returns the specified ones.
578
     *
579
     * @param string $tableName table name.
580
     * @param string $returnType return type:
581
     * - primaryKey
582
     * - foreignKeys
583
     * - uniques
584
     *
585
     * @throws Exception
586
     * @throws InvalidConfigException
587
     * @throws Throwable
588
     *
589
     * @return array|Constraint|null (Constraint|ForeignKeyConstraint)[]|Constraint|null constraints.
590
     */
591
    private function loadTableConstraints(string $tableName, string $returnType): array|Constraint|null
592
    {
593
        $sql = <<<SQL
594
        SELECT
595
            `kcu`.`CONSTRAINT_NAME` AS `name`,
596
            `kcu`.`COLUMN_NAME` AS `column_name`,
597
            `tc`.`CONSTRAINT_TYPE` AS `type`,
598
        CASE
599
            WHEN :schemaName IS NULL AND `kcu`.`REFERENCED_TABLE_SCHEMA` = DATABASE() THEN NULL
600
        ELSE `kcu`.`REFERENCED_TABLE_SCHEMA`
601
        END AS `foreign_table_schema`,
602
            `kcu`.`REFERENCED_TABLE_NAME` AS `foreign_table_name`,
603
            `kcu`.`REFERENCED_COLUMN_NAME` AS `foreign_column_name`,
604
            `rc`.`UPDATE_RULE` AS `on_update`,
605
            `rc`.`DELETE_RULE` AS `on_delete`,
606
            `kcu`.`ORDINAL_POSITION` AS `position`
607
        FROM `information_schema`.`KEY_COLUMN_USAGE` AS `kcu`
608
        JOIN `information_schema`.`REFERENTIAL_CONSTRAINTS` AS `rc` ON
609
            `rc`.`CONSTRAINT_SCHEMA` = `kcu`.`TABLE_SCHEMA` AND
610
            `rc`.`TABLE_NAME` = `kcu`.`TABLE_NAME` AND
611
            `rc`.`CONSTRAINT_NAME` = `kcu`.`CONSTRAINT_NAME`
612
        JOIN `information_schema`.`TABLE_CONSTRAINTS` AS `tc` ON
613
            `tc`.`TABLE_SCHEMA` = `kcu`.`TABLE_SCHEMA` AND
614
            `tc`.`TABLE_NAME` = `kcu`.`TABLE_NAME` AND
615
            `tc`.`CONSTRAINT_NAME` = `kcu`.`CONSTRAINT_NAME` AND
616
            `tc`.`CONSTRAINT_TYPE` = 'FOREIGN KEY'
617 71
        WHERE
618
            `kcu`.`TABLE_SCHEMA` = COALESCE(:schemaName, DATABASE()) AND
619 71
            `kcu`.`CONSTRAINT_SCHEMA` = `kcu`.`TABLE_SCHEMA` AND
620
            `kcu`.`TABLE_NAME` = :tableName
621 71
        UNION
622 71
        SELECT
623 71
            `kcu`.`CONSTRAINT_NAME` AS `name`,
624 71
            `kcu`.`COLUMN_NAME` AS `column_name`,
625
            `tc`.`CONSTRAINT_TYPE` AS `type`,
626
        NULL AS `foreign_table_schema`,
627 71
        NULL AS `foreign_table_name`,
628 71
        NULL AS `foreign_column_name`,
629
        NULL AS `on_update`,
630 71
        NULL AS `on_delete`,
631 71
            `kcu`.`ORDINAL_POSITION` AS `position`
632 71
        FROM `information_schema`.`KEY_COLUMN_USAGE` AS `kcu`
633 71
        JOIN `information_schema`.`TABLE_CONSTRAINTS` AS `tc` ON
634 71
            `tc`.`TABLE_SCHEMA` = `kcu`.`TABLE_SCHEMA` AND
635
            `tc`.`TABLE_NAME` = `kcu`.`TABLE_NAME` AND
636
            `tc`.`CONSTRAINT_NAME` = `kcu`.`CONSTRAINT_NAME` AND
637
            `tc`.`CONSTRAINT_TYPE` IN ('PRIMARY KEY', 'UNIQUE')
638
        WHERE
639
            `kcu`.`TABLE_SCHEMA` = COALESCE(:schemaName, DATABASE()) AND
640 71
            `kcu`.`TABLE_NAME` = :tableName
641
        ORDER BY `position` ASC
642
        SQL;
643
644
        $resolvedName = $this->resolveTableName($tableName);
645 66
646
        $constraints = $this->db->createCommand($sql, [
647 66
            ':schemaName' => $resolvedName->getSchemaName(),
648 47
            ':tableName' => $resolvedName->getName(),
649 47
        ])->queryAll();
650 47
651 58
        /** @var array<array-key, array> $constraints */
652 16
        $constraints = $this->normalizeRowKeyCase($constraints, true);
653 16
        $constraints = ArrayHelper::index($constraints, null, ['type', 'name']);
654 16
655 16
        $result = [
656 16
            self::PRIMARY_KEY => null,
657 16
            self::FOREIGN_KEYS => [],
658 16
            self::UNIQUES => [],
659 16
        ];
660 16
661 48
        /**
662 48
         * @var string $type
663 48
         * @var array $names
664 48
         */
665 48
        foreach ($constraints as $type => $names) {
666
            /**
667
             * @psalm-var object|string|null $name
668
             * @psalm-var ConstraintArray $constraint
669
             */
670 71
            foreach ($names as $name => $constraint) {
671 71
                switch ($type) {
672
                    case 'PRIMARY KEY':
673
                        $result[self::PRIMARY_KEY] = (new Constraint())
674 71
                            ->columnNames(ArrayHelper::getColumn($constraint, 'column_name'));
675
                        break;
676
                    case 'FOREIGN KEY':
677
                        $result[self::FOREIGN_KEYS][] = (new ForeignKeyConstraint())
678
                            ->foreignSchemaName($constraint[0]['foreign_table_schema'])
679
                            ->foreignTableName($constraint[0]['foreign_table_name'])
680
                            ->foreignColumnNames(ArrayHelper::getColumn($constraint, 'foreign_column_name'))
681
                            ->onDelete($constraint[0]['on_delete'])
682
                            ->onUpdate($constraint[0]['on_update'])
683
                            ->columnNames(ArrayHelper::getColumn($constraint, 'column_name'))
684
                            ->name($name);
685
                        break;
686 15
                    case 'UNIQUE':
687
                        $result[self::UNIQUES][] = (new Constraint())
688 15
                            ->columnNames(ArrayHelper::getColumn($constraint, 'column_name'))
689
                            ->name($name);
690
                        break;
691
                }
692
            }
693
        }
694
695
        foreach ($result as $type => $data) {
696
            $this->setTableMetadata($tableName, $type, $data);
697
        }
698
699
        return $result[$returnType];
700
    }
701
702 9
    /**
703
     * Loads all default value constraints for the given table.
704 9
     *
705
     * @param string $tableName table name.
706 9
     *
707
     * @throws NotSupportedException
708
     *
709
     * @return array default value constraints for the given table.
710
     */
711
    protected function loadTableDefaultValues(string $tableName): array
712
    {
713
        throw new NotSupportedException(__METHOD__ . ' is not supported by MySQL.');
714
    }
715
716
    /**
717
     * Loads all foreign keys for the given table.
718
     *
719
     * @param string $tableName table name.
720 38
     *
721
     * @throws Exception
722 38
     * @throws InvalidConfigException
723
     * @throws Throwable
724
     *
725
     * @return array foreign keys for the given table.
726
     */
727
    protected function loadTableForeignKeys(string $tableName): array
728
    {
729
        $tableForeignKeys = $this->loadTableConstraints($tableName, self::FOREIGN_KEYS);
730
731
        return is_array($tableForeignKeys) ? $tableForeignKeys : [];
732
    }
733
734 38
    /**
735
     * Loads all indexes for the given table.
736 38
     *
737
     * @param string $tableName table name.
738 38
     *
739 38
     * @throws Exception
740 38
     * @throws InvalidConfigException
741 38
     * @throws Throwable
742
     *
743
     * @return IndexConstraint[] indexes for the given table.
744 38
     */
745 38
    protected function loadTableIndexes(string $tableName): array
746 38
    {
747
        $sql = <<<SQL
748
        SELECT
749
            `s`.`INDEX_NAME` AS `name`,
750
            `s`.`COLUMN_NAME` AS `column_name`,
751
            `s`.`NON_UNIQUE` ^ 1 AS `index_is_unique`,
752 38
            `s`.`INDEX_NAME` = 'PRIMARY' AS `index_is_primary`
753 38
        FROM `information_schema`.`STATISTICS` AS `s`
754
        WHERE
755 38
            `s`.`TABLE_SCHEMA` = COALESCE(:schemaName, DATABASE()) AND
756 38
            `s`.`INDEX_SCHEMA` = `s`.`TABLE_SCHEMA` AND
757 38
            `s`.`TABLE_NAME` = :tableName
758 38
        ORDER BY `s`.`SEQ_IN_INDEX` ASC
759
        SQL;
760 38
761
        $resolvedName = $this->resolveTableName($tableName);
762
763 38
        $indexes = $this->db->createCommand($sql, [
764
            ':schemaName' => $resolvedName->getSchemaName(),
765
            ':tableName' => $resolvedName->getName(),
766
        ])->queryAll();
767
768
        /** @var array[] $indexes */
769
        $indexes = $this->normalizeRowKeyCase($indexes, true);
770
        $indexes = ArrayHelper::index($indexes, null, 'name');
771
        $result = [];
772
773
        /**
774
         * @psalm-var object|string|null $name
775
         * @psalm-var array[] $index
776
         */
777 45
        foreach ($indexes as $name => $index) {
778
            $ic = new IndexConstraint();
779 45
780
            $ic->primary((bool) $index[0]['index_is_primary']);
781 45
            $ic->unique((bool) $index[0]['index_is_unique']);
782
            $ic->name($name !== 'PRIMARY' ? $name : null);
783
            $ic->columnNames(ArrayHelper::getColumn($index, 'column_name'));
784
785
            $result[] = $ic;
786
        }
787
788
        return $result;
789
    }
790
791
    /**
792
     * Loads a primary key for the given table.
793
     *
794 147
     * @param string $tableName table name.
795
     *
796 147
     * @throws Exception
797 147
     * @throws InvalidConfigException
798 147
     * @throws Throwable
799
     *
800 147
     * @return Constraint|null primary key for the given table, `null` if the table has no primary key.*
801 130
     */
802
    protected function loadTablePrimaryKey(string $tableName): Constraint|null
803 130
    {
804
        $tablePrimaryKey = $this->loadTableConstraints($tableName, self::PRIMARY_KEY);
805
806 30
        return $tablePrimaryKey instanceof Constraint ? $tablePrimaryKey : null;
807
    }
808
809
    /**
810
     * Loads the metadata for the specified table.
811
     *
812
     * @param string $name table name.
813
     *
814
     * @throws Exception
815
     * @throws Throwable
816
     *
817
     * @return TableSchemaInterface|null DBMS-dependent table metadata, `null` if the table does not exist.
818
     */
819
    protected function loadTableSchema(string $name): TableSchemaInterface|null
820 17
    {
821
        $table = $this->resolveTableName($name);
822 17
        $this->resolveTableCreateSql($table);
823
        $this->findTableComment($table);
824 17
825
        if ($this->findColumns($table)) {
826
            $this->findConstraints($table);
827
828
            return $table;
829
        }
830
831
        return null;
832
    }
833
834 191
    /**
835
     * Loads all unique constraints for the given table.
836 191
     *
837
     * @param string $tableName table name.
838 191
     *
839 191
     * @throws Exception
840 191
     * @throws InvalidConfigException
841
     * @throws Throwable
842 191
     *
843 191
     * @return array unique constraints for the given table.
844
     */
845 191
    protected function loadTableUniques(string $tableName): array
846 191
    {
847 191
        $tableUniques = $this->loadTableConstraints($tableName, self::UNIQUES);
848 191
849
        return is_array($tableUniques) ? $tableUniques : [];
850 191
    }
851
852
    /**
853
     * Resolves the table name and schema name (if any).
854
     *
855
     * @param string $name the table name.
856
     *
857
     * {@see TableSchemaInterface}
858 147
     */
859
    protected function resolveTableName(string $name): TableSchemaInterface
860 147
    {
861 147
        $resolvedName = new TableSchema();
862
863
        $parts = array_reverse(
864
            $this->db->getQuoter()->getTableNameParts($name)
865
        );
866
867
        $resolvedName->name($parts[0] ?? '');
868
        $resolvedName->schemaName($parts[1] ?? $this->defaultSchema);
869
870
        $resolvedName->fullName(
871 131
            $resolvedName->getSchemaName() !== $this->defaultSchema ?
872
            implode('.', array_reverse($parts)) : $resolvedName->getName()
873 131
        );
874
875
        return $resolvedName;
876
    }
877
878
    /**
879
     * @throws Exception
880
     * @throws InvalidConfigException
881
     * @throws Throwable
882
     */
883
    protected function resolveTableCreateSql(TableSchemaInterface $table): void
884
    {
885
        $sql = $this->getCreateTableSql($table);
886
        $table->createSql($sql);
887
    }
888
889
    /**
890
     * Creates a column schema for the database.
891
     *
892
     * This method may be overridden by child classes to create a DBMS-specific column schema.
893
     *
894
     * @return ColumnSchema column schema instance.
895
     */
896
    private function createColumnSchema(): ColumnSchema
897
    {
898
        return new ColumnSchema();
899
    }
900
901
    private function getJsonColumns(TableSchemaInterface $table): array
902
    {
903
        $sql = $this->getCreateTableSql($table);
904
        $result = [];
905
906
        $regexp = '/json_valid\([\`"](.+)[\`"]\s*\)/mi';
907
908
        if (preg_match_all($regexp, $sql, $matches, PREG_SET_ORDER)) {
909
            foreach ($matches as $match) {
910
                $result[] = $match[1];
911
            }
912
        }
913
914
        return $result;
915
    }
916
}
917