updater.php

<?php
namespace Bitrix\Perfmon\Sql;
 
use Bitrix\Main\NotSupportedException;
use Bitrix\Perfmon\Php;
 
class Updater
{
    protected $dbType = '';
    protected $delimiter = '';
    protected $tableCheck = null;
    protected $columns = [];
 
    protected  $statements = [];
 
    public function setDbType($dbType = '')
    {
        $this->dbType = (string)$dbType;
        return $this;
    }
 
    public function setDelimiter($delimiter = '')
    {
        $this->delimiter = (string)$delimiter;
        return $this;
    }
 
    public function getStatements()
    {
        return $this->statements;
    }
 
    public function generate($sourceSql, $targetSql)
    {
        $source = new Schema;
        $source->createFromString($sourceSql, $this->delimiter);
 
        $target = new Schema;
        $target->createFromString($targetSql, $this->delimiter);
 
        $diff = Compare::diff($source ,$target);
        if ($diff)
        {
            $sourceTables = $source->tables->getList();
            if ($sourceTables)
            {
                $this->tableCheck = array_shift($sourceTables);
            }
            else
            {
                $targetTables = $target->tables->getList();
                if ($targetTables)
                {
                    $this->tableCheck = array_shift($targetTables);
                }
                else
                {
                    $this->tableCheck = null;
                }
            }
 
            if (!$this->tableCheck)
            {
                throw new NotSupportedException('no CHECK TABLE found.');
            }
 
            return $this->handle($diff);
        }
        else
        {
            return '';
        }
    }
 
    protected function handle(array $diff)
    {
        $this->columns = [];
        foreach ($diff as $pair)
        {
            if (!isset($pair[0]))
            {
                $this->handleCreate($pair[1]);
            }
            elseif (!isset($pair[1]))
            {
                $this->handleDrop($pair[0]);
            }
            else
            {
                $this->handleChange($pair[0], $pair[1]);
            }
        }
 
        foreach ($this->columns as $columns)
        {
            $ddl = ''; $predicate2 = [];
            foreach ($columns as $column)
            {
                $predicate2[] = $column[0];
                if ($ddl)
                {
                    $ddl .= ', ' . preg_replace('/^ALTER TABLE [^ ]+ /', '', $column[1]);
                }
                else
                {
                    $ddl = $column[1];
                }
            }
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '");');
            $stmt->dependOn = $columns[0][2];
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($columns[0][3]);
            $stmt->addCondition($columns[0][0]);
 
            $this->statements[] = $stmt;
        }
 
        $updaterSteps = $this->getStatements();
        $codeTree = new \Bitrix\Perfmon\Php\CodeTree($updaterSteps);
        $result = $codeTree->getCode(0);
 
