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

Complexity

Conditions	16
Paths	364

Size

Total Lines	95
Code Lines	50

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	42
CRAP Score	16.5

Importance

Changes

0

<?php
    private function loadTableConstraints($tableName, $returnType)
    {
        $indexes = $this->getDb()->createCommand(
            'PRAGMA INDEX_LIST (' . $this->quoteValue($tableName) . ')'
        )->queryAll();

        $indexes = $this->normalizePdoRowKeyCase($indexes, true);
        $tableColumns = null;

        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);

            ArrayHelper::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();

            $ic->setIsPrimary($index['origin'] === 'pk');
            $ic->setIsUnique((bool) $index['unique']);
            $ic->setName($index['name']);
            $ic->setColumnNames(ArrayHelper::getColumn($columns, 'name'));

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

            if ($index['origin'] === 'u') {
                $ct = new Constraint();

                $ct->setName($index['name']);
                $ct->setColumnNames(ArrayHelper::getColumn($columns, 'name'));

                $result['uniques'][] = $ct;
            } elseif ($index['origin'] === 'pk') {
                $ct = new Constraint();

                $ct->setColumnNames(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();
                    $ct->setColumnNames([$tableColumn['name']]);

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

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

        return $result[$returnType];
    }