AlterPrimaryKeyScript::convertIdField() - Code Metrics - locomotivemtl/charcoal-admin - Measure and Improve Code Quality continuously with Scrutinizer

AlterPrimaryKeyScript::convertIdField() B
last analyzed 2019-03-06 21:44 UTC

↳ Parent: AlterPrimaryKeyScript
Complexity

Conditions	8
Paths	10
Size

Total Lines	83
Code Lines	47
Duplication

Lines	0
Ratio	0 %
Importance

Changes
Metric	Value
cc	8
eloc	47
nc	10
nop	4
dl	0
loc	83
rs	7.9119
c	0
b	0
f	0
How to fix Long Method

<?php

namespace Charcoal\Admin\Script\Object\Table;

use PDO;
use PDOStatement;

use Countable;
use Traversable;
use Exception;
use RuntimeException;
use UnexpectedValueException;
use InvalidArgumentException;

// From PSR-7
use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;

// From 'charcoal-core'
use Charcoal\Model\ModelInterface;

// From 'charcoal-property'
use Charcoal\Property\IdProperty;
use Charcoal\Property\PropertyField;
use Charcoal\Property\PropertyInterface;

// From 'charcoal-app'
use Charcoal\App\Script\ArgScriptTrait;

// From 'charcoal-admin'
use Charcoal\Admin\AdminScript;

/**
 * Alter an object's primary key (SQL source).
 */
class AlterPrimaryKeyScript extends AdminScript
{
    use ArgScriptTrait;

    /**
     * The model to alter.
     *
     * @var ModelInterface|null
     */
    protected $targetModel;

    /**
     * The related models to update.
     *
     * @var ModelInterface[]|null
     */
    protected $relatedModels;

    /**
     * The related model properties to update.
     *
     * @var PropertyInterface[]|null
     */
    protected $relatedProperties;

    /**
     * @return void
     */
    protected function init()
    {
        parent::init();

        $this->setDescription(
            'The <underline>object/table/alter-primary-key</underline> script replaces '.
            'the existing primary key with the new definition from the given model\'s metadata.'
        );
    }

    /**
     * Run the script.
     *
     * @param  RequestInterface  $request  A PSR-7 compatible Request instance.
     * @param  ResponseInterface $response A PSR-7 compatible Response instance.
     * @return ResponseInterface
     */
    public function run(RequestInterface $request, ResponseInterface $response)
    {
        unset($request);

        try {
            $this->start();
        } catch (Exception $e) {
            $this->climate()->error($e->getMessage());
        }

        return $response;
    }

