<#

.SYNOPSIS
PSAppDeployToolkit - This script performs the installation or uninstallation of an application(s).

.DESCRIPTION
- The script is provided as a template to perform an install, uninstall, or repair of an application(s).
- The script either performs an "Install", "Uninstall", or "Repair" deployment type.
- The install deployment type is broken down into 3 main sections/phases: Pre-Install, Install, and Post-Install.

The script imports the PSAppDeployToolkit module which contains the logic and functions required to install or uninstall an application.

PSAppDeployToolkit is licensed under the GNU LGPLv3 License - (C) 2025 PSAppDeployToolkit Team (Sean Lillis, Dan Cunningham, Muhammad Mashwani, Mitch Richters, Dan Gough).

This program is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the
Free Software Foundation, either version 3 of the License, or any later version. This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details. You should have received a copy of the GNU Lesser General Public License along with this program. If not, see <http://www.gnu.org/licenses/>.

.PARAMETER DeploymentType
The type of deployment to perform.

.PARAMETER DeployMode
Specifies whether the installation should be run in Interactive (shows dialogs), Silent (no dialogs), or NonInteractive (dialogs without prompts) mode.

NonInteractive mode is automatically set if it is detected that the process is not user interactive.

.PARAMETER AllowRebootPassThru
Allows the 3010 return code (requires restart) to be passed back to the parent process (e.g. SCCM) if detected from an installation. If 3010 is passed back to SCCM, a reboot prompt will be triggered.

.PARAMETER TerminalServerMode
Changes to "user install mode" and back to "user execute mode" for installing/uninstalling applications for Remote Desktop Session Hosts/Citrix servers.

.PARAMETER DisableLogging
Disables logging to file for the script.

.EXAMPLE
powershell.exe -File Invoke-AppDeployToolkit.ps1 -DeployMode Silent

.EXAMPLE
powershell.exe -File Invoke-AppDeployToolkit.ps1 -AllowRebootPassThru

.EXAMPLE
powershell.exe -File Invoke-AppDeployToolkit.ps1 -DeploymentType Uninstall

.EXAMPLE
Invoke-AppDeployToolkit.exe -DeploymentType "Install" -DeployMode "Silent"

.INPUTS
None. You cannot pipe objects to this script.

.OUTPUTS
None. This script does not generate any output.

.NOTES
Toolkit Exit Code Ranges:
- 60000 - 68999: Reserved for built-in exit codes in Invoke-AppDeployToolkit.ps1, and Invoke-AppDeployToolkit.exe
- 69000 - 69999: Recommended for user customized exit codes in Invoke-AppDeployToolkit.ps1
- 70000 - 79999: Recommended for user customized exit codes in PSAppDeployToolkit.Extensions module.

.LINK
https://psappdeploytoolkit.com

#>

[CmdletBinding()]
param
(
    [Parameter(Mandatory = $false)]
    [ValidateSet('Install', 'Uninstall', 'Repair')]
    [PSDefaultValue(Help = 'Install', Value = 'Install')]
    [System.String]$DeploymentType,

    [Parameter(Mandatory = $false)]
    [ValidateSet('Interactive', 'Silent', 'NonInteractive')]
    [PSDefaultValue(Help = 'Interactive', Value = 'Interactive')]
    [System.String]$DeployMode,

    [Parameter(Mandatory = $false)]
    [System.Management.Automation.SwitchParameter]$AllowRebootPassThru = $false,

    [Parameter(Mandatory = $false)]
    [System.Management.Automation.SwitchParameter]$TerminalServerMode = $false,

    [Parameter(Mandatory = $false)]
    [System.Management.Automation.SwitchParameter]$DisableLogging
)


##================================================
## MARK: Variables
##================================================

