copy parsers

Author: ray-x

lua/tests/install-parsers.lua

@@ -29,56 +29,79 @@ if not ok then
   os.exit(1)
 end
 
-print('Installation completed, searching for compiled parsers...')
-
--- In treesitter main branch, parsers might be in different locations
-local search_paths = {
-  vim.fn.stdpath('cache') .. '/nvim-treesitter',
-  vim.fn.stdpath('data') .. '/nvim-treesitter',
-  vim.fn.stdpath('state') .. '/nvim-treesitter',
-  install_dir,
-  '~/.local/share/nvim/site',
-  '~/.cache/nvim/nvim-treesitter',
+print('Installation completed, searching for go parser...')
+
+-- Check the build directory specifically for go parser
+local build_locations = {
+  vim.fn.stdpath('cache') .. '/nvim-treesitter/tree-sitter-go',
+  vim.fn.stdpath('data') .. '/nvim-treesitter/tree-sitter-go',
+  install_dir .. '/parser',
+  '/tmp/tree-sitter-go',
 }
 
--- Search recursively for .so files
-for _, base_path in ipairs(search_paths) do
-  local expanded = vim.fn.expand(base_path)
+local go_parser_found = false
+
+for _, location in ipairs(build_locations) do
+  local expanded = vim.fn.expand(location)
+  print('Checking: ' .. expanded)
   if vim.fn.isdirectory(expanded) == 1 then
-    print('Searching in: ' .. expanded)
-    local so_files = vim.fn.glob(expanded .. '/**/*.so', false, true)
-    if #so_files > 0 then
-      print('Found .so files: ' .. vim.inspect(so_files))
-      -- Copy all .so files to install_dir/parser
-      for _, so_file in ipairs(so_files) do
-        local filename = vim.fn.fnamemodify(so_file, ':t')
-        local dest = install_dir .. '/parser/' .. filename
-        print('Copying ' .. so_file .. ' to ' .. dest)
+    print('  Directory exists, listing contents:')
+    local files = vim.fn.glob(expanded .. '/*', false, true)
+    for _, file in ipairs(files) do
+      print('    ' .. file)
+    end
+    
+    -- Look for go.so specifically
+    local go_so = vim.fn.glob(expanded .. '/**/go.so', false, true)
+    if #go_so > 0 then
+      print('  Found go.so: ' .. vim.inspect(go_so))
+      for _, so_file in ipairs(go_so) do
+        local dest = install_dir .. '/parser/go.so'
+        print('  Copying ' .. so_file .. ' to ' .. dest)
         vim.fn.system(string.format('cp "%s" "%s"', so_file, dest))
+        go_parser_found = true
       end
     end
   end
 end
 
--- Also look for queries
-local queries_found = false
-for _, base_path in ipairs(search_paths) do
-  local expanded = vim.fn.expand(base_path)
-  local queries_path = expanded .. '/queries/go'
-  if vim.fn.isdirectory(queries_path) == 1 then
-    print('Found queries in: ' .. queries_path)
-    vim.fn.system(string.format('cp -r "%s" "%s/"', queries_path, install_dir .. '/queries'))
-    queries_found = true
-    break
+-- If not found, search everywhere
+if not go_parser_found then
+  print('\nSearching entire filesystem for go.so...')
+  local search_roots = {
+    vim.fn.stdpath('cache'),
+    vim.fn.stdpath('data'),
+    vim.fn.stdpath('state'),
+    '/tmp',
+  }
+  
+  for _, root in ipairs(search_roots) do
+    local go_files = vim.fn.glob(root .. '/**/go.so', false, true)
+    if #go_files > 0 then
+      print('Found go.so files: ' .. vim.inspect(go_files))
+      for _, so_file in ipairs(go_files) do
+        local dest = install_dir .. '/parser/go.so'
+        print('Copying ' .. so_file .. ' to ' .. dest)
+        vim.fn.system(string.format('cp "%s" "%s"', so_file, dest))
+        go_parser_found = true
+      end
+      break
+    end
   end
 end
 
--- If queries not found, copy from nvim-treesitter source
-if not queries_found then
-  print('Copying queries from nvim-treesitter source...')
-  local ts_queries = vim.fn.expand('~/.local/share/nvim/site/pack/vendor/start/nvim-treesitter/queries/go')
+-- Copy queries from nvim-treesitter source
+print('\nCopying queries...')
+local ts_queries_paths = {
+  vim.fn.expand('~/.local/share/nvim/site/pack/vendor/start/nvim-treesitter/queries/go'),
+  '../nvim-treesitter/queries/go',
+}
+
+for _, ts_queries in ipairs(ts_queries_paths) do
   if vim.fn.isdirectory(ts_queries) == 1 then
+    print('Copying queries from: ' .. ts_queries)
     vim.fn.system(string.format('cp -r "%s" "%s/"', ts_queries, install_dir .. '/queries'))
+    break
   end
 end
 
@@ -90,8 +113,13 @@ local queries_dir = install_dir .. '/queries/go'
 if vim.fn.isdirectory(parser_dir) == 1 then
   local files = vim.fn.readdir(parser_dir)
   print('Parser files: ' .. vim.inspect(files))
-  if #files == 0 then
-    print('ERROR: No parser files found!')
+  
+  -- Check specifically for go.so
+  local go_so_exists = vim.fn.filereadable(parser_dir .. '/go.so') == 1
+  print('go.so exists: ' .. tostring(go_so_exists))
+  
+  if not go_so_exists then
+    print('ERROR: go.so not found!')
     os.exit(1)
   end
 else
@@ -103,17 +131,8 @@ if vim.fn.isdirectory(queries_dir) == 1 then
   local files = vim.fn.readdir(queries_dir)
   print('Query files: ' .. vim.inspect(files))
 else
-  print('WARNING: Queries directory does not exist!')
-end
-
--- Try to load the parser
-for _, parser in ipairs(parsers) do
-  local ok, lang = pcall(vim.treesitter.language.add, parser)
-  if ok then
-    print("✓ Parser " .. parser .. " language added")
-  else
-    print("✗ Parser " .. parser .. " language failed: " .. tostring(lang))
-  end
+  print('ERROR: Queries directory does not exist!')
+  os.exit(1)
 end
 
 print('Parser installation complete!')