atuin 18.22.0

use std::ops::Range;

use atuin_client::settings::WordJumpMode;
use itertools::Itertools;

/// Like [`str::char_indices`], but only yields characters whose byte positions
/// are in `start..end`.
///
/// # Panics
///
/// This function does not panic.
fn chars_within(
    string: &str,
    range: Range<usize>,
) -> impl DoubleEndedIterator<Item = (usize, char)> + use<'_> {
    let start = string.ceil_char_boundary(range.start);
    // `ceil_char_boundary` clamps to `string.len()` if out of bounds. Use
    // `.max` to prevent `end` from being less than `start`.
    let end = string.ceil_char_boundary(range.end).max(start);
    string[start..end].char_indices().map(move |(i, c)| (start + i, c))
}

pub struct Cursor {
    source: String,
    index: usize,
}

impl From<String> for Cursor {
    fn from(source: String) -> Self {
        Self { source, index: 0 }
    }
}

pub struct WordJumper<'a> {
    word_chars: &'a str,
    word_jump_mode: WordJumpMode,
}

impl WordJumper<'_> {
    fn is_word_boundary(&self, c: char, next_c: char) -> bool {
        (c.is_whitespace() && !next_c.is_whitespace())
            || (!c.is_whitespace() && next_c.is_whitespace())
            || (self.word_chars.contains(c) && !self.word_chars.contains(next_c))
            || (!self.word_chars.contains(c) && self.word_chars.contains(next_c))
    }

    fn emacs_get_next_word_pos(&self, source: &str, index: usize) -> usize {
        let end = source.len().saturating_sub(1);
        let index = chars_within(source, index + 1..end)
            .find_map(|(i, c)| self.word_chars.contains(c).then_some(i))
            .unwrap_or(source.len());
        chars_within(source, index + 1..end)
            .find_map(|(i, c)| (!self.word_chars.contains(c)).then_some(i))
            .unwrap_or(source.len())
    }

    fn emacs_get_prev_word_pos(&self, source: &str, index: usize) -> usize {
        let index = chars_within(source, 1..index)
            .rev()
            .find_map(|(i, c)| self.word_chars.contains(c).then_some(i))
            .unwrap_or(0);
        chars_within(source, 1..index)
            .rev()
            .find_map(|(i, c)| (!self.word_chars.contains(c)).then_some(i + c.len_utf8()))
            .unwrap_or(0)
    }

    fn subl_get_next_word_pos(&self, source: &str, index: usize) -> usize {
        let Some(index) = chars_within(source, index..source.len())
            .tuple_windows()
            .find_map(|((i1, c1), (_, c2))| self.is_word_boundary(c1, c2).then_some(i1))
        else {
            return source.len();
        };
        chars_within(source, index + 1..source.len())
            .find_map(|(i, c)| (!c.is_whitespace()).then_some(i))
            .unwrap_or(source.len())
    }

    fn subl_get_prev_word_pos(&self, source: &str, index: usize) -> usize {
        let Some(index) = chars_within(source, 1..index)
            .rev()
            .find_map(|(i, c)| (!c.is_whitespace()).then_some(i))
        else {
            return 0;
        };
        chars_within(source, 0..index)
            .rev()
            .tuple_windows()
            .find_map(|((i1, c1), (_, c2))| self.is_word_boundary(c2, c1).then_some(i1))
            .unwrap_or(0)
    }

    fn get_next_word_pos(&self, source: &str, index: usize) -> usize {
        match self.word_jump_mode {
            WordJumpMode::Emacs => self.emacs_get_next_word_pos(source, index),
            WordJumpMode::Subl => self.subl_get_next_word_pos(source, index),
        }
    }

    fn get_prev_word_pos(&self, source: &str, index: usize) -> usize {
        match self.word_jump_mode {
            WordJumpMode::Emacs => self.emacs_get_prev_word_pos(source, index),
            WordJumpMode::Subl => self.subl_get_prev_word_pos(source, index),
        }
    }
}

impl Cursor {
    pub fn as_str(&self) -> &str {
        self.source.as_str()
    }

    pub fn into_inner(self) -> String {
        self.source
    }