$adtSession = @{
    # App variables.
    AppVendor = 'NCH IS'
    AppName = 'Default PDF Picker'
    AppVersion = '0.1'
    AppArch = 'x64'
    AppLang = 'EN'
    AppRevision = '01'
    AppSuccessExitCodes = @(0)
    AppRebootExitCodes = @(1641, 3010)
    AppScriptVersion = '1.0.0'
    AppScriptDate = '07/25/2025'
    AppScriptAuthor = 'jxp066admin'

    # Install Titles (Only set here to override defaults set by the toolkit).
    InstallName = ''
    InstallTitle = ''

    # Script variables.
    DeployAppScriptFriendlyName = $MyInvocation.MyCommand.Name
    DeployAppScriptVersion = '4.0.6'
    DeployAppScriptParameters = $PSBoundParameters
}

function Install-ADTDeployment
{
    ##================================================
    ## MARK: Pre-Install
    ##================================================
    $adtSession.InstallPhase = "Pre-$($adtSession.DeploymentType)"

    ##================================================
    ## MARK: Install
    ##================================================
    $adtSession.InstallPhase = $adtSession.DeploymentType

    Add-Type -AssemblyName System.Windows.Forms
    
    $SetFTAPath = "C:\ProgramData\SetUserFTA\SetUserFTA.exe"
    
    # Prepare form early so we can use it as a message box owner
    $form = New-Object System.Windows.Forms.Form
    $form.Text = "Set Default PDF Application"
    $form.Size = New-Object System.Drawing.Size(400, 250)
    $form.StartPosition = "CenterScreen"
    $form.TopMost = $true
    
    # Check if SetUserFTA.exe exists
    if (-not (Test-Path $SetFTAPath)) {
        [System.Windows.Forms.MessageBox]::Show($form, "SetUserFTA.exe not found.`nExpected at:`n$SetFTAPath", "Missing Dependency", [System.Windows.Forms.MessageBoxButtons]::OK, [System.Windows.Forms.MessageBoxIcon]::Error)
        $form.Close()
        return
    }
    
    # Known ProgIDs and Friendly Names
    $friendlyMap = @{
        "AcroExch.Document.DC" = "Adobe Acrobat"
        "AcroExch.Document"    = "Adobe Reader"
        "NitroPDF.Pro.v13"     = "Nitro PDF Pro 13"
        "NitroPDF.Pro.v14"     = "Nitro PDF Pro 14"
        "Bluebeam.Revu.PDF"    = "Bluebeam Revu"
        "RedlinePDFfile"       = "Bluebeam Revu PDF"
        "MSEdgePDF"            = "Microsoft Edge"
    }
    
    # Check if a ProgID exists in the registry
    function Is-ProgIDPresent($progId) {
        $paths = @(
            "HKCR:\$progId",
            "HKCU:\Software\Classes\$progId",
            "HKLM:\Software\Classes\$progId"
        )
        foreach ($path in $paths) {
            if (Test-Path $path) { return $true }
        }
        return $false
    }
    
    # Get only installed PDF handlers
    function Get-InstalledPDFProgIDs {
        $installed = @()
        foreach ($progId in $friendlyMap.Keys) {
            if (Is-ProgIDPresent $progId) {
                $installed += $progId
            }
        }
        return $installed
    }
    
    # Get current default handler
    function Get-CurrentPDFHandler {
        try {
            $userSID = ([System.Security.Principal.WindowsIdentity]::GetCurrent()).User.Value
            $keyPath = "Registry::HKEY_USERS\$userSID\Software\Microsoft\Windows\CurrentVersion\Explorer\FileExts\.pdf\UserChoice"
            if (Test-Path $keyPath) {
                return (Get-ItemProperty -Path $keyPath -Name ProgId).ProgId
            } else {
                return "Unknown"
            }
        } catch {
            return "Error"
        }
    }
    
    # Set PDF default using SetUserFTA with a mapping file
    function Set-PDFDefaultHandler($progId) {
        try {
            $mapFile = Join-Path $env:TEMP "SetFTA.txt"
            "pdf $progId" | Out-File -FilePath $mapFile -Encoding ASCII -Force
            Start-Process -FilePath $SetFTAPath -ArgumentList "`"$mapFile`"" -Wait -NoNewWindow
        } catch {
            [System.Windows.Forms.MessageBox]::Show($form, "Failed to apply default:`n$_", "Error", [System.Windows.Forms.MessageBoxButtons]::OK, [System.Windows.Forms.MessageBoxIcon]::Error)
        }
    }
    
    # Add controls
    $labelCurrent = New-Object System.Windows.Forms.Label
    $labelCurrent.Text = "Current Default Handler:"
    $labelCurrent.Location = New-Object System.Drawing.Point(20, 20)
    $labelCurrent.AutoSize = $true
    $form.Controls.Add($labelCurrent)
    
    $labelCurrentValue = New-Object System.Windows.Forms.Label
    $labelCurrentValue.Text = Get-CurrentPDFHandler
    $labelCurrentValue.Location = New-Object System.Drawing.Point(180, 20)
    $labelCurrentValue.AutoSize = $true
    $form.Controls.Add($labelCurrentValue)
    
    $comboBox = New-Object System.Windows.Forms.ComboBox
    $comboBox.Location = New-Object System.Drawing.Point(20, 60)
    $comboBox.Size = New-Object System.Drawing.Size(340, 20)
    $comboBox.DropDownStyle = "DropDownList"
    $form.Controls.Add($comboBox)
    
    # Build dropdown from installed handlers
    $displayToProgID = @{}
    $installedProgIDs = Get-InstalledPDFProgIDs
    $currentHandler = Get-CurrentPDFHandler
    
    foreach ($progId in $installedProgIDs) {
        $friendlyName = $friendlyMap[$progId]
        if (-not $friendlyName) { $friendlyName = $progId }
        $displayToProgID[$friendlyName] = $progId
        $comboBox.Items.Add($friendlyName)
    
        if ($progId -eq $currentHandler) {
            $comboBox.SelectedItem = $friendlyName
        }
    }
    
    # Apply Button
    $buttonApply = New-Object System.Windows.Forms.Button
    $buttonApply.Text = "Set as Default"
    $buttonApply.Location = New-Object System.Drawing.Point(20, 100)
    $buttonApply.Size = New-Object System.Drawing.Size(120, 30)
    $buttonApply.Add_Click({
        if ($comboBox.SelectedItem -and $comboBox.Enabled) {
            $selectedDisplayName = $comboBox.SelectedItem
            $selectedProgId = $displayToProgID[$selectedDisplayName]
            Set-PDFDefaultHandler $selectedProgId
            [System.Windows.Forms.MessageBox]::Show($form, "Default PDF app set to: $selectedDisplayName", "Success", [System.Windows.Forms.MessageBoxButtons]::OK, [System.Windows.Forms.MessageBoxIcon]::Information)
            $labelCurrentValue.Text = Get-CurrentPDFHandler
        } else {
            [System.Windows.Forms.MessageBox]::Show($form, "Please select an application.", "Error", [System.Windows.Forms.MessageBoxButtons]::OK, [System.Windows.Forms.MessageBoxIcon]::Error)
        }
    })
    $form.Controls.Add($buttonApply)
    
    # Close Button
    $buttonClose = New-Object System.Windows.Forms.Button
    $buttonClose.Text = "Close"
    $buttonClose.Location = New-Object System.Drawing.Point(240, 100)
    $buttonClose.Size = New-Object System.Drawing.Size(120, 30)
    $buttonClose.Add_Click({ $form.Close() })
    $form.Controls.Add($buttonClose)
    
    # Show GUI
    [void]$form.ShowDialog()

    ##================================================
    ## MARK: Post-Install
    ##================================================
    $adtSession.InstallPhase = "Post-$($adtSession.DeploymentType)"

    ## Master Wrapper detection
    Set-ADTRegistryKey -Key "HKLM\SOFTWARE\InstalledApps\NCH IS_Default PDF Picker_0.1"
}

