<?php

class CPerfomanceTableList extends CDBResult
{
	public static function GetList($bFull = true)
	{
		global $DB;
		if ($bFull)
		{
			$rsTables = $DB->Query('show table status');
		}
		else
		{
			$rsTables = $DB->Query('show tables from ' . CPerfomanceTable::escapeTable($DB->DBName));
		}
		return new CPerfomanceTableList($rsTables);
	}

	public function Fetch()
	{
		global $DB;
		$ar = parent::Fetch();
		if ($ar)
		{
			if (isset($ar['Tables_in_' . $DB->DBName]))
			{
				$ar = [
					'TABLE_NAME' => $ar['Tables_in_' . $DB->DBName],
					'ENGINE_TYPE' => '',
					'NUM_ROWS' => '',
					'BYTES' => '',
					'BYTES_INDEX' => '',
				];
			}
			else
			{
				$ar = [
					'TABLE_NAME' => $ar['Name'],
					'ENGINE_TYPE' => $ar['Comment'] === 'VIEW' ? 'VIEW' : $ar['Engine'],
					'NUM_ROWS' => $ar['Rows'],
					'BYTES' => $ar['Data_length'],
					'BYTES_INDEX' => $ar['Index_length'],
				];
			}
		}
		return $ar;
	}
}

class CPerfomanceTable extends CAllPerfomanceTable
{
	public function Init($TABLE_NAME)
	{
		$TABLE_NAME = trim($TABLE_NAME, '`');
		$this->TABLE_NAME = $TABLE_NAME;
	}

	public function IsExists($TABLE_NAME = false)
	{
		global $DB;

		if ($TABLE_NAME === false)
		{
			$TABLE_NAME = $this->TABLE_NAME;
		}
		if ($TABLE_NAME == '')
		{
			return false;
		}

		$TABLE_NAME = trim($TABLE_NAME, '`');

		$strSql = "
			SHOW TABLES LIKE '" . $DB->ForSql($TABLE_NAME) . "'
		";
		$rs = $DB->Query($strSql);
		if ($rs->Fetch())
		{
			return true;
		}
		else
		{
			return false;
		}
	}

	public function GetIndexes($TABLE_NAME = false)
	{
		global $DB;
		static $cache = [];

		if ($TABLE_NAME === false)
		{
			$TABLE_NAME = $this->TABLE_NAME;
		}
		if ($TABLE_NAME == '')
		{
			return [];
		}

		$TABLE_NAME = trim($TABLE_NAME, '`');

		if (!array_key_exists($TABLE_NAME, $cache))
		{
			$strSql = 'SHOW INDEXES FROM ' . static::escapeTable($TABLE_NAME);
			$arResult = [];
			$rsInd = $DB->Query($strSql, true);
			if ($rsInd)
			{
				while ($arInd = $rsInd->Fetch())
				{
					$arResult[$arInd['Key_name']][$arInd['Seq_in_index']] = $arInd['Column_name'];
				}
			}
			$cache[$TABLE_NAME] = $arResult;
		}

		return $cache[$TABLE_NAME];
	}

	public function GetUniqueIndexes($TABLE_NAME = false)
	{
		global $DB;
		static $cache = [];

		if ($TABLE_NAME === false)
		{
			$TABLE_NAME = $this->TABLE_NAME;
		}
		if ($TABLE_NAME == '')
		{
			return [];
		}

		$TABLE_NAME = trim($TABLE_NAME, '`');

		if (!array_key_exists($TABLE_NAME, $cache))
		{
			$strSql = 'SHOW INDEXES FROM ' . static::escapeTable($TABLE_NAME);
			$arResult = [];
			$rsInd = $DB->Query($strSql, true);
			if ($rsInd)
			{
				while ($arInd = $rsInd->Fetch())
				{
					if (!$arInd['Non_unique'])
					{
						$arResult[$arInd['Key_name']][$arInd['Seq_in_index']] = $arInd['Column_name'];
					}
				}
			}
			$cache[$TABLE_NAME] = $arResult;
		}

		return $cache[$TABLE_NAME];
	}

