baseModules/PPoShSqlTools/1.0.2/Public/Start-RestoreSqlDatabase.ps1

function Start-RestoreSqlDatabase {
    <#
    .SYNOPSIS
        Restores database on MSSQL Server.

    .DESCRIPTION
        Uses Start-RestoreSqlDatabase.sql sql script to restore database.

    .EXAMPLE
        Start-RestoreSqlDatabase -DatabaseName "DbName" -ConnectionString "data source=localhost;integrated security=True" -Path "C:\database.bak"
    #>
 

    [CmdletBinding()]
    [OutputType([void])]
    param(
        [Parameter(Mandatory=$true)]
        [string]
        $ConnectionString,

        # Database name - if not specified, Initial Catalog from ConnectionString will be used.
        [Parameter(Mandatory=$false)]
        [string]
        $DatabaseName, 
        
        # Backup file path.
        [Parameter(Mandatory=$true)]
        [string]
        $Path,

        # Remote share credential to use if $Path is an UNC path. Note the file will be copied to localhost if this set, and this will work only if
        # you're connecting to local database.
        [Parameter(Mandatory=$false)]
        [PSCredential] 
        $RemoteShareCredential,

        # Timeout for executing sql restore command.
        [Parameter(Mandatory=$false)] 
        [int]
        $QueryTimeoutInSeconds = 3600
    )

    try { 
        if (!$DatabaseName) { 
            $csb = New-Object -TypeName System.Data.SqlClient.SqlConnectionStringBuilder -ArgumentList $ConnectionString
            $DatabaseName = $csb.InitialCatalog
        }
        if (!$DatabaseName) {
            throw "No database name - please specify -DatabaseName or add Initial Catalog to ConnectionString."
        }

        if ($RemoteShareCredential) {
            $shareDir = Split-Path -Path $Path -Parent
            $username = $RemoteShareCredential.UserName
            $password = $RemoteShareCredential.GetNetworkCredential().Password
            Write-Log -Info "Connecting to $Path using username $username."
            & net use $shareDir $password /user:$username
            $tempDir = Join-Path -Path $env:TEMP -ChildPath 'PPoshSqlToolsTemp'
            if (Test-Path -LiteralPath $tempDir) {
              Write-Log -Info "Deleting temp directory '$tempDir'."
              [void](Remove-Item -LiteralPath $tempDir -Force -Recurse)
            }
            [void](New-Item -Path $tempDir -ItemType Directory -Force)
            Write-Log -Info "Copying '$Path' to '$tempDir'"
            Copy-Item -Path $Path -Destination $tempDir -Force
            #TODO: unhardcode this user
            Set-SimpleAcl -Path $tempDir -User 'NT Service\MSSQLSERVER' -Permission 'Read' -Type 'Allow'
            $Path = Join-Path -Path $tempDir -ChildPath (Split-Path -Path $Path -Leaf)
        }

        $sqlScript = Join-Path -Path $PSScriptRoot -ChildPath "Start-RestoreSqlDatabase.sql"
        $parameters =  @{ "DatabaseName" = $DatabaseName }
        $parameters += @{ "Path" = $Path }
        [void](Invoke-Sql -ConnectionString $ConnectionString -InputFile $sqlScript -SqlCmdVariables $parameters -QueryTimeoutInSeconds $QueryTimeoutInSeconds -DatabaseName '')
    } finally {
        if ($RemoteShareCredential) {
            & net use $shareDir /DELETE           
            Remove-TempDirectory
        }
    }
}