Fix coverage of comment sections

Should extend to the next section.
If a nested chunk extends beyond its encompassing section, this confuses
the frontend

Author: lionel-

crates/ark/src/lsp/folding_range.rs

@@ -90,13 +90,9 @@ fn parse_ts_node(
             // Nested comment section handling
             let comment_line = get_line_text(document, start.row, None, None);
 
-            if let Err(err) = nested_processor(
-                document,
-                comment_stack,
-                folding_ranges,
-                start.row,
-                &comment_line,
-            ) {
+            if let Err(err) =
+                nested_processor(comment_stack, folding_ranges, start.row, &comment_line)
+            {
                 lsp::log_error!("Can't process comment: {err:?}");
             };
             region_processor(folding_ranges, region_marker, start.row, &comment_line);
@@ -236,7 +232,6 @@ pub static RE_COMMENT_SECTION: LazyLock<Regex> =
     LazyLock::new(|| Regex::new(r"^\s*(#+)\s*(.*?)\s*[#=-]{4,}\s*$").unwrap());
 
 fn nested_processor(
-    document: &Document,
     comment_stack: &mut Vec<Vec<(usize, usize)>>,
     folding_ranges: &mut Vec<FoldingRange>,
     line_num: usize,
@@ -269,20 +264,14 @@ fn nested_processor(
             },
             Ordering::Equal => {
                 let start_line = comment_stack.last_or_error()?.last_or_error()?.1;
-                folding_ranges.push(comment_range(
-                    start_line,
-                    find_last_non_empty_line(document, start_line, line_num - 1),
-                ));
+                folding_ranges.push(comment_range(start_line, line_num - 1));
                 comment_stack.last_mut_or_error()?.pop();
                 comment_stack.last_mut_or_error()?.push((level, line_num));
                 break;
             },
             Ordering::Greater => {
                 let start_line = comment_stack.last_or_error()?.last_or_error()?.1;
-                folding_ranges.push(comment_range(
-                    start_line,
-                    find_last_non_empty_line(document, start_line, line_num - 1),
-                ));
+                folding_ranges.push(comment_range(start_line, line_num - 1));
                 comment_stack.last_mut_or_error()?.pop(); // Safe: the loop exits early if the stack becomes empty
             },
         }
@@ -496,7 +485,6 @@ h"
 
     #[test]
     fn test_folding_section_comments() {
-        // FIXME: First section doesn't span whole section
         insta::assert_debug_snapshot!(test_folding_range(
             "
 # Section ----

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

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# Empty section ----\n\n# Another empty sect
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 1,
+        end_line: 2,
         end_character: None,
         kind: Some(
             Region,
@@ -16,7 +16,7 @@ expression: "test_folding_range(\"\n# Empty section ----\n\n# Another empty sect
     FoldingRange {
         start_line: 3,
         start_character: None,
-        end_line: 3,
+        end_line: 4,
         end_character: None,
         kind: Some(
             Region,

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

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# First section ----\nfunction() {\n  # #reg
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 10,
+        end_line: 11,
         end_character: None,
         kind: Some(
             Region,
@@ -38,7 +38,7 @@ expression: "test_folding_range(\"\n# First section ----\nfunction() {\n  # #reg
     FoldingRange {
         start_line: 8,
         start_character: None,
-        end_line: 10,
+        end_line: 11,
         end_character: None,
         kind: Some(
             Region,

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

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# Level 1 ----\na\n\n## Level 2 ----\nb\n\n#
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 11,
+        end_line: 12,
         end_character: None,
         kind: Some(
             Region,
@@ -16,7 +16,7 @@ expression: "test_folding_range(\"\n# Level 1 ----\na\n\n## Level 2 ----\nb\n\n#
     FoldingRange {
         start_line: 4,
         start_character: None,
-        end_line: 8,
+        end_line: 9,
         end_character: None,
         kind: Some(
             Region,
@@ -26,7 +26,7 @@ expression: "test_folding_range(\"\n# Level 1 ----\na\n\n## Level 2 ----\nb\n\n#
     FoldingRange {
         start_line: 7,
         start_character: None,
-        end_line: 8,
+        end_line: 9,
         end_character: None,
         kind: Some(
             Region,
@@ -36,7 +36,7 @@ expression: "test_folding_range(\"\n# Level 1 ----\na\n\n## Level 2 ----\nb\n\n#
     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_chunks_with_sections.snap

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# Section ----\na\n\n#+ Cell\nb\n\n# Section
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 5,
+        end_line: 6,
         end_character: None,
         kind: Some(
             Region,

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

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# Section ----\na\n\nb\nc\n\n# Section ----\
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 5,
+        end_line: 6,
         end_character: None,
         kind: Some(
             Region,

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

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# First section ----\na\nb\n\n# Second secti
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 3,
+        end_line: 4,
         end_character: None,
         kind: Some(
             Region,