use super::{Bias, DisplayPoint, DisplaySnapshot, SelectionGoal, ToDisplayPoint};
use crate::{char_kind, CharKind, ToPoint};
use language::Point;
use std::ops::Range;

pub fn left(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
    if point.column() > 0 {
        *point.column_mut() -= 1;
    } else if point.row() > 0 {
        *point.row_mut() -= 1;
        *point.column_mut() = map.line_len(point.row());
    }
    map.clip_point(point, Bias::Left)
}

pub fn right(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
    let max_column = map.line_len(point.row());
    if point.column() < max_column {
        *point.column_mut() += 1;
    } else if point.row() < map.max_point().row() {
        *point.row_mut() += 1;
        *point.column_mut() = 0;
    }
    map.clip_point(point, Bias::Right)
}

pub fn up(
    map: &DisplaySnapshot,
    start: DisplayPoint,
    goal: SelectionGoal,
    preserve_column_at_start: bool,
) -> (DisplayPoint, SelectionGoal) {
    let mut goal_column = if let SelectionGoal::Column(column) = goal {
        column
    } else {
        map.column_to_chars(start.row(), start.column())
    };

    let prev_row = start.row().saturating_sub(1);
    let mut point = map.clip_point(
        DisplayPoint::new(prev_row, map.line_len(prev_row)),
        Bias::Left,
    );
    if point.row() < start.row() {
        *point.column_mut() = map.column_from_chars(point.row(), goal_column);
    } else if preserve_column_at_start {
        return (start, goal);
    } else {
        point = DisplayPoint::new(0, 0);
        goal_column = 0;
    }

    let clip_bias = if point.column() == map.line_len(point.row()) {
        Bias::Left
    } else {
        Bias::Right
    };

    (
        map.clip_point(point, clip_bias),
        SelectionGoal::Column(goal_column),
    )
}

pub fn down(
    map: &DisplaySnapshot,
    start: DisplayPoint,
    goal: SelectionGoal,
    preserve_column_at_end: bool,
) -> (DisplayPoint, SelectionGoal) {
    let mut goal_column = if let SelectionGoal::Column(column) = goal {
        column
    } else {
        map.column_to_chars(start.row(), start.column())
    };

    let next_row = start.row() + 1;
    let mut point = map.clip_point(DisplayPoint::new(next_row, 0), Bias::Right);
    if point.row() > start.row() {
        *point.column_mut() = map.column_from_chars(point.row(), goal_column);
    } else if preserve_column_at_end {
        return (start, goal);
    } else {
        point = map.max_point();
        goal_column = map.column_to_chars(point.row(), point.column())
    }

    let clip_bias = if point.column() == map.line_len(point.row()) {
        Bias::Left
    } else {
        Bias::Right
    };

    (
        map.clip_point(point, clip_bias),
        SelectionGoal::Column(goal_column),
    )
}

pub fn line_beginning(
    map: &DisplaySnapshot,
    display_point: DisplayPoint,
    stop_at_soft_boundaries: bool,
) -> DisplayPoint {
    let point = display_point.to_point(map);
    let soft_line_start = map.clip_point(DisplayPoint::new(display_point.row(), 0), Bias::Right);
    let indent_start = Point::new(
        point.row,
        map.buffer_snapshot.indent_size_for_line(point.row).len,
    )
    .to_display_point(map);
    let line_start = map.prev_line_boundary(point).1;

    if stop_at_soft_boundaries && soft_line_start > indent_start && display_point != soft_line_start
    {
        soft_line_start
    } else if stop_at_soft_boundaries && display_point != indent_start {
        indent_start
    } else {
        line_start
    }
}

pub fn line_end(
    map: &DisplaySnapshot,
    display_point: DisplayPoint,
    stop_at_soft_boundaries: bool,
) -> DisplayPoint {
    let soft_line_end = map.clip_point(
        DisplayPoint::new(display_point.row(), map.line_len(display_point.row())),
        Bias::Left,
    );
    if stop_at_soft_boundaries && display_point != soft_line_end {
        soft_line_end
    } else {
        map.next_line_boundary(display_point.to_point(map)).1
    }
}

pub fn previous_word_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
    find_preceding_boundary(map, point, |left, right| {
        (char_kind(left) != char_kind(right) && !right.is_whitespace()) || left == '\n'
    })
}

