MsSql support for tables, columns, indexes, foreign keys (#215)

src/Database/Drivers/MsSqlDriver.php

@@ -19,8 +19,13 @@ class MsSqlDriver implements Nette\Database\ISupplementalDriver
{
	use Nette\SmartObject;

	/** @var Nette\Database\Connection */
	private $connection;


	public function initialize(Nette\Database\Connection $connection, array $options): void
	{
		$this->connection = $connection;
	}


@@ -81,25 +86,123 @@ public function applyLimit(string &$sql, ?int $limit, ?int $offset): void

	public function getTables(): array
	{
		throw new Nette\NotImplementedException;
		$tables = [];
		foreach ($this->connection->query('SELECT TABLE_SCHEMA, TABLE_NAME, TABLE_TYPE FROM INFORMATION_SCHEMA.TABLES') as $row) {
			$tables[] = [
				'name' => $row['TABLE_SCHEMA'] . '.' . $row['TABLE_NAME'],
				'view' => ($row['TABLE_TYPE'] ?? null) === 'VIEW',
			];
		}
		return $tables;
	}


	public function getColumns(string $table): array
	{
		throw new Nette\NotImplementedException;
		[$table_schema, $table_name] = explode('.', $table);
		$columns = [];

		$query = "
			SELECT
				COLUMN_NAME,
				DATA_TYPE,
				CHARACTER_MAXIMUM_LENGTH,
				NUMERIC_PRECISION,
				IS_NULLABLE,
				COLUMN_DEFAULT,
				DOMAIN_NAME
			FROM
				INFORMATION_SCHEMA.COLUMNS
			WHERE
				TABLE_SCHEMA = {$this->connection->quote($table_schema)}
				AND TABLE_NAME = {$this->connection->quote($table_name)}";

		foreach ($this->connection->query($query) as $row) {
			$columns[] = [
				'name' => $row['COLUMN_NAME'],
				'table' => $table,
				'nativetype' => strtoupper($row['DATA_TYPE']),
				'size' => $row['CHARACTER_MAXIMUM_LENGTH'] ?? ($row['NUMERIC_PRECISION'] ?? null),
				'unsigned' => false,
				'nullable' => $row['IS_NULLABLE'] === 'YES',
				'default' => $row['COLUMN_DEFAULT'],
				'autoincrement' => $row['DOMAIN_NAME'] === 'COUNTER',
				'primary' => $row['COLUMN_NAME'] === 'ID',
				'vendor' => (array) $row,
			];
		}
		return $columns;
	}


	public function getIndexes(string $table): array
	{
		throw new Nette\NotImplementedException;
		[, $table_name] = explode('.', $table);
		$indexes = [];

		$query = "
			SELECT
				 name_index = ind.name,
				 id_column = ic.index_column_id,
				 name_column = col.name,
				 ind.is_unique,
				 ind.is_primary_key
			FROM
				sys.indexes ind
				INNER JOIN sys.index_columns ic ON  ind.object_id = ic.object_id and ind.index_id = ic.index_id
				INNER JOIN sys.columns col ON ic.object_id = col.object_id and ic.column_id = col.column_id
				INNER JOIN sys.tables t ON ind.object_id = t.object_id
			WHERE
				 t.name = {$this->connection->quote($table_name)}
			ORDER BY
				 t.name, ind.name, ind.index_id, ic.index_column_id";

		foreach ($this->connection->query($query) as $row) {
			$indexes[$row['name_index']]['name'] = $row['name_index'];
			$indexes[$row['name_index']]['unique'] = $row['is_unique'] !== 'False';
			$indexes[$row['name_index']]['primary'] = $row['is_primary_key'] !== 'False';
			$indexes[$row['name_index']]['columns'][$row['id_column'] - 1] = $row['name_column'];
		}
		return array_values($indexes);
	}


	public function getForeignKeys(string $table): array
	{
		throw new Nette\NotImplementedException;
		[$table_schema, $table_name] = explode('.', $table);
		$keys = [];

		$query = "
			SELECT
				obj.name AS [fk_name],
				col1.name AS [column],
				tab2.name AS [referenced_table],
				col2.name AS [referenced_column]
			FROM
				sys.foreign_key_columns fkc
				INNER JOIN sys.objects obj
					ON obj.object_id = fkc.constraint_object_id
				INNER JOIN sys.tables tab1
					ON tab1.object_id = fkc.parent_object_id
				INNER JOIN sys.schemas sch
					ON tab1.schema_id = sch.schema_id
				INNER JOIN sys.columns col1
					ON col1.column_id = parent_column_id AND col1.object_id = tab1.object_id
				INNER JOIN sys.tables tab2
					ON tab2.object_id = fkc.referenced_object_id
				INNER JOIN sys.columns col2
				ON col2.column_id = referenced_column_id AND col2.object_id = tab2.object_id
			WHERE
				tab1.name = {$this->connection->quote($table_name)}";

		foreach ($this->connection->query($query) as $id => $row) {
			$keys[$id]['name'] = $row['fk_name'];
			$keys[$id]['local'] = $row['column'];
			$keys[$id]['table'] = $table_schema . '.' . $row['referenced_table'];
			$keys[$id]['foreign'] = $row['referenced_column'];
		}

		return array_values($keys);
	}