    /// Returns the string before the cursor
    pub fn substring(&self) -> &str {
        &self.source[..self.index]
    }

    /// Returns the currently selected [`char`]
    pub fn char(&self) -> Option<char> {
        self.source[self.index..].chars().next()
    }

    pub fn right(&mut self) {
        if self.index < self.source.len() {
            loop {
                self.index += 1;
                if self.source.is_char_boundary(self.index) {
                    break;
                }
            }
        }
    }

    pub fn left(&mut self) -> bool {
        if self.index > 0 {
            loop {
                self.index -= 1;
                if self.source.is_char_boundary(self.index) {
                    break true;
                }
            }
        } else {
            false
        }
    }

    pub fn next_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {
        let word_jumper = WordJumper {
            word_chars,
            word_jump_mode,
        };
        self.index = word_jumper.get_next_word_pos(&self.source, self.index);
    }

    pub fn prev_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {
        let word_jumper = WordJumper {
            word_chars,
            word_jump_mode,
        };
        self.index = word_jumper.get_prev_word_pos(&self.source, self.index);
    }

    /// Move cursor to the end of the current/next word (vim `e` motion).
    ///
    /// If cursor is in the middle of a word, moves to the end of that word.
    /// If cursor is at the end of a word (or on whitespace), moves to the
    /// end of the next word.
    pub fn word_end(&mut self, word_chars: &str) {
        let len = self.source.len();
        if self.index >= len {
            return;
        }

        let chars: Vec<char> = self.source.chars().collect();
        let mut char_idx = self.source[..self.index].chars().count();

        if char_idx >= chars.len() {
            return;
        }

        let current = chars[char_idx];

        // Check if we're at a word boundary (end of current word or on whitespace)
        let at_word_boundary = current.is_whitespace() || char_idx + 1 >= chars.len() || {
            let next = chars[char_idx + 1];
            next.is_whitespace() || (word_chars.contains(current) != word_chars.contains(next))
        };

        // If at word boundary, advance past it and skip whitespace to find next word
        if at_word_boundary {
            char_idx += 1;
            while char_idx < chars.len() && chars[char_idx].is_whitespace() {
                char_idx += 1;
            }
        }

        // If we've gone past end, go to end of string
        if char_idx >= chars.len() {
            self.index = len;
            return;
        }

        // Find end of word: advance until next char is whitespace or different word type
        let in_word_chars = word_chars.contains(chars[char_idx]);
        while char_idx < chars.len() {
            let next_idx = char_idx + 1;
            if next_idx >= chars.len() {
                // At last char, move past it
                char_idx = next_idx;
                break;
            }
            let next_c = chars[next_idx];
            if next_c.is_whitespace() || (word_chars.contains(next_c) != in_word_chars) {
                // Next char is start of new word/whitespace, so current char is end
                char_idx = next_idx;
                break;
            }
            char_idx += 1;
        }

        // Convert char index back to byte index
        self.index = chars.iter().take(char_idx).map(|c| c.len_utf8()).sum();
    }

    pub fn insert(&mut self, c: char) {
        self.source.insert(self.index, c);
        self.index += c.len_utf8();
    }

    pub fn remove(&mut self) -> Option<char> {
        if self.index < self.source.len() {
            Some(self.source.remove(self.index))
        } else {
            None
        }
    }

    pub fn remove_next_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {
        let word_jumper = WordJumper {
            word_chars,
            word_jump_mode,
        };
        let next_index = word_jumper.get_next_word_pos(&self.source, self.index);
        self.source.replace_range(self.index..next_index, "");
    }

    pub fn remove_prev_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {
        let word_jumper = WordJumper {
            word_chars,
            word_jump_mode,
        };
        let next_index = word_jumper.get_prev_word_pos(&self.source, self.index);
        self.source.replace_range(next_index..self.index, "");
        self.index = next_index;
    }

    pub fn back(&mut self) -> Option<char> {
        if self.left() {
            self.remove()
        } else {
            None
        }
    }

    pub fn clear(&mut self) {
        self.source.clear();
        self.index = 0;
    }

    pub fn clear_to_start(&mut self) {
        self.source.replace_range(..self.index, "");
        self.index = 0;
    }

    pub fn clear_to_end(&mut self) {
        self.source.replace_range(self.index.., "");
        self.index = self.source.len();
    }

    pub fn end(&mut self) {
        self.index = self.source.len();
    }

    pub fn start(&mut self) {
        self.index = 0;
    }

    pub fn position(&self) -> usize {
        self.index
    }
}

#[cfg(test)]
mod cursor_tests {
    use rstest::rstest;

    use super::{Cursor, *};

    static EMACS_WORD_JUMPER: WordJumper = WordJumper {
        word_chars: "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789",
        word_jump_mode: WordJumpMode::Emacs,
    };

    static SUBL_WORD_JUMPER: WordJumper = WordJumper {
        word_chars: "./\\()\"'-:,.;<>~!@#$%^&*|+=[]{}`~?",
        word_jump_mode: WordJumpMode::Subl,
    };

    #[test]
    fn right() {
        // ö is 2 bytes
        let mut c = Cursor::from(String::from("öaöböcödöeöfö"));
        let indices = [0, 2, 3, 5, 6, 8, 9, 11, 12, 14, 15, 17, 18, 20, 20, 20, 20];
        for i in indices {
            assert_eq!(c.index, i);
            c.right();
        }
    }

    #[test]
    fn left() {
        // ö is 2 bytes
        let mut c = Cursor::from(String::from("öaöböcödöeöfö"));
        c.end();
        let indices = [20, 18, 17, 15, 14, 12, 11, 9, 8, 6, 5, 3, 2, 0, 0, 0, 0];
        for i in indices {
            assert_eq!(c.index, i);
            c.left();
        }
    }

    const JUMPER_SUBJECT: &str = "   aaa   ((()))bbb   ((()))   ";
    const EMACS_SUBJECT: &str = " 😀 test";
    const SUBL_SUBJECT: &str = " hi 😀 ((test";
    const TRAILING_MULTIBYTE_SUBJECT: &str = "aa bb😀";

    #[rstest]
    #[case::from_start_to_end_of_aaa(JUMPER_SUBJECT, 0, 6)]
    #[case::from_within_aaa_to_end_of_aaa(JUMPER_SUBJECT, 3, 6)]
    #[case::from_gap_skips_parens_to_end_of_bbb(JUMPER_SUBJECT, 7, 18)]
    #[case::from_after_bbb_to_end_of_string(JUMPER_SUBJECT, 19, 30)]
    #[case::non_ascii_from_start_to_test(EMACS_SUBJECT, 0, 10)]
    #[case::non_ascii_from_emoji_to_test(EMACS_SUBJECT, 1, 10)]
    #[case::trailing_multibyte(TRAILING_MULTIBYTE_SUBJECT, 3, 5)]
    #[case::empty_string("", 0, 0)]
    fn emacs_get_next_word_pos(#[case] subject: &str, #[case] from: usize, #[case] to: usize) {
        assert_eq!(EMACS_WORD_JUMPER.get_next_word_pos(subject, from), to);
    }

    #[rstest]
    #[case::from_end_of_string_to_start_of_bbb(JUMPER_SUBJECT, 30, 15)]
    #[case::from_trailing_space_to_start_of_bbb(JUMPER_SUBJECT, 29, 15)]
    #[case::from_start_of_bbb_to_start_of_aaa(JUMPER_SUBJECT, 15, 3)]
    #[case::from_start_of_aaa_to_string_start(JUMPER_SUBJECT, 3, 0)]
    #[case::non_ascii_from_test_to_start(EMACS_SUBJECT, 6, 0)]
    #[case::non_ascii_from_emoji_to_start(EMACS_SUBJECT, 1, 0)]
    #[case::trailing_multibyte(TRAILING_MULTIBYTE_SUBJECT, 5, 3)]
    #[case::empty_string("", 0, 0)]
    fn emacs_get_prev_word_pos(#[case] subject: &str, #[case] from: usize, #[case] to: usize) {
        assert_eq!(EMACS_WORD_JUMPER.get_prev_word_pos(subject, from), to);
    }

