feat: goprompt learns about prompt start marks that you can set for iterm and other terminal emulators

Author: NonLogicalDev

cmd/goprompt/cmdRender.go

@@ -21,14 +21,18 @@ var (
 		"prompt-loading", false,
 		"is prompt query not yet done rendering",
 	)
-	flgRColorMode = cmdRender.PersistentFlags().String(
-		"color-mode", "none",
-		"color rendering mode of the prompt (zsh, ascii, none)",
+	flgREscapeMode = cmdRender.PersistentFlags().String(
+		"escape-mode", "none",
+		"color / escape rendering mode of the prompt (zsh, ascii, none)",
 	)
 	flgRMode = cmdRender.PersistentFlags().String(
 		"prompt-mode", "normal",
 		"mode of the prompt (normal, edit)",
 	)
+	flgRPromptStartMark = cmdRender.PersistentFlags().String(
+		"prompt-mark-start", "",
+		"mark to place at the start of the prompt (first prompt line)",
+	)
 )
 
 func init() {
@@ -61,7 +65,7 @@ func setColorMode(mode string) {
 		magentaC = wrapC("%F{magenta}", "%F{reset}")
 		greyC = wrapC("%F{black}", "%F{reset}")
 		newline = "\n%{\r%}"
-		
+
 	} else if mode == "ascii" {
 		redC = color.Red.Render
 		greenC = color.Green.Render
@@ -74,7 +78,7 @@ func setColorMode(mode string) {
 }
 
 func cmdRenderRun(_ *cobra.Command, _ []string) error {
-	setColorMode(*flgRColorMode)
+	setColorMode(*flgREscapeMode)
 
 	if _, err := os.Stdin.Stat(); err != nil {
 		fmt.Printf("%#v", err)
@@ -258,7 +262,14 @@ func cmdRenderRun(_ *cobra.Command, _ []string) error {
 	}
 	promptLines = append(promptLines, promptMarker)
 
-	fmt.Print(strings.Join(promptLines, newline))
+	// Add prompt mark to last line
+	lastLine := len(promptLines) - 1
+	if lastLine >= 0 {
+		promptLines[lastLine] = fmt.Sprintf("%v%v", *flgRPromptStartMark, promptLines[lastLine])
+	}
+
+	fullPrompt := strings.Join(promptLines, newline)
+	fmt.Print(fullPrompt)
 
 	return nil
 }

plugin/zsh/prompt_asynczle_setup.zsh

@@ -1,6 +1,7 @@
 # In a file `prompt_asynczle_setup` available on `fpath`
 emulate -L zsh
 
+typeset -g ZSH_ASYNC_PROMPT_START_MARK=${ZSH_ASYNC_PROMPT_START_MARK:-}
 typeset -g ZSH_ASYNC_PROMPT_TIMEOUT=${ZSH_ASYNC_PROMPT_TIMEOUT:-5s}
 typeset -g ZSH_ASYNC_PROMPT_EXEC=${GOPROMPT}
 
@@ -59,7 +60,8 @@ __async_prompt_render() {
   ${ZSH_ASYNC_PROMPT_EXEC} render \
     --prompt-mode "$MODE" \
     --prompt-loading="$LOADING" \
-    --color-mode "zsh"
+    --prompt-mark-start "$ZSH_ASYNC_PROMPT_START_MARK" \
+    --escape-mode "zsh"
 }
 
 #-------------------------------------------------------------------------------