<?php
/**
 * PHP Server Monitor
 * Monitor your servers and websites.
 *
 * This file is part of PHP Server Monitor.
 * PHP Server Monitor 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, or
 * (at your option) any later version.
 *
 * PHP Server Monitor 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 PHP Server Monitor.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @package     phpservermon
 * @author      Pepijn Over <pep@neanderthal-technology.com>
 * @copyright   Copyright (c) 2008-2014 Pepijn Over <pep@neanderthal-technology.com>
 * @license     http://www.gnu.org/licenses/gpl.txt GNU GPL v3
 * @version     Release: @package_version@
 * @link        http://www.phpservermonitor.org/
 * @since		phpservermon 2.1.0
 **/

namespace psm\Util\Install;

/**
 * Installer class.
 *
 * Executes the queries to install/upgrade phpservermon.
 */
class Installer {

	/**
	 * Database service
	 * @var \psm\Service\Database $db
	 */
	protected $db;

	/**
	 * Log callback
	 * @var callable $logger
	 */
	protected $logger;

	/**
	 * Log of executed queries
	 * @var array $queries
	 */
	protected $queries = array();

	/**
	 * Open a new installer instance
	 * @param \psm\Service\Database $db
	 * @param callable $logger
	 */
	function __construct(\psm\Service\Database $db, $logger = null) {
		$this->db = $db;
		$this->logger = $logger;
	}

	/**
	 * Check if an upgrade is required for the current version.
	 * @return boolean
	 * @see upgrade()
	 */
	public function isUpgradeRequired() {
		$version_db = psm_get_conf('version');

		if(version_compare(PSM_VERSION, $version_db, '==')) {
			// version is up to date
			return false;
		}

		// different DB version, check if the version requires any changes
		// @todo this is currently a manual check for each version, similar to upgrade().. not a clean way
		if(version_compare($version_db, '3.1.0', '<')) {
			return true;
		} else {
			// change database version to current version so this check won't be required next time
			psm_update_conf('version', PSM_VERSION);
		}
		return false;
	}

	/**
	 * Log a message to the logger callable (if any)
	 * @param string|array $msg
	 * @return \psm\Util\Install\Installer
	 */
	protected function log($msg) {
		if(is_callable($this->logger)) {
			$msg = (!is_array($msg)) ? array($msg) : $msg;

			foreach($msg as $m) {
				call_user_func($this->logger, $m);
			}
		}
		return $this;
	}

	/**
	 * Execute one or more queries. Does no fetching or anything, so execute only.
	 * @param string|array $query
	 * @return \psm\Util\Install\Installer
	 */
	protected function execSQL($query) {
		$query = (!is_array($query)) ? array($query) : $query;

		foreach($query as $q) {
			$this->queries[] = $q;
			$this->db->exec($q);
		}
		return $this;
	}

	/**
	 * Retrieve table queries for install
	 */
	public function install() {
		$this->installTables();

		$this->log('Populating database...');
		$queries = array();
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "servers` (`ip`, `port`, `label`, `type`, `pattern`, `status`, `error`, `rtime`, `last_online`, `last_check`, `active`, `email`, `sms`, `pushover`) VALUES ('http://sourceforge.net/index.php', 80, 'SourceForge', 'website', '', 'on', '', '', '0000-00-00 00:00:00', '0000-00-00 00:00:00', 'yes', 'yes', 'yes', 'yes'), ('smtp.gmail.com', 465, 'Gmail SMTP', 'service', '', 'on', '', '', '0000-00-00 00:00:00', '0000-00-00 00:00:00', 'yes', 'yes', 'yes', 'yes')";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "users_servers` (`user_id`,`server_id`) VALUES (1, 1), (1, 2);";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUE
					('language', 'en_US'),
					('email_status', '1'),
					('email_from_email', 'monitor@example.org'),
					('email_from_name', 'Server Monitor'),
					('email_smtp', ''),
					('email_smtp_host', ''),
					('email_smtp_port', ''),
					('email_smtp_security', ''),
					('email_smtp_username', ''),
					('email_smtp_password', ''),
					('sms_status', '0'),
					('sms_gateway', 'mollie'),
					('sms_gateway_username', 'username'),
					('sms_gateway_password', 'password'),
					('sms_from', '1234567890'),
					('pushover_status', '0'),
					('pushover_api_token', ''),
					('alert_type', 'status'),
					('log_status', '1'),
					('log_email', '1'),
					('log_sms', '1'),
					('log_pushover', '1'),
					('log_retention_period', '365'),
					('version', '" . PSM_VERSION . "'),
					('version_update_check', '" . PSM_VERSION . "'),
					('auto_refresh_servers', '0'),
					('show_update', '1'),
					('last_update_check', '0'),
					('cron_running', '0'),
					('cron_running_time', '0');";
		$this->execSQL($queries);
	}

