feat(picker): add focus option to resume()

lua/snacks/picker/actions.lua

@@ -778,6 +778,11 @@ function M.cycle_win(picker)
     end
   end
   win = wins[idx % #wins + 1] or 1 -- cycle
+  for name, layout in pairs(picker.layout.wins) do
+    if layout.win == win then
+      picker.init_opts.focus = name
+    end
+  end
   vim.api.nvim_set_current_win(win)
 end
 

lua/snacks/picker/init.lua

@@ -53,6 +53,7 @@ M.meta = {
 ---@field source? string
 ---@field include? string[]
 ---@field exclude? string[]
+---@field focus? "input"|"list"
 
 -- create actual picker functions for autocomplete
 vim.defer_fn(function()

lua/snacks/picker/resume.lua

@@ -58,6 +58,9 @@ function M.resume(opts)
 
   for _, source in ipairs(sources) do
     if M.state[source] and not vim.tbl_contains(opts.exclude or {}, source) then
+      if opts.focus then
+        M.state[source].opts.focus = opts.focus
+      end
       states[#states + 1] = M.state[source]
     end
   end