fix: take into account prompt size in nui and builtin selects (#135)

stevearc · Jan 14, 2024 · d7dde6a · d7dde6a
1 parent 2caff86
commit d7dde6a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lua/dressing/select/builtin.lua b/lua/dressing/select/builtin.lua
@@ -55,7 +55,7 @@ M.select = function(config, items, opts, on_choice)
   end
   local lines = {}
   local highlights = {}
-  local max_width = 1
+  local max_width = opts.prompt and vim.api.nvim_strwidth(opts.prompt) or 1
   for idx, item in ipairs(items) do
     local prefix = ""
     if config.show_numbers then

diff --git a/lua/dressing/select/nui.lua b/lua/dressing/select/nui.lua
@@ -8,7 +8,7 @@ M.select = function(config, items, opts, on_choice)
   local Menu = require("nui.menu")
   local event = require("nui.utils.autocmd").event
   local lines = {}
-  local line_width = 1
+  local line_width = opts.prompt and vim.api.nvim_strwidth(opts.prompt) or 1
   for i, item in ipairs(items) do
     local line = opts.format_item(item)
     line_width = math.max(line_width, vim.api.nvim_strwidth(line))