fix: Ignore Java annotations

Author: RedFlameKen

lua/treesitter-context/context.lua

@@ -377,6 +377,9 @@ function M.get(winid)
           local range0 = context_range(parent, bufnr, query)
           if range0 and range_is_valid(range0) then
             local range, lines = get_text_for_range(range0, bufnr)
+            if range[1] > contexts_end_row - 1 then
+              return
+            end
             if range_is_valid(range) then
               local last_context = context_ranges[#context_ranges]
               if last_context and parent_start_row == last_context[1] then

queries/java/context.scm

@@ -3,6 +3,12 @@
 ) @context
 
 (method_declaration
+  type: (_) @context.start
+  body: (_) @context.end
+) @context
+
+(constructor_declaration
+  name: (_) @context.start
   body: (_) @context.end
 ) @context
 
@@ -15,6 +21,7 @@
 ) @context
 
 (class_declaration
+  name: (_) @context.start
   body: (_) @context.end
 ) @context
 

test/lang/test.java

@@ -3,15 +3,15 @@
 import stuff1;
 import stuff2;
 
-@class_annot_1 // {{CONTEXT}}
-@class_annot_2 // {{CONTEXT}}
+@class_annot_1
+@class_annot_2
 public class MyClass { // {{CONTEXT}}
 
 
 
     // {{CURSOR}}
-    @method_annot_1 // {{CONTEXT}}
-    @method_annot_2 // {{CONTEXT}}
+    @method_annot_1
+    @method_annot_2
     public void my_method(int param) { // {{CONTEXT}}