<?php
/**
 * Akeeba Engine
 * The modular PHP5 site backup engine
 * @copyright Copyright (c)2006-2017 Nicholas K. Dionysopoulos / Akeeba Ltd
 * @license   GNU GPL version 3 or, at your option, any later version
 * @package   akeebaengine
 *
 */

namespace Akeeba\Engine\Dump\Reverse;

// Protection against direct access
defined('AKEEBAENGINE') or die();

use Akeeba\Engine\Driver\Base as DriverBase;
use Akeeba\Engine\Dump\Native\Mysql as NativeMysql;
use Akeeba\Engine\Factory;
use Psr\Log\LogLevel;


/**
 * A SQL Server database dump class, using reverse engineering of the
 * INFORMATION_SCHEMA views to deduce the DDL of the database entities.
 *
 * Configuration parameters:
 * host            <string>    SQL database server host name or IP address
 * port            <string>    SQL database server port (optional)
 * username        <string>    SQL user name, for authentication
 * password        <string>    SQL password, for authentication
 * database        <string>    SQL database
 * dumpFile        <string>    Absolute path to dump file; must be writable (optional; if left blank it is automatically calculated)
 */
class Sqlsrv extends NativeMysql
{
	/**
	 * Return the current database name by querying the database connection object (e.g. SELECT DATABASE() in MySQL)
	 *
	 * @return  string
	 */
	protected function getDatabaseNameFromConnection()
	{
		$db = $this->getDB();

		try
		{
			$ret = $db->setQuery('SELECT DB_NAME()')->loadResult();
		}
		catch (\Exception $e)
		{
			return '';
		}

		return empty($ret) ? '' : $ret;
	}

	/**
	 * Implements the constructor of the class
	 *
	 * @return Sqlsrv
	 */
	public function __construct()
	{
		Factory::getLog()->log(LogLevel::DEBUG, __CLASS__ . " :: New instance");

		$this->postProcessValues = true;
	}

	/**
	 * Applies the SQL compatibility setting
	 */
	protected function enforceSQLCompatibility()
	{
		//  Do nothing for SQL server
	}

	protected function getTablesToBackup()
	{
		$configuration = Factory::getConfiguration();
		$notracking = $configuration->get('engine.dump.native.nodependencies', 0);

		// First, get a map of table names <--> abstract names
		$this->reverse_engineer_db();
		if ($this->getError())
		{
			return;
		}

		if (!$notracking)
		{
			// Process dependencies and rearrange tables respecting them
			$this->process_dependencies();
			if ($this->getError())
			{
				return;
			}

			// Remove dependencies array
			$this->dependencies = array();
		}
	}

	protected function  reverse_engineer_db()
	{
		// Get a database connection
		Factory::getLog()->log(LogLevel::DEBUG, __CLASS__ . " :: Reverse engineering database");
		$db = $this->getDB();
		if ($this->getError())
		{
			return;
		}

		// Reset internal tables
		$this->table_name_map = array();
		$this->tables_data = array();
		$this->dependencies = array();

		// Get the list of all database tables and views
		Factory::getLog()->log(LogLevel::DEBUG, __CLASS__ . " :: Reverse engineering Tables");
		$this->reverse_engineer_tables($db);
		$this->reverse_engineer_views($db);

		// Optional backup of triggers, functions and procedures
		$registry = Factory::getConfiguration();
		$enable_entities = $registry->get('engine.dump.native.advanced_entitites', true);
		$notracking = $registry->get('engine.dump.native.nodependencies', 0);

		if ($enable_entities && ($notracking == 0))
		{
			// @todo Triggers
			// Factory::getLog()->log(LogLevel::DEBUG, __CLASS__." :: Reverse engineering Triggers");

			// @todo Functions
			// Factory::getLog()->log(LogLevel::DEBUG, __CLASS__." :: Reverse engineering Functions");

			// @todo Procedures
			// Factory::getLog()->log(LogLevel::DEBUG, __CLASS__." :: Reverse engineering Procedures");
		}
	}

