1use std::ops::Range;
 2
 3use git::diff::{DiffHunk, DiffHunkStatus};
 4use language::Point;
 5
 6use crate::{
 7    display_map::{DisplaySnapshot, ToDisplayPoint},
 8    AnchorRangeExt,
 9};
10
11#[derive(Debug, Clone, PartialEq, Eq)]
12pub enum DisplayDiffHunk {
13    Folded {
14        display_row: u32,
15    },
16
17    Unfolded {
18        display_row_range: Range<u32>,
19        status: DiffHunkStatus,
20    },
21}
22
23impl DisplayDiffHunk {
24    pub fn display_row_range(&self) -> Range<u32> {
25        match self {
26            &DisplayDiffHunk::Folded { display_row } => display_row..display_row + 1,
27            DisplayDiffHunk::Unfolded {
28                display_row_range, ..
29            } => display_row_range.clone(),
30        }
31    }
32}
33
34pub fn diff_hunk_to_display(hunk: DiffHunk<u32>, snapshot: &DisplaySnapshot) -> DisplayDiffHunk {
35    let hunk_start_point = Point::new(hunk.buffer_range.start, 0);
36    let hunk_end_point = Point::new(hunk.buffer_range.end, 0);
37    let hunk_start_point_sub = Point::new(hunk.buffer_range.start.saturating_sub(1), 0);
38    let hunk_end_point_sub = Point::new(
39        hunk.buffer_range
40            .end
41            .saturating_sub(1)
42            .max(hunk.buffer_range.start),
43        0,
44    );
45
46    let is_removal = hunk.status() == DiffHunkStatus::Removed;
47
48    let folds_start = Point::new(hunk.buffer_range.start.saturating_sub(1), 0);
49    let folds_end = Point::new(hunk.buffer_range.end + 1, 0);
50    let folds_range = folds_start..folds_end;
51
52    let containing_fold = snapshot.folds_in_range(folds_range).find(|fold_range| {
53        let fold_point_range = fold_range.to_point(&snapshot.buffer_snapshot);
54        let fold_point_range = fold_point_range.start..=fold_point_range.end;
55
56        let folded_start = fold_point_range.contains(&hunk_start_point);
57        let folded_end = fold_point_range.contains(&hunk_end_point_sub);
58        let folded_start_sub = fold_point_range.contains(&hunk_start_point_sub);
59
60        (folded_start && folded_end) || (is_removal && folded_start_sub)
61    });
62
63    if let Some(fold) = containing_fold {
64        let row = fold.start.to_display_point(snapshot).row();
65        DisplayDiffHunk::Folded { display_row: row }
66    } else {
67        let start = hunk_start_point.to_display_point(snapshot).row();
68        let end = hunk_end_point.to_display_point(snapshot).row();
69        DisplayDiffHunk::Unfolded {
70            display_row_range: start..end,
71            status: hunk.status(),
72        }
73    }
74}