schema.php

<?php
namespace Bitrix\Perfmon\Sql;
 
use Bitrix\Main\NotSupportedException;
 
//Sample usage:
/*
\Bitrix\Main\Loader::includeModule('perfmon');
$dir = new \Bitrix\Main\IO\Directory("/opt/php03.cp1251.www/mercurial/bitrix/modules");
foreach ($dir->getChildren()  as $child)
{
    if ($child->isDirectory() && $child->getName()!=='xxx')
    {
        echo $child->getName(),": ";
 
        foreach (array("mysql"=>";", "mssql"=>"GO", "oracle"=>"/") as $db=>$delimiter)
        {
            $path = $child->getPath()."/install/db/$db/install.sql";
            if (!\Bitrix\Main\IO\File::isFileExists($path))
                $path = $child->getPath()."/install/$db/install.sql";
            if (!\Bitrix\Main\IO\File::isFileExists($path))
                continue;
            //echo "<br>$path<br>";
            $sql = \Bitrix\Main\IO\File::getFileContents($path);
            $s = new \Bitrix\Perfmon\Sql\Schema;
            $s->createFromString($sql, $delimiter);
            // p r i n t _ r ($s->tables);
            echo count($s->tables->getList())," ";
        }
        echo "\n";
    }
}
*/
class Schema
{
    public $tables = null;
    public $procedures = null;
    public $sequences = null;
 
    public function __construct()
    {
        $this->tables = new Collection;
        $this->procedures = new Collection;
        $this->sequences = new Collection;
    }
 
    public function createFromString($str, $delimiter)
    {
        $tokenizer = Tokenizer::createFromString($str);
        foreach ($this->splitStatements($tokenizer, $delimiter) as $statement)
        {
            $this->executeStatement($statement);
        }
    }
 
    protected function splitStatements(Tokenizer $tokenizer, $delimiter = ';')
    {
        $result = [];
        $index = 0;
        $result[$index] = [];
 
        $prevToken = null;
        foreach ($tokenizer->getTokens() as $token)
        {
            if (
                $token->text === $delimiter
                && $prevToken
                && mb_strpos($prevToken->text, "\n") !== false
            )
            {
                $index++;
                $result[$index] = [];
            }
            elseif (
                mb_strpos($token->text, "\n") !== false
                && $prevToken
                && $prevToken->text === $delimiter
            )
            {
                array_pop($result[$index]);
                $index++;
                $result[$index] = [];
            }
            else
            {
                $result[$index][] = $token;
            }
            $prevToken = $token;
        }
 
        foreach ($result as $i => $tokens)
        {
            $result[$i] = Tokenizer::createFromTokens($tokens);
        }
 
        return $result;
    }
 