	/**
	 * Reverse engineers the Table definitions of this database
	 *
	 * @param   DriverBase $dbi Database connection to INFORMATION_SCHEMA
	 */
	protected function reverse_engineer_tables(&$dbi)
	{
		$schema_name = $this->database;
		$sql = 'SELECT * FROM sys.objects WHERE type = ' . $dbi->quote('U') . ' ORDER BY [name] ASC';
		$dbi->setQuery($sql);
		$all_tables = $dbi->loadObjectList();

		$registry = Factory::getConfiguration();
		$root = $registry->get('volatile.database.root', '[SITEDB]');

		// If we have filters, make sure the tables pass the filtering
		$filters = Factory::getFilters();
		if (!empty($all_tables))
		{
			foreach ($all_tables as $table_object)
			{
				// Extract the table name
				$table_name = $table_object->name;

				// Filter and convert
				if (substr($table_name, 0, 3) == '#__')
				{
					$this->setWarning(__CLASS__ . " :: Table $table_name has a prefix of #__. This would cause restoration errors; table skipped.");

					continue;
				}
				$table_abstract = $this->getAbstract($table_name);
				if (substr($table_abstract, 0, 4) != 'bak_') // Skip backup tables
				{
					// Apply exclusion filters
					if (!$filters->isFiltered($table_abstract, $root, 'dbobject', 'all'))
					{
						Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Adding $table_name (internal name $table_abstract)");
						$this->table_name_map[$table_name] = $table_abstract;
					}
					else
					{
						Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Skipping $table_name (internal name $table_abstract)");
						continue;
					}
				}
				else
				{
					Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Backup table $table_name automatically skipped.");
					continue;
				}

				// Still here? The table is added. We now have to store its
				// create command, dependency info and so on
				$new_entry = array(
					'type'         => 'table',
					'dump_records' => true
				);

				// Table Data Filter - skip dumping table contents of filtered out tables
				if ($filters->isFiltered($table_abstract, $root, 'dbobject', 'content'))
				{
					$new_entry['dump_records'] = false;
				}

				$new_entry['create'] = $this->get_create_table($dbi, $table_name, $table_abstract, $table_object, $dependencies);
				$new_entry['dependencies'] = $dependencies;

				$this->tables_data[$table_name] = $new_entry;
			}
		}
	}

