<?php
IncludeModuleLangFile(__FILE__);

abstract class CAllClusterDBNode
{
	public function CheckFields(&$arFields, $ID)
	{
		global $APPLICATION;
		$aMsg = [];

		if (array_key_exists('NAME', $arFields))
		{
			$arFields['NAME'] = trim($arFields['NAME']);
		}

		if (array_key_exists('ACTIVE', $arFields))
		{
			$arFields['ACTIVE'] = $arFields['ACTIVE'] === 'Y' ? 'Y' : 'N';
		}

		if (array_key_exists('SELECTABLE', $arFields))
		{
			$arFields['SELECTABLE'] = $arFields['SELECTABLE'] === 'N' ? 'N' : 'Y';
		}

		if (array_key_exists('WEIGHT', $arFields))
		{
			$weight = intval($arFields['WEIGHT']);
			if ($weight < 0)
			{
				$weight = 0;
			}
			elseif ($weight > 100)
			{
				$weight = 100;
			}
			$arFields['WEIGHT'] = $weight;
		}

		if (
			(isset($arFields['ACTIVE']) && $arFields['ACTIVE'] == 'Y')
			&& $arFields['ROLE_ID'] != 'SLAVE'
		)
		{
			$obCheck = new CClusterDBNodeCheck;
			$nodeDB = $obCheck->SlaveNodeConnection(
				$arFields['DB_HOST'],
				$arFields['DB_NAME'],
				$arFields['DB_LOGIN'],
				$arFields['DB_PASSWORD'],
				$arFields['ROLE_ID'] == 'MASTER' ? $arFields['MASTER_HOST'] : false,
				$arFields['ROLE_ID'] == 'MASTER' ? $arFields['MASTER_PORT'] : false
			);
			if (is_object($nodeDB))
			{
				$arFields['SERVER_ID'] = intval($obCheck->GetServerVariable($nodeDB, 'server_id'));
			}
			else
			{
				if (!array_key_exists('STATUS', $arFields))
				{
					$arFields['STATUS'] = 'OFFLINE';
				}
				$aMsg[] = ['id' => '', 'text' => $nodeDB];
			}
		}

		if (!empty($aMsg))
		{
			$e = new CAdminException($aMsg);
			$APPLICATION->ThrowException($e);
			return false;
		}
		return true;
	}

	abstract public static function GetUpTime($node_id);

	public static function GetByID($node_id, $arVirtNode=false)
	{
		global $DB, $CACHE_MANAGER;
		static $arNodes = false;
		static $arVirtNodes = [];

		//This code sets and gets virtual nodes
		//needed to test connection just before
		//save node credentials into db
		if (preg_match('/^v(\d+)$/', $node_id))
		{
			if (is_array($arVirtNode))
			{
				$arVirtNodes[$node_id] = $arVirtNode;
				return true;
			}
			else
			{
				return $arVirtNodes[$node_id];
			}
		}

		//Normal method continues here
		$node_id = intval($node_id);
		if ($arNodes === false)
		{
			$cache_id = 'db_nodes';
			if (
				CACHED_b_cluster_dbnode !== false
				&& $CACHE_MANAGER->Read(CACHED_b_cluster_dbnode, $cache_id, 'b_cluster_dbnode')
			)
			{
				$arNodes = $CACHE_MANAGER->Get($cache_id);
			}
			else
			{
				$arNodes = [];

				$rs = $DB->Query('SELECT * FROM b_cluster_dbnode ORDER BY ID', false, '', ['fixed_connection' => true]);
				while ($ar = $rs->Fetch())
				{
					$arNodes[intval($ar['ID'])] = $ar;
				}

				if (CACHED_b_cluster_dbnode !== false)
				{
					$CACHE_MANAGER->Set($cache_id, $arNodes);
				}
			}
		}

		return $arNodes[$node_id];
	}

	public function Add($arFields)
	{
		global $DB, $CACHE_MANAGER;

		if (!$this->CheckFields($arFields, 0))
		{
			return false;
		}

		if (!array_key_exists('ACTIVE', $arFields))
		{
			$arFields['ACTIVE'] = 'Y';
		}

		$ID = $DB->Add('b_cluster_dbnode', $arFields);

		if (CACHED_b_cluster_dbnode !== false)
		{
			$CACHE_MANAGER->CleanDir('b_cluster_dbnode');
		}

		return $ID;
	}

	public static function Delete($ID, $bStopSlave = true)
	{
		global $DB, $CACHE_MANAGER;
		$ID = intval($ID);

		$res = $DB->Query('select ID from b_cluster_dbnode WHERE ID=1 OR MASTER_ID = ' . $ID, false, '', ['fixed_connection' => true]);
		if ($bStopSlave)
		{
			while ($ar = $res->Fetch())
			{
				if (!CClusterSlave::Stop($ar['ID']))
				{
					return false;
				}
			}
		}

		if ($res)
		{
			$res = $DB->Query('DELETE FROM b_cluster_dbnode WHERE ID = ' . $ID, false, '', ['fixed_connection' => true]);
		}

		if (CACHED_b_cluster_dbnode !== false)
		{
			$CACHE_MANAGER->CleanDir('b_cluster_dbnode');
		}

		return $res;
	}

