function show_mode_rp {
  RPS1="${${KEYMAP/vicmd/-- NORMAL --}/(main|viins)/-- INSERT --}"
  # RPS2=$RPS1
  zle reset-prompt
}

function show_mode_cursor {
  case ${KEYMAP} in
    (vicmd)
      print -n '\e[1 q'
      print -n '\e]12;cyan\a'
      ;;
    (main|viins|*)
      print -n '\e[1 q'
      print -n '\e]12;pink\a'
      ;;
  esac
}

function zle-line-init zle-keymap-select {
  show_mode_cursor
  show_mode_rp
}
zle -N zle-line-init
zle -N zle-keymap-select