SudoEsc

1.0.6

SudoEsc.psm1

# SudoEsc.psm1

$script:debugMode = $false

function Write-DebugMessage {
    param([string]$message)
    if ($script:debugMode) {
        Write-Host "DEBUG: $message" -ForegroundColor Yellow
    }
}

function Get-PSReadLineVersion {
    $module = Get-Module PSReadLine -ListAvailable | Sort-Object Version -Descending | Select-Object -First 1
    return $module.Version
}

function Switch-SudoCommand {
    $line = $null
    $cursor = $null
    [Microsoft.PowerShell.PSConsoleReadLine]::GetBufferState([ref]$line, [ref]$cursor)

    if ([string]::IsNullOrWhiteSpace($line)) {
        $line = (Get-History -Count 1).CommandLine
    }

    if (![string]::IsNullOrWhiteSpace($line)) {
        if ($line.TrimStart().StartsWith("sudo ")) {
            $newLine = $line -replace '^(\s*)sudo\s+', '$1'
        }
        else {
            $newLine = $line -replace '^(\s*)', '$1sudo '
        }
        [Microsoft.PowerShell.PSConsoleReadLine]::SetCursorPosition(0)
        [Microsoft.PowerShell.PSConsoleReadLine]::DeleteLine()
        [Microsoft.PowerShell.PSConsoleReadLine]::Insert($newLine)
    }
}

function Add-SudoEscToProfile {
    $profileContent = @"
 
# SudoEsc Autoload
if (-not (Get-Module -Name SudoEsc -ListAvailable)) {
    Install-Module -Name SudoEsc -Scope CurrentUser -Force
}
Import-Module SudoEsc
Enable-SudoEsc
"@

    if (!(Test-Path -Path $PROFILE)) {
        New-Item -ItemType File -Path $PROFILE -Force | Out-Null
    }

    $currentContent = Get-Content $PROFILE -Raw -ErrorAction SilentlyContinue
    if ($currentContent -notmatch "# SudoEsc Autoload") {
        Add-Content -Path $PROFILE -Value "`n$profileContent"
        Write-Host "SudoEsc has been added to your PowerShell profile. It will be automatically loaded in future sessions." -ForegroundColor Green
        return $true
    }
    else {
        return $false
    }
}

function Enable-SudoEsc {
    $psReadLineVersion = Get-PSReadLineVersion

    if ($psReadLineVersion -ge [Version]"2.2.0") {
        if (!(Get-PSReadLineKeyHandler -Chord 'Escape,Escape' | Where-Object { $_.Function -eq 'SudoEscHandler' })) {
            Set-PSReadLineKeyHandler -Chord 'Escape,Escape' -ScriptBlock {
                Write-DebugMessage "Double Esc detected"
                Switch-SudoCommand
                [Microsoft.PowerShell.PSConsoleReadLine]::EndOfLine()
            } -Description 'SudoEscHandler'
        }
    }
    else {
        $handlers = Get-PSReadLineKeyHandler
        if (!($handlers | Where-Object { $_.Key -eq 'Escape' -and $_.Function -eq 'SudoEscHandler' })) {
            Set-PSReadLineKeyHandler -Key 'Escape' -ScriptBlock {
                $key = $host.UI.RawUI.ReadKey("NoEcho,IncludeKeyDown")
                if ($key.VirtualKeyCode -eq 27) {
                    Write-DebugMessage "Double Esc detected"
                    Switch-SudoCommand
                    [Microsoft.PowerShell.PSConsoleReadLine]::EndOfLine()
                }
                else {
                    [Microsoft.PowerShell.PSConsoleReadLine]::Insert([char]27 + $key.Character)
                }
            } -Description 'SudoEscHandler'
        }
    }
}

function Disable-SudoEsc {
    $psReadLineVersion = Get-PSReadLineVersion

    if ($psReadLineVersion -ge [Version]"2.2.0") {
        Remove-PSReadLineKeyHandler -Chord 'Escape,Escape'
    }
    else {
        Remove-PSReadLineKeyHandler -Key 'Escape'
    }

    Write-Host "SudoEsc functionality disabled."
}

Export-ModuleMember -Function Enable-SudoEsc, Disable-SudoEsc