feat(install): Add db upgrade command

CLI/FetchToncoinTransactions.php

@@ -23,7 +23,9 @@ class FetchToncoinTransactions extends Command
 
     public function __construct()
     {
-        $this->transactions = DatabaseConnection::i()->getContext()->table("cryptotransactions");
+        $ctx = DatabaseConnection::i()->getContext();
+        if (in_array("cryptotransactions", $ctx->getStructure()->getTables()))
+            $this->transactions = $ctx->table("cryptotransactions");
 
         parent::__construct();
     }

CLI/RebuildImagesCommand.php

@@ -20,7 +20,9 @@ class RebuildImagesCommand extends Command
 
     public function __construct()
     {
-        $this->images = DatabaseConnection::i()->getContext()->table("photos");
+        $ctx = DatabaseConnection::i()->getContext();
+        if (in_array("photos", $ctx->getStructure()->getTables()))
+            $this->images = $ctx->table("photos");
 
         parent::__construct();
     }

CLI/UpgradeCommand.php

@@ -0,0 +1,259 @@
+<?php
+
+declare(strict_types=1);
+
+namespace openvk\CLI;
+
+use Nette\Database\Connection;
+use Chandler\Database\DatabaseConnection;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Helper\ProgressBar;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
+use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\Style\SymfonyStyle;
+
+class UpgradeCommand extends Command {
+    protected static $defaultName = "upgrade";
+    
+    private Connection $db;
+    
+    private array $chandlerTables = [
+        "CHANDLERACLPERMISSIONALIASES",
+        "CHANDLERACLGROUPSPERMISSIONS",
+        "CHANDLERACLUSERSPERMISSIONS",
+        "CHANDLERACLRELATIONS",
+        "CHANDLERGROUPS",
+        "CHANDLERTOKENS",
+        "CHANDLERUSERS"
+    ];
+    
+    public function __construct()
+    {
+        $this->db = DatabaseConnection::i()->getConnection();
+        
+        parent::__construct();
+    }
+    
+    protected function configure(): void
+    {
+        $this->setDescription("Upgrade OpenVK installation")
+            ->setHelp("This command upgrades database schema after OpenVK was updated")
+            ->addOption("quick", "Q", InputOption::VALUE_NEGATABLE,
+                "Don't display warning before migrating database", false)
+            ->addOption("repair", "R", InputOption::VALUE_NEGATABLE,
+                "Attempt to repair database schema if tables are missing", false)
+            ->addArgument("chandler", InputArgument::OPTIONAL,
+                "Location of Chandler installation");
+    }
+    
+    protected function checkDatabaseReadiness(bool &$chandlerOk, bool &$ovkOk, bool &$migrationsOk): void
+    {
+        $tables = $this->db->query("SHOW TABLES")->fetchAll();
+        $tables = array_map(fn ($x) => strtoupper($x->offsetGet(0)), $tables);
+        
+        $missingTables = array_diff($this->chandlerTables, $tables);
+        if (sizeof($missingTables) == 0)
+            $chandlerOk = true;
+        else if (sizeof($missingTables) == sizeof($this->chandlerTables))
+            $chandlerOk = null;
+        else
+            $chandlerOk = false;
+        
+        $ovkOk = in_array("PROFILES", $tables);
+        $migrationsOk = in_array("OVK_UPGRADE_HISTORY", $tables);
+    }
+    
+    protected function executeSqlScript(int $errCode, string $script, SymfonyStyle $io, bool $transaction = false): int
+    {
+        $pdo = $this->db->getPdo();
+        
+        $res = false;
+        try {
+            if ($transaction)
+                $res = $pdo->beginTransaction();
+            
+            $res = $pdo->exec($script);
+            
+            if ($transaction)
+                $res = $pdo->commit();
+        } catch (\PDOException $e) {}
+        
+        if ($res === false)
+            goto error;
+        
+        return 0;
+        
+        error:
+        $io->getErrorStyle()->error([
+            "Failed to execute SQL statement:",
+            implode("\t", $pdo->errorInfo())
+        ]);
+        
+        return $errCode;
+    }
+    
+    protected function getNextLevel(): int
+    {
+        $record = $this->db->query("SELECT level FROM ovk_upgrade_history ORDER BY level DESC LIMIT 1");
+        if (!$record->getRowCount())
+            return 0;
+        
+        return $record->fetchField() + 1;
+    }
+    
+    protected function getMigrationFiles(): array
+    {
+        $files = [];
+        $root = dirname(__DIR__ . "/../install/init-static-db.sql");
+        
+        foreach (glob("$root/sqls/*.sql") as $file)
+            $files[(int) basename($file)] = basename($file);
+        
+        ksort($files);
+        
+        return $files;
+    }
+    
+    protected function installChandler(InputInterface $input, SymfonyStyle $io, bool $drop = false): int
+    {
+        $chandlerLocation = $input->getArgument("chandler") ?? (__DIR__ . "/../../../../");
+        $chandlerConfigLocation = "$chandlerLocation/chandler.yml";
+        
+        if (!file_exists($chandlerConfigLocation)) {
+            $err = ["Could not find chandler location. Perhaps your config is too unique?"];
+            if (!$input->getOption("chandler"))
+                $err[] = "Specify absolute path to your chandler installation using the --chandler option.";
+            
+            $io->getErrorStyle()->error($err);
+            
+            return 21;
+        }
+        
+        if ($drop) {
+            $bar = new ProgressBar($io, sizeof($this->chandlerTables));
+            $io->writeln("Dropping chandler tables...");
+            
+            foreach ($bar->iterate($this->chandlerTables) as $table)
+                $this->db->query("DROP TABLE IF EXISTS $table;");
+            
+            $io->newLine();
+        }
+        
+        $installFile = file_get_contents("$chandlerLocation/install/init-db.sql");
+        
+        return $this->executeSqlScript(22, $installFile, $io);
+    }
+    
+    protected function initSchema(SymfonyStyle $io): int
+    {
+        $installFile = file_get_contents(__DIR__ . "/../install/init-static-db.sql");
+        
+        return $this->executeSqlScript(31, $installFile, $io);
+    }
+    
+    protected function initUpgradeLog(SymfonyStyle $io): int
+    {
+        $installFile = file_get_contents(__DIR__ . "/../install/init-migration-table.sql");
+        
+        return $this->executeSqlScript(31, $installFile, $io);
+    }
+    
+    protected function runMigrations(SymfonyStyle $io): int
+    {
+        $nextLevel = $this->getNextLevel();
+        $migrations = array_filter($this->getMigrationFiles(), fn ($id) => $id >= $nextLevel, ARRAY_FILTER_USE_KEY);
+        
+        if (!sizeof($migrations)) {
+            $io->writeln("Database up to date. Nothing left to do.");
+            
+            return 24;
+        }
+        
+        $uname = addslashes(`whoami`);
+        $bar = new ProgressBar($io, sizeof($migrations));
+        $bar->setFormat("very_verbose");
+        
+        foreach ($bar->iterate($migrations) as $num => $migration) {
+            $script = file_get_contents(__DIR__ . "/../install/sqls/$migration");
+            $res = $this->executeSqlScript(100 + $num, $script, $io, true);
+            if ($res != 0) {
+                $io->getErrorStyle()->error("Error while executing migration №$num");
+                
+                return $res;
+            }
+            
+            $t = time();
+            $this->db->query("INSERT INTO ovk_upgrade_history VALUES ($num, $t, \"$uname\");");
+        }
+        
+        return 0;
+    }
+    
+    protected function execute(InputInterface $input, OutputInterface $output): int
+    {
+        $io = new SymfonyStyle($input, $output);
+        
+        if (!$input->getOption("quick")) {
+            $io->writeln("Do full backup of the database before executing this command!");
+            $io->writeln("Command will resume execution after 5 seconds.");
+            $io->writeln("You can skip this warning with --quick option.");
+            sleep(5);
+        }
+        
+        $migrationsOk = false;
+        $chandlerOk = false;
+        $ovkOk = false;
+        
+        $this->checkDatabaseReadiness($chandlerOk, $ovkOk, $migrationsOk);
+        
+        $res = 0;
+        if ($chandlerOk === null) {
+            $io->writeln("Chandler schema not detected, attempting to install...");
+            
+            $res = $this->installChandler($input, $io);
+        } else if ($chandlerOk === false) {
+            if ($input->getOption("repair")) {
+                $io->warning("Chandler schema detected but is broken, attempting to repair...");
+                
+                $res = $this->installChandler($input, $io, true);
+            } else {
+                $io->writeln("Chandler schema detected but is broken");
+                $io->writeln("Run command with --repair to repair (PERMISSIONS WILL BE LOST)");
+                
+                return 1;
+            }
+        }
+        
+        if ($res > 0)
+            return $res;
+        
+        if (!$ovkOk) {
+            $io->writeln("Initializing OpenVK schema...");
+            $res = $this->initSchema($io);
+            if ($res > 0)
+                return $res;
+        }
+        
+        if (!$migrationsOk) {
+            $io->writeln("Initializing upgrade log...");
+            $res = $this->initUpgradeLog($io);
+            if ($res > 0)
+                return $res;
+        }
+        
+        $io->writeln("Upgrading database...");
+        $res = $this->runMigrations($io);
+        
+        if (!$res) {
+            $io->success("Database has been upgraded!");
+            
+            return 0;
+        } else if ($res != 24) {
+            return $res;
+        }
+        
+        return 0;
+    }
+}

install/init-migration-table.sql

@@ -0,0 +1,9 @@
+START TRANSACTION;
+
+CREATE TABLE `ovk_upgrade_history` (
+    `level` smallint UNSIGNED NOT NULL,
+    `timestamp` bigint(20) UNSIGNED NOT NULL,
+    `operator` varchar(256) COLLATE utf8mb4_unicode_520_ci NOT NULL DEFAULT "Maintenance Script"
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_520_ci;
+
+COMMIT;

openvkctl

@@ -12,6 +12,7 @@ $bootstrap = require(__DIR__ . "/../../../chandler/Bootstrap.php");
 $bootstrap->ignite(true);
 
 $application = new Application();
+$application->add(new CLI\UpgradeCommand());
 $application->add(new CLI\RebuildImagesCommand());
 $application->add(new CLI\FetchToncoinTransactions());