sql_cmd 0.3.0

data/sql_scripts/Database/GetBackupFiles.sql

@@ -0,0 +1,31 @@
+DECLARE @Depth int
+DECLARE @FileFlag int
+DECLARE @TargetFolder nvarchar(255)
+DECLARE @BackupBaseName nvarchar(255)
+DECLARE @LogOnly bit
+DECLARE @BackupFileExtension nvarchar(3)
+
+SET @Depth = 1
+SET @FileFlag = 1
+SET @TargetFolder = '$(targetfolder)'
+SET @BackupBaseName = '$(bkupname)'
+SET @LogOnly = '$(logonly)'
+
+SET @BackupFileExtension = CASE WHEN @LogOnly = 1 THEN 'trn' ELSE 'bak' END
+
+DECLARE @DirTree TABLE (
+    FileName nvarchar(255),
+    Depth smallint,
+    FileFlag smallint
+  )
+
+INSERT INTO @DirTree
+EXEC xp_dirtree @TargetFolder, @Depth, @FileFlag
+
+SELECT * FROM @DirTree
+WHERE FileFlag = @FileFlag AND (
+		[FileName] LIKE @BackupBaseName + '[_][0-9][0-9][0-9][0-9][0-9][0-9].part[0-9]%.' + @BackupFileExtension
+		OR [FileName] LIKE @BackupBaseName + '[_][0-9][0-9][0-9][0-9][0-9][0-9].' + @BackupFileExtension
+		OR [FileName] LIKE @BackupBaseName + '.part[0-9]%.' + @BackupFileExtension
+		OR [FileName] LIKE @BackupBaseName + '.' + @BackupFileExtension
+	)

data/sql_scripts/Database/GetBackupHeaders.sql