        return $result;
    }
 
    protected function handleCreate(BaseObject $object)
    {
        $stmt = null;
 
        if ($object instanceof Sequence || $object instanceof Procedure)
        {
            $ddl = $object->getCreateDdl($this->dbType);
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->dependOn = $this->tableCheck->getLowercasedName();
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition('$updater->TableExists("' . EscapePHPString($this->tableCheck->getLowercasedName()) . '")');
        }
        elseif ($object instanceof Table)
        {
            $ddl = $object->getCreateDdl($this->dbType);
            $predicate = '!$updater->TableExists("' . EscapePHPString($object->name) . '")';
            $cond = "\tif (${predicate})\n";
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->tableName = $object->getLowercasedName();
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            if ($this->tableCheck->getLowercasedName() !== $object->getLowercasedName())
            {
                $stmt->addCondition('$updater->TableExists("' . EscapePHPString($this->tableCheck->getLowercasedName()) . '")');
            }
            $stmt->addCondition('!$updater->TableExists("' . EscapePHPString($object->getLowercasedName()) . '")');
        }
        elseif ($object instanceof Column)
        {
            $ddl = $object->getCreateDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->getLowercasedName()) . '")';
            $cond = "\t\tif (${predicate})\n";
            $predicate2 = '!$DB->Query("SELECT ' . EscapePHPString($object->name) . ' FROM ' . EscapePHPString($object->parent->getLowercasedName()) . ' WHERE 1=0", true)';
 
            $this->columns[$cond][] = [$predicate2, $ddl, $object->parent->getLowercasedName(), $predicate, $predicate2];
        }
        elseif ($object instanceof Index)
        {
            $ddl = $object->getCreateDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->getLowercasedName()) . '")';
            $predicate2 = '!$DB->IndexExists("' . EscapePHPString($object->parent->getUnquotedName()) . '", array(' . $this->multiLinePhp('"', $object->getUnquotedName($object->columns), '", ') . '), true)';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '");');
            $stmt->dependOn = $object->parent->getLowercasedName();
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
            $stmt->addCondition($predicate2);
        }
        elseif ($object instanceof Trigger || $object instanceof Constraint)
        {
            $ddl = $object->getCreateDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->getLowercasedName()) . '")';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->dependOn = $object->parent->getLowercasedName();
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
        }
        else
        {
            $stmt = $this->createStatement('', '//create for ' . get_class($object) . ' not supported yet', '');
        }
 
        if ($stmt)
        {
            $this->statements[] = $stmt;
        }
    }
 
    protected function handleDrop(BaseObject $object)
    {
        $stmt = false;
 
        if ($object instanceof Sequence || $object instanceof Procedure)
        {
            $ddl = $object->getDropDdl($this->dbType);
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition('$updater->TableExists("' . EscapePHPString($this->tableCheck->getLowercasedName()) . '")');
        }
        elseif ($object instanceof Table)
        {
            $ddl = $object->getDropDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->getLowercasedName()) . '")';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '");');
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
        }
        elseif ($object instanceof Column)
        {
            $ddl = $object->getDropDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->name) . '")';
            $cond = "\t\tif (${predicate})\n";
            $predicate2 = '$DB->Query("SELECT ' . EscapePHPString($object->name) . ' FROM ' . EscapePHPString($object->parent->getLowercasedName()) . ' WHERE 1=0", true)';
 
            $this->columns[$cond][] = [$predicate2, $ddl, $object->parent->getLowercasedName(), $predicate, $predicate2];
        }
        elseif ($object instanceof Index)
        {
            $ddl = $object->getDropDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->getLowercasedName()) . '")';
            $predicate2 = '$DB->IndexExists("' . EscapePHPString($object->parent->getUnquotedName()) . '", array(' . $this->multiLinePhp('"', $object->getUnquotedName($object->columns), '", ') . '), true)';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '");');
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
            $stmt->addCondition($predicate2);
        }
        elseif ($object instanceof Trigger || $object instanceof Constraint)
        {
            $ddl = $object->getDropDdl($this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($object->parent->getLowercasedName()) . '")';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
        }
        else
        {
            $stmt = $this->createStatement('', '//drop for ' . get_class($object) . ' not supported yet', '');
        }
 
        if ($stmt)
        {
            $this->statements[] = $stmt;
        }
    }
 
    protected function handleChange(BaseObject $source, BaseObject $target)
    {
        $stmt = null;
 
        if ($source instanceof Sequence || $source instanceof Procedure)
        {
            $dropStmt = $this->createStatement('$DB->Query("', $source->getDropDdl($this->dbType), '", true);');
            $createStmt = $this->createStatement('$DB->Query("', $target->getCreateDdl($this->dbType), '", true);');
            $stmt = new Php\Statement;
            $stmt->dependOn = $this->tableCheck->getLowercasedName();
            $stmt->merge($dropStmt);
            $stmt->merge($createStmt);
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition('$updater->TableExists("' . EscapePHPString($this->tableCheck->getLowercasedName()) . '")');
        }
        elseif ($target instanceof Column)
        {
            $ddl = $source->getModifyDdl($target, $this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($source->parent->getLowercasedName()) . '")';
            $cond = "\t\tif (${predicate})\n";
            $predicate2 = '$DB->Query("SELECT ' . EscapePHPString($source->name) . ' FROM ' . EscapePHPString($source->parent->getLowercasedName()) . ' WHERE 1=0", true)';
 
            $this->columns[$cond][] = [$predicate2, $ddl, $source->parent->getLowercasedName(), $predicate, $predicate2];
        }
        elseif ($source instanceof Index)
        {
            $predicate = '$updater->TableExists("' . EscapePHPString($source->parent->getLowercasedName()) . '")';
            $predicate2 = '$DB->IndexExists("' . EscapePHPString($source->parent->getUnquotedName()) . '", array(' . $this->multiLinePhp('"', $source->getUnquotedName($source->columns), '", ') . '), true)';
 
            $dropStmt = $this->createStatement('$DB->Query("', $source->getDropDdl($this->dbType), '", true);');
            $createStmt = $this->createStatement('$DB->Query("', $target->getCreateDdl($this->dbType), '", true);');
            $stmt = new Php\Statement;
            $stmt->dependOn = $source->parent->getLowercasedName();
            $stmt->merge($dropStmt);
            $stmt->merge($createStmt);
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
            $stmt->addCondition($predicate2);
            $stmt->addCondition('!$DB->IndexExists("' . EscapePHPString($target->parent->getUnquotedName()) . '", array(' . $this->multiLinePhp('"', $source->getUnquotedName($source->columns), '", ') . '), true)');
        }
        elseif ($source instanceof Trigger || $source instanceof Constraint)
        {
            $ddl = $source->getModifyDdl($target, $this->dbType);
            $predicate = '$updater->TableExists("' . EscapePHPString($source->parent->getLowercasedName()) . '")';
 
            $stmt = $this->createStatement('$DB->Query("', $ddl, '", true);');
            $stmt->dependOn = $source->parent->getLowercasedName();
            $stmt->addCondition('$updater->CanUpdateDatabase()');
            $stmt->addCondition('$DB->type == "' . EscapePHPString($this->dbType) . '"');
            $stmt->addCondition($predicate);
        }
        else
        {
            $stmt = $this->createStatement('', '//change for ' . get_class($source) . ' not supported yet', '');
        }
 
        if ($stmt)
        {
            $this->statements[] = $stmt;
        }
    }
 
    protected function multiLinePhp($prefix, $body, $suffix)
    {
        $result  = [];
        if (is_array($body))
        {
            foreach ($body as $line)
            {
                $result[] = $prefix . EscapePHPString($line) . $suffix;
            }
        }
        else
        {
            $result[] = $prefix . EscapePHPString($body) . $suffix;
        }
        return implode('', $result);
    }
 
    protected function createStatement($prefix, $body, $suffix)
    {
        $result  = new Php\Statement;
        if (is_array($body))
        {
            foreach ($body as $line)
            {
                $result->addLine($prefix . EscapePHPString($line) . $suffix);
            }
        }
        else
        {
            $result->addLine($prefix . EscapePHPString($body) . $suffix);
        }
        return $result;
    }
}