CT-PS-Standard.psm1

<#
--------------------------------------------------------------------------------------------------
 
This is a standard module with a set of standard functions used across multiple scripts within CT.
 
Any changes to this module need to be published using the powershell script "Build_CT_Module.ps1"
 
--------------------------------------------------------------------------------------------------
HOW TO IMPORT INTO SCRIPT:
--------------------------------------------------------------------------------------------------
This module should be imported using the commands below (do not copy the asterix's, just whats between).
This will import this module AND initialise the script with all the standard features required by scripts,
including all the log files for each transaction
*****************
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
if ($null -eq (Get-PSRepository)) {
    Register-PSRepository -Default -Verbose:$VerbosePreference
}
Install-PackageProvider -Name NuGet -MinimumVersion 2.8.5.201 -Force -ErrorAction SilentlyContinue | Out-Null
Remove-Module -Name CT-PS-Standard -Force -Verbose:$VerbosePreference -ErrorAction SilentlyContinue
Uninstall-Module -Name "CT-PS-Standard" -AllVersions -Force -ErrorAction SilentlyContinue -Verbose:$VerbosePreference
Install-Module -Name CT-PS-Standard -Force -AllowClobber -Verbose:$VerbosePreference
Import-Module -Name CT-PS-Standard -Force -Verbose:$VerbosePreference
*****************
 
--------------------------------------------------------------------------------------------------
 
 
--------------------------------------------------------------------------------------------------
LOG FILES
--------------------------------------------------------------------------------------------------
There are four log files initialised by this module that can be used for output.
You can write to each of these logs accordingly.
 
$Output_Log: This is the standard console output.
$Transcript_log: This is where the transcript is written to. You will need to start and stop the transcript inside your script by using the command "Start-Transcript -Path $Transcript_log -append | Out-Null"
$API_log: This is where the output from API posts should be sent.
$Install_log: This is where output from MSIEXEC commands should be logged
 
--------------------------------------------------------------------------------------------------
 
 
--------------------------------------------------------------------------------------------------
The following commands will run the init script when the module is imported
#>


    #Begin {
    # Display some troubleshoot info
    [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
    Write-verbose "MyInvocation info:
     
    MyCommand: $($MyInvocation.MyCommand)
    ScriptName: $($MyInvocation.ScriptName)
    PSScriptRoot: $($MyInvocation.PSScriptRoot)
    PSCommandPath: $($MyInvocation.PSCommandPath)
    InvocationName: $($MyInvocation.InvocationName)
    ScriptLineNumber: $($MyInvocation.ScriptLineNumber)
    "


    Write-verbose "MyInvocation info:
     
    ScriptName: $($ScriptName)
    PSScriptRoot: $($PSScriptRoot)
    PSCommandPath: $($PSCommandPath)
    "


    <#
    $P_MyInvocation = $PSCmdlet.SessionState.PSVariable.Get('MyInvocation').Value
 
    Write-Verbose "Parent MyInvocation info:
     
    MyCommand: $($P_MyInvocation.MyCommand)
    ScriptName: $($P_MyInvocation.ScriptName)
    PSScriptRoot: $($P_MyInvocation.PSScriptRoot)
    PSCommandPath: $($P_MyInvocation.PSCommandPath)
    InvocationName: $($P_MyInvocation.InvocationName)
    ScriptLineNumber: $($P_MyInvocation.ScriptLineNumber)
    "
    #>

    $global:ErrorActionPreference = "Stop"

    $global:Script_Path = $MyInvocation.PSScriptRoot

    if ($MyInvocation.ScriptName.Length -gt 4) {
        try {
            $Global:Script_Name = ($MyInvocation.ScriptName).Replace("$($MyInvocation.PSScriptRoot)\","")
            $Global:Script_Name = ($Global:Script_Name).Substring(0,($Global:Script_Name).Length-4)
        } catch {
            $Global:Script_Name = "Terminal"
        }
    } else {
        $Global:Script_Name = "Terminal"
    }

    $global:CT_DEST="C:\CT"  # Where the files are downloaded to
    $global:DateStamp = get-date -Format yyyyMMddTHHmmss # A formatted date strong

    $global:Output_log = "$CT_DEST\logs\$($Script_Name)\$($DateStamp)_output.log"  # The output
    $global:Transcript_log = "$CT_DEST\logs\$($Script_Name)\$($DateStamp)_transcript.log"  # The powershell transcript file
    $global:API_log = "$($CT_DEST)\logs\$($Script_Name)\$($DateStamp)_API.log"
    $global:Install_log = "$CT_DEST\logs\$($Script_Name)\$($DateStamp)_install.log"  # The powershell installation file

    #} #End begin block

    
    #Process {

        
    # ComputerType will report if the machine is a workstation, DC, or non-DC server
    # 1 for workstations, 2 for DCs, and 3 for non-DC servers
    try {
        $ComputerType = (Get-CimInstance -ClassName Win32_OperatingSystem -Debug:$DebugPreference).ProductType
    } catch {
        write-host "There is a problem with this computer and updates are required for this script to continue."
        #Stop-Transcript | Out-Null
        $exiterror = 16
        throw "There is a problem with this computer and updates are required for this script to continue."
        exit 16
    }



    try{

        # Check for a CT folder on the C: and if not, create it, however that location should already exist as part of the Start-Transcript command.
        if(-not( Test-Path -Path $CT_DEST )) {
            try{
                mkdir $CT_DEST > $null
                #Transcript-Log "New folder created at $CT_DEST."
            }catch{
                #Can't create the folder, therefore cannot continue
                Write-Error "Cannot create folder $CT_DEST. $($Error[0].Exception.Message)" -ErrorAction Continue
                Write-Error $_ -verbose:$VerbosePreference -ErrorAction Continue
                #Stop-Transcript
                exit 1
            }
        }

        if(-not( Test-Path -Path "$($CT_DEST)\logs" )) {
            try{
                mkdir "$($CT_DEST)\logs" > $null
                #Transcript-Log "New logs folder created at $CT_DEST."
            }catch{
                #Can't create the folder, therefore cannot continue
                Write-Error "Cannot create logs folder in $CT_DEST. $($Error[0].Exception.Message)" -ErrorAction Continue
                Write-Error $_ -verbose:$VerbosePreference -ErrorAction Continue
                #Stop-Transcript
                exit 3
            }
        }

        if(-not( Test-Path -Path "$($CT_DEST)\logs\$($Script_Name)" )) {
            try{
                mkdir "$($CT_DEST)\logs\$($Script_Name)" > $null
                #Transcript-Log "New logs folder for $($Script_Name) created at $CT_DEST."
            }catch{
                #Can't create the folder, therefore cannot continue
                Write-Error "Cannot create logs folder for $($Script_Name) in $CT_DEST. $($Error[0].Exception.Message)" -ErrorAction Continue
                Write-Error $_ -ErrorAction Continue
                #Stop-Transcript
                exit 4
            }
        }

        $global:CT_Reg_Path = "HKLM:\Software\CT\Monitoring"
        $global:CT_Reg_Key = "$($CT_Reg_Path)\$($Script_Name)"
        if(-not( Test-Path -Path $CT_Reg_Key )) {
            try{
                $CTMonitoringReg = New-Item -Path $CT_Reg_Path -Name $Script_Name -Force
                Set-ItemProperty -Path "HKLM:\Software\CT" -Name "CustomerNo" -Value $customer
            }catch{
                #Can't create the regkey, therefore cannot continue
                Write-Error "Cannot create registry key at $($CT_Reg_Key). $($Error[0].Exception.Message)" -ErrorAction Continue
                Write-Error "$($CTMonitoringReg)" -ErrorAction Continue
                Write-Error $_
                #Stop-Transcript
                exit 5
            }
        }


        #Setup TLS 1.1 and 1.2
        $Name         = 'DisabledByDefault'
        $Value        = '0'

        if(-not( Test-Path -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.1" )) {
            try{
                New-Item -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols" -Name "TLS 1.1" -Force | Out-Null
                New-Item -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.1" -Name "Client" -Force | Out-Null
                New-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.1\Client" -Name $Name -Value $Value -PropertyType DWORD -Force | Out-Null
            }catch{
                #Can't create the regkey, therefore cannot continue
                Write-Error "Cannot set TLS 1.1." -ErrorAction Continue
                Write-Error $_
            }
        } else {
            Set-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.1\Client" -Name $Name -Value $Value | Out-Null
        }

        if(-not( Test-Path -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.2" )) {
            try{
                New-Item -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols" -Name "TLS 1.2" -Force | Out-Null
                New-Item -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.2" -Name "Client" -Force | Out-Null
                New-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.2\Client" -Name $Name -Value $Value -PropertyType DWORD -Force | Out-Null
            }catch{
                #Can't create the regkey, therefore cannot continue
                Write-Error "Cannot set TLS 1.2." -ErrorAction Continue
                Write-Error $_
            }
        } else {
            Set-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\SecurityProviders\SCHANNEL\Protocols\TLS 1.1\Client" -Name $Name -Value $Value | Out-Null
        }


        # ComputerType will report if the machine is a workstation, DC, or non-DC server
        # 1 for workstations, 2 for DCs, and 3 for non-DC servers
        try {
            $global:ComputerType = (Get-CimInstance -ClassName Win32_OperatingSystem -Debug:$DebugPreference).ProductType
        } catch {
            Write-Error "There is a problem with this computer and updates are required for this script to continue." -ErrorAction Continue
            Throw $_
        }
        
        

        # Create Transcript header
        Write-Verbose "**********************"
        Write-Verbose "Script: $($Script_Name)."
        Write-Verbose "Start time: $($DateStamp)"
        Write-Verbose "Username: $($env:USERDOMAIN)\$($env:USERNAME)"
        Write-Verbose "Execution Policy Preference: $($env:PSExecutionPolicyPreference)"
        Write-Verbose "Machine: $($env:COMPUTERNAME) ($($env:OS))"
        Write-Verbose "Process ID: $($PID)"
        Write-Verbose "PSVersion: $($PSVersionTable.PSVersion)"
        Write-Verbose "PSEdition: $($PSVersionTable.PSEdition)"
        Write-Verbose "Operating System: $($PSVersionTable.OS)"
        Write-Verbose "WSManStackVersion: $($PSVersionTable.WSManStackVersion)"
        Write-Verbose "PSRemotingProtocolVersion: $($PSVersionTable.PSRemotingProtocolVersion)"
        Write-Verbose "SerializationVersion: $($PSVersionTable.SerializationVersion)"
        Write-Verbose "**********************"


        <#
        ---- END STANDARD SCRIPT BLOCK----
        #>

    } catch {
        Throw $_
    }

    #} #End process block


<#
This ends the script block that run on module import
--------------------------------------------------------------------------------------------------
 
 
#>



<#
Initialize-Script is the old command for when the module is imported, its kept in here for backwards compatibility but it now only calls the import-module command
#>

function Initialize-Script {
    [CmdletBinding()]
    param()

    Import-Module CT-PS-Standard -Force
}




Function Get-UserInput {
    [CmdletBinding()]
    Param(
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory=$True)]
        $message,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory=$True)]
        $title,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        $defaultvalue
    )
    Process{
        [void][Reflection.Assembly]::LoadWithPartialName('Microsoft.VisualBasic')
        Add-Type -AssemblyName Microsoft.VisualBasic

        return [Microsoft.VisualBasic.Interaction]::InputBox($message, $title, $defaultvalue)
    }
}



Function Write-OutputLog {
    [CmdletBinding()]
    Param(
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        $output
    )
    Process{
        if ($PSCmdlet.MyInvocation.BoundParameters['Verbose'] -eq $true) {
            Write-Verbose "(Line $($MyInvocation.ScriptLineNumber)) $($output)"
        }
        $output | Out-File -FilePath "$($Output_log)" -Append
        Write-Host $output
    }
}


# Writes to the API log and optionally console if -Verbose flag is set at script level
Function Write-APILog {
    [CmdletBinding()]
    Param(
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        $output
    )
    Process{
        if ($PSCmdlet.MyInvocation.BoundParameters['Verbose'] -eq $true) {
            Write-Verbose "(Line $($MyInvocation.ScriptLineNumber)) $($output)"
        }
        $output | Out-File -FilePath "$($API_log)" -Append
    }
}


Function Request-Download {
    # Downloads a file using BITS if possible, and if BITS is not available, downloads directly from URL
    [CmdletBinding()]
    Param(
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [string[]] $FILE_URL,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [string[]] $FILE_LOCAL,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        [switch] $NoBITS, # This is for when BITS should not be used
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        [string[]] $BasicUsername, # This is for auth for downloading
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True)]
        [String[]] $BasicPassword # This is for auth for downloading
    )

    Process{
        if ($BasicUsername -and $BasicPassword) {
            $Credentials = New-Object System.Management.Automation.PSCredential ($userName, (ConvertTo-SecureString $BasicPassword -AsPlainText -Force))
        }
        # Test for existing file and remove if it exists
        if(Test-Path -Path $FILE_LOCAL -PathType Leaf ) {
            try {
                Remove-Item $FILE_LOCAL -Force
            } catch {
                #Can't remove the MSI, therefore cannot continue
                write-host "Cannot remove $($FILE_LOCAL). Unable to continue."
                Write-host $_
                $PScmdlet.ThrowTerminatingError($_)
            }
        }
        
        try {
            $ComputerType = (Get-CimInstance -ClassName Win32_OperatingSystem).ProductType
        } catch {
            write-host "There is a problem with this computer and updates are required for this script to continue."
            $PScmdlet.ThrowTerminatingError($_)
        }

        try {
            if ($ComputerType -ne 1) {
                Install-WindowsFeature BranchCache | Out-Null
            }
        } catch {
            $NoBITS = $true
        }

        if (!(Get-Module -ListAvailable -Name "BitsTransfer") -and !($NoBITS)) {
            try{
                Import-Module BitsTransfer -Force
            } catch {
                $NoBITS = $true
            }
        }



        if (!($NoBITS)) {
            # Check if BranchCache Distributed Mode is enabled, and if not, enable it so BITS uses computers on the subnet to download where available
            $BCStatus = Get-BCStatus
            if ($BCStatus.ClientConfiguration.CurrentClientMode -ne "DistributedCache") {
                try {
                    Enable-BCDistributed -Verbose -Force
                    Write-OutputLog "BranchCache Distributed Mode is now enabled" -Verbose:$VerbosePreference
                } catch {
                    #BranchCache cannot be enabled to work with BITS. BITS will download over the internet connection instead of cached copies on the local subnet
                    Write-OutputLog "Cannot enable BranchCache Distributed Mode. $($Error[0].ErrorDetails). The installation files will download over the internet connection instead of cached copies on the local subnet" -Verbose:$VerbosePreference
                }
            } else {
                Write-OutputLog "BranchCache Distributed Mode is already enabled in distributed mode on this computer" -Verbose:$VerbosePreference
            }
            try {
                if ($Credentials) {
                    $DownloadJob = Start-BitsTransfer -Priority Normal -DisplayName "$($DateStamp) $($FILE_LOCAL)" -Source "$($FILE_URL)" -Destination "$($FILE_LOCAL)" -Credential $Credentials
                } else {
                    $DownloadJob = Start-BitsTransfer -Priority Normal -DisplayName "$($DateStamp) $($FILE_LOCAL)" -Source "$($FILE_URL)" -Destination "$($FILE_LOCAL)"
                }
                
                #Complete-BitsTransfer -BitsJob $DownloadJob
                Write-OutputLog "Downloaded $($FILE_URL) using BITS to $($FILE_LOCAL)" -Verbose:$VerbosePreference
            } catch {
                Write-OutputLog "Cannot download $($FILE_URL) using BITS. Now trying through standard HTTP request." -Verbose:$VerbosePreference
                Write-OutputLog "$($_ | Out-String)" -Verbose:$VerbosePreference
                try {
                    if ($Credentials) {
                        $DownloadJob = Invoke-WebRequest -Uri "$($FILE_URL)" -OutFile "$($FILE_LOCAL)" -PassThru -Credential $Credentials -UseBasicParsing
                    } else {
                        $DownloadJob = Invoke-WebRequest -Uri "$($FILE_URL)" -OutFile "$($FILE_LOCAL)" -PassThru -UseBasicParsing
                    }
                } catch {
                    Write-OutputLog "Cannot download $($FILE_URL) using standard HTTP request." -Verbose:$VerbosePreference
                    Write-OutputLog "$($_ | Out-String)" -Verbose:$VerbosePreference
                    $PScmdlet.ThrowTerminatingError($_)
                }
            }
        } else {
            try {
                if ($Credentials) {
                    $DownloadJob = Invoke-WebRequest -Uri "$($FILE_URL)" -OutFile "$($FILE_LOCAL)" -PassThru -Credential $Credentials -UseBasicParsing
                } else {
                    $DownloadJob = Invoke-WebRequest -Uri "$($FILE_URL)" -OutFile "$($FILE_LOCAL)" -PassThru -UseBasicParsing
                }
        } catch {
                Write-OutputLog "Cannot download $($FILE_URL) using standard HTTP request." -Verbose:$VerbosePreference
                Write-OutputLog "$($_ | Out-String)" -Verbose:$VerbosePreference
                $PScmdlet.ThrowTerminatingError($_)
            }
        }
        return $DownloadJob

    }
}


