From d7dde6afff84e440073a2763ee0a22855e118b5f Mon Sep 17 00:00:00 2001
From: Alexandre Fonseca <alexandrefonseca@live.com>
Date: Sun, 14 Jan 2024 17:30:26 +0000
Subject: [PATCH] fix: take into account prompt size in nui and builtin selects
 (#135)

---
 lua/dressing/select/builtin.lua | 2 +-
 lua/dressing/select/nui.lua     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/lua/dressing/select/builtin.lua b/lua/dressing/select/builtin.lua
index a33b872..460b2e8 100644
--- 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
index 6cae3f9..b7fd0ec 100644
--- 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))