Yiisoft\Db\Sqlite\Schema\Schema::loadTableConstraints()

Complexity

Conditions	16
Paths	364

Size

Total Lines	106
Code Lines	61

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	47
CRAP Score	16.3713

Importance

Changes

0

<?php
    private function loadTableConstraints(string $tableName, string $returnType)
    {
        $tableColumns = null;

        $index = $this->getDb()->createCommand(
            'PRAGMA INDEX_LIST (' . $this->quoteValue($tableName) . ')'
        )->queryAll();

        $unique = $this->getDb()->createCommand(
            "SELECT
                '0' as 'seq',
                name,
                '1' as 'unique',
                'u' as 'origin',
                '0' as 'partial'
            FROM sqlite_master
            WHERE type='index' AND sql LIKE 'CREATE UNIQUE INDEX%' AND tbl_name='$tableName'"
        )->queryAll();

        $indexes = array_merge($index, $unique);
        $indexes = $this->normalizePdoRowKeyCase($indexes, true);

        if (!empty($indexes) && !isset($indexes[0]['origin'])) {
            /**
             * SQLite may not have an "origin" column in INDEX_LIST.
             *
             * {See https://www.sqlite.org/src/info/2743846cdba572f6}
             */
            $tableColumns = $this->loadTableColumnsInfo($tableName);
        }

        $result = [
            'primaryKey' => null,
            'indexes' => [],
            'uniques' => [],
        ];

        foreach ($indexes as $index) {
            $columns = $this->getDb()->createCommand(
                'PRAGMA INDEX_INFO (' . $this->quoteValue($index['name']) . ')'
            )->queryAll();

            $columns = $this->normalizePdoRowKeyCase($columns, true);

            ArraySorter::multisort($columns, 'seqno', SORT_ASC, SORT_NUMERIC);

            if ($tableColumns !== null) {
                // SQLite may not have an "origin" column in INDEX_LIST
                $index['origin'] = 'c';

                if (!empty($columns) && $tableColumns[$columns[0]['cid']]['pk'] > 0) {
                    $index['origin'] = 'pk';
                } elseif ($index['unique'] && $this->isSystemIdentifier($index['name'])) {
                    $index['origin'] = 'u';
                }
            }

            $ic = (new IndexConstraint())
                ->primary($index['origin'] === 'pk')
                ->unique((bool) $index['unique'])
                ->name($index['name'])
                ->columnNames(ArrayHelper::getColumn($columns, 'name'));

            $result['indexes'][] = $ic;

            if ($index['origin'] === 'u') {
                $ct = (new Constraint())
                    ->name($index['name'])
                    ->columnNames(ArrayHelper::getColumn($columns, 'name'));

                $result['uniques'][] = $ct;
            } elseif ($index['origin'] === 'pk') {
                $ct = (new Constraint())
                    ->columnNames(ArrayHelper::getColumn($columns, 'name'));

                $result['primaryKey'] = $ct;
            }
        }

        if ($result['primaryKey'] === null) {
            /**
             * Additional check for PK in case of INTEGER PRIMARY KEY with ROWID.
             *
             * {@See https://www.sqlite.org/lang_createtable.html#primkeyconst}
             */

            if ($tableColumns === null) {
                $tableColumns = $this->loadTableColumnsInfo($tableName);
            }

            foreach ($tableColumns as $tableColumn) {
                if ($tableColumn['pk'] > 0) {
                    $ct = (new Constraint())
                        ->columnNames([$tableColumn['name']]);

                    $result['primaryKey'] = $ct;
                    break;
                }
            }
        }

        foreach ($result as $type => $data) {
            $this->setTableMetadata($tableName, $type, $data);
        }

        return $result[$returnType];
    }