pub fn previous_subword_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
    find_preceding_boundary(map, point, |left, right| {
        let is_word_start = char_kind(left) != char_kind(right) && !right.is_whitespace();
        let is_subword_start =
            left == '_' && right != '_' || left.is_lowercase() && right.is_uppercase();
        is_word_start || is_subword_start || left == '\n'
    })
}

pub fn next_word_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
    find_boundary(map, point, |left, right| {
        (char_kind(left) != char_kind(right) && !left.is_whitespace()) || right == '\n'
    })
}

pub fn next_subword_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
    find_boundary(map, point, |left, right| {
        let is_word_end = (char_kind(left) != char_kind(right)) && !left.is_whitespace();
        let is_subword_end =
            left != '_' && right == '_' || left.is_lowercase() && right.is_uppercase();
        is_word_end || is_subword_end || right == '\n'
    })
}

/// Scans for a boundary from the start of each line preceding the given end point until a boundary
/// is found, indicated by the given predicate returning true. The predicate is called with the
/// character to the left and right of the candidate boundary location, and will be called with `\n`
/// characters indicating the start or end of a line. If the predicate returns true multiple times
/// on a line, the *rightmost* boundary is returned.
pub fn find_preceding_boundary(
    map: &DisplaySnapshot,
    end: DisplayPoint,
    mut is_boundary: impl FnMut(char, char) -> bool,
) -> DisplayPoint {
    let mut point = end;
    loop {
        *point.column_mut() = 0;
        if point.row() > 0 {
            if let Some(indent) = map.soft_wrap_indent(point.row() - 1) {
                *point.column_mut() = indent;
            }
        }

        let mut boundary = None;
        let mut prev_ch = if point.is_zero() { None } else { Some('\n') };
        for ch in map.chars_at(point) {
            if point >= end {
                break;
            }

            if let Some(prev_ch) = prev_ch {
                if is_boundary(prev_ch, ch) {
                    boundary = Some(point);
                }
            }

            if ch == '\n' {
                break;
            }

            prev_ch = Some(ch);
            *point.column_mut() += ch.len_utf8() as u32;
        }

        if let Some(boundary) = boundary {
            return boundary;
        } else if point.row() == 0 {
            return DisplayPoint::zero();
        } else {
            *point.row_mut() -= 1;
        }
    }
}

/// Scans for a boundary following the given start point until a boundary is found, indicated by the
/// given predicate returning true. The predicate is called with the character to the left and right
/// of the candidate boundary location, and will be called with `\n` characters indicating the start
/// or end of a line.
pub fn find_boundary(
    map: &DisplaySnapshot,
    mut point: DisplayPoint,
    mut is_boundary: impl FnMut(char, char) -> bool,
) -> DisplayPoint {
    let mut prev_ch = None;
    for ch in map.chars_at(point) {
        if let Some(prev_ch) = prev_ch {
            if is_boundary(prev_ch, ch) {
                break;
            }
        }

        if ch == '\n' {
            *point.row_mut() += 1;
            *point.column_mut() = 0;
        } else {
            *point.column_mut() += ch.len_utf8() as u32;
        }
        prev_ch = Some(ch);
    }
    map.clip_point(point, Bias::Right)
}

pub fn is_inside_word(map: &DisplaySnapshot, point: DisplayPoint) -> bool {
    let ix = map.clip_point(point, Bias::Left).to_offset(map, Bias::Left);
    let text = &map.buffer_snapshot;
    let next_char_kind = text.chars_at(ix).next().map(char_kind);
    let prev_char_kind = text.reversed_chars_at(ix).next().map(char_kind);
    prev_char_kind.zip(next_char_kind) == Some((CharKind::Word, CharKind::Word))
}