    protected function executeStatement(Tokenizer $tokenizer)
    {
        $tokenizer->resetState();
        $tokenizer->skipWhiteSpace();
        if ($tokenizer->testUpperText('CREATE'))
        {
            $this->executeCreate($tokenizer);
        }
        elseif ($tokenizer->testUpperText('INSERT'))
        {
            //skip insert into
        }
        elseif ($tokenizer->testUpperText('SET'))
        {
            //skip set identity_insert
        }
        elseif ($tokenizer->testUpperText('ALTER'))
        {
            $this->executeAlter($tokenizer);
        }
        elseif ($tokenizer->testUpperText('IF'))
        {
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testUpperText('OBJECT_ID'))
            {
                while (!$tokenizer->endOfInput())
                {
                    if ($tokenizer->nextToken()->upper === 'CREATE')
                    {
                        break;
                    }
                }
                $tokenizer->nextToken();
                $tokenizer->skipWhiteSpace();
                if ($tokenizer->testUpperText('TABLE'))
                {
                    $this->executeCreateTable($tokenizer);
                }
                else
                {
                    throw new NotSupportedException("'CREATE TABLE' expected. line:" . $tokenizer->getCurrentToken()->line);
                }
            }
            elseif ($tokenizer->testUpperText('NOT'))
            {
                $tokenizer->skipWhiteSpace();
                if ($tokenizer->testUpperText('EXISTS'))
                {
                    while (!$tokenizer->endOfInput())
                    {
                        if ($tokenizer->nextToken()->upper === 'CREATE')
                        {
                            break;
                        }
                    }
                    $tokenizer->nextToken();
                    $tokenizer->skipWhiteSpace();
 
                    if ($tokenizer->testUpperText('UNIQUE'))
                    {
                        $unique = true;
                        $tokenizer->skipWhiteSpace();
                    }
                    else
                    {
                        $unique = false;
                    }
 
                    if ($tokenizer->testUpperText('INDEX'))
                    {
                        $this->executeCreateIndex($tokenizer, $unique);
                    }
                    else
                    {
                        throw new NotSupportedException("'CREATE INDEX' expected. line:" . $tokenizer->getCurrentToken()->line);
                    }
                }
                else
                {
                    throw new NotSupportedException("'NOT EXISTS' expected. line:" . $tokenizer->getCurrentToken()->line);
                }
            }
            else
            {
                throw new NotSupportedException("'OBJECT_ID' expected. line:" . $tokenizer->getCurrentToken()->line);
            }
        }
        elseif (!$tokenizer->endOfInput())
        {
            throw new NotSupportedException("'CREATE' expected. line:" . $tokenizer->getCurrentToken()->line);
        }
    }
 
    protected function executeCreate(Tokenizer $tokenizer)
    {
        $tokenizer->skipWhiteSpace();
        if ($tokenizer->testUpperText('OR'))
        {
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testUpperText('REPLACE'))
            {
                $tokenizer->skipWhiteSpace();
            }
            else
            {
                throw new NotSupportedException("'OR REPLACE' expected. line:" . $tokenizer->getCurrentToken()->line);
            }
        }
 
        if ($tokenizer->testUpperText('TABLE'))
        {
            $this->executeCreateTable($tokenizer);
        }
        elseif ($tokenizer->testUpperText('INDEX'))
        {
            $this->executeCreateIndex($tokenizer, false);
        }
        elseif ($tokenizer->testUpperText('UNIQUE'))
        {
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testUpperText('INDEX'))
            {
                $tokenizer->skipWhiteSpace();
            }
 
            $this->executeCreateIndex($tokenizer, true);
        }
        elseif ($tokenizer->testUpperText('FULLTEXT'))
        {
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testUpperText('INDEX'))
            {
                $tokenizer->skipWhiteSpace();
            }
 
            $this->executeCreateIndex($tokenizer, false, true);
        }
        elseif ($tokenizer->testUpperText('TRIGGER'))
        {
            $this->executeCreateTrigger($tokenizer);
        }
        elseif (
            $tokenizer->testUpperText('PROCEDURE')
            || $tokenizer->testUpperText('FUNCTION')
            || $tokenizer->testUpperText('TYPE')
        )
        {
            $this->executeCreateProcedure($tokenizer);
        }
        elseif ($tokenizer->testUpperText('SEQUENCE'))
        {
            $this->executeCreateSequence($tokenizer);
        }
        else
        {
            throw new NotSupportedException('TABLE|INDEX|UNIQUE|TRIGGER|PROCEDURE|FUNCTION|TYPE|SEQUENCE expected. line:' . $tokenizer->getCurrentToken()->line);
        }
    }
 
    protected function executeAlter(Tokenizer $tokenizer)
    {
        $tokenizer->skipWhiteSpace();
        if ($tokenizer->testUpperText('TABLE'))
        {
            $tokenizer->skipWhiteSpace();
            $tableName = $tokenizer->getCurrentToken()->text;
            $table = $this->tables->search($tableName);
            if (!$table)
            {
                throw new NotSupportedException("Table [${tableName}] not found. line: " . $tokenizer->getCurrentToken()->line);
            }
            $tokenizer->nextToken();
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testUpperText('ADD'))
            {
                $tokenizer->skipWhiteSpace();
                if ($tokenizer->testUpperText('CONSTRAINT'))
                {
                    $tokenizer->skipWhiteSpace();
                    $table->createConstraint($tokenizer);
                }
            }
            elseif ($tokenizer->testUpperText('MODIFY'))
            {
                $tokenizer->skipWhiteSpace();
                $table->modifyColumn($tokenizer);
            }
            elseif ($tokenizer->testUpperText('NOCHECK') || $tokenizer->testUpperText('CHECK'))
            {
                //(NOCHECK|CHECK) CONSTRAINT ALL
            }
            elseif ($tokenizer->testUpperText('DISABLE') || $tokenizer->testUpperText('ENABLE'))
            {
                //(DISABLE|ENABLE) TRIGGER ALL
            }
            else
            {
                throw new NotSupportedException("'ADD' expected. line:" . $tokenizer->getCurrentToken()->line);
            }
        }
        else
        {
            throw new NotSupportedException("'TABLE' expected. line:" . $tokenizer->getCurrentToken()->line);
        }
    }
 
    protected function executeCreateTable(Tokenizer $tokenizer)
    {
        $tokenizer->skipWhiteSpace();
        $this->tables->add(Table::create($tokenizer));
    }
 
    protected function executeCreateIndex(Tokenizer $tokenizer, $unique, $fulltext = false)
    {
        $tokenizer->skipWhiteSpace();
 
        $tokenizer->setBookmark();
 
        Index::searchTableName($tokenizer);
        $tableName = $tokenizer->getCurrentToken()->text;
 
        $table = $this->tables->search($tableName);
        if (!$table)
        {
            $table = new Table($tableName);
            $this->tables->add($table);
        }
 
        $tokenizer->restoreBookmark();
 
        $table->createIndex($tokenizer, $unique, $fulltext);
    }
 
    protected function executeCreateTrigger(Tokenizer $tokenizer)
    {
        $tokenizer->skipWhiteSpace();
 
        $tokenizer->setBookmark();
 
        Trigger::searchTableName($tokenizer);
        $tableName = $tokenizer->getCurrentToken()->text;
 
        $table = $this->tables->search($tableName);
        if (!$table)
        {
            throw new NotSupportedException("Table [${tableName}] not found. line: " . $tokenizer->getCurrentToken()->line);
        }
 
        $tokenizer->restoreBookmark();
 
        $table->createTrigger($tokenizer);
    }
 
    protected function executeCreateProcedure(Tokenizer $tokenizer)
    {
        $tokenizer->putBack();
        $this->procedures->add(Procedure::create($tokenizer));
    }
 
    protected function executeCreateSequence(Tokenizer $tokenizer)
    {
        $tokenizer->skipWhiteSpace();
        $this->sequences->add(Sequence::create($tokenizer));
    }
}