function Uninstall-ADTDeployment
{
    ##================================================
    ## MARK: Pre-Uninstall
    ##================================================
    $adtSession.InstallPhase = "Pre-$($adtSession.DeploymentType)"

    ##================================================
    ## MARK: Uninstall
    ##================================================
    $adtSession.InstallPhase = $adtSession.DeploymentType

    ##================================================
    ## MARK: Post-Uninstallation
    ##================================================
    $adtSession.InstallPhase = "Post-$($adtSession.DeploymentType)"

    ## Master Wrapper detection
    Remove-ADTRegistryKey -Key "HKLM\SOFTWARE\InstalledApps\NCH IS_Default PDF Picker_0.1"
}

function Repair-ADTDeployment
{
    ##================================================
    ## MARK: Pre-Repair
    ##================================================
    $adtSession.InstallPhase = "Pre-$($adtSession.DeploymentType)"

    ##================================================
    ## MARK: Repair
    ##================================================
    $adtSession.InstallPhase = $adtSession.DeploymentType

    ##================================================
    ## MARK: Post-Repair
    ##================================================
    $adtSession.InstallPhase = "Post-$($adtSession.DeploymentType)"

    ## Master Wrapper detection
    Set-ADTRegistryKey -Key "HKLM\SOFTWARE\InstalledApps\NCH IS_Default PDF Picker_0.1"
}


