Check if parser installed with pcall ts.get_parser

cshuaimin · Oct 13, 2023 · 35b280e · 35b280e
1 parent f6ab487
commit 35b280e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lua/ssr.lua b/lua/ssr.lua
@@ -57,7 +57,7 @@ function Ui.new()
   local origin_buf = api.nvim_win_get_buf(self.origin_win)
   local lang = ts.language.get_lang(vim.bo[origin_buf].filetype)
   if not lang then
-    return u.notify("Treesitter parser not found, please try to install it with :TSInstall " .. self.lang)
+    return u.notify("Treesitter language not found")
   end
   self.lang = lang
 

diff --git a/lua/ssr/utils.lua b/lua/ssr/utils.lua
@@ -62,8 +62,10 @@ end
 ---@param end_col number
 ---@return TSNode?
 function M.node_for_range(buf, lang, start_row, start_col, end_row, end_col)
-  local parser = ts.get_parser(buf, lang)
-  return parser:parse()[1]:root():named_descendant_for_range(start_row, start_col, end_row, end_col)
+  local has_parser, parser = pcall(ts.get_parser, buf, lang)
+  if has_parser then
+    return parser:parse()[1]:root():named_descendant_for_range(start_row, start_col, end_row, end_col)
+  end
 end
 
 ---@param buf buffer