220 lines
8.8 KiB
PowerShell
220 lines
8.8 KiB
PowerShell
## Define the MSP name
|
|
$MSPName = "Emberkom"
|
|
|
|
## Setup the MSP management directories
|
|
If ( Test-Path ${Env:ProgramData} )
|
|
{ $MSPRoot = "${Env:ProgramData}\${MSPName}" } Else { $MSPRoot = "${Env:SystemDrive}\${MSPName}" }
|
|
$MSPScripts = "${MSPRoot}\Scripts"
|
|
$MSPTools = "${MSPRoot}\Tools"
|
|
$MSPLogs = "${MSPRoot}\Logs"
|
|
|
|
## Make sure all the management directories exist
|
|
$MSPRoot, $MSPScripts, $MSPTools, $MSPLogs | ForEach-Object {
|
|
If ( !(Test-Path $_) )
|
|
{
|
|
Try { New-Item -Path $MSPRoot -ItemType Directory -Force -ErrorAction SilentlyContinue | Out-Null }
|
|
Catch { Write-Output $_.Exception.Message ; Exit }
|
|
}
|
|
}
|
|
|
|
## Set a file prefix for any logs that might need to be created
|
|
$LogFilePrefix = Get-Date -Format "yyyyMMddHHmmss"
|
|
|
|
## Function to log activity to the configured log directory
|
|
Function Log
|
|
{
|
|
param(
|
|
[Parameter(Mandatory=$true,ValueFromPipeline=$true)][string]$Message,
|
|
[Parameter(Mandatory=$false)][switch]$Error=$false,
|
|
[Parameter(Mandatory=$false)][string]$Name
|
|
)
|
|
$Timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
|
|
If ( $Error ) { $LogEntry = "${Timestamp} - Error: ${Message}" }
|
|
Else { $LogEntry = "${Timestamp} - ${Message}" }
|
|
If ( $Name )
|
|
{
|
|
$LogFile = "${MSPLogs}\${LogFilePrefix}-${Name}.log"
|
|
If ( !(Test-Path $LogFile) ) { New-Item -Path $LogFile -ItemType File -Force | Out-Null }
|
|
Add-Content -Path $LogFile -Value $LogEntry
|
|
Write-Output $LogEntry
|
|
}
|
|
Else { Write-Output $LogEntry }
|
|
}
|
|
|
|
## Function to determine if the current user context is an Administrator
|
|
Function IsAdmin
|
|
{
|
|
If ( ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator) )
|
|
{ Return $true } Else { Return $false }
|
|
}
|
|
|
|
## Run a script from the live repo or from a local path
|
|
Function Run-Script
|
|
{
|
|
param(
|
|
## Parameters for live-ps scripts
|
|
[Parameter(Mandatory=$false,ParameterSetName='live-ps')]
|
|
[string]$Type='management-scripts',
|
|
[Parameter(Mandatory=$true,ParameterSetName='live-ps')]
|
|
[string]$LivePSScript,
|
|
## Parameters for local scripts
|
|
[Parameter(Mandatory=$true,ParameterSetName='local')]
|
|
[string]$Path,
|
|
#### $NoWait will prevent waiting for the specified script to finish running
|
|
[Parameter(Mandatory=$false,ParameterSetName='local')]
|
|
[switch]$NoWait=$false,
|
|
## Parameters for all scripts
|
|
[Parameter(Mandatory=$false,ParameterSetName='live-ps')]
|
|
[Parameter(Mandatory=$false,ParameterSetName='local')]
|
|
[string]$Arguments='null',
|
|
#### $HaltIfRunning is a collection of app names which, if running, will prevent the script from running
|
|
[Parameter(Mandatory=$false,ParameterSetName='live-ps')]
|
|
[Parameter(Mandatory=$false,ParameterSetName='local')]
|
|
[string[]]$HaltIfRunning='null',
|
|
#### $ForceClose will forcefully close all apps specified in $HaltIfRunning
|
|
[Parameter(Mandatory=$false,ParameterSetName='live-ps')]
|
|
[Parameter(Mandatory=$false,ParameterSetName='local')]
|
|
[switch]$ForceClose=$false
|
|
)
|
|
$LogName = $MyInvocation.MyCommand
|
|
If ( $Arguments -eq "null" ) { $Arguments = $null }
|
|
If ( $HaltIfRunning -eq "null" ) { $HaltIfRunning = $null }
|
|
If ( $LivePSScript )
|
|
{
|
|
$URL = "https://dev.emberkom.com/emberkom/${Type}/raw/branch/master/${LivePSScript}.ps1"
|
|
If ( !(IsURLValid -URL $URL) ) { Log "The specified script does not exist: ${LivePSScript}" -Name $LogName ; Exit }
|
|
}
|
|
If ( $Path ) { If ( !(Test-Path $Path) ) { Log "The specified script does not exist: ${Path}" -Name $LogName ; Exit } }
|
|
If ( ($HaltIfRunning -ne $null ) -and ($ForceClose) -and ( !(IsAdmin) ) ) { Log "Cannot close dependent apps because this task does not have administrative privileges" -Name $LogName ; Exit }
|
|
If ( $HaltIfRunning )
|
|
{
|
|
$Halt = $false
|
|
Log "Checking for running instances of the following dependent apps:" -Name $LogName
|
|
ForEach ( $name in $HaltIfRunning)
|
|
{
|
|
$RunningInstances = Get-Process | Where { $_.Name -like "${name}" }
|
|
If ( $RunningInstances )
|
|
{
|
|
If ( $ForceClose -eq $true )
|
|
{
|
|
Try { $name | Stop-Process -Force }
|
|
Catch { $_.Exception.Message | Log -Name $LogName ; Exit }
|
|
Log " ""${name}"" (force closed)" -Name $LogName
|
|
}
|
|
Else { $Halt = $true ; Log " ""${name}"" (running)" -Name $LogName }
|
|
}
|
|
Else { Log " ""${name}"" (not running)" -Name $LogName }
|
|
}
|
|
If ( $Halt -eq $true ) { Log "Dependent apps are currently in use and are preventing the specified script from running" -Name $LogName ; Exit }
|
|
}
|
|
switch ($PSCmdlet.ParameterSetName)
|
|
{
|
|
'live-ps'
|
|
{
|
|
Log "Retrieving : ${LivePSScript}.ps1" -Name $LogName
|
|
Try { $Script = (New-Object Net.WebClient).DownloadString($URL) }
|
|
Catch { $_.Exception.Message | Log -Name $LogName ; Exit }
|
|
Try { $ScriptBlock = [Scriptblock]::Create($Script) }
|
|
Catch { $_.Exception.Message | Log -Name $LogName ; Exit }
|
|
Log "Executing: ${LivePSScript}.ps1 ${Arguments}"
|
|
Try { Invoke-Command -ScriptBlock $ScriptBlock -ArgumentList $Arguments }
|
|
Catch { $_.Exception.Message | Log -Name $LogName ; Exit }
|
|
}
|
|
|
|
'local'
|
|
{
|
|
Log "Executing: ${Path} ${Arguments}" -Name $LogName
|
|
Try
|
|
{
|
|
If ( $NoWait ) { Start-Process "${Path}" -ArgumentList "{$Arguments}" }
|
|
Else
|
|
{
|
|
Log " Waiting for script to finish..." -Name $LogName
|
|
Start-Process "${Path}" -ArgumentList "{$Arguments}" -Wait
|
|
Log " Finished" -Name $LogName
|
|
}
|
|
}
|
|
Catch { $_.Exception.Message | Log -Name $LogName ; Exit }
|
|
}
|
|
}
|
|
}
|
|
|
|
## Function to return a Powershell version object from a string
|
|
Function Get-Version
|
|
{
|
|
param([Parameter(Mandatory=$true)][string]$Version)
|
|
[int]$Sets = 4
|
|
$VersionArray = $Version.Split('.')
|
|
If ( $VersionArray.Count -le $Sets ) { $Sets = $VersionArray.Count }
|
|
For ($i = 0; $i -le ($Sets - 1); $i++)
|
|
{ $Return = '{0}.{1}' -f $Return, $VersionArray[$i] }
|
|
Return [version]$Return.Trim('.')
|
|
}
|
|
|
|
## Determines whether a URL is valid
|
|
Function IsURLValid
|
|
{
|
|
param([Parameter(Mandatory=$true,ValueFromPipeline=$true)][string]$URL)
|
|
$HTTPRequest = [System.Net.WebRequest]::Create($URL)
|
|
$HTTPResponse = $HTTPRequest.GetResponse()
|
|
$HTTPStatus = [int]$HTTPResponse.StatusCode
|
|
$HTTPResponse.Close()
|
|
If ($HTTPStatus -eq 200) { Return $true }
|
|
Else { Return $false }
|
|
}
|
|
|
|
## Function to compare current Windows version with a supplied version
|
|
## The value supplied must be a string and must include at least 1 decimal
|
|
Function IsWindowsVersion
|
|
{
|
|
param(
|
|
[Parameter(Mandatory=$true,ParameterSetName='gt')][string]$gt,
|
|
[Parameter(Mandatory=$true,ParameterSetName='ge')][string]$ge,
|
|
[Parameter(Mandatory=$true,ParameterSetName='lt')][string]$lt,
|
|
[Parameter(Mandatory=$true,ParameterSetName='le')][string]$le,
|
|
[Parameter(Mandatory=$true,ParameterSetName='eq')][string]$eq
|
|
)
|
|
[version]$WindowsVersion = [System.Environment]::OSVersion.Version
|
|
switch ($PSCmdlet.ParameterSetName)
|
|
{
|
|
'gt' { If ( $WindowsVersion -gt (Get-Version $gt) ) { Return $true } Else { Return $false } }
|
|
'ge' { If ( $WindowsVersion -ge (Get-Version $ge) ) { Return $true } Else { Return $false } }
|
|
'lt' { If ( $WindowsVersion -lt (Get-Version $lt) ) { Return $true } Else { Return $false } }
|
|
'le' { If ( $WindowsVersion -le (Get-Version $le) ) { Return $true } Else { Return $false } }
|
|
'eq' { If ( $WindowsVersion -eq (Get-Version $eq) ) { Return $true } Else { Return $false } }
|
|
}
|
|
}
|
|
|
|
## Function to compare current Windows build with a supplied version
|
|
## The value supplied must be an int
|
|
Function IsWindowsBuild
|
|
{
|
|
param(
|
|
[Parameter(Mandatory=$true,ParameterSetName='gt')][int]$gt,
|
|
[Parameter(Mandatory=$true,ParameterSetName='ge')][int]$ge,
|
|
[Parameter(Mandatory=$true,ParameterSetName='lt')][int]$lt,
|
|
[Parameter(Mandatory=$true,ParameterSetName='le')][int]$le,
|
|
[Parameter(Mandatory=$true,ParameterSetName='eq')][int]$eq
|
|
)
|
|
[int]$WindowsBuild = [System.Environment]::OSVersion.Version.Build
|
|
switch ($PSCmdlet.ParameterSetName)
|
|
{
|
|
'gt' { If ( $WindowsBuild -gt $gt ) { Return $true } Else { Return $false } }
|
|
'ge' { If ( $WindowsBuild -ge $ge ) { Return $true } Else { Return $false } }
|
|
'lt' { If ( $WindowsBuild -lt $lt ) { Return $true } Else { Return $false } }
|
|
'le' { If ( $WindowsBuild -le $le ) { Return $true } Else { Return $false } }
|
|
'eq' { If ( $WindowsBuild -eq $eq ) { Return $true } Else { Return $false } }
|
|
}
|
|
}
|
|
|
|
## Function to determine if the system is 64-bit or not
|
|
Function Is64bit
|
|
{
|
|
switch ( $(Get-CimInstance -ClassName Win32_OperatingSystem -Property OSArchitecture | Select-Object -ExpandProperty OSArchitecture) )
|
|
{ '64-bit' { Return $true } '32-bit' { Return $false } }
|
|
}
|
|
|
|
## Function to determine if the Powershell process is 64-bit or not
|
|
Function Is64bitShell
|
|
{ If ( [Environment]::Is64BitProcess ) { Return $true } Else { Return $false } }
|