benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-10 23:40:04 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

workaround(indents): avoid reparsing for indent sensitive yaml/python

Browse source
This commit is contained in:
Stephan Seitz 2022-01-23 15:52:03 +01:00
parent d615497e7e
commit aa035cdf2a
1 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

21
lua/nvim-treesitter/indent.lua
View file

@ -2,6 +2,13 @@ local parsers = require "nvim-treesitter.parsers"
local queries = require "nvim-treesitter.query" local queries = require "nvim-treesitter.query"
local tsutils = require "nvim-treesitter.ts_utils" local tsutils = require "nvim-treesitter.ts_utils"
local M = {}
M.avoid_force_reparsing = {
python = true,
yaml = true,
}
local function get_first_node_at_line(root, lnum) local function get_first_node_at_line(root, lnum)
local col = vim.fn.indent(lnum) local col = vim.fn.indent(lnum)
return root:descendant_for_range(lnum - 1, col, lnum - 1, col) return root:descendant_for_range(lnum - 1, col, lnum - 1, col)
@ -23,8 +30,6 @@ local function find_delimiter(bufnr, node, delimiter)
end end
end end
local M = {}
local get_indents = tsutils.memoize_by_buf_tick(function(bufnr, root, lang) local get_indents = tsutils.memoize_by_buf_tick(function(bufnr, root, lang)
local map = { local map = {
auto = {}, auto = {},
@ -51,15 +56,19 @@ end, {
---@param lnum number (1-indexed) ---@param lnum number (1-indexed)
function M.get_indent(lnum) function M.get_indent(lnum)
local parser = parsers.get_parser() local bufnr = vim.api.nvim_get_current_buf()
local parser = parsers.get_parser(bufnr)
if not parser or not lnum then if not parser or not lnum then
return -1 return -1
end end
-- Reparse in case we got triggered by ":h indentkeys" local root_lang = parsers.get_buf_lang(bufnr)
parser:parse()
local bufnr = vim.api.nvim_get_current_buf() -- some languages like Python will actually have worse results when re-parsing at opened new line
if not M.avoid_force_reparsing[root_lang] then
-- Reparse in case we got triggered by ":h indentkeys"
parser:parse()
end
-- get_root_for_position is 0-based. -- get_root_for_position is 0-based.
local root, _, lang_tree = tsutils.get_root_for_position(lnum - 1, 0, parser) local root, _, lang_tree = tsutils.get_root_for_position(lnum - 1, 0, parser)