From d747f1b511c59d9dd45b0cfaaa71aa3063f03c40 Mon Sep 17 00:00:00 2001
From: root <rdyro@stanford.edu>
Date: Mon, 25 Mar 2024 10:16:53 -0700
Subject: [PATCH] adding support for changing cursor in GNU screen

---
 zsh-vi-mode.zsh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/zsh-vi-mode.zsh b/zsh-vi-mode.zsh
index 93c80c2..9e5c8f4 100644
--- a/zsh-vi-mode.zsh
+++ b/zsh-vi-mode.zsh
@@ -3224,6 +3224,11 @@ function zvm_cursor_style() {
     fi
   fi
 
+  # in GNU screen, we need to surround an escape sequence by '\eP' and '\e\\'
+  if [[ $term =~ "^screen.*" ]]; then
+    style='\eP'"$style"'\e\\'
+  fi
+
   echo $style
 }