Quickly imports for lua modules

[chrisgrieser]

A heavily customized grep_words picker that allows you to easily import lua modules you have already used somewhere else. Confirming inserts the statement as a new line below the current cursor line.

local function import(text) return vim.trim(text:gsub(".-:", "")) end

Snacks.picker.grep_word {
	title = "󰢱 Import module",
	cmd = "rg",
	args = { "--only-matching" },
	regex = true,
	search = [[local (\w+) ?= ?require\(["'](.*?)["']\)(\.[\w.]*)?]],
	ft = "lua",

	live = false,
	layout = {
		preset = "vertical",
		preview = false, ---@diagnostic disable-line: assign-type-mismatch faulty annotation
		layout = { width = 0.75, height = 0.6, min_height = 10 },
	},
	transform = function(item, ctx)
		-- ensure items are unique
		ctx.meta.done = ctx.meta.done or {} ---@type table<string, boolean>
		local imp = import(item.text)
		if ctx.meta.done[imp] then return false end
		ctx.meta.done[imp] = true
	end,
	format = function(item, _picker)
		-- only display the grepped line
		local out = {}
		Snacks.picker.highlight.format(item, item.line, out)
		return out
	end,
	confirm = function(picker, item)
		-- insert the grepped line below the current one
		picker:close()
		local lnum = vim.api.nvim_win_get_cursor(0)[1]
		vim.api.nvim_buf_set_lines(0, lnum, lnum, false, { import(item.text) })
		vim.cmd.normal { "j==", bang = true }
	end,
}