Support for keymap highlighting

README.md

@@ -107,6 +107,15 @@ function install_powerline_precmd() {
 if [ "$TERM" != "linux" ]; then
     install_powerline_precmd
 fi
+
+function zle-line-init zle-keymap-select {
+    export KEYMAP_POWERLINE=$KEYMAP
+    powerline_precmd
+    zle reset-prompt
+}
+
+zle -N zle-line-init
+zle -N zle-keymap-select
 ```
 
 ### Fish

main.go

@@ -82,6 +82,7 @@ var modules = map[string](func(*powerline)){
 	"time":     segmentTime,
 	"user":     segmentUser,
 	"venv":     segmentVirtualEnv,
+	"keymap":   segmentKeymap,
 }
 
 func main() {

segment-keymap.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"os"
+)
+
+var editSegment = segment{
+	content:    "\u270E",
+	foreground: 15,
+	background: 161,
+}
+
+var commandSegment = segment{
+	content:    "\u26CF",
+	foreground: 15,
+	background: 31,
+}
+
+func segmentKeymap(p *powerline) {
+	var keymap string
+	if keymap == "" {
+		keymap, _ = os.LookupEnv("KEYMAP_POWERLINE")
+	}
+	if keymap == "main" {
+		p.appendSegment("keymap", editSegment)
+	}
+	if keymap == "vicmd" {
+		p.appendSegment("keymap", commandSegment)
+	}
+
+}