function Get-CurrentLineNumber {
    # Downloads a file using BITS if possible, and if BITS is not available, downloads directly from URL
    [CmdletBinding()]
    param()
    #$LineNumber =
    #$LineNo = Get-ChildItem $MyInvocation.ScriptLineNumber
    return $PSCmdlet.MyInvocation.ScriptLineNumber
}



function Update-WMF {
    [CmdletBinding()]
    param(
        [Parameter()] [switch] $ForceReboot # Forces a reboot of the machine after update has completed
    )


    $OSInfo = (Get-WMIObject win32_operatingsystem)
    $OSBuild = $OSInfo.buildnumber
    $OSArch = $OSInfo.OSArchitecture
    $PowerShellVersion = $PSVersionTable.PSVersion.Major + ($PSVersionTable.PSVersion.Minor/10)
    $dotnetversion = (Get-ItemProperty "HKLM:\SOFTWARE\Microsoft\NET Framework Setup\NDP\v4\Full" -Name Release).Release
    Write-Host "DotNet Framework version $($dotnetversion) found."
    Write-Host "Powershell version $($PSVersionTable.PSVersion.ToString()) found."
    [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
    
    if ($dotnetversion -lt 379893) {
        Write-Host "Updating DotNet Framework to 4.5.2"
        $dotnet_URL = "https://download.microsoft.com/download/E/2/1/E21644B5-2DF2-47C2-91BD-63C560427900/NDP452-KB2901907-x86-x64-AllOS-ENU.exe"
        $dotnet_File = "C:\CT\NDP452-KB2901907-x86-x64-AllOS-ENU.exe"
        try {
            Invoke-WebRequest -Uri $dotnet_URL -OutFile $dotnet_File -Verbose:$VerbosePreference -UseBasicParsing
        } catch {
            write-host "Cannot download DotNet Framework 4.5.2. $_"
            $PScmdlet.ThrowTerminatingError($_)
        }

        write-host "Installing DotNet Framework 4.5.2"
        $DotNetInstall = Start-Process -FilePath $dotnet_File -ArgumentList "/q /norestart" -Wait -NoNewWindow -PassThru -Verbose:$VerbosePreference
        if (@(0,3010) -contains $DotNetInstall.ExitCode) {
            write-host "DotNet Framework 4.5.2 installed successfully. A reboot of this computer is required to complete the installation."
        } else {
            write-host "Unable to install DotNet Framework 4.5.2. Error code $($DotNetInstall.ExitCode) - $_"
            #Stop-Transcript
            $PScmdlet.ThrowTerminatingError($_)
        }
    }



    if($OSBuild -eq "9600" -and $PowerShellVersion -lt 5.1) {
        # Windows 8.1 and Windows Server 2012r2
        $WMF_URL = "https://download.microsoft.com/download/6/F/5/6F5FF66C-6775-42B0-86C4-47D41F2DA187/Win8.1AndW2K12R2-KB3191564-x64.msu"
    } elseif($OSBuild -eq "9200" -and $PowerShellVersion -lt 5.1) {
        # Windows 8.1 and Windows Server 2012r2
        if($OSArch -eq "64-bit"){
            $WMF_URL = "https://download.microsoft.com/download/6/F/5/6F5FF66C-6775-42B0-86C4-47D41F2DA187/W2K12-KB3191565-x64.msu"
        } else {
            $WMF_URL = "https://download.microsoft.com/download/6/F/5/6F5FF66C-6775-42B0-86C4-47D41F2DA187/Win8.1-KB3191564-x86.msu"
        }    
    }

    if ($WMF_URL) {
        
        $WMF_File = "C:\CT\WMF51.msu"

        # Test for existing WMF file and remove if it exists
        if(Test-Path -Path $WMF_File -PathType Leaf ) {
            try {
                Remove-Item $WMF_File -Force -Verbose:$VerbosePreference
                #write-host "Found old WMF update and removed."
            } catch {
                #Can't remove the WMF, therefore cannot continue
                write-host "Cannot remove $WMF_File. Unable to continue. $($Error[0].Exception.Message)"
                #Stop-Transcript
                $PScmdlet.ThrowTerminatingError($_)
            }
        }
        
        # Download WMF
        try {
            Invoke-WebRequest -Uri $WMF_URL -OutFile $WMF_File -Verbose:$VerbosePreference -UseBasicParsing
        } catch {
            write-host "Cannot download WMF. $($WMFjob.ErrorContextDescription)"
            #Stop-Transcript
            $PScmdlet.ThrowTerminatingError($_)
        }
            

        write-host "Installing WMF"
        $WMFUpgrade = Start-Process -FilePath "C:\Windows\System32\wusa.exe" -ArgumentList "$($WMF_File) /quiet /norestart" -Wait -NoNewWindow -PassThru -Verbose:$VerbosePreference
        if (@(0,3010) -contains $WMFUpgrade.ExitCode) {
            write-host "WMF installed successfully. A reboot of this computer is required to complete the installation."
        } else {
            write-host "Unable to install WMF. Error code $($WMFUpgrade.ExitCode)"
            #Stop-Transcript
            $PScmdlet.ThrowTerminatingError($_)
        }
    }

    if ($ForceReboot) {
        Start-Sleep -Seconds 60
        Restart-Computer -Force -Verbose:$VerbosePreference
    }
    return $true

}



function Update-PowerShell {
    [CmdletBinding()]
    param(
        [Parameter()] [switch] $ForceReboot # Forces a reboot of the machine after update has completed
    )

    $WMFupgrade = $false
    $OSInfo = (Get-WMIObject win32_operatingsystem)
    $OSBuild = $OSInfo.buildnumber
    $PowerShellVersion = $PSVersionTable.PSVersion.Major + ($PSVersionTable.PSVersion.Minor/10)
    $dotnetversion = (Get-ItemProperty "HKLM:\SOFTWARE\Microsoft\NET Framework Setup\NDP\v4\Full" -Name Release).Release
    Write-Host "DotNet Framework version $($dotnetversion) found."
    Write-Host "Powershell version $($PSVersionTable.PSVersion.ToString()) found."
    [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12

    
    if ($dotnetversion -lt 379893) {
        $WMFupgrade = $true
    }



    if($OSBuild -eq "9600" -and $PowerShellVersion -lt 5.1) {
        # Windows 8.1 and Windows Server 2012r2
        $WMFupgrade = $true
    } elseif($OSBuild -eq "9200" -and $PowerShellVersion -lt 5.1) {
        # Windows 8.1 and Windows Server 2012r2
        $WMFupgrade = $true
    }

    if($WMFupgrade -eq $true) {
        try {
            Update-WMF 
        }
        catch {
            write-host "Failed to upgrade WMF to 5.1. Please install DotNet Framework 4.5.2 and WMF 5.1 before upgrading PowerShell"
            $PScmdlet.ThrowTerminatingError($_)
        }
    }

    Write-Host "Now will attempt to install latest PowerShell version alongside Windows PowerShell 5.1."
    try {
        Invoke-Expression -Command "& { $(Invoke-RestMethod -Uri 'https://aka.ms/install-powershell.ps1') } -UseMSI -Quiet" -Verbose:$VerbosePreference
        #iex "& { $(irm https://aka.ms/install-powershell.ps1) } -UseMSI -Quiet"
    } catch {
        Write-Host "Unable to install latest PowerShell"
        Write-Host $_
        $PScmdlet.ThrowTerminatingError($_)
    }
    return $true

}


Function New-APIPost {
    [CmdletBinding()]
    Param(
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [string[]] $BASE_URL,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [string[]] $EndPoint_URL,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [Parameter()] $headers,
        [Parameter(ValueFromPipeline=$True, ValueFromPipelineByPropertyName=$True, Mandatory = $true)]
        [Parameter()] [hashtable] $PostData, # The hashtable that needs to be posted to the API
        [Parameter()] [int] $Retry = 2 # Attempts a retry of the post if it fails
    )

    # Post to API
    Write-OutputLog "Posting $($PostData.Count) items to API" -Verbose:$VerbosePreference
    Write-APILog "$($PostData | Out-String)" -Verbose:$VerbosePreference
    $nullfound = $false
    foreach ($APIData in $PostData) {
        #Write-OutputLog $member.name
        foreach ($APIentry in $APIData.GetEnumerator()) {
            Write-Host "$($APIentry.Name) : $($APIentry.Value)"
            if ($APIentry.Value -eq $null) {$nullfound = $true}
        }
        if($nullfound -ne $true) {
            $body = $APIData | ConvertTo-Json
        } else {
            $body = $null
        }
        Write-APILog $body -Verbose:$VerbosePreference
        if ($null -ne $body) {
            $retryCount = $Retry # performs a retry after 60 seconds if it fails
            do{
                try {
                    $SendToAPI = Invoke-WebRequest -URI "$($BASE_URL)$($EndPoint_URL)" -Method 'POST' -Headers $headers -Body $body -PassThru -Debug:$DebugPreference
                    $retryCount = 0
                    #$ReturnValue = $SendToAPI
                } catch {
                    Write-OutputLog $_.Exception.Message -Verbose:$VerbosePreference
                    Write-OutputLog $_.Exception -Verbose:$VerbosePreference
                    Write-OutputLog $SendToAPI -Verbose:$VerbosePreference
                    Write-OutputLog $body -Verbose:$VerbosePreference
                    
                    # Dig into the exception to get the Response details.
                    Write-OutputLog "API Error: $($SendToAPI.StatusCode) [$($_.Exception.Response.StatusCode.value__)] - $($_.Exception.Response.StatusDescription)" -Verbose:$VerbosePreference
                    $line = $_.InvocationInfo.ScriptLineNumber
                    #if ($DebugPreference -eq "Continue") { write-error "API error: $($_.Exception.Response.StatusCode.value__) - $($_.Exception.Response.StatusDescription)" -Category InvalidData -ErrorAction Continue }
                    #$exiterror = $_.Exception.Response.StatusCode.value__
                    Set-ItemProperty -Path "$($CT_Reg_Key)" -Name "API-Post-$($BASE_URL)$($EndPoint_URL)" -Value "$($_.Exception.Response.StatusCode.value__)"
                    start-sleep -Seconds 60
                    $retryCount = $retryCount - 1
                    if($retryCount -lt 1) {$PScmdlet.ThrowTerminatingError($_)}
                }
            } while ($retryCount -gt 0)
        }
    }
    
}




Function Get-AVStatus {
    
    [cmdletbinding(DefaultParameterSetName = "computer")]

    Param(
        #The name of a computer to query.
        [Parameter(
            Position = 0,
            ValueFromPipeline,
            ValueFromPipelineByPropertyName,
            ParameterSetName = "computer"
            )]
        [ValidateNotNullorEmpty()]
        [string[]]$Computername = $env:COMPUTERNAME,

        #An existing CIMsession.
        [Parameter(ValueFromPipeline, ParameterSetName = "session")]
        [Microsoft.Management.Infrastructure.CimSession[]]$CimSession,

        #The default is enabled products only.
        [switch]$All
    )

    Begin {
        Write-OutputLog "[BEGIN ] Starting: $($MyInvocation.Mycommand)" -Verbose:$VerbosePreference

        $CTPSModules = (Get-Module CT-PS-Standard -ListAvailable)

        $CTPSModPath = $CTPSModules[0].ModuleBase

        $AVSearchList = import-csv -Path "$($CTPSModPath)\antiviruslist.csv"
        Function ConvertTo-Hex {
            Param([int]$Number)
            '0x{0:x}' -f $Number
        }
        [system.Version]$OSVersion = (Get-WmiObject win32_operatingsystem -computername $Computername).version

        #initialize an hashtable of paramters to splat to Get-CimInstance
        IF ($OSVersion -ge [system.version]'6.0.0.0') 
        {
            Write-OutputLog "OS Windows Vista/Server 2008 or newer detected" -Verbose:$VerbosePreference
            $cimParams = @{
                Namespace   = "root/SecurityCenter2"
                ClassName   = "AntiVirusProduct"
# ErrorAction = "Stop"
            }
        } 
        Else 
        {
            Write-OutputLog "Windows 2000, 2003, XP detected" -Verbose:$VerbosePreference
            $cimParams = @{
                Namespace   = "root/SecurityCenter"
                ClassName   = "AntiVirusProduct"
# ErrorAction = "Stop"
            }
        } # end IF ($OSVersion -ge 6.0)

        If ($All) {
            write-outputlog "[BEGIN ] Getting all AV products" -Verbose:$VerbosePreference
        }

        #Test for SecurityCenter(2) existance and if not, run as server
        try {
            $CIMTest = Get-CimInstance @CimParams -ErrorAction SilentlyContinue
            if ($CIMTest) {
                $runAsServer = $False    
            } else {
                $runAsServer = $True
            }
        } catch {
            $runAsServer = $True
        }

        $results = @()
    } #begin

    Process {

        #Check against WMI if workstation
        if($ComputerType -eq 1 -and $runAsServer -eq $False) {

            #initialize an empty array to hold results
            $AV = @()

            write-outputlog "[PROCESS] Using parameter set: $($pscmdlet.ParameterSetName)" -Verbose:$VerbosePreference
            write-outputlog "[PROCESS] PSBoundparameters: " -Verbose:$VerbosePreference
            write-outputlog ($PSBoundParameters | Out-String) -Verbose:$VerbosePreference

            if ($pscmdlet.ParameterSetName -eq 'computer') {
                foreach ($computer in $Computername) {

                    write-outputlog "[PROCESS] Querying $($computer.ToUpper())" -Verbose:$VerbosePreference
                    #$cimParams.ComputerName = $computer
                    Try {
                        $AV += Get-CimInstance @CimParams -Verbose | Where-Object {$_.displayName -ne $null}
                    }
                    Catch {
                        Write-Warning "[$($computer.ToUpper())] $($_.Exception.Message)"
                        $cimParams.ComputerName = $null
                    }

                } #foreach computer
            }
            else {
                foreach ($session in $CimSession) {

                    write-outputlog "[PROCESS] Using session $($session.computername.toUpper())" -Verbose:$VerbosePreference
                    $cimParams.CimSession = $session
                    Try {
                        $AV += Get-CimInstance @CimParams -Verbose | Where-Object {$_.displayName -ne $null}
                    }
                    Catch {
                        Write-Warning "[$($session.computername.ToUpper())] $($_.Exception.Message)"
                        $cimParams.cimsession = $null
                    }

                } #foreach computer
            }

            foreach ($item in $AV) {
                write-outputlog "[PROCESS] Found $($item.Displayname)" -Verbose:$VerbosePreference
                $hx = ConvertTo-Hex $item.ProductState
                $mid = $hx.Substring(3, 2)
                if ($mid -match "00|01") {
                    $Enabled = $False
                }
                else {
                    $Enabled = $True
                }
                $end = $hx.Substring(5)
                if ($end -eq "00") {
                    $UpToDate = $True
                }
                else {
                    $UpToDate = $False
                }

                if(!($item.pathToSignedProductExe)) {
                    $results += $item | Select-Object @{Name = "DisplayName"; Expression = { ($_.Displayname).trim() } }, ProductState,
                    @{Name = "Enabled"; Expression = { $Enabled } },
                    @{Name = "UpToDate"; Expression = { $UptoDate } },
                    @{Name = "Path"; Expression = { $_.pathToSignedProductExe } },
                    @{Name = "Version"; Expression = { $_.VersionNumber } },
                    Timestamp,
                    @{Name = "Computername"; Expression = { $_.PSComputername.toUpper() } }
                } else {
                    if($AVproduct.displayName -match "Defender"){
                        $DefenderInfo = Get-MpComputerStatus
                        $AVversion = $DefenderInfo.AMProductVersion
                    } else {
                        if(Test-Path -Path $item.pathToSignedProductExe) {
                            $AVversion = (Get-Item $item.pathToSignedProductExe -ErrorAction Stop).VersionInfo.fileversion
                            $results += $item | Select-Object @{Name = "DisplayName"; Expression = { ($_.Displayname).trim() } }, ProductState,
                            @{Name = "Enabled"; Expression = { $Enabled } },
                            @{Name = "UpToDate"; Expression = { $UptoDate } },
                            @{Name = "Path"; Expression = { $_.pathToSignedProductExe } },
                            @{Name = "Version"; Expression = { $AVversion } },
                            Timestamp,
                            @{Name = "Computername"; Expression = { $_.PSComputername.toUpper() } }
                        }
                    }
                }

            } #foreach
        
        } else {

            $vbsDL = Request-Download -FILE_URL https://raw.githubusercontent.com/alexisspencer/ScriptsAndAutomationPolicies/master/AV%20Status/AVStatus.vbs -FILE_LOCAL "$($Script_Path)\avstatus.vbs" -NoBITS
            $vbsexe = Start-Process -FilePath "C:\Windows\System32\cscript.exe" -ArgumentList "'$($Script_Path)\avstatus.vbs' WRITE" -Wait

            $AV = @()
            $cimParams = @{
                Namespace   = "root/SecurityCenter"
                ClassName   = "AntiVirusProduct"
# ErrorAction = "Stop"
            }

            write-outputlog "[PROCESS] Using parameter set: $($pscmdlet.ParameterSetName)" -Verbose:$VerbosePreference
            write-outputlog "[PROCESS] PSBoundparameters: " -Verbose:$VerbosePreference
            write-outputlog ($PSBoundParameters | Out-String)

            if ($pscmdlet.ParameterSetName -eq 'computer') {
                foreach ($computer in $Computername) {

                    write-outputlog "[PROCESS] Querying $($computer.ToUpper())" -Verbose:$VerbosePreference
                    #$cimParams.ComputerName = $computer
                    Try {
                        $AV += Get-CimInstance @CimParams -Verbose | Where-Object {$_.displayName -ne $null}
                    }
                    Catch {
                        Write-Warning "[$($computer.ToUpper())] $($_.Exception.Message)"
                        $cimParams.ComputerName = $null
                    }

                } #foreach computer
            }
            else {
                foreach ($session in $CimSession) {

                    write-outputlog "[PROCESS] Using session $($session.computername.toUpper())" -Verbose:$VerbosePreference
                    $cimParams.CimSession = $session
                    Try {
                        $AV += Get-CimInstance @CimParams -Verbose
                    }
                    Catch {
                        Write-Warning "[$($session.computername.ToUpper())] $($_.Exception.Message)"
                        $cimParams.cimsession = $null
                    }

                } #foreach computer
            }

            foreach ($item in $AV) {
                write-outputlog "[PROCESS] Found $($item.Displayname)" -Verbose:$VerbosePreference
                $hx = ConvertTo-Hex $item.ProductState
                $mid = $hx.Substring(3, 2)
                if ($mid -match "00|01") {
                    $Enabled = $False
                }
                else {
                    $Enabled = $True
                }
                $end = $hx.Substring(5)
                if ($end -eq "00") {
                    $UpToDate = $True
                }
                else {
                    $UpToDate = $False
                }

                if(!($item.pathToSignedProductExe)) {
                    $results += $item | Select-Object @{Name = "DisplayName"; Expression = { ($_.Displayname).trim() } }, ProductState,
                    @{Name = "Enabled"; Expression = { $Enabled } },
                    @{Name = "UpToDate"; Expression = { $UptoDate } },
                    @{Name = "Path"; Expression = { $_.pathToSignedProductExe } },
                    @{Name = "Version"; Expression = { $_.VersionNumber } },
                    Timestamp,
                    @{Name = "Computername"; Expression = { $_.PSComputername.toUpper() } }
                } else {
                    if($AVproduct.displayName -match "Defender"){
                        $DefenderInfo = Get-MpComputerStatus
                        $AVversion = $DefenderInfo.AMProductVersion
                    } else {
                        if(Test-Path -Path $item.pathToSignedProductExe) {
                            $AVversion = (Get-Item $item.pathToSignedProductExe -ErrorAction Stop).VersionInfo.fileversion
                            $results += $item | Select-Object @{Name = "DisplayName"; Expression = { ($_.Displayname).trim() } }, ProductState,
                            @{Name = "Enabled"; Expression = { $Enabled } },
                            @{Name = "UpToDate"; Expression = { $UptoDate } },
                            @{Name = "Path"; Expression = { $_.pathToSignedProductExe } },
                            @{Name = "Version"; Expression = { $AVversion } },
                            Timestamp,
                            @{Name = "Computername"; Expression = { $_.PSComputername.toUpper() } }
                        }
                    }
                }

            } #foreach

        } #if/else

    } #process

    End {
        If ($All) {
            write-outputlog "[RETURN] Returning:" -Verbose:$VerbosePreference
            write-outputlog "$($results | Out-String)" -Verbose:$VerbosePreference
            return $results
        }
        else {
            #filter for enabled only
            write-outputlog "[RETURN] Returning:" -Verbose:$VerbosePreference
            write-outputlog "$(($results).Where( { $_.enabled }) | Out-String)" -Verbose:$VerbosePreference
            return ($results).Where( { $_.enabled })
        }

        write-outputlog "[END ] Ending: $($MyInvocation.Mycommand)" -Verbose:$VerbosePreference
    } #end

} #end function