@@ -0,0 +1,94 @@
+SET NOCOUNT ON
+
+DECLARE @bkup_files nvarchar(max)
+DECLARE @credential nvarchar(max)
+DECLARE @rowcount int
+
+DECLARE @HeaderOnly_cmd nvarchar(max)
+DECLARE @BackupSetPosition int
+DECLARE @Version14Plus bit
+DECLARE @SQLMajorVersion nvarchar(128)
+
+SET @bkup_files = '$(bkupfiles)'
+SET @credential = '$(credential)'
+
+SET @SQLMajorVersion = CONVERT(NVARCHAR(128), SERVERPROPERTY('ProductVersion'))
+SET @SQLMajorVersion = LEFT(@SQLMajorVersion, CHARINDEX('.', @SQLMajorVersion) - 1)
+
+SET @HeaderOnly_cmd =
+'DECLARE @BackupSets TABLE(
+	BackupName nvarchar(128),
+    BackupDescription nvarchar(255),
+    BackupType smallint,
+    ExpirationDate datetime,
+    Compressed bit,
+    Position smallint,
+    DeviceType tinyint,
+    UserName nvarchar(128),
+    ServerName nvarchar(128),
+    DatabaseName nvarchar(128),
+    DatabaseVersion int,
+    DatabaseCreationDate datetime,
+    BackupSize numeric(20,0),
+    FirstLSN numeric(25,0),
+    LastLSN numeric(25,0),
+    CheckpointLSN numeric(25,0),
+    DatabaseBackupLSN numeric(25,0),
+    BackupStartDate datetime,
+    BackupFinishDate datetime,
+    SortOrder smallint,
+    CodePage smallint,
+    UnicodeLocaleId int,
+    UnicodeComparisonStyle int,
+    CompatibilityLevel tinyint,
+    SoftwareVendorId int,
+    SoftwareVersionMajor int,
+    SoftwareVersionMinor int,
+    SoftwareVersionBuild int,
+    MachineName nvarchar(128),
+    Flags int,
+    BindingID uniqueidentifier,
+    RecoveryForkID uniqueidentifier,
+    Collation nvarchar(128),
+    FamilyGUID uniqueidentifier,
+    HasBulkLoggedData bit,
+    IsSnapshot bit,
+    IsReadOnly bit,
+    IsSingleUser bit,
+    HasBackupChecksums bit,
+    IsDamaged bit,
+    BeginsLogChain bit,
+    HasIncompleteMetaData bit,
+    IsForceOffline bit,
+    IsCopyOnly bit,
+    FirstRecoveryForkID uniqueidentifier,
+    ForkPointLSN numeric(25,0),
+    RecoveryModel nvarchar(60),
+    DifferentialBaseLSN numeric(25,0),
+    DifferentialBaseGUID uniqueidentifier,
+    BackupTypeDescription nvarchar(60),
+    BackupSetGUID uniqueidentifier,
+    CompressedBackupSize bit,'
+    + CASE WHEN (@SQLMajorVersion >= 11) THEN ' Containment tinyint,' ELSE '' END
+	+ CASE WHEN (@SQLMajorVersion >= 12) THEN
+		'KeyAlgorithm nvarchar(32),
+		EncryptorThumbprint varbinary(20),
+		EncryptorType nvarchar(32),' ELSE '' END
+	+ '--
+    -- This field added to retain order by
+    --
+    Seq int NOT NULL identity(1,1)
+)
+
+INSERT INTO @BackupSets
+exec (''
+RESTORE HEADERONLY
+FROM ' + REPLACE(@bkup_files, '''', '''''') + '
+WITH ' + CASE WHEN @credential NOT LIKE '' THEN 'CREDENTIAL = ''''' + @credential + ''''', ' ELSE '' END + '
+    NOUNLOAD'')
+
+SELECT * FROM @BackupSets
+ORDER BY BackupFinishDate DESC
+'
+
+EXEC sp_executesql @HeaderOnly_cmd

data/sql_scripts/Database/GetFileInfoFromBackup.sql

@@ -0,0 +1,9 @@
+SET NOCOUNT ON
+
+DECLARE @sql nvarchar(max)
+DECLARE @credential nvarchar(max)
+
+SET @credential = '$(credential)'
+SET @sql = 'RESTORE FILELISTONLY FROM $(bkupfiles)' + CASE WHEN @credential NOT LIKE '' THEN ' WITH CREDENTIAL = ''' + @credential + '''' ELSE '' END
+
+EXEC(@sql)

data/sql_scripts/Database/RestoreDatabase.sql

@@ -0,0 +1,185 @@
+SET NOCOUNT ON
+
+DECLARE @db_name nvarchar(max)
+DECLARE @credential nvarchar(max)
+DECLARE @db_state int
+DECLARE @bkup_files nvarchar(max)
+DECLARE @data_file nvarchar(max)
+DECLARE @data_file_logical_name nvarchar(255)
+DECLARE @log_file nvarchar(max)
+DECLARE @log_file_logical_name nvarchar(255)
+DECLARE @database_cmd nvarchar(max)
+DECLARE @setproperties_cmd nvarchar(max)
+DECLARE @prepareolddb_cmd nvarchar(max)
+DECLARE @replcheck_cmd nvarchar(max)
+DECLARE @change_logical_names_cmd nvarchar(max)
+DECLARE @rowcount int
+DECLARE @log_only bit
+DECLARE @recovery bit
+DECLARE @replace bit
+DECLARE @keep_replication bit
+DECLARE @unload bit
+DECLARE @stats nvarchar(3)
+
+DECLARE @GetPosition_cmd nvarchar(max)
+DECLARE @BackupSetPosition int
+DECLARE @Version14Plus bit
+DECLARE @SQLMajorVersion nvarchar(128)
+
+SET @db_name = '$(databasename)'
+SET @credential = '$(credential)'
+SET @bkup_files = '$(bkupfiles)'
+SET @data_file = '$(datafile)'
+SET @data_file_logical_name = '$(datafilelogicalname)'
+SET @log_file = '$(logfile)'
+SET @log_file_logical_name = '$(logfilelogicalname)'
+SET @log_only = '$(logonly)'
+SET @recovery = '$(recovery)'
+SET @replace = '$(replace)'
+SET @keep_replication = '$(keepreplication)'
+SET @unload = '$(unload)'
+SET @stats = '$(stats)'
+
+SET @SQLMajorVersion = CONVERT(NVARCHAR(128), SERVERPROPERTY('ProductVersion'))
+SET @SQLMajorVersion = LEFT(@SQLMajorVersion, CHARINDEX('.', @SQLMajorVersion) - 1)
+SELECT @db_state = [state] FROM sys.databases WHERE [Name] = @db_name
+
+SET @GetPosition_cmd =
+'DECLARE @BackupSets TABLE(
+	BackupName nvarchar(128),
+    BackupDescription nvarchar(255),
+    BackupType smallint,
+    ExpirationDate datetime,
+    Compressed bit,
+    Position smallint,
+    DeviceType tinyint,
+    UserName nvarchar(128),
+    ServerName nvarchar(128),
+    DatabaseName nvarchar(128),
+    DatabaseVersion int,
+    DatabaseCreationDate datetime,
+    BackupSize numeric(20,0),
+    FirstLSN numeric(25,0),
+    LastLSN numeric(25,0),
+    CheckpointLSN numeric(25,0),
+    DatabaseBackupLSN numeric(25,0),
+    BackupStartDate datetime,
+    BackupFinishDate datetime,
+    SortOrder smallint,
+    CodePage smallint,
+    UnicodeLocaleId int,
+    UnicodeComparisonStyle int,
+    CompatibilityLevel tinyint,
+    SoftwareVendorId int,
+    SoftwareVersionMajor int,
+    SoftwareVersionMinor int,
+    SoftwareVersionBuild int,
+    MachineName nvarchar(128),
+    Flags int,
+    BindingID uniqueidentifier,
+    RecoveryForkID uniqueidentifier,
+    Collation nvarchar(128),
+    FamilyGUID uniqueidentifier,
+    HasBulkLoggedData bit,
+    IsSnapshot bit,
+    IsReadOnly bit,
+    IsSingleUser bit,
+    HasBackupChecksums bit,
+    IsDamaged bit,
+    BeginsLogChain bit,
+    HasIncompleteMetaData bit,
+    IsForceOffline bit,
+    IsCopyOnly bit,
+    FirstRecoveryForkID uniqueidentifier,
+    ForkPointLSN numeric(25,0),
+    RecoveryModel nvarchar(60),
+    DifferentialBaseLSN numeric(25,0),
+    DifferentialBaseGUID uniqueidentifier,
+    BackupTypeDescription nvarchar(60),
+    BackupSetGUID uniqueidentifier,
+    CompressedBackupSize bit,'
+    + CASE WHEN (@SQLMajorVersion >= 11) THEN ' Containment tinyint,' ELSE '' END
+	+ CASE WHEN (@SQLMajorVersion >= 12) THEN
+		'KeyAlgorithm nvarchar(32),
+		EncryptorThumbprint varbinary(20),
+		EncryptorType nvarchar(32),' ELSE '' END
+	+ '--
+    -- This field added to retain order by
+    --
+    Seq int NOT NULL identity(1,1)
+)
+
+INSERT INTO @BackupSets
+exec (''
+RESTORE HEADERONLY
+FROM ' + REPLACE(@bkup_files, '''', '''''') + '
+WITH NOUNLOAD' + CASE WHEN @credential NOT LIKE '' THEN ', CREDENTIAL = ''''' + @credential + '''''' ELSE '' END + ''')
+
+SELECT TOP(1) @BackupSetPosition = Position FROM @BackupSets
+ORDER BY BackupFinishDate DESC
+'
+
+SET @replcheck_cmd =
+N'IF EXISTS (SELECT * FROM [' + @db_name + '].[INFORMATION_SCHEMA].[TABLES]
+			WHERE TABLE_NAME = ''syspublications'')
+BEGIN
+	SELECT @rowcount=count(*) FROM [' + @db_name + '].[dbo].syspublications
+END
+ELSE
+BEGIN
+	SET @rowcount = 0
+END
+'
+
+-- Check if replication is enabled. If so, notify user and exit.
+IF EXISTS (SELECT * FROM sys.databases WHERE [Name] = @db_name AND state != 1)
+BEGIN
+	EXEC sp_executesql @replcheck_cmd, N'@rowcount int output', @rowcount output;
+	IF (@rowcount > 0)
+	BEGIN
+		PRINT ('Replication is enabled on ' + @db_name + '. Remove replication before proceeding...')
+		RETURN
+	END
+END
+
+SET @prepareolddb_cmd =
+'
+IF EXISTS (SELECT * FROM sys.databases WHERE Name = ''' + @db_name + ''')
+BEGIN
+	ALTER DATABASE [' + @db_name + '] SET  SINGLE_USER WITH ROLLBACK IMMEDIATE
+	ALTER DATABASE [' + @db_name + '] SET  MULTI_USER WITH ROLLBACK IMMEDIATE
+END
+'
+
+-- Get the position of the most recent backup set.
+EXEC sp_executesql @GetPosition_cmd, N'@BackupSetPosition int output', @BackupSetPosition output;
+IF (@BackupSetPosition < 1)
+BEGIN
+	PRINT ('Failed to get backup set position. Using position 1...')
+	SET @BackupSetPosition = 1
+END
+
+-- Restore database query
+SET @database_cmd = 'RESTORE ' + CASE WHEN @log_only = 1 THEN 'LOG' ELSE 'DATABASE' END + ' [' + @db_name + '] FROM ' + @bkup_files + ' WITH FILE = ' + CONVERT(NVARCHAR(5), @BackupSetPosition) + ', ' +
+                    CASE WHEN @data_file_logical_name != '' THEN 'MOVE N''' + @data_file_logical_name + ''' TO N''' + @data_file + ''', ' ELSE '' END +
+                    CASE WHEN @log_file_logical_name != '' THEN 'MOVE N''' + @log_file_logical_name + ''' TO N''' + @log_file + ''', ' ELSE '' END +
+                    CASE WHEN @replace = 1 THEN 'REPLACE, ' ELSE '' END +
+                    CASE WHEN @recovery = 1 THEN 'RECOVERY' ELSE 'NORECOVERY' END + ', ' +
+                    CASE WHEN @keep_replication = 1 THEN 'KEEP_REPLICATION, ' ELSE '' END +
+                    CASE WHEN @credential NOT LIKE '' THEN 'CREDENTIAL = ''' + @credential + ''', ' ELSE '' END +
+                    CASE WHEN @unload = 1 THEN 'UNLOAD' ELSE 'NOUNLOAD' END + ', STATS = ' + @stats
+
+SET @change_logical_names_cmd = 'ALTER DATABASE [' + @db_name + '] MODIFY FILE ( NAME = ' + @data_file_logical_name + ', NEWNAME = ' + @db_name + '_Data );' +
+                               'ALTER DATABASE [' + @db_name + '] MODIFY FILE ( NAME = ' + @log_file_logical_name + ', NEWNAME = ' + @db_name + '_Log );'
+
+IF @db_state != 1 -- If it's not restoring
+BEGIN
+    EXEC (@prepareolddb_cmd)
+END
+
+EXEC (@database_cmd)
+
+IF @db_state != 1
+BEGIN
+    EXEC (@change_logical_names_cmd)
+END

data/sql_scripts/Database/SetFullRecovery.sql

@@ -0,0 +1,19 @@
+IF (SELECT recovery_model_desc FROM sys.databases WHERE name = '$(databasename)') NOT LIKE 'FULL'
+BEGIN
+	PRINT ('Setting recovery to full...')
+
+	ALTER DATABASE [$(databasename)] SET RECOVERY FULL WITH NO_WAIT
+
+	DECLARE @currentrecoverymodel nvarchar(50)
+	SELECT @currentrecoverymodel = recovery_model_desc FROM sys.databases WHERE name = '$(databasename)'
+	IF (@currentrecoverymodel NOT LIKE 'FULL')
+	BEGIN
+		RAISERROR('Failed to set recovery to full!', 20, 1) WITH LOG
+	END
+	PRINT ('Recovery model set to ' + @currentrecoverymodel)
+	SELECT CAST(1 AS bit) AS RecoveryModelUpdated
+END
+ELSE
+BEGIN
+	SELECT CAST(0 AS bit) AS RecoveryModelUpdated
+END

data/sql_scripts/Database/SetSQLCompatibility.sql

@@ -0,0 +1,33 @@
+SET NOCOUNT ON
+
+DECLARE @dbname nvarchar(128)
+DECLARE @alter_db_command nvarchar(255)
+DECLARE @compatibility_level nvarchar(3)
+DECLARE @current_compatibility_level nvarchar(3)
+DECLARE @out_msg nvarchar(max)
+
+SET @dbname = '$(databasename)'
+SET @compatibility_level = '$(compatibility_level)'
+SET @alter_db_command = 'ALTER DATABASE [' + @dbname + '] SET COMPATIBILITY_LEVEL = ' + @compatibility_level
+
+SELECT @current_compatibility_level = compatibility_level FROM sys.databases WHERE name = @dbname
+
+IF @current_compatibility_level < @compatibility_level
+BEGIN
+	SET @out_msg = 'Updating compatibility level from ' + @current_compatibility_level +  ' to ' + @compatibility_level + '.'
+	PRINT @out_msg
+	exec sp_executesql @alter_db_command
+END
+ELSE IF @current_compatibility_level = @compatibility_level
+BEGIN
+	SET @out_msg = 'Compatibility level already set to ' + @current_compatibility_level + '. No change made.'
+	PRINT @out_msg
+END
+ELSE
+BEGIN
+  SET @out_msg = 'WARNING! Current compatibility level (' + @current_compatibility_level +  ') is higher than ' + @compatibility_level + '! Downgrading compatibility level!'
+	PRINT @out_msg
+	exec sp_executesql @alter_db_command
+END
+
+SELECT compatibility_level FROM sys.databases WHERE name = @dbname

data/sql_scripts/Security/AssignDatabaseRoles.sql

@@ -0,0 +1,44 @@
+USE [$(databasename)]
+
+DECLARE @login_name nvarchar(max)
+DECLARE @db_name nvarchar(max)
+DECLARE @user_cmd nvarchar(max)
+DECLARE @roles nvarchar(max)
+DECLARE @roles_cmd nvarchar(max)
+DECLARE @database_roles TABLE (
+	DatabaseRole nvarchar(max)
+)
+
+SET @login_name = '$(user)'
+SET @db_name = '$(databasename)'
+SET @roles = '$(databaseroles)' -- comma separated list - do not use commas in the values
+
+SET @roles_cmd = '
+DECLARE @database_roles TABLE ( DatabaseRole nvarchar(max) )
+INSERT INTO @database_roles ( DatabaseRole )
+VALUES ( ' + '''' + REPLACE(@roles, ',', '''), (''') + ''')
+SELECT RTRIM(LTRIM(DatabaseRole)) FROM @database_roles'
+INSERT INTO @database_roles
+exec sp_executesql @roles_cmd
+
+-- Add user and assign security query
+SET @user_cmd =
+'
+USE [' + @db_name + ']
+IF  EXISTS (SELECT * FROM sys.schemas WHERE name = N''' + @login_name + ''')
+BEGIN
+	DROP SCHEMA [' + @login_name + ']
+END
+
+IF EXISTS (SELECT * FROM sys.database_principals WHERE name=N''' + @login_name + ''')
+BEGIN
+	DROP USER [' + @login_name + ']
+END
+
+CREATE USER [' + @login_name + '] FOR LOGIN [' + @login_name + '];'
+
+SELECT @user_cmd += 'EXEC sp_addrolemember N''' + DatabaseRole + ''', N''' + @login_name + ''';' FROM @database_roles
+
+EXEC (@user_cmd)
+
+SELECT * FROM sys.database_principals WHERE name=@login_name

data/sql_scripts/Security/CreateOrUpdateCredential.sql

@@ -0,0 +1,11 @@
+IF EXISTS (SELECT * FROM sys.credentials
+           WHERE name = '$(credential_name)')
+BEGIN
+  ALTER CREDENTIAL [$(credential_name)] WITH IDENTITY = '$(identity)', SECRET = '$(secret)';
+  SELECT 'Credential [$(credential_name)] updated.'
+END
+ELSE
+BEGIN
+  CREATE CREDENTIAL [$(credential_name)] WITH IDENTITY = '$(identity)', SECRET = '$(secret)';
+  SELECT 'Credential [$(credential_name)] created.'
+END

data/sql_scripts/Security/CreateSqlLogin.sql

@@ -0,0 +1,20 @@
+DECLARE @login_name nvarchar(max)
+DECLARE @password nvarchar(max)
+DECLARE @user_cmd nvarchar(max)
+
+SET @login_name = '$(user)'
+SET @password = '$(password)'
+
+-- Create login
+SET @user_cmd =
+'USE [master]
+IF EXISTS (SELECT name FROM master.dbo.syslogins WHERE name=''' + @login_name + ''')
+BEGIN
+	DROP LOGIN [' + @login_name + ']
+END
+
+CREATE LOGIN [' + @login_name + '] WITH PASSWORD=N''' + @password + ''', DEFAULT_DATABASE=[master], CHECK_EXPIRATION=OFF, CHECK_POLICY=OFF'
+
+EXEC (@user_cmd)
+
+SELECT name, CONVERT(NVARCHAR(max), sid, 1) FROM master.dbo.syslogins WHERE name=@login_name