update setup script

Author: ray-x

lua/tests/install-parsers.lua

@@ -1,10 +1,30 @@
 #!/usr/bin/env -S nvim -l
-vim.opt.runtimepath:append(os.getenv('NVIM_TS_DIR'))
+vim.opt.runtimepath:append('../nvim-treesitter')
 vim.opt.runtimepath:append('.')
 
 local parsers = { 'go' }
 for i = 1, #_G.arg do
   parsers[#parsers + 1] = _G.arg[i] ---@type string
 end
 
+require('nvim-treesitter').setup({
+  -- Directory to install parsers and queries to
+  install_dir = vim.fn.stdpath('data') .. '/site',
+})
+
 require('nvim-treesitter').install(parsers, { force = true }):wait(1800000) -- wait max. 30 minutes
+
+
+-- Verify installation
+print("Verifying parser installation...")
+local install_info = require('nvim-treesitter.info')
+local installed = install_info.installed_parsers()
+print('Installed parsers: ' .. vim.inspect(installed))
+
+for _, parser in ipairs(parsers) do
+  if vim.tbl_contains(installed, parser) then
+    print("✓ Parser " .. parser .. " successfully installed")
+  else
+    print("✗ Parser " .. parser .. " failed to install")
+  end
+end

lua/tests/minimal.lua

@@ -42,10 +42,14 @@ vim.api.nvim_create_autocmd('FileType', {
   pattern = { 'go' },
   callback = function()
     pcall(vim.treesitter.start)
-    local queries = require('nvim-treesitter.config').get_installed('queries')
-    if not vim.tbl_contains(queries, 'go') then
-      print('Installed queries: ' .. vim.inspect(queries))
-      error('No queries for go found')
+    -- Check if go parser is available using main branch API
+    local ok, parser = pcall(vim.treesitter.get_parser, 0, 'go')
+    if not ok then
+      -- Check what parsers are actually installed
+      local install_info = require('nvim-treesitter.info')
+      local installed = install_info.installed_parsers()
+      print('Installed parsers: ' .. vim.inspect(installed))
+      error('No parser for go found')
     end
   end,
 })