fix(base_picker): snacks width and matching

Author: cameronr

lua/opencode/ui/base_picker.lua

@@ -86,9 +86,9 @@ local function telescope_ui(opts)
     return {
       value = item,
       display = function(entry)
-        return displayer(opts.format_fn(entry.value):to_formatted_text())
+        return displayer(opts.format_fn(entry.value, opts.width):to_formatted_text())
       end,
-      ordinal = opts.format_fn(item):to_string(),
+      ordinal = opts.format_fn(item, opts.width):to_string(),
     }
   end
 
@@ -189,7 +189,7 @@ local function fzf_ui(opts)
 
   fzf_lua.fzf_exec(function(fzf_cb)
     for _, item in ipairs(opts.items) do
-      fzf_cb(opts.format_fn(item):to_string())
+      fzf_cb(opts.format_fn(item, opts.width):to_string())
     end
     fzf_cb()
   end, {
@@ -201,7 +201,7 @@ local function fzf_ui(opts)
     actions = actions_config,
     fn_fzf_index = function(line)
       for i, item in ipairs(opts.items) do
-        if opts.format_fn(item):to_string() == line then
+        if opts.format_fn(item, opts.width):to_string() == line then
           return i
         end
       end
@@ -217,7 +217,7 @@ local function mini_pick_ui(opts)
 
   ---@type MiniPickItem[]
   local items = vim.tbl_map(function(item)
-    return { text = opts.format_fn(item):to_string(), item = item }
+    return { text = opts.format_fn(item, opts.width):to_string(), item = item }
   end, opts.items)
 
   local mappings = {}
@@ -234,7 +234,7 @@ local function mini_pick_ui(opts)
               opts.items = new_items
               ---@type MiniPickItem[]
               items = vim.tbl_map(function(it)
-                return { text = opts.format_fn(it):to_string(), item = it }
+                return { text = opts.format_fn(it, opts.width):to_string(), item = it }
               end, opts.items)
               mini_pick.set_picker_items(items)
             end
@@ -286,8 +286,15 @@ local function snacks_picker_ui(opts)
     finder = function()
       return opts.items
     end,
-    format = function(item)
-      return opts.format_fn(item):to_formatted_text()
+    transform = function(item, ctx)
+      -- Snacks requires item.text to be set to do matching
+      if not item.text then
+        local picker_item = opts.format_fn(item, ctx.picker.layout.width)
+        item.text = picker_item:to_string()
+      end
+    end,
+    format = function(item, snacks_picker)
+      return opts.format_fn(item, snacks_picker.layout.width):to_formatted_text()
     end,
     actions = {
       confirm = function(_picker, item)
@@ -392,11 +399,6 @@ function M.pick(opts)
     opts.width = config.ui.picker_width
   end
 
-  local original_format_fn = opts.format_fn
-  opts.format_fn = function(item)
-    return original_format_fn(item, opts.width)
-  end
-
   local title_str = type(opts.title) == 'function' and opts.title() or opts.title --[[@as string]]
   opts.title = build_title(title_str, opts.actions)