feat: debug option
This commit is contained in:
parent
29f9a5f7e4
commit
ae727159b9
4 changed files with 68 additions and 6 deletions
|
|
@ -25,6 +25,10 @@ SETUP *fugitive-ts-setup*
|
||||||
-- Enable/disable highlighting (default: true)
|
-- Enable/disable highlighting (default: true)
|
||||||
enabled = true,
|
enabled = true,
|
||||||
|
|
||||||
|
-- Enable debug logging (default: false)
|
||||||
|
-- Outputs to :messages with [fugitive-ts] prefix
|
||||||
|
debug = false,
|
||||||
|
|
||||||
-- Custom filename -> language mappings (optional)
|
-- Custom filename -> language mappings (optional)
|
||||||
languages = {},
|
languages = {},
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,9 +1,17 @@
|
||||||
local M = {}
|
local M = {}
|
||||||
|
|
||||||
|
---@param msg string
|
||||||
|
---@param ... any
|
||||||
|
local function dbg(msg, ...)
|
||||||
|
local formatted = string.format(msg, ...)
|
||||||
|
vim.notify('[fugitive-ts] ' .. formatted, vim.log.levels.DEBUG)
|
||||||
|
end
|
||||||
|
|
||||||
---@param bufnr integer
|
---@param bufnr integer
|
||||||
---@param ns integer
|
---@param ns integer
|
||||||
---@param hunk fugitive-ts.Hunk
|
---@param hunk fugitive-ts.Hunk
|
||||||
function M.highlight_hunk(bufnr, ns, hunk)
|
---@param debug? boolean
|
||||||
|
function M.highlight_hunk(bufnr, ns, hunk, debug)
|
||||||
local lang = hunk.lang
|
local lang = hunk.lang
|
||||||
if not lang then
|
if not lang then
|
||||||
return
|
return
|
||||||
|
|
@ -22,19 +30,29 @@ function M.highlight_hunk(bufnr, ns, hunk)
|
||||||
|
|
||||||
local ok, parser_obj = pcall(vim.treesitter.get_string_parser, code, lang)
|
local ok, parser_obj = pcall(vim.treesitter.get_string_parser, code, lang)
|
||||||
if not ok or not parser_obj then
|
if not ok or not parser_obj then
|
||||||
|
if debug then
|
||||||
|
dbg('failed to create parser for lang: %s', lang)
|
||||||
|
end
|
||||||
return
|
return
|
||||||
end
|
end
|
||||||
|
|
||||||
local trees = parser_obj:parse()
|
local trees = parser_obj:parse()
|
||||||
if not trees or #trees == 0 then
|
if not trees or #trees == 0 then
|
||||||
|
if debug then
|
||||||
|
dbg('parse returned no trees for lang: %s', lang)
|
||||||
|
end
|
||||||
return
|
return
|
||||||
end
|
end
|
||||||
|
|
||||||
local query = vim.treesitter.query.get(lang, 'highlights')
|
local query = vim.treesitter.query.get(lang, 'highlights')
|
||||||
if not query then
|
if not query then
|
||||||
|
if debug then
|
||||||
|
dbg('no highlights query for lang: %s', lang)
|
||||||
|
end
|
||||||
return
|
return
|
||||||
end
|
end
|
||||||
|
|
||||||
|
local extmark_count = 0
|
||||||
for id, node, _ in query:iter_captures(trees[1]:root(), code) do
|
for id, node, _ in query:iter_captures(trees[1]:root(), code) do
|
||||||
local capture_name = '@' .. query.captures[id]
|
local capture_name = '@' .. query.captures[id]
|
||||||
local sr, sc, er, ec = node:range()
|
local sr, sc, er, ec = node:range()
|
||||||
|
|
@ -50,6 +68,11 @@ function M.highlight_hunk(bufnr, ns, hunk)
|
||||||
hl_group = capture_name,
|
hl_group = capture_name,
|
||||||
priority = 200,
|
priority = 200,
|
||||||
})
|
})
|
||||||
|
extmark_count = extmark_count + 1
|
||||||
|
end
|
||||||
|
|
||||||
|
if debug then
|
||||||
|
dbg('hunk %s:%d applied %d extmarks', hunk.filename, hunk.start_line, extmark_count)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,6 @@
|
||||||
---@class fugitive-ts.Config
|
---@class fugitive-ts.Config
|
||||||
---@field enabled boolean
|
---@field enabled boolean
|
||||||
|
---@field debug boolean
|
||||||
---@field languages table<string, string>
|
---@field languages table<string, string>
|
||||||
---@field debounce_ms integer
|
---@field debounce_ms integer
|
||||||
|
|
||||||
|
|
@ -17,6 +18,7 @@ local ns = vim.api.nvim_create_namespace('fugitive_ts')
|
||||||
---@type fugitive-ts.Config
|
---@type fugitive-ts.Config
|
||||||
local default_config = {
|
local default_config = {
|
||||||
enabled = true,
|
enabled = true,
|
||||||
|
debug = false,
|
||||||
languages = {},
|
languages = {},
|
||||||
debounce_ms = 50,
|
debounce_ms = 50,
|
||||||
}
|
}
|
||||||
|
|
@ -27,6 +29,16 @@ local config = vim.deepcopy(default_config)
|
||||||
---@type table<integer, boolean>
|
---@type table<integer, boolean>
|
||||||
local attached_buffers = {}
|
local attached_buffers = {}
|
||||||
|
|
||||||
|
---@param msg string
|
||||||
|
---@param ... any
|
||||||
|
local function dbg(msg, ...)
|
||||||
|
if not config.debug then
|
||||||
|
return
|
||||||
|
end
|
||||||
|
local formatted = string.format(msg, ...)
|
||||||
|
vim.notify('[fugitive-ts] ' .. formatted, vim.log.levels.DEBUG)
|
||||||
|
end
|
||||||
|
|
||||||
---@param bufnr integer
|
---@param bufnr integer
|
||||||
local function highlight_buffer(bufnr)
|
local function highlight_buffer(bufnr)
|
||||||
if not config.enabled then
|
if not config.enabled then
|
||||||
|
|
@ -39,9 +51,10 @@ local function highlight_buffer(bufnr)
|
||||||
|
|
||||||
vim.api.nvim_buf_clear_namespace(bufnr, ns, 0, -1)
|
vim.api.nvim_buf_clear_namespace(bufnr, ns, 0, -1)
|
||||||
|
|
||||||
local hunks = parser.parse_buffer(bufnr, config.languages)
|
local hunks = parser.parse_buffer(bufnr, config.languages, config.debug)
|
||||||
|
dbg('found %d hunks in buffer %d', #hunks, bufnr)
|
||||||
for _, hunk in ipairs(hunks) do
|
for _, hunk in ipairs(hunks) do
|
||||||
highlight.highlight_hunk(bufnr, ns, hunk)
|
highlight.highlight_hunk(bufnr, ns, hunk, config.debug)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
@ -77,6 +90,8 @@ function M.attach(bufnr)
|
||||||
end
|
end
|
||||||
attached_buffers[bufnr] = true
|
attached_buffers[bufnr] = true
|
||||||
|
|
||||||
|
dbg('attaching to buffer %d', bufnr)
|
||||||
|
|
||||||
local debounced = create_debounced_highlight(bufnr)
|
local debounced = create_debounced_highlight(bufnr)
|
||||||
|
|
||||||
highlight_buffer(bufnr)
|
highlight_buffer(bufnr)
|
||||||
|
|
|
||||||
|
|
@ -6,16 +6,27 @@
|
||||||
|
|
||||||
local M = {}
|
local M = {}
|
||||||
|
|
||||||
|
---@param msg string
|
||||||
|
---@param ... any
|
||||||
|
local function dbg(msg, ...)
|
||||||
|
local formatted = string.format(msg, ...)
|
||||||
|
vim.notify('[fugitive-ts] ' .. formatted, vim.log.levels.DEBUG)
|
||||||
|
end
|
||||||
|
|
||||||
---@param filename string
|
---@param filename string
|
||||||
---@param custom_langs? table<string, string>
|
---@param custom_langs? table<string, string>
|
||||||
|
---@param debug? boolean
|
||||||
---@return string?
|
---@return string?
|
||||||
local function get_lang_from_filename(filename, custom_langs)
|
local function get_lang_from_filename(filename, custom_langs, debug)
|
||||||
if custom_langs and custom_langs[filename] then
|
if custom_langs and custom_langs[filename] then
|
||||||
return custom_langs[filename]
|
return custom_langs[filename]
|
||||||
end
|
end
|
||||||
|
|
||||||
local ft = vim.filetype.match({ filename = filename })
|
local ft = vim.filetype.match({ filename = filename })
|
||||||
if not ft then
|
if not ft then
|
||||||
|
if debug then
|
||||||
|
dbg('no filetype for: %s', filename)
|
||||||
|
end
|
||||||
return nil
|
return nil
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
@ -25,6 +36,11 @@ local function get_lang_from_filename(filename, custom_langs)
|
||||||
if ok then
|
if ok then
|
||||||
return lang
|
return lang
|
||||||
end
|
end
|
||||||
|
if debug then
|
||||||
|
dbg('no parser for lang: %s (ft: %s)', lang, ft)
|
||||||
|
end
|
||||||
|
elseif debug then
|
||||||
|
dbg('no ts lang for filetype: %s', ft)
|
||||||
end
|
end
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|
@ -32,8 +48,9 @@ end
|
||||||
|
|
||||||
---@param bufnr integer
|
---@param bufnr integer
|
||||||
---@param custom_langs? table<string, string>
|
---@param custom_langs? table<string, string>
|
||||||
|
---@param debug? boolean
|
||||||
---@return fugitive-ts.Hunk[]
|
---@return fugitive-ts.Hunk[]
|
||||||
function M.parse_buffer(bufnr, custom_langs)
|
function M.parse_buffer(bufnr, custom_langs, debug)
|
||||||
local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
|
local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
|
||||||
---@type fugitive-ts.Hunk[]
|
---@type fugitive-ts.Hunk[]
|
||||||
local hunks = {}
|
local hunks = {}
|
||||||
|
|
@ -65,7 +82,10 @@ function M.parse_buffer(bufnr, custom_langs)
|
||||||
if filename then
|
if filename then
|
||||||
flush_hunk()
|
flush_hunk()
|
||||||
current_filename = filename
|
current_filename = filename
|
||||||
current_lang = get_lang_from_filename(filename, custom_langs)
|
current_lang = get_lang_from_filename(filename, custom_langs, debug)
|
||||||
|
if debug and current_lang then
|
||||||
|
dbg('file: %s -> lang: %s', filename, current_lang)
|
||||||
|
end
|
||||||
elseif line:match('^@@.-@@') then
|
elseif line:match('^@@.-@@') then
|
||||||
flush_hunk()
|
flush_hunk()
|
||||||
hunk_start = i
|
hunk_start = i
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue