Yiisoft\Db\Pgsql\Schema::findColumns()

Complexity

Conditions	18
Paths	24

Size

Total Lines	135
Code Lines	64

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	49
CRAP Score	18.0025

Importance

Changes	5
Bugs	1	Features	0

<?php
            a.attlen AS character_maximum_length,
            pg_catalog.col_description(c.oid, a.attnum) AS column_comment,
            information_schema._pg_truetypmod(a, t) AS modifier,
            NOT (a.attnotnull OR t.typnotnull) AS is_nullable,
            COALESCE(t.typdefault, pg_get_expr(ad.adbin, ad.adrelid)) AS column_default,
            COALESCE(pg_get_expr(ad.adbin, ad.adrelid) ~ 'nextval', false) $orIdentity AS is_autoinc,
            pg_get_serial_sequence(quote_ident(d.nspname) || '.' || quote_ident(c.relname), a.attname)
            AS sequence_name,
            CASE WHEN COALESCE(td.typtype, tb.typtype, t.typtype) = 'e'::char
                THEN array_to_string(
                    (
                        SELECT array_agg(enumlabel)
                        FROM pg_enum
                        WHERE enumtypid = COALESCE(td.oid, tb.oid, a.atttypid)
                    )::varchar[],
                ',')
                ELSE NULL
            END AS enum_values,
            information_schema._pg_numeric_precision(
                COALESCE(td.oid, tb.oid, a.atttypid),
                information_schema._pg_truetypmod(a, t)
            ) AS numeric_precision,
            information_schema._pg_numeric_scale(
                COALESCE(td.oid, tb.oid, a.atttypid),
                information_schema._pg_truetypmod(a, t)
            ) AS numeric_scale,
            information_schema._pg_char_max_length(
                COALESCE(td.oid, tb.oid, a.atttypid),
                information_schema._pg_truetypmod(a, t)
            ) AS size,
            a.attnum = any (ct.conkey) as is_pkey,
            COALESCE(NULLIF(a.attndims, 0), NULLIF(t.typndims, 0), (t.typcategory='A')::int) AS dimension
        FROM
            pg_class c
            LEFT JOIN pg_attribute a ON a.attrelid = c.oid
            LEFT JOIN pg_attrdef ad ON a.attrelid = ad.adrelid AND a.attnum = ad.adnum
            LEFT JOIN pg_type t ON a.atttypid = t.oid
            LEFT JOIN pg_type tb ON (a.attndims > 0 OR t.typcategory='A') AND t.typelem > 0 AND t.typelem = tb.oid
                                        OR t.typbasetype > 0 AND t.typbasetype = tb.oid
            LEFT JOIN pg_type td ON t.typndims > 0 AND t.typbasetype > 0 AND tb.typelem = td.oid
            LEFT JOIN pg_namespace d ON d.oid = c.relnamespace
            LEFT JOIN pg_constraint ct ON ct.conrelid = c.oid AND ct.contype = 'p'
        WHERE
            a.attnum > 0 AND t.typname != '' AND NOT a.attisdropped
            AND c.relname = :tableName
            AND d.nspname = :schemaName
        ORDER BY
            a.attnum;
        SQL;

        $columns = $this->db->createCommand($sql, [
            ':schemaName' => $table->getSchemaName(),
            ':tableName' => $table->getName(),
        ])->queryAll();

        if (empty($columns)) {
            return false;
        }

        /** @psalm-var ColumnArray $info */
        foreach ($columns as $info) {
            /** @psalm-var ColumnArray $info */
            $info = $this->normalizeRowKeyCase($info, false);

            /** @psalm-var ColumnSchema $column */
            $column = $this->loadColumnSchema($info);

            $table->column($column->getName(), $column);

            if ($column->isPrimaryKey()) {
                $table->primaryKey($column->getName());

                if ($table->getSequenceName() === null) {
                    $table->sequenceName($column->getSequenceName());
                }
            }
        }

        return true;
    }

    /**
     * Loads the column information into a {@see ColumnSchemaInterface} object.
     *
     * @psalm-param ColumnArray $info Column information.
     *
     * @return ColumnSchemaInterface The column schema object.
     */
    protected function loadColumnSchema(array $info): ColumnSchemaInterface
    {
        $column = $this->createColumnSchema($info['column_name']);
        $column->allowNull($info['is_nullable']);
        $column->autoIncrement($info['is_autoinc']);
        $column->comment($info['column_comment']);

        if (!in_array($info['type_scheme'], [$this->defaultSchema, 'pg_catalog'], true)) {
            $column->dbType($info['type_scheme'] . '.' . $info['data_type']);
        } else {
            $column->dbType($info['data_type']);
        }

        $column->enumValues($info['enum_values'] !== null
            ? explode(',', str_replace(["''"], ["'"], $info['enum_values']))
            : null);
        $column->unsigned(false); // has no meaning in PG
        $column->primaryKey((bool) $info['is_pkey']);
        $column->precision($info['numeric_precision']);
        $column->scale($info['numeric_scale']);
        $column->size($info['size'] === null ? null : (int) $info['size']);
        $column->dimension($info['dimension']);

        /**
         * pg_get_serial_sequence() doesn't track DEFAULT value change.
         * GENERATED BY IDENTITY columns always have a null default value.
         */
        $defaultValue = $info['column_default'];

        if (
            $defaultValue !== null
            && preg_match("/nextval\\('\"?\\w+\"?\.?\"?\\w+\"?'(::regclass)?\\)/", $defaultValue) === 1
        ) {
            $column->sequenceName(preg_replace(
                ['/nextval/', '/::/', '/regclass/', '/\'\)/', '/\(\'/'],
                '',
                $defaultValue
            ));
        } elseif ($info['sequence_name'] !== null) {
            $column->sequenceName($this->resolveTableName($info['sequence_name'])->getFullName());
        }

        $column->type($this->typeMap[(string) $column->getDbType()] ?? self::TYPE_STRING);
        $column->phpType($this->getColumnPhpType($column));
        $column->defaultValue($this->normalizeDefaultValue($defaultValue, $column));

        return $column;
    }