Merge pull request #34 from desdic/20240112apidoc

fix: fix documentation and change treesitter api to vim.treesitter

README.md

@@ -109,7 +109,7 @@ Some file formats just doesn't fit into the category labels so custom ones have
 
 ## Custom queries
 
-You can place your custom queries in `~/.config/nvim/queries` you can load them just as with the build-in queries. So adding `myspecial.scm` as `~/.config/nvim/queries/c/myspecial.scm` enables you to run
+You can place your custom queries in `~/.config/nvim/queries` you can load them just as with the build-in queries. So adding `myspecial` as `~/.config/nvim/queries/c/agrolens.myspecial.scm` enables you to run
 
 ```
 :Telescope agrolens query=myspecial

doc/agrolens.txt

@@ -6,7 +6,7 @@ Agrolens is an extention for telescope that runs pre-defined (or custom)
 tree-sitter queries on a buffer (or all buffers) and gives a quick
 view via telescope.
 
-Usage~
+Usage ~
 Using Lazy plugin manager
 
 {
@@ -68,10 +68,10 @@ Default options when generation a query:
                           `agrolens.generate`({opts})
 Generate treesitter query
 
-Parameters~
+Parameters ~
 {opts} `(table)` Options for generating query.
 
-Usage~
+Usage ~
 `require('agrolens').generate({})`
 
 

lua/telescope/_extensions/agrolenslib.lua

@@ -3,7 +3,6 @@ local finders = require("telescope.finders")
 local make_entry = require("telescope.make_entry")
 local actions = require("telescope.actions")
 local conf = require("telescope.config").values
-local queries = require("nvim-treesitter.query")
 local ppath = require("plenary.path")
 local plenary = require("plenary")
 local utils = require("telescope._extensions.utils")
@@ -138,9 +137,8 @@ agrolens._add_entries = function(
         local root = trees[1]:root()
 
         for _, capture_name in ipairs(capture_names) do
-            local okgetq, iter_query =
-                pcall(queries.get_query, filetype, "agrolens." .. capture_name)
-            if okgetq and iter_query then
+            local iter_query =  vim.treesitter.query.get(filetype, "agrolens." .. capture_name)
+            if iter_query then
                 for _, matches, _ in iter_query:iter_matches(root, bufnr) do
                     local entry = agrolens._create_entry(
                         filename,