fix: error when disable config returns false

Fixes #31
nvim-neorocks · Aug 29, 2024 · 63b711c · 63b711c
1 parent fd0c830
commit 63b711c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lua/rocks_treesitter/highlight.lua b/lua/rocks_treesitter/highlight.lua
@@ -138,8 +138,12 @@ function highlight.create_autocmd()
             local filetype = vim.bo[bufnr].filetype
             local lang = get_lang(filetype)
             ---@type boolean | nil
-            local disable = type(config.disable) == "function" and config.disable(lang, bufnr)
-                or config.disable[filetype]
+            local disable
+            if type(config.disable) == "function" then
+                disable = config.disable(lang, bufnr)
+            else 
+                disable = config.disable[filetype]
+            end
             if not disable and config.auto_highlight == "all" or config.auto_highlight[lang] then
                 do_highlight(lang)
             end