	/**
	 * Gets the CREATE TABLE command of a given table
	 *
	 * @param   DriverBase $dbi            The db connection to the INFORMATION_SCHEMA db
	 * @param   string           $table_name     The name of the table
	 * @param   string           $table_abstract The abstract name of the table
	 * @param   \stdClass         $table_object   The SYS.OBJECTS record for this table
	 * @param   array            $dependencies   Dependency tracking information
	 *
	 * @return  string  The CREATE TABLE definition
	 */
	protected function get_create_table(&$dbi, $table_name, $table_abstract, $table_object, &$dependencies)
	{
		$configuration = Factory::getConfiguration();
		$notracking = $configuration->get('engine.dump.native.nodependencies', 0);
		$useabstract = Factory::getEngineParamsProvider()->getScriptingParameter('db.abstractnames', 1);

		$columns_sql = array();
		$keys_sql = array();
		$constraints_sql = array();
		$indexes_sql = array();

		// =====================================================================
		// ========== GENERATE SQL FOR COLUMNS
		// =====================================================================
		// Get identity columns for this table
		$sysObjectID = $table_object->object_id;
		$query = 'SELECT COUNT(*) FROM sys.identity_columns WHERE object_id = ' . $dbi->quote($sysObjectID);
		$dbi->setQuery($query);
		$countIdentityColumns = $dbi->loadResult();
		if ($countIdentityColumns)
		{
			/**
			 * $query = 'SELECT column_id, name, seed_value, increment_value FROM sys.identity_columns WHERE object_id = ' . $dbi->quote($sysObjectID)
			 * . ' ORDER BY column_id ASC';
			 **/
			$query = 'select * from INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = ' . $dbi->quote($table_name) .
				'and COLUMNPROPERTY(object_id(TABLE_NAME), COLUMN_NAME, \'IsIdentity\') = 1';
			$dbi->setQuery($query);
			$identityColumns = $dbi->loadAssocList('COLUMN_NAME');
		}
		else
		{
			$identityColumns = array();
		}

		// Get columns
		$query = 'SELECT * FROM information_schema.columns WHERE table_catalog = ' . $dbi->quote($this->database)
			. ' AND table_name = ' . $dbi->quote($table_name) . ' ORDER BY ordinal_position ASC';
		$dbi->setQuery($query);
		$allColumns = $dbi->loadObjectList();
		foreach ($allColumns as $oColumn)
		{
			$line = '[' . $oColumn->COLUMN_NAME . '] [' . $oColumn->DATA_TYPE . ']';
			switch ($oColumn->DATA_TYPE)
			{
				case 'bigint':
				case 'int':
				case 'smallint':
				case 'tinyint':
					/* NOT SUPPORTED *
					$precision = $oColumn->NUMERIC_PRECISION;
					$scale = $oColumn->NUMERIC_SCALE;
					if ($precision)
					{
						$line .= '(';
						if ($precision && $scale)
						{
							$line .= $precision . ', ' . $scale;
						}
						else
						{
							$line .= $precision;
						}
						$line .= ')';
					}
					*/
					break;

				case 'nvarchar':
				case 'nchar':
					$len = $oColumn->CHARACTER_MAXIMUM_LENGTH;
					if ($len < 0)
					{
						$len = 'max';
					}
					$line .= '(' . $len . ')';
					break;

				case 'datetime':
				case 'datetime2':
					/* NOT SUPPORTED *
					$precision = $oColumn->DATETIME_PRECISION;
					if ($precision)
					{
						$line .= '(' . $precision . ')';
					}
					*/
					break;

				case 'float':
				case 'real':
					break;
			}
			$line .= ($oColumn->IS_NULLABLE == 'YES') ? 'NULL ' : 'NOT NULL ';

			if (array_key_exists($oColumn->COLUMN_NAME, $identityColumns))
			{
				/**
				 * $seed = $identityColumns[$oColumn->COLUMN_NAME]['seed_value'];
				 * $increment = $identityColumns[$oColumn->COLUMN_NAME]['increment_value'];
				 **/
				// fake the seed and increment because Microsoft sucks and their API is broken!

				$qLala = 'SELECT MAX(' . $oColumn->COLUMN_NAME . ') FROM ' . $dbi->quoteName($table_name);
				$dbi->setQuery($qLala);
				$seed = (int)($dbi->loadResult());

				$increment = 1;

				$line .= ' IDENTITY (' . $seed . ', ' . $increment . ')';
			}

			$line .= ($oColumn->COLUMN_DEFAULT == '') ? '' : (' DEFAULT ' . $oColumn->COLUMN_DEFAULT . ' ');

			$columns_sql[] = $line;
		}

		// =====================================================================
		// ========== GENERATE SQL FOR KEYS AND INDICES
		// =====================================================================
		// Get the primary and unique key names
		$query = 'SELECT * from sys.indexes where object_id = OBJECT_ID(' . $dbi->q($table_name) . ')';
		$dbi->setQuery($query);
		$allKeys = $dbi->loadObjectList('name');

		// Get the columns per key and key information
		$query = 'select c.name, ic.* from sys.index_columns as ic inner join sys.columns as c on(c.object_id = ic.object_id and c.column_id = ic.column_id) '
			. 'where ic.object_id = OBJECT_ID(' . $dbi->q($table_name) . ') order by index_id ASC, index_column_id ASC';
		$dbi->setQuery($query);
		$allColumns = $dbi->loadObjectList();

		$rawKeys = array();
		if (!empty($allKeys))
		{
			foreach ($allKeys as $currentKey)
			{
				if (empty($currentKey->name))
				{
					continue;
				}

				$isUnique = $currentKey->is_unique == 1;
				$isPrimary = $currentKey->is_primary_key == 1;

				$keyName = $currentKey->name;
				if ($useabstract && strlen($this->prefix) && substr($this->prefix, -1) == '_')
				{
					$keyName = str_replace($this->prefix, '#__', $keyName);
				}

				if ($isPrimary)
				{
					$line = 'CONSTRAINT [' . $keyName . '] ';
					$line .= 'PRIMARY KEY ' . $currentKey->type_desc;
				}
				elseif ($isUnique)
				{
					$line = 'CONSTRAINT [' . $keyName . '] ';
					$line .= 'UNIQUE ' . $currentKey->type_desc;
				}
				else
				{
					//$line = 'CREATE ' . $currentKey->type_desc . ' INDEX [' . $this->getAbstract($currentKey->name) . '] ON [' . $table_abstract . ']';
					if ($useabstract)
					{
						$line = 'CREATE INDEX [' . $keyName . '] ON [' . $table_abstract . ']';
					}
					else
					{
						$line = 'CREATE INDEX [' . $keyName . '] ON [' . $table_name . ']';
					}
				}
				$line .= '(';

				// Add columns
				$cols = array();
				foreach ($allColumns as $oColumn)
				{
					if ($oColumn->index_id != $currentKey->index_id)
					{
						continue;
					}

					$cols[] = '[' . $oColumn->name . '] ' . ($oColumn->is_descending_key ? 'DESC' : 'ASC');
				}
				$line .= implode(', ', $cols);

				$line .= ')';

				// append WITH (...blah...) ON [PRIMARY]
				$line .= ' WITH (';
				if ($isPrimary || $isUnique)
				{
					$line .= 'PAD_INDEX = ' . ($currentKey->is_padded ? 'ON' : 'OFF');
					$line .= ', STATISTICS_NORECOMPUTE  = OFF, IGNORE_DUP_KEY = OFF, ALLOW_ROW_LOCKS  = ON, ALLOW_PAGE_LOCKS  = ON) ON [PRIMARY]';
				}
				else
				{
					$line .= 'STATISTICS_NORECOMPUTE  = OFF, IGNORE_DUP_KEY = OFF, DROP_EXISTING = OFF, ONLINE = OFF)';
				}

				// Add primary/unique keys to $keys_sql, add indices to $indexes_sql and paste them as new lines below the create command
				if ($isPrimary || $isUnique)
				{
					$keys_sql[] = $line;
				}
				else
				{
					$indexes_sql[] = $line . ';';
				}
			}
		}

		// =====================================================================
		// ========== GENERATE SQL FOR FOREIGN KEYS
		// =====================================================================
		// Get the foreign key names
		$query = 'SELECT * FROM sys.foreign_keys WHERE parent_object_id = ' . $dbi->quote($sysObjectID)
			. ' AND [type] = ' . $dbi->quote('F');
		$dbi->setQuery($query);
		$foreignKeyInfo = $dbi->loadObjectList('name');


		$rawConstraints = array();

		if (!empty($foreignKeyInfo)) foreach ($foreignKeyInfo as $oKey)
		{
			// Get the columns per key and key information
			$query = 'SELECT * FROM sys.foreign_key_columns WHERE parent_object_id = ' . $dbi->quote($sysObjectID) .
				' AND constraint_object_id = ' . $dbi->q($oKey->object_id);
			$dbi->setQuery($query);
			$allFKColumns = $dbi->loadObjectList();

			// Get referenced table's name
			$refID = $oKey->referenced_object_id;
			$query = 'SELECT name FROM sys.objects WHERE object_id = ' . $dbi->q($refID);
			$dbi->setQuery($query);
			$refObjectName = $dbi->loadResult();

			// Initialise column maps
			$FKcolumnMap = array();

			// Loop through each column and map parent_object_id / parent_column_id,  referenced_object_id / referenced_column_id to column names
			foreach ($allFKColumns as $oColumn)
			{
				$objectIDs = array($oColumn->parent_object_id, $oColumn->referenced_object_id);
				foreach ($objectIDs as $oid)
				{
					if (!array_key_exists($oid, $FKcolumnMap))
					{
						$query = $dbi->getQuery(true)
							->select(array('name', 'column_id'))
							->from('sys.columns')
							->where('object_id = ' . $dbi->q($oid));
						$dbi->setQuery($query);
						$FKcolumnMap[$oid] = $dbi->loadObjectList('column_id');
					}
				}
			}

			$keyName = $oKey->name;
			if (strlen($this->prefix) && substr($this->prefix, -1) == '_')
			{
				$keyName = str_replace($this->prefix, '#__', $keyName);
			}
			$line = 'CONSTRAINT [' . $keyName . '] FOREIGN KEY (';

			$tempCols = array();
			foreach ($allFKColumns as $oColumn)
			{
				$oid = $oColumn->parent_object_id;
				$cid = $oColumn->parent_column_id;
				$tempCols[] = '[' . $FKcolumnMap[$oid][$cid]->name . ']';
			}
			$line .= implode(', ', $tempCols);

			// Add a reference hit
			$this->dependencies[$refObjectName][] = $table_name;
			// Add the dependency to this table's metadata
			$dependencies[] = $refObjectName;

			if ($useabstract)
			{
				$refObjectName = $this->getAbstract($refObjectName);
			}

			$line .= ') REFERENCES [' . $refObjectName . '] (';

			$tempCols = array();
			foreach ($allFKColumns as $oColumn)
			{
				$oid = $oColumn->referenced_object_id;
				$cid = $oColumn->referenced_column_id;
				$tempCols[] = '[' . $FKcolumnMap[$oid][$cid]->name . ']';
			}
			$line .= implode(', ', $tempCols);

			$line .= ') ';

			// Tuck the delete and update actions
			$line .= ' ON DELETE ';
			switch ($oKey->delete_referential_action_desc)
			{
				case 'NO_ACTION';
					$line .= 'NO ACTION';
					break;

				case 'CASCADE';
					$line .= 'CASCADE';
					break;

				case 'SET_NULL';
					$line .= 'SET NULL';
					break;

				case 'SET_DEFAULT';
					$line .= 'SET DEFAULT';
					break;
			}

			$line .= ' ON UPDATE ';
			switch ($oKey->update_referential_action_desc)
			{
				case 'NO_ACTION';
					$line .= 'NO ACTION';
					break;

				case 'CASCADE';
					$line .= 'CASCADE';
					break;

				case 'SET_NULL';
					$line .= 'SET NULL';
					break;

				case 'SET_DEFAULT';
					$line .= 'SET DEFAULT';
					break;
			}

			if ($oKey->is_not_for_replication)
			{
				$line .= ' NOT FOR REPLICATION';
			}

			// add to the $constraints_sql array
			$constraints_sql[] = $line;
		}

		// =====================================================================
		// ==========CONSTRUCT THE TABLE CREATE STATEMENT
		// =====================================================================
		// Create the SQL output
		if ($useabstract)
		{
			$table_sql = "CREATE TABLE [$table_abstract] (";
		}
		else
		{
			$table_sql = "CREATE TABLE [$table_name] (";
		}

		$table_sql .= implode(',', $columns_sql);
		if (count($keys_sql))
		{
			$table_sql .= ',' . implode(',', $keys_sql);
		}
		if (count($constraints_sql))
		{
			$table_sql .= ',' . implode(',', $constraints_sql);
		}
		$table_sql .= ")";

		$table_sql .= ";\n";

		$table_sql .= implode(";\n", $indexes_sql);
		if (count($indexes_sql) >= 1)
		{
			$table_sql .= "\n";
		}

		return $table_sql;
	}

