From 65d0818fede50cb92697701a6afb1c77c7c33ae8 Mon Sep 17 00:00:00 2001
From: Akin Sowemimo <22454918+akinsho@users.noreply.github.com>
Date: Wed, 31 Aug 2022 19:08:27 +0100
Subject: fix(highlight): ensure links are rebuilt on colorscheme change

---
 lua/nvim-treesitter/highlight.lua | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/lua/nvim-treesitter/highlight.lua b/lua/nvim-treesitter/highlight.lua
index 9c4ede21..c50c4f89 100644
--- a/lua/nvim-treesitter/highlight.lua
+++ b/lua/nvim-treesitter/highlight.lua
@@ -130,8 +130,10 @@ elseif not vim.g.skip_ts_default_groups then
   end
 end
 
-for capture, hlgroup in pairs(default_map) do
-  link_captures(capture, hlgroup)
+local function link_all_captures()
+  for capture, hlgroup in pairs(default_map) do
+    link_captures(capture, hlgroup)
+  end
 end
 
 local function should_enable_vim_regex(config, lang)
@@ -177,6 +179,9 @@ function M.set_custom_captures(captures)
 end
 
 function M.set_default_hlgroups()
+  if not ts.highlighter.hl_map then
+    link_all_captures()
+  end
   local highlights = {
     TSNone = { default = true },
     TSPunctDelimiter = { link = "Delimiter", default = true },
-- 
cgit v1.2.3