feat(sources): added snacks picker source

lua/trouble/sources/snacks.lua

@@ -0,0 +1,117 @@
+---@diagnostic disable: inject-field
+local Item = require("trouble.item")
+
+---@module 'snacks'
+
+---@class trouble.Source.snacks: trouble.Source
+local M = {}
+
+---@type trouble.Item[]
+M.items = {}
+
+M.config = {
+  modes = {
+    snacks = {
+      desc = "Snacks results previously opened with `require('trouble.sources.snacks').open()`.",
+      source = "snacks",
+      groups = {
+        { "cmd", format = "{hl:Title}Snacks{hl} {cmd:Comment} {count}" },
+        { "filename", format = "{file_icon} {filename} {count}" },
+      },
+      sort = { "filename", "pos" },
+      format = "{text:ts} {pos}",
+    },
+    snacks_files = {
+      desc = "Snacks results previously opened with `require('trouble.sources.snacks').open()`.",
+      source = "snacks",
+      groups = {
+        { "cmd", format = "{hl:Title}Snacks{hl} {cmd:Comment} {count}" },
+      },
+      sort = { "filename", "pos" },
+      format = "{file_icon} {filename}",
+    },
+  },
+}
+
+---@param item snacks.picker.Item
+function M.item(item)
+  return Item.new({
+    source = "snacks",
+    buf = item.buf,
+    filename = item.file,
+    pos = item.pos,
+    end_pos = item.end_pos,
+    text = item.line or item.comment or item.label or item.name or item.detail or false,
+    item = item,
+  })
+end
+
+---@param cb trouble.Source.Callback
+---@param _ctx trouble.Source.ctx)
+function M.get(cb, _ctx)
+  cb(M.items)
+end
+
+-- Returns the mode based on the items.
+function M.mode()
+  for _, item in ipairs(M.items) do
+    if item.text then
+      return "snacks"
+    end
+  end
+  return "snacks_files"
+end
+
+---@param picker snacks.Picker
+---@param opts? { type?: "all" | "selected" | "smart", add?: boolean }
+function M.open(picker, opts)
+  opts = opts or {}
+  if not opts.add then
+    M.items = {}
+  end
+  local sitems = {} ---@type snacks.picker.Item[]
+  local selected = picker:selected()
+  opts.type = opts.type or "smart"
+  if opts.type == "smart" then
+    opts.type = #selected == 0 and "all" or "selected"
+  end
+  if opts.type == "all" then
+    vim.list_extend(sitems, picker.finder.items)
+  else
+    vim.list_extend(sitems, selected)
+  end
+  for _, i in ipairs(sitems) do
+    local item = M.item(i)
+    table.insert(M.items, item)
+  end
+  picker:close()
+  vim.schedule(function()
+    require("trouble").open({ mode = M.mode() })
+  end)
+end
+
+---@param opts? { type?: "all" | "selected" | "smart", add?: boolean }
+function M.wrap(opts)
+  ---@param picker snacks.Picker
+  return function(picker)
+    M.open(picker, vim.deepcopy(opts or {}))
+  end
+end
+
+---@type table<string, snacks.picker.Action.spec>
+M.actions = {
+  -- Open selected or all items in the trouble list.
+  trouble_open = { action = M.wrap({ type = "smart" }), desc = "smart-open-with-trouble" },
+  -- Open selected items in the trouble list.
+  trouble_open_selected = { action = M.wrap({ type = "selected" }), desc = "open-with-trouble" },
+  -- Open all items in the trouble list.
+  trouble_open_all = { action = M.wrap({ type = "all" }), desc = "open-all-with-trouble" },
+  -- Add selected or all items to the trouble list.
+  trouble_add = { action = M.wrap({ type = "smart", add = true }), desc = "smart-add-to-trouble" },
+  -- Add selected items to the trouble list.
+  trouble_add_selected = { action = M.wrap({ type = "selected", add = true }), desc = "add-to-trouble" },
+  -- Add all items to the trouble list.
+  trouble_add_all = { action = M.wrap({ type = "all" }), desc = "add-all-to-trouble" },
+}
+
+return M