	/**
	 * Reverse engineers the View definitions of this database
	 *
	 * @param   DriverBase $dbi Database connection to INFORMATION_SCHEMA
	 */
	protected function reverse_engineer_views(&$dbi)
	{
		$schema_name = $this->database;
		$sql = 'SELECT * FROM [INFORMATION_SCHEMA].[VIEWS] WHERE [table_catalog] = ' . $dbi->quote($schema_name);
		$dbi->setQuery($sql);
		$all_views = $dbi->loadObjectList();

		$registry = Factory::getConfiguration();
		$root = $registry->get('volatile.database.root', '[SITEDB]');

		// If we have filters, make sure the tables pass the filtering
		$filters = Factory::getFilters();
		// First pass: populate the table_name_map
		if (!empty($all_views)) foreach ($all_views as $table_object)
		{
			// Extract the table name
			$table_name = $table_object->TABLE_NAME;

			// Filter and convert
			if (substr($table_name, 0, 3) == '#__')
			{
				$this->setWarning(__CLASS__ . " :: Table $table_name has a prefix of #__. This would cause restoration errors; table skipped.");

				continue;
			}
			$table_abstract = $this->getAbstract($table_name);
			if (substr($table_abstract, 0, 4) != 'bak_') // Skip backup tables
			{
				// Apply exclusion filters
				if (!$filters->isFiltered($table_abstract, $root, 'dbobject', 'all'))
				{
					Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Adding $table_name (internal name $table_abstract)");
					$this->table_name_map[$table_name] = $table_abstract;
				}
				else
				{
					Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Skipping $table_name (internal name $table_abstract)");
					continue;
				}
			}
			else
			{
				Factory::getLog()->log(LogLevel::INFO, __CLASS__ . " :: Backup view $table_name automatically skipped.");
				continue;
			}
		}

		// Second pass: get the create commands
		if (!empty($all_views)) foreach ($all_views as $table_object)
		{
			// Extract the table name
			$table_name = $table_object->TABLE_NAME;

			if (!in_array($table_name, $this->table_name_map))
			{
				// Skip any views which have been filtered out
				continue;
			}

			$table_abstract = $this->getAbstract($table_name);

			// Still here? The view is added. We now have to store its
			// create command, dependency info and so on
			$new_entry = array(
				'type'         => 'view',
				'dump_records' => false
			);

			$dependencies = array();
			$table_sql = $table_object->VIEW_DEFINITION;
			$old_table_sql = $table_sql;
			foreach ($this->table_name_map as $ref_normal => $ref_abstract)
			{
				if ($pos = strpos($table_sql, ".$ref_normal"))
				{
					// Add a reference hit
					$this->dependencies[$ref_normal][] = $table_name;
					// Add the dependency to this table's metadata
					$dependencies[] = $ref_normal;
					// Do the replacement
					$table_sql = str_replace(".$ref_normal", ".$ref_abstract", $table_sql);
				}
			}

			// On DB only backup we don't want any replacing to take place, do we?
			if (!Factory::getEngineParamsProvider()->getScriptingParameter('db.abstractnames', 1))
			{
				$table_sql = $old_table_sql;
			}

			// Replace newlines with spaces
			$table_sql = str_replace("\n", " ", $table_sql) . ";\n";
			$table_sql = str_replace("\r", " ", $table_sql);
			$table_sql = str_replace("\t", " ", $table_sql);

			$new_entry['create'] = $table_sql;
			$new_entry['dependencies'] = $dependencies;

			$this->tables_data[$table_name] = $new_entry;
		}
	}

