feat: Display Vi mode as PROMPT (#169)

Add Vi-mode indicator for zsh

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index c95b5fd21..523b2d9bc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -43,6 +43,14 @@ fn main() {
         .help("The execution duration of the last command, in seconds")
         .takes_value(true);
 
+    let keymap_arg = Arg::with_name("keymap")
+        .short("k")
+        .long("keymap")
+        .value_name("KEYMAP")
+        // zsh only
+        .help("The keymap of zsh")
+        .takes_value(true);
+
     let jobs_arg = Arg::with_name("jobs")
         .short("j")
         .long("jobs")
@@ -69,6 +77,7 @@ fn main() {
                 .arg(&status_code_arg)
                 .arg(&path_arg)
                 .arg(&cmd_duration_arg)
+                .arg(&keymap_arg)
                 .arg(&jobs_arg),
         )
         .subcommand(
@@ -82,6 +91,7 @@ fn main() {
                 .arg(&status_code_arg)
                 .arg(&path_arg)
                 .arg(&cmd_duration_arg)
+                .arg(&keymap_arg)
                 .arg(&jobs_arg),
         )
         .get_matches();