install.ps1

[CmdletBinding()]
param (
	[string]$Path,
	[switch]$Beta
)

try {
	Update-Module dbatools -Erroraction Stop
	Write-Output "Updated using the PowerShell Gallery"
	return
}
catch {
	Write-Output "dbatools was not installed by the PowerShell Gallery, continuing with web install."
}

$module = Import-Module -Name dbatools -ErrorAction SilentlyContinue -Force
$localpath = $module.ModuleBase

if ($null -eq $localpath) {
	$localpath = "$HOME\Documents\WindowsPowerShell\Modules\dbatools"
}
else {
	Write-Output "Updating current install"
}

try {
	if (-not $path) {
		if ($PSCommandPath.Length -gt 0) {
			$path = Split-Path $PSCommandPath
			if ($path -match "github") {
				Write-Output "Looks like this installer is run from your GitHub Repo, defaulting to psmodulepath"
				$path = $localpath
			}
		}
		else {
			$path = $localpath
		}
	}
}
catch {
	$path = $localpath
}

if (-not $path -or (Test-Path -Path "$path\.git")) {
	$path = $localpath
}

Write-Output "Installing module to $path"

Remove-Module -Name dbatools -ErrorAction SilentlyContinue

if ($beta) {
	$url = 'https://dbatools.io/devzip'
	$branch = "development"
}
else {
	$url = 'https://dbatools.io/zip'
	$branch = "master"
}

$temp = ([System.IO.Path]::GetTempPath()).TrimEnd("\")
$zipfile = "$temp\dbatools.zip"

if (!(Test-Path -Path $path)) {
	try {
		Write-Output "Creating directory: $path"
		New-Item -Path $path -ItemType Directory | Out-Null
	}
	catch {
		throw "Can't create $Path. You may need to Run as Administrator"
	}
}
else {
	try {
		Write-Output "Deleting previously installed module"
		Remove-Item -Path "$path\*" -Force -Recurse
	}
	catch {
		throw "Can't delete $Path. You may need to Run as Administrator"
	}
}

Write-Output "Downloading archive from github"
try {
	Invoke-WebRequest $url -OutFile $zipfile
}
catch {
	#try with default proxy and usersettings
	Write-Output "Probably using a proxy for internet access, trying default proxy settings"
	(New-Object System.Net.WebClient).Proxy.Credentials = [System.Net.CredentialCache]::DefaultNetworkCredentials
	Invoke-WebRequest $url -OutFile $zipfile
}

# Unblock if there's a block
Unblock-File $zipfile -ErrorAction SilentlyContinue

Write-Output "Unzipping"

# Keep it backwards compatible
$shell = New-Object -ComObject Shell.Application
$zipPackage = $shell.NameSpace($zipfile)
$destinationFolder = $shell.NameSpace($temp)
$destinationFolder.CopyHere($zipPackage.Items())

Write-Output "Cleaning up"
Move-Item -Path "$temp\dbatools-$branch\*" $path
Remove-Item -Path "$temp\dbatools-$branch"
Remove-Item -Path $zipfile

Write-Output "Done! Please report any bugs to dbatools.io/issues or clemaire@gmail.com."
if ((Get-Command -Module dbatools).count -eq 0) { Import-Module "$path\dbatools.psd1" -Force }
Get-Command -Module dbatools
Write-Output "`n`nIf you experience any function missing errors after update, please restart PowerShell or reload your profile."