Chaining picker callbacks - reverts to normal mode when opening the next picker

[VishalSubramanyam]

I have something along the lines of

Snacks.picker.pick(
...
confirm = function(picker, item)
    picker:close()
    ...
    Snacks.picker.pick(...)
end
)

When the second picker opens, it's put in normal mode. How do I preserve the insert mode that I was in in the first picker?

[drowning-cat]

Open in Insert mode:

confirm = function(picker)
  Snacks.picker.pick(...)
  picker:close()
end,

Restore previous mode:

confirm = function(picker)
  local prev_mode = vim.fn.mode()
  Snacks.picker.pick {
    on_show = function(picker_inner)
      if not prev_mode:match '^i' then
        vim.schedule(function()
          picker_inner.input:stopinsert()
        end)
      end
    end,
  }
  picker:close()
end,

[VishalSubramanyam]

Thanks, this worked!