powershell: support exe extension

carapace-sh · Oct 29, 2024 · a7dc936 · a7dc936
1 parent 548e451
commit a7dc936
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/shell/powershell/snippet.go b/internal/shell/powershell/snippet.go
@@ -3,13 +3,18 @@ package powershell
 
 import (
 	"fmt"
+	"runtime"
 
 	"github.com/carapace-sh/carapace/pkg/uid"
 	"github.com/spf13/cobra"
 )
 
 // Snippet creates the powershell completion script.
 func Snippet(cmd *cobra.Command) string {
+	prefix := "# "
+	if runtime.GOOS == "windows" {
+		prefix = ""
+	}
 	return fmt.Sprintf(`using namespace System.Management.Automation
 using namespace System.Management.Automation.Language
 Function _%v_completer {
@@ -54,6 +59,7 @@ Function _%v_completer {
 
     $completions
 }
-Register-ArgumentCompleter -Native -CommandName '%v' -ScriptBlock (Get-Item "Function:_%v_completer").ScriptBlock
-`, cmd.Name(), uid.Executable(), uid.Executable(), cmd.Name(), cmd.Name())
+Register-ArgumentCompleter -Native -CommandName '%v'     -ScriptBlock (Get-Item "Function:_%v_completer").ScriptBlock
+%vRegister-ArgumentCompleter -Native -CommandName '%v.exe' -ScriptBlock (Get-Item "Function:_%v_completer").ScriptBlock
+`, cmd.Name(), uid.Executable(), uid.Executable(), cmd.Name(), cmd.Name(), prefix, cmd.Name(), cmd.Name())
 }