fix(ai): update tree-sitter textobject to use correct language in error

Author: echasnovski

lua/mini/ai.lua

@@ -1566,7 +1566,7 @@ H.get_matched_ranges_builtin = function(captures)
 
   -- Get query file depending on the local language
   local query = vim.treesitter.query.get(lang, 'textobjects')
-  if query == nil then H.error_treesitter('query') end
+  if query == nil then H.error_treesitter('query', lang) end
 
   -- Compute ranges of matched captures
   local capture_is_requested = vim.tbl_map(function(c) return vim.tbl_contains(captures, '@' .. c) end, query.captures)
@@ -1602,10 +1602,12 @@ H.get_nodes_range_builtin = function(nodes, buf_id, metadata)
   return { left[1], left[2], left[3], right[4], right[5], right[6] }
 end
 
-H.error_treesitter = function(failed_get)
+H.error_treesitter = function(failed_get, lang)
   local buf_id, ft = vim.api.nvim_get_current_buf(), vim.bo.filetype
-  local has_lang, lang = pcall(vim.treesitter.language.get_lang, ft)
-  lang = has_lang and lang or ft
+  if lang == nil then
+    local has_lang, ft_lang = pcall(vim.treesitter.language.get_lang, ft)
+    lang = has_lang and ft_lang or ft
+  end
   local msg = string.format('Can not get %s for buffer %d and language "%s".', failed_get, buf_id, lang)
   H.error(msg)
 end

tests/test_ai.lua

@@ -883,18 +883,28 @@ T['gen_spec']['treesitter()']['validates builtin treesitter presence'] = functio
     '%(mini%.ai%) Can not get query for buffer 1 and language "vim"%.'
   )
 
+  -- - Should show local language in error message
+  child.cmd('edit tmp.lua')
+  set_lines({ 'vim.cmd([[', 'setlocal cursorline', ']])' })
+  set_cursor(2, 0)
+  expect.error(
+    function() child.lua('MiniAi.find_textobject("a", "F")') end,
+    '%(mini%.ai%) Can not get query for buffer 2 and language "vim"%.'
+  )
+
   -- Parser
+  child.cmd('enew')
   child.bo.filetype = 'aaa'
   expect.error(
     function() child.lua('MiniAi.find_textobject("a", "F")') end,
-    '%(mini%.ai%) Can not get parser for buffer 1 and language "aaa"%.'
+    '%(mini%.ai%) Can not get parser for buffer 3 and language "aaa"%.'
   )
 
   -- - Should respect registered language for a filetype
   child.lua('vim.treesitter.language.register("my_aaa", "aaa")')
   expect.error(
     function() child.lua('MiniAi.find_textobject("a", "F")') end,
-    '%(mini%.ai%) Can not get parser for buffer 1 and language "my_aaa"%.'
+    '%(mini%.ai%) Can not get parser for buffer 3 and language "my_aaa"%.'
   )
 end