pub fn surrounding_word(map: &DisplaySnapshot, position: DisplayPoint) -> Range<DisplayPoint> {
    let position = map
        .clip_point(position, Bias::Left)
        .to_offset(map, Bias::Left);
    let (range, _) = map.buffer_snapshot.surrounding_word(position);
    let start = range
        .start
        .to_point(&map.buffer_snapshot)
        .to_display_point(map);
    let end = range
        .end
        .to_point(&map.buffer_snapshot)
        .to_display_point(map);
    start..end
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{test::marked_display_snapshot, Buffer, DisplayMap, ExcerptRange, MultiBuffer};
    use language::Point;
    use settings::Settings;

    #[gpui::test]
    fn test_previous_word_start(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                previous_word_start(&snapshot, display_points[1]),
                display_points[0]
            );
        }

        assert("\n|   |lorem", cx);
        assert("|\n|   lorem", cx);
        assert("    |lorem|", cx);
        assert("|    |lorem", cx);
        assert("    |lor|em", cx);
        assert("\nlorem\n|   |ipsum", cx);
        assert("\n\n|\n|", cx);
        assert("    |lorem  |ipsum", cx);
        assert("lorem|-|ipsum", cx);
        assert("lorem|-#$@|ipsum", cx);
        assert("|lorem_|ipsum", cx);
        assert(" |defγ|", cx);
        assert(" |bcΔ|", cx);
        assert(" ab|——|cd", cx);
    }

    #[gpui::test]
    fn test_previous_subword_start(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                previous_subword_start(&snapshot, display_points[1]),
                display_points[0]
            );
        }

        // Subword boundaries are respected
        assert("lorem_|ip|sum", cx);
        assert("lorem_|ipsum|", cx);
        assert("|lorem_|ipsum", cx);
        assert("lorem_|ipsum_|dolor", cx);
        assert("lorem|Ip|sum", cx);
        assert("lorem|Ipsum|", cx);

        // Word boundaries are still respected
        assert("\n|   |lorem", cx);
        assert("    |lorem|", cx);
        assert("    |lor|em", cx);
        assert("\nlorem\n|   |ipsum", cx);
        assert("\n\n|\n|", cx);
        assert("    |lorem  |ipsum", cx);
        assert("lorem|-|ipsum", cx);
        assert("lorem|-#$@|ipsum", cx);
        assert(" |defγ|", cx);
        assert(" bc|Δ|", cx);
        assert(" |bcδ|", cx);
        assert(" ab|——|cd", cx);
    }

    #[gpui::test]
    fn test_find_preceding_boundary(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(
            marked_text: &str,
            cx: &mut gpui::MutableAppContext,
            is_boundary: impl FnMut(char, char) -> bool,
        ) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                find_preceding_boundary(&snapshot, display_points[1], is_boundary),
                display_points[0]
            );
        }

        assert("abc|def\ngh\nij|k", cx, |left, right| {
            left == 'c' && right == 'd'
        });
        assert("abcdef\n|gh\nij|k", cx, |left, right| {
            left == '\n' && right == 'g'
        });
        let mut line_count = 0;
        assert("abcdef\n|gh\nij|k", cx, |left, _| {
            if left == '\n' {
                line_count += 1;
                line_count == 2
            } else {
                false
            }
        });
    }

    #[gpui::test]
    fn test_next_word_end(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                next_word_end(&snapshot, display_points[0]),
                display_points[1]
            );
        }

        assert("\n|   lorem|", cx);
        assert("    |lorem|", cx);
        assert("    lor|em|", cx);
        assert("    lorem|    |\nipsum\n", cx);
        assert("\n|\n|\n\n", cx);
        assert("lorem|    ipsum|   ", cx);
        assert("lorem|-|ipsum", cx);
        assert("lorem|#$@-|ipsum", cx);
        assert("lorem|_ipsum|", cx);
        assert(" |bcΔ|", cx);
        assert(" ab|——|cd", cx);
    }

    #[gpui::test]
    fn test_next_subword_end(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                next_subword_end(&snapshot, display_points[0]),
                display_points[1]
            );
        }

        // Subword boundaries are respected
        assert("lo|rem|_ipsum", cx);
        assert("|lorem|_ipsum", cx);
        assert("lorem|_ipsum|", cx);
        assert("lorem|_ipsum|_dolor", cx);
        assert("lo|rem|Ipsum", cx);
        assert("lorem|Ipsum|Dolor", cx);

        // Word boundaries are still respected
        assert("\n|   lorem|", cx);
        assert("    |lorem|", cx);
        assert("    lor|em|", cx);
        assert("    lorem|    |\nipsum\n", cx);
        assert("\n|\n|\n\n", cx);
        assert("lorem|    ipsum|   ", cx);
        assert("lorem|-|ipsum", cx);
        assert("lorem|#$@-|ipsum", cx);
        assert("lorem|_ipsum|", cx);
        assert(" |bc|Δ", cx);
        assert(" ab|——|cd", cx);
    }

    #[gpui::test]
    fn test_find_boundary(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(
            marked_text: &str,
            cx: &mut gpui::MutableAppContext,
            is_boundary: impl FnMut(char, char) -> bool,
        ) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                find_boundary(&snapshot, display_points[0], is_boundary),
                display_points[1]
            );
        }

        assert("abc|def\ngh\nij|k", cx, |left, right| {
            left == 'j' && right == 'k'
        });
        assert("ab|cdef\ngh\n|ijk", cx, |left, right| {
            left == '\n' && right == 'i'
        });
        let mut line_count = 0;
        assert("abc|def\ngh\n|ijk", cx, |left, _| {
            if left == '\n' {
                line_count += 1;
                line_count == 2
            } else {
                false
            }
        });
    }

    #[gpui::test]
    fn test_surrounding_word(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
            let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
            assert_eq!(
                surrounding_word(&snapshot, display_points[1]),
                display_points[0]..display_points[2]
            );
        }

        assert("||lorem|  ipsum", cx);
        assert("|lo|rem|  ipsum", cx);
        assert("|lorem||  ipsum", cx);
        assert("lorem| |  |ipsum", cx);
        assert("lorem\n|||\nipsum", cx);
        assert("lorem\n||ipsum|", cx);
        assert("lorem,|| |ipsum", cx);
        assert("|lorem||, ipsum", cx);
    }

    #[gpui::test]
    fn test_move_up_and_down_with_excerpts(cx: &mut gpui::MutableAppContext) {
        cx.set_global(Settings::test(cx));
        let family_id = cx.font_cache().load_family(&["Helvetica"]).unwrap();
        let font_id = cx
            .font_cache()
            .select_font(family_id, &Default::default())
            .unwrap();

        let buffer = cx.add_model(|cx| Buffer::new(0, "abc\ndefg\nhijkl\nmn", cx));
        let multibuffer = cx.add_model(|cx| {
            let mut multibuffer = MultiBuffer::new(0);
            multibuffer.push_excerpts(
                buffer.clone(),
                [
                    ExcerptRange {
                        context: Point::new(0, 0)..Point::new(1, 4),
                        primary: None,
                    },
                    ExcerptRange {
                        context: Point::new(2, 0)..Point::new(3, 2),
                        primary: None,
                    },
                ],
                cx,
            );
            multibuffer
        });
        let display_map =
            cx.add_model(|cx| DisplayMap::new(multibuffer, font_id, 14.0, None, 2, 2, cx));
        let snapshot = display_map.update(cx, |map, cx| map.snapshot(cx));

        assert_eq!(snapshot.text(), "\n\nabc\ndefg\n\n\nhijkl\nmn");

        // Can't move up into the first excerpt's header
        assert_eq!(
            up(
                &snapshot,
                DisplayPoint::new(2, 2),
                SelectionGoal::Column(2),
                false
            ),
            (DisplayPoint::new(2, 0), SelectionGoal::Column(0)),
        );
        assert_eq!(
            up(
                &snapshot,
                DisplayPoint::new(2, 0),
                SelectionGoal::None,
                false
            ),
            (DisplayPoint::new(2, 0), SelectionGoal::Column(0)),
        );

        // Move up and down within first excerpt
        assert_eq!(
            up(
                &snapshot,
                DisplayPoint::new(3, 4),
                SelectionGoal::Column(4),
                false
            ),
            (DisplayPoint::new(2, 3), SelectionGoal::Column(4)),
        );
        assert_eq!(
            down(
                &snapshot,
                DisplayPoint::new(2, 3),
                SelectionGoal::Column(4),
                false
            ),
            (DisplayPoint::new(3, 4), SelectionGoal::Column(4)),
        );

        // Move up and down across second excerpt's header
        assert_eq!(
            up(
                &snapshot,
                DisplayPoint::new(6, 5),
                SelectionGoal::Column(5),
                false
            ),
            (DisplayPoint::new(3, 4), SelectionGoal::Column(5)),
        );
        assert_eq!(
            down(
                &snapshot,
                DisplayPoint::new(3, 4),
                SelectionGoal::Column(5),
                false
            ),
            (DisplayPoint::new(6, 5), SelectionGoal::Column(5)),
        );

        // Can't move down off the end
        assert_eq!(
            down(
                &snapshot,
                DisplayPoint::new(7, 0),
                SelectionGoal::Column(0),
                false
            ),
            (DisplayPoint::new(7, 2), SelectionGoal::Column(2)),
        );
        assert_eq!(
            down(
                &snapshot,
                DisplayPoint::new(7, 2),
                SelectionGoal::Column(2),
                false
            ),
            (DisplayPoint::new(7, 2), SelectionGoal::Column(2)),
        );
    }
}