<#
.NOTES
    Author         : Chris Titus @christitustech
    Runspace Author: @DeveloperDurp
    GitHub         : https://github.com/ChrisTitusTech
    Version        : #{replaceme}
#>
param (
    [switch]$Debug,
    [string]$Config,
    [switch]$Run
)

& {
    $ErrorActionPreference = "Stop"

    # Run as administrator if not already (only for script on disk, not in memory)
    if (-not ([Security.Principal.WindowsPrincipal]([Security.Principal.WindowsIdentity]::GetCurrent())).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
        $MyScriptPath = $MyInvocation.PSCommandPath
        
        # If script not on disk, show error message and quit
        if (-not $MyScriptPath) {
            Write-Host "===========================================" -Foregroundcolor Red
            Write-Host "-- Scripts must be run as Administrator ---" -Foregroundcolor Red
            Write-Host "-- Right-Click Start -> Terminal(Admin) ---" -Foregroundcolor Red
            Write-Host "===========================================" -Foregroundcolor Red
            exit
        }
        
        # Use the modern 'PowerShell' if available, otherwise use 'Windows PowerShell'
        if (Get-Command "pwsh" -ErrorAction SilentlyContinue) {
            $PwshExecutable = "pwsh.exe"
        }
        else {
            $PwshExecutable = "powershell.exe"
        }

        # Elevate the script and exit the current non-elevated script
        $PwshArgList = @(
            "-NoLogo",                                  # Don't print PowerShell header in CLI
            "-NoProfile",                               # Don't load PowerShell profile
            "-File", $MyScriptPath,                     # Script path
            $args | ForEach-Object { $_ }               # Script arguments
        )

        if ($Debug) {
            $PwshArgList = @(
                "-NoExit"                               # Don't exit after running the command
            ) + $PwshArgList
        }

        $PwshArgList = $PwshArgList | ForEach-Object { "`"$_`"" }

        $WorkingDirectory = Get-Location

        $ProcessParameters = @{
            FilePath            = $PwshExecutable;
            ArgumentList        = $PwshArgList;
            WorkingDirectory    = $WorkingDirectory;
            Verb                = 'RunAs';
            PassThru            = $true;
        }

        # Use 'wt' if available
        if (Get-Command "wt" -ErrorAction SilentlyContinue) {
            $WtArgList = @(
                "new-tab",
                $ProcessParameters.FilePath
            ) + $PwshArgList

            $ProcessParameters.FilePath = "wt.exe"
            $ProcessParameters.ArgumentList = $WtArgList
        }

        $process = $null
        try {
            $process = Start-Process @ProcessParameters
        }
        catch {
            $exception = $_.Exception
        
            # Optional: Check for specific error messages or types
            if ($exception.Message -like "*The operation was canceled by the user*") {
                Write-Host "===========================================" -Foregroundcolor Red
                Write-Host "---- This must be run as Administrator ----" -Foregroundcolor Red
                Write-Host "------ Click 'Yes' in the UAC Prompt ------" -Foregroundcolor Red
                Write-Host "------------------ (OR) -------------------" -Foregroundcolor Red
                Write-Host "-- Right-Click Start -> Terminal(Admin) ---" -Foregroundcolor Red
                Write-Host "===========================================" -Foregroundcolor Red
            }
            else {
                $ErrorMessage = @(
                    "An unexpected error occurred:",
                    ($exception.Message -split ":" | Select-Object -Index 1).Trim()
                ) -join "`n"

                Write-Host $ErrorMessage -Foregroundcolor Red

                $process.Kill()
            }
        }
        finally {
            exit
        }
    }
}

# Set DebugPreference based on the -Debug switch
if ($Debug) {
    $DebugPreference = "Continue"
}

if ($Config) {
    $PARAM_CONFIG = $Config
}

$PARAM_RUN = $false
# Handle the -Run switch
if ($Run) {
    Write-Host "Running config file tasks..."
    $PARAM_RUN = $true
}

if (!(Test-Path -Path $ENV:TEMP)) {
    New-Item -ItemType Directory -Force -Path $ENV:TEMP
}

Start-Transcript $ENV:TEMP\Winutil.log -Append

# Load DLLs
Add-Type -AssemblyName PresentationFramework
Add-Type -AssemblyName System.Windows.Forms

# Variable to sync between runspaces
$sync = [Hashtable]::Synchronized(@{})
$sync.PSScriptRoot = $PSScriptRoot
$sync.version = "#{replaceme}"
$sync.configs = @{}
$sync.ProcessRunning = $false

# Set PowerShell window title
$Host.UI.RawUI.WindowTitle = $myInvocation.MyCommand.Definition + "(Admin)"
clear-host