fix(powershell): fix powershell hook script override original prompt …

…function (#101) fix powershell hook script override original prompt function. --------- Co-authored-by: taowong <[email protected]> Co-authored-by: Han Li <[email protected]>
version-fox · Mar 13, 2024 · 9a77ef5 · 9a77ef5
1 parent eef222d
commit 9a77ef5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/internal/shell/powershell.go b/internal/shell/powershell.go
@@ -33,12 +33,13 @@ const hook = `
 {{.EnvContent}}
 
 $__VFOX_PID=$pid;
+$originalPrompt = $function:prompt;
 function prompt {
     $export = &"{{.SelfPath}}" env -s pwsh;
     if ($export) {
-      Invoke-Expression -Command $export;
+		Invoke-Expression -Command $export;
     }
-	return "PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) ";
+    &$originalPrompt;
 }
 
 Register-EngineEvent -SourceIdentifier PowerShell.Exiting -SupportEvent -Action {