Merge pull request #48 from desdic/neovim0.11

fix: iter_matches() changed
desdic · Sep 24, 2024 · 8d1e4fd · 8d1e4fd
2 parents 75d4c0d + bc9e754
commit 8d1e4fd
Showing 1 changed file with 22 additions and 18 deletions.
diff --git a/lua/agrolens/core.lua b/lua/agrolens/core.lua
@@ -84,25 +84,29 @@ M.add_entries = function(
             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 = M.create_entry(
-                        filename,
-                        relfilename,
-                        matches,
-                        iter_query,
-                        bufnr,
-                        capture_name
-                    )
-
-                    if opts.disable_indentation then
-                        entry.line = utils.all_trim(entry.line)
-                    end
+                for _, matches, _ in
+                    iter_query:iter_matches(root, bufnr, 0, -1, { all = false })
+                do
+                    if matches ~= nil then
+                        local entry = M.create_entry(
+                            filename,
+                            relfilename,
+                            matches,
+                            iter_query,
+                            bufnr,
+                            capture_name
+                        )
+
+                        if opts.disable_indentation then
+                            entry.line = utils.all_trim(entry.line)
+                        end
 
-                    local formated_entry = utils.format_entry(entry)
-                    if not dublicates[formated_entry] then
-                        dublicates[formated_entry] = true
-                        if M.does_match(opts, entry) then
-                            table.insert(entries, entry)
+                        local formated_entry = utils.format_entry(entry)
+                        if not dublicates[formated_entry] then
+                            dublicates[formated_entry] = true
+                            if M.does_match(opts, entry) then
+                                table.insert(entries, entry)
+                            end
                         end
                     end
                 end