Sort prefix keys to the end (#151)

liuchengxu · Oct 2, 2020 · 30c0810 · 30c0810
1 parent 641df56
commit 30c0810
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/autoload/which_key/renderer.vim b/autoload/which_key/renderer.vim
@@ -92,7 +92,18 @@ function! s:create_rows(layout, mappings) abort
   let rows = []
   let row = 0
   let col = 0
-  let smap = sort(filter(keys(mappings), 'v:val !=# "name"'), 'i')
+
+  let leaf_keys = []
+  let dict_keys = []
+  for key in sort(filter(keys(mappings), 'v:val !=# "name"'), 'i')
+    if type(mappings[key]) == s:TYPE.dict
+      call add(dict_keys, key)
+    else
+      call add(leaf_keys, key)
+    endif
+  endfor
+
+  let smap = leaf_keys + dict_keys
 
   let displaynames = which_key#renderer#get_displaynames()
   if get(g:, 'which_key_align_by_seperator', 1)