##================================================
## MARK: Initialization
##================================================

# Set strict error handling across entire operation.
$ErrorActionPreference = [System.Management.Automation.ActionPreference]::Stop
$ProgressPreference = [System.Management.Automation.ActionPreference]::SilentlyContinue
Set-StrictMode -Version 1

# Import the module and instantiate a new session.
try
{
    $moduleName = if ([System.IO.File]::Exists("$PSScriptRoot\PSAppDeployToolkit\PSAppDeployToolkit.psd1"))
    {
        Get-ChildItem -LiteralPath $PSScriptRoot\PSAppDeployToolkit -Recurse -File | Unblock-File -ErrorAction Ignore
        "$PSScriptRoot\PSAppDeployToolkit\PSAppDeployToolkit.psd1"
    }
    else
    {
        'PSAppDeployToolkit'
    }
    Import-Module -FullyQualifiedName @{ ModuleName = $moduleName; Guid = '8c3c366b-8606-4576-9f2d-4051144f7ca2'; ModuleVersion = '4.0.6' } -Force
    try
    {
        $iadtParams = Get-ADTBoundParametersAndDefaultValues -Invocation $MyInvocation
        $adtSession = Open-ADTSession -SessionState $ExecutionContext.SessionState @adtSession @iadtParams -PassThru
    }
    catch
    {
        Remove-Module -Name PSAppDeployToolkit* -Force
        throw
    }
}
catch
{
    $Host.UI.WriteErrorLine((Out-String -InputObject $_ -Width ([System.Int32]::MaxValue)))
    exit 60008
}


##================================================
## MARK: Invocation
##================================================

try
{
    Get-Item -Path $PSScriptRoot\PSAppDeployToolkit.* | & {
        process
        {
            Get-ChildItem -LiteralPath $_.FullName -Recurse -File | Unblock-File -ErrorAction Ignore
            Import-Module -Name $_.FullName -Force
        }
    }
    & "$($adtSession.DeploymentType)-ADTDeployment"
    Close-ADTSession
}
catch
{
    Write-ADTLogEntry -Message ($mainErrorMessage = Resolve-ADTErrorRecord -ErrorRecord $_) -Severity 3
    Show-ADTDialogBox -Text $mainErrorMessage -Icon Stop | Out-Null
    Close-ADTSession -ExitCode 60001
}
finally
{
    Remove-Module -Name PSAppDeployToolkit* -Force
}