FMRegJump.ps1

using namespace Microsoft.VisualBasic
using namespace System.Windows.Forms
using namespace System.Diagnostics
using namespace System.IO
using namespace System.Security.Principal

param (
    [Parameter(
        Position = 0,
        HelpMessage = "The file whose extension determines which registry " +
            "key to open in RegEdit or Registry Finder."
    )]
    [string] $InputFile,

    [Parameter(
        HelpMessage = "Root hive path for the extension lookup: " +
            "'SystemFileAssociations' or 'HKEY_CLASSES_ROOT'."
    )]
    [ValidateSet('SystemFileAssociations', 'HKEY_CLASSES_ROOT')]
    [string] $InputFileRoot = 'SystemFileAssociations',

    [Parameter(
        HelpMessage = "Override file input with a preset registry key."
    )]
    [ValidateSet(
        'DirectoryBackground', 'Directory', 'DesktopBackground',
        'AllFileSystemObjects', 'AllFiles', 'Drive', 'MyComputer',
        'EnvVariablesCurrentUser', 'EnvVariablesSystem', 'Folder'
    )]
    [string] $PresetDestination,

    [Parameter(
        HelpMessage = "Which application to use for registry navigation."
    )]
    [ValidateSet('RegistryFinder', 'RegEdit')]
    [string] $AppToUse = 'RegistryFinder',

    [Parameter(
        HelpMessage = "Re-use a single instance instead of opening " +
            "multiple windows."
    )]
    [switch] $SingleInstance
)

# ── Resolve script/exe directory (ps2exe-safe) ──────────────
[string] $ScriptDir = if ($PSScriptRoot) {
    $PSScriptRoot
} else {
    [string] $exePath = [Environment]::GetCommandLineArgs()[0]
    if ($exePath) {
        Split-Path -Parent -Path (Convert-Path -LiteralPath $exePath)
    } else { '.' }
}

# ─────────────────────────────────────────────────────────────────────────────────────────────
# Region - INTERNAL FUNCTIONS
# ─────────────────────────────────────────────────────────────────────────────────────────────

function Invoke-InternalMessageBox {
    [CmdletBinding()]
    [OutputType([string])]
    param (
        [Parameter(Position = 0, Mandatory)]
        [ValidateNotNullOrEmpty()]
        [string] $Prompt,
        [ValidateNotNullOrEmpty()]
        [string] $Title = 'FMRegJump',
        [ValidateSet('Information', 'Question', 'Critical', 'Exclamation')]
        [string] $Icon = 'Information',
        [ValidateSet(
            'OKOnly', 'OKCancel', 'AbortRetryIgnore',
            'YesNoCancel', 'YesNo', 'RetryCancel'
        )]
        [string] $BoxType = 'OKOnly',
        [ValidateSet(1, 2, 3)]
        [int] $DefaultButton = 1,
        [switch] $NonTopMost
    )

    if (-not ('FMDevToolbox.Win32.DpiHelper' -as [type])) {
        Add-Type -AssemblyName System.Windows.Forms
        Add-Type -TypeDefinition @'
using System.Runtime.InteropServices;
namespace FMDevToolbox.Win32 {
    public static class DpiHelper {
        [DllImport("user32.dll")]
        public static extern bool SetProcessDPIAware();
    }
}
'@ -Language CSharp
        [void][FMDevToolbox.Win32.DpiHelper]::SetProcessDPIAware()
        [Application]::EnableVisualStyles()
    }

    [string] $boxEnumName = switch ($BoxType) {
        'OKOnly'   { 'OkOnly' }
        'OKCancel' { 'OkCancel' }
        default    { $BoxType }
    }

    [MsgBoxStyle] $style =
        [MsgBoxStyle]::$Icon -bor
        [MsgBoxStyle]::$boxEnumName -bor
        [MsgBoxStyle]::"DefaultButton$DefaultButton" -bor
        $(if ($NonTopMost) {
            [MsgBoxStyle]::ApplicationModal
        } else {
            [MsgBoxStyle]::SystemModal
        })

    [Interaction]::MsgBox($Prompt, $style, $Title) -as [string]
}

