generated from smedev/Template-for-SMEServer-Contribs-Package
Add in software files and templates
This commit is contained in:
121
root/opt/dmarc-srg/classes/Database/DatabaseUpgrader.php
Normal file
121
root/opt/dmarc-srg/classes/Database/DatabaseUpgrader.php
Normal file
@@ -0,0 +1,121 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* dmarc-srg - A php parser, viewer and summary report generator for incoming DMARC reports.
|
||||
* Copyright (C) 2020 Aleksey Andreev (liuch)
|
||||
*
|
||||
* Available at:
|
||||
* https://github.com/liuch/dmarc-srg
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the Free
|
||||
* Software Foundation, either version 3 of the License.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||
* more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along with
|
||||
* this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
namespace Liuch\DmarcSrg\Database;
|
||||
|
||||
use PDO;
|
||||
use Exception;
|
||||
use Liuch\DmarcSrg\Settings\SettingString;
|
||||
|
||||
class DatabaseUpgrader
|
||||
{
|
||||
public static function go()
|
||||
{
|
||||
$ver = (new SettingString('version'))->value();
|
||||
if ($ver == '') {
|
||||
$ver = 'null';
|
||||
}
|
||||
|
||||
while ($ver !== Database::REQUIRED_VERSION) {
|
||||
if (!isset(self::$upways['ver_' . $ver])) {
|
||||
throw new Exception('Upgrading failed: There is no way to upgrade from ' . $ver . ' to ' . Database::REQUIRED_VERSION, -1);
|
||||
}
|
||||
$um = self::$upways['ver_' . $ver];
|
||||
$ver = self::$um();
|
||||
}
|
||||
}
|
||||
|
||||
private static $upways = [
|
||||
'ver_null' => 'upNull',
|
||||
'ver_0.1' => 'up01',
|
||||
'ver_1.0' => 'up10'
|
||||
];
|
||||
|
||||
private static function upNull()
|
||||
{
|
||||
$db = Database::connection();
|
||||
$db->beginTransaction();
|
||||
try {
|
||||
$db->query(
|
||||
'INSERT INTO `' . Database::tablePrefix('system') . '` (`key`, `value`) VALUES ("version", "0.1")'
|
||||
);
|
||||
$db->commit();
|
||||
} catch (Exception $e) {
|
||||
$db->rollBack();
|
||||
throw $e;
|
||||
}
|
||||
return '0.1';
|
||||
}
|
||||
|
||||
private static function up01()
|
||||
{
|
||||
$db = Database::connection();
|
||||
$db->beginTransaction();
|
||||
try {
|
||||
$dom_tn = Database::tablePrefix('domains');
|
||||
if (!self::columnExists($db, $dom_tn, 'active')) {
|
||||
$db->query('ALTER TABLE `' . $dom_tn . '` ADD COLUMN `active` boolean NOT NULL AFTER `fqdn`');
|
||||
}
|
||||
if (!self::columnExists($db, $dom_tn, 'created_time')) {
|
||||
$db->query('ALTER TABLE `' . $dom_tn . '` ADD COLUMN `created_time` datetime NOT NULL');
|
||||
}
|
||||
if (!self::columnExists($db, $dom_tn, 'updated_time')) {
|
||||
$db->query('ALTER TABLE `' . $dom_tn . '` ADD COLUMN `updated_time` datetime NOT NULL');
|
||||
}
|
||||
$db->query('UPDATE `' . $dom_tn . '` SET `active` = TRUE, `created_time` = NOW(), `updated_time` = NOW()');
|
||||
$db->query('UPDATE `' . Database::tablePrefix('system') . '` SET `value` = "1.0" WHERE `key` = "version"');
|
||||
$db->commit();
|
||||
} catch (Exception $e) {
|
||||
$db->rollBack();
|
||||
throw $e;
|
||||
}
|
||||
return '1.0';
|
||||
}
|
||||
|
||||
private static function up10()
|
||||
{
|
||||
$db = Database::connection();
|
||||
$db->beginTransaction();
|
||||
try {
|
||||
$sys_tn = Database::tablePrefix('system');
|
||||
$db->query('ALTER TABLE `' . $sys_tn . '` MODIFY COLUMN `key` varchar(64) NOT NULL');
|
||||
$db->query('UPDATE `' . $sys_tn . '` SET `value` = "2.0" WHERE `key` = "version"');
|
||||
$db->commit();
|
||||
} catch (Exception $d) {
|
||||
$db->rollBack();
|
||||
throw $e;
|
||||
}
|
||||
return '2.0';
|
||||
}
|
||||
|
||||
private static function columnExists($db, $table, $column)
|
||||
{
|
||||
$st = $db->prepare('SELECT NULL FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `table_schema` = ? AND `table_name` = ? AND `column_name` = ?');
|
||||
$st->bindValue(1, Database::name(), PDO::PARAM_STR);
|
||||
$st->bindValue(2, $table, PDO::PARAM_STR);
|
||||
$st->bindValue(3, $column, PDO::PARAM_STR);
|
||||
$st->execute();
|
||||
$res = $st->fetch(PDO::FETCH_NUM);
|
||||
$st->closeCursor();
|
||||
return $res ? true : false;
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user