xNightR00T File Manager

<?php

require_once dirname(__FILE__) . "/zs_component_upgrader.php";

class MonitorDBHandler extends ZSComponentUpgrader {


	protected function getSqliteDbName($version = "6.0.0") {
		if ($version == "5.6.0") {
			return "monitor";
		} else {
			return "monitor.db";
		}
	}

	protected function getMysqlSqlFilenames() {
		return array("mysql_create_monitor_db.sql");
	}

	protected function getSqliteSqlFilenames() {
		return array("create_monitor_db.sql");
	}

	public function getComponent() { return "monitor"; }

	public function getSchemaVersion() { return "2018.0.0"; }

	public function getSqliteSchemaVersion() {
		$dbh = $this->getSqliteDbh();
		$res = $dbh->query("SELECT version as value FROM version WHERE key = 'SCHEMA_VERSION'")->fetchAll();
		foreach ($res as $row) {
			return $this->getParsedVersionString($row['value']);
		}

		return false;
	}

	public function getMysqlSchemaVersion() {
		$dbh = $this->getMySqlDbh();
		$res = $dbh->query("SELECT property_value as value FROM schema_properties WHERE property = 'MONITOR_SCHEMA_VERSION'")->fetchAll();

		if (!$res) {
			return false;
		}

		foreach ($res as $row) {
			return $row['value'];
		}

		return false;
	}

	/**
	 *
	 * @param PDO $dbh
	 * @param unknown $version
	 * @return unknown|boolean
	 */
	public function setSqliteSchemaVersion($dbh, $version) {
		$stmt = $dbh->prepare("INSERT OR REPLACE INTO version VALUES('SCHEMA_VERSION', :version)");
		$stmt->bindValue(":version", $version);
		return $stmt->execute();
	}

	/**
	 *
	 * @param PDO $dbh
	 * @param unknown $version
	 * @return unknown|boolean
	 */
	public function setMysqlSchemaVersion($dbh, $version) {
		$stmt = $dbh->prepare("REPLACE INTO schema_properties VALUES('MONITOR_SCHEMA_VERSION', :version)");
		$stmt->bindValue(":version", $version);
		return $stmt->execute();
	}

	protected function upgradeSqlite() {

		$upgradeFromVersion = $this->getUpgradeFromVersion();
		if (version_compare($upgradeFromVersion, "6.0.0", ">=")) {
			message("Keeping monitor db from $upgradeFromVersion");
		        $this->copySqliteFromPreviousVersions();
		}

		if (file_exists($this->getSqliteDbPath())) {
			message("Database at " . $this->getSqliteDbPath() . " already exists");
		} else {

		        if ($this->copySqliteFromPreviousVersions("6.0.0")){
		        	message("Using monitor db from version 6.0.0");
		        } else {
		        	$this->createCleanSqliteDatabase();
		        }
                }

		$dbh = $this->getSqliteDbh();
		$schemaVersion = $this->getSqliteSchemaVersion();
		if (!$schemaVersion || version_compare($schemaVersion, "8.5.0", "<")) {
			message ("Upgrading to 8.5.0 (schema version is $schemaVersion)");
			$dbh->exec("DELETE FROM request_components;
					ALTER TABLE request_components ADD unique_hash TINYBLOB;
					CREATE UNIQUE INDEX IDX_MONITOR_ROUTE_HASH on request_components (unique_hash);
			");

			$this->setSqliteSchemaVersion($dbh, "8.5.0");
		}

		if (!$schemaVersion || version_compare($schemaVersion, "9.1.0", "<")) {
		    message ("Upgrading to 9.1.0 (schema version is $schemaVersion)");
		    $dbh->exec("  CREATE TABLE IF NOT EXISTS events_jobs (event_id INTEGER NOT NULL PRIMARY KEY, job_id INTEGER);
		                  CREATE INDEX EVENTS_JOBS_IDX on EVENTS_JOBS (job_id);");

		    $this->setSqliteSchemaVersion($dbh, "9.1.0");
		}

		if (!$schemaVersion || version_compare($schemaVersion, "2018.0.0", "<")) {
		    message ("Upgrading to 2018.0.0 (schema version is $schemaVersion)");
		    $this->setSqliteSchemaVersion($dbh, "2018.0.0");
		}
	}

	protected function upgradeMysql() {

		message("Upgrading Mysql schema");

		$this->createCleanMysqlDatabase();

		$dbh6 = $this->getMySqlDbh("6.0.0");
		if ($dbh6) {
			message("Upgrade from 6.0.0 - doing nothing");
			return;
		}

		$schemaVersion = $this->getMysqlSchemaVersion();
		if (!$schemaVersion || version_compare($schemaVersion, "8.5.0", "<")) {
			message ("Upgrading to 8.5.0 (schema version is $schemaVersion)");

			$dbh = $this->getMySqlDbh();
			$dbh->exec("DELETE FROM request_components;
			ALTER TABLE request_components ADD unique_hash TINYBLOB;
			CREATE UNIQUE INDEX IDX_MONITOR_ROUTE_HASH on request_components (unique_hash(40));
		    DELETE FROM schema_properties WHERE property = 'SCHEMA_VERSION';
			");
			$this->setMysqlSchemaVersion($dbh, "8.5.0");
		}

		if (!$schemaVersion || version_compare($schemaVersion, "9.1.0", "<")) {
		    message ("Upgrading to 9.1.0 (schema version is $schemaVersion). Create events_jobs and EVENTS_JOBS_IDX");

		    $dbh = $this->getMySqlDbh();
		    $dbh->exec("  CREATE TABLE IF NOT EXISTS events_jobs (event_id INTEGER NOT NULL PRIMARY KEY, job_id INTEGER);
		        CREATE INDEX EVENTS_JOBS_IDX on events_jobs (job_id);");
		    $this->setMysqlSchemaVersion($dbh, "9.1.0");
		}

		if (!$schemaVersion || version_compare($schemaVersion, "2018.0.0", "<")) {
		    message ("Upgrading to 2018.0.0 (schema version is $schemaVersion).");

		    $dbh = $this->getMySqlDbh();
		    $this->setMysqlSchemaVersion($dbh, "2018.0.0");
		}
	}

	public function upgrade() {
		$this->upgradeSqlite();

		if ($this->isClusterUpgrade()) {
			$this->upgradeMysql();
		}

        // Remove the obsolete monitor directive 'zend_monitor.event_generate_trace_file'
        // from the INI file
        $iniFile = $this->getConfdDir() . DIRECTORY_SEPARATOR . "monitor.ini";
        $this->removeDirectiveFromINI("zend_monitor.event_generate_trace_file", $iniFile);
        message("Removing directive: zend_monitor.event_generate_trace_file from file: $iniFile");
    }
}

?>