mirrors/snacks.nvim
1
0
Fork 0
mirror of https://github.com/folke/snacks.nvim synced 2025-12-23 08:47:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

perf(picker): re-use existing string parsers per language to prevent needing to create new parsers

Browse source
This commit is contained in:
Folke Lemaitre 2025-10-24 17:00:34 +02:00
parent 9c4236cae4
commit efa304a078
No known key found for this signature in database
GPG key ID: 9B52594D560070AB
1 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

29
lua/snacks/picker/util/highlight.lua
View file

@ -2,18 +2,21 @@
local M = {}
M.langs = {} ---@type table<string, boolean>
M._scratch_buf = nil ---@type number?
M._scratch = {} ---@type table<string, number>
---@param source string
function M.scratch_buf(source)
if not (M._scratch_buf and vim.api.nvim_buf_is_valid(M._scratch_buf)) then
M._scratch_buf = vim.api.nvim_create_buf(false, true)
vim.api.nvim_buf_set_name(M._scratch_buf, "snacks://picker/highlight")
---@param lang string
function M.scratch_buf(source, lang)
local buf = M._scratch[lang]
if not (buf and vim.api.nvim_buf_is_valid(buf)) then
buf = vim.api.nvim_create_buf(false, true)
vim.api.nvim_buf_set_name(buf, "snacks://picker/highlight/" .. lang)
M._scratch[lang] = buf
end
vim.bo[M._scratch_buf].fixeol = false
vim.bo[M._scratch_buf].eol = false
vim.api.nvim_buf_set_lines(M._scratch_buf, 0, -1, false, vim.split(source, "\n", { plain = true }))
return M._scratch_buf
vim.bo[buf].fixeol = false
vim.bo[buf].eol = false
vim.api.nvim_buf_set_lines(buf, 0, -1, false, vim.split(source, "\n", { plain = true }))
return buf
end
---@param opts? {buf?:number, code?:string, ft?:string, lang?:string, file?:string, extmarks?:boolean}
@ -22,8 +25,6 @@ function M.get_highlights(opts)
assert(opts.buf or opts.code, "buf or code is required")
assert(not (opts.buf and opts.code), "only one of buf or code is allowed")
local buf = opts.buf or M.scratch_buf(opts.code)
local ret = {} ---@type table<number, snacks.picker.Extmark[]>
local ft = opts.ft
@ -32,14 +33,16 @@ function M.get_highlights(opts)
or vim.bo.filetype
local lang = Snacks.util.get_lang(opts.lang or ft)
lang = lang and lang:lower() or nil
local parser ---@type vim.treesitter.LanguageTree?
local parser, buf ---@type vim.treesitter.LanguageTree?, number?
if lang then
local ok = false
buf = opts.buf or M.scratch_buf(opts.code, lang)
ok, parser = pcall(vim.treesitter.get_parser, buf, lang)
parser = ok and parser or nil
end
if parser then
if parser and buf then
parser:parse(true)
parser:for_each_tree(function(tstree, tree)
if not tstree then