Functions/GenXdev.AI/Set-GenXdevCommandNotFoundAction.ps1
|
<############################################################################## Part of PowerShell module : GenXdev.AI Original cmdlet filename : Set-GenXdevCommandNotFoundAction.ps1 Original author : René Vaessen / GenXdev Version : 3.26.2026 ################################################################################ Copyright (c) 2026 René Vaessen / GenXdev Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. ################################################################################> # Don't remove this line [dontrefactor] ############################################################################### <# .SYNOPSIS Sets up custom command not found handling with optional AI assistance. .DESCRIPTION Configures PowerShell to handle unknown commands by either navigating to directories or using AI to interpret user intent. The handler first tries any existing command not found handler, then checks if the command is a valid path for navigation. AI resolution is only enabled when the -UseAIResolve switch is explicitly specified. .PARAMETER UseAIResolve When specified, enables AI-powered resolution of unknown commands. The user will be prompted to confirm before any AI-generated command is executed. .PARAMETER NoPrompt When specified together with -UseAIResolve, skips the "What did you want to do?" question and uses the typed command name directly as the AI query. Command execution still requires explicit user confirmation. .EXAMPLE Set-GenXdevCommandNotFoundAction -UseAIResolve Enables AI resolution with interactive prompting for unknown commands. .EXAMPLE Set-GenXdevCommandNotFoundAction -UseAIResolve -NoPrompt Enables AI resolution where the unknown command is sent directly to the AI without asking the user what they meant. #> function Set-GenXdevCommandNotFoundAction { [CmdletBinding(SupportsShouldProcess)] [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSAvoidGlobalVars', '')] [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseBOMForUnicodeEncodedFile', '')] param( ########################################################################### [Parameter( Mandatory = $false, HelpMessage = 'The type of LLM query' )] [ValidateSet( 'SimpleIntelligence', 'Knowledge', 'Pictures', 'TextTranslation', 'Coding', 'ToolUse' )] [string] $LLMQueryType = 'Coding', ########################################################################### [Parameter( Mandatory = $false, HelpMessage = ('The model identifier or pattern to use for AI ' + 'operations') )] [string] $Model, ########################################################################### [Parameter( Mandatory = $false, HelpMessage = 'The API endpoint URL for AI operations' )] [string] $ApiEndpoint, ########################################################################### [Parameter( Mandatory = $false, HelpMessage = 'The API key for authenticated AI operations' )] [string] $ApiKey, ################################################################### [Parameter( Mandatory = $false, HelpMessage = 'Indicates that LLM has no support for JSON schemas' )] [string] $NoSupportForJsonSchema, ####################################################################### [Parameter( Mandatory = $false, HelpMessage = 'Enable AI-powered resolution of unknown commands. ' )] [switch] $UseAIResolve, ####################################################################### [Parameter( Mandatory = $false, HelpMessage = 'Skip the intent prompt and use the command name directly as the AI query (requires -UseAIResolve).' )] [switch] $NoPrompt ####################################################################### ) begin { Microsoft.PowerShell.Utility\Write-Verbose 'Starting Set-GenXdevCommandNotFoundAction' # store reference to existing handler if it's not already our handler $script:originalHandler = $null $currentHandler = $ExecutionContext.InvokeCommand.CommandNotFoundAction # check if handler is already installed if ($null -ne $currentHandler) { $handlerString = $currentHandler.ToString() if (-not $handlerString.Contains('Microsoft.PowerShell.Management\Set-Location "+"')) { $script:originalHandler = $currentHandler Microsoft.PowerShell.Utility\Write-Verbose 'Stored original command handler for chaining' } } } process { if (-not $PSCmdlet.ShouldProcess('Command not found handling', 'Set AI assistance handler')) { return } try { # Add flag to prevent recursion $script:insideCommandHandler = $false $script:UseAIResolve = $true -eq $UseAIResolve $global:UseAIUnknownCommandPrompt = $true -ne $NoPrompt $global:AIPowershellCommandParams = GenXdev\Copy-IdenticalParamValues ` -BoundParameters $PSBoundParameters ` -FunctionName 'GenXdev\Invoke-AIPowershellCommand' ` -DefaultValues (Microsoft.PowerShell.Utility\Get-Variable -Scope Local -ErrorAction SilentlyContinue) Microsoft.PowerShell.Utility\Write-Verbose 'Configuring new CommandNotFoundAction handler' # define the command not found action handler $ExecutionContext.InvokeCommand.CommandNotFoundAction = { param($CommandName, $CommandLookupEventArgs) # prevent recursion if ($script:insideCommandHandler) { Microsoft.PowerShell.Utility\Write-Debug "Preventing recursive call for command: $CommandName" return } $script:insideCommandHandler = $true $origPSDebugPreference = $PSDebugPreference $origErrorActionPreference = $ErrorActionPreference $origVerbosePreference = $VerbosePreference $origWarningPreference = $WarningPreference try { # suppress unnecessary output during handler execution $PSDebugPreference = 'continue' $ErrorActionPreference = 'SilentlyContinue' $VerbosePreference = 'SilentlyContinue' $WarningPreference = 'SilentlyContinue' # skip .NET method calls if ($CommandName -match '^\[.*\]::') { return } # try original handler first if ($null -ne $script:originalHandler) { try { & $script:originalHandler $CommandName $CommandLookupEventArgs if ($CommandLookupEventArgs.StopSearch) { return } } catch { Microsoft.PowerShell.Utility\Write-Debug "Original handler failed: $_" } } } finally { # restore original preferences $PSDebugPreference = $origPSDebugPreference $ErrorActionPreference = $origErrorActionPreference $VerbosePreference = $origVerbosePreference $WarningPreference = $origWarningPreference $script:insideCommandHandler = $false } # handle directory navigation if (Microsoft.PowerShell.Management\Test-Path -LiteralPath $CommandName -PathType Container) { $CommandLookupEventArgs.CommandScriptBlock = { Microsoft.PowerShell.Management\Set-Location -LiteralPath $CommandName Microsoft.PowerShell.Management\Get-ChildItem }.GetNewClosure() $CommandLookupEventArgs.StopSearch = $true return } try { [Uri] $uri; if ([Uri]::TryCreate($CommandName, [UriKind]::Absolute, [ref] $uri)) { return; } } catch { } # skip internal and get- commands if ($CommandLookupEventArgs.CommandOrigin -eq 'Internal' -or $CommandName -like 'get-*') { if (@(Microsoft.PowerShell.Core\Get-Command $CommandName -ErrorAction SilentlyContinue).Length -eq 0) { return } } if (($CommandName.trim() -like "$([char]21)") -or ($CommandName.trim() -like "get-$([char]21)") -or ($CommandName.trim() -like '^U') -or ($CommandName.trim() -like 'get-^U') -or ($CommandName.trim() -eq "")) { $CommandLookupEventArgs.CommandScriptBlock = { } $CommandLookupEventArgs.StopSearch = $true return } if ($CommandName -like '-' -or $CommandName -like "cd -" -or $CommandName -like "Set-Location +") { $CommandLookupEventArgs.CommandScriptBlock = { Microsoft.PowerShell.Management\Set-Location "-" } $CommandLookupEventArgs.StopSearch = $true return } if ($CommandName -like '+' -or $CommandName -like "cd +" -or $CommandName -like "Set-Location -") { $CommandLookupEventArgs.CommandScriptBlock = { Microsoft.PowerShell.Management\Set-Location "+" } $CommandLookupEventArgs.StopSearch = $true return } # AI assistance block — only active when -UseAIResolve was set if ($script:UseAIResolve) { $CommandLookupEventArgs.CommandScriptBlock = { $faultyLine = $MyInvocation.Line if ($global:UseAIUnknownCommandPrompt) { $userChoice = $host.ui.PromptForChoice( 'Command not found', 'Do you want AI to figure out what you ' + 'want?', @('&Nah', '&Yes'), 0) if ($userChoice -eq 0) { return } } $splat = $global:AIPowershellCommandParams Microsoft.PowerShell.Utility\Write-Host "Command not found" -ForegroundColor Cyan Microsoft.PowerShell.Utility\Write-Host "Ok, hold on looking up your instruction using AI.. press Ctrl-C to abort" -ForegroundColor Cyan GenXdev\Invoke-AIPowershellCommand @splat $faultyLine }.GetNewClosure() } $CommandLookupEventArgs.StopSearch = $true } } catch { Microsoft.PowerShell.Utility\Write-Error "Failed to set up command not found handler: $_" } } end { Microsoft.PowerShell.Utility\Write-Verbose 'Command not found handler configuration completed' } } |