	/**
	 * Install the tables for the monitor
	 */
	protected function installTables() {
		$tables = array(
			PSM_DB_PREFIX . 'config' => "CREATE TABLE `" . PSM_DB_PREFIX . "config` (
							`key` varchar(255) NOT NULL,
							`value` varchar(255) NOT NULL,
							PRIMARY KEY (`key`)
						) ENGINE=MyISAM DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'users' => "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "users` (
							`user_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
							`user_name` varchar(64) NOT NULL COMMENT 'user''s name, unique',
							`password` varchar(255) NOT NULL COMMENT 'user''s password in salted and hashed format',
							`password_reset_hash` char(40) DEFAULT NULL COMMENT 'user''s password reset code',
							`password_reset_timestamp` bigint(20) DEFAULT NULL COMMENT 'timestamp of the password reset request',
							`rememberme_token` varchar(64) DEFAULT NULL COMMENT 'user''s remember-me cookie token',
							`level` tinyint(2) unsigned NOT NULL DEFAULT '20',
							`name` varchar(255) NOT NULL,
							`mobile` varchar(15) NOT NULL,
							`pushover_key` varchar(255) NOT NULL,
							`pushover_device` varchar(255) NOT NULL,
							`email` varchar(255) NOT NULL,
							PRIMARY KEY (`user_id`),
							UNIQUE KEY `unique_username` (`user_name`)
						  ) ENGINE=MyISAM  DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'users_preferences' => "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "users_preferences` (
							`user_id` int(11) unsigned NOT NULL,
							`key` varchar(255) NOT NULL,
							`value` varchar(255) NOT NULL,
							PRIMARY KEY (`user_id`, `key`)
						  ) ENGINE=MyISAM DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'users_servers' => "CREATE TABLE `" . PSM_DB_PREFIX . "users_servers` (
							`user_id` INT( 11 ) UNSIGNED NOT NULL ,
							`server_id` INT( 11 ) UNSIGNED NOT NULL ,
							PRIMARY KEY ( `user_id` , `server_id` )
							) ENGINE = MYISAM ;",
			PSM_DB_PREFIX . 'log' => "CREATE TABLE `" . PSM_DB_PREFIX . "log` (
						  `log_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						  `server_id` int(11) unsigned NOT NULL,
						  `type` enum('status','email','sms','pushover') NOT NULL,
						  `message` varchar(255) NOT NULL,
						  `datetime` timestamp NOT NULL default CURRENT_TIMESTAMP,
						  `user_id` varchar(255) NOT NULL,
						  PRIMARY KEY  (`log_id`)
						) ENGINE=MyISAM  DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'servers' => "CREATE TABLE `" . PSM_DB_PREFIX . "servers` (
						  `server_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						  `ip` varchar(500) NOT NULL,
						  `port` int(5) unsigned NOT NULL,
						  `label` varchar(255) NOT NULL,
						  `type` enum('service','website') NOT NULL default 'service',
						  `pattern` varchar(255) NOT NULL,
						  `status` enum('on','off') NOT NULL default 'on',
						  `error` varchar(255) NULL,
						  `rtime` FLOAT(9, 7) NULL,
						  `last_online` datetime NULL,
						  `last_check` datetime NULL,
						  `active` enum('yes','no') NOT NULL default 'yes',
						  `email` enum('yes','no') NOT NULL default 'yes',
						  `sms` enum('yes','no') NOT NULL default 'no',
						  `pushover` enum('yes','no') NOT NULL default 'yes',
                          `warning_threshold` mediumint(1) unsigned NOT NULL DEFAULT '1',
                          `warning_threshold_counter` mediumint(1) unsigned NOT NULL DEFAULT '0',
                          `timeout` smallint(1) unsigned NULL DEFAULT NULL,
						  PRIMARY KEY  (`server_id`)
						) ENGINE=MyISAM  DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'servers_uptime' => "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "servers_uptime` (
						`servers_uptime_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						`server_id` int(11) unsigned NOT NULL,
						`date` datetime NOT NULL,
						`status` tinyint(1) unsigned NOT NULL,
						`latency` float(9,7) DEFAULT NULL,
						PRIMARY KEY (`servers_uptime_id`),
						KEY `server_id` (`server_id`)
					  ) ENGINE=MyISAM  DEFAULT CHARSET=utf8;",
			PSM_DB_PREFIX . 'servers_history' => "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "servers_history` (
						  `servers_history_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						  `server_id` int(11) unsigned NOT NULL,
						  `date` date NOT NULL,
						  `latency_min` float(9,7) NOT NULL,
						  `latency_avg` float(9,7) NOT NULL,
						  `latency_max` float(9,7) NOT NULL,
						  `checks_total` int(11) unsigned NOT NULL,
						  `checks_failed` int(11) unsigned NOT NULL,
						  PRIMARY KEY (`servers_history_id`),
						  UNIQUE KEY `server_id_date` (`server_id`,`date`)
						) ENGINE=MyISAM  DEFAULT CHARSET=utf8;",
		);

		foreach($tables as $name => $sql) {
			$if_table_exists = $this->db->query("SHOW TABLES LIKE '{$name}'");

			if(!empty($if_table_exists)) {
				$this->log('Table ' . $name . ' already exists in your database!');
			} else {
				$this->execSQL($sql);
				$this->log('Table ' . $name . ' added.');
			}
		}
	}

	/**
	 * Populate the tables and perform upgrades if necessary
	 * @param string $version_from
	 * @param string $version_to
	 * @see isUpgradeRequired()
	 */
	public function upgrade($version_from, $version_to) {
		if(version_compare($version_from, '2.1.0', '<')) {
			$this->upgrade210();
		}
		if(version_compare($version_from, '3.0.0', '<')) {
			$this->upgrade300();
		}
		if(version_compare($version_from, '3.1.0', '<')) {
			$this->upgrade310();
		}
		if(version_compare($version_from, '3.2.0', '<')) {
			$this->upgrade320();
		}
		psm_update_conf('version', $version_to);
	}

	/**
	 * Upgrade for v2.1.0 release
	 */
	protected function upgrade210() {
		$queries = array();
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "config` DROP `config_id`;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "config` ADD PRIMARY KEY ( `key` );";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "config` DROP INDEX `key`;";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('cron_running', '0');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('cron_running_time', '0');";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `error` `error` VARCHAR( 255 ) CHARACTER SET utf8 COLLATE utf8_general_ci NULL;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `rtime` `rtime` FLOAT( 9, 7 ) NULL;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `last_online` `last_online` DATETIME NULL;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `last_check` `last_check` DATETIME NULL;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` ADD  `pattern` VARCHAR( 255 ) NOT NULL AFTER  `type`;";

		$this->execSQL($queries);
	}

	/**
	 * Upgrade for v3.0.0 release
	 */
	protected function upgrade300() {
		$queries = array();
		// language is now stored as language code (ISO 639-1) + country code (ISO 3166-1)
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='bg_BG' WHERE `key`='language' AND `value`='bg';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='de_DE' WHERE `key`='language' AND `value`='de';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='en_US' WHERE `key`='language' AND `value`='en';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='fr_FR' WHERE `key`='language' AND `value`='fr';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='ko_KR' WHERE `key`='language' AND `value`='kr';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='nl_NL' WHERE `key`='language' AND `value`='nl';";
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "config` SET `value`='pt_BR' WHERE `key`='language' AND `value`='br';";

		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('version_update_check', '" . PSM_VERSION . "');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('email_smtp', '');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('email_smtp_host', '');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('email_smtp_port', '');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('email_smtp_username', '');";
		$queries[] = "INSERT INTO `" . PSM_DB_PREFIX . "config` (`key`, `value`) VALUES ('email_smtp_password', '');";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "log` CHANGE `log_id` `log_id` INT( 11 ) UNSIGNED NOT NULL AUTO_INCREMENT;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "log` CHANGE `server_id` `server_id` INT( 11 ) UNSIGNED NOT NULL;";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `server_id` `server_id` INT( 11 ) UNSIGNED NOT NULL AUTO_INCREMENT;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` ADD `warning_threshold` MEDIUMINT( 1 ) UNSIGNED NOT NULL DEFAULT '1';";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` ADD `warning_threshold_counter` MEDIUMINT( 1 ) UNSIGNED NOT NULL DEFAULT '0';";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "users` CHANGE `user_id` `user_id` INT( 11 ) UNSIGNED NOT NULL AUTO_INCREMENT;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "users`
			ADD `user_name` varchar(64) COLLATE utf8_general_ci NOT NULL COMMENT 'user\'s name, unique' AFTER `user_id`,
			ADD `password` varchar(255) COLLATE utf8_general_ci NOT NULL COMMENT 'user\'s password in salted and hashed format' AFTER `user_name`,
			ADD `password_reset_hash` char(40) COLLATE utf8_general_ci DEFAULT NULL COMMENT 'user\'s password reset code' AFTER `password`,
			ADD `password_reset_timestamp` bigint(20) DEFAULT NULL COMMENT 'timestamp of the password reset request' AFTER `password_reset_hash`,
			ADD `rememberme_token` varchar(64) COLLATE utf8_general_ci DEFAULT NULL COMMENT 'user\'s remember-me cookie token' AFTER `password_reset_timestamp`,
			ADD `level` TINYINT( 2 ) UNSIGNED NOT NULL DEFAULT '20' AFTER `rememberme_token`;";
		// make sure all current users are admins (previously we didnt have non-admins):
		$queries[] = "UPDATE `" . PSM_DB_PREFIX . "users` SET `user_name`=`email`, `level`=10;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "users` ADD UNIQUE `unique_username` ( `user_name` );";

		$queries[] = "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "servers_uptime` (
						`servers_uptime_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						`server_id` int(11) unsigned NOT NULL,
						`date` datetime NOT NULL,
						`status` tinyint(1) unsigned NOT NULL,
						`latency` float(9,7) DEFAULT NULL,
						PRIMARY KEY (`servers_uptime_id`),
						KEY `server_id` (`server_id`)
					  ) ENGINE=MyISAM  DEFAULT CHARSET=utf8;";

		$queries[] = "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "servers_history` (
						  `servers_history_id` int(11) unsigned NOT NULL AUTO_INCREMENT,
						  `server_id` int(11) unsigned NOT NULL,
						  `date` date NOT NULL,
						  `latency_min` float(9,7) NOT NULL,
						  `latency_avg` float(9,7) NOT NULL,
						  `latency_max` float(9,7) NOT NULL,
						  `checks_total` int(11) unsigned NOT NULL,
						  `checks_failed` int(11) unsigned NOT NULL,
						  PRIMARY KEY (`servers_history_id`),
						  UNIQUE KEY `server_id_date` (`server_id`,`date`)
						) ENGINE=MyISAM  DEFAULT CHARSET=utf8;";

		$queries[] = "CREATE TABLE `" . PSM_DB_PREFIX . "users_servers` (
						`user_id` INT( 11 ) UNSIGNED NOT NULL ,
						`server_id` INT( 11 ) UNSIGNED NOT NULL ,
						PRIMARY KEY ( `user_id` , `server_id` )
						) ENGINE = MYISAM ;";
		$this->execSQL($queries);

		// from 3.0 all user-server relations are in a separate table
		$users = $this->db->select(PSM_DB_PREFIX . 'users', null, array('user_id', 'server_id'));
		foreach($users as $user) {
			$idc = array();
			if($user['server_id'] == '') {
				continue;
			}
			if(strpos($user['server_id'], ',') === false) {
				$idc[] = $user['server_id'];
			} else {
				$idc = explode(',', $user['server_id']);
			}
			foreach($idc as $id) {
				$this->db->save(PSM_DB_PREFIX . 'users_servers', array(
					'user_id' => $user['user_id'],
					'server_id' => $id,
				));
			}
		}
		$this->execSQL("ALTER TABLE `".PSM_DB_PREFIX."users` DROP `server_id`;");
	}

	protected function upgrade310() {
		$queries = array();
		psm_update_conf('log_retention_period', '365');

		psm_update_conf('pushover_status', 0);
		psm_update_conf('log_pushover', 1);
		psm_update_conf('pushover_api_token', '');
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "users` ADD  `pushover_key` VARCHAR( 255 ) NOT NULL AFTER `mobile`;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "users` ADD  `pushover_device` VARCHAR( 255 ) NOT NULL AFTER `pushover_key`;";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` ADD  `pushover` ENUM( 'yes','no' ) NOT NULL DEFAULT 'yes' AFTER  `sms`;";
		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "log` CHANGE `type` `type` ENUM( 'status', 'email', 'sms', 'pushover' ) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL;";

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` ADD `timeout` smallint(1) unsigned NULL DEFAULT NULL;";

		$queries[] = "CREATE TABLE IF NOT EXISTS `" . PSM_DB_PREFIX . "users_preferences` (
						`user_id` int(11) unsigned NOT NULL,
						`key` varchar(255) NOT NULL,
						`value` varchar(255) NOT NULL,
						PRIMARY KEY (`user_id`, `key`)
					  ) ENGINE=MyISAM DEFAULT CHARSET=utf8;";

		$this->execSQL($queries);
	}

	protected function upgrade320() {
		$queries = array();

		$queries[] = "ALTER TABLE `" . PSM_DB_PREFIX . "servers` CHANGE `ip` `ip` VARCHAR(500) NOT NULL;";

		$this->execSQL($queries);
	}
}