From bebefb601a7a202b9f02b4fc3da057095f0bbb61 Mon Sep 17 00:00:00 2001
From: Jonathan Apodaca <jrapodaca@gmail.com>
Date: Sat, 9 Aug 2025 08:59:32 -0600
Subject: [PATCH] (range) bounds check for TS caps

---
 lua/u/range.lua | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lua/u/range.lua b/lua/u/range.lua
index 1fcfa3f..d49787e 100644
--- a/lua/u/range.lua
+++ b/lua/u/range.lua
@@ -256,6 +256,9 @@ function Range.from_tsquery_caps(bufnr, query, opts)
       Pos.new(bufnr, stop_row0 + 1, stop_col0 + 1),
       'v'
     )
+    if range.stop.lnum > vim.api.nvim_buf_line_count(bufnr) then
+      range.stop = range.stop:must_next(-1)
+    end
     if not opts.contains_cursor or opts.contains_cursor and range:contains(cursor) then
       local capture_name = q.captures[id]
       if not ranges[capture_name] then ranges[capture_name] = {} end