	public function Update($ID, $arFields)
	{
		global $DB, $CACHE_MANAGER;
		$ID = intval($ID);

		if ($ID <= 0)
		{
			return false;
		}

		if (!$this->CheckFields($arFields, $ID))
		{
			return false;
		}

		unset($arFields['GROUP_ID']);

		$strUpdate = $DB->PrepareUpdate('b_cluster_dbnode', $arFields);
		if ($strUpdate <> '')
		{
			$strSql = '
				UPDATE b_cluster_dbnode SET
				' . $strUpdate . '
				WHERE ID = ' . $ID . '
			';
			if (!$DB->Query($strSql, false, '', ['fixed_connection' => true]))
			{
				return false;
			}
		}

		if (CACHED_b_cluster_dbnode !== false)
		{
			$CACHE_MANAGER->CleanDir('b_cluster_dbnode');
		}

		return true;
	}

	public static function SetOffline($node_id)
	{
		global $DB, $CACHE_MANAGER, $APPLICATION;

		$rs = $DB->Query("
			UPDATE b_cluster_dbnode SET
			STATUS = 'OFFLINE'
			WHERE ID = " . intval($node_id) . "
			AND STATUS <> 'OFFLINE'
		", false, '', ['fixed_connection' => true]);
		if ($rs->AffectedRowsCount() > 0)
		{
			if (CACHED_b_cluster_dbnode !== false)
			{
				$CACHE_MANAGER->CleanDir('b_cluster_dbnode');
			}

			if (!CAgent::AddAgent('CClusterDBNode::BringOnline();', 'cluster', 'N', 10))
			{
				$APPLICATION->ResetException();
			}
		}
	}

	public static function SetOnline($node_id)
	{
		global $DB, $CACHE_MANAGER;

		$DB->Query("
			UPDATE b_cluster_dbnode SET
			STATUS = 'ONLINE'
			WHERE ID = " . intval($node_id) . '
		', false, '', ['fixed_connection' => true]);

		if (CACHED_b_cluster_dbnode !== false)
		{
			$CACHE_MANAGER->CleanDir('b_cluster_dbnode');
		}
	}

	public static function BringOnline()
	{
		$rsOfflineNodes = CClusterDBNode::GetList([], ['=STATUS' => 'OFFLINE'], ['ID']);
		if ($arNode = $rsOfflineNodes->Fetch())
		{
			ob_start();
			$nodeDB = CDatabase::GetDBNodeConnection($arNode['ID'], true, false);
			ob_end_clean();
			if (is_object($nodeDB))
			{
				CClusterDBNode::SetOnline($arNode['ID']);
			}

			return 'CClusterDBNode::BringOnline();';
		}
		return '';
	}

	public static function GetList($arOrder=false, $arFilter=false, $arSelect=false)
	{
		global $DB;

		if (!is_array($arSelect))
		{
			$arSelect = [];
		}
		if (count($arSelect) < 1)
		{
			$arSelect = [
				'ID',
				'ACTIVE',
				'ROLE_ID',
				'NAME',
				'DESCRIPTION',
				'DB_HOST',
				'DB_NAME',
				'DB_LOGIN',
				'DB_PASSWORD',
				'MASTER_ID',
				'SERVER_ID',
				'STATUS',
				'WEIGHT',
				'SELECTABLE',
				'GROUP_ID'
			];
		}

		if (!is_array($arOrder))
		{
			$arOrder = [];
		}

		$arQueryOrder = [];
		foreach ($arOrder as $strColumn => $strDirection)
		{
			$strColumn = mb_strtoupper($strColumn);
			$strDirection = mb_strtoupper($strDirection) === 'ASC' ? 'ASC' : 'DESC';
			switch ($strColumn)
			{
				case 'ID':
				case 'NAME':
					$arSelect[] = $strColumn;
					$arQueryOrder[$strColumn] = $strColumn . ' ' . $strDirection;
					break;
			}
		}

		$arQuerySelect = [];
		foreach ($arSelect as $strColumn)
		{
			$strColumn = mb_strtoupper($strColumn);
			switch ($strColumn)
			{
				case 'ID':
				case 'ACTIVE':
				case 'ROLE_ID':
				case 'NAME':
				case 'DESCRIPTION':
				case 'DB_HOST':
				case 'DB_NAME':
				case 'DB_LOGIN':
				case 'DB_PASSWORD':
				case 'MASTER_ID':
				case 'SERVER_ID':
				case 'STATUS':
				case 'WEIGHT':
				case 'SELECTABLE':
				case 'GROUP_ID':
					$arQuerySelect[$strColumn] = 'n.' . $strColumn;
					break;
			}
		}
		if (count($arQuerySelect) < 1)
		{
			$arQuerySelect = ['ID' => 'n.ID'];
		}

		$obQueryWhere = new CSQLWhere;
		$arFields = [
			'ID' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
			'GROUP_ID' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.GROUP_ID',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'ROLE_ID' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.ROLE_ID',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'ACTIVE' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.ACTIVE',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'SERVER_ID' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.SERVER_ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
			'MASTER_ID' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.MASTER_ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
			'STATUS' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.STATUS',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'NAME' => [
				'TABLE_ALIAS' => 'n',
				'FIELD_NAME' => 'n.NAME',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
		];
		$obQueryWhere->SetFields($arFields);

		if (!is_array($arFilter))
		{
			$arFilter = [];
		}
		$strQueryWhere = $obQueryWhere->GetQuery($arFilter);

		$bDistinct = $obQueryWhere->bDistinctReqired;

		$strSql = '
			SELECT ' . ($bDistinct ? 'DISTINCT' : '') . '
			' . implode(', ', $arQuerySelect) . '
			FROM
				b_cluster_dbnode n
			' . $obQueryWhere->GetJoins() . '
		';

		if ($strQueryWhere)
		{
			$strSql .= '
				WHERE
				' . $strQueryWhere . '
			';
		}

		if (count($arQueryOrder) > 0)
		{
			$strSql .= '
				ORDER BY
				' . implode(', ', $arQueryOrder) . '
			';
		}

		return $DB->Query($strSql, false, '', ['fixed_connection' => true]);
	}

	public static function getServerList()
	{
		global $DB;
		$result = [];
		$rsData = $DB->Query('
			SELECT ID, GROUP_ID, ROLE_ID, DB_HOST
			FROM b_cluster_dbnode
			ORDER BY GROUP_ID IS NULL, GROUP_ID, ID
		');
		while ($arData = $rsData->Fetch())
		{
			if ($arData['ROLE_ID'] == 'SLAVE' || $arData['ROLE_ID'] == 'MASTER')
			{
				$edit_url = '/bitrix/admin/cluster_slave_edit.php?lang=' . LANGUAGE_ID . '&group_id=' . $arData['GROUP_ID'] . '&ID=' . $arData['ID'];
			}
			elseif ($arData['ROLE_ID'] == 'MODULE')
			{
				$edit_url = '/bitrix/admin/cluster_dbnode_edit.php?lang=' . LANGUAGE_ID . '&ID=' . $arData['ID'];
			}
			else
			{
				$edit_url = '';
			}

			$host = '';
			$match = [];
			if (
				preg_match('/\\(\\s*HOST\\s*=\\s*([^)]+)\\)/i', $arData['DB_HOST'], $match)
				|| preg_match('/^([^:]+)(:|)/', $arData['DB_HOST'], $match)
			)
			{
				if ($match[1] !== 'localhost' && $match[1] !== '127.0.0.1')
				{
					$host = $arData['DB_HOST'];
				}
			}

			$result[] = [
				'ID' => $arData['ID'],
				'GROUP_ID' => $arData['GROUP_ID'],
				'SERVER_TYPE' => 'database',
				'ROLE_ID' => $arData['ROLE_ID'],
				'HOST' => $arData['DB_HOST'],
				'DEDICATED' => $host === '' ? 'N' : 'Y',
				'EDIT_URL' => $edit_url,
			];
		}
		return $result;
	}

	public static function GetModules($node_id)
	{
		global $DB;
		static $arCache = false;
		if ($arCache === false)
		{
			$arCache = [];
			$rs = $DB->Query("SELECT * from b_option WHERE NAME='dbnode_id'", false, '', ['fixed_connection' => true]);
			while ($ar = $rs->Fetch())
			{
				if (is_numeric($ar['VALUE']))
				{
					$arCache[intval($ar['VALUE'])][$ar['MODULE_ID']] = $ar['MODULE_ID'];
				}
			}
		}
		if (isset($arCache[$node_id]))
		{
			return $arCache[$node_id];
		}
		else
		{
			return [];
		}
	}

	public static function GetModulesForSharding()
	{
		$connection = \Bitrix\Main\Application::getConnection();
		$arModules = [];

		if ($connection->getType() === 'mysql')
		{
			foreach (GetModuleEvents('cluster', 'OnGetTableList', true) as $arEvent)
			{
				$ar = ExecuteModuleEventEx($arEvent);
				if (is_array($ar))
				{
					$cur_node_id = intval(COption::GetOptionString($ar['MODULE']->MODULE_ID, 'dbnode_id'));
					if ($cur_node_id < 1)
					{
						$cur_node_id = 1;
					}
					$cur_node_id = CClusterDBNode::GetByID($cur_node_id);
					$ar['TITLE'] = $ar['MODULE']->MODULE_NAME . ' (' . $cur_node_id['NAME'] . ')';
					$arModules[$ar['MODULE']->MODULE_ID] = $ar;
				}
			}
		}

		return $arModules;
	}

	public static function GetListForModuleInstall()
	{
		return CClusterDBNode::GetList(
			['NAME' => 'ASC','ID' => 'ASC']
			,['=ACTIVE' => 'Y', '=ROLE_ID' => 'MODULE', '=STATUS' => 'READY']
			,['ID', 'NAME']
		);
	}
}