	/**
	 * Creates a drop query from a CREATE query
	 *
	 * @param $query string The CREATE query to process
	 *
	 * @return string The DROP statement
	 */
	protected function createDrop($query)
	{
		$db = $this->getDB();

		// Initialize
		$dropQuery = '';

		// Parse CREATE TABLE commands
		if (substr($query, 0, 12) == 'CREATE TABLE')
		{
			// Try to get the table name
			$restOfQuery = trim(substr($query, 12, strlen($query) - 12)); // Rest of query, after CREATE TABLE
			// Is there a backtick?
			if (substr($restOfQuery, 0, 1) == '[')
			{
				// There is... Good, we'll just find the matching backtick
				$pos = strpos($restOfQuery, ']', 1);
				$tableName = substr($restOfQuery, 1, $pos - 1);
			}
			else
			{
				// Nope, let's assume the table name ends in the next blank character
				$pos = strpos($restOfQuery, ' ', 1);
				$tableName = substr($restOfQuery, 1, $pos - 1);
			}

			unset($restOfQuery);

			/**
			 * Defense against CVE-2016-5483 ("Bad Dump") affecting MySQL, Percona, MariaDB and other MySQL clones.
			 * Possibly not affecting Microsoft SQL Server but we'd better be safe than sorry.
			 */
			$tableName = str_replace(array("\r", "\n"), array('', ''), $tableName);

			// Try to drop the table anyway
			$dropQuery = 'IF OBJECT_ID(' . $db->q($tableName) . ') IS NOT NULL DROP TABLE ' . $db->qn($tableName) . ';';
		}
		// Parse CREATE VIEW commands
		elseif ((substr($query, 0, 7) == 'CREATE ') && (strpos($query, ' VIEW ') !== false))
		{
			// Try to get the view name
			$view_pos = strpos($query, ' VIEW ');
			$restOfQuery = trim(substr($query, $view_pos + 6)); // Rest of query, after VIEW string
			// Is there a backtick?
			if (substr($restOfQuery, 0, 1) == '[')
			{
				// There is... Good, we'll just find the matching backtick
				$pos = strpos($restOfQuery, ']', 1);
				$tableName = substr($restOfQuery, 1, $pos - 1);
			}
			else
			{
				// Nope, let's assume the table name ends in the next blank character
				$pos = strpos($restOfQuery, ' ', 1);
				$tableName = substr($restOfQuery, 1, $pos - 1);
			}

			unset($restOfQuery);

			/**
			 * Defense against CVE-2016-5483 ("Bad Dump") affecting MySQL, Percona, MariaDB and other MySQL clones.
			 * Possibly not affecting Microsoft SQL Server but we'd better be safe than sorry.
			 */
			$tableName = str_replace(array("\r", "\n"), array('', ''), $tableName);

			$dropQuery = 'IF OBJECT_ID(' . $db->q($tableName) . ') IS NOT NULL DROP VIEW ' . $db->qn($tableName) . ';';
		}

		return $dropQuery;
	}

