LaunchScript.ps1

[CmdletBinding(SupportsShouldProcess)]
param (
	[Parameter(Mandatory)]
	[string[]]$Client,
	
	[Parameter(Mandatory)]
	[ValidateSet('Install', 'Upgrade', 'Uninstall', 'Detect')]
	[string]$Type,
	
	[Parameter(Mandatory)]
	[ValidateScript({ Test-Path -Path $_ -PathType Container })]
	[string]$ServerSideClientDeploymentFolderPath,
	
	[Parameter()]
	[string]$ServerModuleFolderPath = 'C:\Demos\DeploymentShare',
	
	[Parameter()]
	[ValidateScript({ Test-Path -Path $_ -PathType Leaf })]
	[string]$ModuleFilePath = "$ServerModuleFolderPath\PSSoftware.psm1",
	
	[Parameter()]
	[string]$AllowedLaunchServer = 'WINSRV2012R2-1.MYLAB.LOCAL',
	
	[Parameter()]
	[string]$ClientDeploymentFolder = 'C:\MyDeployment'
	
)

$ErrorActionPreference = 'Stop'
try
{
	#region Configuration Values
	
	#endregion
	
	#region Testing the waters
	
	## Test to ensure our shared module exists where we think it does
	if (-not (Test-Path -Path $ModuleFilePath -PathType Leaf))
	{
		throw "PSSoftware module: BAD"
	}
	else
	{
		Write-Verbose -Message "PSSoftware module: GOOD"
	}
	## Test to ensure the deployment script is where we think it is
	if (-not (Test-Path -Path $ServerSideClientDeploymentFolderPath -PathType Container))
	{
		throw "Deployment folder: BAD"
	}
	else
	{
		Write-Verbose -Message "Deployment folder: GOOD"
	}
	if (-not (Test-Path -Path "$ServerSideClientDeploymentFolderPath\$Type.ps1" -PathType Leaf))
	{
		throw "Deployment script: BAD"
	}
	else
	{
		Write-Verbose -Message "Deployment script: GOOD"
	}
	## Test to ensure this script is launching on the "launch server"
	if ("$env:COMPUTERNAME.$env:USERDNSDOMAIN" -ne $AllowedLaunchServer)
	{
		throw "Appropriate launch server: BAD"
	}
	else
	{
		Write-Verbose -Message "Appropriate launch server: GOOD"
	}
	#endregion
	
	#region Inserting the Import-Module and Start-Log lines into the top of the deployment script
	
	$DeploymentScriptText = Get-Content -Path "$ServerSideClientDeploymentFolderPath\$Type.ps1"
	if ((($DeploymentScriptText | Select-Object -First 2) -join '|') -ne "Import-Module $ClientDeploymentFolder\PSSoftware.psm1|Start-Log")
	{
		## The deployment script doesn't have the required Import-Module as the first line and Start-Log as the second line
		## Insert Import-Module as the first line and Start-Log as the second line
		
		## If there's already some other Import-Module line, delete it
		if (($DeploymentScriptText | Select-Object -First 1) -like 'Import-Module *')
		{
			## Remove the top Import-Module line
			$DeploymentScriptText = ($DeploymentScriptText | Select-Object -Skip 1)
		}
		## If some other Start-Log line then delete it also
		if (($DeploymentScriptText | Select-Object -First 1) -like 'Start-Log*')
		{
			## Remove the top Start-Log line
			$DeploymentScriptText = ($DeploymentScriptText | Select-Object -Skip 1)
		}
		## If some other working dir line then delete it also
		if (($DeploymentScriptText | Select-Object -First 1) -like '$WorkingDir*')
		{
			## Remove the top Start-Log line
			$DeploymentScriptText = ($DeploymentScriptText | Select-Object -Skip 1)
		}
		
		## Insert the new Import-Module line and the Start-Log line at the top of the deployment script
		$DeploymentScriptText = "Import-Module $ClientDeploymentFolder\PSSoftware.psm1", 'Start-Log', '$WorkingDir = $MyInvocation.MyCommand.Path | Split-Path -Parent', $DeploymentScriptText
		
		Set-Content -Path "$ServerSideClientDeploymentFolderPath\$Type.ps1" –value $DeploymentScriptText
	}
	
	#endregion
	
	#region Removing existing jobs
	Write-Verbose -Message 'Cleaning up any pre-existing jobs'
	Get-Job | Remove-Job -Force
	#endregion
	
	#region Testing and deployment
	
	## We will be creating a folder in the root of the C drive called MyDeployment. We'll then copy
	## all files including the deployment script and all installer files into that folder.
	## Once copied, we'll then use PS remoting to execute the script
	## We need to make sure all the services that require this are available; ping, c$ share 
	## available and PS remoting is working
	foreach ($pc in $Client)
	{
		try
		{
			if (-not (Test-Connection -ComputerName $pc -Quiet -Count 1))
			{
				throw "$pc`: OFFLINE"
				continue
			}
			else
			{
				Write-Verbose -Message "$pc`: ONLINE"
				## Convert the mydeployment local path to a UNC so it can be tested and created
				$RemoteFilePathDrive = ($ClientDeploymentFolder | Split-Path -Qualifier).TrimEnd(':')
				$RemoteServerSideClientServerSideClientDeploymentFolderPath = "\\$pc\$RemoteFilePathDrive`$$($ClientDeploymentFolder | Split-Path -NoQualifier)"
				if (-not (Test-Path -Path "\\$pc\c$"))
				{
					throw "$pc`: C`$ share is NOT available"
					continue
				}
				else
				{
					Write-Verbose -Message "$pc`: C`$ share is available"
					if (-not (Test-WsMan -ComputerName $pc))
					{
						throw "$pc`: PS Remoting Problem"
						continue
					}
					else
					{
						Write-Verbose -Message "$pc`: PS Remoting OK"
						if ($PsCmdlet.ShouldProcess($pc, 'Invoke deployment on client'))
						{
							if (-not (Test-Path -Path $RemoteServerSideClientServerSideClientDeploymentFolderPath -PathType Container))
							{
								Write-Verbose -Message "The deployment folder $ClientDeploymentFolder does not exist on client $pc. Creating it."
								$null = mkdir $RemoteServerSideClientServerSideClientDeploymentFolderPath
							}
							Write-Verbose -Message "$pc`: Copying deployment scripts to client"
							Copy-Item -Path $ServerSideClientDeploymentFolderPath\* -Destination $RemoteServerSideClientServerSideClientDeploymentFolderPath -Force
							Write-Verbose -Message "$pc`: Copying module folder to client"
							Copy-Item -Path $ServerModuleFolderPath\* -Destination $RemoteServerSideClientServerSideClientDeploymentFolderPath -Force -Recurse
							Write-Verbose -Message "$pc`: Launching deployment script"
							$null = Invoke-Command -ComputerName $pc -ScriptBlock { & "$using:ClientDeploymentFolder\$using:Type.ps1" } -AsJob -ThrottleLimit 5
							Write-Verbose -Message "$pc`: Deployment script launched"
						}
					}
				}
			}
		}
		catch
		{
			Write-Warning "$pc - $($_.Exception.Message)"
		}
	}
	
	#endregion
	
	#region Deployment monitoring
	$Timeout = 240
	$RunningTime = 0
	if ($PsCmdlet.ShouldProcess('Running deployments', 'Check status'))
	{
		do
		{
			$JobsLeft = (Get-Job | Where-Object { $_.State -in @('Running', 'NotStarted') }).Count
			if ($JobsLeft -gt 1)
			{
				Write-Verbose -Message "Waiting for all the deployments to finish up. There are $JobsLeft left."
			}
			else
			{
				Write-Verbose -Message 'Waiting for all the deployments to finish up. There is 1 left.'
			}
			Start-Sleep -Seconds 5
			$RunningTime += 5
			if ($RunningTime -eq $Timeout)
			{
				Write-Warning -Message "Timeout of $Timeout seconds has been exceeded waiting for jobs"
				break
			}
		}
		while (Get-Job | Where-Object { $_.State -in @('Running', 'NotStarted') })
	}
	#endregion
	
	#region Deployment result and cleanup
	
	Get-Job | Receive-Job
	
	Write-Verbose 'Cleaning up deployment folders on clients'
	foreach ($pc in $Client)
	{
		Remove-Item -Path "\\$pc\c$\$($ClientDeploymentFolder | Split-Path -Leaf)" -Recurse -ErrorAction SilentlyContinue
	}
	
	#endregion
	
}
catch
{
	Write-Warning $_.Exception.Message
	
}