<?php
IncludeModuleLangFile(__FILE__);

class CClusterWebNode
{
	public static $errno = 0;
	public static $errstr = '';

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

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

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

		return $ID;
	}

	public static function Delete($ID)
	{
		global $DB;
		$ID = intval($ID);

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

		return $res;
	}

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

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

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

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

		return true;
	}

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

		$bHost = false;
		if (isset($arFields['HOST']))
		{
			if (preg_match('/^([0-9a-zA-Z-_.]+)$/', $arFields['HOST']))
			{
				$bHost = true;
			}

			if (!$bHost)
			{
				$aMsg[] = ['id' => 'HOST', 'text' => GetMessage('CLU_WEBNODE_WRONG_IP')];
			}
		}

		$bStatus = true;
		if ($bHost && isset($arFields['PORT']))
		{
			if ($arFields['STATUS_URL'] <> '')
			{
				$arStatus = static::GetStatus($arFields['HOST'], $arFields['PORT'], $arFields['STATUS_URL']);
				$bStatus = is_array($arStatus);
			}
		}

		if (!$bStatus)
		{
			//$aMsg[] = array("id" => "STATUS_URL", "text" => GetMessage("CLU_WEBNODE_WRONG_STATUS_URL"));
		}

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

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

		if (!is_array($arSelect))
		{
			$arSelect = [];
		}
		if (count($arSelect) < 1)
		{
			$arSelect = [
				'ID',
				'NAME',
				'DESCRIPTION',
				'HOST',
				'PORT',
				'STATUS_URL',
			];
		}

		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 'NAME':
				case 'DESCRIPTION':
				case 'HOST':
				case 'PORT':
				case 'STATUS_URL':
					$arQuerySelect[$strColumn] = 'w.' . $strColumn;
					break;
			}
		}
		if (count($arQuerySelect) < 1)
		{
			$arQuerySelect = ['ID' => 'w.ID'];
		}

		$obQueryWhere = new CSQLWhere;
		$arFields = [
			'ID' => [
				'TABLE_ALIAS' => 'w',
				'FIELD_NAME' => 'w.ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
			'GROUP_ID' => [
				'TABLE_ALIAS' => 'w',
				'FIELD_NAME' => 'w.GROUP_ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
			'NAME' => [
				'TABLE_ALIAS' => 'w',
				'FIELD_NAME' => 'w.NAME',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'HOST' => [
				'TABLE_ALIAS' => 'w',
				'FIELD_NAME' => 'w.HOST',
				'FIELD_TYPE' => 'string',
				'JOIN' => false,
			],
			'PORT' => [
				'TABLE_ALIAS' => 'w',
				'FIELD_NAME' => 'w.PORT',
				'FIELD_TYPE' => 'int',
				'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_webnode w
			' . $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 GetStatus($host, $port, $url)
	{
		self::$errno = 0;
		self::$errstr = '';
		$protocol = ($port === '443') ? 'ssl://' : '';
		$FP = @fsockopen($protocol . $host, $port, self::$errno, self::$errstr, 2);
		if ($FP)
		{
			$strRequest = 'GET ' . $url . " HTTP/1.0\r\n";
			$strRequest .= "User-Agent: BitrixSMCluster\r\n";
			$strRequest .= "Accept: */*\r\n";
			$strRequest .= 'Host: ' . $host . "\r\n";
			$strRequest .= "Accept-Language: en\r\n";
			$strRequest .= "\r\n";
			fputs($FP, $strRequest);

			$headers = '';
			while (!feof($FP))
			{
				$line = fgets($FP, 4096);
				if ($line == "\r\n")
				{
					break;
				}
				$headers .= $line;
			}

			$text = '';
			while (!feof($FP))
			{
				$text .= fread($FP, 4096);
			}

			fclose($FP);

			$match = [];
			if (preg_match_all('#<dt>(.*?)\\s*:\\s*(.*?)</dt>#', $text, $match))
			{
				$arResult = [];
				foreach ($match[0] as $i => $_)
				{
					$key = $match[1][$i];
					$value = $match[2][$i];
					if ($key === 'Total accesses')
					{
						$accessMatch = [];
						if (preg_match('/^(.*) - (.*)\\s*:\\s*(.*)$/', $value, $accessMatch))
						{
							$value = $accessMatch[1];
							$arResult[$accessMatch[2]] = $accessMatch[3];
						}
					}
					$arResult[$key] = $value;
				}
				return $arResult;
			}
		}

		return false;
	}

	public static function getServerList()
	{
		global $DB;
		$result = [];
		$rsData = $DB->Query('
			SELECT ID, GROUP_ID, HOST
			FROM b_cluster_webnode
			ORDER BY GROUP_ID, ID
		');
		while ($arData = $rsData->Fetch())
		{
			$host = $arData['HOST'] === '127.0.0.1' || $arData['HOST'] === 'localhost' ? '' : $arData['HOST'];
			$result[] = [
				'ID' => $arData['ID'],
				'GROUP_ID' => $arData['GROUP_ID'],
				'SERVER_TYPE' => 'web',
				'ROLE_ID' => '',
				'HOST' => $host,
				'DEDICATED' => 'Y',
				'EDIT_URL' => '/bitrix/admin/cluster_webnode_edit.php?lang=' . LANGUAGE_ID . '&group_id=' . $arData['GROUP_ID'] . '&ID=' . $arData['ID'],
			];
		}
		return $result;
	}

	public static function ParseDateTime($str)
	{
		static $search = false;
		static $replace = false;
		if ($search === false)
		{
			$search = [];
			$replace = [];
			for ($i = 1; $i <= 12; $i++)
			{
				$time = mktime(0, 0, 0, $i, 1, 2010);
				$search[] = date('M', $time);
				$replace[] = date('m', $time);
			}
		}

		$str = str_replace($search, $replace, $str);
		$dateMatch = [];
		if (preg_match('/(\\d{2}-\\d{2}-\\d{4} \\d{2}:\\d{2}:\\d{2})/', $str, $dateMatch))
		{
			return MakeTimeStamp($dateMatch[1], 'DD-MM-YYYY HH:MI:SS');
		}
		else
		{
			return false;
		}
	}
}