function Read-InternalTomlConfig {
    [CmdletBinding()]
    [OutputType([hashtable])]
    param (
        [Parameter(Mandatory)]
        [string] $Path
    )

    if (-not (Test-Path -LiteralPath $Path -PathType Leaf)) {
        return $null
    }

    [hashtable] $config = @{}
    foreach ($line in [File]::ReadAllLines($Path)) {
        [string] $trimmed = $line.Trim()

        # Skip blank lines, comments, and table headers
        if (-not $trimmed -or
            $trimmed.StartsWith('#') -or
            $trimmed.StartsWith('[')) {
            continue
        }

        # Match: key = 'value' or key = "value"
        if ($trimmed -match
            "^([A-Za-z0-9_-]+)\s*=\s*(['""])(.*)(\2)\s*$") {
            $config[$Matches[1]] = $Matches[3]
        }
    }
    return $config
}

function Invoke-InternalRegistryKey {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory, Position = 0)]
        [ValidateNotNullOrEmpty()]
        [string[]] $RegKey,

        [switch] $SingleInstance
    )

    [string] $lastKeyPath =
        'HKCU:\Software\Microsoft\Windows\CurrentVersion\Applets\Regedit'

    [hashtable] $hiveMap = @{
        'HKLM' = 'HKEY_LOCAL_MACHINE'
        'HKCU' = 'HKEY_CURRENT_USER'
        'HKU'  = 'HKEY_USERS'
        'HKCC' = 'HKEY_CURRENT_CONFIG'
        'HKCR' = 'HKEY_CLASSES_ROOT'
    }

    [string[]] $keysToOpen = if ($SingleInstance) {
        , $RegKey[0]
    } else { $RegKey }

    foreach ($key in $keysToOpen) {

        [string] $normalized = $key.Trim() -replace ':\\\\?', '\'
        foreach ($abbr in $hiveMap.Keys) {
            if ($normalized -match "^$abbr\\") {
                $normalized = $normalized -replace "^$abbr",
                    $hiveMap[$abbr]
                break
            }
        }
        $normalized = $normalized.TrimEnd('\')

        [string] $testPath = $normalized
        [bool] $found = $false
        while ($testPath) {
            if (Test-Path -LiteralPath "Registry::$testPath") {
                $found = $true
                break
            }
            [string] $parent = Split-Path $testPath
            if ($parent -eq $testPath) { break }
            $testPath = $parent
        }

        if (-not $found) {
            Invoke-InternalMessageBox `
                -Prompt "Registry key not found:`n`n$key" `
                -Icon 'Exclamation'
            continue
        }

        Write-Verbose "Opening: $testPath"
        Set-ItemProperty -LiteralPath $lastKeyPath `
            -Name LastKey -Value $testPath -Force

        [hashtable] $startParams = @{
            FilePath = 'regedit'
            Verb     = 'RunAs'
            PassThru = $true
        }
        if (-not $SingleInstance) {
            $startParams.ArgumentList = '-m'
        }

        [Process] $proc = Start-Process @startParams

        # Wait up to 10 seconds for the window to appear
        [int] $waited = 0
        while ($proc.MainWindowHandle -eq 0 -and $waited -lt 10000) {
            Start-Sleep -Milliseconds 100
            $waited += 100
        }
        if ($proc.MainWindowHandle -eq 0) {
            Invoke-InternalMessageBox `
                -Prompt ("RegEdit failed to open within 10 seconds." +
                    "`n`nThe UAC prompt may have been dismissed.") `
                -Icon 'Critical'
        }
    }
}

function New-InternalVerboseLog {
    [CmdletBinding()]
    [OutputType([string])]
    param (
        [Parameter(Mandatory,Position=0)]
        [ValidateNotNullOrEmpty()]
        [string] $Message,
        [ValidateSet("ERROR", "WARNING", "INFO", "SUCCESS", "DEBUG")]
        [string] $Level = "INFO",
        [switch] $IncludeCallerInfo,
        [switch] $PassThru
    )
    $vLevel = $Level.ToUpper()
    $ts = [DateTime]::UtcNow.ToString("yyyy-MM-dd HH:mm:ss")
    $prefix = "[$ts] [$vLevel]"
    if ($IncludeCallerInfo) {
        $caller = (Get-PSCallStack)[1].FunctionName
        $prefix += " [$caller]"
    }
    $formatted = "${prefix} $Message"
    Write-Verbose $formatted
    if ($PassThru) { $formatted }
}

# EndRegion

# ─────────────────────────────────────────────────────────────────────────────────────────────
# Region - VALIDATION
# ─────────────────────────────────────────────────────────────────────────────────────────────

# Mutual exclusivity
if ($InputFile -and $PresetDestination) {
    Invoke-InternalMessageBox `
        -Prompt ("-InputFile and -PresetDestination are mutually" +
            " exclusive. Provide one or the other.") `
        -Icon 'Critical'
    return
}

# At least one input required
if (-not $InputFile -and -not $PresetDestination) {
    Invoke-InternalMessageBox `
        -Prompt ("No input provided. Pass either -InputFile or" +
            " -PresetDestination.") `
        -Icon 'Critical'
    return
}

# Validate InputFile exists (when used)
if ($InputFile -and -not (Test-Path -LiteralPath $InputFile)) {
    Invoke-InternalMessageBox `
        -Prompt "The specified path does not exist:`n`n$InputFile" `
        -Icon 'Critical'
    return
}

# Registry Finder config validation (only when needed)
[string] $rfPath = $null
if ($AppToUse -eq 'RegistryFinder') {

    [string] $configPath = Join-Path $ScriptDir 'FMRegJumpConfig.toml'

    New-InternalVerboseLog -Message "Config Path Loaded: $configPath" -Level INFO

    if (-not (Test-Path -LiteralPath $configPath -PathType Leaf)) {
        New-InternalVerboseLog -Message "Configuration file not found: $configPath" -Level ERROR
        Invoke-InternalMessageBox `
            -Prompt "Configuration file not found:`n`n$configPath" `
            -Icon 'Critical'
        return
    }

    [hashtable] $config = Read-InternalTomlConfig -Path $configPath
    New-InternalVerboseLog -Message "Configuration file loaded: `n`n$($config.ToString())" -Level INFO

    if (-not $config -or -not $config['RegistryFinderPath']) {
        New-InternalVerboseLog -Message "Missing 'RegistryFinderPath' in config." -Level ERROR
        Invoke-InternalMessageBox `
            -Prompt ("Missing 'RegistryFinderPath' in config:" +
                "`n`n$configPath") `
            -Icon 'Critical'
        return
    }

    $rfPath = $config['RegistryFinderPath']
    New-InternalVerboseLog -Message "Extracted Registry Finder Path from TOML Config: $rfPath" -Level SUCCESS

    # Accept .exe (preferred — GUI subsystem, no console window) or .com (legacy —
    # console subsystem, spawns a console window alongside Registry Finder).
    if ([Path]::GetExtension($rfPath) -notin @('.com', '.exe')) {
        New-InternalVerboseLog -Message "Registry Finder Path from TOML must be .exe or .com." -Level ERROR
        Invoke-InternalMessageBox `
            -Prompt ("RegistryFinderPath must point to a .exe or .com file." +
                "`n`nCurrent value:`n$rfPath") `
            -Icon 'Critical'
        return
    }

    if (-not (Test-Path -LiteralPath $rfPath -PathType Leaf)) {
        New-InternalVerboseLog -Message "Registry Finder not found at '$rfPath'" -Level ERROR
        Invoke-InternalMessageBox `
            -Prompt "Registry Finder not found at:`n`n$rfPath" `
            -Icon 'Critical'
        return
    }
}

# EndRegion

# ─────────────────────────────────────────────────────────────────────────────────────────────
# Region - KEY RESOLUTION
# ─────────────────────────────────────────────────────────────────────────────────────────────

[string] $key = ''

if ($PresetDestination) {
    [string] $cuSID =
        [System.Security.Principal.WindowsIdentity]::GetCurrent().User.Value
    [hashtable] $presets = @{
        DirectoryBackground     = 'HKCR\Directory\Background\shell'
        Directory               = 'HKCR\Directory\shell'
        Folder                  = 'HKCR\Folder'
        DesktopBackground       = 'HKCR\DesktopBackground\shell'
        AllFileSystemObjects    = 'HKCR\AllFileSystemObjects\shell'
        AllFiles                = 'HKCR\*\shell'
        Drive                   = 'HKCR\Drive\shell'
        MyComputer              = 'HKCR\CLSID\{20D04FE0-3AEA-1069-A2D8-08002B30309D}\shell'
        EnvVariablesCurrentUser = "HKU\$cuSID\Environment"
        EnvVariablesSystem      = 'HKLM\SYSTEM\CurrentControlSet\Control\Session Manager\Environment'
    }
    $key = $presets[$PresetDestination]
}
else {
    if (Test-Path -LiteralPath $InputFile -PathType Leaf) {
        [string] $ext = [Path]::GetExtension($InputFile)
        [string] $root = if ($InputFileRoot -eq 'SystemFileAssociations') {
            "HKCR\SystemFileAssociations\$ext"
        } else { "HKCR\$ext" }
        $key = "$root\Shell"
    }
    elseif (Test-Path -LiteralPath $InputFile -PathType Container) {
        $key = 'HKCR\Directory\shell'
    }
}

# Safety net: key must be resolved before launching anything
if (-not $key) {
    Invoke-InternalMessageBox `
        -Prompt ("Could not determine a registry key from the" +
            " provided input.") `
        -Icon 'Critical'
    return
}

New-InternalVerboseLog -Message "Registry Key has been populated: '$key'" -Level INFO


# EndRegion

# ─────────────────────────────────────────────────────────────────────────────────────────────
# Region - LAUNCH APPLICATION
# ─────────────────────────────────────────────────────────────────────────────────────────────

if ($AppToUse -eq 'RegistryFinder') {
    New-InternalVerboseLog -Message "App to launch is Registry Finder. Building arguments." -Level INFO

    [hashtable] $rfStartParams = @{
        FilePath = $rfPath
        PassThru = $true
    }

    # Build argument list
    [string[]] $rfArgs = @()
    if (-not $SingleInstance) { $rfArgs += '--multiInst' }
    $rfArgs += '--navigate'
    $rfArgs += $key
    $rfStartParams.ArgumentList = $rfArgs
    New-InternalVerboseLog -Message "Pre Registry Finder Call, the arguments are: $rfArgs" -Level INFO

    # We're already elevated via the SELF-ELEVATION region, so Registry Finder
    # inherits our admin token. No -Verb RunAs needed (which would fail anyway:
    # .com has no 'runas' shell verb on Windows 10/11). No auto-elevating compat
    # flag needed on Registry Finder either.
    [Process] $rfProc = Start-Process @rfStartParams

    # Wait up to 10 seconds for the window to appear
    [int] $waited = 0
    while ($rfProc.MainWindowHandle -eq 0 -and $waited -lt 10000) {
        Start-Sleep -Milliseconds 100
        $waited += 100
    }
    if ($rfProc.MainWindowHandle -eq 0) {
        Invoke-InternalMessageBox `
            -Prompt "Registry Finder failed to open within 10 seconds." `
            -Icon 'Critical'
    }
    New-InternalVerboseLog -Message "Registry Finder was called." -Level INFO
}
else {
    New-InternalVerboseLog -Message "App to launch is RegEdit. Building arguments." -Level INFO
    [hashtable] $regeditSplat = @{
        RegKey         = $key
        SingleInstance = $SingleInstance
    }
    Invoke-InternalRegistryKey @regeditSplat
    New-InternalVerboseLog -Message "RegEdit was called." -Level INFO
}

# EndRegion