	/**
	 * Post process a quoted value before it's written to the database dump.
	 * So far it's only required for SQL Server which has a problem escaping
	 * newline characters...
	 *
	 * @param   string $value The quoted value to post-process
	 *
	 * @return  string
	 */
	protected function postProcessQuotedValue($value)
	{
		$value = str_replace("\r\n", "' + CHAR(10) + CHAR(13) + N'", $value);
		$value = str_replace("\r", "' + CHAR(13) + N'", $value);
		$value = str_replace("\n", "' + CHAR(10) + N'", $value);

		return $value;
	}

	/**
	 * Returns a preamble for the data dump portion of the SQL backup. This is
	 * used to output commands before the first INSERT INTO statement for a
	 * table when outputting a plain SQL file.
	 *
	 * Practical use: the SET IDENTITY_INSERT sometable ON required for SQL Server
	 *
	 * @param   string  $tableAbstract Abstract name of the table, e.g. #__foobar
	 * @param   string  $tableName     Real name of the table, e.g. abc_foobar
	 * @param   integer $maxRange      Row count on this table
	 *
	 * @return  string   The SQL commands you want to be written in the dump file
	 */
	protected function getDataDumpPreamble($tableAbstract, $tableName, $maxRange)
	{
		if ($maxRange > 0)
		{
			// Do we have an identity column?
			$db = $this->getDB();
			$query = $db->getQuery(true)
				->select('COUNT(*)')
				->from('sys.identity_columns')
				->where("object_id = OBJECT_ID(" . $db->q($tableName) . ")");
			$db->setQuery($query);
			$idColumns = $db->loadResult();

			if ($idColumns < 1)
			{
				return '';
			}

			return "SET IDENTITY_INSERT [$tableName] ON;\n";
		}

		return '';
	}

