# Install HVCE v4 context-menu entries for the current user.
# Run in PowerShell from the HVCE package directory:
#   powershell -ExecutionPolicy Bypass -File windows\install_context_menu.ps1

$Root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
$Python = "python"
$Hvce = Join-Path $Root "hvce.py"
$Gui = Join-Path $Root "hvce_gui.py"

function New-KeyValue($Path, $Name, $Value) {
    New-Item -Path $Path -Force | Out-Null
    New-ItemProperty -Path $Path -Name $Name -Value $Value -PropertyType String -Force | Out-Null
}

$folderShell = "HKCU:\Software\Classes\Directory\shell\HVCECompress"
New-KeyValue $folderShell "MUIVerb" "Compress with HVCE"
New-KeyValue $folderShell "Icon" "python.exe"
New-KeyValue "$folderShell\command" "(default)" "$Python `"$Hvce`" compress `"%1`" `"%1.hvce`" --profile balanced --recovery-percent 10"

$fileShell = "HKCU:\Software\Classes\*\shell\HVCECompress"
New-KeyValue $fileShell "MUIVerb" "Compress with HVCE"
New-KeyValue $fileShell "Icon" "python.exe"
New-KeyValue "$fileShell\command" "(default)" "$Python `"$Hvce`" compress `"%1`" `"%1.hvce`" --profile balanced --recovery-percent 10"

$archiveShell = "HKCU:\Software\Classes\.hvce\shell\HVCEExtract"
New-KeyValue $archiveShell "MUIVerb" "Extract HVCE here"
New-KeyValue $archiveShell "Icon" "python.exe"
New-KeyValue "$archiveShell\command" "(default)" "$Python `"$Hvce`" extract `"%1`" `"%1_extracted`" --overwrite"

$guiShell = "HKCU:\Software\Classes\.hvce\shell\HVCEGui"
New-KeyValue $guiShell "MUIVerb" "Open HVCE GUI"
New-KeyValue "$guiShell\command" "(default)" "$Python `"$Gui`""

Write-Host "HVCE context menu installed for current user."