	public function GetTableFields($TABLE_NAME = false, $bExtended = false)
	{
		static $cache = [];

		if ($TABLE_NAME === false)
		{
			$TABLE_NAME = $this->TABLE_NAME;
		}
		if ($TABLE_NAME == '')
		{
			return false;
		}

		$TABLE_NAME = trim($TABLE_NAME, '`');

		if (!array_key_exists($TABLE_NAME, $cache))
		{
			global $DB;

			$strSql = 'SHOW COLUMNS FROM ' . static::escapeTable($TABLE_NAME);
			$rs = $DB->Query($strSql);
			$arResult = [];
			$arResultExt = [];
			while ($ar = $rs->Fetch())
			{
				$canSort = true;
				$match = [];
				if (preg_match('/^(varchar|char|varbinary)\\((\\d+)\\)/', $ar['Type'], $match))
				{
					$ar['DATA_TYPE'] = 'string';
					$ar['DATA_LENGTH'] = $match[2];
					if ($match[2] == 1 && ($ar['Default'] === 'N' || $ar['Default'] === 'Y'))
					{
						$ar['ORM_DATA_TYPE'] = 'boolean';
					}
					else
					{
						$ar['ORM_DATA_TYPE'] = 'string';
					}
				}
				elseif (preg_match('/^(varchar|char)/', $ar['Type']))
				{
					$ar['DATA_TYPE'] = 'string';
					$ar['ORM_DATA_TYPE'] = 'string';
				}
				elseif (preg_match('/^(text|longtext|mediumtext|longblob)/', $ar['Type']))
				{
					$canSort = false;
					$ar['DATA_TYPE'] = 'string';
					$ar['ORM_DATA_TYPE'] = 'text';
				}
				elseif (preg_match('/^(datetime|timestamp)/', $ar['Type']))
				{
					$ar['DATA_TYPE'] = 'datetime';
					$ar['ORM_DATA_TYPE'] = 'datetime';
				}
				elseif (preg_match('/^(date)/', $ar['Type']))
				{
					$ar['DATA_TYPE'] = 'date';
					$ar['ORM_DATA_TYPE'] = 'date';
				}
				elseif (preg_match('/^(int|smallint|bigint|tinyint)/', $ar['Type']))
				{
					$ar['DATA_TYPE'] = 'int';
					$ar['ORM_DATA_TYPE'] = 'integer';
				}
				elseif (preg_match('/^(float|double|decimal)/', $ar['Type']))
				{
					$ar['DATA_TYPE'] = 'double';
					$ar['ORM_DATA_TYPE'] = 'float';
				}
				else
				{
					$canSort = false;
					$ar['DATA_TYPE'] = 'unknown';
					$ar['ORM_DATA_TYPE'] = 'UNKNOWN';
				}
				$arResult[$ar['Field']] = $ar['DATA_TYPE'];
				$arResultExt[$ar['Field']] = [
					'type' => $ar['DATA_TYPE'],
					'length' => $ar['DATA_LENGTH'] ?? null,
					'nullable' => $ar['Null'] !== 'NO',
					'default' => $ar['Default'],
					'sortable' => $canSort,
					'orm_type' => $ar['ORM_DATA_TYPE'],
					'increment' => ($ar['Extra'] === 'auto_increment'),
					//"info" => $ar,
				];
			}
			$cache[$TABLE_NAME] = [$arResult, $arResultExt];
		}

		if ($bExtended)
		{
			return $cache[$TABLE_NAME][1];
		}
		else
		{
			return $cache[$TABLE_NAME][0];
		}
	}

	public static function escapeColumn($column)
	{
		return '`' . str_replace('`', '``', $column) . '`';
	}

	public static function escapeTable($tableName)
	{
		return '`' . str_replace('`', '``', $tableName) . '`';
	}

	public function getCreateIndexDDL($TABLE_NAME, $INDEX_NAME, $INDEX_COLUMNS)
	{
		$tableFields = $this->GetTableFields($TABLE_NAME, true);
		foreach ($INDEX_COLUMNS as $i => $field)
		{
			if ($tableFields[trim($field, '`[]"')]['orm_type'] === 'text')
			{
				$INDEX_COLUMNS[$i] = $field . '(100)';
			}
		}
		return parent::getCreateIndexDDL($TABLE_NAME, $INDEX_NAME, $INDEX_COLUMNS);
	}
}