    /**
     * Execute the prime directive.
     *
     * @return self
     */
    public function start()
    {
        $cli = $this->climate();

        $cli->br();
        $cli->bold()->underline()->out('Alter Model\'s Primary Key');
        $cli->br();

        $objType = $this->argOrInput('target_model');
        $this->setTargetModel($objType);
        $this->setTargetModel(/** @scrutinizer ignore-type */ $objType);

        $model  = $this->targetModel();
        $source = $model->source();
        $table  = $source->table();
        $table  = $source->table();

        $cli->comment(sprintf('The "%s" table will be altered.', $table));
        $cli->br();
        $cli->shout('This process is destructive. A backup should be made before proceeding.');
        $cli->br();
        $cli->red()->flank('Analyse your tables before proceeding. Not all fields can or might be affected.', '!');
        $cli->br();

        $input = $cli->confirm('Continue?');
        if ($input->confirmed()) {
            $cli->info('Starting Conversion');
        } else {
            $cli->info('Canceled Conversion');

            return $this;
        }

        $cli->br();

        $db = $source->db();
        $db = $source->db();
        if (!$db) {
            $cli->error(
                'Could not instantiate a database connection.'
            );

            return $this;
        }

        if (!$source->tableExists()) {
        if (!$source->/** @scrutinizer ignore-call */ tableExists()) {
            $cli->error(
                sprintf(
                    'The table "%s" does not exist. This script can only alter existing tables.',
                    $table
                )
            );

            return $this;
        }

        $oldKey = $model->key();
        $newKey = sprintf('%s_new', $model->key());

        $db->query(
            strtr('LOCK TABLES `%table` WRITE;', ['%table' => $table])
        );

        $this->prepareProperties($oldKey, $newKey, $oldProp, $newProp);

        if ($newProp->mode() === $oldProp->mode()) {
            $cli->error(
                sprintf(
                    'The ID is already %s. Canceling conversion.',
                    $this->labelFromMode($newProp)
                )
            );
            $db->query('UNLOCK TABLES;');

            return $this;
        }

        $newField = $this->propertyField($newProp);
        $oldField = $this->propertyField($oldProp);
        $oldField->setExtra('');

        if (!$this->quiet()) {
            $this->describeConversion($newProp);
        }

        $this->convertIdField($newProp, $newField, $oldProp, $oldField);

        $db->query('UNLOCK TABLES;');

        if (!$this->quiet()) {
            $cli->br();
            $cli->info('Success!');
        }

        return $this;
    }



    // Alter Table
    // =========================================================================

    /**
     * Retrieve the old and new ID properties.
     *
     * @param  string          $oldKey  The previous key.
     * @param  string          $newKey  The new key.
     * @param  IdProperty|null $oldProp If provided, then it is filled with an instance of IdProperty.
     * @param  IdProperty|null $newProp If provided, then it is filled with an instance of IdProperty.
     * @return IdProperty[]
     */
    protected function prepareProperties($oldKey, $newKey, &$oldProp = null, &$newProp = null)
    {
        $model  = $this->targetModel();
        $source = $model->source();

        $oldProp = $model->property($oldKey)->setAllowNull(false);
        $newProp = clone $oldProp;
        $newProp->setIdent($newKey);

        $sql  = strtr('SHOW COLUMNS FROM `%table`;', ['%table' => $source->table()]);
        $cols = $source->db()->query($sql, PDO::FETCH_ASSOC);
        foreach ($cols as $col) {
            if ($col['Field'] !== $oldKey) {
                continue;
            }

            if (!$this->quiet()) {
                $this->climate()->comment(
                    sprintf('Evaluating the current `%s` column.', $oldKey)
                );
            }

            if (preg_match('~\bINT\(?(?:$|\b)~i', $col['Type'])) {
                $oldProp->setMode(IdProperty::MODE_AUTO_INCREMENT);
            } elseif (preg_match('~(?:^|\b)(?:VAR)?CHAR\(13\)(?:$|\b)~i', $col['Type'])) {
                $oldProp->setMode(IdProperty::MODE_UNIQID);
            } elseif (preg_match('~(?:^|\b)(?:VAR)?CHAR\(36\)(?:$|\b)~i', $col['Type'])) {
                $oldProp->setMode(IdProperty::MODE_UUID);
            }
        }

        return [
            'old' => $oldProp,
            'new' => $newProp,
        ];
    }

    /**
     * Retrieve a label for the ID's mode.
     *
     * @param  string|IdProperty $mode The mode or property to resolve.
     * @throws UnexpectedValueException If the ID mode is invalid.
     * @return string
     */
    protected function labelFromMode($mode)
    {
        if ($mode instanceof IdProperty) {
            $mode = $mode->mode();
        }

        switch ($mode) {
            case IdProperty::MODE_AUTO_INCREMENT:
                return 'auto-increment';

            case IdProperty::MODE_UNIQID:
                return 'uniqid()';

            case IdProperty::MODE_UUID:
                return 'RFC-4122 UUID';
        }

        throw new UnexpectedValueException(sprintf(
            'The ID mode was not recognized: %s',
            is_object($mode) ? get_class($mode) : gettype($mode)

        ));
    }

    /**
     * Describe what we are converting to.
     *
     * @param  IdProperty $prop The property to analyse.
     * @return self
     */
    protected function describeConversion(IdProperty $prop)
    {
        $cli  = $this->climate();
        $mode = $prop->mode();
        if ($mode === IdProperty::MODE_AUTO_INCREMENT) {
            $cli->comment('Converting to auto-increment ID.');
        } else {
            $label = $this->labelFromMode($mode);
            if ($label) {
                $cli->comment(
                    sprintf('Converting to auto-generated ID (%s).', $label)
                );
            } else {
                $cli->comment('Converting to auto-generated ID.');
            }
        }

        return $this;
    }

    /**
     * Retrieve the given property's field.
     *
     * @param  IdProperty $prop The property to retrieve the field from.
     * @return PropertyField
     */
    protected function propertyField(IdProperty $prop)
    {
        $fields = $prop->fields('');

        return reset($fields);
    }

    /**
     * Retrieve the target model's rows.
     *
     * @return array|Traversable
     */
    private function fetchTargetRows()
    {
        $model  = $this->targetModel();
        $source = $model->source();

        $sql = strtr('SELECT %key FROM `%table`', [
            '%table' => $source->table(),
            '%key'   => $model->key(),
        ]);

        return $source->db()->query($sql, PDO::FETCH_ASSOC);
    }

    /**
     * Describe the given count.
     *
     * @param  array|Traversable $rows The target model's existing rows.
     * @throws InvalidArgumentException If the given argument is not iterable.
     * @return boolean
     */
    private function describeCount($rows = null)
    {
        if ($rows === null) {
            $rows = $this->fetchTargetRows();
        }

        if (!is_array($rows) && !($rows instanceof Traversable)) {

            throw new InvalidArgumentException(
                sprintf(
                    'The rows must be iterable; received %s',
                    is_object($rows) ? get_class($rows) : gettype($rows)
                )
            );
        }

        $cli   = $this->climate();
        $model = $this->targetModel();

        if (is_array($rows) || $rows instanceof Countable) {
            $count = count($rows);
        } elseif ($rows instanceof PDOStatement) {
            $count = $rows->rowCount();
        } else {
            $count = iterator_count($rows);
        }

        if ($count === 0) {
            $cli->comment('The object table is empty.');
            $cli->comment(
                sprintf('Only changing `%s` column.', $model->key())
            );

            return false;
        } elseif ($count === 1) {
            if (!$this->quiet()) {
                $cli->comment('The object table has 1 row.');
            }
        } else {
            if (!$this->quiet()) {
                $cli->comment(
                    sprintf('The object table has %s rows.', $count)
                );
            }
        }

        return true;
    }

    /**
     * Insert the given field.
     *
     * @param  PropertyField $field The new ID field.
     * @param  IdProperty    $prop  The new ID property.
     * @return self
     */
    private function insertNewField(PropertyField $field, IdProperty $prop)
    {
        unset($prop);

        $model  = $this->targetModel();
        $source = $model->source();

        $extra = $field->extra();
        $field->setExtra('');

        // Don't alter table if column name already exists.
        $sql = strtr('SHOW COLUMNS FROM `%table` LIKE "%key"', [
            '%table' => $source->table(),
            '%key'   => $field->ident(),
        ]);

        $res = $source->db()->query($sql);

        if ($res->fetch(1)) {
            // Column name already exists.
            return $this;
        }

        $sql = strtr('ALTER TABLE `%table` ADD COLUMN %field FIRST;', [
            '%table' => $source->table(),
            '%field' => $field->sql(),
        ]);
        $field->setExtra($extra);

        $source->db()->query($sql);

        return $this;
    }

    /**
     * Drop the primary key from the given field.
     *
     * @param  PropertyField $field The previous ID field.
     * @param  IdProperty    $prop  The previous ID property.
     * @return self
     */
    private function dropPrimaryKey(PropertyField $field, IdProperty $prop)
    {
        unset($prop);

        $keepId = $this->climate()->arguments->defined('keep_id');
        $model  = $this->targetModel();
        $source = $model->source();
        $db     = $source->db();
        $key    = $model->key();

        if ($keepId) {
            $field->setIdent(sprintf('%1$s_%2$s', $key, date('Ymd_His')));
            $sql = strtr(
                'ALTER TABLE `%table`
                    CHANGE COLUMN `%key` %field,
                    DROP PRIMARY KEY;',
                [
                    '%table' => $source->table(),
                    '%field' => $field->sql(),
                    '%key'   => $key,
                ]
            );
        } else {
            $sql = strtr(
                'ALTER TABLE `%table`
                    MODIFY COLUMN %field,
                    DROP PRIMARY KEY;',
                [
                    '%table' => $source->table(),
                    '%field' => $field->sql(),
                ]
            );
        }

        $db->query($sql);

        return $this;
    }

    /**
     * Set the given field as the primary key.
     *
     * @param  PropertyField $field The new ID field.
     * @param  IdProperty    $prop  The new ID property.
     * @return self
     */
    private function applyPrimaryKey(PropertyField $field, IdProperty $prop)
    {
        unset($prop);

        $model  = $this->targetModel();
        $source = $model->source();

        $sql = strtr(
            'ALTER TABLE `%table` ADD PRIMARY KEY (`%key`);',
            [
                '%table' => $source->table(),
                '%key'   => $field->ident(),
            ]
        );
        $source->db()->query($sql);

        return $this;
    }

    /**
     * Rename the given field.
     *
     * @param  PropertyField $field The field to rename.
     * @param  string        $from  The original field key.
     * @param  string        $to    The new field key.
     * @return self
     */
    private function renameColumn(PropertyField $field, $from, $to)
    {
        $model  = $this->targetModel();
        $source = $model->source();
        $key    = $model->key();


        $field->setIdent($to);
        $sql = strtr(
            'ALTER TABLE `%table` CHANGE COLUMN `%from` %field;',
            [
                '%table' => $source->table(),
                '%field' => $field->sql(),
                '%from'  => $from,
                '%to'    => $to,
            ]
        );
        $source->db()->query($sql);

        return $this;
    }

    /**
     * Remove the given field.
     *
     * @param  PropertyField $field The field to remove.
     * @return self
     */
    private function removeColumn(PropertyField $field)
    {
        $source = $this->targetModel()->source();

        $sql = strtr(
            'ALTER TABLE `%table` DROP COLUMN `%key`;',
            [
                '%table' => $source->table(),
                '%key'   => $field->ident(),
            ]
        );
        $source->db()->query($sql);

        return $this;
    }

    /**
     * Convert a primary key column from one format to another.
     *
     * @param  IdProperty    $newProp  The new ID property.
     * @param  PropertyField $newField The new ID field.
     * @param  IdProperty    $oldProp  The previous ID property.
     * @param  PropertyField $oldField The previous ID field.
     * @throws InvalidArgumentException If the new property does not implement the proper mode.
     * @return self
     */
    protected function convertIdField(
        IdProperty $newProp,
        PropertyField $newField,
        IdProperty $oldProp,
        PropertyField $oldField
    ) {
        $cli = $this->climate();

        $keepId = $cli->arguments->defined('keep_id');
        $model  = $this->targetModel();
        $source = $model->source();
        $table  = $source->table();
        $db     = $source->db();


        $newKey = $newProp->ident();
        $oldKey = $oldProp->ident();

        $this->insertNewField($newField, $newProp);

        $rows = $this->fetchTargetRows();
        if ($this->describeCount($rows)) {
            if (!$this->quiet()) {
                $cli->br();
                $progress = $cli->progress($rows->rowCount());
                $progress = $cli->progress($rows->/** @scrutinizer ignore-call */ rowCount());
            }

            if ($newProp->mode() === IdProperty::MODE_AUTO_INCREMENT) {
                $pool = 0;
                $ids  = function () use (&$pool) {
                    return ++$pool;
                };
            } else {
                $pool = [];
                $ids  = function () use (&$pool, $newProp) {
                    $id = $newProp->autoGenerate();
                    while (in_array($id, $pool)) {
                        $id = $newProp->autoGenerate();
                    }

                    $pool[] = $id;

                    return $id;
                };
            }

            foreach ($rows as $row) {
                $id  = $ids();
                $sql = strtr('UPDATE `%table` SET `%newKey` = :new WHERE `%oldKey` = :old;', [
                    '%table'  => $table,
                    '%newKey' => $newKey,
                    '%oldKey' => $oldKey,
                ]);
                $source->dbQuery(
                         dbQuery(
                    $sql,
                    [
                        'new' => $id,
                        'old' => $row[$oldKey],
                    ],
                    [
                        'new' => $newField->sqlPdoType(),
                        'old' => $oldField->sqlPdoType(),
                    ]
                );

                if (!$this->quiet()) {
                    $progress->advance();

                }
            }
        }

        $this->dropPrimaryKey($oldField, $oldProp);
        $this->applyPrimaryKey($newField, $newProp);

        /** @todo Alter related tables */
        $this->syncRelatedFields($newProp, $newField, $oldProp, $oldField);

        if (!$keepId) {
            $this->removeColumn($oldField);
        }

        $this->renameColumn($newField, $newKey, $oldKey);

        return $this;
    }

    /**
     * Sync the new primary keys to related models.
     *
     * @param  IdProperty    $newProp  The new ID property.
     * @param  PropertyField $newField The new ID field.
     * @param  IdProperty    $oldProp  The previous ID property.
     * @param  PropertyField $oldField The previous ID field.
     * @throws InvalidArgumentException If the new property does not implement the proper mode.
     * @return self
     */
    protected function syncRelatedFields(
        IdProperty $newProp,
        PropertyField $newField,
        IdProperty $oldProp,
        PropertyField $oldField
    ) {
        unset($newProp, $oldProp, $oldField);

        $cli = $this->climate();
        if (!$this->quiet()) {
            $cli->br();
            $cli->comment('Syncing new IDs to related tables.');
        }

        $related = $cli->arguments->get('related_model');
        if (!$related) {
            $cli->br();
            $input = $cli->confirm('Are there any model(s) related to the target?');
            if (!$input->confirmed()) {
                return $this;
            }

            $related = $this->argOrInput('related_model');
        }
        $this->setRelatedModels($related);

        $target = $this->targetModel();
        $table  = $target->source()->table();
        foreach ($this->relatedModels() as $model) {
            $src = $model->source();
            $tbl = $src->table();
            $db  = $src->db();

            $db->query(
                strtr(
                    'LOCK TABLES
                        `%relatedTable` AS a WRITE,
                        `%sourceTable` AS b WRITE;',
                    [
                        '%relatedTable' => $tbl,
                        '%sourceTable'  => $table,
                    ]
                )
            );

            $sql = strtr(
                'UPDATE `%relatedTable` AS a
                    JOIN `%sourceTable` AS b ON a.`%prop` = b.`%oldKey`
                    SET a.`%prop` = b.`%newKey`;',
                [
                    '%relatedTable' => $tbl,
                    '%prop'         => $this->relatedProperties[$model->objType()],
                    '%prop'         => $this->relatedProperties[$model->/** @scrutinizer ignore-call */ objType()],
                    '%sourceTable'  => $table,
                    '%newKey'       => $newField->ident(),
                    '%oldKey'       => $target->key(),
                ]
            );
            $db->query($sql);

            $db->query('UNLOCK TABLES;');
        }

        return $this;
    }



    // CLI Arguments
    // =========================================================================

    /**
     * Retrieve the script's supported arguments.
     *
     * @return array
     */
    public function defaultArguments()
    {
        static $arguments;

        if ($arguments === null) {
            $validateModel = function ($response) {
                if (strlen($response) === 0) {
                    return false;
                }

                try {
                    $this->modelFactory()->get($response);
                } catch (Exception $e) {
                    unset($e);

                    return false;
                }

                return true;
            };

            $validateModels = function ($response) {
                if (strlen($response) === 0) {
                    return false;
                }

                try {
                    $arr = $this->parseAsArray($response);
                    foreach ($arr as $model) {
                        $this->resolveRelatedModel($model);
                    }
                } catch (Exception $e) {
                    unset($e);

                    return false;
                }

                return true;
            };

            $arguments = [
                'keep_id'       => [
                    'longPrefix'  => 'keep-id',
                    'noValue'     => true,
                    'description' => 'Skip the deletion of the ID field to be replaced.',
                ],
                'target_model'  => [
                    'prefix'      => 'o',
                    'longPrefix'  => 'obj-type',
                    'required'    => true,
                    'description' => 'The object type to alter.',
                    'prompt'      => 'What model must be altered?',
                    'acceptValue' => $validateModel->bindTo($this)
                ],
                'related_model' => [
                    'prefix'      => 'r',
                    'longPrefix'  => 'related-obj-type',
                    'description' => 'Properties of related object types to synchronize (ObjType:propertyIdent,…).',
                    'prompt'      => 'List related models and properties (ObjType:propertyIdent,…):',
                    'acceptValue' => $validateModels->bindTo($this)
                ]
            ];

            $arguments = array_merge(parent::defaultArguments(), $arguments);
        }

        return $arguments;
    }

    /**
     * Retrieve the script's parent arguments.
     *
     * Useful for specialty classes extending this one that might not want
     * options for selecting specific objects.
     *
     * @return array
     */
    public function parentArguments()
    {
        return parent::defaultArguments();
    }

    /**
     * Set the model to alter.
     *
     * @param  string|ModelInterface $model An object model.
     * @throws InvalidArgumentException If the given argument is not a model.
     * @return self
     */
    public function setTargetModel($model)
    {
        if (is_string($model)) {
            $model = $this->modelFactory()->get($model);
        }

        if (!$model instanceof ModelInterface) {
            throw new InvalidArgumentException(
                sprintf(
                    'The model must be an instance of "%s"',
                    ModelInterface::class
                )
            );
        }

        $this->targetModel = $model;

        return $this;
    }

    /**
     * Retrieve the model to alter.
     *
     * @throws RuntimeException If a target model has not been defined.
     * @return ModelInterface
     */
    public function targetModel()
    {
        if (!isset($this->targetModel)) {
            throw new RuntimeException('A model must be targeted.');
        }

        return $this->targetModel;
    }

    /**
     * Set the related models to update.
     *
     * @param  string|array $models One or more object models.
     * @throws InvalidArgumentException If the given argument is not a model.
     * @return self
     */
    public function setRelatedModels($models)
    {
        $models = $this->parseAsArray($models);
        foreach ($models as $i => $model) {
            if (is_string($model)) {
                list($model, $prop) = $this->resolveRelatedModel($model);
                $models[$i]                                 = $model;
                $this->relatedProperties[$model->objType()] = $prop;
            } elseif ($model instanceof ModelInterface) {
                if (!isset($this->relatedProperties[$model->objType()])) {
                    throw new InvalidArgumentException(
                        sprintf(
                            'The related model [%s] requires a target property',
                            get_class($model)
                        )
                    );
                }
                $models[$i] = $model;
            } else {
                throw new InvalidArgumentException(
                    sprintf(
                        'A related model must be defined as "%s"',
                        'ObjType:propertyIdent'
                    )
                );
            }
        }

        $this->relatedModels = $models;

        return $this;
    }

    /**
     * Resolve the given related model.
     *
     * @param  string $pattern A 'model:property' identifier.
     * @throws InvalidArgumentException If the identifier is invalid.
     * @return array Returns an array containing a ModelInterface and a property identifier.
     */
    protected function resolveRelatedModel($pattern)
    {
        list($class, $prop) = array_pad($this->parseAsArray($pattern, ':'), 2, null);
        $model = $this->modelFactory()->get($class);

        if (!$prop) {
            throw new InvalidArgumentException(
                sprintf(
                    'The related model [%s] requires a target property',
                    get_class($model)
                )
            );
        }

        $metadata = $model->metadata();
        if (!$metadata->property($prop)) {
            throw new InvalidArgumentException(
                sprintf(
                    'The related model [%1$s] does not have the target property [%2$s]',
                    $class,
                    (is_string($prop) ? $prop : gettype($prop))
                )
            );
        }

        return [$model, $prop];
    }

    /**
     * Retrieve the related models to update.
     *
     * @return ModelInterface[]|null
     */
    public function relatedModels()
    {
        return $this->relatedModels;
    }
}

locomotivemtl / charcoal-admin

AlterPrimaryKeyScript::convertIdField() B last analyzed 2019-03-06 21:44 UTC

Complexity

Size

Duplication

Importance

How to fix Long Method

Long Method

Duplication Side-by-Side

Filter issues like

AlterPrimaryKeyScript::convertIdField() B
last analyzed 2019-03-06 21:44 UTC
