Edit File by line

<?php namespace la\core\db;

[0] Fix | Delete

if ( ! defined( 'WPINC' ) ) die;

[1] Fix | Delete

[2] Fix | Delete

use Exception;

[3] Fix | Delete

use la\core\db\migrations\ILADBMigration;

[4] Fix | Delete

use la\core\LAUtils;

[5] Fix | Delete

use ReflectionClass;

[6] Fix | Delete

use ReflectionException;

[7] Fix | Delete

[8] Fix | Delete

/**

[9] Fix | Delete

* Flow-Flow.

[10] Fix | Delete

[11] Fix | Delete

* @package FlowFlow

[12] Fix | Delete

* @author Looks Awesome <email@looks-awesome.com>

[13] Fix | Delete

[14] Fix | Delete

* @link http://looks-awesome.com

[15] Fix | Delete

* @copyright Looks Awesome

[16] Fix | Delete

[17] Fix | Delete

abstract class LADBMigrationManager{

[18] Fix | Delete

const INIT_MIGRATION = '0.9999';

[19] Fix | Delete

[20] Fix | Delete

protected $context;

[21] Fix | Delete

[22] Fix | Delete

public function __construct($context) {

[23] Fix | Delete

$this->context = $context;

[24] Fix | Delete

}

[25] Fix | Delete

[26] Fix | Delete

/**

[27] Fix | Delete

* @throws ReflectionException

[28] Fix | Delete

* @throws Exception

[29] Fix | Delete

[30] Fix | Delete

public final function migrate(){

[31] Fix | Delete

$version = $this->getDBVersion();

[32] Fix | Delete

if (!$this->hasMigrations4Perform($version)){

[33] Fix | Delete

return;

[34] Fix | Delete

}

[35] Fix | Delete

[36] Fix | Delete

$dbm = LAUtils::dbm($this->context);

[37] Fix | Delete

$conn = $this->connection();

[38] Fix | Delete

try{

[39] Fix | Delete

if ( $conn->autocommit(false) ){

[40] Fix | Delete

if ($this->needStartInitMigration($version)){

[41] Fix | Delete

foreach ($this->getInitMigration() as $max_version => $migration){

[42] Fix | Delete

$migration->execute($conn, $dbm);

[43] Fix | Delete

$dbm->setOption('db_version', $max_version);

[44] Fix | Delete

}

[45] Fix | Delete

}

[46] Fix | Delete

else {

[47] Fix | Delete

foreach ( $this->getMigrations() as $migration ) {

[48] Fix | Delete

if ($this->needExecuteMigration($version, $migration->version())){

[49] Fix | Delete

$migration->execute($conn, $dbm);

[50] Fix | Delete

$dbm->setOption('db_version', $migration->version());

[51] Fix | Delete

}

[52] Fix | Delete

}

[53] Fix | Delete

}

[54] Fix | Delete

$conn->commit();

[55] Fix | Delete

}

[56] Fix | Delete

} catch ( Exception $e ){

[57] Fix | Delete

error_log($e->getTraceAsString());

[58] Fix | Delete

$conn->rollback();

[59] Fix | Delete

$conn->close();

[60] Fix | Delete

throw $e;

[61] Fix | Delete

}

[62] Fix | Delete

}

[63] Fix | Delete

[64] Fix | Delete

/**

[65] Fix | Delete

* Return the list of migration

[66] Fix | Delete

* @return array

[67] Fix | Delete

[68] Fix | Delete

protected abstract function migrations();

[69] Fix | Delete

[70] Fix | Delete

/**

[71] Fix | Delete

* @return LASafeMySQL

[72] Fix | Delete

[73] Fix | Delete

protected function connection() {

[74] Fix | Delete

return LAUtils::dbm($this->context)->conn();

[75] Fix | Delete

}

[76] Fix | Delete

[77] Fix | Delete

/**

[78] Fix | Delete

* @return string

[79] Fix | Delete

* @throws Exception

[80] Fix | Delete

[81] Fix | Delete

private function getDBVersion(){

[82] Fix | Delete

global $wpdb;

[83] Fix | Delete

$version = self::INIT_MIGRATION;

[84] Fix | Delete

$table = LAUtils::dbm($this->context)->option_table_name;

[85] Fix | Delete

if (null != $wpdb->get_var($wpdb->prepare('SHOW TABLES LIKE %s', $table))){

[86] Fix | Delete

$option = LAUtils::slug_down($this->context) . '_db_version';

[87] Fix | Delete

$version = $wpdb->get_var($wpdb->prepare('select value from ' . $table . ' where id = %s', $option));

[88] Fix | Delete

if (null == $version){

[89] Fix | Delete

$e = new Exception('Can`t get the db version of plugin');

[90] Fix | Delete

error_log($e->getTraceAsString());

[91] Fix | Delete

return self::INIT_MIGRATION;

[92] Fix | Delete

}

[93] Fix | Delete

}

[94] Fix | Delete

return $version;

[95] Fix | Delete

}

[96] Fix | Delete

[97] Fix | Delete

private function needStartInitMigration($version){

[98] Fix | Delete

return self::INIT_MIGRATION == $version || $version === false;

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

/**

[102] Fix | Delete

* @return array

[103] Fix | Delete

* @throws ReflectionException

[104] Fix | Delete

[105] Fix | Delete

private function getInitMigration(){

[106] Fix | Delete

$migrations = $this->getMigrations();

[107] Fix | Delete

[108] Fix | Delete

$max = self::INIT_MIGRATION;

[109] Fix | Delete

foreach ($migrations as $version => $migration) {

[110] Fix | Delete

if ($max < $version){

[111] Fix | Delete

$max = $version;

[112] Fix | Delete

}

[113] Fix | Delete

}

[114] Fix | Delete

return [ $max => $migrations[self::INIT_MIGRATION] ];

[115] Fix | Delete

}

[116] Fix | Delete

[117] Fix | Delete

/**

[118] Fix | Delete

* @return array

[119] Fix | Delete

* @throws ReflectionException

[120] Fix | Delete

[121] Fix | Delete

private function getMigrations(){

[122] Fix | Delete

$migrations = [];

[123] Fix | Delete

foreach ($this->migrations() as $class) {

[124] Fix | Delete

$clazz = new ReflectionClass($class);

[125] Fix | Delete

/** @var ILADBMigration $migration */

