mirror of
https://github.com/nvim-treesitter/nvim-treesitter.git
synced 2026-07-21 21:00:08 -04:00
80 lines
2.1 KiB
Lua
80 lines
2.1 KiB
Lua
local api = vim.api
|
|
local ts = vim.treesitter
|
|
|
|
local queries = require'nvim-treesitter.query'
|
|
local parsers = require'nvim-treesitter.parsers'
|
|
local configs = require'nvim-treesitter.configs'
|
|
|
|
local M = {
|
|
highlighters = {}
|
|
}
|
|
|
|
local hlmap = vim.treesitter.TSHighlighter.hl_map
|
|
|
|
-- Misc
|
|
hlmap.error = "TSError"
|
|
hlmap["punctuation.delimiter"] = "TSPunctDelimiter"
|
|
hlmap["punctuation.bracket"] = "TSPunctBracket"
|
|
hlmap["punctuation.special"] = "TSPunctSpecial"
|
|
|
|
-- Constants
|
|
hlmap["constant"] = "TSConstant"
|
|
hlmap["constant.builtin"] = "TSConstBuiltin"
|
|
hlmap["constant.macro"] = "TSConstMacro"
|
|
hlmap["string"] = "TSString"
|
|
hlmap["string.regex"] = "TSStringRegex"
|
|
hlmap["string.escape"] = "TSStringEscape"
|
|
hlmap["character"] = "TSCharacter"
|
|
hlmap["number"] = "TSNumber"
|
|
hlmap["boolean"] = "TSBoolean"
|
|
hlmap["float"] = "TSFloat"
|
|
hlmap["annotation"] = "TSAnnotation"
|
|
|
|
-- Functions
|
|
hlmap["function"] = "TSFunction"
|
|
hlmap["function.builtin"] = "TSFuncBuiltin"
|
|
hlmap["function.macro"] = "TSFuncMacro"
|
|
hlmap["parameter"] = "TSIdentifier"
|
|
hlmap["method"] = "TSMethod"
|
|
hlmap["field"] = "TSField"
|
|
hlmap["property"] = "TSProperty"
|
|
hlmap["constructor"] = "TSConstructor"
|
|
|
|
-- Keywords
|
|
hlmap["conditional"] = "TSConditional"
|
|
hlmap["repeat"] = "TSRepeat"
|
|
hlmap["label"] = "TSLabel"
|
|
hlmap["operator"] = "TSOperator"
|
|
hlmap["keyword"] = "TSKeyword"
|
|
hlmap["exception"] = "TSException"
|
|
|
|
hlmap["type"] = "TSType"
|
|
hlmap["type.builtin"] = "TSTypeBuiltin"
|
|
hlmap["structure"] = "TSStructure"
|
|
hlmap["include"] = "TSInclude"
|
|
|
|
function M.attach(bufnr, lang)
|
|
local bufnr = bufnr or api.nvim_get_current_buf()
|
|
local lang = lang or parsers.get_buf_lang(bufnr)
|
|
local config = configs.get_module('highlight')
|
|
|
|
for k, v in pairs(config.custom_captures) do
|
|
hlmap[k] = v
|
|
end
|
|
|
|
local query = queries.get_query(lang, "highlights")
|
|
if not query then return end
|
|
|
|
M.highlighters[bufnr] = ts.TSHighlighter.new(query, bufnr, lang)
|
|
end
|
|
|
|
function M.detach(bufnr)
|
|
local buf = bufnr or api.nvim_get_current_buf()
|
|
if M.highlighters[buf] then
|
|
M.highlighters[buf]:set_query("")
|
|
M.highlighters[buf] = nil
|
|
end
|
|
api.nvim_buf_set_option(buf, 'syntax', 'on')
|
|
end
|
|
|
|
return M
|