	/**
	 * Returns an epilogue for the data dump portion of the SQL backup. This is
	 * used to output commands after the last INSERT INTO statement for a
	 * table when outputting a plain SQL file.
	 *
	 * Practical use: the SET IDENTITY_INSERT sometable OFF required for SQL Server
	 *
	 * @param   string  $tableAbstract Abstract name of the table, e.g. #__foobar
	 * @param   string  $tableName     Real name of the table, e.g. abc_foobar
	 * @param   integer $maxRange      Row count on this table
	 *
	 * @return  string   The SQL commands you want to be written in the dump file
	 */
	protected function getDataDumpEpilogue($tableAbstract, $tableName, $maxRange)
	{
		if ($maxRange > 0)
		{
			// Do we have an identity column?
			$db = $this->getDB();
			$query = $db->getQuery(true)
				->select('COUNT(*)')
				->from('sys.identity_columns')
				->where("object_id = OBJECT_ID(" . $db->q($tableName) . ")");
			$db->setQuery($query);
			$idColumns = $db->loadResult();

			if ($idColumns < 1)
			{
				return '';
			}

			return "SET IDENTITY_INSERT [$tableName] OFF;\n";
		}

		return '';
	}

	/**
	 * Return a list of field names for the INSERT INTO statements. This is only
	 * required for Microsoft SQL Server because without it the SET IDENTITY_INSERT
	 * has no effect.
	 *
	 * @param   array   $fieldNames  A list of field names in array format
	 * @param   integer $numOfFields The number of fields we should be dumping
	 *
	 * @return  string
	 */
	protected function getFieldListSQL($fieldNames, $numOfFields)
	{
		if (count($fieldNames) < $numOfFields)
		{
			return '';
		}
		elseif (count($fieldNames) > $numOfFields)
		{
			$fieldNames = array_slice($fieldNames, 0, $numOfFields);
		}

		$db = $this->getDB();

		$temp = array();
		foreach ($fieldNames as $f)
		{
			$temp[] = $db->quoteName($f);
		}

		return '(' . implode(', ', $temp) . ')';
	}

	protected function setAutoIncrementInfo()
	{
		// Does nothing, there is no way to know if the table has an autoincrement field in SQL server
	}
}