rust-lang · Muscraft · Mar 21, 2024 · Mar 21, 2024 · Mar 21, 2024 · Mar 21, 2024
diff --git a/src/renderer/display_list.rs b/src/renderer/display_list.rs
@@ -696,17 +696,19 @@ impl<'a> Iterator for CursorLines<'a> {
 }
 
 fn format_message(
-    snippet::Message {
+    message: snippet::Message<'_>,
+    term_width: usize,
+    anonymized_line_numbers: bool,
+    primary: bool,
+) -> Vec<DisplaySet<'_>> {
+    let snippet::Message {
         level,
         id,
         title,
         footer,
         snippets,
-    }: snippet::Message<'_>,
-    term_width: usize,
-    anonymized_line_numbers: bool,
-    primary: bool,
-) -> Vec<DisplaySet<'_>> {
+    } = message;
+
     let mut sets = vec![];
     let body = if !snippets.is_empty() || primary {
         vec![format_title(level, id, title)]
@@ -715,6 +717,7 @@ fn format_message(
     };
 
     for (idx, snippet) in snippets.into_iter().enumerate() {
+        let snippet = fold_prefix_suffix(snippet);
         sets.push(format_snippet(
             snippet,
             idx == 0,
@@ -876,6 +879,46 @@ fn format_header<'a>(
     None
 }
 
+fn fold_prefix_suffix(mut snippet: snippet::Snippet<'_>) -> snippet::Snippet<'_> {
+    if !snippet.fold {
+        return snippet;
+    }
+
+    let ann_start = snippet
+        .annotations
+        .iter()
+        .map(|ann| ann.range.start)
+        .min()
+        .unwrap_or(0);
+    if let Some(before_new_start) = snippet.source[0..ann_start].rfind('\n') {
+        let new_start = before_new_start + 1;
+
+        let line_offset = snippet.source[..new_start].lines().count();
+        snippet.line_start += line_offset;
+
+        snippet.source = &snippet.source[new_start..];
+
+        for ann in &mut snippet.annotations {
+            let range_start = ann.range.start - new_start;
+            let range_end = ann.range.end - new_start;
+            ann.range = range_start..range_end;
+        }
+    }
+
+    let ann_end = snippet
+        .annotations
+        .iter()
+        .map(|ann| ann.range.end)
+        .max()
+        .unwrap_or(snippet.source.len());
+    if let Some(end_offset) = snippet.source[ann_end..].find('\n') {
+        let new_end = ann_end + end_offset;
+        snippet.source = &snippet.source[..new_end];
+    }
+
+    snippet
+}
+
 fn fold_body(mut body: Vec<DisplayLine<'_>>) -> Vec<DisplayLine<'_>> {
     enum Line {
         Fold(usize),

diff --git a/...xtures/no-color/multiline_annotation2.svg → tests/fixtures/no-color/ann_multiline.svg b/...xtures/no-color/multiline_annotation2.svg → tests/fixtures/no-color/ann_multiline.svg
diff --git a/...tures/no-color/multiline_annotation2.toml → tests/fixtures/no-color/ann_multiline.toml b/...tures/no-color/multiline_annotation2.toml → tests/fixtures/no-color/ann_multiline.toml
diff --git a/...xtures/no-color/multiline_annotation3.svg → tests/fixtures/no-color/ann_multiline2.svg b/...xtures/no-color/multiline_annotation3.svg → tests/fixtures/no-color/ann_multiline2.svg
diff --git a/...tures/no-color/multiline_annotation3.toml → tests/fixtures/no-color/ann_multiline2.toml b/...tures/no-color/multiline_annotation3.toml → tests/fixtures/no-color/ann_multiline2.toml
diff --git a/...ixtures/no-color/multiline_annotation.svg → .../fixtures/no-color/fold_ann_multiline.svg b/...ixtures/no-color/multiline_annotation.svg → .../fixtures/no-color/fold_ann_multiline.svg
diff --git a/...xtures/no-color/multiline_annotation.toml → ...fixtures/no-color/fold_ann_multiline.toml b/...xtures/no-color/multiline_annotation.toml → ...fixtures/no-color/fold_ann_multiline.toml
diff --git a/tests/fixtures/no-color/issue_52.svg → ...ixtures/no-color/fold_bad_origin_line.svg b/tests/fixtures/no-color/issue_52.svg → ...ixtures/no-color/fold_bad_origin_line.svg
diff --git a/tests/fixtures/no-color/issue_52.toml → ...xtures/no-color/fold_bad_origin_line.toml b/tests/fixtures/no-color/issue_52.toml → ...xtures/no-color/fold_bad_origin_line.toml
diff --git a/tests/fixtures/no-color/fold_leading.svg b/tests/fixtures/no-color/fold_leading.svg
diff --git a/tests/fixtures/no-color/fold_leading.toml b/tests/fixtures/no-color/fold_leading.toml
@@ -0,0 +1,26 @@
+[message]
+level = "Error"
+id = "E0308"
+title = "invalid type: integer `20`, expected a bool"
+
+[[message.snippets]]
+source = """
+[workspace]
+
+[package]
+name = "hello"
+version = "1.0.0"
+license = "MIT"
+rust-version = "1.70"
+edition = "2021"
+
+[lints]
+workspace = 20
+"""
+line_start = 1
+origin = "Cargo.toml"
+fold = true
+[[message.snippets.annotations]]
+label = ""
+level = "Error"
+range = [132, 134]
diff --git a/tests/fixtures/no-color/fold_trailing.svg b/tests/fixtures/no-color/fold_trailing.svg
diff --git a/tests/fixtures/no-color/fold_trailing.toml b/tests/fixtures/no-color/fold_trailing.toml
@@ -0,0 +1,25 @@
+[message]
+level = "Error"
+id = "E0308"
+title = "invalid type: integer `20`, expected a lints table"
+
+[[message.snippets]]
+source = """
+lints = 20
+
+[workspace]
+
+[package]
+name = "hello"
+version = "1.0.0"
+license = "MIT"
+rust-version = "1.70"
+edition = "2021"
+"""
+line_start = 1
+origin = "Cargo.toml"
+fold = true
+[[message.snippets.annotations]]
+label = ""
+level = "Error"
+range = [8, 10]
diff --git a/tests/formatter.rs b/tests/formatter.rs
@@ -14,7 +14,6 @@ fn test_i_29() {
 error: oops
  --> <current file>:2:8
   |
-1 | First line
 2 | Second oops line
   |        ^^^^ oops
   |"#]]