    #[rstest]
    #[case::from_string_start_to_start_of_aaa(JUMPER_SUBJECT, 0, 3)]
    #[case::from_within_leading_spaces_to_start_of_aaa(JUMPER_SUBJECT, 1, 3)]
    #[case::from_start_of_aaa_to_first_parens_block(JUMPER_SUBJECT, 3, 9)]
    #[case::from_first_parens_block_to_start_of_bbb(JUMPER_SUBJECT, 9, 15)]
    #[case::from_start_of_bbb_to_second_parens_block(JUMPER_SUBJECT, 15, 21)]
    #[case::from_second_parens_block_to_end_of_string(JUMPER_SUBJECT, 21, 30)]
    #[case::non_ascii_from_start_to_hi(SUBL_SUBJECT, 0, 1)]
    #[case::non_ascii_from_hi_to_emoji(SUBL_SUBJECT, 1, 4)]
    #[case::non_ascii_from_emoji_to_paren(SUBL_SUBJECT, 4, 9)]
    #[case::trailing_multibyte(TRAILING_MULTIBYTE_SUBJECT, 3, 9)]
    #[case::boundary_at_last_char("a.", 0, 1)]
    #[case::empty_string("", 0, 0)]
    fn subl_get_next_word_pos(#[case] subject: &str, #[case] from: usize, #[case] to: usize) {
        assert_eq!(SUBL_WORD_JUMPER.get_next_word_pos(subject, from), to);
    }

    #[rstest]
    #[case::from_end_of_string_to_second_parens_block(JUMPER_SUBJECT, 30, 21)]
    #[case::from_second_parens_block_to_start_of_bbb(JUMPER_SUBJECT, 21, 15)]
    #[case::from_start_of_bbb_to_first_parens_block(JUMPER_SUBJECT, 15, 9)]
    #[case::from_first_parens_block_to_start_of_aaa(JUMPER_SUBJECT, 9, 3)]
    #[case::from_start_of_aaa_to_string_start(JUMPER_SUBJECT, 3, 0)]
    #[case::non_ascii_from_hi_to_start(SUBL_SUBJECT, 1, 0)]
    #[case::non_ascii_from_paren_to_after_hi(SUBL_SUBJECT, 9, 3)]
    #[case::trailing_multibyte(TRAILING_MULTIBYTE_SUBJECT, 5, 3)]
    #[case::boundary_at_first_char(".aa", 3, 1)]
    #[case::empty_string("", 0, 0)]
    fn subl_get_prev_word_pos(#[case] subject: &str, #[case] from: usize, #[case] to: usize) {
        assert_eq!(SUBL_WORD_JUMPER.get_prev_word_pos(subject, from), to);
    }
    #[test]
    fn pop() {
        let mut s = String::from("öaöböcödöeöfö");
        let mut c = Cursor::from(s.clone());
        c.end();
        while !s.is_empty() {
            let c1 = s.pop();
            let c2 = c.back();
            assert_eq!(c1, c2);
            assert_eq!(s.as_str(), c.substring());
        }
        let c1 = s.pop();
        let c2 = c.back();
        assert_eq!(c1, c2);
    }

    #[test]
    fn back() {
        let mut c = Cursor::from(String::from("öaöböcödöeöfö"));
        // move to                                 ^
        for _ in 0..4 {
            c.right();
        }
        assert_eq!(c.substring(), "öaöb");
        assert_eq!(c.back(), Some('b'));
        assert_eq!(c.back(), Some('ö'));
        assert_eq!(c.back(), Some('a'));
        assert_eq!(c.back(), Some('ö'));
        assert_eq!(c.back(), None);
        assert_eq!(c.as_str(), "öcödöeöfö");
    }

    #[test]
    fn insert() {
        let mut c = Cursor::from(String::from("öaöböcödöeöfö"));
        // move to                                 ^
        for _ in 0..4 {
            c.right();
        }
        assert_eq!(c.substring(), "öaöb");
        c.insert('ö');
        c.insert('g');
        c.insert('ö');
        c.insert('h');
        assert_eq!(c.substring(), "öaöbögöh");
        assert_eq!(c.as_str(), "öaöbögöhöcödöeöfö");
    }
}