Fix end coverage of chunks

Author: lionel-

crates/ark/src/lsp/folding_range.rs

@@ -125,7 +125,6 @@ fn parse_ts_node(
         }
         // End of node handling
         end_node_handler(
-            document,
             folding_ranges,
             end.row + 1,
             &mut child_comment_stack,
@@ -214,15 +213,6 @@ fn get_line_text(
     line[start_idx..end_idx].to_string()
 }
 
-fn find_last_non_empty_line(document: &Document, start_line: usize, end_line: usize) -> usize {
-    for idx in (start_line..=end_line).rev() {
-        if !get_line_text(document, idx, None, None).trim().is_empty() {
-            return idx;
-        }
-    }
-    start_line
-}
-
 fn count_leading_whitespaces(document: &Document, line_num: usize) -> usize {
     let line_text = get_line_text(document, line_num, None, None);
     line_text.chars().take_while(|c| c.is_whitespace()).count()
@@ -371,7 +361,6 @@ fn cell_processor(
 }
 
 fn end_node_handler(
-    document: &Document,
     folding_ranges: &mut Vec<FoldingRange>,
     line_idx: usize,
     comment_stack: &mut Vec<Vec<(usize, usize)>>,
@@ -402,10 +391,7 @@ fn end_node_handler(
     // End cell Handling
     if let Some(cell_start) = cell_marker {
         // For the last cell, include the current line in the folding range
-        let folding_range = comment_range(
-            *cell_start,
-            find_last_non_empty_line(document, *cell_start, line_idx),
-        );
+        let folding_range = comment_range(*cell_start, line_idx - 1);
         folding_ranges.push(folding_range);
         *cell_marker = None;
     }
@@ -776,21 +762,6 @@ function() {
         }
     }
 
-    // Test for correct last non-empty line detection
-    #[test]
-    fn test_find_last_non_empty_line() {
-        let doc = Document::new("\nline1\nline2\n\nline3\n", None);
-
-        assert_eq!(find_last_non_empty_line(&doc, 1, 5), 4);
-        assert_eq!(find_last_non_empty_line(&doc, 1, 2), 2);
-        assert_eq!(find_last_non_empty_line(&doc, 3, 5), 4);
-        assert_eq!(find_last_non_empty_line(&doc, 5, 5), 5);
-
-        // Test with empty document
-        let empty_doc = Document::new("\n\n", None);
-        assert_eq!(find_last_non_empty_line(&empty_doc, 1, 2), 1);
-    }
-
     // Test for whitespace counting
     #[test]
     fn test_count_leading_whitespaces() {

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_section_chunks_with_section_in_middle.snap

@@ -26,7 +26,7 @@ expression: "test_folding_range(\"\n#+ Cell\na\n\n# Section ----\nb\n\n#+ Other
     FoldingRange {
         start_line: 7,
         start_character: None,
-        end_line: 8,
+        end_line: 9,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_section_chunks_with_sections.snap

@@ -36,7 +36,7 @@ expression: "test_folding_range(\"\n# Section ----\na\n\n#+ Cell\nb\n\n# Section
     FoldingRange {
         start_line: 10,
         start_character: None,
-        end_line: 11,
+        end_line: 12,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_section_comments_basic.snap

@@ -36,7 +36,7 @@ expression: "test_folding_range(\"\n# First section ----\na\nb\n\n# Second secti
     FoldingRange {
         start_line: 13,
         start_character: None,
-        end_line: 16,
+        end_line: 17,
         end_character: None,
         kind: Some(
             Region,