[126] Fix | Delete

$migration = $clazz->newInstance();

[127] Fix | Delete

$migrations[$migration->version()] = $migration;

[128] Fix | Delete

}

[129] Fix | Delete

uksort($migrations, 'version_compare');

[130] Fix | Delete

[131] Fix | Delete

return $migrations;

[132] Fix | Delete

}

[133] Fix | Delete

[134] Fix | Delete

private function needExecuteMigration($db_version, $migration_version){

[135] Fix | Delete

$db = explode('.', $db_version);

[136] Fix | Delete

$migration = explode('.', $migration_version);

[137] Fix | Delete

if (intval($migration[0]) == intval($db[0])){

[138] Fix | Delete

return (intval($migration[1]) > $db[1]);

[139] Fix | Delete

}

[140] Fix | Delete

return (intval($migration[0]) > intval($db[0]));

[141] Fix | Delete

}

[142] Fix | Delete

[143] Fix | Delete

/**

[144] Fix | Delete

* @param $version

[145] Fix | Delete

[146] Fix | Delete

* @return bool

[147] Fix | Delete

* @throws ReflectionException

[148] Fix | Delete

[149] Fix | Delete

private function hasMigrations4Perform($version){

[150] Fix | Delete

if ($this->needStartInitMigration($version)){

[151] Fix | Delete

return true;

[152] Fix | Delete

}

[153] Fix | Delete

foreach ( $this->getMigrations() as $migration ) {

[154] Fix | Delete

if ( $this->needExecuteMigration( $version, $migration->version() ) ) {

[155] Fix | Delete

return true;

[156] Fix | Delete

}

[157] Fix | Delete

}

[158] Fix | Delete

return false;

[159] Fix | Delete

}

[160] Fix | Delete

}

[161] Fix | Delete